[Nemo CICD] Try trigger cicd run on comment (NVIDIA#9111)

* try trigger cicd run on comment * try trigger cicd run on comment
HuiyingLi · May 4, 2024 · b198a11 · b198a11
1 parent be606a0
commit b198a11
Showing 1 changed file with 29 additions and 3 deletions.
diff --git a/.github/workflows/cicd-main.yml b/.github/workflows/cicd-main.yml
@@ -17,6 +17,8 @@ on:
   pull_request:
     branches: [ "main" ]
     types: [ labeled ]
+  issue_comment:
+    types: [ created ]
 
 concurrency:
   group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
@@ -25,7 +27,15 @@ concurrency:
 jobs:
   gpu-test:
     runs-on: self-hosted-azure
-    if: ${{ github.event.label.name == 'Run CICD' }}
+    if: >
+      (
+        github.event.label.name == 'Run CICD'
+      ) || (
+        github.event_name == 'issue_comment' &&
+        github.event.issue.pull_request &&
+        contains(fromJSON('["MEMBER", "COLLABORATOR", "OWNER"]'), github.event.comment.author_association) &&
+        contains(github.event.comment.body, '/cicd')
+      )
     steps:
     - name: Run nvidia-smi test
       run: |
@@ -34,7 +44,15 @@ jobs:
 
   cicd-cluster-clean:
     runs-on: self-hosted-azure-builder
-    if: ${{ github.event.label.name == 'Run CICD' }}
+    if: >
+      (
+        github.event.label.name == 'Run CICD'
+      ) || (
+        github.event_name == 'issue_comment' &&
+        github.event.issue.pull_request &&
+        contains(fromJSON('["MEMBER", "COLLABORATOR", "OWNER"]'), github.event.comment.author_association) &&
+        contains(github.event.comment.body, '/cicd')
+      )
     steps:
     - name: Clean server from old files
       run: |
@@ -57,7 +75,15 @@ jobs:
   cicd-test-container-setup:
     needs: [cicd-cluster-clean]
     runs-on: self-hosted-azure-builder
-    if: ${{ github.event.label.name == 'Run CICD' }}
+    if: >
+      (
+        github.event.label.name == 'Run CICD'
+      ) || (
+        github.event_name == 'issue_comment' &&
+        github.event.issue.pull_request &&
+        contains(fromJSON('["MEMBER", "COLLABORATOR", "OWNER"]'), github.event.comment.author_association) &&
+        contains(github.event.comment.body, '/cicd')
+      )
     # uses: actions/cache@v2
     #container:
 #      image: nvcr.io/nvidia/pytorch:24.02-py3