minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[AMD] Upgrade aiter version (#19936)

Browse Source 
Co-authored-by: HaiShaw <hixiao@gmail.com>
This commit is contained in:
Thomas Wang
2026-03-06 02:31:04 -08:00
committed by GitHub
co-authored by HaiShaw
parent 84aaa69795
commit 550506894a
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/rocm.Dockerfile
+5 -4
View File
@@ -27,7 +27,7 @@ ENV BUILD_TRITON="0"
ENV BUILD_LLVM="0" ENV BUILD_LLVM="0"
ENV BUILD_AITER_ALL="1" ENV BUILD_AITER_ALL="1"
ENV BUILD_MOONCAKE="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 # Base image 942 with rocm720 and args
@@ -37,7 +37,7 @@ ENV BUILD_TRITON="1"
ENV BUILD_LLVM="0" ENV BUILD_LLVM="0"
ENV BUILD_AITER_ALL="1" ENV BUILD_AITER_ALL="1"
ENV BUILD_MOONCAKE="1" ENV BUILD_MOONCAKE="1"
ENV AITER_COMMIT="v0.1.10.post3" ENV AITER_COMMIT="v0.1.11.post1"
# =============================== # ===============================
# Base image 950 and args # Base image 950 and args
@@ -47,7 +47,7 @@ ENV BUILD_TRITON="0"
ENV BUILD_LLVM="0" ENV BUILD_LLVM="0"
ENV BUILD_AITER_ALL="1" ENV BUILD_AITER_ALL="1"
ENV BUILD_MOONCAKE="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 # Base image 950 with rocm720 and args
@@ -57,7 +57,7 @@ ENV BUILD_TRITON="1"
ENV BUILD_LLVM="0" ENV BUILD_LLVM="0"
ENV BUILD_AITER_ALL="1" ENV BUILD_AITER_ALL="1"
ENV BUILD_MOONCAKE="1" ENV BUILD_MOONCAKE="1"
ENV AITER_COMMIT="v0.1.10.post3" ENV AITER_COMMIT="v0.1.11.post1"
# =============================== # ===============================
# Chosen arch and args # Chosen arch and args
@@ -172,6 +172,7 @@ RUN if [ "$BUILD_LLVM" = "1" ]; then \
# leak into AITER's version when AITER uses setuptools_scm) # leak into AITER's version when AITER uses setuptools_scm)
ENV SETUPTOOLS_SCM_PRETEND_VERSION= ENV SETUPTOOLS_SCM_PRETEND_VERSION=
RUN pip uninstall -y aiter \ RUN pip uninstall -y aiter \
&& pip install flydsl==0.0.1.dev95158637 \
&& pip install psutil pybind11 # Required by AITER setup.py && pip install psutil pybind11 # Required by AITER setup.py
RUN git clone ${AITER_REPO} \ RUN git clone ${AITER_REPO} \
&& cd aiter \ && cd aiter \