From 57414dfb446150ead6e0ddadef20c965f9b51047 Mon Sep 17 00:00:00 2001 From: tankya2 Date: Mon, 3 Apr 2023 10:18:53 +0800 Subject: [PATCH] added in decorator to mark GPU --- tests/test_cuquantum_cutensor_backend.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_cuquantum_cutensor_backend.py b/tests/test_cuquantum_cutensor_backend.py index 9c989a3..07bb337 100644 --- a/tests/test_cuquantum_cutensor_backend.py +++ b/tests/test_cuquantum_cutensor_backend.py @@ -22,7 +22,7 @@ def time(func): time = end - start return time, res - +@pytest.mark.gpu @pytest.mark.parametrize("nqubits", [1, 2, 5, 10]) def test_eval(nqubits: int): import qibotn.cutn