Cleanup CI

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

Author: kevalmorabia97

.github/codecov.yml

@@ -0,0 +1,8 @@
+# Allow atmost 5% coverage drop from main branch.
+coverage:
+  status:
+    project:
+      default:
+        target: auto
+        threshold: 5%
+    patch: false

.github/workflows/_wait_for_checks.yml

@@ -0,0 +1,29 @@
+name: Wait for checks
+
+on:
+  workflow_call:
+    inputs:
+      match_pattern:
+        required: true
+        type: string
+      delay:
+        required: false
+        type: string
+        default: 10s
+
+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 }}
+          delay: ${{ inputs.delay }}
+      - name: No-op for non-PR events
+        if: github.event_name != 'pull_request' && !startsWith(github.ref, 'refs/heads/pull-request/')
+        run: echo "Not a pull_request event"

.github/workflows/gpu_tests.yml

@@ -14,11 +14,28 @@ on:
 
 # Cancel previous runs if new commit is pushed to the same PR
 concurrency:
-  group: ${{ github.workflow }}-${{ github.event.pull_request.number }}
+  group: ${{ github.workflow }}-${{ github.ref }}
   cancel-in-progress: true
 
 jobs:
+  wait-dco:
+    uses: ./.github/workflows/_wait_for_checks.yml
+    permissions:
+      checks: read
+    secrets: inherit
+    with:
+      match_pattern: '^DCO$'
+  wait-unit-tests:
+    needs: [wait-dco]
+    uses: ./.github/workflows/_wait_for_checks.yml
+    permissions:
+      checks: read
+    secrets: inherit
+    with:
+      match_pattern: '^Unit tests / linux$'
+      delay: 60s
   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/multi_version_unit_tests.yml

@@ -26,7 +26,15 @@ concurrency:
   cancel-in-progress: true
 
 jobs:
-  unit-tests:
+  wait-dco:
+    uses: ./.github/workflows/_wait_for_checks.yml
+    permissions:
+      checks: read
+    secrets: inherit
+    with:
+      match_pattern: '^DCO$'
+  linux:
+    needs: [wait-dco]
     runs-on: ubuntu-latest
     timeout-minutes: 30
     steps:
@@ -42,8 +50,9 @@ jobs:
           token: ${{ secrets.CODECOV_TOKEN }}
           fail_ci_if_error: true
           verbose: true
-  unit-tests-windows:
+  windows:
     if: github.event_name == 'pull_request'
+    needs: [linux]
     runs-on: windows-latest
     timeout-minutes: 30
     steps:
@@ -53,3 +62,48 @@ jobs:
           python-version: "3.12"
       - name: Run unit tests (without coverage)
         run: pip install tox && tox -e py312-torch28-unit
+  multi-py:
+    if: github.event_name == 'pull_request'
+    needs: [linux]
+    runs-on: ubuntu-latest
+    timeout-minutes: 30
+    strategy:
+      matrix:
+        py: [10, 11]
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.${{ matrix.py }}"
+      - name: Run unit tests
+        run: pip install tox && tox -e py3${{ matrix.py }}-torch28-unit
+  multi-torch:
+    if: github.event_name == 'pull_request'
+    needs: [linux]
+    runs-on: ubuntu-latest
+    timeout-minutes: 30
+    strategy:
+      matrix:
+        torch: [25, 26, 27]
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+      - name: Run unit tests
+        run: pip install tox && tox -e py312-torch${{ matrix.torch }}-unit
+  partial-install:
+    if: github.event_name == 'pull_request'
+    needs: [linux]
+    runs-on: ubuntu-latest
+    timeout-minutes: 30
+    strategy:
+      matrix:
+        test-env: [onnx, torch]
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+      - name: Run unit tests
+        run: pip install tox && tox -e py312-partial-unit-${{ matrix.test-env }}