x

Dockerfile

@@ -4,11 +4,15 @@ FROM registry.fedoraproject.org/fedora:43
 ARG HTTP_PROXY=""
 ARG HTTPS_PROXY=""
 ARG SOCKS_PROXY=""
+ARG MAX_JOBS="32"
 
 # Set environment variables if proxies are provided
 ENV http_proxy=$HTTP_PROXY
 ENV https_proxy=$HTTPS_PROXY
 ENV no_proxy="localhost,127.0.0.1,::1"
+ENV MAX_JOBS=$MAX_JOBS
+ENV CMAKE_BUILD_PARALLEL_LEVEL=$MAX_JOBS
+ENV NINJAFLAGS="-j$MAX_JOBS"
 
 # Configure DNF proxy if provided
 RUN if [ -n "$HTTP_PROXY" ]; then \
@@ -167,7 +171,7 @@ ENV VLLM_TARGET_DEVICE="rocm"
 ENV PYTORCH_ROCM_ARCH="gfx1201"
 ENV HIP_ARCHITECTURES="gfx1201"
 ENV AMDGPU_TARGETS="gfx1201"
-ENV MAX_JOBS="32"
+ENV MAX_JOBS=$MAX_JOBS
 
 ENV CC="/opt/rocm/llvm/bin/clang"
 ENV CXX="/opt/rocm/llvm/bin/clang++"

build_and_run.sh

@@ -330,7 +330,7 @@ build_image() {
     BUILD_START=$(date +%s)
     
     # 准备构建参数
-    BUILD_ARGS="--network=host --build-arg MAX_JOBS=32"
+    BUILD_ARGS="--network=host --build-arg MAX_JOBS=${MAX_JOBS:-32}"
     if [ -n "$HTTP_PROXY_ARG" ]; then
         BUILD_ARGS="${BUILD_ARGS} --build-arg HTTP_PROXY=${HTTP_PROXY_ARG} --build-arg HTTPS_PROXY=${HTTP_PROXY_ARG}"
     fi