gpu_triton_custom_call_test is cuda-only test

ROCm · Jan 8, 2025 · 184ecb0 · 184ecb0
1 parent d6d8535
commit 184ecb0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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",