mirror of
https://github.com/deepseek-ai/DeepInfer.git
synced 2026-10-28 14:57:09 +08:00
Attention is head-sharded within lanes of 8 consecutive ranks of the CP group and replicated across lanes, as vllm-neuron#71 does: a 64-rank all-reduce is ~1.9 ms of latency against ~0.1 ms for 8 ranks on two adjacent devices. With 64 heads and 8 groups each rank holds exactly one wv group. Host tiny cp=8 with lanes of 2: golden tokens up to near ties. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com> |
||
|---|---|---|
| .. | ||
| mtp | ||
| v4_modules | ||
| vision_encoder | ||
| __init__.py | ||
| decoder_layer.py | ||
| deepseek_vl.py | ||
| deepseek.py | ||