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

[AMD] Update ROCm AITER pin to c16d44b (#35810)

Browse Source
This commit is contained in:
Bingxu Chen
2026-08-21 00:02:17 -07:00
committed by GitHub
parent f3fe81583e
commit 6a12583679
1 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/rocm.Dockerfile
+9 -9
View File
@@ -46,7 +46,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_DEFAULT="d9e5ef7ce08ee7045d583aed768cff41aa9210fe" ENV AITER_COMMIT_DEFAULT="c16d44b93a528b2a4bfd6d8d3409116d465872a9"
# =============================== # ===============================
# Base image 942 with rocm720 and args # Base image 942 with rocm720 and args
@@ -56,7 +56,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_DEFAULT="d9e5ef7ce08ee7045d583aed768cff41aa9210fe" ENV AITER_COMMIT_DEFAULT="c16d44b93a528b2a4bfd6d8d3409116d465872a9"
# =============================== # ===============================
# Base image 942 with rocm724 and args (Python 3.12 + torch 2.11) # Base image 942 with rocm724 and args (Python 3.12 + torch 2.11)
@@ -66,7 +66,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_DEFAULT="d9e5ef7ce08ee7045d583aed768cff41aa9210fe" ENV AITER_COMMIT_DEFAULT="c16d44b93a528b2a4bfd6d8d3409116d465872a9"
# Pin the ROCm torch stack for every pip invocation in this flavor. The file is # Pin the ROCm torch stack for every pip invocation in this flavor. The file is
# filled in after the torch 2.11 upgrade below; it must already exist (empty is # filled in after the torch 2.11 upgrade below; it must already exist (empty is
# valid) because pip reads PIP_CONSTRAINT from the first pip call onwards. # valid) because pip reads PIP_CONSTRAINT from the first pip call onwards.
@@ -90,7 +90,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_DEFAULT="d9e5ef7ce08ee7045d583aed768cff41aa9210fe" ENV AITER_COMMIT_DEFAULT="c16d44b93a528b2a4bfd6d8d3409116d465872a9"
# =============================== # ===============================
# Base image 950 with rocm720 and args # Base image 950 with rocm720 and args
@@ -100,7 +100,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_DEFAULT="d9e5ef7ce08ee7045d583aed768cff41aa9210fe" ENV AITER_COMMIT_DEFAULT="c16d44b93a528b2a4bfd6d8d3409116d465872a9"
# =============================== # ===============================
# Base image 950 with rocm724 and args (Python 3.12 + torch 2.11) # Base image 950 with rocm724 and args (Python 3.12 + torch 2.11)
@@ -110,7 +110,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_DEFAULT="d9e5ef7ce08ee7045d583aed768cff41aa9210fe" ENV AITER_COMMIT_DEFAULT="c16d44b93a528b2a4bfd6d8d3409116d465872a9"
# Pin the ROCm torch stack for every pip invocation in this flavor. The file is # Pin the ROCm torch stack for every pip invocation in this flavor. The file is
# filled in after the torch 2.11 upgrade below; it must already exist (empty is # filled in after the torch 2.11 upgrade below; it must already exist (empty is
# valid) because pip reads PIP_CONSTRAINT from the first pip call onwards. # valid) because pip reads PIP_CONSTRAINT from the first pip call onwards.
@@ -351,9 +351,9 @@ ENV AITER_USE_SYSTEM_TRITON=1
RUN pip uninstall -y aiter RUN pip uninstall -y aiter
# Use `checkout -f` so the smudge-filter-induced "dirty" working tree from # Use `checkout -f` so the smudge-filter-induced "dirty" working tree from
# AITER's .gitattributes (*.csv text eol=lf, added in ROCm/aiter#3370) does not # AITER's .gitattributes (*.csv text eol=lf, added in ROCm/aiter#3370) does not
# block switching to commits that predate that rule (e.g. the current default # block AITER_COMMIT overrides that predate that rule. The working tree was just
# AITER_COMMIT_DEFAULT). The working tree was just produced by a fresh # produced by a fresh `git clone` above, so there are no real user changes to
# `git clone` above, so there are no real user changes to preserve. # preserve.
RUN git clone ${AITER_REPO} \ RUN git clone ${AITER_REPO} \
&& cd aiter \ && cd aiter \
&& git checkout -f ${AITER_COMMIT} \ && git checkout -f ${AITER_COMMIT} \