DeepSpeed
add moe topk(k>2) gate support
#5881
Merged

Commits
  • Mingzhi/topk prob (#28)
    inkcherry committed 2 years ago
  • refine
    inkcherry committed 2 years ago
  • remove empty line
    inkcherry committed 2 years ago
  • Merge branch 'master' into topk_2
    tjruwase committed 2 years ago
  • fix format
    inkcherry committed 2 years ago
  • Merge branch 'master' into topk_2
    loadams committed 2 years ago
  • Merge branch 'master' into topk_2
    loadams committed 2 years ago
  • Merge branch 'master' into topk_2
    tohtana committed 2 years ago
Loading