From 3e44751596f7e65b557d639a1a082c2dddab3469 Mon Sep 17 00:00:00 2001 From: Henning Becker Date: Wed, 2 Oct 2024 14:38:36 -0700 Subject: [PATCH] Add missing cuda-only tags to stream_executor/cuda targets PiperOrigin-RevId: 681596514 --- xla/stream_executor/cuda/BUILD | 3 +++ 1 file changed, 3 insertions(+) diff --git a/xla/stream_executor/cuda/BUILD b/xla/stream_executor/cuda/BUILD index 4d561f61ea66a..60d25bec6c369 100644 --- a/xla/stream_executor/cuda/BUILD +++ b/xla/stream_executor/cuda/BUILD @@ -299,6 +299,7 @@ xla_test( name = "cuda_collectives_test", srcs = ["cuda_collectives_test.cc"], backends = ["gpu_any"], + tags = ["cuda-only"], deps = [ ":cuda_collectives", "//xla/service/gpu/runtime:nccl_api", @@ -594,6 +595,7 @@ xla_test( name = "cuda_kernel_test", srcs = ["cuda_kernel_test.cc"], backends = ["gpu_any"], + tags = ["cuda-only"], deps = [ ":cuda_kernel", ":cuda_runtime", @@ -1069,6 +1071,7 @@ cc_library( cc_library( name = "stream_executor_cuda", + tags = ["cuda-only"], deps = [ "//xla/stream_executor", "//xla/stream_executor:dnn",