diff --git a/stablehlo/testdata/BUILD.bazel b/stablehlo/testdata/BUILD.bazel index c2c54a80480..5340d4389fe 100644 --- a/stablehlo/testdata/BUILD.bazel +++ b/stablehlo/testdata/BUILD.bazel @@ -55,7 +55,29 @@ RUNFILES_DIR = LITE_CFG_PY.parents[2].absolute().as_posix()''', tags = ["stablehlo_testdata_tests"], deps = ["@rules_python//python/runfiles"], ) - for src in glob(["**/*.mlir"]) + for src in glob( + ["**/*.mlir"], + exclude = ["**/*.large.mlir"], + ) +] + +# Give large tests more RAM / timeout +[ + lit_test( + name = "%s.test" % src, + size = "large", + srcs = [src], + data = [ + "lit.cfg.py", + "lit.site.cfg.py", + "//:stablehlo-opt", + "//:stablehlo-translate", + "@llvm-project//llvm:FileCheck", + ], + tags = ["stablehlo_testdata_tests"], + deps = ["@rules_python//python/runfiles"], + ) + for src in glob(["**/*.large.mlir"]) ] test_suite( diff --git a/stablehlo/testdata/igamma_float64_20_20_float64_20_20_chlo.mlir b/stablehlo/testdata/igamma_float64_20_20_float64_20_20_chlo.large.mlir similarity index 99% rename from stablehlo/testdata/igamma_float64_20_20_float64_20_20_chlo.mlir rename to stablehlo/testdata/igamma_float64_20_20_float64_20_20_chlo.large.mlir index 7d19428fddd..767a62d047c 100644 --- a/stablehlo/testdata/igamma_float64_20_20_float64_20_20_chlo.mlir +++ b/stablehlo/testdata/igamma_float64_20_20_float64_20_20_chlo.large.mlir @@ -1,4 +1,4 @@ -// RUN-DISABLED(#2590, timeout): stablehlo-opt --chlo-pre-serialization-pipeline -inline %s | stablehlo-translate --interpret +// RUN: stablehlo-opt --chlo-pre-serialization-pipeline -inline %s | stablehlo-translate --interpret // RUN: stablehlo-opt --chlo-pre-serialization-pipeline %s | stablehlo-translate --serialize --target=current | stablehlo-translate --deserialize | stablehlo-opt > %t.0 // RUN: stablehlo-opt --chlo-pre-serialization-pipeline %s > %t.1 // RUN: diff %t.0 %t.1 diff --git a/stablehlo/testdata/igammac_float64_20_20_float64_20_20_chlo.mlir b/stablehlo/testdata/igammac_float64_20_20_float64_20_20_chlo.large.mlir similarity index 99% rename from stablehlo/testdata/igammac_float64_20_20_float64_20_20_chlo.mlir rename to stablehlo/testdata/igammac_float64_20_20_float64_20_20_chlo.large.mlir index 42d097dc18f..03e60172d75 100644 --- a/stablehlo/testdata/igammac_float64_20_20_float64_20_20_chlo.mlir +++ b/stablehlo/testdata/igammac_float64_20_20_float64_20_20_chlo.large.mlir @@ -1,4 +1,4 @@ -// RUN-DISABLED(#2590, timeout): stablehlo-opt --chlo-pre-serialization-pipeline -inline %s | stablehlo-translate --interpret +// RUN: stablehlo-opt --chlo-pre-serialization-pipeline -inline %s | stablehlo-translate --interpret // RUN: stablehlo-opt --chlo-pre-serialization-pipeline %s | stablehlo-translate --serialize --target=current | stablehlo-translate --deserialize | stablehlo-opt > %t.0 // RUN: stablehlo-opt --chlo-pre-serialization-pipeline %s > %t.1 // RUN: diff %t.0 %t.1 diff --git a/stablehlo/testdata/quantized/conv_general_dilated_tf_conversion_path_3d_lhs_float32_1_4_28_28_1__rhs_float32_2_3_3_1_16__windowst-8282934958224398022_qi8.mlir b/stablehlo/testdata/quantized/conv_general_dilated_tf_conversion_path_3d_lhs_float32_1_4_28_28_1__rhs_float32_2_3_3_1_16__windowst-8282934958224398022_qi8.large.mlir similarity index 100% rename from stablehlo/testdata/quantized/conv_general_dilated_tf_conversion_path_3d_lhs_float32_1_4_28_28_1__rhs_float32_2_3_3_1_16__windowst-8282934958224398022_qi8.mlir rename to stablehlo/testdata/quantized/conv_general_dilated_tf_conversion_path_3d_lhs_float32_1_4_28_28_1__rhs_float32_2_3_3_1_16__windowst-8282934958224398022_qi8.large.mlir diff --git a/stablehlo/testdata/quantized/conv_general_dilated_tf_conversion_path_3d_lhs_float32_1_4_28_28_1__rhs_float32_2_3_3_1_16__windowst8524424959485066052_qi8.mlir b/stablehlo/testdata/quantized/conv_general_dilated_tf_conversion_path_3d_lhs_float32_1_4_28_28_1__rhs_float32_2_3_3_1_16__windowst8524424959485066052_qi8.large.mlir similarity index 100% rename from stablehlo/testdata/quantized/conv_general_dilated_tf_conversion_path_3d_lhs_float32_1_4_28_28_1__rhs_float32_2_3_3_1_16__windowst8524424959485066052_qi8.mlir rename to stablehlo/testdata/quantized/conv_general_dilated_tf_conversion_path_3d_lhs_float32_1_4_28_28_1__rhs_float32_2_3_3_1_16__windowst8524424959485066052_qi8.large.mlir