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_AITER_ALL="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
@@ -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.11.post1"
ENV AITER_COMMIT_DEFAULT="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.11.post1"
ENV AITER_COMMIT_DEFAULT="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.11.post1"
ENV AITER_COMMIT_DEFAULT="v0.1.11.post1"
# ===============================
# Chosen arch and args
@@ -79,6 +79,8 @@ ARG TRITON_REPO="https://github.com/triton-lang/triton.git"
ARG TRITON_COMMIT="42270451990532c67e69d753fbd026f28fcc4840"
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_BRANCH="MainOpSelV2"