1. 22 Mar, 2023 1 commit
  2. 08 Mar, 2023 1 commit
    • Xin Yao's avatar
      [Refactor] Replace third_party/nccl with PyTorch's NCCL backend (#4989) · 8d5d8962
      Xin Yao authored
      * expose GeneratePermutation
      
      * add sparse_all_to_all_push
      
      * add sparse_all_to_all_pull
      
      * add unit test
      
      * handle world_size=1
      
      * remove python nccl wrapper
      
      * remove the nccl dependency
      
      * use pinned memory to speedup D2H copy
      
      * fix lint
      
      * resolve comments
      
      * fix lint
      
      * fix ut
      
      * resolve comments
      8d5d8962
  3. 01 Mar, 2023 1 commit
  4. 21 Feb, 2023 11 commits
  5. 17 Feb, 2023 2 commits
  6. 16 Feb, 2023 1 commit
  7. 14 Feb, 2023 1 commit
  8. 13 Feb, 2023 1 commit
    • Quan (Andy) Gan's avatar
      enable sparse on windows and mac (#5277) · f62669b0
      Quan (Andy) Gan authored
      
      
      * enable sparse on windows and mac
      
      * that was stupid
      
      * let's see what's going on..
      
      * [Sparse] Fix the import error on Mac OS.
      
      When using template functions that are defined in source files from DGL,
      the loader of MacOS somehow cannot find their definitions. This fix simply
      avoids depending on template functions from DGL headers.
      
      With this fix, the sparse tests all pass on the MAC environment.
      
      * ok this is the problem
      
      * make errors clearer
      
      * uh
      
      * test
      
      * Update __init__.py
      
      * disabling ddp on windows
      
      ---------
      Co-authored-by: default avatarczkkkkkk <zekucai@gmail.com>
      f62669b0
  9. 10 Feb, 2023 1 commit
  10. 16 Jan, 2023 1 commit
  11. 06 Jan, 2023 1 commit
  12. 29 Dec, 2022 1 commit
  13. 15 Dec, 2022 1 commit
    • Rhett Ying's avatar
      [Dist] enable to chunk node/edge data into arbitrary number of chunks (#4930) · 9731e023
      Rhett Ying authored
      
      
      * [Dist] enable to chunk node/edge data into arbitrary number of chunks
      
      * [Dist] enable to split node/edge data into arbitrary parts
      
      * refine code
      
      * Format boolean to uint8 forcely to avoid dist.scatter failure
      
      * convert boolean to int8 before scatter and revert it after scatter
      
      * refine code
      
      * fix test
      
      * refine code
      
      * move test utilities into utils.py
      
      * update comment
      
      * fix empty data
      
      * update
      
      * update
      
      * fix empty data issue
      
      * release unnecessary mem
      
      * release unnecessary mem
      
      * release unnecessary mem
      
      * release unnecessary mem
      
      * release unnecessary mem
      
      * remove unnecessary shuffle data
      
      * separate array_split into standalone utility
      
      * add example
      Co-authored-by: default avatarxiang song(charlie.song) <classicxsong@gmail.com>
      9731e023
  14. 07 Nov, 2022 1 commit
    • czkkkkkk's avatar
      [Sparse] Add sparse matrix C++ implementation (#4773) · 7c059e86
      czkkkkkk authored
      * [Sparse] Add sparse matrix C++ implementation
      
      * Add documentation
      
      * Update
      
      * Minor fix
      
      * Move Python code to dgl/mock_sparse2
      
      * Move headers to include
      
      * lint
      
      * Update
      
      * Add dgl_sparse directory
      
      * Move src code to dgl_sparse
      
      * Add __init__.py in tests to avoid naming conflict
      
      * Add dgl sparse so in Jenkinsfile
      
      * Complete docstring & SparseMatrix basic op
      
      * lint
      
      * Disable win tests
      7c059e86
  15. 06 Nov, 2022 1 commit
    • Xin Yao's avatar
      [Feature] Add bfloat16 (bf16) support (#4648) · 96297fb8
      Xin Yao authored
      * add bf16 specializations
      
      * remove SWITCH_BITS
      
      * enable amp for bf16
      
      * remove SWITCH_BITS for cpu kernels
      
      * enbale bf16 based on CUDART
      
      * fix compiling for sm<80
      
      * fix cpu build
      
      * enable unit tests
      
      * update doc
      
      * disable test for CUDA < 11.0
      
      * address comments
      
      * address comments
      96297fb8
  16. 11 Oct, 2022 1 commit
  17. 22 Sep, 2022 1 commit
  18. 21 Sep, 2022 1 commit
  19. 16 Sep, 2022 2 commits
  20. 31 Aug, 2022 1 commit
  21. 19 Aug, 2022 1 commit
  22. 17 Aug, 2022 2 commits
  23. 30 Jul, 2022 1 commit
  24. 28 Jul, 2022 1 commit
  25. 14 Jul, 2022 1 commit
  26. 05 Jul, 2022 2 commits
    • Chang Liu's avatar
      [Example][Refactor] Refactor GCN example (#4160) · 885be178
      Chang Liu authored
      
      
      * 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: default avatarXin Yao <xiny@nvidia.com>
      885be178
    • Rhett Ying's avatar
      [CI] Add new CI stage for testing cugraph (#4171) · 85f28117
      Rhett Ying authored
      * [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
      85f28117