DeepSpeed
Reduce redundant work in AutoEP token routing
#8209
Merged

Reduce redundant work in AutoEP token routing #8209

delock merged 4 commits into master from hongwei/moe_routing
hwchen2017
hwchen2017 save changes for reduce redundant work; avoid torch.bincount
e60b591f
hwchen2017 replace torch.bincount with scatter add
091d9ebb
hwchen2017 Fix format error
8f59e509
hwchen2017 hwchen2017 requested a review from delock delock 7 days ago
hwchen2017 hwchen2017 requested a review from sfc-gh-truwase sfc-gh-truwase 7 days ago
hwchen2017 hwchen2017 requested a review from tohtana tohtana 7 days ago
hwchen2017 hwchen2017 requested a review from loadams loadams 7 days ago
hwchen2017 hwchen2017 requested a review from tjruwase tjruwase 7 days ago
delock
delock approved these changes on 2026-08-04
delock Merge branch 'master' into hongwei/moe_routing
c54ba577
delock delock enabled auto-merge 7 days ago
delock
delock delock merged 58aab420 into master 7 days ago
delock delock deleted the hongwei/moe_routing branch 7 days ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone