nx_zimage_sdpa_verify.nx
buildroot/runtime/nx_zimage_sdpa_verify.nx
about
nx_zimage_sdpa_verify.nx -- SOVEREIGN Z-Image attention sub-piece 2b+2c: qk-norm+rope+SDPA, verified.
Small organ (no big linears): takes the oracle's dumped qkv [4,11520], applies per-head QK-RMSNorm(qn/kn)
+ axial RoPE(cosr/sinr) to Q,K, then per-head SDPA (softmax(q.k/sqrt128) v) -> ao [4,3840]; verifies vs
the oracle's dumped ao_pre (attention output BEFORE out-proj). Reference dumps oracle-only; organ = Nishi.
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
| 14 | const K_MAGIC_3840: i64 = 3840 |
| 15 | const K_MAGIC_11520: i64 = 11520 |
| 16 | const K_MAGIC_100000: i64 = 100000 |
functions
| 18 | func zsd_load(name: *u8, nl: i64, n_floats: i64) -> *u8 |
| 45 | func zsd_qknorm_rope(qkv: *i64, off: i64, w: *u8, cosr: *u8, sinr: *u8, t: i64, eps: i64) -> i64 |
| 65 | func main() -> i64 |