Skip to content
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.

Commit 1efc4bb

Browse files
authoredApr 18, 2023
Merge branch 'DeepRec-AI:main' into main
2 parents b750f29 + 3bc930a commit 1efc4bb

File tree

746 files changed

+67530
-12799
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

746 files changed

+67530
-12799
lines changed
 

‎.bazelrc

Lines changed: 15 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -111,25 +111,29 @@ build --define framework_shared_object=true
111111
build --define open_source_build=true
112112
test --define open_source_build=true
113113

114+
# For workaround https://github.com/bazelbuild/bazel/issues/8772 with Bazel >= 0.29.1
115+
build --java_toolchain=//third_party/toolchains/java:tf_java_toolchain
116+
build --host_java_toolchain=//third_party/toolchains/java:tf_java_toolchain
117+
114118
# Please note that OneDNN on MacOS or windows is still not supported.
115119
# If you would like to use a local OneDNN instead of downloading, please set the
116120
# environment variable "TF_MKL_ROOT" every time before build.
117121
build:mkl --define=build_with_mkl=true --define=enable_mkl=true
118122
build:mkl --define=tensorflow_dnnl_contraction_kernel=0
119-
build:mkl -c opt
123+
build:mkl -c opt --copt=-O3
120124

121125
# config to build OneDNN backend with a user specified threadpool.
122126
build:mkl_threadpool --define=build_with_mkl=true --define=enable_mkl=true
123127
build:mkl_threadpool --define=tensorflow_dnnl_contraction_kernel=0
124128
build:mkl_threadpool --define=build_with_mkl_opensource=true
125129
build:mkl_threadpool --define=build_with_mkldnn_threadpool=true
126-
build:mkl_threadpool -c opt
130+
build:mkl_threadpool -c opt --copt=-O3
127131

128132
# Config setting to build with oneDNN and without the binary blob
129133
build:mkl_opensource_only --define=build_with_mkl=true --define=enable_mkl=true
130134
build:mkl_opensource_only --define=tensorflow_dnnl_contraction_kernel=0
131135
build:mkl_opensource_only --define=build_with_mkl_opensource=true
132-
build:mkl_opensource_only -c opt
136+
build:mkl_opensource_only -c opt --copt=-O3
133137

134138
# This config option is used to enable OneDNN open source library only,
135139
# without depending on OneDNN binary version.
@@ -138,6 +142,13 @@ build:mkl_open_source_only --define=build_with_mkl_dnn_v1_only=true
138142
build:mkl_open_source_only --define=build_with_mkl=true --define=enable_mkl=true
139143
build:mkl_open_source_only --define=tensorflow_dnnl_contraction_kernel=0
140144

145+
# Config setting to build oneDNN with Compute Library for the Arm Architecture (ACL).
146+
# This build is for the inference regime only.
147+
build:mkl_aarch64 --define=build_with_mkl_aarch64=true
148+
build:mkl_aarch64 --define=tensorflow_mkldnn_contraction_kernel=0
149+
build:mkl_aarch64 --define=build_with_openmp=true
150+
build:mkl_aarch64 -c opt
151+
141152
# This config refers to building with CUDA available. It does not necessarily
142153
# mean that we build CUDA op kernels.
143154
build:using_cuda --define=using_cuda=true
@@ -197,7 +208,7 @@ build --define=use_fast_cpp_protos=true
197208
build --define=allow_oversize_protos=true
198209

199210
build --spawn_strategy=standalone
200-
build -c opt
211+
build -c opt --copt=-O3
201212

202213
# Make Bazel print out all options from rc files.
203214
build --announce_rc

‎.bazelversion

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
0.26.1
1+
5.3.1

0 commit comments

Comments
 (0)
Please sign in to comment.