mirror/tensorflow
1
0
Fork 0
mirror of https://github.com/tensorflow/tensorflow.git synced 2024-11-21 21:05:19 +00:00
Code Issues Actions 8 Packages Projects Releases Wiki Activity

Add nccl_headers alias to the content of BUILD file in NCCL repository when NCCL stub is not used.

Browse Source 
PiperOrigin-RevId: 676578558
This commit is contained in:
A. Unique TensorFlower 2024-09-19 15:06:22 -07:00 committed by TensorFlower Gardener
parent eda2d321b2
commit cf3b94729e
2 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
third_party/nccl/hermetic/nccl_configure.bzl vendored
View File

@ -60,6 +60,15 @@ alias(
visibility = ["//visibility:public"],
)
alias(
name = "nccl_headers",
actual = select({
"@local_config_cuda//cuda:cuda_tools_and_libs": "@cuda_nccl//:headers",
"//conditions:default": "@nccl_archive//:nccl_headers",
}),
visibility = ["//visibility:public"],
)
cc_library(
name = "hermetic_nccl_config",
hdrs = ["nccl_config.h"],

9
third_party/xla/third_party/tsl/third_party/nccl/hermetic/nccl_configure.bzl vendored
View File

@ -60,6 +60,15 @@ alias(
visibility = ["//visibility:public"],
)
alias(
name = "nccl_headers",
actual = select({
"@local_config_cuda//cuda:cuda_tools_and_libs": "@cuda_nccl//:headers",
"//conditions:default": "@nccl_archive//:nccl_headers",
}),
visibility = ["//visibility:public"],
)
cc_library(
name = "hermetic_nccl_config",
hdrs = ["nccl_config.h"],