Adding the first version of qibo's qasm tested with Quimb backend
This commit is contained in:
Nitin Shivaraman
275
src/qibotn/test_qasm_quimb_backend.py
Normal file
275
src/qibotn/test_qasm_quimb_backend.py
Normal file
@@ -0,0 +1,275 @@
|
|||||||
|
import random
|
||||||
|
from turtle import delay
|
||||||
|
import quimb as qu
|
||||||
|
import quimb.tensor as qtn
|
||||||
|
import numpy as np
|
||||||
|
import re
|
||||||
|
|
||||||
|
from timeit import default_timer as timer
|
||||||
|
import cirq
|
||||||
|
|
||||||
|
nqubits = 18
|
||||||
|
|
||||||
|
# define dictionary
|
||||||
|
gate_dict_cirq = {
|
||||||
|
#'i': I,
|
||||||
|
'h': cirq.H,
|
||||||
|
't': cirq.T,
|
||||||
|
#'z': Z,
|
||||||
|
'cz': cirq.CZ,
|
||||||
|
#'cX': cX,
|
||||||
|
#'rz': ZPhase,
|
||||||
|
#'rX': XPhase,
|
||||||
|
#'x': X,
|
||||||
|
#'y': Y,
|
||||||
|
'x_1_2': cirq.rx(0.5*np.pi),
|
||||||
|
'y_1_2': cirq.ry(0.5*np.pi)
|
||||||
|
#'hz_1_2': W_1_2,
|
||||||
|
#'fs': fSim
|
||||||
|
}
|
||||||
|
|
||||||
|
# search 'GATE_FUNCTIONS' in Source code for quimb.tensor.circuit for pre-defined gates in quimb
|
||||||
|
gate_dict = {
|
||||||
|
#'i': I,
|
||||||
|
'h': 'H',
|
||||||
|
't': 'T',
|
||||||
|
#'z': Z,
|
||||||
|
'cz': 'cZ',
|
||||||
|
#'cX': cX,
|
||||||
|
#'rz': ZPhase,
|
||||||
|
#'rX': XPhase,
|
||||||
|
#'x': X,
|
||||||
|
#'y': Y,
|
||||||
|
'x_1_2': 'X_1_2',
|
||||||
|
'y_1_2': 'Y_1_2'
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def QI_QFT(nqubits: int, with_swaps: bool = True, psi0 = None):
|
||||||
|
## constructs qft circuit
|
||||||
|
#from qibo import gates
|
||||||
|
#circuit = Circuit(nqubits)
|
||||||
|
circ = qtn.Circuit(nqubits, psi0 = psi0)
|
||||||
|
|
||||||
|
for i1 in range(nqubits):
|
||||||
|
#circuit.add(gates.H(i1))
|
||||||
|
circ.apply_gate('H', i1)
|
||||||
|
for i2 in range(i1 + 1, nqubits):
|
||||||
|
theta = np.pi / 2 ** (i2 - i1)
|
||||||
|
#circuit.add(gates.CU1(i2, i1, theta))
|
||||||
|
circ.apply_gate('CU1', theta, i2, i1)
|
||||||
|
|
||||||
|
if with_swaps:
|
||||||
|
for i in range(nqubits // 2):
|
||||||
|
#circuit.add(gates.SWAP(i, nqubits - i - 1))
|
||||||
|
circ.apply_gate('SWAP', i, nqubits - i - 1)
|
||||||
|
|
||||||
|
return circ
|
||||||
|
|
||||||
|
def get_gate_params(operation):
|
||||||
|
if "h" in operation:
|
||||||
|
qbit_no = [int(re.findall(r'\d+', operation)[0])]
|
||||||
|
qbit_no.insert(0, "H")
|
||||||
|
elif "x" in operation:
|
||||||
|
qbit_no = [int(re.findall(r'\d+', operation)[0])]
|
||||||
|
qbit_no.insert(0, "X")
|
||||||
|
elif "y" in operation:
|
||||||
|
qbit_no = [int(re.findall(r'\d+', operation)[0])]
|
||||||
|
qbit_no.insert(0, "Y")
|
||||||
|
elif "z" in operation:
|
||||||
|
qbit_no = [int(re.findall(r'\d+', operation)[0])]
|
||||||
|
qbit_no.insert(0, "Z")
|
||||||
|
elif "s" in operation:
|
||||||
|
qbit_no = [int(re.findall(r'\d+', operation)[0])]
|
||||||
|
qbit_no.insert(0, "S")
|
||||||
|
elif "t" in operation:
|
||||||
|
qbit_no = [int(re.findall(r'\d+', operation)[0])]
|
||||||
|
qbit_no.insert(0, "T")
|
||||||
|
elif "cu1" in operation:
|
||||||
|
lamda = float('.'.join(re.findall(r'\b\d+(?:[Ee][+-]?\d+)?',
|
||||||
|
operation.split(" ")[0])))
|
||||||
|
qbit_no = re.findall(r'\d+', operation.split(" ")[1])
|
||||||
|
qbit_no = [int(x) for x in qbit_no]
|
||||||
|
qbit_no[0:0] = ["CU1", lamda]
|
||||||
|
elif "cu2" in operation:
|
||||||
|
angles = re.findall(r'\b\d+(?:[Ee][+-]?\d+)?',operation.split(" ")[0])
|
||||||
|
phi = float('.'.join(angles[0:2]))
|
||||||
|
lamba = float('.'.join(angles[2:]))
|
||||||
|
qbit_no = re.findall(r'\d+', operation.split(" ")[1])
|
||||||
|
qbit_no = [int(x) for x in qbit_no]
|
||||||
|
qbit_no[0:0] = ["CU2", phi, lamda]
|
||||||
|
elif "cu3" in operation:
|
||||||
|
angles = re.findall(r'\b\d+(?:[Ee][+-]?\d+)?',operation.split(" ")[0])
|
||||||
|
theta = float('.'.join(angles[0:2]))
|
||||||
|
phi = float('.'.join(angles[2:4]))
|
||||||
|
lamba = float('.'.join(angles[4:]))
|
||||||
|
qbit_no = re.findall(r'\d+', operation.split(" ")[1])
|
||||||
|
qbit_no = [int(x) for x in qbit_no]
|
||||||
|
qbit_no[0:0] = ["CU3", theta, phi, lamda]
|
||||||
|
elif "cx" in operation:
|
||||||
|
qbit_no = re.findall(r'\d+', operation.split(" ")[1])
|
||||||
|
qbit_no = [int(x) for x in qbit_no]
|
||||||
|
qbit_no.insert(0, "CX")
|
||||||
|
elif "cy" in operation:
|
||||||
|
qbit_no = re.findall(r'\d+', operation.split(" ")[1])
|
||||||
|
qbit_no = [int(x) for x in qbit_no]
|
||||||
|
qbit_no.insert(0, "CY")
|
||||||
|
elif "cz" in operation:
|
||||||
|
qbit_no = re.findall(r'\d+', operation.split(" ")[1])
|
||||||
|
qbit_no = [int(x) for x in qbit_no]
|
||||||
|
qbit_no.insert(0, "CZ")
|
||||||
|
elif "rx" in operation:
|
||||||
|
theta = float('.'.join(re.findall(r'\b\d+(?:[Ee][+-]?\d+)?',
|
||||||
|
operation.split(" ")[0])))
|
||||||
|
qbit_no = [int(re.findall(r'\d+', operation)[0])]
|
||||||
|
qbit_no[0:0] = ["RX", theta]
|
||||||
|
elif "ry" in operation:
|
||||||
|
theta = float('.'.join(re.findall(r'\b\d+(?:[Ee][+-]?\d+)?',
|
||||||
|
operation.split(" ")[0])))
|
||||||
|
qbit_no = [int(re.findall(r'\d+', operation)[0])]
|
||||||
|
qbit_no[0:0] = ["RY", theta]
|
||||||
|
elif "rz" in operation:
|
||||||
|
theta = float('.'.join(re.findall(r'\b\d+(?:[Ee][+-]?\d+)?',
|
||||||
|
operation.split(" ")[0])))
|
||||||
|
qbit_no = [int(re.findall(r'\d+', operation)[0])]
|
||||||
|
qbit_no[0:0] = ["RZ", theta]
|
||||||
|
elif "rzz" in operation:
|
||||||
|
theta = float('.'.join(re.findall(r'\b\d+(?:[Ee][+-]?\d+)?',
|
||||||
|
operation.split(" ")[0])))
|
||||||
|
qbit_no = re.findall(r'\d+', operation.split(" ")[1])
|
||||||
|
qbit_no = [int(x) for x in qbit_no]
|
||||||
|
qbit_no[0:0] = ["RZZ", theta]
|
||||||
|
elif "u1" in operation:
|
||||||
|
lamda = float('.'.join(re.findall(r'\b\d+(?:[Ee][+-]?\d+)?',
|
||||||
|
operation.split(" ")[0])))
|
||||||
|
qbit_no = [int(re.findall(r'\d+', operation)[0])]
|
||||||
|
qbit_no[0:0] = ["U1", lamda]
|
||||||
|
elif "u2" in operation:
|
||||||
|
angles = re.findall(r'\b\d+(?:[Ee][+-]?\d+)?',operation.split(" ")[0])
|
||||||
|
phi = float('.'.join(angles[0:2]))
|
||||||
|
lamba = float('.'.join(angles[2:]))
|
||||||
|
qbit_no = int(re.findall(r'\d+', operation)[0])
|
||||||
|
qbit_no[0:0] = ["U2", phi, lamda]
|
||||||
|
elif "u3" in operation:
|
||||||
|
angles = re.findall(r'\b\d+(?:[Ee][+-]?\d+)?',operation.split(" ")[0])
|
||||||
|
theta = float('.'.join(angles[0:2]))
|
||||||
|
phi = float('.'.join(angles[2:4]))
|
||||||
|
lamba = float('.'.join(angles[4:]))
|
||||||
|
qbit_no = int(re.findall(r'\d+', operation)[0])
|
||||||
|
qbit_no[0:0] = ["U3", theta, phi, lamda]
|
||||||
|
else:
|
||||||
|
assert("Unsupported gate")
|
||||||
|
|
||||||
|
return qbit_no
|
||||||
|
|
||||||
|
|
||||||
|
def get_gate_functions(qasm_str, start_idx):
|
||||||
|
# func_list = []
|
||||||
|
# param_list = {}
|
||||||
|
# for line in qasm_str[start_idx:]:
|
||||||
|
# if "gate" in line:
|
||||||
|
# line = line.split(" ")
|
||||||
|
# for i in line[3:]:
|
||||||
|
# if ',' in i:
|
||||||
|
# params = i.split(",")
|
||||||
|
# param_list.append([int(j) for j in params])
|
||||||
|
# elif "(" in i:
|
||||||
|
# params = re.findall(r'\w+', i)
|
||||||
|
# param_list.append([int(j) for j in params])
|
||||||
|
# elif "{" in i:
|
||||||
|
# break
|
||||||
|
# elif "}" in line:
|
||||||
|
# return func_list
|
||||||
|
# else:
|
||||||
|
# func_list.append(line)
|
||||||
|
pass
|
||||||
|
|
||||||
|
def qasm_QFT(nqubits:int, qasm_str:str, with_swaps: bool = True, psi0 = None):
|
||||||
|
circ = qtn.Circuit(nqubits, psi0 = psi0)
|
||||||
|
# circ = qtn.Circuit.qasm(nqubits, psi0 = psi0)
|
||||||
|
gate_functions = {}
|
||||||
|
qasm_str = qasm_str.split('\n')
|
||||||
|
for idx, line in enumerate(qasm_str):
|
||||||
|
command = line.split(" ")[0]
|
||||||
|
if re.search("include|//|OPENQASM", command):
|
||||||
|
continue
|
||||||
|
elif "qreg" in command:
|
||||||
|
nbits = int(re.findall(r'\d+', line)[0])
|
||||||
|
assert(nbits == nqubits)
|
||||||
|
elif "swap" in command:
|
||||||
|
break
|
||||||
|
elif "gate" in command: # TODO: Complete gate handling
|
||||||
|
gate_name = line.split(" ")[1]
|
||||||
|
# gate_func = get_gate_functions(qasm_str, idx)
|
||||||
|
# gate_funtions[gate_name] = gate_func
|
||||||
|
pass
|
||||||
|
elif "barrier" in command: # TODO: Complete barrier handling
|
||||||
|
pass
|
||||||
|
elif "measure" in command: # TODO: Complete measure handling
|
||||||
|
pass
|
||||||
|
else:
|
||||||
|
params = get_gate_params(line)
|
||||||
|
circ.apply_gate(*params)
|
||||||
|
|
||||||
|
if with_swaps:
|
||||||
|
for i in range(nqubits // 2): # TODO: Ignore the barrier indices?
|
||||||
|
circ.apply_gate('SWAP', i, nqubits - i - 1)
|
||||||
|
|
||||||
|
return circ
|
||||||
|
|
||||||
|
|
||||||
|
def eval_QI_qft(nqubits, bond_dim=0, backend='numpy', qibo_backend='numpy',
|
||||||
|
with_swaps=True, compare_qibo=False):
|
||||||
|
# backend (quimb): numpy, cupy, jax. Passed to ``opt_einsum``.
|
||||||
|
# qibo_backend: qibojit, qibotf, tensorflow, numpy
|
||||||
|
|
||||||
|
# generate random statevector as initial state
|
||||||
|
init_state = np.random.random(2 ** nqubits) + 1j
|
||||||
|
* np.random.random(2 ** nqubits)
|
||||||
|
init_state = init_state / np.sqrt((np.abs(init_state)**2).sum())
|
||||||
|
|
||||||
|
# Qibo part
|
||||||
|
if compare_qibo==True:
|
||||||
|
import qibo
|
||||||
|
qibo.set_backend(qibo_backend)
|
||||||
|
# qibo.set_backend(backend="qibojit", platform="numba")
|
||||||
|
from qibo.models import QFT as qibo_qft
|
||||||
|
start = timer()
|
||||||
|
circ_qibo = qibo_qft(nqubits, with_swaps)
|
||||||
|
amplitudes_reference = np.array(circ_qibo(init_state))
|
||||||
|
end = timer()
|
||||||
|
print("qibo time is " + str(end-start))
|
||||||
|
qasm_circ = circ_qibo.to_qasm()
|
||||||
|
|
||||||
|
|
||||||
|
#####################################################################
|
||||||
|
|
||||||
|
# Quimb part
|
||||||
|
qtn.tensor_core.set_contract_backend(backend)
|
||||||
|
## convert vector to MPS
|
||||||
|
dims = tuple(2*np.ones(nqubits, dtype=int))
|
||||||
|
start = timer()
|
||||||
|
init_state_MPS =
|
||||||
|
qtn.tensor_1d.MatrixProductState.from_dense(init_state, dims)
|
||||||
|
end = timer()
|
||||||
|
MPS_time = end-start
|
||||||
|
# print('MPS conversion time: ', MPS_time)
|
||||||
|
|
||||||
|
# construct quimb qft circuit
|
||||||
|
start = timer()
|
||||||
|
if compare_qibo == True:
|
||||||
|
circ_quimb =
|
||||||
|
qasm_QFT(nqubits, qasm_circ, with_swaps, psi0=init_state_MPS)
|
||||||
|
else:
|
||||||
|
circ_quimb = QI_QFT(nqubits, with_swaps, psi0=init_state_MPS)
|
||||||
|
|
||||||
|
result = circ_quimb.to_dense(backend=backend)
|
||||||
|
amplitudes = result.flatten()
|
||||||
|
end = timer()
|
||||||
|
quimb_qft_time = end-start
|
||||||
|
print("quimb time is " + str(quimb_qft_time))
|
||||||
|
assert(np.allclose(amplitudes,amplitudes_reference))
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
print("Testing for %d nqubits" % (nqubits))
|
||||||
|
result = eval_QI_qft(nqubits, compare_qibo=True)
|
||||||
Reference in New Issue
Block a user