Fixing amd image building functionality.

vllm-project · Jan 16, 2025 · 552e7f3 · 552e7f3
1 parent c899dbb
commit 552e7f3
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/scripts/test-template-aws.j2 b/scripts/test-template-aws.j2
@@ -174,7 +174,10 @@ 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 ."
+          #- "docker build --build-arg max_jobs=16 --tag {{ docker_image_amd }} -f Dockerfile.rocm --progress plain ."
+          - "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: