Commits · 077e002fe54d3438071a41772f2a0d3fdbd8814c · OpenDAS / dgl

23 Jun, 2022 1 commit

[Bugfix][Rework] Automatically unpin tensors pinned by DGL (rework #3997) (#4135) · 077e002f

Xin Yao authored Jun 23, 2022



* 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>

077e002f

22 Jun, 2022 1 commit
- [Bug] Fix problem with ShaDowKHopSampler working with reverse edge type exclusion (#4145) · 71157b05
  Quan (Andy) Gan authored Jun 22, 2022
```
* fix

* fix

* Update utils.py
```
  71157b05
21 Jun, 2022 1 commit

[DGL-Go] Inference for Node Prediction Pipeline (full & ns) (#4095) · 31e4a89b

Mufei Li authored Jun 21, 2022

* 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

31e4a89b

20 Jun, 2022 1 commit

[Dist] defer to load node/edge feats (#4143) · 69226588

Rhett Ying authored Jun 20, 2022



* [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>

69226588

16 Jun, 2022 1 commit
- [Dist] set socket as default backend for RPC (#4120) · b258729b
  Rhett Ying authored Jun 16, 2022
```
* [Dist] set socket as default backend for RPC

* add tests both for socket and tensorpipe
```
  b258729b
15 Jun, 2022 1 commit

[Dist] Add env var for non-default SSH configs in tests (#4098) · 652f4c07

Serge Panev authored Jun 14, 2022


Signed-off-by: Serge Panev <spanev@nvidia.com>
Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com>

652f4c07

14 Jun, 2022 2 commits

[Dataset] Add Flickr and Yelp dataset (#4099) · defa292b

RecLusIve-F authored Jun 14, 2022



* 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>

defa292b

[Dist] master port should be fixed for all trainers (#4108) · 9501ed6a
Rhett Ying authored Jun 14, 2022
```
* [Dist] master port should be fixed for all trainers

* add tests for tools/launch.py
```
9501ed6a

12 Jun, 2022 1 commit

[dataset] Add a `reorder` flag to builtin datasets (#4104) · 92e77330

Huarui HE authored Jun 12, 2022



* 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>

92e77330

09 Jun, 2022 1 commit
- disable multiple groups tests due to random failure in CI (#4101) · abcc9cce
  Rhett Ying authored Jun 09, 2022
  
  abcc9cce
08 Jun, 2022 3 commits

[Dist] enable time out when fetching msg (#4043) · cac3720b

Rhett Ying authored Jun 08, 2022

* [ist] enable time out when fetching msg

* fix lint error

* minor refinements

* improve minor log

* fix dist test

* fix timeout issue in tensorpipe

cac3720b

[DistTest] add python test of RPC (#4093) · 2de80dde
Rhett Ying authored Jun 08, 2022
```
* [DistTest] add python test of RPC

* remove return
```
2de80dde

[DistTest] add basic pipeline for dist test across machines (#3984) · c1ff4c9b

Rhett Ying authored Jun 08, 2022



* [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>

c1ff4c9b

07 Jun, 2022 1 commit

[Bug][Feature] Added cublasGemm<__half> specialization (#3988) (#4029) · eabcc58e

ndickson-nvidia authored Jun 07, 2022

* * 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>

eabcc58e

04 Jun, 2022 1 commit

[DGL-Go] Graph Property Prediction Pipeline (#3927) · 6a91d181

Mufei Li authored Jun 04, 2022



* 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>

6a91d181

03 Jun, 2022 1 commit

[NN] Enhance EGATConv branch (#4062) · efd909e6

Riju Mukherjee authored Jun 03, 2022



* 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>

efd909e6

02 Jun, 2022 2 commits

[Dataset] Add WikiCS Dataset (#4035) · 89655cfd

RecLusIve-F authored Jun 02, 2022



* 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>

89655cfd

[Data] AsGraphPredDataset (#4073) · d9c25521
Mufei Li authored Jun 02, 2022
```
* Update

* CI

* Update

* Update

* Fix

* Fix
```
d9c25521

28 May, 2022 2 commits
- Revert "[bugfix] Explicitly unpin tensoradapter allocated arrays (#3997)" (#4061) · 00c09b9f
  Quan (Andy) Gan authored May 28, 2022
```
This reverts commit fdd1fe19.
```
  00c09b9f
- add sanity check (#4050) · c577dc9f
  Quan (Andy) Gan authored May 28, 2022
  
  c577dc9f
26 May, 2022 1 commit

[Build][Tests] Enable FP16 for GPU builds in CI (#4030) · 7a065a9c

nv-dlasalle authored May 26, 2022

* Enable FP16 for GPU builds in CI

* Limit default GPU archs to pascal and above

* Disable FP16 dispatching for cuda architectures less than 60

* Fix linting

* Fix typos

7a065a9c

25 May, 2022 1 commit
- [Bugfix] Cython CAPI holding GIL causes deadlock when Python callback is asynchronous (#4036) · 3c129ad7
  Minjie Wang authored May 25, 2022
```
* cython nogil

* move APIs to internal and add unit test

* fix lint

* disable callback array test
```
  3c129ad7
24 May, 2022 1 commit

[Bug fix] Misc Fix for Transforms and NN Modules (#4038) · 230b886e

Mufei Li authored May 24, 2022

* Update module.py

* Update utils.py

* Update utils.py

* Update utils.py

* Update module.py

* Update

* Update

* Update

230b886e

23 May, 2022 1 commit
- Fix test naming (#4023) · 14251504
  nv-dlasalle authored May 23, 2022
```
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
```
  14251504
20 May, 2022 1 commit

[NN] Label Propagation & Directional Graph Networks (#4017) · 3fe5eea7

rudongyu authored May 20, 2022



* add label propagation module

* fix prev bug in example

* add dgn

* fix linting and doc issues

* update label propagation & dgn

* update label propagation & dgn

* update example

* fix unit test

* fix agg heritage issue

* fix agg issue

* fix lint

* fix idx

* fix lp gpu issue

* Update

* Update
Co-authored-by: mufeili <mufeili1996@gmail.com>

3fe5eea7

18 May, 2022 1 commit
- [Dist][BugFix] enable sampling on bipartite (#4014) · 53835bdb
  Rhett Ying authored May 18, 2022
```
* [Dist][BugFix] enable sampling on bipartite

* add comments for tests
```
  53835bdb
17 May, 2022 1 commit

[Feature] Added floating-point conversion functions to dgl.transforms.functional (#3890) · e0e8736f

ndickson-nvidia authored May 16, 2022

* * Added half_(), float_(), and double_() functions to DGLHeteroGraph, HeteroNodeDataView, and HeteroEdgeDataView, for converting floating-point tensor data to float16, float32, or float64 precision

* * Extracted out private functions for floating-point type conversion, to reduce code duplication

* * Added test for floating-point data conversion functions, half_(), float_(), and double_()

* * Moved half_(), float_(), and double_() functions from HeteroNodeDataView and HeteroEdgeDataView to Frame class

* * Updated test_float_cast() to use dgl.heterograph instead of dgl.graph

* Added to CONTRIBUTORS.md

* * Changed data type conversion to be deferred until the data is accessed, to avoid redundant conversions of data that isn't used.

* * Addressed issues flagged by linter

* * Worked around a bug in the old version of mxnet that's currently used for DGL testing

* * Only defer Column data type co...

e0e8736f

16 May, 2022 4 commits

Prevent users from attempting to pin PyTorch non-contiguous tensors or views... · 1f2e6960

nv-dlasalle authored May 16, 2022


Prevent users from attempting to pin PyTorch non-contiguous tensors or views only encompassing part of tensor. (#3992)

* Disable pinning non-contiguous memory

* Prevent views from being converted for write

* Fix linting

* Add unit tests

* Improve error message for users

* Switch to pytest function

* exclude mxnet and tensorflow from inplace pinning

* Add skip

* Restrict to pytorch backend

* Use backend to retrieve device

* Fix capitalization in decorator
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>

1f2e6960

[bugfix] Explicitly unpin tensoradapter allocated arrays (#3997) · fdd1fe19
nv-dlasalle authored May 16, 2022
```
* Explicitly unpin tensoradapter allocated arrays

* Undo unrelated change

* Add unit test

* update unit test
```
fdd1fe19

[Dataset & Transform] Synthetic Datasets for Explainability and SIGNDiffusion Transform (#3982) · 65b0b9e8

Mufei Li authored May 16, 2022

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Fix

* Update

* Update

* Update

65b0b9e8

[Peformance] Remove unnecessary induced vertices in EdgeSubgraph (#3978) · 03024f95
Xin Yao authored May 16, 2022
```
* remove unnecessary induced vertices in EdgeSubgraph

* add unit test
```
03024f95

13 May, 2022 1 commit
- [Build] enable USE_AVX in CI and regression tests (#4004) · 65e6b04d
  Rhett Ying authored May 13, 2022
  
  65e6b04d
11 May, 2022 2 commits

Supporting DGL build with arch specific optimization (#3996) · cb39eae1
Vikram Sharma authored May 11, 2022
```
Based on the pull request: https://github.com/dmlc/dgl/pull/3983
```
cb39eae1

[Dist] Enable maximum try times for socket backend via DGL_DIST_MAX_T… (#3977) · 22e218d3

Rhett Ying authored May 11, 2022

* [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

22e218d3

10 May, 2022 1 commit
- [CI] print commit/build ID into comment (#3989) · 22d7f924
  Rhett Ying authored May 10, 2022
  
  22d7f924
07 May, 2022 1 commit

[Transform] Add FeatMask and RowFeatNormalizer transforms (#3968) · b2ec5a2e

RecLusIve-F authored May 07, 2022



* [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>

b2ec5a2e

06 May, 2022 1 commit

[DGL-Go] CI for DGL-Go (#3959) · 334e6434

Mufei Li authored May 06, 2022

* Update

* Update

* Fix

* Update

* CI

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

334e6434

29 Apr, 2022 2 commits
- [BugFix] fix CI status (#3963) · 3f792071
  Rhett Ying authored Apr 29, 2022
  
  3f792071
- [BugFix] fix job status in master CI (#3962) · 56614f32
  Rhett Ying authored Apr 29, 2022
```
* [BugFix] fix job status in master CI

* finalize
```
  56614f32
28 Apr, 2022 1 commit

[Feature] Dataloader worker affinitization (#3723) · 5561af45

Daniil Sizov authored Apr 28, 2022



* 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>

5561af45