def smb_fix

bendtorch@1.3.0.0 · main.bend:1416

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

1416# dy[j] = y[j] * (dy[j] - s), j = idx..idx + k; rd and ry hold dy[idx] and y[idx]
1417def smb_fix(k: Nat, +idx: U32, +s: F32, rd: Array<F32> & F32, ry: Array<F32> & F32) -> A2:
1418  match k rd ry:
1419    case 0n Tuple{d, +g} Tuple{y, +v}:
1420      A2{Array.set(F32, d, idx, (v * (g - s : F32) : F32)), y}
1421    case 1n+q Tuple{d, +g} Tuple{y, +v}:
1422      smb_fix(q, (idx + 1 : U32), s, Array.get(F32, Array.set(F32, d, idx, (v * (g - s : F32) : F32)), (idx + 1 : U32)), Array.get(F32, y, (idx + 1 : U32)))

Uses (12)

Used by (1)