commit f70dfea26f6ecc9faccea4800ca28ddfc129e79f
author A. Unique TensorFlower <gardener@tensorflow.org> Fri Jun 29 01:44:59 2018 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Fri Jun 29 01:48:36 2018 -0700
tree 3dd8cef6d15511c14b3d78499b95b4609479747d
parent 6874e1ef40c4189d96c105227f60b507953f95d3

Checking that the NCCL 2 license file is found, see GitHub issue 19679.

PiperOrigin-RevId: 202613754

configure.py

diff --git a/configure.py b/configure.py
index ad585fa..5243e09 100644
--- a/configure.py
+++ b/configure.py
@@ -1134,7 +1134,9 @@
 
     nccl_lib_path = os.path.join(nccl_install_path, nccl_lib_path)
     nccl_hdr_path = os.path.join(nccl_install_path, 'include/nccl.h')
-    if os.path.exists(nccl_lib_path) and os.path.exists(nccl_hdr_path):
+    nccl_license_path = os.path.join(nccl_install_path, 'NCCL-SLA.txt')
+    if os.path.exists(nccl_lib_path) and os.path.exists(
+        nccl_hdr_path) and os.path.exists(nccl_license_path):
       # Set NCCL_INSTALL_PATH
       environ_cp['NCCL_INSTALL_PATH'] = nccl_install_path
       write_action_env_to_bazelrc('NCCL_INSTALL_PATH', nccl_install_path)