def tsx8
import bendtorch-gpu@1.3.0.0/main.bend as BendtorchGpuimport Base
1489# tsx over 2^(l+3) cells, its last three levels in blocks of 8
1490def tsx8(l: Nat, +lo: U32, +h: U32, +m: F32, a: Array<F32>) -> TS:
1491 match l:
1492 case 0n: x8_0(lo, m, Array.get(F32, a, lo))
1493 case 1n++q: ts_then(tsx8(q, lo, U32.shr(h), m, a), +s => a2 => ts_plus(s, tsx8(q, (lo + h : U32), U32.shr(h), m, a2)))