whn09 e5663a0a1d Neuron: attention TP over lanes of 8 consecutive ranks (DEEPINFER_NEURON_ATTN_TP) (#28)
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>
2026-10-24 09:16:01 +08:00
..
eplb initial commit 2026-10-20 12:09:42 +08:00
input_metadata initial commit 2026-10-20 12:09:42 +08:00
layers Neuron sampler: skip the top-p/top-k bisection when no row filters (#27) 2026-10-23 12:05:16 +08:00
models Neuron: attention TP over lanes of 8 consecutive ranks (DEEPINFER_NEURON_ATTN_TP) (#28) 2026-10-24 09:16:01 +08:00
parallel_utils initial commit 2026-10-20 12:09:42 +08:00
__init__.py initial commit 2026-10-20 12:09:42 +08:00
layer_utils.py initial commit 2026-10-20 12:09:42 +08:00
model_loader.py Neuron: CP group as a replicated-decode group; MoE all-reduce mode (#28) 2026-10-23 22:46:04 +08:00
model_metadata_utils.py Neuron: prompt chunks through the decode region graphs (issue #26) 2026-10-23 03:19:56 +08:00
model_output.py initial commit 2026-10-20 12:09:42 +08:00
sampling_inputs.py initial commit 2026-10-20 12:09:42 +08:00
utils.py Neuron: static decode graphs, static-shape MoE, trn2 e4m3 fp8 (issues #9, #12, #24) 2026-10-22 11:59:37 +08:00