Update Dockerfile.rocm

Use pre-OptimizeLDSUsage Triton
prarit · Jul 26, 2024 · fcab51c · fcab51c
1 parent 02679fe
commit fcab51c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile.rocm b/Dockerfile.rocm
@@ -104,7 +104,7 @@ FROM export_cupy_${BUILD_CUPY} AS export_cupy
 # -----------------------
 # Triton build stages
 FROM base AS build_triton
-ARG TRITON_BRANCH="main"
+ARG TRITON_BRANCH="6ddb79b"
 ARG TRITON_REPO="https://github.com/OpenAI/triton.git"
 RUN git clone ${TRITON_REPO} \
     && cd triton \