def lpar

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

import Base

144def lpar(d: Nat, +cnt: Nat, +mm1: Nat, +kn: Nat, +i: U32, +arow: U32, +sa: U32, +sb: U32, +bcol: U32, ra: Array<F32> & Array<F32>, rb: Array<F32> & Array<F32>) -> List<&2, F32>:
145  match d ra rb:
146    case 0n Tuple{a, a2} Tuple{b, b2}:
147      lblock(cnt, mm1, kn, i, arow, sa, sb, bcol, a, b)
148    case 1n++q Tuple{a, a2} Tuple{b, b2}:
149      x y = lpar(q, Nat.div(cnt, 2n), mm1, kn, i, arow, sa, sb, bcol, Array.clone(F32, a), Array.clone(F32, b)) lpar(q, Nat.sub(cnt, Nat.div(cnt, 2n)), mm1, kn, (i + U32.from_nat(Nat.div(cnt, 2n)) : U32), arow, sa, sb, bcol, Array.clone(F32, a2), Array.clone(F32, b2))
150      List.append(&2, F32, x, y)