• ranzhejiang's avatar
    New fused edge_softmax op (#3650) · bc8f8b0b
    ranzhejiang authored
    
    
    * [feature] edge softmax refact.
    
    * delete file
    
    * fix backward and cmake version
    
    * fix backward
    
    * format function
    
    * fix setting
    
    * refix
    
    * refix
    
    * refix
    
    * refix
    
    * refix
    
    * refix
    
    * refix
    
    * refix
    
    * refix
    
    * refix
    
    * refix
    
    * refix
    
    * add cuda kernel for backward and rename some function
    
    * add benchmark for edge_softmax
    
    * fix format
    
    * remove cuda_backwrd
    
    * fix code format and add comment for op on CPU
    
    * fix lint
    Co-authored-by: default avatarJinjing Zhou <VoVAllen@users.noreply.github.com>
    bc8f8b0b
spmm.h 22.4 KB