diff --git a/tests/e2e/linalg/BUILD.bazel b/tests/e2e/linalg/BUILD.bazel index ab8360219b06..80d1f03eb50b 100644 --- a/tests/e2e/linalg/BUILD.bazel +++ b/tests/e2e/linalg/BUILD.bazel @@ -73,8 +73,10 @@ iree_check_single_backend_test_suite( VMVX_SRCS = enforce_glob( # keep sorted [ + "argmax.mlir", "conv2d.mlir", "gather_like_ops.mlir", + "index.mlir", "narrow_n_matmuls.mlir", "pack.mlir", "pack_dynamic_inner_tiles.mlir", @@ -84,10 +86,8 @@ VMVX_SRCS = enforce_glob( ], include = ["*.mlir"], exclude = [ - "argmax.mlir", "fp_to_subbyte.mlir", "fp4_f32_conversion.mlir", - "index.mlir", "large_linalg_matmul.mlir", "subbyte_to_fp.mlir", ], @@ -124,18 +124,18 @@ iree_check_single_backend_test_suite( VULKAN_SRCS = enforce_glob( # keep sorted [ + "argmax.mlir", "conv2d.mlir", "gather_like_ops.mlir", + "index.mlir", "narrow_n_matmuls.mlir", "softmax.mlir", "subbyte_to_fp.mlir", ], include = ["*.mlir"], exclude = [ - "argmax.mlir", "fp_to_subbyte.mlir", "fp4_f32_conversion.mlir", - "index.mlir", "large_linalg_matmul.mlir", "pack.mlir", "pack_dynamic_inner_tiles.mlir", @@ -221,20 +221,20 @@ ROCM_SRCS = enforce_glob( # keep sorted [ "argmax.mlir", + "conv2d.mlir", + "fp4_f32_conversion.mlir", + "fp_to_subbyte.mlir", "gather_like_ops.mlir", + "index.mlir", + "narrow_n_matmuls.mlir", "pack_i8.mlir", "softmax.mlir", + "subbyte_to_fp.mlir", "unpack.mlir", ], include = ["*.mlir"], exclude = [ - "conv2d.mlir", - "fp_to_subbyte.mlir", - "fp4_f32_conversion.mlir", - "index.mlir", "large_linalg_matmul.mlir", - "narrow_n_matmuls.mlir", - "subbyte_to_fp.mlir", # https://github.com/llvm/llvm-project/issues/131386 causes # See bug #20294 "pack.mlir", diff --git a/tests/e2e/linalg/CMakeLists.txt b/tests/e2e/linalg/CMakeLists.txt index 9f941a2f6024..33db43f627b4 100644 --- a/tests/e2e/linalg/CMakeLists.txt +++ b/tests/e2e/linalg/CMakeLists.txt @@ -55,8 +55,10 @@ iree_check_single_backend_test_suite( NAME check_vmvx_local-task SRCS + "argmax.mlir" "conv2d.mlir" "gather_like_ops.mlir" + "index.mlir" "narrow_n_matmuls.mlir" "pack.mlir" "pack_dynamic_inner_tiles.mlir" @@ -89,8 +91,10 @@ iree_check_single_backend_test_suite( NAME check_vulkan-spirv_vulkan SRCS + "argmax.mlir" "conv2d.mlir" "gather_like_ops.mlir" + "index.mlir" "narrow_n_matmuls.mlir" "softmax.mlir" "subbyte_to_fp.mlir" @@ -156,9 +160,15 @@ iree_check_single_backend_test_suite( check_rocm_hip SRCS "argmax.mlir" + "conv2d.mlir" + "fp4_f32_conversion.mlir" + "fp_to_subbyte.mlir" "gather_like_ops.mlir" + "index.mlir" + "narrow_n_matmuls.mlir" "pack_i8.mlir" "softmax.mlir" + "subbyte_to_fp.mlir" "unpack.mlir" TARGET_BACKEND "rocm" diff --git a/tests/e2e/linalg_ext_ops/BUILD.bazel b/tests/e2e/linalg_ext_ops/BUILD.bazel index 38442ae241e3..68295287a35f 100644 --- a/tests/e2e/linalg_ext_ops/BUILD.bazel +++ b/tests/e2e/linalg_ext_ops/BUILD.bazel @@ -69,6 +69,7 @@ VMVX_SRCS = enforce_glob( # keep sorted [ "arg_compare.mlir", + "attention.mlir", "gather.mlir", "map_gather.mlir", "map_scatter.mlir", @@ -81,7 +82,6 @@ VMVX_SRCS = enforce_glob( ], include = ["*.mlir"], exclude = [ - "attention.mlir", "attention_i1_mask.mlir", "attention_i1_mask_encoding.mlir", ], @@ -195,9 +195,12 @@ iree_check_single_backend_test_suite( [ "arg_compare.mlir", "gather.mlir", + "map_gather.mlir", + "map_scatter.mlir", "scan.mlir", "scatter.mlir", "sort.mlir", + "top-k.mlir", "winograd_input.mlir", "winograd_output.mlir", ], @@ -206,9 +209,6 @@ iree_check_single_backend_test_suite( "attention.mlir", "attention_i1_mask.mlir", "attention_i1_mask_encoding.mlir", - "map_gather.mlir", - "map_scatter.mlir", - "top-k.mlir", ], ), driver = "vulkan", diff --git a/tests/e2e/linalg_ext_ops/CMakeLists.txt b/tests/e2e/linalg_ext_ops/CMakeLists.txt index ce5279321bc7..8ce864dcfdb2 100644 --- a/tests/e2e/linalg_ext_ops/CMakeLists.txt +++ b/tests/e2e/linalg_ext_ops/CMakeLists.txt @@ -57,6 +57,7 @@ iree_check_single_backend_test_suite( check_vmvx_local-task SRCS "arg_compare.mlir" + "attention.mlir" "gather.mlir" "map_gather.mlir" "map_scatter.mlir" @@ -138,9 +139,12 @@ iree_check_single_backend_test_suite( SRCS "arg_compare.mlir" "gather.mlir" + "map_gather.mlir" + "map_scatter.mlir" "scan.mlir" "scatter.mlir" "sort.mlir" + "top-k.mlir" "winograd_input.mlir" "winograd_output.mlir" TARGET_BACKEND diff --git a/tests/e2e/stablehlo_ops/BUILD.bazel b/tests/e2e/stablehlo_ops/BUILD.bazel index 9118e78eb292..c8297028824f 100644 --- a/tests/e2e/stablehlo_ops/BUILD.bazel +++ b/tests/e2e/stablehlo_ops/BUILD.bazel @@ -241,6 +241,7 @@ iree_check_single_backend_test_suite( "reduce_window.mlir", "remainder.mlir", "reshape.mlir", + "reverse.mlir", "rng_normal.mlir", "rng_uniform.mlir", "round.mlir", @@ -264,7 +265,6 @@ iree_check_single_backend_test_suite( exclude = [ "exponential_fp16.mlir", "fft.mlir", # TODO(#9583) - "reverse.mlir", # TODO(#12415): disabled due to miscompilation on Pixel 6. ], ), compiler_flags = [ diff --git a/tests/e2e/stablehlo_ops/CMakeLists.txt b/tests/e2e/stablehlo_ops/CMakeLists.txt index 2d357621cee5..c76b852331cb 100644 --- a/tests/e2e/stablehlo_ops/CMakeLists.txt +++ b/tests/e2e/stablehlo_ops/CMakeLists.txt @@ -287,6 +287,7 @@ iree_check_single_backend_test_suite( "reduce_window.mlir" "remainder.mlir" "reshape.mlir" + "reverse.mlir" "rng_normal.mlir" "rng_uniform.mlir" "round.mlir"