OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-06 12:20:11 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Undo unintentional edits to oneDNN configs from cl/755989315

Browse Source 
cl/755989315 should have only added the single flag to affect Python compiling,
but somehow it ended up reverting the edits from cl/755781948

Manually restore the edits made by cl/755781948

PiperOrigin-RevId: 761660001
This commit is contained in:
Richard Levasseur 2025-05-21 14:01:38 -07:00 committed by TensorFlower Gardener
parent 368743da9c
commit 4a8e5c4a6b
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
.bazelrc
View File

@ -255,15 +255,17 @@ build:mkl_threadpool --define=tensorflow_mkldnn_contraction_kernel=0
build:mkl_threadpool --define=build_with_mkl_opensource=true
build:mkl_threadpool -c opt
# Config setting to build oneDNN with Compute Library for the Arm Architecture (ACL).
build:mkl_aarch64 --define=build_with_mkl_aarch64=true
build:mkl_aarch64 --define=build_with_openmp=true
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 --define=build_with_acl=true
build:mkl_aarch64_threadpool -c opt
# This is an alias for the mkl_aarch64_threadpool build.
build:mkl_aarch64 --config=mkl_aarch64_threadpool
# Default CUDA and CUDNN versions.
build:cuda_version --repo_env=HERMETIC_CUDA_VERSION="12.5.1"
build:cuda_version --repo_env=HERMETIC_CUDNN_VERSION="9.3.0"