Disable TOSA rescale files and tests while broken

PiperOrigin-RevId: 738816558

Author: GleasonK

stablehlo/BUILD

@@ -1541,25 +1541,13 @@ glob_lit_tests(
     data = [":test_data"],
     driver = "@llvm-project//mlir:run_lit.sh",
     exclude = [
-        # TODO(#268): Fix lit checks with floating-point ops having
-        # implementation defined precision.
-        "stablehlo/tests/interpret_rsqrt.mlir",
-        "stablehlo/tests/interpret_sqrt.mlir",
-        "stablehlo/tests/interpret_tanh.mlir",
-
-        # TODO(b/386761235): Re-enable them.
-        "stablehlo/tests/vhlo/stablehlo_legalize_to_vhlo.0_18_0.mlir",
-        "stablehlo/tests/vhlo/stablehlo_legalize_to_vhlo.0_19_0.mlir",
-        "stablehlo/tests/vhlo/stablehlo_legalize_to_vhlo.0_20_0.mlir",
-        "stablehlo/tests/vhlo/stablehlo_legalize_to_vhlo.1_0_0.mlir",
-        "stablehlo/tests/vhlo/stablehlo_legalize_to_vhlo.1_1_0.mlir",
-        "stablehlo/tests/vhlo/stablehlo_legalize_to_vhlo.1_2_0.mlir",
-        "stablehlo/tests/vhlo/stablehlo_legalize_to_vhlo.1_3_0.mlir",
-        "stablehlo/tests/vhlo/stablehlo_legalize_to_vhlo.1_4_0.mlir",
-        "stablehlo/tests/vhlo/stablehlo_legalize_to_vhlo.1_5_0.mlir",
-        "stablehlo/tests/vhlo/stablehlo_legalize_to_vhlo.1_6_0.mlir",
-        "stablehlo/tests/vhlo/stablehlo_legalize_to_vhlo.1_7_0.mlir",
-        "stablehlo/tests/vhlo/stablehlo_legalize_to_vhlo.1_8_0.mlir",
+        # TODO: remove the following excludes once #2751 is fixed.
+        "stablehlo/conversions/tosa/tests/legalize_quant_ops_to_tosa_rescale.mlir",
+        "stablehlo/conversions/tosa/tests/legalize_tosa_rescale_to_stablehlo.mlir",
+        "stablehlo/conversions/tosa/tests/rescale_interpreter.mlir",
+        "stablehlo/conversions/tosa/tests/binary.mlir",
+        "stablehlo/conversions/tosa/tests/nullary.mlir",
+        "stablehlo/conversions/tosa/tests/unary.mlir",
     ],
     test_file_exts = ["mlir"],
 )
