[chore] SGLang tag management in Dockerfile (#12734)
This commit is contained in:
1
.github/workflows/release-docker-cu13.yml
vendored
1
.github/workflows/release-docker-cu13.yml
vendored
@@ -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 \
|
||||
.
|
||||
|
||||
1
.github/workflows/release-docker-dev.yml
vendored
1
.github/workflows/release-docker-dev.yml
vendored
@@ -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"),
|
||||
|
||||
Reference in New Issue
Block a user