From 3d8d3fe84d3906f76257f1517730cce00ef47c20 Mon Sep 17 00:00:00 2001
From: Derek Holloway <derek@mistox.com>
Date: Tue, 12 May 2026 19:56:21 -0700
Subject: [PATCH] fix stage naming

---
 TheRock                      | 1 +
 rocm-base-aarch64-dockerfile | 6 +++---
 2 files changed, 4 insertions(+), 3 deletions(-)
 create mode 160000 TheRock

diff --git a/TheRock b/TheRock
new file mode 160000
index 0000000..b296f29
--- /dev/null
+++ b/TheRock
@@ -0,0 +1 @@
+Subproject commit b296f29cddce6316d52ba0b2b427ee152bcbebf5
diff --git a/rocm-base-aarch64-dockerfile b/rocm-base-aarch64-dockerfile
index d6ca9c4..921d059 100644
--- a/rocm-base-aarch64-dockerfile
+++ b/rocm-base-aarch64-dockerfile
@@ -67,7 +67,7 @@ RUN wget https://www.python.org/ftp/python/3.11.15/Python-3.11.15.tgz && \
     # Tested Working
 
 # Build ROCm latest
-FROM python AS ROCm
+FROM python AS rocm
 WORKDIR /build/therock
 COPY ./TheRock .
 ENV CFLAGS="-Wno-error=discarded-qualifiers -fcommon -w"
@@ -92,7 +92,7 @@ RUN env INSTALL_PREFIX=/usr/local ./dockerfiles/install_pinned_patchelf.sh && \
     # Work in progress [ Breaking due to x86 intrinsics not ported to Aarch64 ]
 
 # Build pytorch
-FROM ROCm AS pytorch
+FROM rocm AS pytorch
 # Setup Envronment
 ENV PYTORCH_ROCM_ARCH=1100
 ENV USE_ROCM=1
@@ -149,5 +149,5 @@ WORKDIR /build/torchaudio
 RUN git clone https://github.com/pytorch/vision.git
 
 # Build final stripped down image
-FROM ubuntu:26.04 as rocm-aarch64
+FROM ubuntu:26.04
 # Copy in rocm, pytorch, torchaudio, torchvision
\ No newline at end of file