mirror of
https://github.com/deepseek-ai/DeepInfer.git
synced 2026-10-28 23:07:09 +08:00
- The state buffer's column regions get their own allocations on Neuron: a layer's uint8 SWA cache and fp32 compressor state are both region graph inputs, and aot_autograd compares offsets of differently typed views of one storage in their own element units (false overlap -> 'input mutations on views with different dtypes'). - paged_mqa_logits chunk path: gather the keys the chunk's last token sees (lens.max()), not the first token's. - Debug: DEEPINFER_NEURON_ALIAS_DEBUG prints aliased graph inputs; a cProfile of decode step 8 under DEEPINFER_NEURON_REGION_TIMING. Tiny V4.1 on a NeuronCore, bucket 128: short 8/9 (= eager), long 715 tokens in 6 chunks 9/9 (eager 8/9) in 26 s instead of 978 s of eager prefill. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com> |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| base.py | ||
| cache_engine.py | ||
| cuda_graph_runner.py | ||
| decode_perf_cache.py | ||
| dflash_cuda_graph_runner.py | ||
| model_runner.py | ||
| neuron_graph_runner.py | ||
| runtime_config.py | ||
| worker.py | ||
| zos_status.py | ||