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

Avoid flatten() so as to keep the shape information of contraction results

Browse Source
This commit is contained in:
Liwei Yang
2023-04-19 16:23:59 +08:00
parent 4d36afb9ef
commit e517b4fe7c
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/qibotn/cutn.py
View File

@@ -7,4 +7,4 @@ def eval(qibo_circ, datatype):
myconvertor = QiboCircuitToEinsum(qibo_circ, dtype=datatype) myconvertor = QiboCircuitToEinsum(qibo_circ, dtype=datatype)
operands_expression = myconvertor.state_vector() operands_expression = myconvertor.state_vector()
results = contract(*operands_expression) results = contract(*operands_expression)
return results.flatten() return results

3
tests/test_cuquantum_cutensor_backend.py
View File

@@ -40,7 +40,8 @@ def test_eval(nqubits: int, dtype="complex128"):
lambda: qibo_qft(nqubits, swaps=True)) lambda: qibo_qft(nqubits, swaps=True))
# Test Cuquantum # Test Cuquantum
cutn_time, result_tn = time(lambda: qibotn.cutn.eval(qibo_circ, dtype)) cutn_time, result_tn = time(
lambda: qibotn.cutn.eval(qibo_circ, dtype).flatten())
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(