A high-throughput and memory-efficient inference and serving engine for LLMs
[ROCm] Bump Torch to 2.11 (#45362)
Signed-off-by: Micah Williamson <micah.williamson@amd.com>
Micah Williamson committed
Jun 12, 2026 at 15:22 UTC
39cb9bf292ec5811b0df9e5461b9504801c1cf91
1 file changed
+2
-4
docker/Dockerfile.rocm_base
+2
-4
@@ -1,7 +1,7 @@
1
ARG BASE_IMAGE=rocm/dev-ubuntu-22.04:7.2.3-complete
2
-ARG TRITON_BRANCH="ba5c1517"
2
+ARG TRITON_BRANCH="0f380657"
3
ARG TRITON_REPO="https://github.com/ROCm/triton.git"
4
-ARG PYTORCH_BRANCH="8514f051" # release/2.10 as of 3/17
4
+ARG PYTORCH_BRANCH="d0c8b1f3" # release/2.11 as of 6/09
5
ARG PYTORCH_REPO="https://github.com/ROCm/pytorch.git"
6
ARG PYTORCH_VISION_BRANCH="v0.24.1"
7
ARG PYTORCH_VISION_REPO="https://github.com/pytorch/vision.git"
@@ -114,12 +114,10 @@ ARG TRITON_REPO
114
RUN git clone ${TRITON_REPO}
115
# Cherry picking the following
116
# https://github.com/triton-lang/triton/pull/8991
117
-# https://github.com/triton-lang/triton/pull/9541
117
RUN cd triton \
118
&& git checkout ${TRITON_BRANCH} \
119
&& git config --global user.email "you@example.com" && git config --global user.name "Your Name" \
120
&& git cherry-pick 555d04f \
122
- && git cherry-pick dd998b6 \
121
&& if [ ! -f setup.py ]; then cd python; fi \
122
&& python3 setup.py bdist_wheel --dist-dir=dist \
123
&& mkdir -p /app/install && cp dist/*.whl /app/install