def precise_c
bend-ml-tensor-gpu@0.1.5.1 · main.bend:557
import bend-ml-tensor-gpu@0.1.5.1/main.bend as BendMlTensorGpu
import Base
import bend-ml-tensor-array@0.1.19.0/main.bend as TA
557def precise_c(+n: Nat, +m: Nat, +k: Nat, +arow: U32, +sa: U32, +sb: U32, +bcol: U32, +a: Array<F32>, +b: Array<F32>) -> Array<F32>:
558 blocks_acc(Nat.sub(U32.to_nat(U32.div((U32.from_nat(k) + 1023 : U32), 1024)), 1n), 1, n, m, U32.from_nat(k), arow, sa, sb, bcol, a, b, bang_new(e_ofo(False{}, n, m, 1024n, arow, sa, sb, bcol, 0, 0), TA.cap_depth(tiles(n, m)), a, b, TA.cap_depth(Nat.mul(n, m))))
559
560@unsafe