Only exclude e2e_matmul_mmt4d_i8_small on RISC-V. (#12764)
The test only failed on RISC-V. We should not disable it for all the platforms.
https://github.com/openxla/iree/issues/12703
diff --git a/build_tools/cmake/test_riscv.sh b/build_tools/cmake/test_riscv.sh
index 0ed3db9..371ba63 100755
--- a/build_tools/cmake/test_riscv.sh
+++ b/build_tools/cmake/test_riscv.sh
@@ -86,6 +86,11 @@
)
fi
+test_exclude_args+=(
+ # TODO(#12703): Enable the test.
+ "iree/tests/e2e/matmul/e2e_matmul_mmt4d_i8_small_llvm-cpu_local-task"
+)
+
tests_label_exclude_regex="($(IFS="|" ; echo "${label_exclude_args[*]}"))"
tests_exclude_regex="($(IFS="|" ; echo "${test_exclude_args[*]}"))"
test_ctest_args=(
diff --git a/tests/e2e/matmul/BUILD.bazel b/tests/e2e/matmul/BUILD.bazel
index 076ea78..b99808f 100644
--- a/tests/e2e/matmul/BUILD.bazel
+++ b/tests/e2e/matmul/BUILD.bazel
@@ -56,8 +56,7 @@
] if lhs_rhs_type == "i8" else []),
trace_runner = "//tools:iree-e2e-matmul-test",
) for lhs_rhs_type in [
- # TODO: https://github.com/openxla/iree/issues/12703
- # "i8",
+ "i8",
"f32",
]]
diff --git a/tests/e2e/matmul/CMakeLists.txt b/tests/e2e/matmul/CMakeLists.txt
index 7a55d90..18f2e60 100644
--- a/tests/e2e/matmul/CMakeLists.txt
+++ b/tests/e2e/matmul/CMakeLists.txt
@@ -44,6 +44,28 @@
iree_generated_trace_runner_test(
NAME
+ e2e_matmul_mmt4d_i8_small
+ GENERATOR
+ "generate_e2e_matmul_tests.py"
+ GENERATOR_ARGS
+ "--lhs_rhs_type=i8"
+ "--shapes=small"
+ TRACE_RUNNER
+ iree-e2e-matmul-test
+ TARGET_BACKENDS
+ "llvm-cpu"
+ DRIVERS
+ "local-task"
+ COMPILER_FLAGS
+ "--iree-flow-enable-data-tiling"
+ TARGET_CPU_FEATURES_VARIANTS
+ "default"
+ "arm_64:+dotprod"
+ "arm_64:+i8mm"
+)
+
+iree_generated_trace_runner_test(
+ NAME
e2e_matmul_mmt4d_f32_small
GENERATOR
"generate_e2e_matmul_tests.py"
