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

fix nccl build bug

Browse Source
This commit is contained in:
Xinan Jiang 2019-05-05 15:45:19 +08:00
parent b14c390fc8
commit 2112f2e9d6
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
third_party/nccl/nccl_configure.bzl vendored
View File

@ -91,7 +91,11 @@ def _nccl_configure_impl(repository_ctx):
else: else:
# Create target for locally installed NCCL. # Create target for locally installed NCCL.
config = find_cuda_config(repository_ctx, ["nccl"]) config = find_cuda_config(repository_ctx, ["nccl"])
repository_ctx.template("BUILD", _label("system.BUILD.tpl"), config) config_wrap = {
"%{nccl_version}": config["nccl_version"],
"%{nccl_header_dir}": config["nccl_include_dir"],
"%{nccl_library_dir}": config["nccl_library_dir"]}
repository_ctx.template("BUILD", _label("system.BUILD.tpl"), config_wrap)
nccl_configure = repository_rule( nccl_configure = repository_rule(
implementation = _nccl_configure_impl, implementation = _nccl_configure_impl,