chore(ci): optimize triggering of gpu workflows

zama-ai · Jun 28, 2024 · c07f89c · c07f89c
1 parent 22e9505
commit c07f89c
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 4 deletions.
diff --git a/.github/workflows/aws_tfhe_integer_gpu_tests.yml b/.github/workflows/aws_tfhe_integer_gpu_tests.yml
@@ -86,6 +86,7 @@ jobs:
           } >> "${GITHUB_ENV}"
 
       - name: Run unsigned integer tests
+        if: github.event_name != 'pull_request' || contains(fromJSON(env.LABELS), 'approved')
         run: |
           make test_unsigned_integer_gpu_ci
 

diff --git a/.github/workflows/aws_tfhe_multi_gpu_tests.yml b/.github/workflows/aws_tfhe_multi_gpu_tests.yml
@@ -16,10 +16,12 @@ on:
   # Allows you to run this workflow manually from the Actions tab as an alternative.
   workflow_dispatch:
   pull_request:
+    types: [ labeled ]
 
 jobs:
   setup-instance:
     name: Setup instance (cuda-tests-multi-gpu)
+    if: github.event_name != 'pull_request' || contains(github.event.label.name, 'approved')
     runs-on: ubuntu-latest
     outputs:
       runner-name: ${{ steps.start-instance.outputs.label }}
@@ -85,9 +87,10 @@ jobs:
             echo "HOME=/home/ubuntu";
           } >> "${GITHUB_ENV}"
 
-      - name: Run core crypto, integer and internal CUDA backend tests
+      # No need to test core_crypto and classic PBS in integer since it's already tested on single GPU.
+      - name: Run multi-bit CUDA integer tests
         run: |
-          make test_gpu
+          make test_integer_multi_bit_gpu_ci
 
       - name: Run user docs tests
         run: |

diff --git a/.github/workflows/aws_tfhe_signed_integer_gpu_tests.yml b/.github/workflows/aws_tfhe_signed_integer_gpu_tests.yml
@@ -86,6 +86,7 @@ jobs:
           } >> "${GITHUB_ENV}"
 
       - name: Run signed integer tests
+        if: github.event_name != 'pull_request' || contains(fromJSON(env.LABELS), 'approved')
         run: |
           make test_signed_integer_gpu_ci
 

diff --git a/.github/workflows/hyperstack_tfhe_gpu_tests.yml b/.github/workflows/hyperstack_tfhe_gpu_tests.yml
@@ -16,10 +16,12 @@ on:
   # Allows you to run this workflow manually from the Actions tab as an alternative.
   workflow_dispatch:
   pull_request:
+      types: [ labeled ]
 
 jobs:
   setup-instance:
     name: Setup instance (cuda-h100-tests)
+    if: github.event_name != 'pull_request' || contains(github.event.label.name, 'approved')
     runs-on: ubuntu-latest
     outputs:
       runner-name: ${{ steps.start-instance.outputs.label }}
@@ -110,9 +112,18 @@ jobs:
         if: ${{ !cancelled() }}
         run: nvidia-smi
 
-      - name: Run core crypto, integer and internal CUDA backend tests
+      - name: Run core crypto and internal CUDA backend tests
         run: |
-          make test_gpu
+          make test_core_crypto_gpu
+          make test_cuda_backend
+
+      - name: Run integer tests
+        run: |
+          make test_integer_gpu_ci
+
+      - name: Run integer multi-bit tests
+        run: |
+          make test_integer_multi_bit_gpu_ci
 
       - name: Run user docs tests
         run: |