ci: Run on main (NVIDIA#354)

* ci: Run gpuci on main
* fix checkout

Signed-off-by: Oliver Koenig <[email protected]>
Signed-off-by: Rucha Apte <[email protected]>

.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

Dockerfile

@@ -18,7 +18,7 @@ RUN bash -exu <<EOF
   cd /opt/NeMo-Curator
   git init
   git remote add origin $FORKED_REPO_URL
-  git fetch origin $CURATOR_COMMIT --depth=1
+  git fetch origin '+refs/pull/*/merge:refs/remotes/pull/*/merge'
   git checkout $CURATOR_COMMIT
 EOF