Moving back the iree-input-type flag to TRANSLATION_FLAGS. (#6110)
This is from #6108 but I let it auto-submit before it could be fixed.
diff --git a/build_tools/cmake/iree_mlir_benchmark_suite.cmake b/build_tools/cmake/iree_mlir_benchmark_suite.cmake
index 1ba78e4..f905453 100644
--- a/build_tools/cmake/iree_mlir_benchmark_suite.cmake
+++ b/build_tools/cmake/iree_mlir_benchmark_suite.cmake
@@ -164,9 +164,6 @@
# The full list of translation flags.
set(_TRANSLATION_ARGS "--iree-mlir-to-vm-bytecode-module")
- if("${_MODULE_SOURCE}" STREQUAL "TensorFlow")
- list(APPEND _TRANSLATION_ARGS "--iree-input-type=mhlo")
- endif()
list(APPEND _TRANSLATION_ARGS "--iree-hal-target-backends=${_RULE_TARGET_BACKEND}")
list(SORT _RULE_TRANSLATION_FLAGS)
list(APPEND _TRANSLATION_ARGS ${_RULE_TRANSLATION_FLAGS})
diff --git a/iree/test/model_benchmarks/CMakeLists.txt b/iree/test/model_benchmarks/CMakeLists.txt
index 0c659bb..6fbc919 100644
--- a/iree/test/model_benchmarks/CMakeLists.txt
+++ b/iree/test/model_benchmarks/CMakeLists.txt
@@ -79,6 +79,7 @@
TARGET_ARCHITECTURE
"CPU-ARM64-v8A"
TRANSLATION_FLAGS
+ "--iree-input-type=mhlo"
"--iree-flow-inline-constants-max-byte-length=2048"
DRIVER
"vmvx"
@@ -109,6 +110,7 @@
TARGET_ARCHITECTURE
"CPU-ARM64-v8A"
TRANSLATION_FLAGS
+ "--iree-input-type=mhlo"
"--iree-llvm-target-triple=aarch64-none-linux-android29"
"--iree-flow-inline-constants-max-byte-length=2048"
# TODO(GH-5857): Enable this after fixing segfault.
@@ -141,6 +143,7 @@
TARGET_ARCHITECTURE
"CPU-ARM64-v8A"
TRANSLATION_FLAGS
+ "--iree-input-type=mhlo"
"--iree-llvm-target-triple=aarch64-none-linux-android29"
"--iree-flow-inline-constants-max-byte-length=2048"
# TODO(GH-5857): Enable this after fixing segfault.
@@ -175,6 +178,7 @@
TARGET_ARCHITECTURE
"CPU-ARM64-v8A"
TRANSLATION_FLAGS
+ "--iree-input-type=mhlo"
"--iree-llvm-target-triple=aarch64-none-linux-android29"
"--iree-flow-inline-constants-max-byte-length=2048"
# TODO(GH-5857): Enable this after fixing segfault.
@@ -208,6 +212,7 @@
TARGET_ARCHITECTURE
"GPU-Adreno"
TRANSLATION_FLAGS
+ "--iree-input-type=mhlo"
"--iree-vulkan-target-triple=adreno-unknown-android11"
"--iree-flow-inline-constants-max-byte-length=2048"
# TODO(GH-6086): Turn on the flag.
@@ -239,6 +244,7 @@
TARGET_ARCHITECTURE
"GPU-Adreno"
TRANSLATION_FLAGS
+ "--iree-input-type=mhlo"
"--iree-vulkan-target-triple=adreno-unknown-android11"
"--iree-flow-inline-constants-max-byte-length=2048"
# TODO(GH-6086): Turn on the flag.
@@ -273,6 +279,7 @@
TARGET_ARCHITECTURE
"GPU-Mali-Valhall"
TRANSLATION_FLAGS
+ "--iree-input-type=mhlo"
"--iree-vulkan-target-triple=valhall-unknown-android11"
"--iree-flow-inline-constants-max-byte-length=16"
# TODO(GH-6086): Turn on the flag.
@@ -304,6 +311,7 @@
TARGET_ARCHITECTURE
"GPU-Mali-Valhall"
TRANSLATION_FLAGS
+ "--iree-input-type=mhlo"
"--iree-vulkan-target-triple=valhall-unknown-android11"
"--iree-flow-inline-constants-max-byte-length=16"
# TODO(GH-6086): Turn on the flag.
