Skip to content

Commit 5f832a7

Browse files
authored
Merge pull request #332 from murste01/murste01/bump-23.05-patches
Bump commits and update patches for 25.06
2 parents 58b44ac + 0115162 commit 5f832a7

3 files changed

Lines changed: 8 additions & 56 deletions

File tree

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

Lines changed: 4 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -20,25 +20,18 @@
2020
source ../utils/git-utils.sh
2121

2222
set -eux -o pipefail
23-
PYTORCH_HASH=e872bf8f888bdbb27a03e03935db61babf7180b8 # 2.8.0.dev20250430 from viable/strict
24-
IDEEP_HASH=2ef932a861439e4cc9bb8baee8424b57573de023 # From ideep_pytorch
25-
ONEDNN_HASH=69150ce5fe1f453af9125ca42a921e017092ccf7 # From main
26-
ACL_HASH=334108c0efc512efdc9576ba957dbcf5b7ee168a # rc_25_04_29_0
23+
PYTORCH_HASH=3040ca6d0f8558e39919b14eebeacc34ddf980f5 # main June 10th
24+
IDEEP_HASH=2ef932a861439e4cc9bb8baee8424b57573de023 # main June 10th
25+
ONEDNN_HASH=106a7b41bc4156297b8a88cd1951304b739cc427 # main June 10th
26+
ACL_HASH=6bc1c7b8d0756272e2a97a7489e13de90f864326 # main June 9th
2727
TORCH_AO_HASH=e1cb44ab84eee0a3573bb161d65c18661dc4a307 # From main
2828

2929
git-shallow-clone https://github.com/pytorch/pytorch.git $PYTORCH_HASH
3030
(
3131
cd pytorch
3232

33-
# https://github.com/pytorch/pytorch/pull/143190 - Enable AArch64 CI scripts to be used for local dev
34-
apply-github-patch pytorch/pytorch 6e5628b1f648d862e8fdd150ad277120b236ed15
35-
# https://github.com/pytorch/pytorch/pull/140159 - cpu: enable gemm-bf16f32 for SDPA BF16
36-
apply-github-patch pytorch/pytorch ca4a718be80eb88ca6804b91201e4f98a3e236c8
37-
apply-github-patch pytorch/pytorch 406fe1fbd066401774c104d125a7ac0b3d6eb52b
3833
# https://github.com/pytorch/pytorch/pull/152361 - Build libgomp (gcc-11) from source
3934
apply-github-patch pytorch/pytorch 7c54b6b07558c330ee2f95b4793edb3bfbb814c9
40-
# https://github.com/pytorch/pytorch/pull/150833 - Pin all root requirements to major versions
41-
apply-github-patch pytorch/pytorch 02987a7c2e9b249a669723224c8d3cd80c6cb64e
4235

4336
git submodule sync
4437
git submodule update --init --checkout --force --recursive --jobs=$(nproc)

ML-Frameworks/pytorch-aarch64/utils/build_openblas.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525
source /common_utils/git-utils.sh
2626

2727
set -ex
28-
OPENBLAS_HASH="9aa7a0b2a7b2770adec6ff26b34660d3bcd8c49c"
28+
OPENBLAS_HASH="fe220a0d7d7c5188e698643428708063c8c1a9f6"
2929
OPENBLAS_CHECKOUT_DIR="OpenBLAS"
3030

3131
cd /

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)