jaunatisblue/final-qibotn
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove unused imports, format it using black

Browse Source
This commit is contained in:
Liwei Yang
2023-04-19 15:12:08 +08:00
parent bea6b1c166
commit 725c92e75d
1 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/test_cuquantum_cutensor_backend.py
View File

@@ -1,5 +1,3 @@
import copy
import os
from timeit import default_timer as timer from timeit import default_timer as timer
import config import config
@@ -22,18 +20,22 @@ def time(func):
time = end - start time = end - start
return time, res return time, res
@pytest.mark.gpu @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
# Test qibo # Test qibo
qibo.set_backend(backend=config.qibo.backend, platform=config.qibo.platform) qibo.set_backend(backend=config.qibo.backend,
qibo_time, (qibo_circ, result_sv) = time(lambda: qibo_qft(nqubits, swaps=True)) platform=config.qibo.platform)
qibo_time, (qibo_circ, result_sv) = time(
lambda: qibo_qft(nqubits, swaps=True))
# Test Cuquantum # Test Cuquantum
data_type = "complex128" data_type = "complex128"
cutn_time, result_tn = time(lambda: qibotn.cutn.eval(qibo_circ,data_type)) cutn_time, result_tn = time(lambda: qibotn.cutn.eval(qibo_circ, data_type))
assert 1e-2 * qibo_time < cutn_time < 1e2 * qibo_time assert 1e-2 * qibo_time < cutn_time < 1e2 * qibo_time
assert np.allclose(result_sv, result_tn), "Resulting dense vectors do not match" assert np.allclose(
result_sv, result_tn), "Resulting dense vectors do not match"