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)))

Uses (11)

Used by (1)