added in decorator to mark GPU

This commit is contained in:
tankya2
2023-04-03 10:18:53 +08:00
parent 04c4190c98
commit 57414dfb44

View File

@@ -22,7 +22,7 @@ def time(func):
time = end - start time = end - start
return time, res return time, res
@pytest.mark.gpu
@pytest.mark.parametrize("nqubits", [1, 2, 5, 10]) @pytest.mark.parametrize("nqubits", [1, 2, 5, 10])
def test_eval(nqubits: int): def test_eval(nqubits: int):
import qibotn.cutn import qibotn.cutn