From 184ecb0d9d1e85b765799811a31ef4c0fdbebe4e Mon Sep 17 00:00:00 2001 From: Chao Chen Date: Tue, 7 Jan 2025 19:15:48 -0600 Subject: [PATCH] gpu_triton_custom_call_test is cuda-only test --- xla/service/gpu/tests/BUILD | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/xla/service/gpu/tests/BUILD b/xla/service/gpu/tests/BUILD index 35e0b3214a9e1..ca1036ad857b8 100644 --- a/xla/service/gpu/tests/BUILD +++ b/xla/service/gpu/tests/BUILD @@ -316,10 +316,10 @@ xla_test( xla_test( name = "gpu_triton_custom_call_test", srcs = ["gpu_triton_custom_call_test.cc"], + tags = ["cuda-only"], backends = [ "gpu_a100", "gpu_v100", - "gpu_amd_any", ], deps = [ ":gpu_codegen_test",