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

Remove debug print

Browse Source
This commit is contained in:
tankya2
2025-08-26 17:40:50 +08:00
parent 304ae7a6a3
commit 947a2a1583
1 changed files with 0 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/test_cuquantum_cutensor_mpi_backend.py
View File

@@ -73,9 +73,6 @@ def test_eval_mpi(nqubits: int, dtype="complex128"):
result_tn_cp = cp.asarray(result_tn.statevector.flatten()) result_tn_cp = cp.asarray(result_tn.statevector.flatten())
print(f"State vector difference: {abs(result_tn_cp - result_sv_cp).max():0.3e}") print(f"State vector difference: {abs(result_tn_cp - result_sv_cp).max():0.3e}")
print(
f"AndyRank {backend.rank} qibo {result_sv_cp} tn {result_tn_cp} State vector difference: {abs(result_tn_cp - result_sv_cp).max():0.3e}"
)
if backend.rank == 0: if backend.rank == 0:
@@ -209,20 +206,11 @@ def test_eval_nccl(nqubits: int, dtype="complex128"):
result_tn = backend.execute_circuit(circuit=qibo_circ) result_tn = backend.execute_circuit(circuit=qibo_circ)
result_tn_cp = cp.asarray(result_tn.statevector.flatten()) result_tn_cp = cp.asarray(result_tn.statevector.flatten())
# print(
# f"Rank {backend.rank} qibo {result_sv_cp} tn {result_sv_cp} State vector difference: {abs(result_tn_cp - result_sv_cp).max():0.3e}"
# )
if backend.rank == 0: if backend.rank == 0:
print(
f"Rank {backend.rank} qibo {result_sv_cp} tn {result_tn_cp} State vector difference: {abs(result_tn_cp - result_sv_cp).max():0.3e}"
)
assert cp.allclose( assert cp.allclose(
result_sv_cp, result_tn_cp result_sv_cp, result_tn_cp
), "Resulting dense vectors do not match" ), "Resulting dense vectors do not match"
else: else:
print(
f"Rank {backend.rank} qibo {result_sv_cp} tn {result_tn_cp} State vector difference: {abs(result_tn_cp - result_sv_cp).max():0.3e}"
)
assert ( assert (
isinstance(result_tn_cp, cp.ndarray) isinstance(result_tn_cp, cp.ndarray)
and result_tn_cp.size == 1 and result_tn_cp.size == 1