code wiki / (root) / nx_f32_llama_stack_test.nx

nx_f32_llama_stack_test.nx

buildroot/runtime/nx_f32_llama_stack_test.nx

4744 B120 linesdepth 10pulls 19 transitivereach 0 importersview sourcekind gate/prooftopic f32
docsdependenciesstructsconstsfunctions

about

nx_f32_llama_stack_test.nx -- smoke for nx_f32_llama_stack.nx. Build a 3-layer stack with all-zero weights and gamma=1. Forward through all 3 layers should preserve x_in bit-exact (residual passes through every layer, all projections are zero).

dependencies 6 imports · 0 importers

nx_syscalls.nx nx_tier.nx nx_f32.nx nx_f32_kv_cache.nx nx_f32_llama_block.nx nx_f32_llama_stack.nx nx_f32_llama_stack_test.nx

imports: nx_syscalls.nxnx_tier.nxnx_f32.nxnx_f32_kv_cache.nxnx_f32_llama_block.nxnx_f32_llama_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_f32_stk_verdict_is_vali sys_mmap nxa_die sys_write sys_exit nxa_lock_take nxa_lock_addr sys_write ↻ nxa_lock_give nxa_lock_addr ↻ nxa_report_overrun sys_write ↻ nxa_dump_printable sys_write ↻ nxa_dump_sizes sys_write ↻ make_zero_layer nx_f32_llama_layer_alloc sys_mmap ↻ sys_mmap ↻ nx_f32_kv_cache_alloc sys_mmap ↻ nx_f32_llama_stack_forward sys_mmap ↻ nx_f32_llama_block_forward sys_mmap ↻ nx_f32_rmsnorm nx_i32_to_f32 nx_f32_sqrt nx_f32_classify nx_f32_exp_field nx_f32_mant_field nx_f32_sign nx_f32_exp_field ↻ nx_f32_mant_field ↻ _f32_isqrt nx_f32_matmul nx_f32_kv_cache_get_seq_le nx_f32_rope_apply_vector

structs

none

consts

none

functions

14func make_zero_layer(hidden_dim: nx_int, q_dim: nx_int,
called by 1: main calls 2: nx_f32_llama_layer_allocsys_mmap
36func main() -> i64
calls 6: nx_f32_stk_verdict_is_validsys_mmapmake_zero_layernx_f32_kv_cache_allocnx_f32_llama_stack_forwardnx_f32_kv_cache_get_seq_len