Run ROCm job on cron schedule and when push to main.

Author: akashveramd

.github/workflows/integration_test_8gpu_features.yaml

@@ -25,7 +25,33 @@ permissions:
       contents: read
 
 jobs:
+  # checks if job should run for ROCm
+  check-run-job:
+    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
+    outputs:
+      run_job: ${{ steps.set-flag.outputs.run_job }}
+    steps:
+      - name: Decide if job should run for rocm
+        run: |
+          if [[ "${{ matrix.gpu-arch-type }}" == "cuda" ]]; then
+            echo "run_job=1" >> $GITHUB_OUTPUT
+          elif [[ "${{ matrix.gpu-arch-type }}" == "rocm" && "${{ github.event_name }}" == "schedule" ]]; then
+            echo "run_job=1" >> $GITHUB_OUTPUT
+          else
+            echo "run_job=0" >> $GITHUB_OUTPUT
+          fi
+
+  # runs the test
   build-test:
+    needs: check-run-job
+    if: ${{ needs.check-runner.outputs.run_job == '1' }}
     uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
     strategy:
       fail-fast: false