From 550506894a66efff73f4207646ab1f860d4a0564 Mon Sep 17 00:00:00 2001 From: Thomas Wang <1am9trash@gmail.com> Date: Fri, 6 Mar 2026 18:31:04 +0800 Subject: [PATCH] [AMD] Upgrade aiter version (#19936) Co-authored-by: HaiShaw --- docker/rocm.Dockerfile | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/docker/rocm.Dockerfile b/docker/rocm.Dockerfile index 21f8352a4..60631e688 100644 --- a/docker/rocm.Dockerfile +++ b/docker/rocm.Dockerfile @@ -27,7 +27,7 @@ ENV BUILD_TRITON="0" ENV BUILD_LLVM="0" ENV BUILD_AITER_ALL="1" ENV BUILD_MOONCAKE="1" -ENV AITER_COMMIT="v0.1.10.post3" +ENV AITER_COMMIT="v0.1.11.post1" # =============================== # Base image 942 with rocm720 and args @@ -37,7 +37,7 @@ ENV BUILD_TRITON="1" ENV BUILD_LLVM="0" ENV BUILD_AITER_ALL="1" ENV BUILD_MOONCAKE="1" -ENV AITER_COMMIT="v0.1.10.post3" +ENV AITER_COMMIT="v0.1.11.post1" # =============================== # Base image 950 and args @@ -47,7 +47,7 @@ ENV BUILD_TRITON="0" ENV BUILD_LLVM="0" ENV BUILD_AITER_ALL="1" ENV BUILD_MOONCAKE="1" -ENV AITER_COMMIT="v0.1.10.post3" +ENV AITER_COMMIT="v0.1.11.post1" # =============================== # Base image 950 with rocm720 and args @@ -57,7 +57,7 @@ ENV BUILD_TRITON="1" ENV BUILD_LLVM="0" ENV BUILD_AITER_ALL="1" ENV BUILD_MOONCAKE="1" -ENV AITER_COMMIT="v0.1.10.post3" +ENV AITER_COMMIT="v0.1.11.post1" # =============================== # Chosen arch and args @@ -172,6 +172,7 @@ RUN if [ "$BUILD_LLVM" = "1" ]; then \ # leak into AITER's version when AITER uses setuptools_scm) ENV SETUPTOOLS_SCM_PRETEND_VERSION= RUN pip uninstall -y aiter \ + && pip install flydsl==0.0.1.dev95158637 \ && pip install psutil pybind11 # Required by AITER setup.py RUN git clone ${AITER_REPO} \ && cd aiter \