mirror of
https://github.com/deepseek-ai/DeepInfer.git
synced 2026-10-28 23:07: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> |
||
|---|---|---|
| .. | ||
| eplb | ||
| input_metadata | ||
| layers | ||
| models | ||
| parallel_utils | ||
| __init__.py | ||
| layer_utils.py | ||
| model_loader.py | ||
| model_metadata_utils.py | ||
| model_output.py | ||
| sampling_inputs.py | ||
| utils.py | ||