1. 25 Jan, 2021 2 commits
    • Zihao Ye's avatar
      [feature] Implement missing CUDA operators for COO format (part 1). (#2565) · 0f9056ed
      Zihao Ye authored
      
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * fix
      
      * upd
      
      * upd
      Co-authored-by: default avatarJinjing Zhou <VoVAllen@users.noreply.github.com>
      0f9056ed
    • Da Zheng's avatar
      [Distributed] Heterogeneous graph support (#2457) · 25ac3344
      Da Zheng authored
      
      
      * Distributed heterograph (#3)
      
      * heterogeneous graph partition.
      
      * fix graph partition book for heterograph.
      
      * load heterograph partitions.
      
      * update DistGraphServer to support heterograph.
      
      * make DistGraph runnable for heterograph.
      
      * partition a graph and store parts with homogeneous graph structure.
      
      * update DistGraph server&client to use homogeneous graph.
      
      * shuffle node Ids based on node types.
      
      * load mag in heterograph.
      
      * fix per-node-type mapping.
      
      * balance node types.
      
      * fix for homogeneous graph
      
      * store etype for now.
      
      * fix data name.
      
      * fix a bug in example.
      
      * add profiler in rgcn.
      
      * heterogeneous RGCN.
      
      * map homogeneous node ids to hetero node ids.
      
      * fix graph partition book.
      
      * fix DistGraph.
      
      * shuffle eids.
      
      * verify eids and their mappings when loading a partition.
      
      * Id map from homogneous Ids to per-type Ids.
      
      * verify partitioned results.
      
      * add test for distributed sampler.
      
      * add mapping from per-type Ids to homogeneous Ids.
      
      * update example.
      
      * fix DistGraph.
      
      * Revert "add profiler in rgcn."
      
      This reverts commit 36daaed8b660933dac8f61a39faec3da2467d676.
      
      * add tests for homogeneous graphs.
      
      * fix a bug.
      
      * fix test.
      
      * fix for one partition.
      
      * fix for standalone training and evaluation.
      
      * small fix.
      
      * fix two bugs.
      
      * initialize projection matrix.
      
      * small fix on RGCN.
      
      * Fix rgcn performance (#17)
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-62-171.ec2.internal>
      
      * fix lint.
      
      * fix lint.
      
      * fix lint.
      
      * fix lint.
      
      * fix lint.
      
      * fix lint.
      
      * fix.
      
      * fix test.
      
      * fix lint.
      
      * test partitions.
      
      * remove redundant test for partitioning.
      
      * remove commented code.
      
      * fix partition.
      
      * fix tests.
      
      * fix RGCN.
      
      * fix test.
      
      * fix test.
      
      * fix test.
      
      * fix.
      
      * fix a bug.
      
      * update dmlc-core.
      
      * fix.
      
      * fix rgcn.
      
      * update readme.
      
      * add comments.
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-2-202.us-west-1.compute.internal>
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-9-132.us-west-1.compute.internal>
      Co-authored-by: default avatarxiang song(charlie.song) <classicxsong@gmail.com>
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-62-171.ec2.internal>
      
      * fix.
      
      * fix.
      
      * add div_int.
      
      * fix.
      
      * fix.
      
      * fix lint.
      
      * fix.
      
      * fix.
      
      * fix.
      
      * adjust.
      
      * move code.
      
      * handle heterograph.
      
      * return pytorch tensor in GPB.
      
      * remove some tests in example.
      
      * add to_block for distributed training.
      
      * use distributed to_block.
      
      * remove unnecessary function in DistGraph.
      
      * remove distributed to_block.
      
      * use pytorch tensor.
      
      * fix a bug in ntypes and etypes.
      
      * enable norm.
      
      * make the data loader compatible with the old format.
      
      * fix.
      
      * add comments.
      
      * fix a bug.
      
      * add test for heterograph.
      
      * support partition without reshuffle.
      
      * add test.
      
      * support partition without reshuffle.
      
      * fix.
      
      * add test.
      
      * fix bugs.
      
      * fix lint.
      
      * fix dataset.
      
      * fix for mxnet.
      
      * update docstring.
      
      * rename to floor_div
      
      * avoid exposing NodePartitionPolicy and EdgePartitionPolicy.
      
      * fix docstring.
      
      * fix error.
      
      * fixes.
      
      * fix comments.
      
      * rename.
      
      * rename.
      
      * explain IdMap.
      
      * fix docstring.
      
      * fix docstring.
      
      * update docstring.
      
      * remove the code of returning heterograph.
      
      * remove argument.
      
      * fix example.
      
      * make GraphPartitionBook an abstract class.
      
      * fix.
      
      * fix.
      
      * fix a bug.
      
      * fix a bug in example
      
      * fix a bug
      
      * reverse heterograph sampling.
      
      * temp fix.
      
      * fix lint.
      
      * Revert "temp fix."
      
      This reverts commit c450717b9f578b8c48769c675f2a19d6c1e64381.
      
      * compute norm.
      
      * Revert "reverse heterograph sampling."
      
      This reverts commit bd6deb7f52998de76508f800441ff518e2fadcb9.
      
      * fix.
      
      * move id_map.py
      
      * remove check
      
      * add more comments.
      
      * update docstring.
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-2-202.us-west-1.compute.internal>
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-9-132.us-west-1.compute.internal>
      Co-authored-by: default avatarxiang song(charlie.song) <classicxsong@gmail.com>
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-62-171.ec2.internal>
      25ac3344
  2. 21 Jan, 2021 1 commit
  3. 14 Jan, 2021 1 commit
  4. 31 Dec, 2020 1 commit
  5. 26 Dec, 2020 2 commits
  6. 25 Dec, 2020 1 commit
    • Quan (Andy) Gan's avatar
      [Performance] Use allocator from PyTorch if possible (#2328) · 9a7235fa
      Quan (Andy) Gan authored
      * first commit
      
      * some thoughts
      
      * move around
      
      * more commit
      
      * more fixes
      
      * now it uses torch allocator
      
      * fix symbol export error
      
      * fix
      
      * fixes
      
      * test fix
      
      * add script
      
      * building separate library per version
      
      * fix for vs2019
      
      * more fixes
      
      * fix on windows build
      
      * update jenkinsfile
      
      * auto copy built dlls for windows
      
      * lint and installation guide update
      
      * fix
      
      * specify conda environment
      
      * set environment for ci
      
      * fix
      
      * fix
      
      * fix
      
      * fix again
      
      * revert
      
      * fix cmake
      
      * fix
      
      * switch to using python interpreter path
      
      * remove scripts
      
      * debug
      
      * oops sorry
      
      * Update index.rst
      
      * Update index.rst
      
      * copies automatically, no need for this
      
      * do not print message if library not found
      
      * tiny fixes
      
      * debug on nightly
      
      * replace add_compile_definitions to make CMake 3.5 happy
      
      * fix linking to wrong lib for multiple pytorch envs
      
      * changed building strategy
      
      * fix nightly
      
      * fix windows
      
      * fix windows again
      
      * setup bugfix
      
      * address comments
      
      * change README
      9a7235fa
  7. 17 Dec, 2020 1 commit
  8. 15 Dec, 2020 1 commit
  9. 10 Dec, 2020 1 commit
  10. 09 Dec, 2020 1 commit
  11. 27 Nov, 2020 1 commit
  12. 26 Nov, 2020 2 commits
  13. 22 Nov, 2020 1 commit
  14. 17 Nov, 2020 2 commits
  15. 13 Nov, 2020 1 commit
  16. 12 Nov, 2020 2 commits
  17. 06 Nov, 2020 1 commit
  18. 02 Nov, 2020 1 commit
  19. 30 Oct, 2020 1 commit
    • nv-dlasalle's avatar
      [Dataloading] Add class for copying tensors to/from the GPU on a non-default stream (#2284) · f673fc25
      nv-dlasalle authored
      * Add async transferer class
      
      * Add async ndarray copy interface
      
      * Add python bindings
      
      * Fix comment
      
      * Add python class
      
      * Fix linting issues
      
      * Add python unit test
      
      * Update python interface
      
      * move async_transferer to cuda only directory
      
      * Fix linting issue
      
      * Move out of contrib
      
      * Add doc strings
      
      * Move test compute from backend
      
      * Update comment
      
      * Fix test naming
      
      * Fix argument usage
      
      * Wrap/unwrap backend parameters
      
      * Move to dataloading
      
      * Move to 'dataloading'
      
      * Make GPU/CPU compatible
      
      * Fix unit tests
      
      * Add docs
      
      * Use only backend interface for datamovement in unit test
      f673fc25
  20. 17 Oct, 2020 1 commit
  21. 09 Oct, 2020 1 commit
  22. 25 Sep, 2020 1 commit
  23. 15 Sep, 2020 1 commit
  24. 11 Sep, 2020 1 commit
  25. 10 Sep, 2020 4 commits
  26. 30 Aug, 2020 1 commit
  27. 28 Aug, 2020 1 commit
  28. 27 Aug, 2020 2 commits
  29. 19 Aug, 2020 1 commit
  30. 17 Aug, 2020 1 commit
    • Mufei Li's avatar
      [Doc/Feature] Refactor, doc update and behavior fix for graphs (#1983) · be444e52
      Mufei Li authored
      
      
      * Update graph
      
      * Fix for dgl.graph
      
      * from_scipy
      
      * Replace canonical_etypes with relations
      
      * from_networkx
      
      * Update for hetero_from_relations
      
      * Roll back the change of canonical_etypes to relations
      
      * heterograph
      
      * bipartite
      
      * Update doc
      
      * Fix lint
      
      * Fix lint
      
      * Fix test cases
      
      * Fix
      
      * Fix
      
      * Fix
      
      * Fix
      
      * Fix
      
      * Fix
      
      * Update
      
      * Fix test
      
      * Fix
      
      * Update
      
      * Use DGLError
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Fix
      
      * Fix
      
      * Fix
      
      * Fix
      
      * Fix
      
      * Fix
      
      * Fix
      
      * Fix
      
      * Update
      
      * Fix
      
      * Update
      
      * Fix
      
      * Fix
      
      * Fix
      
      * Update
      
      * Fix
      
      * Update
      
      * Fix
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Fix
      
      * Fix
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * rewrite sanity checks
      
      * delete unnecessary checks
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Fix
      
      * Update
      
      * Update
      
      * Update
      
      * Fix
      
      * Fix
      
      * Fix
      
      * Update
      
      * Fix
      
      * Update
      
      * Fix
      
      * Fix
      
      * Update
      
      * Fix
      
      * Update
      
      * Fix
      Co-authored-by: default avatarxiang song(charlie.song) <classicxsong@gmail.com>
      Co-authored-by: default avatarMinjie Wang <wmjlyjemaine@gmail.com>
      Co-authored-by: default avatarQuan Gan <coin2028@hotmail.com>
      be444e52
  31. 14 Aug, 2020 1 commit