88 commands :
99 # #NOTE: torch_cuda_arch_list is derived from upstream PyTorch build files here:
1010 # https://github.com/pytorch/pytorch/blob/main/.ci/aarch64_linux/aarch64_ci_build.sh#L7
11- - " 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=12.9.1 --build-arg VLLM_MAIN_CUDA_VERSION=12.9 --build-arg torch_cuda_arch_list='8.7 8.9 9.0 10.0+PTX 12.0' --tag vllm-ci:build-image --target build --progress plain -f docker/Dockerfile ."
11+ - " 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=12.9.1 --build-arg torch_cuda_arch_list='8.7 8.9 9.0 10.0+PTX 12.0' --tag vllm-ci:build-image --target build --progress plain -f docker/Dockerfile ."
1212 - " mkdir artifacts"
1313 - " 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'"
1414 - " bash .buildkite/scripts/upload-wheels.sh"
@@ -30,19 +30,6 @@ steps:
3030 DOCKER_BUILDKIT : " 1"
3131
3232 # x86 + CUDA builds
33- - label : " Build wheel - CUDA 12.8"
34- depends_on : ~
35- id : build-wheel-cuda-12-8
36- agents :
37- queue : cpu_queue_postmerge
38- commands :
39- - " 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=12.8.1 --tag vllm-ci:build-image --target build --progress plain -f docker/Dockerfile ."
40- - " mkdir artifacts"
41- - " 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'"
42- - " bash .buildkite/scripts/upload-wheels.sh"
43- env :
44- DOCKER_BUILDKIT : " 1"
45-
4633 - label : " Build wheel - CUDA 12.9"
4734 depends_on : ~
4835 id : build-wheel-cuda-12-9
@@ -109,7 +96,6 @@ steps:
10996 - label : " Annotate release workflow"
11097 depends_on :
11198 - create-multi-arch-manifest
112- - build-wheel-cuda-12-8
11399 id : annotate-release-workflow
114100 agents :
115101 queue : cpu_queue_postmerge
0 commit comments