Skip to content

Commit 55fd76b

Browse files
puneetmatharupuneetmatharu
committed
Bump sources and handle latest patches.
1 parent 31079e3 commit 55fd76b

7 files changed

Lines changed: 36 additions & 22 deletions

File tree

‎ML-Frameworks/pytorch-aarch64/CHANGELOG.md‎

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,10 +14,21 @@ where `YY` is the year, and `MM` the month of the increment.
1414
## [unreleased]
1515

1616
### Added
17+
- Adds PyTorch [PR #177009](https://github.com/pytorch/pytorch/pull/177009), to accelerate SDPA on Arm CPUs.
18+
- Adds oneDNN [PR #4911](https://github.com/uxlfoundation/oneDNN/pull/4911), to add LUT-based `gelu_erf` JIT implementations.
1719

1820
### Changed
21+
- Updates hashes for:
22+
- `PYTORCH_HASH` to `fb69aa6b76c7ddbcac13ac3dd8b14625f4352ffd`, 2.12.0.dev20260328 from viable/strict, Mar 28th.
23+
- `IDEEP_HASH` to `cbbfd4ad7c5ac6d7683af571055e95c948d8cf54`, from ideep_pytorch, Mar 17th.
24+
- `ONEDNN_HASH` to `abc14842394f985313191bc1e3c69bb7f8cecd23`, from main, Mar 27th.
25+
- `TORCH_AO_HASH` to `d5814ae2edf434bb7927d3d992b33239e6e2aa1b`, from main, Mar 29th.
26+
- `KLEIDIAI_HASH` to `6c544373d1731d2b74435fc02f8bad5f4631b0b1`, v1.23.0 from main, Mar 25th.
27+
- Updates `OPENBLAS_VERSION` from v0.3.30 to `d26960a21ec5da7f77377f28bd6e230060841ae0` from main, Mar 27th.
28+
- Updates `torchvision` from 0.25.0.dev20260130 to 0.26.0.dev20260329.
1929

2030
### Removed
31+
- Removes PyTorch [PR #159859](https://github.com/pytorch/pytorch/pull/159859), which added PoC LUT optimisation for GELU bf16 operators.
2132

2233
### Fixed
2334

‎ML-Frameworks/pytorch-aarch64/build-torch-ao-wheel.sh‎

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,10 @@ if ! docker container inspect "$TORCH_BUILD_CONTAINER" >/dev/null 2>&1 ; then
6060
docker_exec rm -rf /acl
6161

6262
docker_exec python${PYTHON_VERSION} -m venv ${TEST_VENV_CONTAINER_DIR}
63-
docker_exec bash -c "source ${TEST_VENV_CONTAINER_DIR}/bin/activate && pip install typing_extensions torch wheel numpy --no-deps"
63+
docker_exec bash -c "
64+
source ${TEST_VENV_CONTAINER_DIR}/bin/activate && \
65+
pip install typing_extensions wheel numpy --no-deps && \
66+
pip install torch --index-url https://download.pytorch.org/whl/cpu"
6467

6568
echo "Storing torch build container id in $TORCH_BUILD_CONTAINER_ID_FILE for reuse: $TORCH_BUILD_CONTAINER"
6669
echo "$TORCH_BUILD_CONTAINER" > "$TORCH_BUILD_CONTAINER_ID_FILE"

‎ML-Frameworks/pytorch-aarch64/build-wheel.sh‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ if ! docker container inspect "$TORCH_BUILD_CONTAINER" >/dev/null 2>&1 ; then
123123
# Currently changes in these scripts will not be applied without a clean
124124
# build, which is not ideal for dev work. But we have to balance this with
125125
# extra time/network traffic when rebuilding many times.
126-
docker_exec bash "${PYTORCH_CONTAINER_DIR}/.circleci/scripts/binary_populate_env.sh"
126+
docker_exec bash "${PYTORCH_CONTAINER_DIR}/.ci/pytorch/binary_populate_env.sh"
127127

128128
# Install scons for the Compute Library (ACL) build
129129
docker_exec ${PYTHON_CONTAINER_BIN}/python3 -m pip install scons==4.7.0

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

Lines changed: 7 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -24,20 +24,15 @@ git-shallow-clone https://github.com/pytorch/pytorch.git $PYTORCH_HASH
2424
# https://github.com/pytorch/pytorch/pull/170600 - Gate deletion of clean-up steps in build_common.sh
2525
apply-github-patch pytorch/pytorch e368ec2693b8b2b8ba35d0913f1d663ba2fdc804
2626

27-
# FIXME: Temporarily disabled; to be updated in a later PR
28-
# # https://github.com/pytorch/pytorch/pull/160184 - Draft: separate reqs for manywheel build and pin
29-
# # Note: as part of this patch, setuptools is pinned to ~= 78.1.1 which is not affected by
30-
# # CVE-2025-47273 and CVE-2024-6345
31-
# apply-github-patch pytorch/pytorch 4d344570e5a114fa522e3370c5d59161e2ed8619
32-
33-
# https://github.com/pytorch/pytorch/pull/159859 - PoC LUT optimisation for GELU bf16 operators
34-
apply-github-patch pytorch/pytorch ebcc874e317f9563ab770fc5c27df969e0438a5e
35-
3627
# https://github.com/pytorch/pytorch/pull/167328 - Build cpuinfo into c10 shared library
3728
apply-github-patch pytorch/pytorch 7c053dd1582b778c81101dd452708c4ec6e58233
3829
apply-github-patch pytorch/pytorch b1782bbe0eda5957870e2f6e95b8f167e04843cb
3930
apply-github-patch pytorch/pytorch 337925aed2babb3ef7808f78536bbbc9df346a4f
4031

32+
# https://github.com/pytorch/pytorch/pull/177009 - Accelerate SDPA on Arm CPUs: Unroll exp_sum and max_mul kernels
33+
apply-github-patch pytorch/pytorch f6a6cc94e92636902acbbe443d00cec4a8efa12d
34+
apply-github-patch pytorch/pytorch 253a1ea6db210c01178dd365922eb49fe4892fe7
35+
4136
# Remove deps that we don't need for manylinux AArch64 CPU builds before fetching.
4237
# Only used when jni.h is present (see .ci/pytorch/build.sh:116), which is not the case for manylinux
4338
git rm android/libs/fbjni
@@ -78,6 +73,9 @@ git-shallow-clone https://github.com/pytorch/pytorch.git $PYTORCH_HASH
7873
(
7974
cd mkl-dnn
8075
git fetch origin $ONEDNN_HASH && git clean -f && git checkout -f FETCH_HEAD
76+
77+
# https://github.com/uxlfoundation/oneDNN/pull/4911 - cpu: aarch64: add LUT-based gelu_erf JIT implementations #4911
78+
apply-github-patch uxlfoundation/oneDNN 86661869d4267b49db667ed7946eefba4e44d2c0
8179
)
8280
)
8381
(

‎ML-Frameworks/pytorch-aarch64/versions.sh‎

100644100755
Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -9,15 +9,15 @@
99
# For information on how to update the versions below, read the README.md.
1010

1111
# get-source.sh deps
12-
PYTORCH_HASH=434b8cfdfac8ef7c741385561e340781ad76512f # 2.12.0.dev20260227 from viable/strict, Feb 27th
13-
IDEEP_HASH=bbb9ffb9e0c401ca058b7f35a6ebe7d0e08ffd34 # From ideep_pytorch, Jan 16th
14-
ONEDNN_HASH=a83f8b4c7ca45fb339f3ecdf82895bc496941808 # From main, Feb 27th
15-
TORCH_AO_HASH=9bdc0ca87c1134b7c1dedaa9512233b726f22955 # From main, Feb 27th
16-
KLEIDIAI_HASH=98a6df72bdbb566bc7d8ba13d71991bcd94a8393 # v1.22.0 from main, Feb 19th
12+
PYTORCH_HASH=fb69aa6b76c7ddbcac13ac3dd8b14625f4352ffd # 2.12.0.dev20260328 from viable/strict, Mar 28th
13+
IDEEP_HASH=cbbfd4ad7c5ac6d7683af571055e95c948d8cf54 # From ideep_pytorch, Mar 17th
14+
ONEDNN_HASH=abc14842394f985313191bc1e3c69bb7f8cecd23 # From main, Mar 27th
15+
TORCH_AO_HASH=d5814ae2edf434bb7927d3d992b33239e6e2aa1b # From main, Mar 29th
16+
KLEIDIAI_HASH=6c544373d1731d2b74435fc02f8bad5f4631b0b1 # v1.23.0 from main, Mar 25th
1717

1818
# build-wheel.sh deps
19-
ACL_VERSION="v52.8.0"
20-
OPENBLAS_VERSION="v0.3.30"
19+
ACL_VERSION="v52.8.0" # Jan 23rd
20+
OPENBLAS_VERSION="d26960a21ec5da7f77377f28bd6e230060841ae0" # Mar 27th
2121

2222
# Dockerfile deps
23-
TORCHVISION_NIGHTLY="0.25.0.dev20260130"
23+
TORCHVISION_NIGHTLY="0.26.0.dev20260329"

‎ML-Frameworks/tensorflow-aarch64/CHANGELOG.md‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,8 @@ where `YY` is the year, and `MM` the month of the increment.
1616
### Added
1717

1818
### Changed
19+
- Replaces patch to bump ComputeLibrary version [from v24.12 to v52.4.0](https://github.com/tensorflow/tensorflow/pull/100882) with [one to bump to v52.8.0](https://github.com/tensorflow/tensorflow/pull/113368)
20+
- Updates TensorFlow hash to 25cf6546b44bdb3d3f17277a4b49b027dcdc0ae7 from nightly, Mar 23rd
1921

2022
### Removed
2123

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ source ../utils/git-utils.sh
88

99
set -eux -o pipefail
1010

11-
TENSORFLOW_HASH=535fc05dbac76cec4a446eec0bed866394167c05 # from nightly, Feb 25th
11+
TENSORFLOW_HASH=25cf6546b44bdb3d3f17277a4b49b027dcdc0ae7 # from nightly, Mar 23rd
1212

1313
git-shallow-clone https://github.com/tensorflow/tensorflow.git $TENSORFLOW_HASH
1414

@@ -17,8 +17,8 @@ git-shallow-clone https://github.com/tensorflow/tensorflow.git $TENSORFLOW_HASH
1717

1818
# Apply TensorFlow WIP patches here
1919

20-
# https://github.com/tensorflow/tensorflow/pull/100882 - build(aarch64): Update Compute Library to 52.4.0
21-
apply-github-patch tensorflow/tensorflow f15bec785f25dacaf9ae18250e499274e4ec7fb1
20+
# https://github.com/tensorflow/tensorflow/pull/113368 - Bump Compute Library version from v24.12 to v52.8.0
21+
apply-github-patch tensorflow/tensorflow 189698cf3b3dab5375508ff2e02f6aba3a586323
2222
# https://github.com/tensorflow/tensorflow/pull/102272 - Fix AArch64 CPUIDInfo init
2323
apply-github-patch tensorflow/tensorflow b6a9e1c1173d675533ffbb71c5eb36c7060ae2d0
2424
)

0 commit comments

Comments
 (0)