Skip to content

Commit 5fbb4c3

Browse files
draft intel gaudi 3 integration
Signed-off-by: jakub-sochacki <jakub.sochacki@intel.com>
1 parent 15a7ecb commit 5fbb4c3

File tree

1 file changed

+27
-0
lines changed

1 file changed

+27
-0
lines changed

buildkite/test-template-ci.j2

Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
{% set docker_image_torch_nightly = "public.ecr.aws/q9t5s3a7/vllm-ci-postmerge-repo:$BUILDKITE_COMMIT-torch-nightly" %}
1010
{% set docker_image_cu118 = "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

0 commit comments

Comments
 (0)