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 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 nx_i32_to_f32 ↻ nx_f32_neg nx_f32_div nx_f32_classify ↻ nx_f32_sign ↻ nx_f32_mant_field ↻ nx_f32_exp_field ↻ nx_f32_mul nx_f32_classify ↻ nx_f32_sign ↻ nx_f32_mant_field ↻ nx_f32_exp_field ↻ nx_f32_exp nx_f32_classify ↻

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