- 14 Jun, 2022 5 commits
-
-
nv-dlasalle authored
* Disable non-atomic atomic operations * Improve error message * Make error message more friendly
-
彭齐荣 authored
* 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>
-
Rhett Ying authored
-
RecLusIve-F authored
* 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>
-
Rhett Ying authored
* [Dist] master port should be fixed for all trainers * add tests for tools/launch.py
-
- 12 Jun, 2022 2 commits
-
-
Huarui HE authored
* 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>
-
Quan (Andy) Gan authored
-
- 11 Jun, 2022 1 commit
-
-
Xin Yao authored
* 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>
-
- 09 Jun, 2022 3 commits
-
-
Rhett Ying authored
-
Rhett Ying authored
-
Chang Liu authored
[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
-
- 08 Jun, 2022 4 commits
-
-
Rhett Ying authored
* [ist] enable time out when fetching msg * fix lint error * minor refinements * improve minor log * fix dist test * fix timeout issue in tensorpipe
-
Rhett Ying authored
* [DistTest] add python test of RPC * remove return
-
Rhett Ying authored
* [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>
-
ndickson-nvidia authored
-
- 07 Jun, 2022 2 commits
-
-
ndickson-nvidia authored
* * 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>
-
Tudor Andrei Dumitrascu authored
* 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>
-
- 06 Jun, 2022 3 commits
-
-
ndickson-nvidia authored
* * 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
-
Quan (Andy) Gan authored
Co-authored-by:Xin Yao <xiny@nvidia.com>
-
Xin Yao authored
Co-authored-by:
nv-dlasalle <63612878+nv-dlasalle@users.noreply.github.com> Co-authored-by:
Israt Nisa <neesha295@gmail.com>
-
- 04 Jun, 2022 3 commits
-
-
Sai Kandregula authored
* 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>
-
Mufei Li authored
* 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>
-
- 03 Jun, 2022 3 commits
-
-
RecLusIve-F authored
* Add BGRL example * Update README.md * Update utils.py * Update * Update utils.py
-
Sai Kandregula authored
Co-authored-by:
decoherencer <decoherencer@users.noreply.github.com> Co-authored-by:
Mufei Li <mufeili1996@gmail.com>
-
Riju Mukherjee authored
* 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>
-
- 02 Jun, 2022 4 commits
-
-
Xin Zhang authored
Co-authored-by:Mufei Li <mufeili1996@gmail.com>
-
RecLusIve-F authored
* 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>
-
Chang Liu authored
Co-authored-by:Mufei Li <mufeili1996@gmail.com>
-
Mufei Li authored
* Update * CI * Update * Update * Fix * Fix
-
- 28 May, 2022 4 commits
-
-
Quan (Andy) Gan authored
* change warning message * Update tensordispatch.cc
-
Quan (Andy) Gan authored
This reverts commit fdd1fe19.
-
Quan (Andy) Gan authored
-
Sai Kandregula authored
Co-authored-by:decoherencer <decoherencer@users.noreply.github.com>
-
- 27 May, 2022 1 commit
-
-
Mufei Li authored
* Update * Update * Update dgl.data.rst * CI
-
- 26 May, 2022 2 commits
-
-
nv-dlasalle authored
* 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
-
彭齐荣 authored
-
- 25 May, 2022 1 commit
-
-
Minjie Wang authored
* cython nogil * move APIs to internal and add unit test * fix lint * disable callback array test
-
- 24 May, 2022 1 commit
-
-
Mufei Li authored
* Update module.py * Update utils.py * Update utils.py * Update utils.py * Update module.py * Update * Update * Update
-
- 23 May, 2022 1 commit
-
-
Quan (Andy) Gan authored
Co-authored-by:Mufei Li <mufeili1996@gmail.com>
-