x
This commit is contained in:
+5
-1
@@ -4,11 +4,15 @@ FROM registry.fedoraproject.org/fedora:43
|
|||||||
ARG HTTP_PROXY=""
|
ARG HTTP_PROXY=""
|
||||||
ARG HTTPS_PROXY=""
|
ARG HTTPS_PROXY=""
|
||||||
ARG SOCKS_PROXY=""
|
ARG SOCKS_PROXY=""
|
||||||
|
ARG MAX_JOBS="32"
|
||||||
|
|
||||||
# Set environment variables if proxies are provided
|
# Set environment variables if proxies are provided
|
||||||
ENV http_proxy=$HTTP_PROXY
|
ENV http_proxy=$HTTP_PROXY
|
||||||
ENV https_proxy=$HTTPS_PROXY
|
ENV https_proxy=$HTTPS_PROXY
|
||||||
ENV no_proxy="localhost,127.0.0.1,::1"
|
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
|
# Configure DNF proxy if provided
|
||||||
RUN if [ -n "$HTTP_PROXY" ]; then \
|
RUN if [ -n "$HTTP_PROXY" ]; then \
|
||||||
@@ -167,7 +171,7 @@ ENV VLLM_TARGET_DEVICE="rocm"
|
|||||||
ENV PYTORCH_ROCM_ARCH="gfx1201"
|
ENV PYTORCH_ROCM_ARCH="gfx1201"
|
||||||
ENV HIP_ARCHITECTURES="gfx1201"
|
ENV HIP_ARCHITECTURES="gfx1201"
|
||||||
ENV AMDGPU_TARGETS="gfx1201"
|
ENV AMDGPU_TARGETS="gfx1201"
|
||||||
ENV MAX_JOBS="32"
|
ENV MAX_JOBS=$MAX_JOBS
|
||||||
|
|
||||||
ENV CC="/opt/rocm/llvm/bin/clang"
|
ENV CC="/opt/rocm/llvm/bin/clang"
|
||||||
ENV CXX="/opt/rocm/llvm/bin/clang++"
|
ENV CXX="/opt/rocm/llvm/bin/clang++"
|
||||||
|
|||||||
+1
-1
@@ -330,7 +330,7 @@ build_image() {
|
|||||||
BUILD_START=$(date +%s)
|
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
|
if [ -n "$HTTP_PROXY_ARG" ]; then
|
||||||
BUILD_ARGS="${BUILD_ARGS} --build-arg HTTP_PROXY=${HTTP_PROXY_ARG} --build-arg HTTPS_PROXY=${HTTP_PROXY_ARG}"
|
BUILD_ARGS="${BUILD_ARGS} --build-arg HTTP_PROXY=${HTTP_PROXY_ARG} --build-arg HTTPS_PROXY=${HTTP_PROXY_ARG}"
|
||||||
fi
|
fi
|
||||||
|
|||||||
Reference in New Issue
Block a user