From 9c4e05e964970959755ba9b09373990d65de2e9b Mon Sep 17 00:00:00 2001
From: Julia Guo <153684546+juliagmt-google@users.noreply.github.com>
Date: Tue, 7 May 2024 13:07:36 -0700
Subject: [PATCH] Update validate_docker_images.yml

testing
---
 .github/workflows/validate_docker_images.yml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/validate_docker_images.yml b/.github/workflows/validate_docker_images.yml
index e463e53584..c54c738933 100644
--- 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"