Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "020259895384a1f99844ce6e04f0a226d7b3210b"
Switch branch/tag
  1. 11 Jun, 2021 1 commit
    • nv-dlasalle's avatar
      [Feature] Allow using NCCL for communication in dgl.NodeEmbedding and dgl.SparseOptimizer (#2824) · 17d604b5
      nv-dlasalle authored 
      

* Split from NCCL PR

* Fix type in comment

* Expand documentation for sparse_all_to_all_push

* Restore previous behavior in example

* Re-work optimizer to use NCCL based on gradient location

* Allow for running with embedding on CPU but using NCCL for gradient exchange

* Optimize single partition case

* Fix pylint errors

* Add missing include

* fix gradient indexing

* Fix line continuation

* Migrate 'first_step'

* Skip tests without enough GPUs to run NCCL

* Improve empty tensor handling for pytorch 1.5

* Fix indentation

* Allow multiple NCCL communicator to coexist

* Improve handling of empty message

* Update python/dgl/nn/pytorch/sparse_emb.py
Co-authored-by: default avatarxiang song(charlie.song) <classicxsong@gmail.com>

* Update python/dgl/nn/pytorch/sparse_emb.py
Co-authored-by: default avatarxiang song(charlie.song) <classicxsong@gmail.com>

* Keepy empty tensor dimensionaless

* th.empty -> th.tensor

* Preserve shape for empty non-zero dimension tensors

* Use shared state, when embedding is shared

* Add support for gathering an embedding

* Fix typo

* Fix more typos

* Fix backend call

* Use NodeDataLoader to take advantage of ddp

* Update training script to share memory

* Only squeeze last dimension

* Better handle empty message

* Keep embedding on the target device GPU if dgl_sparse if false in RGCN example

* Fix typo in comment

* Add asserts

* Improve documentation in example
Co-authored-by: default avatarxiang song(charlie.song) <classicxsong@gmail.com>
      17d604b5
  3. 10 Jun, 2021 1 commit
  5. 07 Jun, 2021 1 commit
  7. 04 Jun, 2021 1 commit
  9. 03 Jun, 2021 1 commit
  11. 02 Jun, 2021 1 commit
  13. 26 May, 2021 1 commit
  15. 17 May, 2021 2 commits
  17. 14 May, 2021 1 commit
  19. 12 May, 2021 1 commit
  21. 11 May, 2021 2 commits
  23. 03 May, 2021 1 commit
  25. 27 Apr, 2021 1 commit
  27. 26 Apr, 2021 1 commit
  29. 25 Apr, 2021 1 commit
  31. 23 Apr, 2021 2 commits
  33. 21 Apr, 2021 1 commit
  35. 20 Apr, 2021 1 commit
  37. 19 Apr, 2021 1 commit
    • Tingzhang Zhao's avatar
      [Model] Add RECT example (#2813) · e58eeebf
      Tingzhang Zhao authored 
      

* Update README.md

Add description of RECT

* [Example] Add implementation of RECT

[Example] Add implementation of RECT

* Update classify.py

Modify the class names and the function names mentioned above

* Update main.py

Modify the function names mentioned above

* Update label_utils.py

Adjust the comments

* Update README.md

Add the github information
Co-authored-by: default avatarTianjun Xiao <xiaotj1990327@gmail.com>
      e58eeebf
  39. 15 Apr, 2021 1 commit
    • nv-dlasalle's avatar
      [Performance][GPU] Enable GPU uniform edge sampling (#2716) · e70138bb
      nv-dlasalle authored 
      

* Start on uniform GPU sampling

* Save more work

* Get cu file compiling

* Update sampling

* More changes

* Get GPU sampling for uniform probabilities solved

* Fix batch tensor migration

* Fix

* update kernels

* expand blocking

* Undo testing change

* Cut down on sampling overhead

* Fix replacement

* Update unit tests

* Add option to gpu sample in graphsage

* Copy only csc to gpu

* Add ogbn support

* Fix linting

* Remove nvtx from sample

* Improve documentation and error checking

* Expand documentation

* Update assert checking

* delete extra space

* Use standard dataloader when dataset is a dictionary

* ogb -> ogbn

* Fix edge selection determinism

* Fix typos

* Remove nvtx

* Add comment for self.fanout_arrays and assert

* Fix linting

* Migrate to scalarbatcher

* Fix indentation

* Fix batcher

* Fix indexing

* Only use databatcher for GPU

* Convert to DGL NDArray to PyTorch Tensor

* Add optimization for PyTorch's F.tensor() for list of GPU tensors
Co-authored-by: default avatarDa Zheng <zhengda1936@gmail.com>
      e70138bb
  41. 13 Apr, 2021 1 commit
  43. 12 Apr, 2021 1 commit
    • Chen Sirui's avatar
      Graphsim (#2794) · afc83aa2
      Chen Sirui authored 
      

* Add hgat example

* Add experiment

* Clean code

* clear the code

* Add index in README

* Add index in README

* Add index in README

* Add index in README

* Add index in README

* Add index in README

* Change the code title and folder name

* Ready to merge

* Prepare for rebase and change message passing function

* use git ignore to handle empty file

* change file permission to resolve empty file

* Change permission

* change file mode

* Finish Coding

* working code cpu

* pyg compare

* Accelerate with batching

* FastMode Enabled

* update readme

* Update README.md

* refractor code

* add graphsim code

* modified code

* few fix

* Modified graphsim

* Simple Model Added

* Clean up code

* Refractor the code for Merge

* Bugfix enable gradient when train

* update readme and format
Co-authored-by: default avatarChen <chesirui@3c22fbe5458c.ant.amazon.com>
Co-authored-by: default avatarTianjun Xiao <xiaotj1990327@gmail.com>
Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-4-63.ap-northeast-1.compute.internal>
Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-45-47.ap-northeast-1.compute.internal>
      afc83aa2
  45. 11 Apr, 2021 1 commit
  47. 08 Apr, 2021 1 commit
  49. 07 Apr, 2021 2 commits
  51. 06 Apr, 2021 1 commit
  53. 05 Apr, 2021 2 commits
  55. 03 Apr, 2021 1 commit
  57. 30 Mar, 2021 1 commit
  59. 29 Mar, 2021 2 commits
  61. 27 Mar, 2021 1 commit
  63. 23 Mar, 2021 1 commit
  65. 22 Mar, 2021 2 commits