191681d851
[DistDGL] fix device mismatch when calling all_to_all with gloo backend ( #7409 )
2024-05-17 08:08:24 +08:00
5dfaf99e3c
[Performance] Add NCCL support ( #5929 )
...
Co-authored-by: Israt Nisa <nisisrat@amazon.com >
2023-10-12 11:17:15 -04:00
0b3a447ba4
auto format distributed ( #5317 )
...
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal >
2023-02-19 08:33:09 +08:00
2caa6bd02d
[Dist] Enable save and load for Distributed Optimizer ( #4752 )
...
* add save/load for distributed optimizer
Co-authored-by: Ubuntu <ubuntu@ip-172-31-16-19.ap-northeast-1.compute.internal >
2022-11-01 14:30:16 +08:00
a208e8868b
[Misc] Black auto fix. ( #4680 )
...
* [Misc] Black auto fix.
* fix pylint disable
Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal >
2022-10-08 14:15:56 +08:00
7cd531c4d6
[Dist][Optim] Change op order in SparseAdagrad to be numerically closer to PyTorch ( #4253 )
...
Signed-off-by: Serge Panev <spanev@nvidia.com >
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
2022-07-25 14:53:53 +08:00
edf2d52666
[Dist][Optim] Fixed race conditions in distributed SparseAdam and SparseAdagrad ( #3971 )
...
* * Fixed race condition bug in distributed/optim/pytorch/sparse_optim.py's SparseAdam::update, corresponding with the bug fixed in the non-distributed version in https://github.com/dmlc/dgl/pull/3013 , though using the newer Event-based approach from that corresponding function. The race condition would often result in NaNs, like the previously fixed bug. https://github.com/dmlc/dgl/issues/2760
* * Fixed race condition bug in SparseAdagrad::update corresponding with the one fixed in SparseAdam::update in the previous commit. Same info applies.
* * Fixed typo in all copies of a repeatedly-copied comment near bug fixed 3 commits ago, checking all implementations nearby for a corresponding bug. (All of them appear to have been fixed as of 2 commits ago.)
* * Removed trailing whitespace
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com >
Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com >
2022-05-09 09:05:31 -07:00
bb6cec23b4
[BugFix] fix incorrect name when fetch data in sparse optim ( #3808 )
2022-03-06 10:43:43 -08:00
e0f0fa2a78
Fix a bug in sparse optimizer. ( #3779 )
...
* fix a bug.
* Update pylintrc
* fix.
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com >
2022-02-28 12:04:32 -08:00
156c17f3a6
[BugFix] Fix a bug of distributed SparseAdam optimizer. ( #3561 )
...
* Fix dist user embed bug
* Add some comments
* trigger
Co-authored-by: xiangsx <xiangsx@ip-10-3-83-86.eu-west-1.compute.internal >
2021-12-03 21:29:25 -08:00
d7390763f0
[Distributed] Deprecate old DistEmbedding impl, use synchronized embedding impl ( #3111 )
...
* fix.
* fix.
* fix.
* fix.
* Fix test
* Deprecate old DistEmbedding impl, use synchronized embedding impl
* update doc
Co-authored-by: Ubuntu <ubuntu@ip-172-31-71-112.ec2.internal >
Co-authored-by: Ubuntu <ubuntu@ip-172-31-2-66.ec2.internal >
Co-authored-by: Da Zheng <zhengda1936@gmail.com >
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com >
2021-07-14 00:15:53 +08:00
485c04cff6
[Distributed] Fix a few bugs in distributed API ( #3094 )
...
* fix.
* fix.
* fix.
* fix.
* Fix test
Co-authored-by: Ubuntu <ubuntu@ip-172-31-71-112.ec2.internal >
Co-authored-by: Ubuntu <ubuntu@ip-172-31-2-66.ec2.internal >
2021-07-05 15:12:02 +08:00
975eb8fc5d
[Distributed] Distributed node embedding and sparse optimizer ( #2733 )
...
* Draft for sparse emb
* add some notes
* Fix
* Add sparse optim for dist pytorch
* Update test
* Fix
* upd
* upd
* Fix
* Fix
* Fix bug
* add transductive exmpale
* Fix example
* Some fix
* Upd
* Fix lint
* lint
* lint
* lint
* upd
* Fix lint
* lint
* upd
* remove dead import
* update
* lint
* update unitest
* update example
* Add adam optimizer
* Add unitest and update data
* upd
* upd
* upd
* Fix docstring and fix some bug in example code
* Update rgcn readme
Co-authored-by: Ubuntu <ubuntu@ip-172-31-57-25.ec2.internal >
Co-authored-by: Ubuntu <ubuntu@ip-172-31-24-210.ec2.internal >
Co-authored-by: Ubuntu <ubuntu@ip-172-31-2-66.ec2.internal >
2021-05-03 08:37:40 +08:00
Rhett Ying