提交

提交图

3215 次代码提交

作者 SHA1 备注 提交日期
Chang Liu 598d746e59 [Example][Bugfix] Remove all torchtext legacy-related APIs for pytorch/pinsage example (#4130)
* Remove all torchtext legacy-related APIs

* Remove unused BagOfWordsPretrained class, and fix some typos

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-06-23 16:06:18 +08:00
Xin Yao 077e002fe5 [Bugfix][Rework] Automatically unpin tensors pinned by DGL (rework #3997) (#4135)
* Explicitly unpin tensoradapter allocated arrays

* Undo unrelated change

* Add unit test

* update unit test

* add pinned_by_dgl flag to NDArray::Container

* use dgl.ndarray for holding the pinning status

* update multi-gpu uva inference

* reinterpret cast NDArray::Container* to DLTensor* in MoveAsDLTensor

* update unpin column and examples

* add unit test for unpin column

Co-authored-by: Dominique LaSalle <dlasalle@nvidia.com>
Co-authored-by: nv-dlasalle <63612878+nv-dlasalle@users.noreply.github.com>
2022-06-23 13:56:54 +08:00
Triston 1ad65879a1 [Fix] Fix compiler warnings - part 1 (#4051)
* Fix a cub compile error for CUDA 11.5

* Fix comparison of integer expressions of different signedness in coo_sort.cu file

* Fix comparison of integer expressions of different signedness in cuda_compact_graph.cu file

* Remove never referenced variable in spmm.cu

* Fix comparison of integer expressions of different signedness in rowwise_pick.h file

* Fix comparison of integer expressions of different signedness in choice.cc file

* Remove never referenced variable col_data in spat_op_impl_coo.cc

* Remove never referenced variable allowed in global_uniform.cc

* Fix comparison of integer expressions of different signedness in graph.cc

* Fix comparison of integer expressions of different signedness in graph_apis.cc

* Fix the un-used ctx variable in ndarray_partition.cc file for cpu only build

* Fix comparison of integer expressions of different signedness in libra_partition.cc

* Fix comparison of integer expressions of different signedness in graph_op.cc

Co-authored-by: Triston Cao <tristonc@nvidia.com>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2022-06-23 11:57:43 +08:00
Rhett Ying ab1b28111d [Dist] etype is not guaranteed to be sorted (#4156) 2022-06-23 09:23:08 +08:00
Mufei Li 4d3c01d671 [Bug Fix] Fix the case when reverse_edge is False for citation graphs (#3840)
* Update citation_graph.py

* Update

* Update

* Update

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2022-06-22 23:52:02 +08:00
Quan (Andy) Gan 71157b05a8 [Bug] Fix problem with ShaDowKHopSampler working with reverse edge type exclusion (#4145)
* fix

* fix

* Update utils.py
2022-06-22 21:37:13 +08:00
maqy 794ec4a440 [BugFix] fix unstable sort when using dataloader with HeteroGraph (#4147)
* fix unstable sort

* add torch version check

* reformat

* split too long comments

* Update dataloader.py

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2022-06-22 18:05:42 +08:00
Mufei Li 31e4a89b23 [DGL-Go] Inference for Node Prediction Pipeline (full & ns) (#4095)
* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update
2022-06-21 17:45:44 +08:00
Rhett Ying 69226588a5 [Dist] defer to load node/edge feats (#4143)
* [Dist] defer to load node/edge feats

* fix lint

* Update python/dgl/distributed/partition.py

Co-authored-by: Minjie Wang <minjie.wang@nyu.edu>

* Update python/dgl/distributed/partition.py

Co-authored-by: Minjie Wang <minjie.wang@nyu.edu>

* fix lint

Co-authored-by: Minjie Wang <minjie.wang@nyu.edu>
2022-06-20 19:44:37 +08:00
Anthony Mahanna 532d4ac3d9 [Doc] Add ArangoDB-DGL in DGL-powered projects (#4139)
* Update README.md

* fix: ArangoDB hyperlink

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2022-06-20 16:44:00 +08:00
Rhett Ying 3ffe0c09b2 [Dist] re-try to receive rpc ndarray msg (#4142) 2022-06-20 15:15:50 +08:00
RuisiZhang 4a9be030df [Doc] Add distributed link prediction tutorial (#3993)
* add dist tutorial

* add predictor in dist prediction

* refine after rendering

* change links

* Update 2_link_prediction.py

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2022-06-17 18:51:44 +08:00
Rhett Ying b258729b3f [Dist] set socket as default backend for RPC (#4120)
* [Dist] set socket as default backend for RPC

* add tests both for socket and tensorpipe
2022-06-16 13:08:19 +08:00
Yunxin Sang 702d08dbf9 add gtrick in DGL-powered projects (#4128) 2022-06-15 20:04:16 +08:00
ndickson-nvidia 5f04fc2b31 [Doc] Updated transform ops list in dgl.rst (Issue #4087) (#4123)
* * Added functions from dgl.transforms.functional that were missing from the list for documentation in dgl.rst

* * Sorted transform ops list in dgl.rst in alphabetical order

Co-authored-by: Xin Yao <xiny@nvidia.com>
2022-06-15 14:52:37 +08:00
Serge Panev 652f4c0743 [Dist] Add env var for non-default SSH configs in tests (#4098)
Signed-off-by: Serge Panev <spanev@nvidia.com>

Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
2022-06-15 12:38:50 +08:00
nv-dlasalle 473bf15f8c [Bugfix] Disable non-atomic atomic operations (#4117)
* Disable non-atomic atomic operations

* Improve error message

* Make error message more friendly
2022-06-14 15:52:39 -07:00
彭齐荣 9a6f292493 [Bugfix] Fix fail to create_shared_mem_array in ddp spawn train #4110 (#4111)
* Fix fail to create_shared_mem_array in ddp spawn train #4110

Fix fail to create_shared_mem_array in ddp spawn train #4110

* [Bugfix] Fix fail to create_shared_mem_array in ddp spawn train #4110

[Bugfix] Fix fail to create_shared_mem_array in ddp spawn train #4110
Replace random.seed() to random_ = random.Random()

* Update pytorch.py

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2022-06-14 23:39:36 +08:00
Rhett Ying 7936e2eda1 [CI] loose time liimit for unit tests on Win64 (#4119) 2022-06-14 20:13:03 +08:00
RecLusIve-F defa292bc0 [Dataset] Add Flickr and Yelp dataset (#4099)
* Add Flickr and Yelp dataset

* Update flickr.py

* update

* Update yelp.py

* Update yelp.py

* update

* Update yelp.py

* Update test_data.py

* Update yelp.py

* update

* Update test_data.py

* Update yelp.py

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-06-14 16:14:12 +08:00
Rhett Ying 9501ed6a07 [Dist] master port should be fixed for all trainers (#4108)
* [Dist] master port should be fixed for all trainers

* add tests for tools/launch.py
2022-06-14 11:22:22 +08:00
Huarui HE 92e7733065 [dataset] Add a reorder flag to builtin datasets (#4104)
* add argument reorder=False for citation_graph

* add description of the argument reorder

* add reordered/un_reordered save_path

* add version number postfix

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-06-12 12:35:30 +08:00
Quan (Andy) Gan 148575e489 Update README.md (#4105) 2022-06-12 09:09:39 +08:00
Xin Yao 60b1c99241 [Fix] Wrap all CUDA runtime API/CUB calls with macro (#4083)
* Wrap all CUDA runtime API/CUB calls with macro

* remove the usage of explicit cudaMalloc in favor of AllocWorkspace

* fix typo

Co-authored-by: Israt Nisa <neesha295@gmail.com>
2022-06-11 12:18:01 +08:00
Rhett Ying 966d1aa8b7 [Dist] avoid busy ssh connection (#4096) 2022-06-09 19:35:42 +08:00
Rhett Ying abcc9cce83 disable multiple groups tests due to random failure in CI (#4101) 2022-06-09 17:40:53 +08:00
Chang Liu 549df65a77 [Bugfix] Fix example case: examples/pytorch/ogb/ogbn-proteins and examples/pytorch/ogb/ogbn-products (#4080)
* [Bugfix] Fix ogbn-gat-proteins/products examples

* Remove unused BatchSampler definition

* Remove comments to ease reading/reviewing

* Remove dataloader wrapper
2022-06-09 15:45:28 +08:00
Rhett Ying cac3720b48 [Dist] enable time out when fetching msg (#4043)
* [ist] enable time out when fetching msg

* fix lint error

* minor refinements

* improve minor log

* fix dist test

* fix timeout issue in tensorpipe
2022-06-08 20:20:03 +08:00
Rhett Ying 2de80dde7d [DistTest] add python test of RPC (#4093)
* [DistTest] add python test of RPC

* remove return
2022-06-08 17:03:02 +08:00
Rhett Ying c1ff4c9b41 [DistTest] add basic pipeline for dist test across machines (#3984)
* [DistTest] add basic pipeline for dist test across machines

* move launch remote cmd to separate file

* add test for rpc

* fix function naming rule

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2022-06-08 12:37:02 +08:00
ndickson-nvidia d92a4e8b64 [Bug] Fixed missing dgl.transforms.functional __all__ entries (#4089) 2022-06-08 09:31:44 +08:00
ndickson-nvidia eabcc58e41 [Bug][Feature] Added cublasGemm<__half> specialization (#3988) (#4029)
* * Added specialization of cublasGemm function for `__half` type, to try to address https://github.com/dmlc/dgl/issues/3988

* * Added USE_FP16 guard

* * Added test cases to test_segment_mm, to test newly-added FP16 specialization of cublasGemm

* * Replaced for loop in test_segment_mm with pytest.mark.parametrize, as recommended

Co-authored-by: Xin Yao <xiny@nvidia.com>
2022-06-07 16:00:02 +08:00
Tudor Andrei Dumitrascu 85c2ff7131 [Doc] Update link to correct destination. (#3966)
* Update link to correct destination.

* Update 4_rgcn.py

* Update 4_rgcn.py

* Update tutorials/models/1_gnn/4_rgcn.py

* Update tutorials/models/1_gnn/4_rgcn.py

* Update tutorials/models/1_gnn/4_rgcn.py

Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
2022-06-07 09:56:34 +08:00
ndickson-nvidia ea44da50de [Bug] Added common operations for FP16 on older GPUs (#4079)
* * Added support for common operations on FP16 (`half` or `__half`) for older GPU architectures
* Fixed an issue with previous check for FP16 support

* * Removing FP16 type checks, since they should no longer be needed

* * Fixed AtomicAdd to be atomic for `float` and `double` for old GPU architectures.  Unfortunately, it seems that atomicCAS for unsigned short seems to be unavailable until architecture 70, so half will have to stay non-atomic on old GPUs.

* * Fixed non-atomic version of `AtomicAdd<half>` for older GPUs to return old value instead value of new
2022-06-07 01:48:33 +08:00
Quan (Andy) Gan 31a81438fc parallelize csr2coo (#4081)
Co-authored-by: Xin Yao <xiny@nvidia.com>
2022-06-06 15:14:25 +08:00
Xin Yao 6014623d8f wrap all cuda kernel calls with macro (#4066)
Co-authored-by: nv-dlasalle <63612878+nv-dlasalle@users.noreply.github.com>
Co-authored-by: Israt Nisa <neesha295@gmail.com>
2022-06-06 09:16:44 +08:00
Sai Kandregula a3ea4873d3 [Example] Remove Bias in Linear layer because of BatchNorm1d usage (#4071)
* typo fix in TUDataset docs

* remove bias in Linear layer because of BatchNorm1d

Co-authored-by: decoherencer <decoherencer@users.noreply.github.com>
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-06-04 17:16:20 +08:00
Mufei Li 6a91d18151 [DGL-Go] Graph Property Prediction Pipeline (#3927)
* Update

* Update

* Update

* Fix

* Update

* Update

* Update

* Update

* Update

* Fix

* Update

* Update

* Update

* Fix

* Update

* Update

* update

* Update

* Update

* Update

* Fix

* Fix

* Update

* Update

* Update

* Update

* lr_scheduler

* Update

* Update

* Update

* Update

* Update

* Update

* Fix

* Fix

* Fix

* Fix

* Update

* Update

* Update

* Update

* Update

* Fix

* Update

* Update

* Update

* Update

* Fix

* Update

* Update

* Update

* Fix

* Fix

* Fix

* Update

* Update

* Fix

* Fix

* Update

* Update

* Update

* Update

* Update

* Update

* update

* CI

* Update

* Update

* Update

* Update

* Update

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2022-06-04 15:43:34 +08:00
Chang Liu d31448dd24 [Bugfix] Fix GCMC examples (#4082)
* [Example][Bug] Fix GCMC examples

* Revert the change of model.py from #74f01405
2022-06-04 11:19:33 +08:00
RecLusIve-F 0f2ff47de6 [Example] Add BGRL example (#4077)
* Add BGRL example

* Update README.md

* Update utils.py

* Update

* Update utils.py
2022-06-03 23:13:30 +08:00
Sai Kandregula c9c79b4401 typo fix in TUDataset docs (#4056)
Co-authored-by: decoherencer <decoherencer@users.noreply.github.com>
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-06-03 13:29:32 +08:00
Riju Mukherjee efd909e62e [NN] Enhance EGATConv branch (#4062)
* enhance EGATConv| nfeats as tuples

* egatconv modified for bipartite graphs

* modified docstrings

* added/modified unittests for EGATConv

* Update egatconv.py

* rectified lint errors

Co-authored-by: rijulizer <riju.mukherjee@gmail.com>
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-06-03 11:53:10 +08:00
Xin Zhang 92063d88f0 fix indent in minibatch-custom-sampler.rst (#4063)
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-06-03 01:00:31 +08:00
RecLusIve-F 89655cfda2 [Dataset] Add WikiCS Dataset (#4035)
* Fix bugs & Update dataset

* Update

* Update wikics.py

* Update wikics.py

* Update test_data.py

* Update wikics.py

* Update wikics.py

* Update wikics.py

* update

* Update module.py

* Update dgl.data.rst

* Update wikics.py

* Update wikics.py

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-06-02 23:03:17 +08:00
Chang Liu 9e46423e4f [Bugfix] Fix cluster-gat examples (#4068)
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-06-02 21:39:59 +08:00
Mufei Li d9c25521bc [Data] AsGraphPredDataset (#4073)
* Update

* CI

* Update

* Update

* Fix

* Fix
2022-06-02 16:18:48 +08:00
Quan (Andy) Gan 9922f41f06 Change warning message for tensoradapter when not found (#4055)
* change warning message

* Update tensordispatch.cc
2022-05-29 02:04:31 +08:00
Quan (Andy) Gan 00c09b9f91 Revert "[bugfix] Explicitly unpin tensoradapter allocated arrays (#3997)" (#4061)
This reverts commit fdd1fe1908.
2022-05-28 20:44:20 +08:00
Quan (Andy) Gan c577dc9fc3 add sanity check (#4050) 2022-05-28 18:51:33 +08:00
Sai Kandregula 7ec165c2c5 default value typo fix in GlobalUniform negative sampler (#4060)
Co-authored-by: decoherencer <decoherencer@users.noreply.github.com>
2022-05-28 13:22:41 +08:00