experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

internal change

Browse Source 
PiperOrigin-RevId: 204832902
This commit is contained in:
Toby Boyd 2018-07-16 17:04:48 -07:00 committed by TensorFlower Gardener
parent 8e27c65acf
commit e4c0dbcab8
2 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
configure.py
View File

@ -1138,9 +1138,7 @@ 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')
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):
if os.path.exists(nccl_lib_path) and os.path.exists(nccl_hdr_path):
# Set NCCL_INSTALL_PATH
environ_cp['NCCL_INSTALL_PATH'] = nccl_install_path
write_action_env_to_bazelrc('NCCL_INSTALL_PATH', nccl_install_path)

2
third_party/nccl/nccl_configure.bzl vendored
View File

@ -47,10 +47,10 @@ alias(
)
"""
# Local build results in dynamic link and the license should not be included.
_NCCL_LOCAL_BUILD_TEMPLATE = """
filegroup(
name = "LICENSE",
data = ["nccl/NCCL-SLA.txt"],
visibility = ["//visibility:public"],
)