MPS execution for quimb

src/qibotn/quimb.py

@@ -3,16 +3,25 @@ import quimb.tensor as qtn
 from qibo.models import Circuit as QiboCircuit
 
 
-def from_qibo(circuit: QiboCircuit, psi0=None):
+def from_qibo(circuit: QiboCircuit, is_mps: False, psi0=None, method='svd', 
+              cutoff=1e-6, cutoff_mode='abs'):
     nqubits = circuit.nqubits
-    tncirc = qtn.Circuit(nqubits, psi0=psi0)
+    gate_opt = {}
+    if (is_mps):
+        tncirc = qtn.CircuitMPS(nqubits, psi0=psi0)
+        gate_opt["method"] = method
+        gate_opt["cutoff"] = cutoff
+        gate_opt["cutoff_mode"] = cutoff_mode
+    else:    
+        tncirc = qtn.Circuit(nqubits, psi0=psi0)
 
     for gate in circuit.queue:
         tncirc.apply_gate(
             gate.name,
             *gate.parameters,
             *gate.qubits,
-            parametrize=len(gate.parameters) > 0
+            parametrize=False if is_mps else (len(gate.parameters) > 0),
+            **gate_opt
         )
 
     return tncirc
@@ -32,7 +41,7 @@ def eval(qasm: str, init_state, backend="numpy"):
     """
     circuit = QiboCircuit.from_qasm(qasm)
     init_state_mps = init_state_tn(circuit.nqubits, init_state)
-    circ_quimb = from_qibo(circuit, psi0=init_state_mps)
+    circ_quimb = from_qibo(circuit, is_mps=True, psi0=init_state_mps)
     interim = circ_quimb.psi.full_simplify(seq="DRC")
     amplitudes = interim.to_dense(backend=backend).flatten()