Add missing cpu features flag to match mmt4d flag

bjacob · Mar 5, 2022 · 89d4981 · 89d4981
1 parent 1f3383a
commit 89d4981
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/benchmarks/TFLite/CMakeLists.txt b/benchmarks/TFLite/CMakeLists.txt
@@ -382,6 +382,7 @@ iree_benchmark_suite(
   TRANSLATION_FLAGS
     ${ANDROID_CPU_TRANSLATION_FLAGS}
     "--iree-flow-mmt4d-target-options=arch=aarch64 features=+dotprod"
+    "-iree-llvm-target-cpu-features=+dotprod"
   BENCHMARK_TOOL
     iree-benchmark-module
   DRIVER
@@ -411,6 +412,7 @@ iree_benchmark_suite(
   TRANSLATION_FLAGS
     ${ANDROID_CPU_TRANSLATION_FLAGS}
     "--iree-flow-mmt4d-target-options=arch=aarch64 features=+dotprod"
+    "-iree-llvm-target-cpu-features=+dotprod"
   BENCHMARK_TOOL
     iree-benchmark-module
   DRIVER
@@ -440,6 +442,7 @@ iree_benchmark_suite(
 #   TRANSLATION_FLAGS
 #     ${ANDROID_CPU_TRANSLATION_FLAGS}
 #     "--iree-flow-mmt4d-target-options=arch=aarch64 features=+dotprod"
+#     "-iree-llvm-target-cpu-features=+dotprod"
 #   BENCHMARK_TOOL
 #     iree-benchmark-module
 #   DRIVER
@@ -467,6 +470,7 @@ iree_benchmark_suite(
 #   TRANSLATION_FLAGS
 #     ${ANDROID_CPU_TRANSLATION_FLAGS}
 #     "--iree-flow-mmt4d-target-options=arch=aarch64 features=+dotprod"
+#     "-iree-llvm-target-cpu-features=+dotprod"
 #   BENCHMARK_TOOL
 #     iree-benchmark-module
 #   DRIVER
@@ -494,6 +498,7 @@ iree_benchmark_suite(
   TRANSLATION_FLAGS
     ${ANDROID_CPU_TRANSLATION_FLAGS}
     "--iree-flow-mmt4d-target-options=arch=aarch64 features=+dotprod"
+    "-iree-llvm-target-cpu-features=+dotprod"
   BENCHMARK_TOOL
     iree-benchmark-module
   DRIVER