Disable some failing XLA tests

ROCm · Jun 2, 2024 · 161cea2 · 161cea2
1 parent b13fd6e
commit 161cea2
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 3 deletions.
diff --git a/third_party/xla/xla/service/gpu/BUILD b/third_party/xla/xla/service/gpu/BUILD
@@ -4637,9 +4637,6 @@ cc_library(
 xla_test(
     name = "cudnn_fused_conv_rewriter_test",
     srcs = ["cudnn_fused_conv_rewriter_test.cc"],
-    tags = [
-        "no_rocm"
-    ],
     backend_tags = {
         "gpu_a100": [
             "noasan",
@@ -5870,6 +5867,7 @@ xla_cc_test(
 xla_test(
     name = "determinism_test",
     srcs = if_gpu_is_configured(["determinism_test.cc"]),
+    tags = ["no_rocm"],
     backends = [
         "gpu_a100",
     ],

diff --git a/third_party/xla/xla/service/gpu/fusions/BUILD b/third_party/xla/xla/service/gpu/fusions/BUILD
@@ -84,6 +84,7 @@ cc_library(
 xla_cc_test(
     name = "in_place_dynamic_update_slice_mlir_test",
     srcs = ["in_place_dynamic_update_slice_mlir_test.cc"],
+    tags = ["no_rocm"], #TODO(rocm) TEMP
     use_gpu = True,
     deps = [
         ":in_place_dynamic_update_slice_mlir",
@@ -438,6 +439,7 @@ cc_library(
 xla_cc_test(
     name = "scatter_mlir_test",
     srcs = ["scatter_mlir_test.cc"],
+    tags = ["no_rocm"],
     use_gpu = True,
     deps = [
         ":mlir_emitter_test_base",
@@ -491,6 +493,7 @@ cc_library(
 xla_cc_test(
     name = "transpose_mlir_test",
     srcs = ["transpose_mlir_test.cc"],
+    tags = ["no_rocm"], #TODO(rocm) TEMP
     use_gpu = True,
     deps = [
         ":mlir_emitter_test_base",
@@ -883,6 +886,7 @@ cc_library(
 xla_cc_test(
     name = "reduction_mlir_test",
     srcs = ["reduction_mlir_test.cc"],
+    tags = ["no_rocm"], #TODO(rocm) TEMP
     use_gpu = True,
     deps = [
         ":mlir_emitter_test_base",
@@ -969,6 +973,7 @@ cc_library(
 xla_cc_test(
     name = "concatenate_mlir_test",
     srcs = ["concatenate_mlir_test.cc"],
+    tags = ["no_rocm"], #TODO(rocm) TEMP
     use_gpu = True,
     deps = [
         ":concatenate_mlir",
@@ -1107,6 +1112,7 @@ cc_library(
 xla_cc_test(
     name = "input_slices_mlir_test",
     srcs = ["input_slices_mlir_test.cc"],
+    tags = ["no_rocm"], #TODO(rocm) TEMP
     use_gpu = True,
     deps = [
         ":input_slices_mlir",

diff --git a/third_party/xla/xla/service/gpu/tests/BUILD b/third_party/xla/xla/service/gpu/tests/BUILD
@@ -447,6 +447,7 @@ xla_cc_test(
 xla_test(
     name = "gpu_triton_custom_call_test",
     srcs = ["gpu_triton_custom_call_test.cc"],
+    tags = ["no_rocm"], #TODO(rocm) TEMP
     backends = [
         "gpu_a100",
         "gpu_v100",

diff --git a/third_party/xla/xla/tests/BUILD b/third_party/xla/xla/tests/BUILD
@@ -900,6 +900,7 @@ xla_test(
     shard_count = 20,
     tags = [
         "optonly",
+        "no_rocm",
     ],
     deps = [
         ":client_library_test_base",
@@ -937,6 +938,7 @@ xla_test(
         "optonly",
         # TODO(b/151340488): Timed out on 2020-03-12.
         "nozapfhahn",
+        "no_rocm",
     ],
     deps = [
         ":client_library_test_base",
@@ -1055,6 +1057,7 @@ xla_test(
     shard_count = 20,
     tags = [
         "optonly",
+        "no_rocm",
     ],
     deps = [
         ":client_library_test_base",