~/jendhub

def blocks_acc

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

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

551def blocks_acc(nb: Nat, +bi: U32, +n: Nat, +m: Nat, +k: U32, +arow: U32, +sa: U32, +sb: U32, +bcol: U32, +a: Array<F32>, +b: Array<F32>, c: Array<F32>) -> Array<F32>:
552  match nb:
553    case 0n: c
554    case 1n++q: blocks_acc(q, (bi + 1 : U32), n, m, k, arow, sa, sb, bcol, a, b, bang(e_ofo(True{}, n, m, blk_len(bi, k), arow, sa, sb, bcol, ((bi * 1024 : U32) * sa : U32), ((bi * 1024 : U32) * sb : U32)), TA.cap_depth(tiles(n, m)), a, b, c))
555
556@unsafe

Uses (12)

Used by (1)