File tree Expand file tree Collapse file tree 1 file changed +30
-0
lines changed Expand file tree Collapse file tree 1 file changed +30
-0
lines changed Original file line number Diff line number Diff line change @@ -15,6 +15,20 @@ steps:
1515 env :
1616 DOCKER_BUILDKIT : " 1"
1717
18+ # aarch64 build.
19+ - label : " Build arm64 CPU wheel"
20+ depends_on : ~
21+ id : build-wheel-arm64-cpu
22+ agents :
23+ queue : arm64_cpu_queue_postmerge
24+ commands :
25+ - " DOCKER_BUILDKIT=1 docker build --build-arg max_jobs=16 --build-arg USE_SCCACHE=1 --build-arg GIT_REPO_CHECK=1 --tag vllm-ci:build-image --target vllm-openai --progress plain -f docker/Dockerfile.cpu ."
26+ - " mkdir artifacts"
27+ - " 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'"
28+ - " bash .buildkite/scripts/upload-wheels.sh"
29+ env :
30+ DOCKER_BUILDKIT : " 1"
31+
1832 - label : " Build wheel - CUDA 12.8"
1933 depends_on : ~
2034 id : build-wheel-cuda-12-8
@@ -142,6 +156,22 @@ steps:
142156 env :
143157 DOCKER_BUILDKIT : " 1"
144158
159+ - block : " Build arm64 CPU release image"
160+ key : block-arm64-cpu-release-image-build
161+ depends_on : ~
162+
163+ - label : " Build and publish arm64 CPU release image"
164+ depends_on : block-arm64-cpu-release-image-build
165+ agents :
166+ queue : arm64_cpu_queue_postmerge
167+ commands :
168+ - " aws ecr-public get-login-password --region us-east-1 | docker login --username AWS --password-stdin public.ecr.aws/q9t5s3a7"
169+ - " DOCKER_BUILDKIT=1 docker build --build-arg max_jobs=16 --build-arg GIT_REPO_CHECK=1 --tag public.ecr.aws/q9t5s3a7/vllm-arm64-cpu-release-repo:$(buildkite-agent meta-data get release-version) --tag public.ecr.aws/q9t5s3a7/vllm-arm64-cpu-release-repo:latest --progress plain --target vllm-openai -f docker/Dockerfile.cpu ."
170+ - " docker push public.ecr.aws/q9t5s3a7/vllm-arm64-cpu-release-repo:latest"
171+ - " docker push public.ecr.aws/q9t5s3a7/vllm-arm64-cpu-release-repo:$(buildkite-agent meta-data get release-version)"
172+ env :
173+ DOCKER_BUILDKIT : " 1"
174+
145175 - label : " Build and publish nightly multi-arch image to DockerHub"
146176 depends_on :
147177 - create-multi-arch-manifest
You can’t perform that action at this time.
0 commit comments