whn09 50483a06df Neuron: context-length buckets for dense compressed-KV scopes and indexer logits (#29)
DEEPINFER_NEURON_CTX_BUCKETS: the model runner sets the step's context limit
to the smallest bucket that holds every sequence, and the dense kv_ca_index /
indexer logits are sized to it instead of max_model_len (one more set of
attention graphs a bucket). Prefill cost scaled with the limit: a 128-token
chunk ~9 s at 16384 against ~5.3 s at 4352. Host tiny check: same tokens as
without buckets up to near ties.

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
2026-10-23 20:08:17 +08:00
..
__init__.py initial commit 2026-10-20 12:09:42 +08:00
base.py Neuron: full DeepSeek-V4.1-Flash load + decode region graphs at EP=64 (issues #21, #25) 2026-10-23 01:39:22 +08:00
cache_engine.py Neuron: separate state regions, chunk-wide indexer context (issue #26) 2026-10-23 03:46:56 +08:00
cuda_graph_runner.py initial commit 2026-10-20 12:09:42 +08:00
decode_perf_cache.py initial commit 2026-10-20 12:09:42 +08:00
dflash_cuda_graph_runner.py initial commit 2026-10-20 12:09:42 +08:00
model_runner.py Neuron: context-length buckets for dense compressed-KV scopes and indexer logits (#29) 2026-10-23 20:08:17 +08:00
neuron_graph_runner.py Neuron: compiled log-softmax + greedy argmax in the sampler; eager op census (#27) 2026-10-23 09:42:08 +08:00
runtime_config.py initial commit 2026-10-20 12:09:42 +08:00
worker.py initial commit 2026-10-20 12:09:42 +08:00
zos_status.py initial commit 2026-10-20 12:09:42 +08:00