Fixing the image building.

Signed-off-by: Alexei V. Ivanov <[email protected]>

Author: Alexei-V-Ivanov-AMD

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: