File tree Expand file tree Collapse file tree 1 file changed +27
-0
lines changed Expand file tree Collapse file tree 1 file changed +27
-0
lines changed Original file line number Diff line number Diff line change 99{% set docker_image_torch_nightly = "public.ecr.aws/q9t5s3a7/vllm-ci-postmerge-repo:$BUILDKITE_COMMIT-torch-nightly" %}
1010{% set docker_image_cu 118 = "public.ecr.aws/q9t5s3a7/vllm-ci-postmerge-repo:$BUILDKITE_COMMIT-cu118" %}
1111{% set docker_image_cpu = "public.ecr.aws/q9t5s3a7/vllm-ci-postmerge-repo:$BUILDKITE_COMMIT-cpu" %}
12+ {% set docker_image_hpu = "public.ecr.aws/q9t5s3a7/vllm-ci-postmerge-repo:$BUILDKITE_COMMIT-hpu" %}
1213{% endif %}
1314{% set docker_image_amd = "rocm/vllm-ci:$BUILDKITE_COMMIT" %}
1415{% set default_working_dir = "/vllm-workspace/tests" %}
@@ -411,6 +412,32 @@ steps:
411412 - exit_status: -10 # Agent was lost
412413 limit: 2
413414
415+ - label: ":docker: build image HPU"
416+ key: image-build-hpu
417+ depends_on: ~
418+ agents:
419+ queue: cpu_queue_postmerge_us_east_1
420+ commands:
421+ - "aws ecr-public get-login-password --region us-east-1 | docker login --username AWS --password-stdin public.ecr.aws/q9t5s3a7"
422+ - |
423+ #!/bin/bash
424+ if [[ -z $(docker manifest inspect {{ docker_image_hpu }}) ]]; then
425+ echo "Image not found, proceeding with build..."
426+ else
427+ echo "Image found"
428+ exit 0
429+ fi
430+ - "docker build --file docker/Dockerfile.hpu --build-arg max_jobs=16 --build-arg buildkite_commit=$BUILDKITE_COMMIT --tag {{ docker_image_hpu }} --progress plain ."
431+ - "docker push {{ docker_image_hpu }}"
432+ env:
433+ DOCKER_BUILDKIT: "1"
434+ retry:
435+ automatic:
436+ - exit_status: -1 # Agent was lost
437+ limit: 2
438+ - exit_status: -10 # Agent was lost
439+ limit: 2
440+
414441 {% for step in steps %}
415442 {% if step .fast_check_only != true %}
416443
You can’t perform that action at this time.
0 commit comments