def gemm_new

bend-ml-tensor-gpu@0.1.5.1 · main.bend:268

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

268def gemm_new(+n: Nat, +m: Nat, +k: Nat, +arow: U32, +sa: U32, +sb: U32, +bcol: U32, +a: Array<F32>, +b: Array<F32>) -> ABC:
269  ABC{a, b, bang_new(e_of(False{}, n, m, k, arow, sa, sb, bcol), TA.cap_depth(tiles(n, m)), a, b, TA.cap_depth(Nat.mul(n, m)))}