Skip to content

Commit

Permalink
Dockerfile.rocm.ubi: do not use nightly pytorch_triton
Browse files Browse the repository at this point in the history
  • Loading branch information
dtrifiro committed Sep 16, 2024
1 parent 4fae090 commit 91c4c51
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions Dockerfile.rocm.ubi
Original file line number Diff line number Diff line change
Expand Up @@ -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 \
Expand All @@ -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

##################################################################################################
Expand Down

0 comments on commit 91c4c51

Please sign in to comment.