Use main triton branch

Revert: #107 After triton fix landed: triton-lang/triton#4413
ROCm · Aug 1, 2024 · 9f273d0 · 9f273d0
1 parent a6414b8
commit 9f273d0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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} \