From 9e46e0a85eb03c9e01d09dc827012127f6bf830f Mon Sep 17 00:00:00 2001
From: Joshua Lang <joshualang@google.com>
Date: Mon, 30 Sep 2024 11:23:39 -0700
Subject: [PATCH] Automated Code Change

PiperOrigin-RevId: 680652154
---
 xla/tsl/BUILD   | 2 +-
 xla/tsl/tsl.bzl | 3 +--
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/xla/tsl/BUILD b/xla/tsl/BUILD
index 401cf088936d31..6771d68463f978 100644
--- 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
index 11bdbf53df968c..cd5f74f6a55ec1 100644
--- 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,
+        "//xla/tsl:is_cuda_enabled": if_true,
         "@local_config_rocm//rocm:using_hipcc": if_true,
         "//conditions:default": if_false,
     })