projects / platform / upstream / tensorflow.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: dfa9921) | patch
Add support for NCCL2. The configure script asks for what version of NCCL to use...
authorA. Unique TensorFlower <gardener@tensorflow.org>
Thu, 5 Apr 2018 10:09:27 +0000 (03:09 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Thu, 5 Apr 2018 10:11:33 +0000 (03:11 -0700)
commit1fda7645d132b71b9084b01945795e97e582adcd
tree7ae6dfa150c4b58dc339db0ae1f0b177c482433etree | snapshot
parentdfa9921e6343727b05f42f8d4a918b19528ff994commit | diff
Add support for NCCL2. The configure script asks for what version of NCCL to use. The default is still NCCL 1 from GitHub. If the user chooses NCCL 2, it asks for the install directory.

The nccl_configure.bzl generates two different BUILD files based on the chose NCCL version. For NCCL 1, it aliases to the existing 'nccl_archive' http_repo on GitHub. For NCCL 2, it creates a target containing the NCCL 2 library and headers from the chosen install directory.

PiperOrigin-RevId: 191718007
configure.py diff | blob | history
tensorflow/contrib/nccl/BUILD diff | blob | history
tensorflow/contrib/nccl/kernels/nccl_manager.h diff | blob | history
tensorflow/contrib/nccl/kernels/nccl_ops.cc diff | blob | history
tensorflow/tools/pip_package/BUILD diff | blob | history
tensorflow/workspace.bzl diff | blob | history
third_party/nccl/LICENSE [new file with mode: 0644] blob
third_party/nccl/nccl_archive.BUILD [moved from third_party/nccl.BUILD with 95% similarity] diff | blob | history
third_party/nccl/nccl_configure.bzl [new file with mode: 0644] blob
Domain: Machine Learning / ML Framework;