Add last aggregator job in unit/gpu CI test as required check

Signed-off-by: Keval Morabia <[email protected]>

Author: kevalmorabia97

.github/workflows/gpu_tests.yml

@@ -19,14 +19,15 @@ jobs:
     if: startsWith(github.ref, 'refs/heads/pull-request/')
     runs-on: ubuntu-latest
     outputs:
-      any_changed: ${{ steps.changed-tests.outputs.any_changed }}
+      any_changed: ${{ steps.compute-any-changed.outputs.any_changed }}
     steps:
       - uses: actions/checkout@v4
       - id: get-pr-info
         uses: nv-gha-runners/get-pr-info@main
       - name: Check for changes in test-relevant directories
         id: changed-tests
         uses: step-security/[email protected]
+        continue-on-error: true
         with:
           files: |
             .github/workflows/gpu_tests.yml
@@ -36,6 +37,17 @@ jobs:
             pyproject.toml
             setup.py
           base_sha: ${{ fromJSON(steps.get-pr-info.outputs.pr-info).base.ref }}
+          # Sometimes fails if commits are force pushed on the PR
+          fail_on_initial_diff_error: true
+      - name: Compute any_changed (fallback to true on failure)
+        id: compute-any-changed
+        if: ${{ always() }}
+        run: |
+          if [ "${{ steps.changed-tests.outcome }}" != "success" ]; then
+            echo "any_changed=true" >> "$GITHUB_OUTPUT"
+          else
+            echo "any_changed=${{ steps.changed-tests.outputs.any_changed }}" >> "$GITHUB_OUTPUT"
+          fi
   wait-checks:
     needs: [check-file-changes]
     if: needs.check-file-changes.outputs.any_changed == 'true'
@@ -70,3 +82,12 @@ jobs:
     timeout-minutes: 90
     container: *gpu_container
     steps: *gpu_steps
+  gpu-pr-required-check:
+    # Run even if gpu-tests-pr is skipped
+    if: ${{ startsWith(github.ref, 'refs/heads/pull-request/') && always() }}
+    needs: [check-file-changes, gpu-tests-pr]
+    runs-on: ubuntu-latest
+    steps:
+      - name: Required GPU tests did not succeed
+        if: ${{ needs.check-file-changes.outputs.any_changed == 'true' && needs.gpu-tests-pr.result != 'success' }}
+        run: exit 1

.github/workflows/unit_tests.yml

@@ -126,3 +126,9 @@ jobs:
           python-version: "3.12"
       - name: Run unit tests
         run: pip install tox && tox -e py312-partial-unit-${{ matrix.test-env }}
+  unit-pr-required-check:
+    if: github.event_name == 'pull_request'
+    needs: [linux, windows, multi-py, multi-torch, multi-transformers, partial-install]
+    runs-on: ubuntu-latest
+    steps:
+      - run: echo "All PR unit test jobs completed"