Skip to content

Commit 6bcf6c2

Browse files
committed
[CI][Release][Arm64] Add Arm64 CPU docker pipeline
Signed-off-by: Ioana Ghiban <ioana.ghiban@arm.com>
1 parent 70b1b33 commit 6bcf6c2

File tree

1 file changed

+30
-0
lines changed

1 file changed

+30
-0
lines changed

.buildkite/release-pipeline.yaml

Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)