def cpar

bend-ml-tensor-array@0.1.19.0 · main.bend:204

import Base

204def cpar(d: Nat, +cnt: Nat, +kn: Nat, +ia0: U32, +j0: U32, +sa: U32, +sb: U32, +bcol: U32, ra: Array<F32> & Array<F32>, rb: Array<F32> & Array<F32>) -> List<&2, F32>:
205  match d ra rb:
206    case 0n Tuple{a, a2} Tuple{b, b2}:
207      cblock(cnt, kn, ia0, j0, sa, sb, bcol, a, b)
208    case 1n++q Tuple{a, a2} Tuple{b, b2}:
209      x y = cpar(q, Nat.div(cnt, 2n), kn, ia0, j0, sa, sb, bcol, Array.clone(F32, a), Array.clone(F32, b)) cpar(q, Nat.sub(cnt, Nat.div(cnt, 2n)), kn, ia0, (j0 + U32.from_nat(Nat.div(cnt, 2n)) : U32), sa, sb, bcol, Array.clone(F32, a2), Array.clone(F32, b2))
210      List.append(&2, F32, x, y)