Merge pull request #439 from puneetmatharu/refactor-dep-versions-into-versions-sh

Refactor dep versions into `versions.sh` and patch `torch` wheel naming

Author: puneetmatharu

ML-Frameworks/pytorch-aarch64/CHANGELOG.md

@@ -21,7 +21,7 @@ where `YY` is the year, and `MM` the month of the increment.
 
 ### Fixed
 
-## [r26.02] 2026-02-09
+## [r26.02] 2026-02-11
 https://github.com/ARM-software/Tool-Solutions/tree/r26.02
 
 ### Added
@@ -30,7 +30,7 @@ https://github.com/ARM-software/Tool-Solutions/tree/r26.02
 
 ### Changed
  - Updates hashes for:
-   - `PYTORCH_HASH` to `77da53a7356e033e3fc1e03fdd960fc4ad117882`, 2.11.0.dev20260130 from viable/strict, Jan 30th.
+   - `PYTORCH_HASH` to `77da53a7356e033e3fc1e03fdd960fc4ad117882`, 2.11.0.dev20260129 from viable/strict, Jan 29th.
    - `IDEEP_HASH` to `bbb9ffb9e0c401ca058b7f35a6ebe7d0e08ffd34`, from ideep_pytorch, Jan 30th.
    - `ONEDNN_HASH` to `804f364c04ad8a763d534abaabc99bf99c2754e0`, from main, Jan 30th.
    - `TORCH_AO_HASH` to `30fcb156945ecacd515775414d37c09bfe60727e`, from main, Jan 30th.

ML-Frameworks/pytorch-aarch64/Dockerfile

@@ -14,6 +14,7 @@ FROM ${DOCKER_IMAGE_MIRROR}${BASE_IMAGE} AS workshop
 
 ARG TORCH_WHEEL
 ARG TORCH_AO_WHEEL
+ARG TORCHVISION_NIGHTLY
 
 ENV DEBIAN_FRONTEND=noninteractive
 ENV DOCKER_USER=debian
@@ -59,7 +60,7 @@ COPY --chown=$DOCKER_USER:$DOCKER_USER ${TORCH_AO_WHEEL} /home/$DOCKER_USER/
 # Install wheels
 RUN set -eux && uv pip install --no-deps "$(basename "$TORCH_WHEEL")" && rm "$(basename "$TORCH_WHEEL")"
 RUN set -eux && uv pip install --no-deps "$(basename "$TORCH_AO_WHEEL")" && rm "$(basename "$TORCH_AO_WHEEL")"
-RUN uv pip install --pre torchvision==0.25.0.dev20260130 --index-url https://download.pytorch.org/whl/nightly/cpu --no-deps
+RUN uv pip install --pre torchvision==${TORCHVISION_NIGHTLY} --index-url https://download.pytorch.org/whl/nightly/cpu --no-deps
 
 # Copy examples/tests into image
 COPY --chown=$DOCKER_USER:$DOCKER_USER examples/ /home/$DOCKER_USER/

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

@@ -15,15 +15,15 @@
 # and
 #   pytorch/.github/workflows/generated-linux-aarch64-binary-manywheel-nightly.yml
 
+source ./versions.sh
+
 set -eux -o pipefail
 
 docker_exec() {
     docker exec "$TORCH_BUILD_CONTAINER" "$@"
 }
 
 PYTHON_VERSION="3.12"
-OPENBLAS_VERSION="v0.3.30"
-ACL_VERSION="v52.8.0"
 
 # Specify DOCKER_IMAGE_MIRROR if you want to use a mirror of hub.docker.com
 IMAGE_NAME="${DOCKER_IMAGE_MIRROR:-}pytorch/manylinux2_28_aarch64-builder:cpu-aarch64-69d4c1f80b5e7da224d4f9c2170ef100e75dfe03"
@@ -155,7 +155,7 @@ docker_exec rm -rf "${PYTORCH_CONTAINER_DIR}/dist"
 # commit, this allows us to also install the matching torch* packages, set in
 # the Dockerfile. This is what PyTorch does in its nightly pipeline, see
 # pytorch/.ci/aarch64_linux/aarch64_wheel_ci_build.py for this logic.
