src/array/cuda/spmm.cu · 2150fcaf230940c94196d590af3dea1450ae3939 · OpenDAS / dgl

"examples/vscode:/vscode.git/clone" did not exist on "48d357b8f6a0b0d5db697760dab966fdce1ed5c7"

[Feature] Added heterograph support to SDDMM_COO and clean up SpMM and SDDMM hetero kernels (#3449) · 2150fcaf

Israt Nisa authored Nov 17, 2021



* Added SDDMMCOO_hetero support

* removed redundant CUDA kernels

* added benchmark for regression test

* fix

* fixed bug for single src node type
Co-authored-by: Israt Nisa <nisisrat@amazon.com>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>

2150fcaf

spmm.cu 30.6 KB

Replace spmm.cu