def gdx_w
bendtorch-gpu@1.3.0.0 · main.bend:343
import bendtorch-gpu@1.3.0.0/main.bend as BendtorchGpu
import Base
343def gdx_w(k: Nat, +j: U32, +off: U32, +r: F32, +m: F32, dx: Array<F32>, rx: Array<F32> & F32, rd: Array<F32> & F32, rg: Array<F32> & F32) -> Array<F32>:
344 match k rx rd rg:
345 case 0n Tuple{x, +a} Tuple{dy, +b} Tuple{g, +c}:
346 +xh = (a * r : F32)
347 Array.set(F32, dx, (off + j : U32), (((b * c : F32) - (xh * m : F32) : F32) * r : F32))
348 case 1n+p Tuple{x, +a} Tuple{dy, +b} Tuple{g, +c}:
349 +xh = (a * r : F32)
350 gdx_w(p, (j + 1 : U32), off, r, m, Array.set(F32, dx, (off + j : U32), (((b * c : F32) - (xh * m : F32) : F32) * r : F32)), Array.get(F32, x, ((off + j : U32) + 1 : U32)), Array.get(F32, dy, ((off + j : U32) + 1 : U32)), Array.get(F32, g, (j + 1 : U32)))