Remove cuda_only_cc_library

Since now we can exclude targets from building using tags, we won't need the `cuda_only_cc_library` rule anymore.

This also required me to remove some wrongly added dependencies, notably
I found several targets depending on `cublas_plugin`, even though those targets
were not CUDA specific and shouldn't directly depend on CUDA-specific targets.

I also found out that the `tsl_gpu_library` macro is not handling its `cuda_deps`
attribute correctly. It was adding those dependencies both for ROCm and for CUDA.
So this change is fixing that as well.

PiperOrigin-RevId: 679516808

xla/BUILD

@@ -1347,6 +1347,7 @@ bzl_library(
     deps = [
         "//xla/tsl:tsl_bzl",
         "@bazel_skylib//lib:paths",
+        "@tsl//tsl/platform/default:cuda_build_defs_bzl",
     ],
 )
 

xla/lit.bzl

@@ -1,6 +1,7 @@
 """Helper rules for writing LIT tests."""
 
 load("@bazel_skylib//lib:paths.bzl", "paths")
+load("@tsl//tsl/platform/default:cuda_build_defs.bzl", "if_cuda_is_configured")
 load("//xla/tsl:tsl.bzl", "if_cuda_tools", "if_google", "if_oss")
 
 def enforce_glob(files, **kwargs):
@@ -209,7 +210,11 @@ def lit_test(
         srcs = tools,
         bin_dir = bin_dir,
         lib_dir = lib_dir,
-        deps = ["//xla/stream_executor/cuda:all_runtime"],
+        deps = if_cuda_is_configured(
+            [
+                "//xla/stream_executor/cuda:all_runtime",
+            ],
+        ),
         visibility = ["//visibility:private"],
         **kwargs
     )

xla/pjrt/gpu/BUILD

@@ -315,7 +315,6 @@ xla_test(
         "//xla/pjrt:pjrt_client",
         "//xla/pjrt:pjrt_compiler",
         "//xla/service:hlo_parser",
-        "//xla/stream_executor/cuda:cublas_plugin",
         "//xla/tests:literal_test_util",
         "@com_google_absl//absl/status",
         "@com_google_googletest//:gtest",

xla/service/gpu/fusions/triton/BUILD

@@ -221,7 +221,6 @@ xla_test(
         "//xla/service/gpu/model:tiled_hlo_instruction_or_computation",
         "//xla/service/gpu/tests:gpu_codegen_test",
         "//xla/stream_executor:device_description",
-        "//xla/stream_executor/cuda:cublas_plugin",
         "//xla/tests:filecheck",
         "//xla/tests:verified_hlo_module",
         "//xla/tests:xla_internal_test_main",  # fixdeps: keep
@@ -269,7 +268,6 @@ xla_test(
         "//xla/service/gpu/model:tiled_hlo_instruction_or_computation",
         "//xla/service/gpu/tests:gpu_codegen_test",
         "//xla/stream_executor:device_description",
-        "//xla/stream_executor/cuda:cublas_plugin",
         "//xla/tests:verified_hlo_module",
         "//xla/tests:xla_internal_test_main",  # fixdeps: keep
         "//xla/tsl/lib/core:status_test_util",
@@ -401,7 +399,6 @@ xla_test(
         "//xla/hlo/ir:hlo",
         "//xla/service/gpu/tests:gpu_codegen_test",
         "//xla/stream_executor:device_description",
-        "//xla/stream_executor/cuda:cublas_plugin",
         "//xla/tests:xla_internal_test_main",  # fixdeps: keep
         "@com_google_absl//absl/base:core_headers",
         "@com_google_absl//absl/strings",

xla/service/gpu/kernels/BUILD

@@ -242,7 +242,6 @@ xla_test(
         "//xla/stream_executor",
         "//xla/stream_executor:platform",
         "//xla/stream_executor:platform_manager",
-        "//xla/stream_executor/cuda:cuda_platform",
         "//xla/tsl/lib/core:status_test_util",
         "@com_google_absl//absl/random",
         "@com_google_absl//absl/strings",

xla/stream_executor/build_defs.bzl

@@ -1,6 +1,5 @@
 """Configurations for StreamExecutor builds"""
 
-load("@local_config_cuda//cuda:build_defs.bzl", "if_cuda_is_configured")
 load(
     "@local_config_rocm//rocm:build_defs.bzl",
     _if_cuda_or_rocm = "if_cuda_or_rocm",
@@ -64,34 +63,5 @@ def gpu_only_cc_library(name, tags = [], **kwargs):
         target_compatible_with = kwargs.get("target_compatible_with"),
     )
 
-def cuda_only_cc_library(name, tags = [], **kwargs):
-    """A library that only gets compiled when CUDA is configured, otherwise it's an empty target.
-
-    Args:
-      name: Name of the target
-      tags: Tags being applied to the implementation target
-      **kwargs: Accepts all arguments that a `cc_library` would also accept
-    """
-    if not native.package_name().startswith("xla/stream_executor"):
-        fail("cuda_only_cc_library may only be used in `xla/stream_executor/...`.")
-
-    cc_library(
-        name = "%s_non_cuda" % name,
-        tags = ["manual"],
-    )
-    cc_library(
-        name = "%s_cuda_only" % name,
-        tags = tags + ["manual", "cuda-only"],
-        **kwargs
-    )
-    native.alias(
-        name = name,
-        actual = if_cuda_is_configured(":%s_cuda_only" % name, ":%s_non_cuda" % name),
-        visibility = kwargs.get("visibility"),
-        compatible_with = kwargs.get("compatible_with"),
-        restricted_to = kwargs.get("restricted_to"),
-        target_compatible_with = kwargs.get("target_compatible_with"),
-    )
-
 def stream_executor_build_defs_bzl_deps():
     return []