diff --git a/experimental/iree/compile_workloads.py b/experimental/iree/compile_workloads.py index 167f4509..0bba15bf 100755 --- a/experimental/iree/compile_workloads.py +++ b/experimental/iree/compile_workloads.py @@ -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" ], } diff --git a/experimental/tflite/build_tflite.sh b/experimental/tflite/build_tflite.sh index e2472939..ea2f26fe 100755 --- 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 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.