From 32981d0faeeb75baddf5137469c5caf5f0ea0a14 Mon Sep 17 00:00:00 2001 From: Vincent Michaud-Rioux Date: Wed, 25 Oct 2023 19:38:52 +0000 Subject: [PATCH] Fix gpu runner --- .github/workflows/tests_gpu_cu11.yml | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/.github/workflows/tests_gpu_cu11.yml b/.github/workflows/tests_gpu_cu11.yml index f590a840e1..185f3f2563 100644 --- 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