def rpar

bend-ml-tensor-fork@0.1.6.0 · main.bend:179

import Base
import bend-ml-tensor-array@0.1.19.0/main.bend as TA

179def rpar(+ac: Bool, d: Nat, +cnt: Nat, +m: Nat, +tk: Nat, +tku: U32, +rot: Nat, +i0: U32, +mu: U32, pa: Array<TA.T8> & Array<TA.T8>, pb: Array<TA.T8> & Array<TA.T8>, pc: Array<F32> & Array<F32>) -> Array<F32>:
180  match d pa pb pc:
181    case 0n Tuple{a, a2} Tuple{b, b2} Tuple{c, c2}:
182      rblock(ac, cnt, m, tk, tku, rot, i0, mu, a, b, c)
183    case 1n++q Tuple{a, a2} Tuple{b, b2} Tuple{c, c2}:
184      l r = rpar(ac, q, Nat.div(cnt, 2n), m, tk, tku, rot, i0, mu, Array.fork(TA.T8, a), Array.fork(TA.T8, b), Array.fork(F32, c)) rpar(ac, q, Nat.sub(cnt, Nat.div(cnt, 2n)), m, tk, tku, rot, (i0 + U32.from_nat(Nat.div(cnt, 2n)) : U32), mu, Array.fork(TA.T8, a2), Array.fork(TA.T8, b2), Array.fork(F32, c2))
185      Array.join(F32, l, r)