nx_vit_encoder_layer_gate.nx
buildroot/runtime/nx_vit_encoder_layer_gate.nx
about
nx_vit_encoder_layer_gate.nx -- validate the ViT encoder block WITHOUT a PyTorch reference: (E1) attention core on
a hand-computable case (identity projections, 2 tokens, 1 head -> softmax([1/sqrt2,0]) mix), (E2) the EXACT
residual identity (all attn/MLP weights zero, LN gamma=1 beta=0 -> out == x, proving the residual wiring), (E3)
end-to-end finiteness with non-trivial weights. expect_exit: 0
dependencies 5 imports · 0 importers
imports: nx_syscalls.nxnx_f32.nxnx_f32_cvt.nxnx_f32_div.nxnx_vit_encoder_layer.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
| none |
functions
| 11 | func gp(s: *u8) -> i64 { var n: i64 = 0; while s[n] != (0 as u8) { n = n + 1 } return sys_write(1, s, n) } |
| 12 | func gn(v: i64) -> i64 { var m: i64=v; if m<0{gp("-" as *u8);m=0-m} let t:*u8=sys_mmap(24); 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; let o:*u8=sys_mmap(24); while i<k{o[i]=t[k-1-i];i=i+1} sys_write(1,o,k); return 0 } |
| 13 | func close(a: i64, b: i64, tol: i64) -> i64 { if nx_f32_lt(nx_f32_abs(nx_f32_sub(a, b)), tol) == 1 { return 1 } return 0 } |
| 14 | func zero(p: *i64, n: i64) -> i64 { var i: i64=0; while i<n { p[i]=0; i=i+1 } return 0 } called by 1: main |
| 15 | func ones(p: *i64, n: i64) -> i64 { var i: i64=0; while i<n { p[i]=nx_i32_to_f32(1); i=i+1 } return 0 } |
| 17 | func main(argc: i64, argv: *i64) -> i64 |