Bug fix in ElidePermutations (#13186)

* elide permutations fix

* extending the test to check the final permutation
This commit is contained in:
Alexander Ivrii 2024-09-19 18:53:00 +03:00 committed by GitHub
parent 65bb09efab
commit adf55e1919
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 38 additions and 2 deletions

View File

@ -96,8 +96,8 @@ class ElidePermutations(TransformationPass):
elif isinstance(node.op, PermutationGate):
starting_indices = [qubit_mapping[dag.find_bit(qarg).index] for qarg in node.qargs]
pattern = node.op.params[0]
pattern_indices = [qubit_mapping[idx] for idx in pattern]
for i, j in zip(starting_indices, pattern_indices):
updated_indices = [starting_indices[idx] for idx in pattern]
for i, j in zip(starting_indices, updated_indices):
qubit_mapping[i] = j
input_qubit_mapping = {qubit: index for index, qubit in enumerate(dag.qubits)}
self.property_set["original_layout"] = Layout(input_qubit_mapping)

View File

@ -0,0 +1,5 @@
---
fixes:
- |
Fixed a bug in the transpiler pass :class:`~.ElidePermutations` where the
qubit mapping was not updated correctly in the presence of :class:`.PermutationGate`\s.

View File

@ -207,6 +207,37 @@ class TestElidePermutations(QiskitTestCase):
res = self.swap_pass(qc)
self.assertEqual(res, expected)
def test_partial_permutation_in_middle(self):
"""Test with a permutation gate in the middle of a circuit,
with the permutation gate defined only on a subset of qubits.
"""
qc = QuantumCircuit(5)
qc.cx(0, 1)
qc.append(PermutationGate([1, 2, 0]), [0, 2, 4])
qc.cx(2, 3)
# The permutation corresponding to the permutation gate maps
# 2 -> 0, 4 -> 2, 0 -> 4, and 1 -> 1, 3 -> 3.
# Instead of the permutation gate, we can relabel the qubits
# 0 -> 2, 1 -> 1, 2 -> 4, 3 -> 3, 4 -> 2.
# Hence cx(2, 3) becomes cx(4, 3).
expected = QuantumCircuit(5)
expected.cx(0, 1)
expected.cx(4, 3)
pass_ = ElidePermutations()
res = pass_(qc)
# Make sure that the pass removes the permutation gate and remaps
# the following gate
self.assertEqual(res, expected)
# Make sure that the transpiled circuit *with* the final permutation
# is equivalent to the original circuit
perm = pass_.property_set["virtual_permutation_layout"].to_permutation(qc.qubits)
res.append(PermutationGate(perm), [0, 1, 2, 3, 4])
self.assertEqual(Operator(res), Operator(qc))
def test_permutation_at_beginning(self):
"""Test permutation in beginning of bell is elided."""
qc = QuantumCircuit(3, 3)