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

Use autopep8 for PEP-8 formatting

Browse Source
This commit is contained in:
Liwei Yang
2023-02-15 17:28:06 +08:00
parent bc13eed8f4
commit 53382b8184
2 changed files with 24 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/qibotn/qasm_quimb.py
View File

@@ -26,7 +26,8 @@ def get_gate_params(operation):
qbit_no.insert(0, "T") qbit_no.insert(0, "T")
elif "cu1" in operation: elif "cu1" in operation:
lambda_ = float( lambda_ = float(
".".join(re.findall(r"\b\d+(?:[Ee][+-]?\d+)?", operation.split(" ")[0])) ".".join(re.findall(
r"\b\d+(?:[Ee][+-]?\d+)?", operation.split(" ")[0]))
) )
qbit_no = re.findall(r"\d+", operation.split(" ")[1]) qbit_no = re.findall(r"\d+", operation.split(" ")[1])
qbit_no = [int(x) for x in qbit_no] qbit_no = [int(x) for x in qbit_no]
@@ -72,32 +73,37 @@ def get_gate_params(operation):
qbit_no.insert(0, "CCZ") qbit_no.insert(0, "CCZ")
elif " rx " in operation: elif " rx " in operation:
theta = float( theta = float(
".".join(re.findall(r"\b\d+(?:[Ee][+-]?\d+)?", operation.split(" ")[0])) ".".join(re.findall(
r"\b\d+(?:[Ee][+-]?\d+)?", operation.split(" ")[0]))
) )
qbit_no = [int(re.findall(r"\d+", operation)[0])] qbit_no = [int(re.findall(r"\d+", operation)[0])]
qbit_no[0:0] = ["RX", theta] qbit_no[0:0] = ["RX", theta]
elif "^ry " in operation: elif "^ry " in operation:
theta = float( theta = float(
".".join(re.findall(r"\b\d+(?:[Ee][+-]?\d+)?", operation.split(" ")[0])) ".".join(re.findall(
r"\b\d+(?:[Ee][+-]?\d+)?", operation.split(" ")[0]))
) )
qbit_no = [int(re.findall(r"\d+", operation)[0])] qbit_no = [int(re.findall(r"\d+", operation)[0])]
qbit_no[0:0] = ["RY", theta] qbit_no[0:0] = ["RY", theta]
elif "^rz " in operation: elif "^rz " in operation:
theta = float( theta = float(
".".join(re.findall(r"\b\d+(?:[Ee][+-]?\d+)?", operation.split(" ")[0])) ".".join(re.findall(
r"\b\d+(?:[Ee][+-]?\d+)?", operation.split(" ")[0]))
) )
qbit_no = [int(re.findall(r"\d+", operation)[0])] qbit_no = [int(re.findall(r"\d+", operation)[0])]
qbit_no[0:0] = ["RZ", theta] qbit_no[0:0] = ["RZ", theta]
elif "^rzz " in operation: elif "^rzz " in operation:
theta = float( theta = float(
".".join(re.findall(r"\b\d+(?:[Ee][+-]?\d+)?", operation.split(" ")[0])) ".".join(re.findall(
r"\b\d+(?:[Ee][+-]?\d+)?", operation.split(" ")[0]))
) )
qbit_no = re.findall(r"\d+", operation.split(" ")[1]) qbit_no = re.findall(r"\d+", operation.split(" ")[1])
qbit_no = [int(x) for x in qbit_no] qbit_no = [int(x) for x in qbit_no]
qbit_no[0:0] = ["RZZ", theta] qbit_no[0:0] = ["RZZ", theta]
elif "^u1 " in operation: elif "^u1 " in operation:
lambda_ = float( lambda_ = float(
".".join(re.findall(r"\b\d+(?:[Ee][+-]?\d+)?", operation.split(" ")[0])) ".".join(re.findall(
r"\b\d+(?:[Ee][+-]?\d+)?", operation.split(" ")[0]))
) )
qbit_no = [int(re.findall(r"\d+", operation)[0])] qbit_no = [int(re.findall(r"\d+", operation)[0])]
qbit_no[0:0] = ["U1", lambda_] qbit_no[0:0] = ["U1", lambda_]
@@ -182,7 +188,7 @@ def init_state_tn(nqubits, init_state_sv, tn_lib="quimb"):
def tn_circ_eval(nqubits, qasm_circ, init_state, swaps=True, tn_lib="quimb", def tn_circ_eval(nqubits, qasm_circ, init_state, swaps=True, tn_lib="quimb",
backend='numpy'): backend='numpy'):
if tn_lib == "quimb": if tn_lib == "quimb":
circ_quimb = qasm_QFT(nqubits, qasm_circ, swaps, psi0=init_state) circ_quimb = qasm_QFT(nqubits, qasm_circ, swaps, psi0=init_state)
@@ -202,4 +208,4 @@ def eval_QI_qft(nqubits, qasm_circ, init_state, backend="numpy", swaps=True):
amplitudes = tn_circ_eval(nqubits=nqubits, qasm_circ=qasm_circ, amplitudes = tn_circ_eval(nqubits=nqubits, qasm_circ=qasm_circ,
init_state=init_state_mps, swaps=swaps, init_state=init_state_mps, swaps=swaps,
tn_lib="quimb") tn_lib="quimb")
return amplitudes return amplitudes

18
tests/test_qasm_quimb_backend.py
View File

@@ -11,7 +11,8 @@ import config
def init_state_sv(nqubits): def init_state_sv(nqubits):
init_state = np.random.random(2**nqubits) + 1j * np.random.random(2**nqubits) 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()) init_state = init_state / np.sqrt((np.abs(init_state) ** 2).sum())
# An unmodified init_state has to be converted to tn format # An unmodified init_state has to be converted to tn format
init_state_for_tn = copy.deepcopy(init_state) init_state_for_tn = copy.deepcopy(init_state)
@@ -32,11 +33,11 @@ def test_eval(nqubits: int):
init_state_qibo, init_state_for_tn = init_state_sv(nqubits=nqubits) init_state_qibo, init_state_for_tn = init_state_sv(nqubits=nqubits)
# Test qibo # Test qibo
qibo.set_backend(backend=config.qibo['backend'], \ qibo.set_backend(backend=config.qibo['backend'],
platform=config.qibo['platform']) platform=config.qibo['platform'])
start_time = timer() start_time = timer()
qibo_circ, result_sv = qibo_qft(nqubits, init_state=init_state_qibo, \ qibo_circ, result_sv = qibo_qft(nqubits, init_state=init_state_qibo,
swaps=config.qibo['swaps']) swaps=config.qibo['swaps'])
end_time = timer() end_time = timer()
qibo_time = end_time - start_time qibo_time = end_time - start_time
@@ -45,9 +46,10 @@ def test_eval(nqubits: int):
# Test quimb # Test quimb
start_time = timer() start_time = timer()
result_tn = qasm_quimb.eval_QI_qft(nqubits=nqubits, qasm_circ=qasm_circ, \ result_tn = qasm_quimb.eval_QI_qft(nqubits=nqubits, qasm_circ=qasm_circ,
init_state=init_state_for_tn, backend=config.quimb['backend'], \ init_state=init_state_for_tn,
swaps=config.quimb['swaps']) backend=config.quimb['backend'],
swaps=config.quimb['swaps'])
end_time = timer() end_time = timer()
quimb_time = end_time - start_time quimb_time = end_time - start_time