Automated Code Change

PiperOrigin-RevId: 681670724
openxla · Oct 3, 2024 · e151a58 · e151a58
1 parent da687fc
commit e151a58
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/xla/tsl/BUILD b/xla/tsl/BUILD
@@ -39,7 +39,7 @@ alias(
     name = "is_cuda_enabled",
     actual = if_oss(
         "@local_config_cuda//:is_cuda_enabled",
-        "@local_config_cuda//cuda:using_clang",
+        "@local_config_cuda//cuda:using_config_cuda",
     ),
     visibility = ["//visibility:public"],
 )

diff --git a/xla/tsl/tsl.bzl b/xla/tsl/tsl.bzl
@@ -87,8 +87,7 @@ def if_cuda_or_rocm(if_true, if_false = []):
 
       """
     return select({
-        "@local_config_cuda//cuda:using_nvcc": if_true,
-        "@local_config_cuda//cuda:using_clang": if_true,
+        clean_dep("//xla/tsl:is_cuda_enabled"): if_true,
         "@local_config_rocm//rocm:using_hipcc": if_true,
         "//conditions:default": if_false,
     })