Skip to content

Commit 0143de9

Browse files
micah-wilusberkeley
authored andcommitted
[ROCm] Update Triton, Torch, and AITER branches for ROCm base Dockerfile (vllm-project#27206)
Signed-off-by: Micah Williamson <[email protected]>
1 parent af429c9 commit 0143de9

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

docker/Dockerfile.rocm_base

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,13 @@
11
ARG BASE_IMAGE=rocm/dev-ubuntu-22.04:7.0-complete
2-
ARG TRITON_BRANCH="f9e5bf54"
2+
ARG TRITON_BRANCH="57c693b6"
33
ARG TRITON_REPO="https://github.com/ROCm/triton.git"
4-
ARG PYTORCH_BRANCH="b2fb6885"
4+
ARG PYTORCH_BRANCH="1c57644d"
55
ARG PYTORCH_VISION_BRANCH="v0.23.0"
66
ARG PYTORCH_REPO="https://github.com/ROCm/pytorch.git"
77
ARG PYTORCH_VISION_REPO="https://github.com/pytorch/vision.git"
88
ARG FA_BRANCH="0e60e394"
99
ARG FA_REPO="https://github.com/Dao-AILab/flash-attention.git"
10-
ARG AITER_BRANCH="2ab9f4cd"
10+
ARG AITER_BRANCH="eef23c7f"
1111
ARG AITER_REPO="https://github.com/ROCm/aiter.git"
1212

1313
FROM ${BASE_IMAGE} AS base

0 commit comments

Comments
 (0)