From 1b5437cec8101677028085fc512b078839051612 Mon Sep 17 00:00:00 2001 From: Micah Williamson Date: Thu, 28 May 2026 11:42:43 -0500 Subject: [PATCH] [ROCm] Bump ROCm to 7.2.3 (#43136) Signed-off-by: Micah Williamson --- .buildkite/release-pipeline.yaml | 2 +- docker/Dockerfile.rocm_base | 27 ++------------------------- 2 files changed, 3 insertions(+), 26 deletions(-) diff --git a/.buildkite/release-pipeline.yaml b/.buildkite/release-pipeline.yaml index df9b80f7f9a..a34f534e54d 100644 --- a/.buildkite/release-pipeline.yaml +++ b/.buildkite/release-pipeline.yaml @@ -737,7 +737,7 @@ steps: - "bash tools/vllm-rocm/generate-rocm-wheels-root-index.sh" env: S3_BUCKET: "vllm-wheels" - VARIANT: "rocm722" + VARIANT: "rocm723" # ROCm Job 6: Build ROCm Release Docker Image - label: ":docker: Build release image - x86_64 - ROCm" diff --git a/docker/Dockerfile.rocm_base b/docker/Dockerfile.rocm_base index cb6d09f5bb6..195067b51a2 100644 --- a/docker/Dockerfile.rocm_base +++ b/docker/Dockerfile.rocm_base @@ -1,4 +1,4 @@ -ARG BASE_IMAGE=rocm/dev-ubuntu-22.04:7.2.2-complete +ARG BASE_IMAGE=rocm/dev-ubuntu-22.04:7.2.3-complete ARG TRITON_BRANCH="ba5c1517" ARG TRITON_REPO="https://github.com/ROCm/triton.git" ARG PYTORCH_BRANCH="8514f051" # release/2.10 as of 3/17 @@ -104,28 +104,6 @@ ENV SCCACHE_REGION=${USE_SCCACHE:+${SCCACHE_REGION_NAME}} ENV SCCACHE_S3_NO_CREDENTIALS=${USE_SCCACHE:+${SCCACHE_S3_NO_CREDENTIALS}} ENV SCCACHE_IDLE_TIMEOUT=${USE_SCCACHE:+0} -# torch profiler hotfix for 7.2.2: rebuild CLR with https://github.com/ROCm/rocm-systems/pull/5062 -# will be removed once we move to ROCm 7.2.3 -RUN apt-get update && apt-get install -y rocm-llvm-dev -RUN pip install CppHeaderParser -RUN git clone --no-checkout --filter=blob:none https://github.com/ROCm/rocm-systems /tmp/rocm-systems \ - && cd /tmp/rocm-systems \ - && git sparse-checkout init --cone \ - && git sparse-checkout set projects/hip projects/clr \ - && git checkout 35e8c7bf8911862e5389509800e65fdf125412b3 \ - && export CLR_DIR=/tmp/rocm-systems/projects/clr \ - && export HIP_DIR=/tmp/rocm-systems/projects/hip \ - && mkdir -p $CLR_DIR/build && cd $CLR_DIR/build \ - && cmake \ - -DHIP_COMMON_DIR=$HIP_DIR \ - -DCMAKE_PREFIX_PATH="/opt/rocm/" \ - -DCLR_BUILD_HIP=ON \ - -DCLR_BUILD_OCL=OFF \ - -DHIP_PLATFORM=amd \ - .. \ - && make -j$(nproc) \ - && make install \ - && rm -rf /tmp/rocm-systems ### ### Triton Build @@ -259,9 +237,8 @@ ARG AITER_REPO ARG USE_SCCACHE RUN --mount=type=bind,from=build_pytorch,src=/app/install/,target=/install \ pip install /install/*.whl -RUN git clone --recursive ${AITER_REPO} +RUN git clone --recursive --branch ${AITER_BRANCH} ${AITER_REPO} RUN cd aiter \ - && git checkout ${AITER_BRANCH} \ && git submodule update --init --recursive \ && pip install -r requirements.txt RUN pip install pyyaml && cd aiter \