From 8ba261949ace09489e87ad847cd2852da8e9e04c Mon Sep 17 00:00:00 2001
From: Ben Vanik <ben.vanik@gmail.com>
Date: Tue, 12 Dec 2023 11:28:18 -0800
Subject: [PATCH] Disabling CUDA transform dialect reduction tests as they are
 flakey. Tracked in #15892.

---
 tests/transform_dialect/cuda/BUILD.bazel    | 15 ++++++++-------
 tests/transform_dialect/cuda/CMakeLists.txt |  7 -------
 2 files changed, 8 insertions(+), 14 deletions(-)

diff --git a/tests/transform_dialect/cuda/BUILD.bazel b/tests/transform_dialect/cuda/BUILD.bazel
index a7903fabfc89..7b5d2595234e 100644
--- a/tests/transform_dialect/cuda/BUILD.bazel
+++ b/tests/transform_dialect/cuda/BUILD.bazel
@@ -27,14 +27,15 @@ iree_lit_test_suite(
     name = "lit",
     srcs = [
         "mma.mlir",
-        "reduction.mlir",
-        "reduction_eltwise.mlir",
-        "reduction_v2.mlir",
-        "reduction_v2_uneven.mlir",
-        "softmax.mlir",
-        "softmax_v2.mlir",
+        # TODO(#15892): reductions have flakes and need to be triaged.
+        # "reduction.mlir",
+        # "reduction_eltwise.mlir",
+        # "reduction_v2.mlir",
+        # "reduction_v2_uneven.mlir",
+        # "softmax.mlir",
+        # "softmax_v2.mlir",
         # First few ops of softmax only, acts as a proxy example.
-        "softmax_partial.mlir",
+        # "softmax_partial.mlir",
     ],
     cfg = "//tests:lit.cfg.py",
     # transform dialect spec files are MLIR files that specify a transformation,
diff --git a/tests/transform_dialect/cuda/CMakeLists.txt b/tests/transform_dialect/cuda/CMakeLists.txt
index ab23d81e1de5..424995d6be19 100644
--- a/tests/transform_dialect/cuda/CMakeLists.txt
+++ b/tests/transform_dialect/cuda/CMakeLists.txt
@@ -19,13 +19,6 @@ iree_lit_test_suite(
     lit
   SRCS
     "mma.mlir"
-    "reduction.mlir"
-    "reduction_eltwise.mlir"
-    "reduction_v2.mlir"
-    "reduction_v2_uneven.mlir"
-    "softmax.mlir"
-    "softmax_partial.mlir"
-    "softmax_v2.mlir"
   TOOLS
     FileCheck
     iree-compile