cirq-core 1.4.0.dev20240509215225__py3-none-any.whl → 1.4.0.dev20240509224851__py3-none-any.whl

cirq/__init__.py

@@ -336,6 +336,7 @@ from cirq.ops import (
 
 from cirq.transformers import (
     AbstractInitialMapper,
+    add_dynamical_decoupling,
     align_left,
     align_right,
     CompilationTargetGateset,

cirq/_version.py

@@ -1 +1 @@
-__version__ = "1.4.0.dev20240509215225"
+__version__ = "1.4.0.dev20240509224851"

cirq/transformers/__init__.py

@@ -78,6 +78,8 @@ from cirq.transformers.drop_empty_moments import drop_empty_moments
 
 from cirq.transformers.drop_negligible_operations import drop_negligible_operations
 
+from cirq.transformers.dynamical_decoupling import add_dynamical_decoupling
+
 from cirq.transformers.eject_z import eject_z
 
 from cirq.transformers.measurement_transformers import (

cirq/transformers/dynamical_decoupling.py

@@ -0,0 +1,122 @@
+# Copyright 2024 The Cirq Developers
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Transformer pass that adds dynamical decoupling operations to a circuit."""
+
+from functools import reduce
+from typing import Dict, Optional, Sequence, Tuple, Union
+
+from cirq.transformers import transformer_api
+import cirq
+import numpy as np
+
+
+def _repeat_sequence(
+    base_sequence: Sequence['cirq.Gate'], num_idle_moments: int
+) -> Sequence['cirq.Gate']:
+    """Returns the longest possible dynamical decoupling sequence."""
+    repeat_times = num_idle_moments // len(base_sequence)
+    return list(base_sequence) * repeat_times
+
+
+def _get_dd_sequence_from_schema_name(schema: str) -> Sequence['cirq.Gate']:
+    """Gets dynamical decoupling sequence from a schema name."""
+    dd_sequence: Sequence['cirq.Gate']
+    match schema:
+        case 'XX_PAIR':
+            dd_sequence = (cirq.X, cirq.X)
+        case 'X_XINV':
+            dd_sequence = (cirq.X, cirq.X**-1)
+        case 'YY_PAIR':
+            dd_sequence = (cirq.Y, cirq.Y)
+        case 'Y_YINV':
+            dd_sequence = (cirq.Y, cirq.Y**-1)
+        case _:
+            raise ValueError('Invalid schema name.')
+    return dd_sequence
+
+
+def _validate_dd_sequence(dd_sequence: Sequence['cirq.Gate']) -> None:
+    """Validates a given dynamical decoupling sequence.
+
+    Args:
+        dd_sequence: Input dynamical sequence to be validated.
+
+    Returns:
+        A tuple containing:
+            - is_valid (bool): True if the dd sequence is valid, False otherwise.
+            - error_message (str): An error message if the dd sequence is invalid, else None.
+
+    Raises:
+        ValueError: If dd_sequence is not valid.
+    """
+    if len(dd_sequence) < 2:
+        raise ValueError('Invalid dynamical decoupling sequence. Expect more than one gates.')
+    matrices = [cirq.unitary(gate) for gate in dd_sequence]
+    product = reduce(np.matmul, matrices)
+
+    if not cirq.equal_up_to_global_phase(product, np.eye(2)):
+        raise ValueError(
+            'Invalid dynamical decoupling sequence. Expect sequence production equals'
+            f' identity up to a global phase, got {product}.'.replace('\n', ' ')
+        )
+
+
+def _parse_dd_sequence(schema: Union[str, Sequence['cirq.Gate']]) -> Sequence['cirq.Gate']:
+    """Parses and returns dynamical decoupling sequence from schema."""
+    if isinstance(schema, str):
+        dd_sequence = _get_dd_sequence_from_schema_name(schema)
+    else:
+        _validate_dd_sequence(schema)
+        dd_sequence = schema
+    return dd_sequence
+
+
+@transformer_api.transformer
+def add_dynamical_decoupling(
+    circuit: 'cirq.AbstractCircuit',
+    *,
+    context: Optional['cirq.TransformerContext'] = None,
+    schema: Union[str, Sequence['cirq.Gate']] = 'X_XINV',
+) -> 'cirq.Circuit':
+    """Adds dynamical decoupling gate operations to idle moments of a given circuit.
+    This transformer preserves the moment structure of the circuit.
+
+    Args:
+          circuit: Input circuit to transform.
+          context: `cirq.TransformerContext` storing common configurable options for transformers.
+          schema: Dynamical decoupling schema name or a dynamical decoupling sequence.
+            If a schema is specified, provided dynamical decouping sequence will be used.
+            Otherwise, customized dynamical decoupling sequence will be applied.
+
+    Returns:
+          A copy of the input circuit with dynamical decoupling operations.
+    """
+    last_busy_moment_by_qubits: Dict['cirq.Qid', int] = {q: 0 for q in circuit.all_qubits()}
+    insert_into: list[Tuple[int, 'cirq.OP_TREE']] = []
+
+    base_dd_sequence = _parse_dd_sequence(schema)
+
+    for moment_id, moment in enumerate(circuit):
+        for q in moment.qubits:
+            insert_gates = _repeat_sequence(
+                base_dd_sequence, num_idle_moments=moment_id - last_busy_moment_by_qubits[q] - 1
+            )
+            for idx, gate in enumerate(insert_gates):
+                insert_into.append((last_busy_moment_by_qubits[q] + idx + 1, gate.on(q)))
+            last_busy_moment_by_qubits[q] = moment_id
+
+    updated_circuit = circuit.unfreeze(copy=True)
+    updated_circuit.batch_insert_into(insert_into)
+    return updated_circuit

cirq/transformers/dynamical_decoupling_test.py

@@ -0,0 +1,123 @@
+# Copyright 2024 The Cirq Developers
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from typing import Sequence, Union
+import cirq
+from cirq import add_dynamical_decoupling
+import pytest
+
+
+def assert_dd(
+    input_circuit: cirq.Circuit,
+    expected_circuit: cirq.Circuit,
+    schema: Union[str, Sequence['cirq.Gate']],
+):
+    updated_circuit = add_dynamical_decoupling(input_circuit, schema=schema)
+    cirq.testing.assert_same_circuits(updated_circuit, expected_circuit)
+
+
+def test_no_insert_due_to_no_consecutive_moments():
+    a = cirq.NamedQubit('a')
+    b = cirq.NamedQubit('b')
+
+    # No insertion as there is no room for a dd sequence.
+    assert_dd(
+        input_circuit=cirq.Circuit(
+            cirq.Moment(cirq.H(a)), cirq.Moment(cirq.CNOT(a, b)), cirq.Moment(cirq.H(b))
+        ),
+        expected_circuit=cirq.Circuit(
+            cirq.Moment(cirq.H(a)), cirq.Moment(cirq.CNOT(a, b)), cirq.Moment(cirq.H(b))
+        ),
+        schema='XX_PAIR',
+    )
+
+
+@pytest.mark.parametrize(
+    'schema,inserted_gates',
+    [
+        ('XX_PAIR', (cirq.X, cirq.X)),
+        ('X_XINV', (cirq.X, cirq.X**-1)),
+        ('YY_PAIR', (cirq.Y, cirq.Y)),
+        ('Y_YINV', (cirq.Y, cirq.Y**-1)),
+    ],
+)
+def test_insert_provided_schema(schema: str, inserted_gates: Sequence['cirq.Gate']):
+    a = cirq.NamedQubit('a')
+    b = cirq.NamedQubit('b')
+    c = cirq.NamedQubit('c')
+
+    input_circuit = cirq.Circuit(
+        cirq.Moment(cirq.H(a)),
+        cirq.Moment(cirq.CNOT(a, b)),
+        cirq.Moment(cirq.CNOT(b, c)),
+        cirq.Moment(cirq.CNOT(b, c)),
+        cirq.Moment(cirq.measure_each(a, b, c)),
+    )
+    expected_circuit = cirq.Circuit(
+        cirq.Moment(cirq.H(a)),
+        cirq.Moment(cirq.CNOT(a, b)),
+        cirq.Moment(cirq.CNOT(b, c), inserted_gates[0](a)),
+        cirq.Moment(cirq.CNOT(b, c), inserted_gates[1](a)),
+        cirq.Moment(cirq.measure_each(a, b, c)),
+    )
+
+    # Insert one dynamical decoupling sequence in idle moments.
+    assert_dd(input_circuit, expected_circuit, schema=schema)
+
+
+def test_insert_by_customized_dd_sequence():
+    a = cirq.NamedQubit('a')
+    b = cirq.NamedQubit('b')
+    c = cirq.NamedQubit('c')
+
+    assert_dd(
+        input_circuit=cirq.Circuit(
+            cirq.Moment(cirq.H(a)),
+            cirq.Moment(cirq.CNOT(a, b)),
+            cirq.Moment(cirq.CNOT(b, c)),
+            cirq.Moment(cirq.CNOT(b, c)),
+            cirq.Moment(cirq.CNOT(b, c)),
+            cirq.Moment(cirq.CNOT(b, c)),
+            cirq.Moment(cirq.measure_each(a, b, c)),
+        ),
+        expected_circuit=cirq.Circuit(
+            cirq.Moment(cirq.H(a)),
+            cirq.Moment(cirq.CNOT(a, b)),
+            cirq.Moment(cirq.CNOT(b, c), cirq.X(a)),
+            cirq.Moment(cirq.CNOT(b, c), cirq.X(a)),
+            cirq.Moment(cirq.CNOT(b, c), cirq.Y(a)),
+            cirq.Moment(cirq.CNOT(b, c), cirq.Y(a)),
+            cirq.Moment(cirq.measure_each(a, b, c)),
+        ),
+        schema=[cirq.X, cirq.X, cirq.Y, cirq.Y],
+    )
+
+
+@pytest.mark.parametrize(
+    'schema,error_msg_regex',
+    [
+        ('INVALID_SCHEMA', 'Invalid schema name.'),
+        ([cirq.X], 'Invalid dynamical decoupling sequence. Expect more than one gates.'),
+        (
+            [cirq.X, cirq.H],
+            'Invalid dynamical decoupling sequence. Expect sequence production equals identity'
+            ' up to a global phase, got',
+        ),
+    ],
+)
+def test_invalid_dd_schema(schema: Union[str, Sequence['cirq.Gate']], error_msg_regex):
+    a = cirq.NamedQubit('a')
+    input_circuit = cirq.Circuit(cirq.H(a))
+    with pytest.raises(ValueError, match=error_msg_regex):
+        add_dynamical_decoupling(input_circuit, schema=schema)

{cirq_core-1.4.0.dev20240509215225.dist-info → cirq_core-1.4.0.dev20240509224851.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cirq-core
-Version: 1.4.0.dev20240509215225
+Version: 1.4.0.dev20240509224851
 Summary: A framework for creating, editing, and invoking Noisy Intermediate Scale Quantum (NISQ) circuits.
 Home-page: http://github.com/quantumlib/cirq
 Author: The Cirq Developers

{cirq_core-1.4.0.dev20240509215225.dist-info → cirq_core-1.4.0.dev20240509224851.dist-info}/RECORD

@@ -1,10 +1,10 @@
-cirq/__init__.py,sha256=Pisx61VoUhqDdggQ31ch0Mx3uMwVtZ0q7Z69Ib4C92s,15736
+cirq/__init__.py,sha256=gQkxjgGyat-egE2jZlrLCVMofvpG_VA1021yh8Tv9j8,15766
 cirq/_compat.py,sha256=2tkJ50ID2PXzmMNuTfQrsZqqRSblTmfu3Y7g742gojs,29342
 cirq/_compat_test.py,sha256=Qq3ZcfgD-Nb81cEppQdJqhAyrVqXKtfXZYGXT0p-Wh0,34718
 cirq/_doc.py,sha256=yDyWUD_2JDS0gShfGRb-rdqRt9-WeL7DhkqX7np0Nko,2879
 cirq/_import.py,sha256=p9gMHJscbtDDkfHOaulvd3Aer0pwUF5AXpL89XR8dNw,8402
 cirq/_import_test.py,sha256=6K_v0riZJXOXUphHNkGA8MY-JcmGlezFaGmvrNhm3OQ,1015
-cirq/_version.py,sha256=XzUu-oSq5dDszma7gs7De3cNgBYMfb9_nx-2PuTDunI,40
+cirq/_version.py,sha256=Oe2oO7fLniWMge5QU8jj7I7z689VwMkUcRfBX4DFRrU,40
 cirq/_version_test.py,sha256=yYS6xm5-nuBRQJa9R3n41WdvFtVyY7Lb5Q8bea3VgBI,133
 cirq/conftest.py,sha256=X7yLFL8GLhg2CjPw0hp5e_dGASfvHx1-QT03aUbhKJw,1168
 cirq/json_resolver_cache.py,sha256=9g_JQMmfBzTuV-3s2flUbXIgcLjs4K7LjAFFgngdG1U,13204
@@ -1013,13 +1013,15 @@ cirq/testing/test_data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3
 cirq/testing/test_data/test_module_missing_json_test_data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cirq/testing/test_data/test_module_missing_testspec/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cirq/testing/test_data/test_module_missing_testspec/json_test_data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cirq/transformers/__init__.py,sha256=Omi-8NLq9lMbMHV-eXxwrpX5_zw-ZyRJKPmIL_mIzMg,4145
+cirq/transformers/__init__.py,sha256=-buJqTF2c6rUM18JDGSb5aMKWtyT24jE_mMrHo6DsdM,4222
 cirq/transformers/align.py,sha256=B4DuX84DWd4tLfMH21JviMZSOFYu7KCOMSEMALYsQpw,2727
 cirq/transformers/align_test.py,sha256=M4etT2cgESv1RdkKASguiGxEuqY7kmI1IswjSi-1jjY,7174
 cirq/transformers/drop_empty_moments.py,sha256=Rtn_BrpwkLXyZBdLzwdnsnEGWTdYuf1xOPakzbpv7-w,1517
 cirq/transformers/drop_empty_moments_test.py,sha256=G8pZmTfi8NG2NpGz_K3LZu5NQoqa-xPMCuZjwEu07xk,1907
 cirq/transformers/drop_negligible_operations.py,sha256=8eyOMy7bra2wJAjORbk6QjwHiLdL5SfwRaz8D2Dazbw,2083
 cirq/transformers/drop_negligible_operations_test.py,sha256=gqL6RoDPm6Zf4RxtprBenFyIsZQPUxmPur9oRl0Yr3U,3823
+cirq/transformers/dynamical_decoupling.py,sha256=iALBcjaO3uwcAla-smjapp2z9YezGecGFo_OGgJkD_8,4704
+cirq/transformers/dynamical_decoupling_test.py,sha256=hGUzTDV1UL3VBIueri91j3_ycxGmr20mGfQSnRAjegM,4149
 cirq/transformers/eject_phased_paulis.py,sha256=mTgRT5aw5_c9ccTkP4Np_4YTWnLzxsMKRO8pOQ7CtYM,13955
 cirq/transformers/eject_phased_paulis_test.py,sha256=-mXsfbi3V0ojC_YqoQM5otzdW4kjGusCx6F-kCv8M98,15834
 cirq/transformers/eject_z.py,sha256=0kOyvh6FDfrCrrTCVfpHKNc_kNC_pBdEKoXv11kuqGA,5803
@@ -1169,8 +1171,8 @@ cirq/work/sampler.py,sha256=JEAeQQRF3bqlO9AkOf4XbrTATDI5f5JgyM_FAUCNxao,19751
 cirq/work/sampler_test.py,sha256=B2ZsuqGT854gQtBIAh8k0LiG9Vj5wSzcGvkxOUoTcW4,13217
 cirq/work/zeros_sampler.py,sha256=x1C7cup66a43n-3tm8QjhiqJa07qcJW10FxNp9jJ59Q,2356
 cirq/work/zeros_sampler_test.py,sha256=JIkpBBFPJe5Ba4142vzogyWyboG1Q1ZAm0UVGgOoZn8,3279
-cirq_core-1.4.0.dev20240509215225.dist-info/LICENSE,sha256=tAkwu8-AdEyGxGoSvJ2gVmQdcicWw3j1ZZueVV74M-E,11357
-cirq_core-1.4.0.dev20240509215225.dist-info/METADATA,sha256=by0PSGyTPZ_yiM5_bg5AGvSYkHW_65ZFRb_hVjgTcHs,2000
-cirq_core-1.4.0.dev20240509215225.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-cirq_core-1.4.0.dev20240509215225.dist-info/top_level.txt,sha256=Sz9iOxHU0IEMLSFGwiwOCaN2e9K-jFbBbtpPN1hB73g,5
-cirq_core-1.4.0.dev20240509215225.dist-info/RECORD,,
+cirq_core-1.4.0.dev20240509224851.dist-info/LICENSE,sha256=tAkwu8-AdEyGxGoSvJ2gVmQdcicWw3j1ZZueVV74M-E,11357
+cirq_core-1.4.0.dev20240509224851.dist-info/METADATA,sha256=VWh4JaX499yyPSZoAlVl2F0SUMTk6zuqukkKI7O7D3g,2000
+cirq_core-1.4.0.dev20240509224851.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+cirq_core-1.4.0.dev20240509224851.dist-info/top_level.txt,sha256=Sz9iOxHU0IEMLSFGwiwOCaN2e9K-jFbBbtpPN1hB73g,5
+cirq_core-1.4.0.dev20240509224851.dist-info/RECORD,,