def mm_precise2

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

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

301def mm_precise2(long: Bool, +n: Nat, +k: Nat, +m: Nat, +par: Nat, a: TA.Mat<n, k>, b: TA.Mat<k, m>) -> TA.MMul<n, k, m>:
302  match long a b:
303    case True{} TA.Mat{x} TA.Mat{y}: TA.mmul_fin(n, k, m, mp_go(U32.to_nat(U32.div((U32.from_nat(k) + 1023 : U32), 1024)), 0, par, n, m, U32.from_nat(k), TA.G{x, y, Array.new(F32, depth(Nat.mul(n, m)), 0.0)}))
304    case False{} a b: matmul(n, k, m, par, a, b)