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