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

Drop circuit-specific dedicated handling for swaps

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

17
src/qibotn/quimb.py
View File

@@ -3,16 +3,17 @@ import quimb.tensor as qtn
from qibo.models import Circuit as QiboCircuit 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 nqubits = circuit.nqubits
tncirc = qtn.Circuit(nqubits, psi0=psi0) tncirc = qtn.Circuit(nqubits, psi0=psi0)
for gate in circuit.queue: for gate in circuit.queue:
tncirc.apply_gate(gate.name) tncirc.apply_gate(
gate.name,
if swaps: *gate.parameters,
for i in range(nqubits // 2): # TODO: Ignore the barrier indices? *gate.qubits,
tncirc.apply_gate("SWAP", i, nqubits - i - 1) parametrize=len(gate.parameters) > 0
)
return tncirc 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) 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 """Evaluate QASM with Quimb
backend (quimb): numpy, cupy, jax. Passed to ``opt_einsum``. 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) circuit = QiboCircuit.from_qasm(qasm)
init_state_mps = init_state_tn(circuit.nqubits, init_state) 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") interim = circ_quimb.psi.full_simplify(seq="DRC")
amplitudes = interim.to_dense(backend=backend).flatten() amplitudes = interim.to_dense(backend=backend).flatten()

1
tests/config.py
View File

@@ -6,7 +6,6 @@ from typing import Optional
class Executor: class Executor:
backend: str backend: str
platform: Optional[str] = None platform: Optional[str] = None
swaps: bool = True
qibo = Executor(backend="qibojit", platform="numpy") qibo = Executor(backend="qibojit", platform="numpy")

18
tests/test_qasm_quimb_backend.py
View File

@@ -1,13 +1,12 @@
import os
import pytest
import qibo
from qibo.models import QFT
import numpy as np
import copy import copy
import os
from timeit import default_timer as timer from timeit import default_timer as timer
import config import config
import numpy as np
import pytest
import qibo
from qibo.models import QFT
def create_init_state(nqubits): def create_init_state(nqubits):
@@ -44,7 +43,7 @@ def test_eval(nqubits: int):
# Test qibo # Test qibo
qibo.set_backend(backend=config.qibo.backend, platform=config.qibo.platform) qibo.set_backend(backend=config.qibo.backend, platform=config.qibo.platform)
qibo_time, (qibo_circ, result_sv) = time( qibo_time, (qibo_circ, result_sv) = time(
lambda: qibo_qft(nqubits, init_state, swaps=config.qibo.swaps) lambda: qibo_qft(nqubits, init_state, swaps=True)
) )
# Convert to qasm for other backends # Convert to qasm for other backends
@@ -53,10 +52,7 @@ def test_eval(nqubits: int):
# Test quimb # Test quimb
quimb_time, result_tn = time( quimb_time, result_tn = time(
lambda: qibotn.quimb.eval( lambda: qibotn.quimb.eval(
qasm_circ, qasm_circ, init_state_tn, backend=config.quimb.backend
init_state_tn,
backend=config.quimb.backend,
swaps=config.quimb.swaps,
) )
) )