def blk_len
import bend-ml-tensor-gpu@0.1.5.1/main.bend as BendMlTensorGpuimport Base
546def blk_len(+bi: U32, +k: U32) -> Nat:
547 U32.to_nat(Bool.pick(U32, U32.is_lt((k - (bi * 1024 : U32) : U32), 1024), (k - (bi * 1024 : U32) : U32), 1024))
import bend-ml-tensor-gpu@0.1.5.1/main.bend as BendMlTensorGpuimport Base
546def blk_len(+bi: U32, +k: U32) -> Nat:
547 U32.to_nat(Bool.pick(U32, U32.is_lt((k - (bi * 1024 : U32) : U32), 1024), (k - (bi * 1024 : U32) : U32), 1024))