DeepSpeed
fixed: Modified the topkgating function and modified the test_moe file for testing
#7163
Merged

fixed: Modified the topkgating function and modified the test_moe file for testing #7163

tohtana merged 6 commits into deepspeedai:master from xiongjyu:master
xiongjyu
xiongjyu xiongjyu requested a review from tohtana tohtana 186 days ago
xiongjyu xiongjyu requested a review from tjruwase tjruwase 186 days ago
xiongjyu xiongjyu requested a review from loadams loadams 186 days ago
xiongjyu xiongjyu closed this 186 days ago
xiongjyu xiongjyu reopened this 186 days ago
xiongjyu fixed: Modified the topkgating function and modified the test_moe fil…
f4a43cb8
xiongjyu xiongjyu force pushed from 62c70c7b to f4a43cb8 186 days ago
tjruwase
xiongjyu style: fix formatting issue in test_moe.py
6123c748
loadams Merge branch 'master' into master
07a8649c
xiongjyu Merge branch 'master' into master
c030f49a
tjruwase Merge branch 'master' into master
23217229
tohtana tohtana assigned tohtana tohtana 109 days ago
tohtana
tohtana approved these changes on 2025-06-06
tohtana Merge branch 'master' into master
7d2a3e11
tohtana tohtana merged 770967f5 into master 108 days ago

Login to write a write a comment.

Login via GitHub

Assignees
Labels
Milestone