def bal

bendtorch@1.3.0.0 · spec.bend:166

import Base

166# the 2^p cells from cell i summed as a balanced tree of pairs: how a GPU leaf sums its block (the device's MSE and
167# cross-entropy block sums)
168def bal(+p: Nat, +i: Nat) -> Sum:
169  match p:
170    case 0n:
171      SCell{i}
172    case 1n+q:
173      SAdd{bal(q, i), bal(q, Nat.add(i, pw(q)))}