Skip to content

Commit dc9bc7a

Browse files
slin1237ch-wan
authored andcommitted
[ci] limit cmake build nproc (#8100)
1 parent b07f32f commit dc9bc7a

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

.github/workflows/release-docker-dev.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,5 +41,5 @@ jobs:
4141

4242
- name: Build and Push Dev Image
4343
run: |
44-
docker buildx build --output type=image,compression=zstd . -f docker/Dockerfile --build-arg CUDA_VERSION=${{ matrix.variant.version }} --build-arg BUILD_TYPE=${{ matrix.variant.type }} -t lmsysorg/sglang:${{ matrix.variant.tag }} --no-cache
44+
docker buildx build --output type=image,compression=zstd . -f docker/Dockerfile --build-arg CUDA_VERSION=${{ matrix.variant.version }} --build-arg BUILD_TYPE=${{ matrix.variant.type }} --build-arg CMAKE_BUILD_PARALLEL_LEVEL=$(nproc) -t lmsysorg/sglang:${{ matrix.variant.tag }} --no-cache
4545
docker push lmsysorg/sglang:${{ matrix.variant.tag }}

docker/Dockerfile

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ FROM nvidia/cuda:${CUDA_VERSION}-cudnn-devel-ubuntu22.04
33

44
ARG BUILD_TYPE=all
55
ARG DEEPEP_COMMIT=b6ce310bb0b75079682d09bc2ebc063a074fbd58
6+
ARG CMAKE_BUILD_PARALLEL_LEVEL=2
67
ENV DEBIAN_FRONTEND=noninteractive \
78
CUDA_HOME=/usr/local/cuda \
89
GDRCOPY_HOME=/usr/src/gdrdrv-2.4.4/ \
@@ -78,7 +79,7 @@ RUN wget https://developer.download.nvidia.com/compute/redist/nvshmem/3.3.9/sour
7879
NVSHMEM_TIMEOUT_DEVICE_POLLING=0 \
7980
NVSHMEM_USE_GDRCOPY=1 \
8081
cmake -S . -B build/ -DCMAKE_INSTALL_PREFIX=${NVSHMEM_DIR} -DCMAKE_CUDA_ARCHITECTURES=90 \
81-
&& cmake --build build --target install -j \
82+
&& cmake --build build --target install -j${CMAKE_BUILD_PARALLEL_LEVEL} \
8283
&& cd /sgl-workspace/DeepEP \
8384
&& NVSHMEM_DIR=${NVSHMEM_DIR} pip install .
8485

0 commit comments

Comments
 (0)