Fix gpu runner

PennyLaneAI · Oct 25, 2023 · 32981d0 · 32981d0
1 parent ce8a959
commit 32981d0
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/.github/workflows/tests_gpu_cu11.yml b/.github/workflows/tests_gpu_cu11.yml
@@ -59,7 +59,11 @@ jobs:
         pl_backend: ["lightning_gpu"]
 
     name: C++ tests (Lightning-GPU)
-    runs-on: ${{ matrix.os }}
+    runs-on:
+      - ${{ matrix.os }}
+      - self-hosted
+      - gpu
+
 
     steps:
       - name: Validate GPU version and installed compiler
@@ -162,7 +166,11 @@ jobs:
         default_backend: ["lightning_qubit"]
 
     name: Python tests with LGPU
-    runs-on: ${{ matrix.os }}
+    runs-on:
+      - ${{ matrix.os }}
+      - self-hosted
+      - gpu
+
 
     steps:
       - name: Validate GPU version and installed compiler