Update validate_docker_images.yml

test adding options on release channel and strip out prefix

.github/workflows/validate_docker_images.yml

@@ -11,13 +11,17 @@ on:
           - 'nightly'
           - 'test'
           - 'release'
+  push:
+      branches:
+      - '*'
 
 jobs:
   generate-matrix:
     uses: pytorch/test-infra/.github/workflows/generate_docker_release_matrix.yml@main
     with:
       channel: ${{ inputs.channel }}
     secrets: inherit
+
   run-gpu-tests:
     needs: generate-matrix
     strategy:
@@ -30,11 +34,13 @@ 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 }}"  # Print docker image
         export MATRIX_GPU_ARCH_TYPE="cuda"
         export MATRIX_GPU_ARCH_VERSION="${{ matrix.cuda }}"
         export TARGET_OS="linux"
         python test/smoke_test/smoke_test.py --package torchonly --runtime-error-check enabled
+