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

PiperOrigin-RevId: 202613754
STCYC · Jun 29, 2018 · f70dfea · f70dfea
1 parent 6874e1e
commit f70dfea
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/configure.py b/configure.py
@@ -1134,7 +1134,9 @@ def set_tf_nccl_install_path(environ_cp):
 
     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)