nx_f32_kv_cache.nx
buildroot/runtime/nx_f32_kv_cache.nx
about
dependencies 2 imports · 56 importers
diagram shows first 10 each side; +0 more imports, +46 more importers in the complete lists below.
imports: nx_syscalls.nxnx_tier.nx
imported by: nx_batched_gate.nxnx_batched_tput.nxnx_chat_warmcache.nxnx_coder_gen_f32.nxnx_companion_chat.nxnx_companion_voice.nxnx_f32_attn_cached.nxnx_f32_attn_cached_test.nxnx_f32_kv_cache_test.nxnx_f32_llama_block.nxnx_f32_llama_block_test.nxnx_f32_llama_block_v4.nxnx_f32_llama_block_v4_test.nxnx_f32_llama_stack.nxnx_f32_llama_stack_test.nxnx_f32_llama_stack_v4.nxnx_f32_llama_stack_v4_test.nxnx_f32_llm.nxnx_f32_llm_bench_test.nxnx_f32_llm_chat_test.nxnx_f32_llm_cohere_test.nxnx_f32_llm_e2e_test.nxnx_f32_llm_live_gen_test.nxnx_f32_llm_live_load_only_test.nxnx_f32_llm_live_load_test.nxnx_f32_llm_load_test.nxnx_f32_llm_probe.nxnx_f32_llm_run.nxnx_f32_llm_run_test.nxnx_f32_llm_run_v2.nxnx_f32_llm_run_v2_test.nxnx_f32_llm_run_v3.nxnx_f32_llm_run_v3_test.nxnx_f32_llm_serve.nxnx_f32_llm_test.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_paged_kv_gate.nxnx_qwen_native_gate.nxnx_qwen_wsl_timing_gate.nxnx_reason_paged_probe.nxnx_reason_probe.nxnx_reasoning.nxnx_reasoning_gate.nxnx_specdec.nxnx_specdec_gate.nx
structs
| 48 | struct NxF32KVCache |
consts
| 36 | const NX_F32_KVC_OK: nx_int = 0 |
| 37 | const NX_F32_KVC_ERR_BAD_DIM: nx_int = 1 |
| 38 | const NX_F32_KVC_ERR_OVERFLOW: nx_int = 2 |
| 39 | const NX_F32_KVC_ERR_BAD_LAYER: nx_int = 3 |
| 40 | const NX_F32_KVC_N_VERDICTS: nx_int = 4 |
| 58 | const NX_F32_KVC_STRUCT_BYTES: nx_int = 56 // 7 fields * 8 |
functions
| 42 | func nx_f32_kvc_verdict_is_valid(v: nx_int) -> nx_int called by 1: main |
| 62 | func nx_f32_kv_cache_alloc(n_layers: nx_int, n_kv_heads: nx_int, |
| 86 | func nx_f32_kv_cache_reset(c: *NxF32KVCache) -> nx_int |
| 92 | func nx_f32_kv_cache_get_seq_len(c: *NxF32KVCache) -> nx_int |
| 99 | func nx_f32_kv_cache_get_K_layer(c: *NxF32KVCache, layer: nx_int) -> *i64 |
| 108 | func nx_f32_kv_cache_get_V_layer(c: *NxF32KVCache, layer: nx_int) -> *i64 |
| 122 | func nx_f32_kv_cache_append_layer(c: *NxF32KVCache, layer: nx_int, called by 2: nx_f32_attn_with_cachemain calls 2: nx_f32_kv_cache_get_K_layernx_f32_kv_cache_get_V_layer |
| 154 | func nx_f32_kv_cache_advance(c: *NxF32KVCache, n_new: nx_int) -> nx_int |