Update validate_docker_images.yml

testing
juliagmt-google · May 7, 2024 · 9c4e05e · 9c4e05e
1 parent 5bc1082
commit 9c4e05e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.github/workflows/validate_docker_images.yml b/.github/workflows/validate_docker_images.yml
@@ -46,10 +46,11 @@ jobs:
       ref: ${{ inputs.ref || github.ref }}
       job-name: cuda${{ matrix.cuda }}-cudnn${{ matrix.cudnn_version }}-${{ matrix.image_type }}
       binary-matrix: ${{ toJSON(matrix) }}
-      docker-image: ${{matrix.docker}}
+      docker-image: ${{ matrix.channel == 'release' && startsWith(matrix.docker, 'ghcr.io/') ? replace(matrix.docker, 'ghcr.io/', '') : matrix.docker }}  # Strip ghcr.io/ if release
       timeout: 180
       script: |
         set -ex
+        echo "Using docker image: ${{ matrix.docker }}"
         export MATRIX_GPU_ARCH_TYPE="cuda"
         export MATRIX_GPU_ARCH_VERSION="${{ matrix.cuda }}"
         export TARGET_OS="linux"