31 Commits

Author SHA1 Message Date
czkkkkkk
59bb5406b3 Revert "[Sparse] Fix Windows.h dependency issue (#6741)" (#6843)
Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
2023-12-28 15:24:49 +08:00
czkkkkkk
0e188b61a6 [Sparse] Fix Windows.h dependency issue (#6741) 2023-12-14 10:15:50 +08:00
xiangyuzhi
fdb4737c35 [Sparse] Add compact operator (#6352)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-24-117.ap-northeast-1.compute.internal>
2023-10-09 08:36:49 +08:00
xiangyuzhi
cbc34705c4 [Sparse] Compact C++ API (#6334) 2023-09-18 20:12:30 +08:00
xiangyuzhi
a23b490df4 [Sparse] Sparse sample implementation (#6303) 2023-09-12 16:42:24 +08:00
xiangyuzhi
dadce86a78 [Sparse] Add sparse matrix slicing operator implementation (#6208)
Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
2023-09-05 16:47:14 +08:00
czkkkkkk
7c465d20fc [Sparse] Support spspdiv (#5541)
Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
2023-04-14 16:46:51 +08:00
czkkkkkk
893098888e [Sparse] Support spspmul (#5543) 2023-04-13 09:07:49 +08:00
czkkkkkk
a03dec059e [Sparse] Support Diag sparse format in C++ (#5432)
* [Sparse] Support Diag sparse format in C++

* update

* Update
2023-03-09 12:49:47 +08:00
czkkkkkk
999c624582 [Sparse] Support converson to/from torch sparse tensor. (#5388)
* [Sparse] Support converson to/from torch sparse tensor.

* Update
2023-02-28 17:27:23 +08:00
czkkkkkk
5ffd2a022c [Sparse] Support column-wise softmax (#5377)
* [Sparse] Support column-wise softmax

* Update python/dgl/sparse/softmax.py

Co-authored-by: Mufei Li <mufeili1996@gmail.com>

---------

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2023-02-24 10:03:43 +08:00
czkkkkkk
73a508e156 [Sparse] Stack SparseMatrix COO row and column coordinates into one tensor. (#5314) 2023-02-23 09:53:19 +08:00
Mufei Li
9169fc3541 [Sparse] Polish sparse matrix creation functions (#5093)
* update

* Update

* Update

* Update

* Update

Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal>
2023-01-03 22:59:57 +08:00
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
14429df661 [Sparse] Support BSpMM and BSDDMM (#5079)
* [Sparse] Support BSpMM and BSDDMM

* Update SpMM and SDDMM error messages

* Use TORCH_CHECK

* Update error string
2022-12-28 17:14:12 +08:00
Minjie Wang
fa5ff2fceb [Sparse] Add coalesce and has_duplicate to SparseMatrix (#5071)
* add coalesce() and has_duplicate()

* Add duplicate check in spspmm

* black

* fix

* clang-format

* lintrunner

* minor fix

* fix
2022-12-28 13:10:10 +08:00
Mufei Li
0f1bcd9927 [Sparse] Softmax based on Reduction and BroadcastOp (#5067)
* Update

* update

* Update

* Update

* Update

* format

* Update

* lint

* Update

* lint

* CI

* CI

* Update

* address comments

Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal>
2022-12-27 14:50:15 +08:00
czkkkkkk
0159c3c11d [Sparse] Add SpSpMM for sparse-sparse and sparse-diag matrix multiplication. (#5050)
* [Sparse] Add SpSpMM

* Update matmul interface

* address comments

* fix test utils to generate only coalesced matrices

* fix linter

* fix ut

* fix

* rm print

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2022-12-24 15:31:09 +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
Mufei Li
11e6190579 [Sparse] Add Two Sparse Matrices with Different Sparsity, DiagMatrix.dense, Add SparseMatrix and DiagMatrix (#5044)
* Update

* undo wrong edit

* lint

* fix import

* Update

Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal>
2022-12-20 20:15:37 +08:00
Quan (Andy) Gan
f1ee3e311d [Sparse] Sparse matrix reduction C++ part (#5012)
* initial commit

* lint

* rename

* address changes

* oops

* fix?

* fix?
2022-12-20 12:37:12 +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
Mufei Li
e088acac00 [Sparse] Add val_like and Disable Setting Nonzero Values (#4972)
* Update

* Update

Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal>
2022-12-01 14:09:36 +08:00
czkkkkkk
60b02f2fc8 [Sparse] Add SparseMatrix transposition (#4940) 2022-11-29 16:54:36 +08:00
czkkkkkk
0a65559276 [Sparse] Support sparse format conversion (#4929)
* [Sparse] Support sparse format conversion

* Update

* Update
2022-11-27 17:01:18 +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
Mufei Li
ff090f698e [Sparse] Set Value and Dense (#4897)
* Update

* dense

* Update

* Update

* Update

* Update

* Update

Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal>
2022-11-16 19:05:56 +08:00
Hongzhi (Steve), Chen
11c866ab37 fix (#4841)
Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
2022-11-08 12:08:07 +08:00
czkkkkkk
7c059e86c7 [Sparse] Add sparse matrix C++ implementation (#4773)
* [Sparse] Add sparse matrix C++ implementation

* Add documentation

* Update

* Minor fix

* Move Python code to dgl/mock_sparse2

* Move headers to include

* lint

* Update

* Add dgl_sparse directory

* Move src code to dgl_sparse

* Add __init__.py in tests to avoid naming conflict

* Add dgl sparse so in Jenkinsfile

* Complete docstring & SparseMatrix basic op

* lint

* Disable win tests
2022-11-07 18:01:19 +08:00