"docs/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "b298484fd0303ba48f1fdceff3100d7068f62b79"
  • 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
kernel_decl.h 6.58 KB