Fix issues in check-run-job job.

Author: akashveramd

.github/workflows/integration_test_8gpu_features.yaml

@@ -27,14 +27,16 @@ permissions:
 jobs:
   # checks if job should run for ROCm
   check-run-job:
-    matrix:
+    strategy:
+      matrix:
         include:
           - name: cuda
             runner: linux.g5.48xlarge.nvidia.gpu
             gpu-arch-type: cuda
           - name: rocm
             runner: linux.rocm.gpu.gfx942.8
             gpu-arch-type: rocm
+    runs-on: ${{ matrix.runner }}
     outputs:
       run_job: ${{ steps.set-flag.outputs.run_job }}
     steps:
@@ -51,7 +53,7 @@ jobs:
   # runs the test
   build-test:
     needs: check-run-job
-    if: ${{ needs.check-runner.outputs.run_job == '1' }}
+    if: ${{ needs.check-run-job.outputs.run_job == '1' }}
     uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
     strategy:
       fail-fast: false