code wiki / (root) / nx_transformer_stack_test.nx

nx_transformer_stack_test.nx

buildroot/runtime/nx_transformer_stack_test.nx

8025 B207 linesdepth 9pulls 23 transitivereach 0 importersview sourcekind gate/prooftopic transformer
docsdependenciesstructsconstsfunctions

about

nx_transformer_stack_test.nx -- smoke for nx_transformer_stack.nx. Builds a 1-layer synthetic Llama-shape GGUF with ALL-ZERO weights (so the forward pass becomes an identity), feeds x_in through the stack, and verifies x_in is unchanged. Test spec: hidden=2, head_dim=2, ffn_dim=4, n_tokens=1. Weights = 0 means: - RMSNorm(x, 0) = 0 (gamma=0 zeros out the norm output) - matmul(0, W) = 0 (zero input/weight = zero output) - attention(0, 0, 0) = 0 (softmax of all-zero scores * zero V) - residual: x + 0 = x (UNCHANGED) - FFN: same identity-via-zero pattern So output x == input x after the entire forward pass.

dependencies 7 imports · 0 importers

nx_syscalls.nx nx_tier.nx nx_le.nx nx_tensor.nx nx_gguf.nx nx_gguf_load.nx nx_transformer_stack.nx nx_transformer_stack_test.nx

imports: nx_syscalls.nxnx_tier.nxnx_le.nxnx_tensor.nxnx_gguf.nxnx_gguf_load.nxnx_transformer_stack.nx

imported by: nobody (leaf or entry point)

call flow from main pre-order; caps 40 nodes / depth 6 declared; ↻ = already shown

main nx_ts_verdict_is_valid sys_mmap nx_le_write_u64 nx_le_write_u32 _mk_blk_name _write_ti nx_le_write_u64 ↻ nx_le_write_u32 ↻ nx_gguf_parse nx_le_read_u32 nx_le_read_u64 nx_le_read_u32 ↻ _gguf_skip_value nx_le_read_u64 ↻ nx_le_read_u32 ↻ _gguf_skip_value ↻ sys_mmap ↻ nx_t_alloc nx_dt_is_valid nx_dt_element_bytes sys_mmap ↻ nx_t_compute_strides_rowma nx_transformer_stack_forwa _ts_alloc_2d sys_mmap ↻ nx_t_alloc ↻ sys_mmap ↻ nx_gguf_load_block_weights sys_mmap ↻ _gbl_load_blk_tensor sys_mmap ↻ _gbl_fmt_blk_name nx_strconv_format_i64 nx_gguf_load_tensor nx_gguf_find_tensor nx_gguf_tensor_at nx_gguf_name_equals nx_gguf_tensor_at ↻ nx_gguf_tensor_n_values

structs

none

consts

none

functions

25func _write_ti(buf: *u8, off: i64,
called by 1: main calls 2: nx_le_write_u64nx_le_write_u32
44func _mk_blk_name(suffix: *u8, suffix_len: nx_int, out: *u8) -> nx_int
called by 1: main
56func main() -> i64