build on arm64
Some checks failed
Build and Push spark-vllm / docker (push) Failing after 1m30s

This commit is contained in:
2026-05-11 14:40:20 -05:00
parent d3dbfb682a
commit 896cdefedf

View File

@@ -471,6 +471,7 @@ fi
COMMON_BUILD_FLAGS+=("--build-arg" "BUILD_JOBS=$BUILD_JOBS") COMMON_BUILD_FLAGS+=("--build-arg" "BUILD_JOBS=$BUILD_JOBS")
COMMON_BUILD_FLAGS+=("--build-arg" "TORCH_CUDA_ARCH_LIST=$GPU_ARCH_LIST") COMMON_BUILD_FLAGS+=("--build-arg" "TORCH_CUDA_ARCH_LIST=$GPU_ARCH_LIST")
COMMON_BUILD_FLAGS+=("--build-arg" "FLASHINFER_CUDA_ARCH_LIST=$GPU_ARCH_LIST") COMMON_BUILD_FLAGS+=("--build-arg" "FLASHINFER_CUDA_ARCH_LIST=$GPU_ARCH_LIST")
COMMON_BUILD_FLAGS+=("--platform" "linux/arm64")
if [ -n "$NETWORK_ARG" ]; then if [ -n "$NETWORK_ARG" ]; then
COMMON_BUILD_FLAGS+=("--network" "$NETWORK_ARG") COMMON_BUILD_FLAGS+=("--network" "$NETWORK_ARG")
fi fi