文件历史

48 次代码提交

作者 SHA1 备注 提交日期
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 0bf51a1671 [Sparse] Fix the NOMINMAX problem on Windows. (#6818) 2023-12-25 15:00:37 +08:00
czkkkkkk 0e188b61a6 [Sparse] Fix Windows.h dependency issue (#6741) 2023-12-14 10:15:50 +08:00
xiangyuzhi 2ebfa6ea5a [Sparse]Fix device check (#6409)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-24-117.ap-northeast-1.compute.internal>
2023-10-11 09:04:29 +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
czkkkkkk 33ba46b7df [Sparse] Add CUDA compilation support. (#6338) 2023-09-25 16:10:25 +08:00
xiangyuzhi cbc34705c4 [Sparse] Compact C++ API (#6334) 2023-09-18 20:12:30 +08:00
xiangyuzhi da1d5de2f1 [Sparse] Add cpu compilation support (#6310) 2023-09-12 16:43: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
Hugo MacDermott-Opeskin 4a42027d4a [Build] Add CMake changes from conda-forge build (#6189)
Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
2023-09-01 11:21:03 +08:00
Chang Liu 4b585ba068 [Build] Build sparse lib with c++17 (#5934) 2023-07-03 10:04:01 -07: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
Ilia Taraban e8579d8be8 [Misc] Add MKL library path for standalone installation (#5510)
Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
2023-04-04 14:42:12 +08:00
czkkkkkk 4463b3d6a5 [Sparse] Support strided tensor when calling old DGL APIs (#5506) 2023-03-31 23:18:55 +08:00
czkkkkkk 866c70dace [Sparse] Fix bugs in Diag format conversions (#5465) 2023-03-21 09:24:53 +08:00
czkkkkkk 92e383c3b8 [Sparse] Remove the Python DiagMatrix class (#5444)
* [Sparse] Remove the Python DiagMatrix class

* Update

* Update

* update

* Update
2023-03-15 18:17:52 +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
Quan (Andy) Gan 5f1babf0bf fix build for multiple pytorch versions (#5289) 2023-02-15 15:22:50 +08:00
Quan (Andy) Gan f62669b05c enable sparse on windows and mac (#5277)
* enable sparse on windows and mac

* that was stupid

* let's see what's going on..

* [Sparse] Fix the import error on Mac OS.

When using template functions that are defined in source files from DGL,
the loader of MacOS somehow cannot find their definitions. This fix simply
avoids depending on template functions from DGL headers.

With this fix, the sparse tests all pass on the MAC environment.

* ok this is the problem

* make errors clearer

* uh

* test

* Update __init__.py

* disabling ddp on windows

---------

Co-authored-by: czkkkkkk <zekucai@gmail.com>
2023-02-13 13:16:17 +08:00
Hongzhi (Steve), Chen 8b47bad5ba [Misc] Remove unnecessary clean up before build to accelerate incremental build. (#5267)
* remove unnecessary clean up.

* more

---------

Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
2023-02-03 15:11:16 +08:00
Quan (Andy) Gan f145097f9a [Release] Conda/CMake changes for 1.0 release (#5186)
* update conda branch

* fix conda

* update conda

* update

* update windows build script

* copy static library to build dir

* try again

* fix

* lint
2023-01-19 22:28:04 +08:00
czkkkkkk 21c4c29a36 [Sparse] Reduce peak memory of SpSpAdd. (#5143)
* [Sparse] Reduce peak memory usage of SpSpAdd

* Update

* Update
2023-01-12 11:53:04 +08:00
Quan (Andy) Gan 9a40d2080e [Sparse] Change CMakeFiles to build DGL Sparse for multiple PyTorch versions (#5106)
* initial commit

* Update build.sh
2023-01-06 14:14:20 +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
czkkkkkk e65fc9ff78 [Sparse] Avoid using macros in DGL headers (#5087) 2022-12-29 17:35:29 +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