@@ -1650,8 +1638,9 @@ cc_library(
     srcs = [
         "stablehlo/conversions/tosa/transforms/StablehloLegalizeToTosa.cpp",
         "stablehlo/conversions/tosa/transforms/StablehloPrepareForTosa.cpp",
-        "stablehlo/conversions/tosa/transforms/StablehloQuantLegalizeToTosaRescale.cpp",
-        "stablehlo/conversions/tosa/transforms/TosaRescaleLegalizeToStablehlo.cpp",
+        # TODO: un-comment the following once #2751 is fixed
+        # "stablehlo/conversions/tosa/transforms/StablehloQuantLegalizeToTosaRescale.cpp",
+        # "stablehlo/conversions/tosa/transforms/TosaRescaleLegalizeToStablehlo.cpp",
     ],
     hdrs = [
         "stablehlo/conversions/tosa/transforms/Passes.h",

stablehlo/BUILD.bazel

@@ -1577,8 +1577,9 @@ cc_library(
     srcs = [
         "stablehlo/conversions/tosa/transforms/StablehloLegalizeToTosa.cpp",
         "stablehlo/conversions/tosa/transforms/StablehloPrepareForTosa.cpp",
-        "stablehlo/conversions/tosa/transforms/StablehloQuantLegalizeToTosaRescale.cpp",
-        "stablehlo/conversions/tosa/transforms/TosaRescaleLegalizeToStablehlo.cpp",
+        # TODO: un-comment the following once #2751 is fixed
+        # "stablehlo/conversions/tosa/transforms/StablehloQuantLegalizeToTosaRescale.cpp",
+        # "stablehlo/conversions/tosa/transforms/TosaRescaleLegalizeToStablehlo.cpp",
     ],
     hdrs = [
         "stablehlo/conversions/tosa/transforms/Passes.h",

stablehlo/docs/generated/stablehlo_tosa_passes.md

@@ -10,17 +10,3 @@ _Prepare StableHLO for legalization to TOSA_
 
 This pass adds rewriters to make StableHLO ops more compatible with TOSA ops.
 Currently simplifies stablehlo.dot_general into stablehlo.dot for easier lowering.
-
-### `-stablehlo-quant-legalize-to-tosa-rescale`
-
-_Legalize StableHLO Quantized operations to TOSA rescale operations_
-
-This pass rewrites StableHLO quantized operations to integer operations
-by inserting TOSA rescale operations at the inputs and outputs of the
-integer operations.
-
-### `-tosa-rescale-legalize-to-stablehlo`
-
-_Legalize TOSA rescales to StableHlo primitive math operations_
-
-This pass rewrites TOSA rescale operations to StableHLO primitive math operations.

stablehlo/stablehlo/conversions/tosa/tests/BUILD.bazel

@@ -55,7 +55,18 @@ RUNFILES_DIR = LITE_CFG_PY.parents[4].absolute().as_posix()''',
         tags = ["stablehlo_tosa_tests"],
         deps = ["@rules_python//python/runfiles"],
     )
-    for src in glob(["**/*.mlir"])
+    # TODO: remove the following excludes once #2751 is fixed.
+    for src in glob(
+        ["**/*.mlir"],
+        exclude = [
+            "legalize_quant_ops_to_tosa_rescale.mlir",
+            "legalize_tosa_rescale_to_stablehlo.mlir",
+            "rescale_interpreter.mlir",
+            "binary.mlir",
+            "nullary.mlir",
+            "unary.mlir",
+        ],
+    )
 ]
 
 test_suite(

stablehlo/stablehlo/conversions/tosa/tests/lit.cfg.py

@@ -25,6 +25,8 @@
 config.test_format = lit.formats.ShTest(not llvm_config.use_lit_shell)
 config.suffixes = ['.mlir']
 config.test_source_root = os.path.dirname(__file__)
+# TODO: remove the following once #2751 is fixed.
+config.excludes = ['legalize_quant_ops_to_tosa_rescale.mlir', 'legalize_tosa_rescale_to_stablehlo.mlir', 'rescale_interpreter.mlir', 'binary.mlir', 'nullary.mlir', 'unary.mlir']
 
 # Disallow reusing variables across CHECK-LABEL matches.
 # A variable can eschew this (be made "global") by prefixing its name with $.

stablehlo/stablehlo/conversions/tosa/transforms/CMakeLists.txt

@@ -25,9 +25,11 @@ add_mlir_pdll_library(StablehloTOSAPDLLPatternsIncGen
 add_mlir_library(StablehloTOSATransforms
   StablehloLegalizeToTosa.cpp
   StablehloPrepareForTosa.cpp
-  StablehloQuantLegalizeToTosaRescale.cpp
-  TosaRescaleLegalizeToStablehlo.cpp
+  # TODO: un-comment the following once #2751 is fixed.
+  # StablehloQuantLegalizeToTosaRescale.cpp
+  # TosaRescaleLegalizeToStablehlo.cpp
 
+  PARTIAL_SOURCES_INTENDED
   DEPENDS
   StablehloTOSATransformsPassIncGen
   StablehloTOSAPDLLPatternsIncGen

stablehlo/stablehlo/conversions/tosa/transforms/Passes.td

@@ -33,24 +33,25 @@ def StablehloPrepareForTosaPass : Pass<"stablehlo-prepare-for-tosa", "mlir::func
   let dependentDialects = ["::mlir::tosa::TosaDialect"];
 }
 
-def StablehloQuantLegalizeToTosaRescalePass : Pass<"stablehlo-quant-legalize-to-tosa-rescale", "mlir::func::FuncOp"> {
-  let summary = "Legalize StableHLO Quantized operations to TOSA rescale operations";
-  let description = [{
-    This pass rewrites StableHLO quantized operations to integer operations
-    by inserting TOSA rescale operations at the inputs and outputs of the
-    integer operations.
-  }];
-  let dependentDialects = [
-    "::mlir::tosa::TosaDialect",
-  ];
-}
-
-def TosaRescaleLegalizeToStablehloPass : Pass<"tosa-rescale-legalize-to-stablehlo", "mlir::func::FuncOp"> {
-  let summary = "Legalize TOSA rescales to StableHlo primitive math operations";
-  let description = [{
-    This pass rewrites TOSA rescale operations to StableHLO primitive math operations.
-  }];
-  let dependentDialects = [
-    "::mlir::stablehlo::StablehloDialect"
-  ];
-}
+// TODO: un-comment the following once #2751 is fixed.
+// def StablehloQuantLegalizeToTosaRescalePass : Pass<"stablehlo-quant-legalize-to-tosa-rescale", "mlir::func::FuncOp"> {
+//   let summary = "Legalize StableHLO Quantized operations to TOSA rescale operations";
+//   let description = [{
+//     This pass rewrites StableHLO quantized operations to integer operations
+//     by inserting TOSA rescale operations at the inputs and outputs of the
+//     integer operations.
+//   }];
+//   let dependentDialects = [
+//     "::mlir::tosa::TosaDialect",
+//   ];
+// }
+//
+// def TosaRescaleLegalizeToStablehloPass : Pass<"tosa-rescale-legalize-to-stablehlo", "mlir::func::FuncOp"> {
+//   let summary = "Legalize TOSA rescales to StableHlo primitive math operations";
+//   let description = [{
+//     This pass rewrites TOSA rescale operations to StableHLO primitive math operations.
+//   }];
+//   let dependentDialects = [
+//     "::mlir::stablehlo::StablehloDialect"
+//   ];
+// }