def Mat.matmul_precise

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

import Base

1309# C = A · B (n x k) · (k x m), k summed in blocks of 1024 (above); opt-in
1310def Mat.matmul_precise(+n: Nat, +k: Nat, +m: Nat, +par: Nat, a: Mat<n, k>, b: Mat<k, m>) -> MMul<n, k, m>:
1311  mm_precise2(Nat.is_lt(1024n, k), n, k, m, par, a, b)