1. 09 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. 17 Feb, 2023 2 commits
  4. 04 Nov, 2022 1 commit
  5. 10 Oct, 2022 1 commit
  6. 23 May, 2022 1 commit
  7. 20 Aug, 2021 1 commit
    • nv-dlasalle's avatar
      [Feature][DistDGL] Add NCCL support for range based partitions (#3213) · 7f927939
      nv-dlasalle authored
      * Implement range based NDArrayPartition
      
      * Finish implement range based partition support
      
      * Add unit test
      
      * Fix whitepace
      
      * Add Kernel suffix
      
      * Fix argument passing
      
      * Add doxygen docs and improve variable naming
      
      * Add unit test
      
      * Add function for converting a partition book
      
      * Add example to partition_op docs
      
      * Fix dtype conversion for mxnet and tensorflow
      7f927939