• R
    [MoE]Assign pos op (#40580) · 305f32d1
    Roc 提交于
    * # This is a combination of 10 commits.
    # The first commit's message is:
    add expert count op
    
    add ut for expert_count
    
    # This is the 2nd commit message:
    
    update UT only for cuda
    
    # This is the 3rd commit message:
    
    fix for rocm
    
    # This is the 4th commit message:
    
    update ut
    
    # This is the 5th commit message:
    
    add moe module
    
    # This is the 6th commit message:
    
    add expert count op
    
    add ut for expert_count
    
    # This is the 7th commit message:
    
    update UT only for cuda
    
    # This is the 8th commit message:
    
    update ut
    
    # This is the 9th commit message:
    
    add moe module
    
    # This is the 10th commit message:
    
    make expert count private
    
    * add assign pos op
    
    * fix upper num name
    
    * add api _assign pos
    
    * add ut for assign pos op
    
    * update date
    
    * fix for win
    
    * update for test (timeout)
    
    * fix ut
    
    * update
    
    * fix ut for number count
    Co-authored-by: Nhlygit66666 <2570058140@qq.com>
    305f32d1
utils.py 3.5 KB