项目文件夹

文件

T

文件历史

nv-dlasalle ae8dbe6d3c [Feature][Performance] Implement NCCL wrapper for communicating NodeEmbeddings and sparse gradients. (#2825 )

* Split NCCL wrapper from sparse optimizer and sparse embedding

* Add more unit tests for single node nccl

* Fix unit test for tf

* Switch to device histogram

* Fix histgram issues

* Finish migration to histogram

* Handle cases with zero send/recieve data

* Start on partition object

* Get compiling

* Updates

* Add unit tests

* Switch to partition object

* Fix linting issues

* Rename partition file

* Add python doc

* Fix python assert and finish doxygen comments

* Remove stubs for range based partition to satisfy pylint

* Wrap unit test in GPU only

* Wrap explicit cuda call in ifdef

* Merge with partition.py

* update docstrings

* Cleanup partition_op

* Add Workspace object

* Switch to using workspace object

* Move last remainder based function out of nccl_api

* Add error messages

* Update docs with examples

* Fix linting erros

Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>

2021-05-20 10:58:17 -07:00

FindCUDA.cmake

[Feature] Add edge coarsening for homogeneous undirected graphs (#2691 )

2021-03-09 16:35:27 +08:00

FindNccl.cmake

[Feature][Performance] Implement NCCL wrapper for communicating NodeEmbeddings and sparse gradients. (#2825 )

2021-05-20 10:58:17 -07:00

MshadowUtil.cmake

[Feature][Kernel] DGL kernel support (#596 )

2019-06-06 15:47:55 -04:00

Util.cmake

[Feature][Kernel] DGL kernel support (#596 )

2019-06-06 15:47:55 -04:00