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
e250ba3d99
[Misc] Rename test_utils -> pytests_utils. ( #5438 )
...
* rename
* sort
* rename
---------
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal >
2023-03-09 09:42:29 +08:00
fe3d29ac4d
[Dataloading] Ignore edge types without reverse types in edge dataloader ( #5411 )
...
* Bugfix for reverse edge issue.
* Resolved copy/paste transcription error
* lint
* Added unit tests for graph with not all reverses.
* linting + added backend device test
---------
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com >
Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com >
2023-03-08 22:04:51 +08:00
ae4a5b7394
[Feature] Add state_dict, load_state_dict, param_groups to dgl.optim.SparseGradOptimizer ( #5311 )
...
* init update
* all get/set optm_state
* add unit tests
* add docstring
* fix for multiple embeddings
* move embedding methods to private
* fix lint
* fix unit tests
* resolve comments
* merge master
2023-03-08 16:46:34 +08:00
8d5d8962ad
[Refactor] Replace third_party/nccl with PyTorch's NCCL backend ( #4989 )
...
* expose GeneratePermutation
* add sparse_all_to_all_push
* add sparse_all_to_all_pull
* add unit test
* handle world_size=1
* remove python nccl wrapper
* remove the nccl dependency
* use pinned memory to speedup D2H copy
* fix lint
* resolve comments
* fix lint
* fix ut
* resolve comments
2023-03-08 12:59:10 +08:00
851d66fa65
[Bugfix] Fix duplicate worker_init_fn argument when provided in DataLoader ( #5420 )
...
* fix duplicate worker_init_fn
* lint
* lint again
* uugh
2023-03-06 23:41:53 +08:00
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
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
45153fc078
[Model] Implemented SubgraphX Explainer for Homogeneous graph ( #5315 )
...
* subgraphx commit
* nits
* newline eof added
* lint fix
* test script updated to use default values
* lint fix
* graphs that are used for test cases are updated to a small graph
* lint formatted
* test paramter adj to complete the test under 20s
* lint fixes
---------
Co-authored-by: kxm180046 <kxm180046@utdallas.edu >
2023-02-23 12:50:31 +08:00
74c9d27d16
[Misc] Auto-format tests. ( #5313 )
...
* [Misc] Auto-format tests.
* more
---------
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal >
2023-02-17 23:33:44 +08:00
86193c26f7
[Misc] Reorg python test (with backend) according to the guideline. ( #5312 )
...
* rename
* remove test transpose
* udpate
* rename
* cleanup
---------
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal >
2023-02-17 16:11:39 +08:00
czkkkkkk