76bb54044e
[Misc] Black auto fix. ( #4682 )
...
Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal >
2022-10-08 14:16:12 +08:00
3d01340d23
[Sparse] Add elementwise operators for diagonal matrix ( #4644 )
...
* Add elementwise operators for diagonal matrix
* minor
* resolve ambiguity of ops
* lint check
* lint check
* lint check
Co-authored-by: Israt Nisa <nisisrat@amazon.com >
2022-09-29 13:08:50 +08:00
fa05ccb961
Add sddmm operator in mock_sparse library ( #4579 )
...
* sddmm init
* SDDMM with N-D nonzero values
* drop support for vector shaped non zero elements
* address comments
* skip cpu test
* skip GPU test too
Co-authored-by: Israt Nisa <nisisrat@amazon.com >
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2022-09-28 11:50:30 -04:00
977b1ba448
[Sparse] Migration of Unary, Softmax, Matmul Ops ( #4616 )
...
* Update
* lint
* Fix
* Update
* Update
* Update
* update
* Fix
* Update
* Fix
* Fix
* Fix
* CI
* Update
* Update
* Update
* update test
Co-authored-by: Ubuntu <ubuntu@ip-172-31-9-26.ap-northeast-1.compute.internal >
2022-09-27 19:34:46 +08:00
544a2cd192
[Sparse] Add reduction operators ( #4600 )
...
* [Sparse] Add reduction operators
* Fix linting problem
* Fix linting problem
* Fix linting problem
* Fix docstring
* Fix test cases
* Fix linting problem
* Skip reduction unittests
2022-09-26 16:06:05 +08:00
73b9f3a1da
[Sparse] Migrate DiagMatrix class ( #4567 )
...
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
Co-authored-by: Ubuntu <ubuntu@ip-172-31-9-26.ap-northeast-1.compute.internal >
2022-09-20 17:17:43 +08:00
cf03592743
Move mock version of dgl_sparse library to DGL main repo ( #4524 )
...
* init
* Add api doc for sparse library
* support op btwn matrices with differnt sparsity
* Fixed docstring
* addresses comments
* lint check
* change keyword format to fmt
Co-authored-by: Israt Nisa <nisisrat@amazon.com >
2022-09-14 12:08:20 -04:00
Hongzhi (Steve), Chen