Skip to content

Commit

Permalink
Update no_nccl_support config_setting.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 679279480
  • Loading branch information
IllogicalMoose authored and Google-ML-Automation committed Oct 2, 2024
1 parent dfb99ba commit 8f56c1a
Showing 1 changed file with 12 additions and 3 deletions.
15 changes: 12 additions & 3 deletions xla/tsl/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ load("@bazel_skylib//lib:selects.bzl", "selects")
load("@bazel_skylib//rules:common_settings.bzl", "bool_flag", "bool_setting")
load("@tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//xla/tsl:package_groups.bzl", "tsl_package_groups")
load("//xla/tsl:tsl.bzl", "if_google", "if_oss")
load("//xla/tsl:tsl.bzl", "if_google", "if_oss", "internal_visibility")
load(
"//xla/tsl:tsl.default.bzl",
"tsl_extra_config_settings",
Expand Down Expand Up @@ -500,11 +500,20 @@ config_setting(
)

config_setting(
name = "no_nccl_support",
name = "using_no_nccl_support_define",
define_values = dict(
if_google({"GOOGLE_CUDA_COMPILER": "clang"}),
no_nccl_support = "true",
),
visibility = internal_visibility(["//visibility:private"]),
)

selects.config_setting_group(
name = "no_nccl_support",
match_all = [
":using_no_nccl_support_define",
] + if_google([
"@local_config_cuda//cuda:using_config_cuda",
]),
visibility = ["//visibility:public"],
)

Expand Down

0 comments on commit 8f56c1a

Please sign in to comment.