提交

2367 次代码提交

作者 SHA1 备注 提交日期
Quan (Andy) Gan c7edb66b8d temporarily removing for 0.9.x due to ux changes 0.9.0 2022-07-18 23:27:45 +08:00
Quan Gan 41240d585a update branch 2022-07-15 12:25:10 +00:00
Quan Gan 9f37e83cc5 update version 2022-07-15 12:25:10 +00:00
Quan (Andy) Gan 9a7ad16eaf decompose (#4259) 2022-07-15 20:24:12 +08:00
Mufei Li fdbf5a0fda [DGL-Go][Doc] Update DGL-Go version to 0.0.2 and misc fix from bug bash (#4236)
* Update

* Update

* Update

* Update

Co-authored-by: Ubuntu <ubuntu@ip-172-31-53-142.us-west-2.compute.internal>
Co-authored-by: Xin Yao <xiny@nvidia.com>
2022-07-14 21:30:38 +08:00
Xin Yao 79b0a50afc [Unittest][Fix] Several unit tests fixes for Ampere+ and PyTorch 1.12+ (#4213)
* Fix test_csrmm for tensor core

* unset allow tf32 flag

* update test unified tensor

* skip fp16 for CPU
2022-07-14 19:59:57 +08:00
Quan (Andy) Gan 2efdaa5df3 [Bug] Revert clearing backward cache for retain_graph flag (#4249)
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2022-07-14 16:32:01 +08:00
Hongyi ZHANG 5c76e47f78 [Docs] fix ffi docs (#4248)
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2022-07-14 13:20:42 +08:00
alxwen711 20fe939741 [Doc] Fixed grammatical typos in README (#4246)
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2022-07-14 13:19:31 +08:00
Xin Yao 82ca781911 [Bug] Record stream when using another CUDA stream for data transfer (#4250)
* record stream when using another cuda stream for data transfer

* fix linting

* fix None stream
2022-07-14 13:17:37 +08:00
Rhett Ying 2f322a94b9 [BugFix] fetch device from graph (#4245)
Co-authored-by: Xin Yao <xiny@nvidia.com>
2022-07-14 10:25:07 +08:00
Chang Liu e26d206437 [Example][Bugfix] Fix link pred example in graphsage (#4255) 2022-07-14 08:30:51 +08:00
PengZhang 1c9528f561 Doc: add a DGL-powered projects DGLD (#4254)
* feature: add a parse parameter degree_as_nlabel for pytorch-gin demo

* fix some typo

* [fix]: allow to benchmark all of the 9 dataset.

* [Feature] add epoch number to log

* [Feature]:simply list the command lines for all datasets (https://github.com/dmlc/dgl/pull/3676#discussion_r790270705) and run a test.

* Update README.md

* update: DGL-powered projects-DGLD

[DGLD](https://github.com/EagleLab-ZJU/DGLD) is an open-source library for Deep Graph Anomaly Detection based on pytorch and DGL.  It provides unified interface of popular graph anomaly detection methods, including the data loader, data augmentation, model training and evaluation. Also, the widely used modules are well organized so that developers and researchers can quickly implement their own designed models.

Co-authored-by: Ubuntu <ubuntu@ip-172-31-10-175.ap-northeast-1.compute.internal>
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-07-13 23:02:29 +08:00
Rhett Ying 4361f8a69d [Test] Fix broken regression benchmarks (#4229)
* dummy

* upgrade torch from 1.8.1 to 1.9.0 for asv

* update torch download link

* remove version for torchvision and torch text

* revert unnecessary change

* fix several test failure

* fix

* remove am dataset

* fix test failure

* fix test failure

* fix rgcn failures
2022-07-13 17:26:45 +08:00
Chang Liu c56e27a807 [Bugfix] Accessing data from the indexes stored in same device (#4242)
* First update to fix two examples

* Update to fix RGCN/graphsage example and dataloader

* Update
2022-07-13 09:02:07 +08:00
Rhett Ying 14e4e1b0f7 [Dist] enable to specify sort_etype for sample_etype_neighbours (#4212)
* [Dist] enable to specify sort_etype for sample_etype_neighbours

* fix lint

* pass argument instead of env

* fix lint and doc string

* refine args

* remove unnecessary lines

* debug only

* debug add sort time log

* change interface

* fix typo

Co-authored-by: Xin Yao <xiny@nvidia.com>
2022-07-11 18:37:25 +08:00
Rhett Ying c65d6fa55f [Dist] format dtypes when loading graph in server (#4228)
* [Dist] format dtypes when loading graph in server

* add test

* refine

* add comments
2022-07-11 14:08:17 +08:00
Xin Yao 1feec870c1 [Bugfix] Add CUDA context availability check before setting curand seed (#4223) 2022-07-09 10:19:10 +08:00
Xin Yao 3e26c3d198 fix caregnn (#4211)
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-07-08 16:16:30 +08:00
Chang Liu 52d43127f7 [Example][Bugfix] Fix arma example (#4218)
* Fix arma example

* Update

Co-authored-by: Xin Yao <xiny@nvidia.com>
2022-07-08 12:25:21 +08:00
Serge Panev e3b6ac8e6c [Dist][Test] Add tests for multi-node DistTensor (#4122)
Signed-off-by: Serge Panev <spanev@nvidia.com>

Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
2022-07-08 10:47:41 +08:00
Chang Liu 28b0904779 [Example][Bugfix] Fix dimenet example (#4219)
Co-authored-by: Xin Yao <xiny@nvidia.com>
2022-07-07 14:52:10 +08:00
Xin Yao 9ee7ced5e3 [Performance] Redirect AllocWorkspace to PyTorch's allocator if available (#4199) 2022-07-07 13:35:44 +08:00
Vibhu Jawa 9ae117d3cd [Feature] Add to_cugraph and from_cugraph to DGL (#4166)
* Added to_cugraph and from_cugraph functionality

* fix from_cugraph example

* Addressed Reviews

* Fix linting

* Apply suggestions to docstrings from code review

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

* Add API docs and remove `from_cugraph` alias

* move cugraph tests to  tests/cugraph/test_basics.py

* remove ununsed imports from test_basics.py

* remove pytest.importorskip as no longer needed

Co-authored-by: Minjie Wang <minjie.wang@nyu.edu>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2022-07-07 11:52:17 +08:00
Chang Liu 0d1b2c5a2c [Example] Update README file of eges example (#4220) 2022-07-07 09:14:40 +08:00
Chang Liu 885be1784b [Example][Refactor] Refactor GCN example (#4160)
* Refactor GCN example

* Refactor GCN based on graphsage

* Readme update

* Minor update

* update

* Remove user-defined GCN implementation

* README update

* Update

* Update CONTRIBUTORS.md

* update task_example_test

Co-authored-by: Xin Yao <xiny@nvidia.com>
2022-07-05 15:58:31 +08:00
Rhett Ying 85f281170f [CI] Add new CI stage for testing cugraph (#4171)
* [CI] add new stage specific forcuda related features based on nvidia+pytorch

* build and test for gpu_nv

* fix build failure

* fix unit tests

* make -j

* install cython beforehand

* copy cython lib

* test cugraph tests only

* fix typo

* separate test script for cugraph

* refactor build dgl shell
2022-07-05 10:57:07 +08:00
Chang Liu 0f0e7c7fad [Example][Bugfix] Bugfix for dgi example (#4201) 2022-07-02 12:48:17 +08:00
Rhett Ying dcf1699280 [BugFix] check whether etype sorted when sampling (#4198) 2022-07-01 15:52:14 +08:00
Chang Liu a9768cb38f [Example][Refactor] Minor update on the golden example (#4197)
* minor update on golden example

* update

* update

* Update README

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2022-07-01 14:34:45 +08:00
Rhett Ying 6a6597a02a [Feature] extend sort_csr/csc_by_tag to edge (#4164)
* [Feature] extend sort_csr/csc_by_tag to edge

* fix test ffailure in tensorflow

* refine sorting by edges

* fix docstring

* remove unnecessary mem

Co-authored-by: Xin Yao <xiny@nvidia.com>
2022-07-01 13:16:11 +08:00
Chang Liu b76d0ed1db [Example][Refactor] Regolden graphsage example for future guide (#4186)
* Regolden graphsage example to guide others

* update golden

* update

* Update example and propagate to original folder

* Update to remove ^M (windows DOS) character

* update

* Merge file changes and update README

* Minor comment update

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-06-30 21:56:13 +08:00
Chang Liu a6bd96aa96 Fix example crashes due to DGL API update (#4194)
Co-authored-by: Xin Yao <xiny@nvidia.com>
2022-06-30 21:10:44 +08:00
Minjie Wang f7dae4533a [CI] Reduce CI workload (#4196)
* try optimize CI

* fix go test; adjust timing report

* disable certain tests for mx/tf backends

* fix ut

* add pydantic
2022-06-30 20:01:16 +08:00
Quan (Andy) Gan 7735473b44 [Doc] fix typo (#4193)
Co-authored-by: Xin Yao <xiny@nvidia.com>
2022-06-30 15:00:43 +08:00
nv-dlasalle d2a22984c1 [bugfix] Implement __setstate__ for Column (fixes #4107) (#4174)
* * Workaround for graph data saving/loading compatibility problem in Column class.  There may be more places in DGL with the same issue, due to using Python serialization, instead of a more cohesive, comprehensive strategy.  This is just a local fix.

* Add checking for non-empty states

* Add unit test

* Handle the case of columns without storage

Co-authored-by: ndickson <ndickson@nvidia.com>
Co-authored-by: Xin Yao <xiny@nvidia.com>
2022-06-30 10:10:44 +08:00
Mufei Li 8b19c28788 Update test_transform.py (#4190) 2022-06-29 21:03:23 +08:00
Xin Yao 32f12ee19e [Doc] Unify the minimal versions required for PyTorch/TensorFlow/MXNet (#4180) 2022-06-29 18:37:19 +08:00
Xin Yao 5bef48dfda [Performance] Optimize the use of alternative streams in dataloader (#4177)
* fix using alternative streams

* use a alternative stream for subgraph transferring

* fix StreamContext when stream is None
2022-06-29 16:02:25 +08:00
Rhett Ying 5640b12969 [CI] Upgrade software version of CI docker image (#4189)
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2022-06-29 13:34:13 +08:00
nv-dlasalle 1dddaad4f0 [bugfix] Allow communicators of size one when NCCL is missing (#3713)
* Update nccl communicator for when NCCL is missing

* Use static_cast

* Add doc string

* Fix whitespace

* Resrtict unit test to GPU runs

Co-authored-by: Xin Yao <xiny@nvidia.com>
2022-06-29 11:48:16 +08:00
Mufei Li a25a14f2fa [Bug Fix] Fix A Bug Related to GroupRevRes (#4181)
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
Co-authored-by: Xin Yao <xiny@nvidia.com>
2022-06-28 21:17:20 +08:00
Rhett Ying 1518861148 [BugFix] fix build issue on mac OS (#4175)
* [BugFix] fix build issue on mac OS

* refine
2022-06-28 18:51:02 +08:00
Mufei Li cb39dbfa13 Update (#4178) 2022-06-28 15:44:03 +08:00
Mufei Li 150e92736f [DGL-Go] Inference for Graph Prediction Pipeline (#4157)
* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update
2022-06-28 11:23:18 +08:00
ndickson-nvidia a5d8460c03 [Bug][Feature] Added more missing FP16 specializations (#4140)
* * Added missing specializations for `__half` of `DLDataTypeTraits`, `IndexSelect`, `Full`, `Scatter_`, `CSRGetData`, `CSRMM`, `CSRSum`, `IndexSelectCPUFromGPU`
* Fixed casting issue in `_LinearSearchKernel` that was preventing it from supporting `__half`
* Added `#if`'d out specializations of `CSRGEMM`, `CSRGEAM`, and `Xgeam`, which would require functions that aren't currently provided by cublas

* * Added more specific error messages for unimplemented FP16 specializations of Xgeam, CSRGEMM, and CSRGEAM

* * Added missing instantiation of DLDataTypeTraits<__half>::dtype

* * Fixed linter error
* Added clearer comment explaining why the cast to long long is necessary

* * Worked around a compile error in some particular setup, where __half can't be constructed on the host side

* * Fixed linter formatting errors

* * Changes to comments as recommended

* * Made recommended changes to logging errors in FP16 specializations
* Also changed the existing Xgeam function for unsupported data types from LOG(INFO) to LOG(FATAL)
2022-06-27 14:26:18 -07:00
Xin Yao b8f905f18b [Bugfix] Fix that pin_prefetcher is not actually enabled (#4169) 2022-06-27 16:07:22 +08:00
Rhett Ying 10db5d0b88 [BugFix] fix rpc-related build issue on mac OS (#4168)
* [BugFix] fix rpc-related build issue on mac OS

* add warning message

* add warning message
2022-06-27 13:13:03 +08:00
Rhett Ying 9d425315da [Dist] enable USE_EPOLL in default (#4167)
* [Dist] enable USE_EPOLL in default

* fix build issue on windows

* fix build issue on windows

* fix build issue on windows

* fix build issue on windows

* fix build issue on windows

* fix build issue
2022-06-27 09:12:54 +08:00
PotatoChipsNinja d1f6f3a848 [Doc] fix a bug in guide_cn (#4149)
Co-authored-by: Xin Yao <xiny@nvidia.com>
2022-06-24 13:08:07 +08:00