@@ -21,12 +21,12 @@ source ../utils/git-utils.sh
2121
2222set -eux -o pipefail
2323
24- TENSORFLOW_HASH=8ed060f0c234f4c069cde7fc7080f166c34c410a # master Apr 03, 2025
24+ TENSORFLOW_HASH=65781570c55d2338106767de200323f123c3f91f
2525
2626# The next oneDNN commits relocate xbyak_aarch64 as a third_party module,
2727# but this is not picked up by TF Bazel build.
2828ONEDNN_HASH=ad06da68524b2b5e63fc1d7a7a749d555394a0a7 # main Jan 03, 2025
29-
29+
3030# The next ACL commit introduces KleidiAI as a third-party module,
3131# but this is not picked up by TF Bazel build.
3232ACL_HASH=0038c52d6c79b76755c087cda1be4bbf752e272c # main Jan 6, 2025
@@ -37,48 +37,7 @@ git-shallow-clone https://github.com/tensorflow/tensorflow.git $TENSORFLOW_HASH
3737 cd tensorflow
3838
3939 # Apply TensorFlow WIP patches here
40- # https://github.com/tensorflow/tensorflow/pull/84975 - build(aarch64): Update to oneDNN-3.7 + ACL-24.12
41- apply-github-patch tensorflow/tensorflow 1ca7978322313cd62733075ea354f2af5d1e54a0
42-
43- cd tensorflow
44-
45- # Set up workspace to point to local versions of Compute Library and oneDNN
46- # Rename existing tf_http_archives
47- sed -i -e ' s/\"mkl_dnn_acl_compatible\"/\"mkl_dnn_acl_compatible_backup\"/g' workspace2.bzl
48- sed -i -e ' s/\"compute_library\"/\"compute_library_backup\"/g' workspace2.bzl
49- # Insert a new_local_repository for oneDNN and ACL in place of the http_archives
50- csplit -f workspace2.bzl. -n 1 workspace2.bzl /' def _tf_repositories():' /+2 ' {0}'
51- onednn_acl_local_repositories=$'
52- native.new_local_repository(
53- name = "mkl_dnn_acl_compatible",
54- build_file = "//third_party/mkl_dnn:mkldnn_acl.BUILD",
55- path = \' ./third_party/mkl_dnn/oneDNN\'
56- )
57- native.local_repository(
58- name = "compute_library",
59- path = \' ./third_party/compute_library/ComputeLibrary\'
60- )'
61- echo " $onednn_acl_local_repositories " > tensorflow_local-repositories.txt
62- cat workspace2.bzl.0 tensorflow_local-repositories.txt workspace2.bzl.1 > workspace2.bzl
63- cd ..
64-
65- # oneDNN patches
66- (
67- cd third_party/mkl_dnn
68- git-shallow-clone https://github.com/oneapi-src/oneDNN.git $ONEDNN_HASH
69-
70- # Apply WIP patches here
71- cd oneDNN
72- # https://github.com/uxlfoundation/oneDNN/pull/2958 - set default num_threads to max for acl_threadpool
73- apply-github-patch uxlfoundation/oneDNN ce72a428594c58e925de38c5eb6fea725fe9d0ff
74- )
75-
76- # ACL patches
77- (
78- cd third_party/compute_library
79- git-shallow-clone https://review.mlplatform.org/ml/ComputeLibrary $ACL_HASH
8040
81- # Apply WIP patches here
82- )
41+ apply-github-patch tensorflow/tensorflow 6eb08485a6312a02636f79d8eddf00a549e32aca # build(aarch64): Update to oneDNN-3.7 + ACL-24.12 (fix)
8342
8443)
0 commit comments