Run tests after checks

Author: kevalmorabia97

.github/workflows/_wait_for_checks.yml

@@ -0,0 +1,22 @@
+name: Wait for checks
+
+on:
+  workflow_call:
+    inputs:
+      match_pattern:
+        description: Regex for check run names to wait for
+        required: true
+        type: string
+
+jobs:
+  wait:
+    runs-on: ubuntu-latest
+    permissions:
+      checks: read
+    steps:
+      - name: Wait for checks (PRs only)
+        if: github.event_name == 'pull_request' || startsWith(github.ref, 'refs/heads/pull-request/')
+        uses: poseidon/[email protected]
+        with:
+          token: ${{ secrets.GITHUB_TOKEN }}
+          match_pattern: ${{ inputs.match_pattern }}

.github/workflows/gpu_tests.yml

@@ -18,7 +18,17 @@ concurrency:
   cancel-in-progress: true
 
 jobs:
+  wait-dco:
+    uses: ./.github/workflows/_wait_for_checks.yml
+    with:
+      match_pattern: '^DCO$'
+  wait-unit-tests:
+    needs: [wait-dco]
+    uses: ./.github/workflows/_wait_for_checks.yml
+    with:
+      match_pattern: 'Unit tests / linux*'
   gpu-tests:
+    needs: [wait-unit-tests]
     # Runner list at https://github.com/nv-gha-runners/enterprise-runner-configuration/blob/main/docs/runner-groups.md
     runs-on: linux-amd64-gpu-h100-latest-1
     timeout-minutes: 60

.github/workflows/unit_tests.yml

@@ -26,7 +26,12 @@ concurrency:
   cancel-in-progress: true
 
 jobs:
+  wait-dco:
+    uses: ./.github/workflows/_wait_for_checks.yml
+    with:
+      match_pattern: '^DCO$'
   linux:
+    needs: [wait-dco]
     runs-on: ubuntu-latest
     timeout-minutes: 30
     steps: