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> |
||
|---|---|---|
| .. | ||
| _C | ||
| distributed | ||
| engine | ||
| entrypoints | ||
| hugging_face | ||
| model_executor | ||
| neuron_ops | ||
| npu_ops | ||
| structure_output | ||
| ve | ||
| worker | ||
| __init__.py | ||
| attrdict.py | ||
| config.py | ||
| environ.py | ||
| load_class.py | ||
| logger.py | ||
| model_config_loader.py | ||
| multimodal_tokens.py | ||
| platform.py | ||
| py.typed | ||
| sampling_params.py | ||
| sequence.py | ||
| shared_memory.py | ||
| testing.py | ||
| timer.py | ||
| utils.py | ||