Clamp capacity to num_tokens in MoE gating drop branches (complete #5353) - #8155

Merged

sfc-gh-truwase merged 3 commits into

deepspeedai:masterfrom

ebarkhordar:fix/topkgating-probs-capacity

Aug 2, 2026

Commits on Jul 18, 2026

Clamp capacity to num_tokens in MoE gating drop branches
ebarkhordar
committed
737050a

Commits on Jul 30, 2026

Bound the top-k selection instead of the dispatch capacity

ebarkhordar
and
Masahiro Tanaka
committed
0c691a5
Merge branch 'master' into fix/topkgating-probs-capacity
tohtana
authored
a1e01ac