Merge pull request #422 from puneetmatharu/bump-builder-images-for-new-libgomp

Author: puneetmatharu

ML-Frameworks/pytorch-aarch64/CHANGELOG.md

@@ -1,5 +1,5 @@
 <!--
-SPDX-FileCopyrightText: Copyright 2022-2025 Arm Limited and affiliates.
+SPDX-FileCopyrightText: Copyright 2022-2026 Arm Limited and affiliates.
 
 SPDX-License-Identifier: Apache-2.0
 -->
@@ -18,6 +18,8 @@ where `YY` is the year, and `MM` the month of the increment.
  - Adds PyTorch [PR #170062](https://github.com/pytorch/pytorch/pull/170062), to add ccache support to ACL/OpenBLAS and manywheel build script.
 
 ### Changed
+ - PyTorch and TorchAO manylinux AArch64 builder image from `cpu-aarch64-d8be0384e085f551506bd739678109fa0f5ee7ac`
+   to `cpu-aarch64-69d4c1f80b5e7da224d4f9c2170ef100e75dfe03`
  - Updates the URL used to download weights for ResNet50 used by `classify_image.py`.
  - Updates `torchvision` from 0.25.0.dev20251104 to 0.25.0.dev20260111.
  - Updates Ubuntu and Python version to 24.04 and 3.12, respectively.

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

@@ -12,7 +12,7 @@ set -eux -o pipefail
 PYTHON_VERSION="3.12"
 
 # 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-d8be0384e085f551506bd739678109fa0f5ee7ac"
+IMAGE_NAME="${DOCKER_IMAGE_MIRROR:-}pytorch/manylinux2_28_aarch64-builder:cpu-aarch64-69d4c1f80b5e7da224d4f9c2170ef100e75dfe03"
 TORCH_BUILD_CONTAINER_ID_FILE="${PWD}/.torch_ao_build_container_id"
 
 TEST_VENV=aarch64_env_test_torch_ao

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

@@ -25,7 +25,7 @@ PYTHON_TAG="cp$(echo "$PYTHON_VERSION" | tr -d .)-cp$(echo "$PYTHON_VERSION" | t
 PYTHON_BIN="/opt/python/${PYTHON_TAG}/bin"
 
 # 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-d8be0384e085f551506bd739678109fa0f5ee7ac"
+IMAGE_NAME="${DOCKER_IMAGE_MIRROR:-}pytorch/manylinux2_28_aarch64-builder:cpu-aarch64-69d4c1f80b5e7da224d4f9c2170ef100e75dfe03"
 TORCH_BUILD_CONTAINER_ID_FILE="${PWD}/.torch_build_container_id"
 
 # Output dir for PyTorch wheel and other artifacts