~/jendhub

def fg_go

bend-ml-tensor-gpu@0.1.5.1 · main.bend:721

import bend-ml-tensor-gpu@0.1.5.1/main.bend as BendMlTensorGpu
import Base

721# pair j of the slice is the global pair q + j (q = i0 / 2), cells 2j - h and 2j - h + 1 of the output (h = i0 mod 2)
722def fg_go(k: Nat, +j: U32, +l: U32, +s: U32, +q: U32, +h: U32, +n: U32, o: Array<F32>) -> Array<F32>:
723  match k:
724    case 0n: o
725    case 1n+r: fg_go(r, (j + l : U32), l, s, q, h, n, fg_put(((j + j : U32) - h : U32), n, rng_gauss2(s, (q + j : U32)), o))

Uses (8)

Used by (1)