ci: Run gpuci on main

Signed-off-by: Oliver Koenig <[email protected]>
NVIDIA · Nov 12, 2024 · 88e782b · 88e782b
1 parent 3d6778d
commit 88e782b
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/.github/workflows/gpuci.yml b/.github/workflows/gpuci.yml
@@ -1,6 +1,9 @@
 name: "GPU CI/CD"
 
 on:
+  push:
+    branches:
+      - main
   pull_request:
     branches:
       # We can run gpuCI on any PR targeting these branches
@@ -19,7 +22,7 @@ jobs:
   # First, we build and push a NeMo-Curator container
   build-container:
     # "build-container" job is run if the "gpuci" label is added to the PR
-    if: ${{ github.event.label.name == 'gpuci' }}
+    if: ${{ github.event.label.name == 'gpuci' || github.ref == 'refs/heads/main' }}
     uses: NVIDIA/NeMo-FW-CI-templates/.github/workflows/[email protected]
     with:
       image-name: nemo_curator_container
@@ -38,7 +41,7 @@ jobs:
     # It has 2 A100 GPUs
     runs-on: self-hosted-azure
     # "run-gpu-tests" job is run if the "gpuci" label is added to the PR
-    if: ${{ github.event.label.name == 'gpuci' }}
+    if: ${{ github.event.label.name == 'gpuci' || github.ref == 'refs/heads/main' }}
 
     steps:
       # If something went wrong during the last cleanup, this step ensures any existing container is removed