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

[AMD][Dockerfile] Support build-arg AITER_COMMIT for rocm.Dockerfile (#21949)

Browse Source
This commit is contained in:
Duyi-Wang
2026-04-03 01:54:28 -07:00
committed by GitHub
parent cd75d54fc5
commit ac593fed90
2 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/rocm.Dockerfile
+6 -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.11.post1" ENV AITER_COMMIT_DEFAULT="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.11.post1" ENV AITER_COMMIT_DEFAULT="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.11.post1" ENV AITER_COMMIT_DEFAULT="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.11.post1" ENV AITER_COMMIT_DEFAULT="v0.1.11.post1"
# =============================== # ===============================
# Chosen arch and args # Chosen arch and args
@@ -79,6 +79,8 @@ ARG TRITON_REPO="https://github.com/triton-lang/triton.git"
ARG TRITON_COMMIT="42270451990532c67e69d753fbd026f28fcc4840" ARG TRITON_COMMIT="42270451990532c67e69d753fbd026f28fcc4840"
ARG AITER_REPO="https://github.com/ROCm/aiter.git" ARG AITER_REPO="https://github.com/ROCm/aiter.git"
ARG AITER_COMMIT=""
ENV AITER_COMMIT="${AITER_COMMIT:-${AITER_COMMIT_DEFAULT}}"
ARG LLVM_REPO="https://github.com/jrbyrnes/llvm-project.git" ARG LLVM_REPO="https://github.com/jrbyrnes/llvm-project.git"
ARG LLVM_BRANCH="MainOpSelV2" ARG LLVM_BRANCH="MainOpSelV2"
scripts/ci/amd/amd_ci_install_dependency.sh
+4 -4
View File
@@ -200,15 +200,15 @@ echo "[CI-AITER-CHECK] Runner GPU_ARCH=${GPU_ARCH}"
if [[ "${GPU_ARCH}" == "mi35x" ]]; then if [[ "${GPU_ARCH}" == "mi35x" ]]; then
echo "[CI-AITER-CHECK] Using gfx950 block from Dockerfile..." echo "[CI-AITER-CHECK] Using gfx950 block from Dockerfile..."
REPO_AITER_COMMIT=$(grep -F -A20 'FROM $BASE_IMAGE_950 AS gfx950' docker/rocm.Dockerfile \ REPO_AITER_COMMIT=$(grep -F -A20 'FROM $BASE_IMAGE_950 AS gfx950' docker/rocm.Dockerfile \
| grep 'AITER_COMMIT=' \ | grep 'AITER_COMMIT_DEFAULT=' \
| head -n1 \ | head -n1 \
| sed 's/.*AITER_COMMIT="\([^"]*\)".*/\1/') | sed 's/.*AITER_COMMIT_DEFAULT="\([^"]*\)".*/\1/')
else else
echo "[CI-AITER-CHECK] Using gfx942 block from Dockerfile..." echo "[CI-AITER-CHECK] Using gfx942 block from Dockerfile..."
REPO_AITER_COMMIT=$(grep -F -A20 'FROM $BASE_IMAGE_942 AS gfx942' docker/rocm.Dockerfile \ REPO_AITER_COMMIT=$(grep -F -A20 'FROM $BASE_IMAGE_942 AS gfx942' docker/rocm.Dockerfile \
| grep 'AITER_COMMIT=' \ | grep 'AITER_COMMIT_DEFAULT=' \
| head -n1 \ | head -n1 \
| sed 's/.*AITER_COMMIT="\([^"]*\)".*/\1/') | sed 's/.*AITER_COMMIT_DEFAULT="\([^"]*\)".*/\1/')
fi fi