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

Merge pull request #28408 from xinan-jiang:dev/fix_nccl_build

Browse Source 
PiperOrigin-RevId: 247561933
This commit is contained in:
TensorFlower Gardener 2019-05-09 23:58:54 -07:00
commit 222219497c
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
third_party/nccl/nccl_configure.bzl vendored
View File

@ -91,7 +91,12 @@ def _nccl_configure_impl(repository_ctx):
else:
# Create target for locally installed 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(
implementation = _nccl_configure_impl,