Merge pull request #57 from Alexei-V-Ivanov-AMD/build_update

Fixing the image building.
vllm-project · Jan 17, 2025 · 027f1b3 · 027f1b3
2 parents c899dbb + 24f2120
commit 027f1b3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/scripts/test-template-aws.j2 b/scripts/test-template-aws.j2
@@ -174,7 +174,8 @@ steps:
       - label: "AMD: :docker: build image"
         depends_on: ~
         commands:
-          - "docker build --build-arg max_jobs=16 --tag {{ docker_image_amd }} -f Dockerfile.rocm --progress plain ."
+          # Handle the introduction of test target in Dockerfile.rocm
+          - "grep -i 'from base as test' Dockerfile.rocm && docker build --build-arg max_jobs=16 --tag {{ docker_image_amd }} -f Dockerfile.rocm --target test --progress plain . || docker build --build-arg max_jobs=16 --tag {{ docker_image_amd }} -f Dockerfile.rocm --progress plain ."
           - "docker push {{ docker_image_amd }}"
         plugins:
           - docker-login#v3.0.0: