Install torch 2.5.0 for ROCm 6.2 as well

Dockerfile.rocm

@@ -41,9 +41,15 @@ RUN case "$(ls /opt | grep -Po 'rocm-[0-9]\.[0-9]')" in \
         *"rocm-6.1"*) \
             python3 -m pip uninstall -y torch torchvision \
             && python3 -m pip install --no-cache-dir --pre \
-                torch==2.5.0.dev20240726 \
-                torchvision==0.20.0.dev20240726 \
+                torch==2.5.0.dev20240826 \
+                torchvision==0.20.0.dev20240826 \
                --index-url https://download.pytorch.org/whl/nightly/rocm6.1;; \
+        *"rocm-6.2"*) \
+        python3 -m pip uninstall -y torch torchvision \
+        && python3 -m pip install --no-cache-dir --pre \
+            torch==2.5.0.dev20240826 \
+            torchvision==0.20.0.dev20240826 \
+            --index-url https://download.pytorch.org/whl/nightly/rocm6.2;; \
         *) ;; esac
 
 ENV LLVM_SYMBOLIZER_PATH=/opt/rocm/llvm/bin/llvm-symbolizer
@@ -107,7 +113,7 @@ FROM export_flash_attn_${BUILD_FA} AS export_flash_attn
 # Triton build stages
 FROM base AS build_triton
 ARG TRITON_BRANCH="e0fc12c"
-ARG TRITON_REPO="https://github.com/OpenAI/triton.git"
+ARG TRITON_REPO="https://github.com/triton-lang/triton.git"
 RUN git clone ${TRITON_REPO} \
     && cd triton \
     && git checkout ${TRITON_BRANCH} \