From e151a584a7e3e641c2c5b62d10cf15d2e458ad38 Mon Sep 17 00:00:00 2001
From: Joshua Lang <joshualang@google.com>
Date: Wed, 2 Oct 2024 18:37:36 -0700
Subject: [PATCH] Automated Code Change

PiperOrigin-RevId: 681670724
---
 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 401cf088936d3..6771d68463f97 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 11bdbf53df968..e3e307a9905a9 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,
+        clean_dep("//xla/tsl:is_cuda_enabled"): if_true,
         "@local_config_rocm//rocm:using_hipcc": if_true,
         "//conditions:default": if_false,
     })