From 442f7369addf23644da54413ba79a1854a453bc5 Mon Sep 17 00:00:00 2001 From: Eugene Rakhmatulin Date: Thu, 18 Dec 2025 22:02:04 -0800 Subject: [PATCH] Updated build script to handle BUILD_JOBS argument --- build-and-copy.sh | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/build-and-copy.sh b/build-and-copy.sh index 19a2c11..01bfd7e 100755 --- a/build-and-copy.sh +++ b/build-and-copy.sh @@ -13,6 +13,7 @@ SSH_USER="$USER" NO_BUILD=false TRITON_REF="v3.5.1" VLLM_REF="main" +BUILD_JOBS="16" # Help function usage() { @@ -22,6 +23,7 @@ usage() { echo " --rebuild-vllm : Set cache bust for vllm" echo " --triton-ref : Triton commit SHA, branch or tag (default: 'v3.5.1')" echo " --vllm-ref : vLLM commit SHA, branch or tag (default: 'main')" + echo " -j, --build-jobs : Number of concurrent build jobs (default: \${BUILD_JOBS})" echo " -h, --copy-to-host : Host address to copy the image to (if not set, don't copy)" echo " -u, --user : Username for ssh command (default: \$USER)" echo " --no-build : Skip building, only copy image (requires --copy-to-host)" @@ -37,6 +39,7 @@ while [[ "$#" -gt 0 ]]; do --rebuild-vllm) REBUILD_VLLM=true ;; --triton-ref) TRITON_REF="$2"; shift ;; --vllm-ref) VLLM_REF="$2"; shift ;; + -j|--build-jobs) BUILD_JOBS="$2"; shift ;; -h|--copy-to-host) COPY_HOST="$2"; shift ;; -u|--user) SSH_USER="$2"; shift ;; --no-build) NO_BUILD=true ;; @@ -74,6 +77,9 @@ if [ "$NO_BUILD" = false ]; then # Add VLLM_REF to build arguments CMD+=("--build-arg" "VLLM_REF=$VLLM_REF") + # Add BUILD_JOBS to build arguments + CMD+=("--build-arg" "BUILD_JOBS=$BUILD_JOBS") + # Add build context CMD+=(".")