From c0eb01b09d821bb0c2587d422f0f1f9766c7ed7e Mon Sep 17 00:00:00 2001 From: Sunita Nadampalli Date: Fri, 28 Jun 2024 01:05:37 +0000 Subject: [PATCH] bazelrc: update aarch64 mkl configs --- .bazelrc | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/.bazelrc b/.bazelrc index 8e52206a2fa..3ce350f72b2 100644 --- a/.bazelrc +++ b/.bazelrc @@ -41,13 +41,15 @@ build:mkl_open_source_only --define=build_with_mkl=true --define=enable_mkl=true build:mkl_open_source_only --define=tensorflow_mkldnn_contraction_kernel=0 # Config setting to build oneDNN with Compute Library for the Arm Architecture (ACL). -# This build is for the inference regime only. -build:mkl_aarch64 --define=build_with_mkl_aarch64=true --define=enable_mkl=true -build:mkl_aarch64 --define=tensorflow_mkldnn_contraction_kernel=0 -build:mkl_aarch64 --define=build_with_mkl_opensource=true +build:mkl_aarch64 --define=build_with_mkl_aarch64=true build:mkl_aarch64 --define=build_with_openmp=true -build:mkl_aarch64 --copt=-march=armv8.2-a -build:mkl_aarch64 --copt=-O3 +build:mkl_aarch64 --define=build_with_acl=true +build:mkl_aarch64 -c opt + +# Config setting to build oneDNN with Compute Library for the Arm Architecture (ACL). +# with Eigen threadpool support +build:mkl_aarch64_threadpool --define=build_with_mkl_aarch64=true +build:mkl_aarch64_threadpool -c opt build --define=build_with_onednn_v2=true build --define=xnn_enable_avxvnni=false