fix stage naming

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