~/jendhub

def gdg_loop

bendtorch-gpu@1.3.0.0 · main.bend:373

import bendtorch-gpu@1.3.0.0/main.bend as BendtorchGpu
import Base

373# acc + dy[i]·(x[i]·r[row]) down a column, i from j by d
374def gdg_loop(k: Nat, +i: U32, +d: U32, +row: U32, acc: F32, rx: Array<F32> & F32, rd: Array<F32> & F32, rr: Array<F32> & F32) -> F32:
375  match k rx rd rr:
376    case 0n Tuple{x, +a} Tuple{dy, +b} Tuple{r, +c}: (acc + (b * (a * c : F32) : F32) : F32)
377    case 1n+p Tuple{x, +a} Tuple{dy, +b} Tuple{r, +c}:
378      gdg_loop(p, (i + d : U32), d, (row + 1 : U32), (acc + (b * (a * c : F32) : F32) : F32), Array.get(F32, x, (i + d : U32)), Array.get(F32, dy, (i + d : U32)), Array.get(F32, r, (row + 1 : U32)))

Uses (10)

Used by (1)