From 9a954982de1aa42cb154e2a5b6a9d1b5dc8a57dc Mon Sep 17 00:00:00 2001
From: Baizhou Zhang <sobereddiezhang@gmail.com>
Date: Wed, 5 Nov 2025 19:03:26 -0800
Subject: [PATCH] [chore] SGLang tag management in Dockerfile (#12734)

---
 .github/workflows/release-docker-cu13.yml | 1 +
 .github/workflows/release-docker-dev.yml  | 1 +
 docker/Dockerfile                         | 6 +++++-
 scripts/release/bump_sglang_version.py    | 1 +
 4 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/release-docker-cu13.yml b/.github/workflows/release-docker-cu13.yml
index 5da630ef5..7a0a7d744 100644
--- a/.github/workflows/release-docker-cu13.yml
+++ b/.github/workflows/release-docker-cu13.yml
@@ -60,6 +60,7 @@ jobs:
                     --build-arg BUILD_TYPE=${{ matrix.build_type }} \
                     --build-arg CMAKE_BUILD_PARALLEL_LEVEL=$(nproc) \
                     --build-arg GRACE_BLACKWELL=${{ matrix.grace_blackwell }} \
+                    --build-arg USE_LATEST_SGLANG=1 \
                     -t lmsysorg/sglang:${{ matrix.tag }} \
                     --no-cache \
                     .
diff --git a/.github/workflows/release-docker-dev.yml b/.github/workflows/release-docker-dev.yml
index 8fbf28b8d..7f54f653e 100644
--- a/.github/workflows/release-docker-dev.yml
+++ b/.github/workflows/release-docker-dev.yml
@@ -61,6 +61,7 @@ jobs:
             --build-arg BUILD_TYPE=${{ matrix.build_type }} \
             --build-arg CMAKE_BUILD_PARALLEL_LEVEL=$(nproc) \
             --build-arg GRACE_BLACKWELL=${{ matrix.grace_blackwell }} \
+            --build-arg USE_LATEST_SGLANG=1 \
             -t lmsysorg/sglang:${{ matrix.tag }} \
             --no-cache \
             .
diff --git a/docker/Dockerfile b/docker/Dockerfile
index 8cbaa63b3..7be7fe48f 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -11,6 +11,8 @@ ARG DEEPEP_COMMIT=9af0e0d0e74f3577af1979c9b9e1ac2cad0104ee
 ARG TRITON_LANG_COMMIT=4caa0328bf8df64896dd5f6fb9df41b0eb2e750a
 ARG BUILD_AND_DOWNLOAD_PARALLEL=8
 ARG SGL_KERNEL_VERSION=0.3.16.post5
+ARG SGL_VERSION=0.5.4.post3
+ARG USE_LATEST_SGLANG=0
 ARG GDRCOPY_VERSION=2.5.1
 ARG NVSHMEM_VERSION=3.4.5
 ARG PIP_DEFAULT_INDEX
@@ -94,8 +96,10 @@ ARG BRANCH_TYPE
 COPY --from=local_src /src /tmp/local_src
 RUN if [ "$BRANCH_TYPE" = "local" ]; then \
         cp -r /tmp/local_src /sgl-workspace/sglang; \
-    else \
+    elif [ "$USE_LATEST_SGLANG" == "1" ]; then \
         git clone --depth=1 https://github.com/sgl-project/sglang.git /sgl-workspace/sglang; \
+    else \
+        git clone --depth=1 --branch v${SGL_VERSION} https://github.com/sgl-project/sglang.git /sgl-workspace/sglang; \
     fi \
  && rm -rf /tmp/local_src
 RUN --mount=type=cache,target=/root/.cache/pip  python3 -m pip install --upgrade pip setuptools wheel html5lib six \
diff --git a/scripts/release/bump_sglang_version.py b/scripts/release/bump_sglang_version.py
index 1bd347a3c..76d26383c 100755
--- a/scripts/release/bump_sglang_version.py
+++ b/scripts/release/bump_sglang_version.py
@@ -20,6 +20,7 @@ def main():
 
     files_to_update = [
         Path("benchmark/deepseek_v3/README.md"),
+        Path("docker/Dockerfile"),
         Path("docker/rocm.Dockerfile"),
         Path("docs/get_started/install.md"),
         Path("docs/platforms/amd_gpu.md"),