Unverified Commit 9a954982 authored by Baizhou Zhang's avatar Baizhou Zhang Committed by GitHub
Browse files

[chore] SGLang tag management in Dockerfile (#12734)

parent 97be66c3
...@@ -60,6 +60,7 @@ jobs: ...@@ -60,6 +60,7 @@ jobs:
--build-arg BUILD_TYPE=${{ matrix.build_type }} \ --build-arg BUILD_TYPE=${{ matrix.build_type }} \
--build-arg CMAKE_BUILD_PARALLEL_LEVEL=$(nproc) \ --build-arg CMAKE_BUILD_PARALLEL_LEVEL=$(nproc) \
--build-arg GRACE_BLACKWELL=${{ matrix.grace_blackwell }} \ --build-arg GRACE_BLACKWELL=${{ matrix.grace_blackwell }} \
--build-arg USE_LATEST_SGLANG=1 \
-t lmsysorg/sglang:${{ matrix.tag }} \ -t lmsysorg/sglang:${{ matrix.tag }} \
--no-cache \ --no-cache \
. .
......
...@@ -61,6 +61,7 @@ jobs: ...@@ -61,6 +61,7 @@ jobs:
--build-arg BUILD_TYPE=${{ matrix.build_type }} \ --build-arg BUILD_TYPE=${{ matrix.build_type }} \
--build-arg CMAKE_BUILD_PARALLEL_LEVEL=$(nproc) \ --build-arg CMAKE_BUILD_PARALLEL_LEVEL=$(nproc) \
--build-arg GRACE_BLACKWELL=${{ matrix.grace_blackwell }} \ --build-arg GRACE_BLACKWELL=${{ matrix.grace_blackwell }} \
--build-arg USE_LATEST_SGLANG=1 \
-t lmsysorg/sglang:${{ matrix.tag }} \ -t lmsysorg/sglang:${{ matrix.tag }} \
--no-cache \ --no-cache \
. .
......
...@@ -11,6 +11,8 @@ ARG DEEPEP_COMMIT=9af0e0d0e74f3577af1979c9b9e1ac2cad0104ee ...@@ -11,6 +11,8 @@ ARG DEEPEP_COMMIT=9af0e0d0e74f3577af1979c9b9e1ac2cad0104ee
ARG TRITON_LANG_COMMIT=4caa0328bf8df64896dd5f6fb9df41b0eb2e750a ARG TRITON_LANG_COMMIT=4caa0328bf8df64896dd5f6fb9df41b0eb2e750a
ARG BUILD_AND_DOWNLOAD_PARALLEL=8 ARG BUILD_AND_DOWNLOAD_PARALLEL=8
ARG SGL_KERNEL_VERSION=0.3.16.post5 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 GDRCOPY_VERSION=2.5.1
ARG NVSHMEM_VERSION=3.4.5 ARG NVSHMEM_VERSION=3.4.5
ARG PIP_DEFAULT_INDEX ARG PIP_DEFAULT_INDEX
...@@ -94,8 +96,10 @@ ARG BRANCH_TYPE ...@@ -94,8 +96,10 @@ ARG BRANCH_TYPE
COPY --from=local_src /src /tmp/local_src COPY --from=local_src /src /tmp/local_src
RUN if [ "$BRANCH_TYPE" = "local" ]; then \ RUN if [ "$BRANCH_TYPE" = "local" ]; then \
cp -r /tmp/local_src /sgl-workspace/sglang; \ 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; \ 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 \ fi \
&& rm -rf /tmp/local_src && rm -rf /tmp/local_src
RUN --mount=type=cache,target=/root/.cache/pip python3 -m pip install --upgrade pip setuptools wheel html5lib six \ RUN --mount=type=cache,target=/root/.cache/pip python3 -m pip install --upgrade pip setuptools wheel html5lib six \
......
...@@ -20,6 +20,7 @@ def main(): ...@@ -20,6 +20,7 @@ def main():
files_to_update = [ files_to_update = [
Path("benchmark/deepseek_v3/README.md"), Path("benchmark/deepseek_v3/README.md"),
Path("docker/Dockerfile"),
Path("docker/rocm.Dockerfile"), Path("docker/rocm.Dockerfile"),
Path("docs/get_started/install.md"), Path("docs/get_started/install.md"),
Path("docs/platforms/amd_gpu.md"), Path("docs/platforms/amd_gpu.md"),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment