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 1 year ago
xiongjyu xiongjyu requested a review from tjruwase tjruwase 1 year ago
xiongjyu xiongjyu requested a review from loadams loadams 1 year ago
xiongjyu xiongjyu closed this 1 year ago
xiongjyu xiongjyu reopened this 1 year ago
xiongjyu fixed: Modified the topkgating function and modified the test_moe fil…
f4a43cb8
xiongjyu xiongjyu force pushed from 62c70c7b to f4a43cb8 1 year 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 1 year ago
tohtana
tohtana approved these changes on 2025-06-06
tohtana Merge branch 'master' into master
7d2a3e11
tohtana tohtana merged 770967f5 into master 1 year ago

Login to write a write a comment.

Login via GitHub

Assignees
Labels
Milestone