1. 23 Jun, 2021 1 commit
    • Quan (Andy) Gan's avatar
      [Feature] Node2vec (#2992) · e667545d
      Quan (Andy) Gan authored
      
      
      * add seal example
      
      * 1. add paper infomation in examples/README
      2. adjust codes
      3. option test
      
      * use latest `to_simple` to replace coalesce graph function
      
      * remove outdated codes
      
      * remove useless comment
      
      * Node2vec
      1.implement node2vec random walk c++ op
      2.implement node2vec model
      3.implement node2vec example
      
      * add CMakeLists file modify
      
      * refine c++ codes
      
      * refine c++ codes
      
      * add missing whitespace
      
      * refine python codes
      
      * add codes
      
      * add node2vec_impl.h
      
      * fix codes
      
      * fix code style problem
      
      * fixes
      
      * remove
      
      * lots of changes
      
      * add benchmark
      
      * fixes
      Co-authored-by: default avatarsmilexuhc <smile.xuhc@gmail.com>
      Co-authored-by: default avatarMinjie Wang <wmjlyjemaine@gmail.com>
      e667545d
  2. 21 Jun, 2021 1 commit
  3. 16 Jun, 2021 4 commits
    • Da Zheng's avatar
      [Distributed] Support hierarchical partitioning (#3000) · aaec3d8a
      Da Zheng authored
      
      
      * add.
      
      * fix.
      
      * fix.
      
      * fix.
      
      * fix.
      
      * add tests.
      
      * support node split and edge split.
      
      * support 1 partition.
      
      * add tests.
      
      * fix.
      
      * fix test.
      
      * use hierarchical partition.
      
      * add check.
      Co-authored-by: default avatarZheng <dzzhen@3c22fba32af5.ant.amazon.com>
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-22-57.us-west-2.compute.internal>
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-71-112.ec2.internal>
      aaec3d8a
    • Hengrui Zhang's avatar
      fix bugs (#3008) · 18eaad17
      Hengrui Zhang authored
      
      Co-authored-by: default avatarzhjwy9343 <6593865@qq.com>
      18eaad17
    • Hengrui Zhang's avatar
      36c6c649
    • espylapiza's avatar
      [Example] update ogbn-arxiv and ogbn-proteins results (#3018) · 305d5c16
      espylapiza authored
      
      
      * [Example] GCN on ogbn-arxiv dataset
      
      * Add README.md
      
      * Update GCN implementation on ogbn-arxiv
      
      * Update GCN on ogbn-arxiv
      
      * Fix typo
      
      * Use evaluator to get results
      
      * Fix duplicated
      
      * Fix duplicated
      
      * Update GCN on ogbn-arxiv
      
      * Add GAT for ogbn-arxiv
      
      * Update README.md
      
      * Update GAT on ogbn-arxiv
      
      * Update README.md
      
      * Update README
      
      * Update README.md
      
      * Update README.md
      
      * Add GAT implementation for ogbn-proteins
      
      * Add GAT implementation for ogbn-proteins
      
      * Add GAT implementation for ogbn-proteins
      
      * Add GAT implementation for ogbn-proteins
      
      * Add GAT implementation for ogbn-proteins
      
      * Update examples for ogbn-arxiv
      
      * Update examples for ogbn-arxiv
      
      * Update examples for ogbn-arxiv
      
      * Update examples for ogbn-arxiv
      
      * Update examples for ogbn-arxiv
      
      * Update examples for ogbn-arxiv
      
      * Update examples for ogbn-arxiv
      
      * Update examples for ogbn-arxiv
      
      * Update examples for ogbn-arxiv
      
      * Update examples for ogbn-arxiv
      
      * Update examples for ogbn-arxiv
      
      * Update examples for ogbn-arxiv
      
      * Update examples for ogbn-arxiv
      
      * Update examples for ogbn-arxiv
      
      * Add examples for ogbn-products.
      
      * Update examples for ogbn-proteins.
      
      * Update examples for ogbn-proteins.
      
      * Update examples for ogbn-proteins.
      
      * Update examples for ogbn-proteins.
      
      * Update examples for ogbn-proteins.
      
      * Update examples for ogbn-proteins.
      
      * Update examples for ogbn-proteins.
      
      * Update examples for ogbn-proteins.
      
      * Update examples for ogbn-proteins.
      
      * Update examples for ogbn-products.
      
      * Update examples for ogbn-products.
      
      * Update examples for ogbn-arxiv.
      
      * Update examples for ogbn-arxiv.
      
      * Update examples for ogbn-arxiv.
      
      * Update examples for ogbn-proteins.
      
      * Update examples for ogbn-products.
      
      * Update examples for ogbn-proteins.
      
      * Update examples for ogbn-arxiv.
      
      * Update examples for ogbn-proteins.
      
      * Update README.md
      
      * [Example] ogbn-arxiv & ogbn-proteins
      
      * [Example] ogbn-arxiv & ogbn-proteins
      
      * [Example] ogbn-arxiv & ogbn-proteins
      
      * [Example] ogbn-arxiv & ogbn-proteins
      Co-authored-by: default avatarMufei Li <mufeili1996@gmail.com>
      Co-authored-by: default avatarZihao Ye <expye@outlook.com>
      305d5c16
  4. 15 Jun, 2021 1 commit
  5. 13 Jun, 2021 2 commits
  6. 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
  7. 10 Jun, 2021 1 commit
  8. 07 Jun, 2021 1 commit
  9. 04 Jun, 2021 1 commit
  10. 03 Jun, 2021 1 commit
  11. 02 Jun, 2021 1 commit
  12. 26 May, 2021 1 commit
  13. 17 May, 2021 2 commits
  14. 14 May, 2021 1 commit
  15. 12 May, 2021 1 commit
  16. 11 May, 2021 2 commits
  17. 03 May, 2021 1 commit
  18. 27 Apr, 2021 1 commit
  19. 26 Apr, 2021 1 commit
  20. 25 Apr, 2021 1 commit
  21. 23 Apr, 2021 2 commits
  22. 21 Apr, 2021 1 commit
  23. 20 Apr, 2021 1 commit
  24. 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
  25. 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
  26. 13 Apr, 2021 1 commit
  27. 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
  28. 11 Apr, 2021 1 commit
  29. 08 Apr, 2021 1 commit
  30. 07 Apr, 2021 2 commits
  31. 06 Apr, 2021 1 commit
  32. 05 Apr, 2021 1 commit