mirrors / dmlc--dgl

文件历史

3,215 提交 34 分支 42 Git 标签
fix-gpu-sampling-doc
提交图

6 次代码提交

作者 SHA1 备注 提交日期
Mufei Li 811e35a6ef [Sparse] Change all CHECK to TORCH_CHECK (#5082) 
* Update

* CI

* Update dgl_sparse/src/utils.h

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>

Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
 2022-12-28 23:26:06 +08:00
czkkkkkk 1f2fcae3e2 [Sparse] Support SpMM and SDDMM (#5032) 
* [Sparse] Support SpMM and SDDMM

* Add SDDMMAutoGrad

* Rename SpMMImpl and SDDMMImpl and move them to dgl_sparse/src/matmul.h

* Update comments

* Update

* Minor fix
 2022-12-21 10:29:23 +08:00
czkkkkkk d02e560e07 Revert "[Sparse] Add SpMM and SDDMM." (#5014) 
* Revert "[Sparse] Add SpMM and SDDMM. (#4999)"

This reverts commit 15365d7855.

* lint
 2022-12-12 19:37:08 +08:00
czkkkkkk 15365d7855 [Sparse] Add SpMM and SDDMM. (#4999) 
* [Sparse] Add SpMM and SDDMM

* Update

* Add CSR and CSC SpMM tests
 2022-12-12 15:37:09 +08:00
czkkkkkk 0cb5f0fdc0 [Sparse] Be compatible with Pytorch 1.13 and later version (#4935) 2022-11-23 15:36:18 -08:00
czkkkkkk 06438d7033 [Sparse] Link to DGL (#4877) 2022-11-17 10:14:08 +08:00