Mark AMD build image step optional

vllm-project · Feb 9, 2025 · 300d52c · 300d52c
1 parent 29b5d3b
commit 300d52c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/scripts/test-template-aws.j2 b/scripts/test-template-aws.j2
@@ -242,8 +242,11 @@ steps:
   - group: "AMD Tests"
     depends_on: ~
     steps:
-      - label: "AMD: :docker: build image"
+      - block: "Build AMD image"
+        key: block-amd-build
         depends_on: ~
+      - label: "AMD: :docker: build image"
+        depends_on: block-amd-build
         commands:
           # 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 ."