nx_nofloat_layernorm_gate.nx
buildroot/runtime/nx_nofloat_layernorm_gate.nx
about
nx_nofloat_layernorm_gate.nx -- the THIRD no-float transformer primitive: integer rsqrt -> LayerNorm.
Same from-the-hardware-rung-up SYSTEM as the exp/softmax one: range-reduce + Newton iterate in PURE
INTEGER -> bit-exact DETERMINISTIC. isqrt(v)=floor(sqrt v) via integer Newton (y=(x+v/x)/2 to a fixed
point); fx_rsqrt = 1/sqrt via isqrt; LayerNorm = (x-mean)/sqrt(var+eps), all Q16 integer. Completes the
trio every transformer layer needs: matmul (CAP-NOFLOAT-GEMM) + softmax (CAP-NOFLOAT-SOFTMAX) + norm.
Float rsqrt/norm is non-deterministic; this is reproducible forever. Verified self-consistently: isqrt
on perfect squares is EXACT; LayerNorm output has mean~0 + variance~1; fwd-sum == rev-sum (deterministic).
No hw writes (Rule 26). expect_exit: 0 license_tier: ORIGINAL
dependencies 2 imports · 0 importers
imports: nx_syscalls.nxnx_gate_verdict.nx
imported by: nobody (leaf or entry point)
call flow from main pre-order; caps 40 nodes / depth 6 declared; ↻ = already shown
structs
| none |
consts
| 12 | const Q16: i64 = 65536 |
functions
| 14 | func ln_puts(s: *u8) -> i64 { var n: i64=0; while s[n]!=(0 as u8){n=n+1} sys_write(1,s,n); return 0 } |
| 15 | func ln_num(v: i64) -> i64 { let b: *u8=sys_mmap(28); var m: i64=v; if m<0{m=0-m;sys_write(1,"-" as *u8,1)} let t: *u8=sys_mmap(28); var k: i64=0; if m==0{t[0]=48 as u8;k=1} while m>0{t[k]=(48+(m%10)) as u8;m=m/10;k=k+1} var i: i64=0; while i<k{b[i]=t[k-1-i];i=i+1} sys_write(1,b,k); return 0 } |
| 16 | func absd(a: i64, b: i64) -> i64 { if a>b { return a-b } return b-a } called by 1: main |
| 19 | func isqrt(v: i64) -> i64 |
| 29 | func fx_rsqrt(x: i64) -> i64 |
| 36 | func fx_layernorm(x: *i64, out: *i64, n: i64, rev: i64) -> i64 |
| 53 | func main() -> i64 |