Fix the docker image tag specified for smoke tests in Release Docker …

…images workflow (#15842)

### Ticket
- #15843

### Problem description
The smoke tests should mention not the latest tag but the tag of the
version being built.


### What's changed
Add {{inputs.version}} to the tag being used to run Smoke tests.
Also modified Docker Run Action to take a new parameter called
`docker_image_tag` that allows to specify the image.

### Checklist
- [x] Post commit CI passes
https://github.com/tenstorrent/tt-metal/actions/runs/12264928767
- [x] Blackhole Post commit (if applicable)
https://github.com/tenstorrent/tt-metal/actions/runs/12264932641
- [ ] Model regression CI testing passes (if applicable)
- [ ] Device performance regression CI testing passes (if applicable)
- [ ] **(For models and ops writers)** Full [new
models](https://github.com/tenstorrent/tt-metal/actions/workflows/full-new-models-suite.yaml)
tests passes
- [ ] New/Existing tests provide coverage for changes

.github/actions/docker-run/action.yml

@@ -9,6 +9,9 @@ inputs:
     description: 'Docker image architecture'
     required: false
     default: tt-metalium/ubuntu-20.04-amd64
+  docker_version:
+    description: 'Specify version for the Docker image tag to use.'
+    required: false
   docker_username:
     description: docker login username
     required: true
@@ -38,6 +41,7 @@ runs:
       uses: ./.github/actions/generate-docker-tag
       with:
         image: ${{ inputs.docker_os_arch }}
+        version: ${{ inputs.docker_version }}
     - name: Set
       shell: bash
       run: |

.github/actions/generate-docker-tag/action.yml

@@ -9,17 +9,24 @@ inputs:
     description: 'Docker image to run commands in - follows os-arch format'
     required: false
     default: ubuntu-20.04-amd64
-
+  version:
+    description: 'Docker image version'
+    required: false
 runs:
   using: "composite"
   steps:
     - name: Determine Docker Tag
       shell: bash
       run: |
-        if [[ "${GITHUB_REF_NAME}" == "main" ]]; then
-          echo "IMAGE_TAG=latest" >> $GITHUB_ENV
+        # If the version was provided use it, otherwise, determine what the version should be.
+        if [ "${{ inputs.version }}" != "" ]; then
+          echo "IMAGE_TAG=${{ inputs.version }}" >> $GITHUB_ENV
         else
-          echo "IMAGE_TAG=dev-${GITHUB_REF_NAME//\//-}" >> $GITHUB_ENV
+          if [[ "${GITHUB_REF_NAME}" == "main" ]]; then
+            echo "IMAGE_TAG=latest" >> $GITHUB_ENV
+          else
+            echo "IMAGE_TAG=dev-${GITHUB_REF_NAME//\//-}" >> $GITHUB_ENV
+          fi
         fi
     - name: Determine Full Docker Image Tag
       shell: bash

.github/workflows/publish-release-image.yaml

@@ -91,6 +91,7 @@ jobs:
         uses: ./.github/actions/docker-run
         with:
           docker_os_arch: tt-metalium-${{ matrix.os }}-amd64-release/${{ matrix.test_group.arch }}
+          docker_version: ${{ inputs.version }}
           docker_password: ${{ secrets.GITHUB_TOKEN }}
           run_args: |
             ${{ matrix.test_group.cmd }}