Fix TFLite build

iree-org · Jun 12, 2024 · ab8e545 · ab8e545
1 parent 38f1750
commit ab8e545
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/.github/workflows/run_mobile_comparative_benchmark.yml b/.github/workflows/run_mobile_comparative_benchmark.yml
@@ -13,6 +13,7 @@ on:
   schedule:
     # Scheduled to run at 09:00 UTC and 21:00 UTC.
     - cron: '0 09,21 * * *'
+  pull_request:
 
 concurrency:
   # A PR number if a pull request and otherwise the commit hash. This cancels

diff --git a/experimental/tflite/build_tflite.sh b/experimental/tflite/build_tflite.sh
@@ -41,7 +41,9 @@ pushd tensorflow
 # Log the git version of Tensorflow repo.
 git log --oneline --graph --max-count=1
 
-bazel build -c opt --define xnnpack_use_latest_ops=true //tensorflow/lite/tools/benchmark:benchmark_model
+# Disable avx512fp16 on build servers due to a gcc compiler error:
+#   gcc: error: unrecognized command-line option '-mavx512fp16'
+bazel build -c opt --define xnnpack_use_latest_ops=true --define avx512fp16_enabled=false //tensorflow/lite/tools/benchmark:benchmark_model
 cp "$(realpath bazel-bin/tensorflow/lite/tools/benchmark/benchmark_model)" "${X86_OUTPUT_DIR}/"
 
 # The compiler flags used here were retrieved by running ./configure in the root Tensorflow repo.