From b8cc6275374c69dbc3c8b7e0564d7ad8b1e7fa51 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 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,
     })