"docs/vscode:/vscode.git/clone" did not exist on "1e84168edd695992d983992997ad02008181c840"
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:
--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 \
.
......
......@@ -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 \
.
......
......@@ -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 \
......
......@@ -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"),
......
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