Drop circuit-specific dedicated handling for swaps

This commit is contained in:
Alessandro Candido
2023-02-15 16:36:42 +01:00
parent 0af53d1ab0
commit 662adfe158
3 changed files with 16 additions and 20 deletions

View File

@@ -3,16 +3,17 @@ import quimb.tensor as qtn
from qibo.models import Circuit as QiboCircuit
def from_qibo(circuit: QiboCircuit, swaps: bool = True, psi0=None):
def from_qibo(circuit: QiboCircuit, psi0=None):
nqubits = circuit.nqubits
tncirc = qtn.Circuit(nqubits, psi0=psi0)
for gate in circuit.queue:
tncirc.apply_gate(gate.name)
if swaps:
for i in range(nqubits // 2): # TODO: Ignore the barrier indices?
tncirc.apply_gate("SWAP", i, nqubits - i - 1)
tncirc.apply_gate(
gate.name,
*gate.parameters,
*gate.qubits,
parametrize=len(gate.parameters) > 0
)
return tncirc
@@ -23,7 +24,7 @@ def init_state_tn(nqubits, init_state_sv):
return qtn.tensor_1d.MatrixProductState.from_dense(init_state_sv, dims)
def eval(qasm: str, init_state, backend="numpy", swaps=True):
def eval(qasm: str, init_state, backend="numpy"):
"""Evaluate QASM with Quimb
backend (quimb): numpy, cupy, jax. Passed to ``opt_einsum``.
@@ -31,7 +32,7 @@ def eval(qasm: str, init_state, backend="numpy", swaps=True):
"""
circuit = QiboCircuit.from_qasm(qasm)
init_state_mps = init_state_tn(circuit.nqubits, init_state)
circ_quimb = from_qibo(circuit, swaps=swaps, psi0=init_state_mps)
circ_quimb = from_qibo(circuit, psi0=init_state_mps)
interim = circ_quimb.psi.full_simplify(seq="DRC")
amplitudes = interim.to_dense(backend=backend).flatten()