Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "e65be580ab2ea6dc64e8d5cd2a7396a9b17b16da"
Switch branch/tag
  1. 23 May, 2022 1 commit
  3. 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