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

Update fucntions to use **kwargs

Browse Source
This commit is contained in:
tankya2
2023-08-17 10:41:32 +08:00
parent c3809d36e6
commit cb21d1d1c0
1 changed files with 8 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/qibotn/MPSUtils.py
View File

@@ -19,7 +19,7 @@ def mps_site_right_swap(
Perform the swap operation between the ith and i+1th MPS tensors. Perform the swap operation between the ith and i+1th MPS tensors.
""" """
# contraction followed by QR decomposition # contraction followed by QR decomposition
a, _, b = contract_decompose('ipj,jqk->iqj,jpk', *mps_tensors[i:i+2], algorithm=list(kwargs.items())[0][1], options=list(kwargs.items())[1][1]) a, _, b = contract_decompose('ipj,jqk->iqj,jpk', *mps_tensors[i:i+2], algorithm=kwargs.get('algorithm',None), options=kwargs.get('options',None))
mps_tensors[i:i+2] = (a, b) mps_tensors[i:i+2] = (a, b)
return mps_tensors return mps_tensors
@@ -27,8 +27,7 @@ def apply_gate(
mps_tensors, mps_tensors,
gate, gate,
qubits, qubits,
algorithm=None, **kwargs
options=None
): ):
""" """
Apply the gate operand to the MPS tensors in-place. Apply the gate operand to the MPS tensors in-place.
@@ -53,24 +52,24 @@ def apply_gate(
if n_qubits == 1: if n_qubits == 1:
# single-qubit gate # single-qubit gate
i = qubits[0] i = qubits[0]
mps_tensors[i] = contract('ipj,qp->iqj', mps_tensors[i], gate, options=options) # in-place update mps_tensors[i] = contract('ipj,qp->iqj', mps_tensors[i], gate, options=kwargs.get('options',None)) # in-place update
elif n_qubits == 2: elif n_qubits == 2:
# two-qubit gate # two-qubit gate
i, j = qubits i, j = qubits
if i > j: if i > j:
# swap qubits order # swap qubits order
return apply_gate(mps_tensors, gate.transpose(1,0,3,2), (j, i), algorithm=algorithm, options=options) return apply_gate(mps_tensors, gate.transpose(1,0,3,2), (j, i), **kwargs)
elif i+1 == j: elif i+1 == j:
# two adjacent qubits # two adjacent qubits
a, _, b = contract_decompose('ipj,jqk,rspq->irj,jsk', *mps_tensors[i:i+2], gate, algorithm=algorithm, options=options) a, _, b = contract_decompose('ipj,jqk,rspq->irj,jsk', *mps_tensors[i:i+2], gate, algorithm=kwargs.get('algorithm',None), options=kwargs.get('options',None))
mps_tensors[i:i+2] = (a, b) # in-place update mps_tensors[i:i+2] = (a, b) # in-place update
else: else:
# non-adjacent two-qubit gate # non-adjacent two-qubit gate
# step 1: swap i with i+1 # step 1: swap i with i+1
mps_site_right_swap(mps_tensors, i, algorithm=algorithm, options=options) mps_site_right_swap(mps_tensors, i, **kwargs)
# step 2: apply gate to (i+1, j) pair. This amounts to a recursive swap until the two qubits are adjacent # step 2: apply gate to (i+1, j) pair. This amounts to a recursive swap until the two qubits are adjacent
apply_gate(mps_tensors, gate, (i+1, j), algorithm=algorithm, options=options) apply_gate(mps_tensors, gate, (i+1, j), **kwargs)
# step 3: swap back i and i+1 # step 3: swap back i and i+1
mps_site_right_swap(mps_tensors, i, algorithm=algorithm, options=options) mps_site_right_swap(mps_tensors, i, **kwargs)
else: else:
raise NotImplementedError("Only one- and two-qubit gates supported") raise NotImplementedError("Only one- and two-qubit gates supported")