Use shareable calculate-docker-image GHA (#4399)

Add a step in Linux job to calculate and build Docker image if needed.
This is similar to how PyTorch CI rebuilds its Docker images atm.

### Testing

pytorch/pytorch#105372, PyTorch Lint job uses
Linux jobs.

Author: huydhn

.github/workflows/linux_job.yml

@@ -47,6 +47,13 @@ on:
         description: Identifies the Docker image by name.
         default: "pytorch/conda-builder"
         type: string
+      docker-build-dir:
+        description: |
+          The directory containing the build.sh shell script to build the docker image.
+          The script parameters can be passed to docker build similar to how it is used
+          in PyTorch, i.e. build.sh "${IMAGE_NAME}" -t "${DOCKER_IMAGE}".
+        default: ".ci/docker"
+        type: string
       gpu-arch-type:
         description: "GPU arch type to use"
         default: "cpu"
@@ -138,11 +145,6 @@ jobs:
         with:
           github-secret: ${{ github.token }}
 
-      - name: Pull docker image
-        uses: ./test-infra/.github/actions/pull-docker-image
-        with:
-          docker-image: ${{ env.DOCKER_IMAGE }}
-
       - name: Checkout repository (${{ inputs.repository || github.repository }}@${{ inputs.ref }})
         uses: actions/checkout@v3
         with:
@@ -152,6 +154,20 @@ jobs:
           path: ${{ inputs.repository || github.repository }}
           fetch-depth: ${{ inputs.fetch-depth }}
 
+      - name: Calculate docker image
+        id: calculate-docker-image
+        uses: ./test-infra/.github/actions/calculate-docker-image
+        with:
+          docker-image-name: ${{ env.DOCKER_IMAGE }}
+          docker-build-dir: ${{ inputs.docker-build-dir }}
+          # This needs to be where the repository is checked out
+          working-directory: ${{ inputs.repository || github.repository }}
+
+      - name: Pull docker image
+        uses: ./test-infra/.github/actions/pull-docker-image
+        with:
+          docker-image: ${{ steps.calculate-docker-image.outputs.docker-image }}
+
       - name: Download artifacts (if any)
         uses: actions/download-artifact@v3
         if: ${{ inputs.download-artifact != '' }}
@@ -172,6 +188,7 @@ jobs:
         working-directory: ${{ inputs.repository }}
         env:
           ALL_SECRETS: ${{ toJSON(secrets) }}
+          DOCKER_IMAGE: ${{ steps.calculate-docker-image.outputs.docker-image }}
         run: |
           set -ex
           {