From bdfd5da53e67bf6a7977995b4e4ba5da304f059a Mon Sep 17 00:00:00 2001 From: Bingxu Chen Date: Thu, 28 May 2026 17:04:58 +0800 Subject: [PATCH] [AMD] AITER Upgrade (#26562) --- docker/rocm.Dockerfile | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/docker/rocm.Dockerfile b/docker/rocm.Dockerfile index b0146deba..1db8997cd 100644 --- a/docker/rocm.Dockerfile +++ b/docker/rocm.Dockerfile @@ -31,7 +31,7 @@ ENV BUILD_TRITON="0" ENV BUILD_LLVM="0" ENV BUILD_AITER_ALL="1" ENV BUILD_MOONCAKE="1" -ENV AITER_COMMIT_DEFAULT="32e1e6d76988e4fbc67cabd9eb72a45a3c6a1bab" +ENV AITER_COMMIT_DEFAULT="46e6c92b3eb33f64823aaa1ff39a14586b059ef5" # =============================== # Base image 942 with rocm720 and args @@ -41,7 +41,7 @@ ENV BUILD_TRITON="1" ENV BUILD_LLVM="0" ENV BUILD_AITER_ALL="1" ENV BUILD_MOONCAKE="1" -ENV AITER_COMMIT_DEFAULT="32e1e6d76988e4fbc67cabd9eb72a45a3c6a1bab" +ENV AITER_COMMIT_DEFAULT="46e6c92b3eb33f64823aaa1ff39a14586b059ef5" # =============================== # Base image 950 and args @@ -51,7 +51,7 @@ ENV BUILD_TRITON="0" ENV BUILD_LLVM="0" ENV BUILD_AITER_ALL="1" ENV BUILD_MOONCAKE="1" -ENV AITER_COMMIT_DEFAULT="32e1e6d76988e4fbc67cabd9eb72a45a3c6a1bab" +ENV AITER_COMMIT_DEFAULT="46e6c92b3eb33f64823aaa1ff39a14586b059ef5" # =============================== # Base image 950 with rocm720 and args @@ -61,7 +61,7 @@ ENV BUILD_TRITON="1" ENV BUILD_LLVM="0" ENV BUILD_AITER_ALL="1" ENV BUILD_MOONCAKE="1" -ENV AITER_COMMIT_DEFAULT="32e1e6d76988e4fbc67cabd9eb72a45a3c6a1bab" +ENV AITER_COMMIT_DEFAULT="46e6c92b3eb33f64823aaa1ff39a14586b059ef5" # =============================== # Chosen arch and args @@ -200,7 +200,6 @@ RUN if [ "$BUILD_LLVM" = "1" ]; then \ # (SETUPTOOLS_SCM_PRETEND_VERSION is set later for SGLang nightly builds and would otherwise # leak into AITER's version when AITER uses setuptools_scm) -# cherry pick b639cb6 commit for aiter_mhc_pre fix, may be removed in next aiter upgrade ENV SETUPTOOLS_SCM_PRETEND_VERSION= # Keep the base image's Torch-compatible Triton by default. Override with # AITER_USE_SYSTEM_TRITON=0 when intentionally testing aiter-managed Triton. @@ -214,7 +213,6 @@ RUN pip uninstall -y aiter RUN git clone ${AITER_REPO} \ && cd aiter \ && git checkout -f ${AITER_COMMIT} \ - && git cherry-pick --no-commit b639cb63bcac4672dce33a731fad042a65cb3649 \ && git submodule update --init --recursive \ && pip install -r requirements.txt