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