Unverified Commit 68444751 authored by Harrison Saturley-Hall's avatar Harrison Saturley-Hall Committed by GitHub
Browse files

fix: limit the number of build jobs on sglang-wideep (#3723)


Signed-off-by: default avatarHarrison Saturley-Hall <hsaturleyhal@nvidia.com>
parent 4b7a806c
...@@ -3,6 +3,7 @@ ...@@ -3,6 +3,7 @@
ARG SGLANG_IMAGE_TAG="v0.5.3.post2" ARG SGLANG_IMAGE_TAG="v0.5.3.post2"
ARG BRANCH_TYPE ARG BRANCH_TYPE
ARG CARGO_BUILD_JOBS
FROM scratch AS local_src FROM scratch AS local_src
COPY . /src COPY . /src
...@@ -16,6 +17,8 @@ WORKDIR /sgl-workspace ...@@ -16,6 +17,8 @@ WORKDIR /sgl-workspace
# Providing --build-arg BRANCH_TYPE=remote will editable install the remote dynamo repo # Providing --build-arg BRANCH_TYPE=remote will editable install the remote dynamo repo
# Default is to install the latest published dynamo version # Default is to install the latest published dynamo version
ARG BRANCH_TYPE ARG BRANCH_TYPE
ARG CARGO_BUILD_JOBS
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/dynamo; \ cp -r /tmp/local_src /sgl-workspace/dynamo; \
...@@ -23,6 +26,8 @@ RUN if [ "$BRANCH_TYPE" = "local" ]; then \ ...@@ -23,6 +26,8 @@ RUN if [ "$BRANCH_TYPE" = "local" ]; then \
git clone https://github.com/ai-dynamo/dynamo.git /sgl-workspace/dynamo; \ git clone https://github.com/ai-dynamo/dynamo.git /sgl-workspace/dynamo; \
fi fi
ENV CARGO_BUILD_JOBS=${CARGO_BUILD_JOBS:-16}
# SGLang does not use a venv in their container # SGLang does not use a venv in their container
RUN if [ "$BRANCH_TYPE" = "local" ]; then \ RUN if [ "$BRANCH_TYPE" = "local" ]; then \
cd dynamo/lib/bindings/python && \ cd dynamo/lib/bindings/python && \
......
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