Merge pull request opendatahub-io#107 from ROCm/Triton-workaround

Update Dockerfile.rocm
prarit · Jul 26, 2024 · edde7cc · edde7cc
2 parents 02679fe + fcab51c
commit edde7cc
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 \