-build_date=$(cd "$PYTORCH_LOCAL_DIR" && git log --pretty=format:%cs -1 | tr -d '-')
+build_date=$(cd "$PYTORCH_LOCAL_DIR" && git show -s --format=%cs "${PYTORCH_HASH}" | tr -d '-')
 version=$(cat "$PYTORCH_LOCAL_DIR/version.txt" | tr -d "[:space:]")
 OVERRIDE_PACKAGE_VERSION="${version%??}.dev${build_date}${TORCH_RELEASE_ID:+"+$TORCH_RELEASE_ID"}"
 

ML-Frameworks/pytorch-aarch64/dockerize.sh

@@ -4,6 +4,8 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
+source ./versions.sh
+
 set -eux -o pipefail
 
 help_str="dockerize.sh takes a PyTorch wheel as the first argument and an ao wheel
@@ -29,6 +31,7 @@ docker buildx \
     --build-arg DOCKER_IMAGE_MIRROR \
     --build-arg TORCH_WHEEL=$1 \
     --build-arg TORCH_AO_WHEEL=$2 \
+    --build-arg TORCHVISION_NIGHTLY="${TORCHVISION_NIGHTLY}" \
     .
 
 [[ $* == *--build-only* ]] && exit 0

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

@@ -5,15 +5,10 @@
 # SPDX-License-Identifier: Apache-2.0
 
 source ../utils/git-utils.sh
+source ./versions.sh
 
 set -eux -o pipefail
 
-PYTORCH_HASH=77da53a7356e033e3fc1e03fdd960fc4ad117882   # 2.11.0.dev20260130 from viable/strict, Jan 30th
-IDEEP_HASH=bbb9ffb9e0c401ca058b7f35a6ebe7d0e08ffd34     # From ideep_pytorch, Jan 30th
-ONEDNN_HASH=804f364c04ad8a763d534abaabc99bf99c2754e0    # From main, Jan 30th
-TORCH_AO_HASH=30fcb156945ecacd515775414d37c09bfe60727e  # From main, Jan 30th
-KLEIDIAI_HASH=5addaad73ebbb02e7dde6c50fff3bdb2ae8c407f  # v1.20.0 from main, Jan 30th
-
 git-shallow-clone https://github.com/pytorch/pytorch.git $PYTORCH_HASH
 (
     # Apply patches to PyTorch build
@@ -102,4 +97,4 @@ git-shallow-clone https://github.com/pytorch/ao.git $TORCH_AO_HASH
     # Remove cutlass directory
     cd ao
     git rm third_party/cutlass
-)
+)

ML-Frameworks/pytorch-aarch64/versions.sh

@@ -0,0 +1,21 @@
+#!/bin/bash
+
+# SPDX-FileCopyrightText: Copyright 2026 Arm Limited and affiliates.
+#
+# SPDX-License-Identifier: Apache-2.0
+
+# Source-of-truth versions and hashes for this repo
+
+# get-source.sh deps
+PYTORCH_HASH=77da53a7356e033e3fc1e03fdd960fc4ad117882   # 2.11.0.dev20260129 from viable/strict, Jan 29th
+IDEEP_HASH=bbb9ffb9e0c401ca058b7f35a6ebe7d0e08ffd34     # From ideep_pytorch, Jan 30th
+ONEDNN_HASH=804f364c04ad8a763d534abaabc99bf99c2754e0    # From main, Jan 30th
+TORCH_AO_HASH=30fcb156945ecacd515775414d37c09bfe60727e  # From main, Jan 30th
+KLEIDIAI_HASH=5addaad73ebbb02e7dde6c50fff3bdb2ae8c407f  # v1.20.0 from main, Jan 30th
+
+# build-wheel.sh deps
+ACL_VERSION="v52.8.0"
+OPENBLAS_VERSION="v0.3.30"
+
+# Dockerfile deps
+TORCHVISION_NIGHTLY="0.25.0.dev20260130"

ML-Frameworks/tensorflow-aarch64/CHANGELOG.md

@@ -21,7 +21,7 @@ where `YY` is the year, and `MM` the month of the increment.
 
 ### Fixed
 
-## [r26.02] 2026-02-09
+## [r26.02] 2026-02-11
 https://github.com/ARM-software/Tool-Solutions/tree/r26.02
 
 ### Added