Run gpu tests if relevant files are changed

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

Author: kevalmorabia97

.github/workflows/_wait_for_checks.yml

@@ -18,12 +18,12 @@ jobs:
       checks: read
     steps:
       - name: Wait for checks (PRs only)
-        if: github.event_name == 'pull_request' || startsWith(github.ref, 'refs/heads/pull-request/')
+        if: github.event_name == 'pull_request' || startsWith(github.sha, '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/')
+        if: github.event_name != 'pull_request' && !startsWith(github.sha, 'refs/heads/pull-request/')
         run: echo "Not a pull_request event"

.github/workflows/gpu_tests.yml

@@ -3,18 +3,11 @@ name: GPU tests
 on:
   push:
     branches: ["pull-request/[0-9]+"]
-    # TODO: paths cannot be used since push happens to copied PR and only latest commit to PR is used
-    # paths:
-    #   - ".github/workflows/gpu_tests.yml"
-    #   - "modelopt/**"
-    #   - "tests/gpu/**"
-    #   - "pyproject.toml"
-    #   - "setup.py"
-    #   - "tox.ini"
+    # NOTE: paths cannot be used since push happens to copied PR and only latest commit to PR is used
 
 # Cancel previous runs if new commit is pushed to the same PR
 concurrency:
-  group: ${{ github.workflow }}-${{ github.ref }}
+  group: ${{ github.workflow }}-${{ github.sha }}
   cancel-in-progress: true
 
 jobs:
@@ -47,7 +40,22 @@ jobs:
         PIP_CONSTRAINT: "" # Disable pip constraint for upgrading packages
     steps:
       - uses: actions/checkout@v4
-      - name: Setup proxy cache
-        uses: nv-gha-runners/setup-proxy-cache@main
+      - uses: nv-gha-runners/setup-proxy-cache@main
+      # Check if relevant files are changed in PR else skip gpu tests
+      - 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]
+        with:
+          files: |
+            .github/workflows/gpu_tests.yml
+            modelopt/**
+            tests/gpu/**
+            tox.ini
+            pyproject.toml
+            setup.py
+          base_sha: ${{ fromJSON(steps.get-pr-info.outputs.pr-info).base.ref }}
       - name: Run gpu tests
+        if: steps.changed-tests.outputs.any_changed == 'true'
         run: pip install tox-current-env && tox -e py312-cuda12-gpu --current-env

.github/workflows/pages.yml

@@ -8,7 +8,7 @@ on:
 
 # Cancel previous runs if new commit is pushed
 concurrency:
-  group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
+  group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.sha }}
   cancel-in-progress: true
 
 # Sets permissions of the GITHUB_TOKEN to allow deployment to GitHub Pages

.github/workflows/unit_tests.yml

@@ -22,7 +22,7 @@ on:
 
 # Cancel previous runs if new commit is pushed
 concurrency:
-  group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
+  group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.sha }}
   cancel-in-progress: true
 
 jobs: