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
178290249e
[Dataset] Add PATTERN dataset ( #5422 )
...
* add PATTERN dataset
* fix bug
* fix bugs
* fix issues
* refine according to dongyu's comments
---------
Co-authored-by: BuptTab <gyzhou2000@gmail.com >
Co-authored-by: rudongyu <ru_dongyu@outlook.com >
2023-03-08 14:08:12 +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
f00cd6efbd
[Utils] Edge and LINKX homophily measure ( #5382 )
...
* Update
* lint
* lint
* r prefix
* CI
* lint
* skip TF
* Update
* edge homophily
* linkx homophily
* format
* skip TF
* fix test
* update
* lint
* lint
* review
* lint
* update
* lint
* update
* CI
---------
Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal >
2023-03-03 16:58:20 +08:00
a53ecd229b
[Dataset] Add CLUSTER dataset ( #5389 )
...
* add CLUSTER dataset
* refine according to dongyu's comments
---------
Co-authored-by: rudongyu <ru_dongyu@outlook.com >
2023-03-01 16:36:12 +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
fcf5ad5f49
[Utils] Node homophily measure ( #5376 )
...
* Update
* lint
* lint
* r prefix
* CI
* lint
* skip TF
* Update
---------
Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal >
2023-02-24 14:26:02 +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
7ff041528c
[Refactor] Add default ffi namespce capi ( #5359 )
2023-02-22 21:33:30 +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
d9e550369f
[Misc] Reorg python test according to the guideline. ( #5307 )
...
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal >
2023-02-17 16:08:20 +08:00
Xin Yao