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

Make tolerance a variable

Browse Source
This commit is contained in:
tankya2
2025-09-03 10:18:13 +08:00
parent 99fbabb9c8
commit b72f6afe49
2 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/test_cuquantum_cutensor_backend.py
View File

@@ -7,6 +7,7 @@ from qibo import construct_backend, hamiltonians
from qibo.models import QFT from qibo.models import QFT
from qibo.symbols import X, Z from qibo.symbols import X, Z
ABS_TOL = 1e-7
def qibo_qft(nqubits, swaps): def qibo_qft(nqubits, swaps):
circ_qibo = QFT(nqubits, swaps) circ_qibo = QFT(nqubits, swaps)
@@ -165,7 +166,7 @@ def test_expectation(nqubits: int, dtype="complex128"):
} }
backend.configure_tn_simulation(computation_settings_1) backend.configure_tn_simulation(computation_settings_1)
result_tn = backend.execute_circuit(circuit=qibo_circ) result_tn = backend.execute_circuit(circuit=qibo_circ)
assert math.isclose(exact_expval.item(), result_tn.real.get().item(), abs_tol=1e-7) assert math.isclose(exact_expval.item(), result_tn.real.get().item(), abs_tol=ABS_TOL)
# Test with user defined hamiltonian using "hamiltonians.SymbolicHamiltonian" object. # Test with user defined hamiltonian using "hamiltonians.SymbolicHamiltonian" object.
computation_settings_2 = { computation_settings_2 = {
@@ -176,7 +177,7 @@ def test_expectation(nqubits: int, dtype="complex128"):
} }
backend.configure_tn_simulation(computation_settings_2) backend.configure_tn_simulation(computation_settings_2)
result_tn = backend.execute_circuit(circuit=qibo_circ) result_tn = backend.execute_circuit(circuit=qibo_circ)
assert math.isclose(exact_expval.item(), result_tn.real.get().item(), abs_tol=1e-7) assert math.isclose(exact_expval.item(), result_tn.real.get().item(), abs_tol=ABS_TOL)
# Test with user defined hamiltonian using Dictionary object form of hamiltonian. # Test with user defined hamiltonian using Dictionary object form of hamiltonian.
ham_dict = build_observable_dict(nqubits) ham_dict = build_observable_dict(nqubits)
@@ -188,4 +189,4 @@ def test_expectation(nqubits: int, dtype="complex128"):
} }
backend.configure_tn_simulation(computation_settings_3) backend.configure_tn_simulation(computation_settings_3)
result_tn = backend.execute_circuit(circuit=qibo_circ) result_tn = backend.execute_circuit(circuit=qibo_circ)
assert math.isclose(exact_expval.item(), result_tn.real.get().item(), abs_tol=1e-7) assert math.isclose(exact_expval.item(), result_tn.real.get().item(), abs_tol=ABS_TOL)

13
tests/test_cuquantum_cutensor_mpi_backend.py
View File

@@ -10,6 +10,7 @@ from qibo import construct_backend, hamiltonians
from qibo.models import QFT from qibo.models import QFT
from qibo.symbols import X, Z from qibo.symbols import X, Z
ABS_TOL = 1e-7
def qibo_qft(nqubits, swaps): def qibo_qft(nqubits, swaps):
circ_qibo = QFT(nqubits, swaps) circ_qibo = QFT(nqubits, swaps)
@@ -118,7 +119,7 @@ def test_expectation_mpi(nqubits: int, dtype="complex128"):
if backend.rank == 0: if backend.rank == 0:
# Compare numerical values # Compare numerical values
assert math.isclose( assert math.isclose(
exact_expval.item(), float(result_tn[0]), abs_tol=1e-7 exact_expval.item(), float(result_tn[0]), abs_tol=ABS_TOL
), f"Rank {backend.rank}: mismatch, expected {exact_expval}, got {result_tn}" ), f"Rank {backend.rank}: mismatch, expected {exact_expval}, got {result_tn}"
else: else:
@@ -142,7 +143,7 @@ def test_expectation_mpi(nqubits: int, dtype="complex128"):
if backend.rank == 0: if backend.rank == 0:
# Compare numerical values # Compare numerical values
assert math.isclose( assert math.isclose(
exact_expval.item(), float(result_tn[0]), abs_tol=1e-7 exact_expval.item(), float(result_tn[0]), abs_tol=ABS_TOL
), f"Rank {backend.rank}: mismatch, expected {exact_expval}, got {result_tn}" ), f"Rank {backend.rank}: mismatch, expected {exact_expval}, got {result_tn}"
else: else:
@@ -167,7 +168,7 @@ def test_expectation_mpi(nqubits: int, dtype="complex128"):
if backend.rank == 0: if backend.rank == 0:
# Compare numerical values # Compare numerical values
assert math.isclose( assert math.isclose(
exact_expval.item(), float(result_tn[0]), abs_tol=1e-7 exact_expval.item(), float(result_tn[0]), abs_tol=ABS_TOL
), f"Rank {backend.rank}: mismatch, expected {exact_expval}, got {result_tn}" ), f"Rank {backend.rank}: mismatch, expected {exact_expval}, got {result_tn}"
else: else:
@@ -251,7 +252,7 @@ def test_expectation_NCCL(nqubits: int, dtype="complex128"):
if backend.rank == 0: if backend.rank == 0:
# Compare numerical values # Compare numerical values
assert math.isclose( assert math.isclose(
exact_expval.item(), float(result_tn[0]), abs_tol=1e-7 exact_expval.item(), float(result_tn[0]), abs_tol=ABS_TOL
), f"Rank {backend.rank}: mismatch, expected {exact_expval}, got {result_tn}" ), f"Rank {backend.rank}: mismatch, expected {exact_expval}, got {result_tn}"
else: else:
@@ -275,7 +276,7 @@ def test_expectation_NCCL(nqubits: int, dtype="complex128"):
if backend.rank == 0: if backend.rank == 0:
# Compare numerical values # Compare numerical values
assert math.isclose( assert math.isclose(
exact_expval.item(), float(result_tn[0]), abs_tol=1e-7 exact_expval.item(), float(result_tn[0]), abs_tol=ABS_TOL
), f"Rank {backend.rank}: mismatch, expected {exact_expval}, got {result_tn}" ), f"Rank {backend.rank}: mismatch, expected {exact_expval}, got {result_tn}"
else: else:
@@ -300,7 +301,7 @@ def test_expectation_NCCL(nqubits: int, dtype="complex128"):
if backend.rank == 0: if backend.rank == 0:
# Compare numerical values # Compare numerical values
assert math.isclose( assert math.isclose(
exact_expval.item(), float(result_tn[0]), abs_tol=1e-7 exact_expval.item(), float(result_tn[0]), abs_tol=ABS_TOL
), f"Rank {backend.rank}: mismatch, expected {exact_expval}, got {result_tn}" ), f"Rank {backend.rank}: mismatch, expected {exact_expval}, got {result_tn}"
else: else: