Use main triton branch

Revert: #107
After triton fix landed: triton-lang/triton#4413

Author: binarman

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} \