• Israt Nisa's avatar
    [Kernel] Add heterograph support in CUDA kernels (SpMM, SDDMM) (#2925) · 1113f674
    Israt Nisa authored
    
    
    * Added heterograph support SpMM, SDDMM
    
    * bug fix cuda stream
    
    * add cudaStrm destroy and fix whitespace
    
    * Added heterograph support SpMM, SDDMM
    
    * bug fix cuda stream
    
    * add cudaStrm destroy and fix whitespace
    
    * changed max stream = 1
    
    * Fixed ctx
    
    * using default stream
    
    * Added heterograph support SpMM, SDDMM
    
    * bug fix cuda stream
    
    * add cudaStrm destroy and fix whitespace
    
    * changed max stream = 1
    
    * Fixed ctx
    
    * using default stream
    
    * fix bug in copy_rhs
    
    * changed by mistake
    
    * minor datatype change
    
    * added datatype check
    Co-authored-by: default avatarIsrat Nisa <nisisrat@amazon.com>
    1113f674
kernel_decl.h 4.93 KB