def gcols

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

import Base

91def gcols(+ac: Bool, mleft: Nat, +kn: Nat, +ia0: U32, +ib0: U32, +sa: U32, +sb: U32, +bcol: U32, +cidx: U32, c: Array<F32>, r: R) -> G:
92  match mleft r:
93    case 0n R{a, b, +x}:
94      G{a, b, cwr(ac, cidx, x, c)}
95    case 1n+p R{a, b, +x}:
96      gcols(ac, p, kn, ia0, (ib0 + bcol : U32), sa, sb, bcol, (cidx + 1 : U32), cwr(ac, cidx, x, c), dotg(kn, ia0, (ib0 + bcol : U32), sa, sb, 0.0, Array.get(F32, a, ia0), Array.get(F32, b, (ib0 + bcol : U32))))