Skip to content

Commit

Permalink
Merge pull request tensorflow#76831 from snadampal:compute_library_de…
Browse files Browse the repository at this point in the history
…f_runtime_fix

PiperOrigin-RevId: 690199035
  • Loading branch information
tensorflower-gardener committed Oct 27, 2024
2 parents 7050b95 + 89bb98e commit 805ae2c
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 0 deletions.
1 change: 1 addition & 0 deletions .bazelrc
Original file line number Diff line number Diff line change
Expand Up @@ -214,6 +214,7 @@ 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 --@compute_library//:openmp=false
build:mkl_aarch64_threadpool -c opt

# CUDA: This config refers to building CUDA op kernels with nvcc.
Expand Down
1 change: 1 addition & 0 deletions third_party/xla/.bazelrc
Original file line number Diff line number Diff line change
Expand Up @@ -214,6 +214,7 @@ 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 --@compute_library//:openmp=false
build:mkl_aarch64_threadpool -c opt

# CUDA: This config refers to building CUDA op kernels with nvcc.
Expand Down
1 change: 1 addition & 0 deletions third_party/xla/third_party/tsl/.bazelrc
Original file line number Diff line number Diff line change
Expand Up @@ -214,6 +214,7 @@ 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 --@compute_library//:openmp=false
build:mkl_aarch64_threadpool -c opt

# CUDA: This config refers to building CUDA op kernels with nvcc.
Expand Down

0 comments on commit 805ae2c

Please sign in to comment.