[CI][ARM64] Restore known-good CUDA 13 builder image (#53374)
Signed-off-by: Kevin Luu <51931015+khluu@users.noreply.github.com> Co-authored-by: OpenAI Codex <codex@openai.com>
Kevin H. Luu committed
Aug 22, 2026 at 03:50 UTC
10704541aaf72567fe9d6229b3e3d84d37f2ddba
3 files changed
+5
-5
.buildkite/image_build/image_build_arm64.sh
+1
-1
@@ -26,7 +26,7 @@ else
26
--platform linux/arm64 \
27
--build-arg max_jobs=16 \
28
--build-arg nvcc_threads=4 \
29
- --build-arg BUILD_BASE_IMAGE=pytorch/manylinuxaarch64-builder:cuda13.0-78e737ad29420ffc4800e677c51e2a852caf8359 \
29
+ --build-arg BUILD_BASE_IMAGE=pytorch/manylinuxaarch64-builder:cuda13.0-b8b5f17a7d9ccfc25bbc5cf17b3fcea12964a042 \
30
--build-arg torch_cuda_arch_list="9.0 10.0 11.0 12.0" \
31
--build-arg USE_SCCACHE=1 \
32
--build-arg buildkite_commit="$BUILDKITE_COMMIT" \
.buildkite/release-pipeline.yaml
+3
-3
@@ -33,7 +33,7 @@ steps:
33
agents:
34
queue: arm64_cpu_queue_release
35
commands:
36
- - "DOCKER_BUILDKIT=1 docker build --build-arg max_jobs=16 --build-arg USE_SCCACHE=1 --build-arg GIT_REPO_CHECK=1 --build-arg CUDA_VERSION=13.0.2 --build-arg torch_cuda_arch_list=\"${CUDA_ARCH_AARCH64}\" --build-arg BUILD_BASE_IMAGE=pytorch/manylinuxaarch64-builder:cuda13.0-78e737ad29420ffc4800e677c51e2a852caf8359 --tag vllm-ci:build-image --target build --progress plain -f docker/Dockerfile ."
36
+ - "DOCKER_BUILDKIT=1 docker build --build-arg max_jobs=16 --build-arg USE_SCCACHE=1 --build-arg GIT_REPO_CHECK=1 --build-arg CUDA_VERSION=13.0.2 --build-arg torch_cuda_arch_list=\"${CUDA_ARCH_AARCH64}\" --build-arg BUILD_BASE_IMAGE=pytorch/manylinuxaarch64-builder:cuda13.0-b8b5f17a7d9ccfc25bbc5cf17b3fcea12964a042 --tag vllm-ci:build-image --target build --progress plain -f docker/Dockerfile ."
37
- "mkdir artifacts"
38
- "docker run --rm -v $(pwd)/artifacts:/artifacts_host vllm-ci:build-image bash -c 'cp -r dist /artifacts_host && chmod -R a+rw /artifacts_host'"
39
- "bash .buildkite/scripts/upload-nightly-wheels.sh"
@@ -260,7 +260,7 @@ steps:
260
--build-arg CUDA_VERSION=13.0.2 \
261
--build-arg torch_cuda_arch_list="${CUDA_ARCH_AARCH64}" \
262
--build-arg INSTALL_KV_CONNECTORS=true \
263
- --build-arg BUILD_BASE_IMAGE=pytorch/manylinuxaarch64-builder:cuda13.0-78e737ad29420ffc4800e677c51e2a852caf8359 \
263
+ --build-arg BUILD_BASE_IMAGE=pytorch/manylinuxaarch64-builder:cuda13.0-b8b5f17a7d9ccfc25bbc5cf17b3fcea12964a042 \
264
--target vllm-openai \
265
--progress plain \
266
-f docker/Dockerfile .
@@ -361,7 +361,7 @@ steps:
361
--build-arg GDRCOPY_OS_VERSION=Ubuntu24_04 \
362
--build-arg torch_cuda_arch_list="${CUDA_ARCH_AARCH64}" \
363
--build-arg INSTALL_KV_CONNECTORS=true \
364
- --build-arg BUILD_BASE_IMAGE=pytorch/manylinuxaarch64-builder:cuda13.0-78e737ad29420ffc4800e677c51e2a852caf8359 \
364
+ --build-arg BUILD_BASE_IMAGE=pytorch/manylinuxaarch64-builder:cuda13.0-b8b5f17a7d9ccfc25bbc5cf17b3fcea12964a042 \
365
--target vllm-openai \
366
--progress plain \
367
-f docker/Dockerfile .
.buildkite/scripts/hardware_ci/run-gh200-test.sh
+1
-1
@@ -15,7 +15,7 @@ DOCKER_BUILDKIT=1 docker build . \
15
-t gh200-test \
16
--build-arg max_jobs=66 \
17
--build-arg nvcc_threads=2 \
18
- --build-arg BUILD_BASE_IMAGE=pytorch/manylinuxaarch64-builder:cuda13.0-78e737ad29420ffc4800e677c51e2a852caf8359 \
18
+ --build-arg BUILD_BASE_IMAGE=pytorch/manylinuxaarch64-builder:cuda13.0-b8b5f17a7d9ccfc25bbc5cf17b3fcea12964a042 \
19
--build-arg torch_cuda_arch_list="9.0+PTX"
20
21
# Setup cleanup