nx_f32_llama_block_v4.nx
buildroot/runtime/nx_f32_llama_block_v4.nx
about
dependencies 13 imports · 42 importers
diagram shows first 10 each side; +3 more imports, +32 more importers in the complete lists below.
imports: nx_syscalls.nxnx_tier.nxnx_f32.nxnx_f32_rmsnorm.nxnx_f32_matmul.nxnx_f32_activations.nxnx_f32_rope.nxnx_f32_attn_multi.nxnx_f32_kv_cache.nxnx_f32_attn_cached.nxnx_f32_lazy_weight.nxnx_thread_pool.nxnx_f32_llama_block.nx
imported by: nx_batched_gate.nxnx_batched_tput.nxnx_chat_warmcache.nxnx_coder_gen_f32.nxnx_companion_chat.nxnx_companion_voice.nxnx_f32_llama_block_v4_test.nxnx_f32_llama_layer_lazy_load.nxnx_f32_llama_layer_lazy_load_test.nxnx_f32_llama_stack_v4.nxnx_f32_llama_stack_v4_test.nxnx_f32_llama_v4b.nxnx_f32_llama_v4p.nxnx_f32_llm_bench_test.nxnx_f32_llm_chat_test.nxnx_f32_llm_cohere_test.nxnx_f32_llm_embed_diag_test.nxnx_f32_llm_live_gen_test.nxnx_f32_llm_live_load_only_test.nxnx_f32_llm_live_load_test.nxnx_f32_llm_probe.nxnx_f32_llm_run_v3.nxnx_f32_llm_run_v3_test.nxnx_f32_llm_serve.nxnx_f32_llm_v4.nxnx_f32_llm_v4_test.nxnx_f32_qwen_capital_gate.nxnx_llm_argmax_probe.nxnx_llm_batch_serve.nxnx_llm_diff_probe.nxnx_llm_forward_profile.nxnx_llm_france_probe.nxnx_llm_layerlens_probe.nxnx_llm_sched_gate.nxnx_lowrank_kv_real.nxnx_paged_fwd_gate.nxnx_qwen_native_gate.nxnx_qwen_wsl_timing_gate.nxnx_reason_paged_probe.nxnx_reason_probe.nxnx_reasoning_gate.nxnx_specdec_gate.nx
structs
| 44 | struct NxF32LlamaLayerLazy |
| 117 | struct NxActCtx |
consts
| 30 | const NX_MAGIC_2026: i64 = 2026 |
| 32 | const NX_BLK4_OK: nx_int = 0 |
| 33 | const NX_BLK4_ERR_BAD_DIM: nx_int = 1 |
| 34 | const NX_BLK4_ERR_NULL: nx_int = 2 |
| 35 | const NX_BLK4_ERR_CACHE: nx_int = 3 |
| 36 | const NX_BLK4_N_VERDICTS: nx_int = 4 |
| 59 | const NX_F32_LLAMA_LAYER_LAZY_BYTES: nx_int = 96 // 12 fields * 8 |
| 70 | const BP_ALLOC: i64 = 0 |
| 71 | const BP_RMSNORM: i64 = 1 |
| 72 | const BP_MATMUL: i64 = 2 |
| 73 | const BP_ROPE: i64 = 3 |
| 74 | const BP_ATTN: i64 = 4 |
| 75 | const BP_ACT: i64 = 5 |
| 76 | const BP_RESID: i64 = 6 |
| 127 | const NX_ACT_CTX_BYTES: i64 = 64 |
functions
| 38 | func nx_blk4_verdict_is_valid(v: nx_int) -> nx_int called by 1: main |
| 61 | func nx_f32_llama_layer_lazy_alloc() -> *NxF32LlamaLayerLazy |
| 81 | func nx_blk4_prof_enable(on: i64) -> i64 |
| 86 | func nx_blk4_prof_reset() -> i64 called by 1: main |
| 94 | func nx_blk4_prof_read(b: i64) -> i64 called by 1: main |
| 99 | func bp_t() -> i64 |
| 103 | func bp_add(b: i64, t0: i64) -> i64 |
| 130 | func _swiglu_task(ctx_i: i64) -> i64 |
| 143 | func nx_blk4_swiglu_pool(gate: *i64, up: *i64, out: *i64, count: i64) -> i64 |
| 172 | func _rope_task(ctx_i: i64) -> i64 calls 1: nx_f32_rope_apply_vector_neox |
| 190 | func nx_blk4_rope_heads_pool(base: i64, n_heads: i64, head_dim: i64, |
| 220 | func nx_f32_llama_block_forward_v4( called by 4: mainnx_f32_llama_stack_forward_v4mainmain calls 12: bp_tsys_mmapbp_addnx_f32_rmsnormnx_f32_lazy_matmulnx_f32_add+6 |