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

Adding feature to pass MPS parameters in quimb

Browse Source
This commit is contained in:
vinitha-balachandran
2024-02-23 14:48:30 +08:00
parent b6bd9bdfd5
commit de92060180
3 changed files with 20 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/qibotn/eval_qu.py
View File

@@ -10,7 +10,7 @@ def init_state_tn(nqubits, init_state_sv):
return qtn.tensor_1d.MatrixProductState.from_dense(init_state_sv, dims)
def dense_vector_tn_qu(qasm: str, initial_state, is_mps, backend="numpy"):
def dense_vector_tn_qu(qasm: str, initial_state, mps_opts, backend="numpy"):
"""Evaluate QASM with Quimb.
backend (quimb): numpy, cupy, jax. Passed to ``opt_einsum``.
@@ -20,14 +20,9 @@ def dense_vector_tn_qu(qasm: str, initial_state, is_mps, backend="numpy"):
nqubits = int(np.log2(len(initial_state)))
initial_state = init_state_tn(nqubits, initial_state)
if is_mps:
gate_opt = {}
gate_opt["method"] = "svd"
gate_opt["cutoff"] = 1e-6
gate_opt["cutoff_mode"] = "abs"
if mps_opts:
circ_quimb = qtn.circuit.CircuitMPS.from_openqasm2_str(
qasm, psi0=initial_state, gate_opts=gate_opt
qasm, psi0=initial_state, gate_opts=mps_opts
)
else: