22e218d335
[Dist] Enable maximum try times for socket backend via DGL_DIST_MAX_T… ( #3977 )
...
* [Dist] Enable maximum try times for socket backend via DGL_DIST_MAX_TRY_TIMES
* reset env before/after test
* print log for info when trying to connect
* fix
* print log in python instead of cpp
2022-05-11 13:03:06 +08:00
74f0140533
[Example] Rename NodeDataLoader to DataLoader in GraphSAGE example ( #3972 )
...
* rename
* Update node_classification.py
* more fixes...
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2022-05-11 09:10:54 +08:00
22d7f92407
[CI] print commit/build ID into comment ( #3989 )
2022-05-10 16:04:21 +08:00
d4d4fa1eaf
Update pinsage readme: torchtext>=0.9.0 ( #3974 )
...
Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com >
2022-05-10 12:41:01 +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
6056ed62c2
Update Jenkinsfile ( #3985 )
2022-05-09 16:58:07 +08:00
34981bd676
[Doc] Fix compact_graphs example docstring ( #3953 )
...
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2022-05-08 15:58:00 +08:00
b2ec5a2ea6
[Transform] Add FeatMask and RowFeatNormalizer transforms ( #3968 )
...
* [Model]P-GNN
* updata
* [Example]P-GNN
* Update README.md
* Add NodeFeatureMasking and NormalizeFeatures
* Update
* Update transforms.rst
* Update
* Update
* Update
* Update test_transform.py
* Update
* Update
* Update test_transform.py
* Update module.py
* Update module.py
* Update module.py
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
2022-05-07 11:59:39 +08:00
334e6434d2
[DGL-Go] CI for DGL-Go ( #3959 )
...
* Update
* Update
* Fix
* Update
* CI
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
2022-05-06 12:26:50 +08:00
44dba197d1
[CI] fetch status and generate report in separate node/machine ( #3973 )
2022-05-05 19:31:32 +08:00
5b97a1a26e
add bias argument ( #3970 )
2022-05-05 10:17:24 +08:00
ae7e3db631
fix bugs in gat training script ( #3505 )
...
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com >
Co-authored-by: Zihao Ye <expye@outlook.com >
2022-04-29 17:53:00 +08:00
3f792071f6
[BugFix] fix CI status ( #3963 )
2022-04-29 14:47:19 +08:00
56614f320f
[BugFix] fix job status in master CI ( #3962 )
...
* [BugFix] fix job status in master CI
* finalize
2022-04-29 12:54:13 +08:00
529fd1075d
[CI] fix job status ( #3954 )
...
* [CI] fix job status
* fetch from master
2022-04-28 15:49:29 +08:00
5561af4517
[Feature] Dataloader worker affinitization ( #3723 )
...
* PR3355 + CSR conversion workaround
* Remove debug code
* Fix convention errors
* Remove wrongly added code section during merge
* Update to reflect dataloading changes
* Fix missing changes
* Remove comment
* Fix linter errors
* Fix trailing whitespace
* Add wrapper around worker init function
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com >
2022-04-28 13:00:04 +08:00
37be02a486
[Feature] enable socket net_type for rpc ( #3951 )
...
* [Feature] enable socket net_type for rpc
* fix lint
* fix lint
* fix build issue on windows
* fix test failure on windows
* fix test failure
* fix cpp unit test failure
* net_type blocking max_try_times
* fix other comments
* fix lint
* fix comment
* fix lint
* fix cpp
2022-04-28 00:10:06 +08:00
c3baf4332b
[Bug] Fix multi-GPU edge classification crashing with pure GPU sampling ( #3946 )
...
* fix
* fix
Co-authored-by: Xin Yao <xiny@nvidia.com >
2022-04-27 22:59:21 +08:00
2eaa58e047
[CI] refine status check ( #3958 )
2022-04-27 15:28:26 +08:00
6e46bbf59c
[Performance][GPU] Improving Disjoint Union kernel for Graph Dataloaders ( #3895 )
...
* Based on issue #3436 . Improving _SegmentCopyKernel s GPU utilization by switching to nonzero based thread assignment
* fixing lint issues
* Update cub for cuda 11.5 compatibility (#3468 )
* fixing type mismatch
* tx guaranteed to be smaller than nnz. Hence removing last check
* minor: updating comment
* adding three unit tests for csr slice method to cover some corner cases
* timing repeatkernel
* clean
* clean
* clean
* updating _SegmentMaskColKernel
* Working on requests: removing sorted array check and adding comments to utility functions
* fixing lint issue
* Optimizing disjoint union kernel
* Trying to resolve compilation issue on CI
* [EMPTY] Relevant commit message here
* applying revision requests on cpu/disjoint_union.cc
* removing unnecessary casts
* remove extra space
Co-authored-by: Abdurrahman Yasar <ayasar@nvidia.com >
Co-authored-by: nv-dlasalle <63612878+nv-dlasalle@users.noreply.github.com >
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com >
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com >
2022-04-26 20:06:48 +08:00
306e0a46e1
[Bug] Fixed optimizer definition in gnn_explainer ( #3944 ) ( #3948 )
...
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
2022-04-25 16:39:44 +08:00
248bece633
[NN] GINEConv ( #3934 )
...
* Update
* Update
* Update
* Update
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2022-04-25 14:41:05 +08:00
df7a61210e
Fix benchmark time measurement ( #3928 )
...
* Fix benchmark time measurement
* Reduce batch size for bench_rgcn_homogeneous_ns am
bench_rgcn_homogeneous_ns am data sample size is too small for 1024 batch size
Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com >
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com >
2022-04-24 21:45:04 +08:00
764b5d0524
Fix test_module_heat_kernel test ( #3941 )
...
Signed-off-by: Serge Panev <spanev@nvidia.com >
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
2022-04-23 13:32:29 +08:00
dc1c0ac28a
[Example][Bug] Fix GAT example ( #3947 )
...
* fix
* oops
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
2022-04-22 23:44:45 +08:00
338db32d21
[Feature] Try upload report to s3 ( #3891 )
...
refactor CI report and log
2022-04-22 16:02:12 +08:00
4375c2d7e4
[Doc] Fix documentation in dgl.multiprocessing namespace ( #3929 )
...
* fix docs
* remove
* oh
* fix
2022-04-16 00:35:06 +08:00
f5bba284f8
[Example] Add EGES example ( #3756 )
...
* add eges example
* remove csv files and add data link
* Update README.md
* Update main.py
* Update model.py
* Update sampler.py
* Update utils.py
* Update model.py
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com >
2022-04-14 16:23:08 +08:00
a0bf5daa65
fix ( #3903 )
...
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2022-04-13 12:28:19 +08:00
c380f0b5fa
Update gatv2.py ( #3921 )
...
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com >
2022-04-13 09:47:52 +08:00
e06e63d5d5
[Feature] Enable UVA sampling with CPU indices ( #3892 )
...
* enable UVA sampling with CPU indices
* add docs
* add more docs
* lint
* fix
* fix
* better error message
* use mp.Barrier instead of queues
* revert
* revert
* oops
* revert dgl.multiprocessing.spawn
* Update pytorch.py
2022-04-12 20:56:51 +08:00
0d878ff8ef
[Example] Cleaned GraphSAGE node classification example with PyTorch Lightning ( #3863 )
...
* cleaned pl node classification example
* conform to PL's method of updating the dataloader
* update
* lint
* fix test
* fix
2022-04-12 17:26:23 +08:00
baa9292899
[bugfix] Fix EdgePredictionSampler for UVA sampling ( #3904 )
...
* Add failing unit test
* Fix negative sampler edge types
* fix test
* oops
* revert
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
Co-authored-by: Quan Gan <coin2028@hotmail.com >
2022-04-12 15:25:21 +08:00
f931c6ba54
[Examples] Add pure gpu mode in the GraphSAGE node classification and link prediction ( #3856 )
...
Signed-off-by: Serge Panev <spanev@nvidia.com >
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com >
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com >
2022-04-12 13:03:03 +08:00
5fcd7f290f
[Feature] Enable UVA for GPU PinSAGE and RandomWalk ( #3857 )
...
* enable uva for pinsage sampler
* unit test
* modify some checks on the python side
* remove legacy random walk code
* update unit test
* update unit test
* fix unit test
* adjust checks
* move some checks to c++
* move max_nodes check to cuda kernel
* fix ci for tf
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com >
Co-authored-by: nv-dlasalle <63612878+nv-dlasalle@users.noreply.github.com >
2022-04-11 20:01:23 +08:00
23a5e674e7
[fix] device cannot be Nonetype object. ( #3822 )
...
* fix device = none to cpu.
* fix singleton-comparison
* modify device is none
* fix use uva
* add docs
* add edge dataloader
* add doc
* fix trailing-whitespace
* modify default
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com >
2022-04-11 14:58:27 +08:00
43b9d40e5e
[Doc] Fix Doc for EGNNConv ( #3919 )
...
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2022-04-11 11:01:15 +08:00
01e50626d3
[Example] Add TorchMetrics in README ( #3913 )
...
* Update README.md
* Update README.md
* Update README.md
Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com >
2022-04-09 23:25:51 +08:00
adb3a7c11e
[BugFix] record/restore pin status when pickle/unpickle ( #3914 )
...
* [BugFix] record/restore pin status when pickle/unpickle
* disable test on TF
* set version as expected
* unpin memory in test
2022-04-09 11:42:23 +08:00
ed15b47147
Update ( #3916 )
2022-04-08 14:29:03 +08:00
dc5035b1f3
[NN] Add EGNN & PNA ( #3901 )
...
* add EGNN & PNA
* fix egnn issues
* fix pna issues
* update pna conv
* add doc strings
* update pnaconv
* fix unused args issue
* fix moment aggregation issue
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
2022-04-07 17:44:57 +08:00
27a6eb564c
[Examples] Update graphsage multi-gpu example to use mutliple GPUs for validation and testing. ( #3827 )
...
* Update graphsage multi-gpu example to use mutliple GPUs for validation and
testing.
* Remove argmax
* Fix rebase error
* Add more documentation to example and simplify
* Switch to name shared memory
* Add comment about how training is distributed
* Restore iteration count
* fix munmap error reporting for better error messages
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com >
2022-04-06 00:32:37 +08:00
11e910b6f8
[Doc] Hotfix for GraphSAINT Doc ( #3898 )
...
* Update graphsaint.py
* Update graphsaint.py
2022-03-31 19:22:21 +08:00
35e66f426a
[Bugfix] Fix UVA sampling with partially specified node types ( #3897 )
...
* fix uva with partial node types
* lint
* skip tensorflow unit test
2022-03-31 17:07:02 +08:00
7a2f494373
[bugfix] Make source build work for systems where the default is python2 ( #3718 )
...
* Make compatible for system where the default is python2
* Update cmakelists with find package
* Switch to find_program
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com >
2022-03-30 08:27:58 -07:00
4c7bd314b4
[Feature] Radius Graph ( #3829 )
...
* radius graph
* remove trailing whitespaces from docs
* disable invalid name for transform func
* disable radius graph module invalid name
* move pylint disable before init
* fix missing nodes from point set
* update docs indexing
* add compute mode as optional param
* radius graph test
* remove trailing whitespaces
* fix precision when comparing tensors
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
2022-03-30 17:23:29 +08:00
290b7c25c2
[Transform] Random Walk Positional Encoding & Laplacian Positional Encoding ( #3869 )
...
* [Transform] Random Walk Positional Encoding & Laplacian Positional Encoding
* Update python/dgl/transforms/module.py
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
* Update python/dgl/transforms/functional.py
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
* Update python/dgl/transforms/functional.py
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
* Update python/dgl/transforms/functional.py
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
* Update python/dgl/transforms/functional.py
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
* Update python/dgl/transforms/functional.py
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
* Update python/dgl/transforms/functional.py & python/dgl/transforms/module.py
* update python/dgl/transforms/functional.py
* update doc indexing
* update unit test
* support weighted graphs for random_walk_pe
* change reference
* fix linter errors
* fix datatype compatibility with MXNet & TF
* fix device errors
* fix result precision errors
* change tensorflow abs api
* reboost CI
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
2022-03-30 15:59:28 +08:00
9fee20b91d
[Sampler] [Example] SAINTSampler and Simplify GraphSAINT Example ( #3879 )
...
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Fix
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com >
2022-03-30 13:51:37 +08:00
0dd500e386
[BugFix] Remove redundant reference of networkx package in pagerank.py ( #3888 )
...
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2022-03-30 11:54:03 +08:00
d8375c10ec
[BugFix] involve indices_devices for other dataset object. ( #3810 )
...
* involve indices_devices for other dataset object.
* modify raise error when device not found.
* remove empty line
* fix line too long
2022-03-30 11:50:03 +08:00
Rhett Ying