AI day rocm base

Signed-off-by:  <>

Author: root

docker/Dockerfile.rocm_base

@@ -2,12 +2,12 @@ ARG BASE_IMAGE=compute-artifactory.amd.com:5000/rocm-plus-docker/framework/compu
 # ARG HIPBLASLT_BRANCH="aa0bda7b"
 # ARG HIPBLAS_COMMON_BRANCH="9b80ba8e"
 # ARG LEGACY_HIPBLASLT_OPTION=
-# ARG TRITON_BRANCH="96316ce5"
-# ARG TRITON_REPO="https://github.com/ROCm/triton.git"
-# ARG PYTORCH_BRANCH="74091a9"
-# ARG PYTORCH_VISION_BRANCH="0.22.0a0"
-# ARG PYTORCH_REPO="https://github.com/ROCm/pytorch.git"
-# ARG PYTORCH_VISION_REPO="https://github.com/pytorch/vision.git"
+ARG TRITON_BRANCH="0e78e54a"
+ARG TRITON_REPO="https://github.com/ROCm/triton.git"
+ARG PYTORCH_BRANCH="37f92bb"
+ARG PYTORCH_VISION_BRANCH="0.22.0a0"
+ARG PYTORCH_REPO="https://github.com/ROCm/pytorch.git"
+ARG PYTORCH_VISION_REPO="https://github.com/pytorch/vision.git"
 ARG FA_BRANCH="8ede036"
 ARG FA_REPO="https://github.com/Dao-AILab/flash-attention.git"
 ARG AITER_BRANCH="498ff21"