def ms_if

bendtorch-gpu@1.3.0.0 · main.bend:1191

import Base
import bend-ml-tensor-gpu@0.1.5.0/main.bend as TG

1191def ms_if(ok: Bool, +t: U32, +p: Nat, +bs: U32, +kk: F32, tg: Array<F32>, y: Array<F32>, ps: Array<F32>) -> TG.O2:
1192  match ok:
1193    case True{}: ms_put(t, ps, ms_blk(p, (t * bs : U32), U32.shr(bs), kk, y, tg))
1194    case False{}: TG.O2{y, ps}