code wiki / (root) / nx_zimage_attn_verify.nx

nx_zimage_attn_verify.nx

buildroot/runtime/nx_zimage_attn_verify.nx

7888 B205 linesdepth 5pulls 11 transitivereach 0 importersview sourcekind tooltopic zimage

docs dependencies structs consts functions

about

nx_zimage_attn_verify.nx -- SOVEREIGN Z-Image block piece #2: JointAttention, verified vs the oracle. z_image.hpp JointAttention: qkv = Linear(h); split Q|K|V each 30 heads x 128; per-head QK-RMSNorm(qn/kn); axial RoPE; per-head SDPA (softmax(q.k^T / sqrt(128)) v); out = Linear. Reads oracle dumps h_attn (input), qkv_w/out_w/qn/kn (weights), cosr/sinr (rope), and golden ao_out; computes sovereignly; verifies to tol. Reference dumps are oracle-only; the ORGAN is pure Nishi (nx_f32_*, raw syscalls, static ELF). license_tier: ORIGINAL

dependencies 7 imports · 0 importers

imports: nx_syscalls.nxnx_le.nxnx_f32.nxnx_f32_div.nxnx_f32_cvt.nxnx_f32_exp.nxnx_strconv.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

15	const K_MAGIC_3840: i64 = 3840
16	const K_MAGIC_11520: i64 = 11520
17	const K_MAGIC_100000: i64 = 100000
18	const K_MAGIC_2000: i64 = 2000

functions

20	func zav_load(name: u8, nl: i64, n_floats: i64) -> u8 called by 1: main calls 4: sys_mmap sys_openat_rd sys_read sys_close
48	func zav_qknorm_rope(qkv: i64, off: i64, w: u8, cosr: u8, sinr: u8, t: i64, eps: i64) -> i64 called by 1: main calls 7: nx_f32_add nx_f32_mul nx_f32_div nx_i32_to_f32 nx_f32_sqrt nx_le_read_u32+1
69	func zstage(n: i64) -> i64 called by 1: main calls 5: sys_openat_wr sys_mmap nx_strconv_format_i64 sys_write sys_close
75	func main() -> i64 calls 17: zav_load zstage nx_f32_div nx_i32_to_f32 nx_f32_sqrt sys_mmap+11