• 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
utils.py 1.9 KB