From 7550346acacab61dd426ad364356fbe6b122a04e Mon Sep 17 00:00:00 2001 From: Milica Makevic Date: Fri, 15 Nov 2024 08:17:39 +0000 Subject: [PATCH] Disable gpu_too_many_blocks_test --- third_party/xla/xla/service/gpu/tests/BUILD | 1 + 1 file changed, 1 insertion(+) diff --git a/third_party/xla/xla/service/gpu/tests/BUILD b/third_party/xla/xla/service/gpu/tests/BUILD index 4d6649be4207d9..19296227680e89 100644 --- a/third_party/xla/xla/service/gpu/tests/BUILD +++ b/third_party/xla/xla/service/gpu/tests/BUILD @@ -191,6 +191,7 @@ xla_test( "gpu_too_many_blocks_test.cc", ], backends = ["gpu"], + tags = ["cuda-only",], #(TODO)(rocm): weekly sync 24-11-05 deps = [ ":gpu_codegen_test", "//xla/hlo/ir:hlo",