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