Merge pull request #339 from nSircombe/feature/r23.07_updates

Updates for 25.07 release.

Author: cfRod

ML-Frameworks/pytorch-aarch64/CHANGELOG.md

@@ -15,6 +15,23 @@ where `YY` is the year, and `MM` the month of the increment.
 
 ### Fixed
 
+## [25.07] 2025-07-11
+https://github.com/ARM-software/Tool-Solutions/tree/r25.07
+
+### Added
+
+### Changed
+ - Updates Dockerfile to minimise layers in the final image.
+ - Updates hashes for:
+   - PyTorch to 5dfd8a9c7a464bb42e81b8594eefd2fa865e5423, to 2.9.0.dev20250703 from viable/strict
+   - ideep to 6eb12eaad5e0f7d8c8613c744ac8ba5a0843cb99, from ideep_pytorch, July 3rd
+   - oneDNN to 0abfca1947b53c03ee74207e4710941ab6456f3b, from main, July 3rd
+   - Compute Library to f69b48afcc59f1b3b0d4544289249bebba489f0a, from main, June 26th
+
+### Removed
+
+### Fixed
+
 ## [r25.06] 2025-06-20
 https://github.com/ARM-software/Tool-Solutions/tree/r25.06
 

ML-Frameworks/pytorch-aarch64/Dockerfile

@@ -18,7 +18,9 @@
 
 # Specify DOCKER_IMAGE_MIRROR if you want to use a mirror of hub.docker.com
 ARG DOCKER_IMAGE_MIRROR=""
-FROM ${DOCKER_IMAGE_MIRROR}ubuntu:22.04
+FROM ${DOCKER_IMAGE_MIRROR}ubuntu:22.04 AS workshop
+
+ARG USERNAME
 
 ARG TORCH_WHEEL
 ENV TORCH_WHEEL=$TORCH_WHEEL
@@ -44,7 +46,7 @@ RUN apt-get update && apt-get install -y \
     && rm -rf /var/lib/apt/lists/*
 
 # DOCKER_USER for the Docker user
-ENV DOCKER_USER=ubuntu
+ENV DOCKER_USER=${USERNAME}
 
 # Setup default user
 RUN useradd --create-home -s /bin/bash -m $DOCKER_USER && echo "$DOCKER_USER:Portland" | chpasswd && adduser $DOCKER_USER sudo
@@ -112,4 +114,16 @@ RUN sudo mkdir -p /home/ubuntu/gen_ai_utils/ && \
     sudo git config --global --add safe.directory /home/ubuntu/gen_ai_utils/torchchat && \
     sudo git checkout 90749d280bbc116fcc121a1eda1b60f1dba5b675
 
+# Move build into final image as a single layer.
+FROM ${DOCKER_IMAGE_MIRROR}ubuntu:22.04
+
+ARG USERNAME
+
+ENV DOCKER_USER=${USERNAME}
+
+COPY --from=workshop / /
+RUN chown $DOCKER_USER:$DOCKER_USER /home/$DOCKER_USER
+
+USER $DOCKER_USER
+WORKDIR /home/$DOCKER_USER
 CMD ["bash", "-l"]

ML-Frameworks/pytorch-aarch64/build.sh

@@ -50,8 +50,9 @@ torch_wheel_name=$(grep -o "torch-.*.whl" $build_log | head -n -1 | tail -n 1)
 # Use the second to last match, otherwise grep finds itself
 torch_ao_wheel_name=$(grep -o "torchao-.*.whl" $build_log | head -n -1 | tail -n 1)
 
-docker build -t toolsolutions-pytorch:latest \
+DOCKER_BUILDKIT=1 docker build -t toolsolutions-pytorch:latest \
     --build-arg TORCH_WHEEL=results/$torch_wheel_name \
     --build-arg DOCKER_IMAGE_MIRROR \
     --build-arg TORCH_AO_WHEEL=ao/dist/$torch_ao_wheel_name \
+    --build-arg USERNAME=ubuntu \
     .

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

@@ -20,7 +20,7 @@
 source ../utils/git-utils.sh
 
 set -eux -o pipefail
-PYTORCH_HASH=5dfd8a9c7a464bb42e81b8594eefd2fa865e5423  # From viable/strict, July 3rd
+PYTORCH_HASH=5dfd8a9c7a464bb42e81b8594eefd2fa865e5423  # 2.9.0.dev20250703, from viable/strict, July 3rd
 IDEEP_HASH=6eb12eaad5e0f7d8c8613c744ac8ba5a0843cb99    # From ideep_pytorch, July 3rd
 ONEDNN_HASH=0abfca1947b53c03ee74207e4710941ab6456f3b   # From main, July 3rd
 ACL_HASH=f69b48afcc59f1b3b0d4544289249bebba489f0a      # From main, June 26th

ML-Frameworks/tensorflow-aarch64/CHANGELOG.md

@@ -15,6 +15,19 @@ where `YY` is the year, and `MM` the month of the increment.
 
 ### Fixed
 
+## [25.07] 2025-07-11
+https://github.com/ARM-software/Tool-Solutions/tree/r25.07
+
+### Added
+
+### Changed
+ - Updates Dockerfile to minimise layers in the final image.
+ - Updates TensorFlow hash to 70d137e8794f97728e094608ec9ea1c6c9fd42af, 2.20.0-dev0 from nightly, July 1st
+
+### Removed
+
+### Fixed
+
 ## [r25.06] 2025-06-20
 https://github.com/ARM-software/Tool-Solutions/tree/r25.06
 

ML-Frameworks/tensorflow-aarch64/Dockerfile

@@ -17,7 +17,9 @@
 
 # Specify DOCKER_IMAGE_MIRROR if you want to use a mirror of hub.docker.com
 ARG DOCKER_IMAGE_MIRROR=""
-FROM ${DOCKER_IMAGE_MIRROR}ubuntu:22.04
+FROM ${DOCKER_IMAGE_MIRROR}ubuntu:22.04 AS workshop
+
+ARG USERNAME
 
 ARG TENSORFLOW_WHEEL
 ENV TENSORFLOW_WHEEL=$TENSORFLOW_WHEEL
@@ -76,4 +78,16 @@ RUN pip install -r requirements.txt --no-deps
 # Setup Examples
 COPY --chown=$DOCKER_USER examples/ /home/$DOCKER_USER/
 
+# Move build into final image as a single layer.
+FROM ${DOCKER_IMAGE_MIRROR}ubuntu:22.04
+
+ARG USERNAME
+
+ENV DOCKER_USER=${USERNAME}
+
+COPY --from=workshop / /
+RUN chown $DOCKER_USER:$DOCKER_USER /home/$DOCKER_USER
+
+USER $DOCKER_USER
+WORKDIR /home/$DOCKER_USER
 CMD ["bash", "-l"]

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

@@ -21,7 +21,7 @@ source ../utils/git-utils.sh
 
 set -eux -o pipefail
 
-TENSORFLOW_HASH=70d137e8794f97728e094608ec9ea1c6c9fd42af # from nightly, July 1st
+TENSORFLOW_HASH=70d137e8794f97728e094608ec9ea1c6c9fd42af # 2.20.0-dev0 from nightly, July 1st
 
 git-shallow-clone https://github.com/tensorflow/tensorflow.git $TENSORFLOW_HASH