Dockerfile.rocm.ubi: do not use nightly pytorch_triton

opendatahub-io · Sep 16, 2024 · 91c4c51 · 91c4c51
1 parent 4fae090
commit 91c4c51
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/Dockerfile.rocm.ubi b/Dockerfile.rocm.ubi
@@ -132,7 +132,6 @@ FROM rocm_devel AS build_triton
 # Whether to build triton on rocm
 ARG BUILD_TRITON="0"
 ARG TRITON_BRANCH="e0fc12c"
-ARG TRITON_WHEEL_SHA=21eae954ef
 
 # Build triton wheel if `BUILD_TRITON` is set to 1
 RUN --mount=type=cache,target=/root/.cache/ccache \
@@ -148,7 +147,7 @@ RUN --mount=type=cache,target=/root/.cache/ccache \
         python3 setup.py bdist_wheel --dist-dir=/install; \
     else \
         mkdir -p /install && \
-        wget -P /install "https://download.pytorch.org/whl/nightly/pytorch_triton_rocm-3.0.0%2B${TRITON_WHEEL_SHA}-cp312-cp312-linux_x86_64.whl"; \
+        wget -P /install "https://download.pytorch.org/whl/pytorch_triton_rocm-3.0.0-cp312-cp312-linux_x86_64.whl#sha256=650ce7a6624ecbbe90c30fccb472a57772c08bc6f12312246f7af584f275b7ca"; \
     fi
 
 ##################################################################################################