From 88e782bb293f56c979f44408cc2b6481a28a96f7 Mon Sep 17 00:00:00 2001 From: Oliver Koenig Date: Tue, 12 Nov 2024 22:17:24 +0100 Subject: [PATCH] ci: Run gpuci on main Signed-off-by: Oliver Koenig --- .github/workflows/gpuci.yml | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/.github/workflows/gpuci.yml b/.github/workflows/gpuci.yml index d7a4aecc..01a3146d 100644 --- 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/_build_container.yml@v0.11.0 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