def b3_q

bendtorch-gpu@1.3.0.0 · main.bend:869

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

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