diff --git a/third_party/nccl/hermetic/nccl_configure.bzl b/third_party/nccl/hermetic/nccl_configure.bzl index 14469acdfc5..c1e49a6b9f1 100644 --- a/third_party/nccl/hermetic/nccl_configure.bzl +++ b/third_party/nccl/hermetic/nccl_configure.bzl @@ -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"], diff --git a/third_party/xla/third_party/tsl/third_party/nccl/hermetic/nccl_configure.bzl b/third_party/xla/third_party/tsl/third_party/nccl/hermetic/nccl_configure.bzl index 14469acdfc5..c1e49a6b9f1 100644 --- a/third_party/xla/third_party/tsl/third_party/nccl/hermetic/nccl_configure.bzl +++ b/third_party/xla/third_party/tsl/third_party/nccl/hermetic/nccl_configure.bzl @@ -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"],