Skip to content

Commit 0115162

Browse files
committed
Bump TensorFlow sources for 25.06 release
1 parent ba3ea0a commit 0115162

1 file changed

Lines changed: 3 additions & 44 deletions

File tree

ML-Frameworks/tensorflow-aarch64/get-source.sh

Lines changed: 3 additions & 44 deletions
Original file line numberDiff line numberDiff line change
@@ -21,12 +21,12 @@ source ../utils/git-utils.sh
2121

2222
set -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.
2828
ONEDNN_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.
3232
ACL_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

Comments
 (0)