• Zihao Ye's avatar
    [kernel] New SpMM & SDDMM kernel on CPU and CUDA (#1644) · 071cba1f
    Zihao Ye authored
    * udp
    
    * simplify
    
    * sddmm dot cpu
    
    * upd
    
    * format
    
    * upd
    
    * compatible with MJ's PR
    
    * lint
    
    * upd
    
    * upd
    
    * upd
    
    * python end
    
    * upd
    
    * upd
    
    * lint
    
    * lint
    
    * upd
    
    * upd
    
    * upd
    
    * upd
    
    * upd
    
    * lint
    
    * fix mxnet
    
    * upd
    
    * lint
    
    * use minjie's ptr
    
    * macro
    
    * upd
    
    * reorg
    
    * lint
    
    * fix corner cases
    
    * upd
    
    * enrich cpu docs
    
    * upd
    
    * upd
    
    * lint
    
    * lint
    
    * pylint
    
    * sx review
    
    * improve docstring
    
    * python doc
    
    * upd
    
    * restructure
    
    * lint
    
    * upd test
    
    * upd
    
    * pylint
    
    * fix corner cases and test
    071cba1f
kernel_decl.h 1.89 KB