parallelize gpu test

PennyLaneAI · Jan 17, 2025 · 0a0b40c · 0a0b40c
1 parent f8d7453
commit 0a0b40c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/.github/workflows/tests_gpu_python.yml b/.github/workflows/tests_gpu_python.yml
@@ -31,7 +31,7 @@ concurrency:
 
 jobs:
   builddeps:
-    if: ${{ github.event_name != 'pull_request' || (github.event.pull_request.draft == false && contains(github.event.pull_request.labels.*.name, 'ci:use-gpu-runner')) }}
+    if: ${{ github.event_name != 'pull_request' || contains(github.event.pull_request.labels.*.name, 'ci:use-gpu-runner') }}
     runs-on:
       - self-hosted
       - ubuntu-22.04
@@ -206,7 +206,8 @@ jobs:
           OMP_PROC_BIND: false
         run: |
           DEVICENAME=`echo ${{ matrix.pl_backend }} | sed "s/_/./g"`
-          PL_DEVICE=${DEVICENAME} python -m pytest tests/ $COVERAGE_FLAGS
+          PL_DEVICE=${DEVICENAME} python -m pytest -n 2 tests/ -k "not unitary_correct" $COVERAGE_FLAGS
+          PL_DEVICE=${DEVICENAME} python -m pytest tests/ -k "unitary_correct" $COVERAGE_FLAGS
 
       - name: Test wheels for Lightning-GPU
         if : matrix.pl_backend == 'lightning_gpu'