def b3_v

bendtorch-gpu@1.3.0.0 · main.bend:877

import Base
import bendtorch@1.3.0.0/main.bend as T

877def b3_v(+e: U32, q: Array<F32>, k: Array<F32>, g: Array<F32>, dq: Array<F32>, dk: Array<F32>, v: Array<F32>, r: T.D2) -> S8:
878  match r:
879    case T.D2{p, d, +s}: S8{q, k, d, g, p, dq, dk, Array.set(F32, v, e, s)}