diff --git a/Dockerfile.rocm b/Dockerfile.rocm
index 39590cef0bd6a..b5788519c53a1 100644
--- a/Dockerfile.rocm
+++ b/Dockerfile.rocm
@@ -104,8 +104,8 @@ FROM export_cupy_${BUILD_CUPY} AS export_cupy
 # -----------------------
 # Triton build stages
 FROM base AS build_triton
-ARG TRITON_BRANCH="6ddb79b"
-ARG TRITON_REPO="https://github.com/OpenAI/triton.git"
+ARG TRITON_BRANCH="main"
+ARG TRITON_REPO="https://github.com/triton-lang/triton.git"
 RUN git clone ${TRITON_REPO} \
     && cd triton \
     && git checkout ${TRITON_BRANCH} \