1. 15 3月, 2022 1 次提交
    • R
      add number count op (#39224) · 9bdee437
      Roc 提交于
      * add expert count op
      
      add ut for expert_count
      
      * update UT only for cuda
      
      * fix for rocm
      
      * update ut
      
      * add moe module
      
      * add expert count op
      
      add ut for expert_count
      
      * update UT only for cuda
      
      * update ut
      
      * add moe module
      
      * make expert count private
      
      * rename expert count op
      Co-authored-by: Nhlygit66666 <2570058140@qq.com>
      9bdee437