Skip to content

Commit

Permalink
Fix TFLite build
Browse files Browse the repository at this point in the history
Signed-off-by: mariecwhite <[email protected]>
  • Loading branch information
mariecwhite committed Jun 24, 2024
1 parent 79870b7 commit e467885
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 2 deletions.
1 change: 1 addition & 0 deletions .github/workflows/run_mobile_comparative_benchmark.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
2 changes: 1 addition & 1 deletion experimental/iree/compile_workloads.py
Original file line number Diff line number Diff line change
Expand Up @@ -67,7 +67,7 @@
],
"pixel-8-pro": [
"--iree-llvmcpu-target-triple=aarch64-none-linux-android34",
"--iree-llvmcpu-target-cpu-features=+v9a,+fullfp16,fp-armv8,+neon,+aes,+sha2,+crc,+lse,+rdm,+complxnum,+rcpc,+sha3,+sm4,+dotprod,+fp16fml,+dit,+flagm,+ssbs,+sb,+sve2-aes,+sve2-bitperm,+sve2-sha3,+sve2-sm4,+altnzcv,+fptoint,+bf16,+i8mm,+bti,+mte,+pauth,+perfmon,+predres,+spe,+ras"
"--iree-llvmcpu-target-cpu-features=+v9.2a,+fullfp16,+fp-armv8,+neon,+aes,+sha2,+crc,+lse,+rdm,+complxnum,+rcpc,+sha3,+sm4,+dotprod,+fp16fml,+dit,+flagm,+ssbs,+sb,+sve2-aes,+sve2-bitperm,+sve2-sha3,+sve2-sm4,+altnzcv,+fptoint,+bf16,+i8mm,+bti,+mte,+pauth,+perfmon,+predres,+spe,+ras"
],
}

Expand Down
4 changes: 3 additions & 1 deletion experimental/tflite/build_tflite.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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 xnn_enable_avx512fp16=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.
Expand Down

0 comments on commit e467885

Please sign in to comment.