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

Expose the precision dtype to the caller so that users can specify the precision for testing

Browse Source
This commit is contained in:
Liwei Yang
2023-04-19 16:11:14 +08:00
parent 5bfa1d9f52
commit 4d36afb9ef
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/test_cuquantum_cutensor_backend.py
View File

@@ -23,7 +23,14 @@ def time(func):
@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, dtype="complex128"):
"""Evaluate QASM with cuQuantum.
Args:
nqubits (int): Total number of qubits in the system.
dtype (str): The data type for precision, 'complex64' for single,
'complex128' for double.
"""
import qibotn.cutn import qibotn.cutn
# Test qibo # Test qibo
@@ -33,8 +40,7 @@ def test_eval(nqubits: int):
lambda: qibo_qft(nqubits, swaps=True)) lambda: qibo_qft(nqubits, swaps=True))
# Test Cuquantum # Test Cuquantum
data_type = "complex128" cutn_time, result_tn = time(lambda: qibotn.cutn.eval(qibo_circ, dtype))
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( assert np.allclose(