CompilationTargetGateset support in GridDeviceMetadata

Author: verult

cirq-core/cirq/devices/grid_device_metadata.py

@@ -33,6 +33,7 @@ def __init__(
         gateset: 'cirq.Gateset',
         gate_durations: Optional[Dict['cirq.GateFamily', 'cirq.Duration']] = None,
         all_qubits: Optional[Iterable['cirq.Qid']] = None,
+        compilation_target_gatesets: Iterable['cirq.CompilationTargetGateset'] = (),
     ):
         """Create a GridDeviceMetadata object.
 
@@ -53,6 +54,10 @@ def __init__(
             all_qubits: Optional iterable specifying all qubits
                 found on the device. If None, all_qubits will
                 be inferred from the entries in qubit_pairs.
+            compilation_target_gatesets: A collection of valid
+                `cirq.CompilationTargetGateset`s which can be used to
+                transform circuits into ones that consist of only
+                operations in `gateset`.
 
         Raises:
             ValueError: if some GateFamily keys in gate_durations are
@@ -95,6 +100,7 @@ def __init__(
         self._qubit_pairs = frozenset({frozenset(pair) for pair in edge_set})
         self._gateset = gateset
         self._isolated_qubits = all_qubits.difference(node_set)
+        self._compilation_target_gatesets = tuple(compilation_target_gatesets)
 
         if gate_durations is not None:
             working_gatefamilies = frozenset(gate_durations.keys())
@@ -126,6 +132,11 @@ def gateset(self) -> 'cirq.Gateset':
         """Returns the `cirq.Gateset` of supported gates on this device."""
         return self._gateset
 
+    @property
+    def compilation_target_gatesets(self) -> Tuple['cirq.CompilationTargetGateset', ...]:
+        """Returns a sequence of valid `cirq.CompilationTargetGateset`s for this device."""
+        return self._compilation_target_gatesets
+
     @property
     def gate_durations(self) -> Optional[Dict['cirq.GateFamily', 'cirq.Duration']]:
         """Get a dictionary mapping from gateset to duration for gates."""
@@ -141,14 +152,15 @@ def _value_equality_values_(self):
             self._gateset,
             tuple(duration_equality),
             tuple(sorted(self.qubit_set)),
+            frozenset(self._compilation_target_gatesets),
         )
 
     def __repr__(self) -> str:
         qubit_pair_tuples = frozenset({tuple(sorted(p)) for p in self._qubit_pairs})
         return (
             f'cirq.GridDeviceMetadata({repr(qubit_pair_tuples)},'
             f' {repr(self._gateset)}, {repr(self._gate_durations)},'
-            f' {repr(self.qubit_set)})'
+            f' {repr(self.qubit_set)}, {repr(self._compilation_target_gatesets)})'
         )
 
     def _json_dict_(self):
@@ -161,8 +173,23 @@ def _json_dict_(self):
             'gateset': self._gateset,
             'gate_durations': duration_payload,
             'all_qubits': sorted(list(self.qubit_set)),
+            'compilation_target_gatesets': list(self._compilation_target_gatesets),
         }
 
     @classmethod
-    def _from_json_dict_(cls, qubit_pairs, gateset, gate_durations, all_qubits, **kwargs):
-        return cls(qubit_pairs, gateset, dict(gate_durations), all_qubits)
+    def _from_json_dict_(
+        cls,
+        qubit_pairs,
+        gateset,
+        gate_durations,
+        all_qubits,
+        compilation_target_gatesets=(),
+        **kwargs,
+    ):
+        return cls(
+            qubit_pairs,
+            gateset,
+            None if gate_durations is None else dict(gate_durations),
+            all_qubits,
+            compilation_target_gatesets,
+        )

cirq-core/cirq/devices/grid_device_metadata_test.py

@@ -29,8 +29,13 @@ def test_griddevice_metadata():
         cirq.GateFamily(cirq.ZPowGate): 1_000,
         # omitting cirq.CZ
     }
+    target_gatesets = (cirq.CZTargetGateset(),)
     metadata = cirq.GridDeviceMetadata(
-        qubit_pairs, gateset, gate_durations=gate_durations, all_qubits=qubits + isolated_qubits
+        qubit_pairs,
+        gateset,
+        gate_durations=gate_durations,
+        all_qubits=qubits + isolated_qubits,
+        compilation_target_gatesets=target_gatesets,
     )
     expected_pairings = frozenset(
         {
@@ -53,6 +58,7 @@ def test_griddevice_metadata():
     assert metadata.nx_graph.nodes() == expected_graph.nodes()
     assert metadata.gate_durations == gate_durations
     assert metadata.isolated_qubits == frozenset(isolated_qubits)
+    assert metadata.compilation_target_gatesets == target_gatesets
 
 
 def test_griddevice_metadata_bad_durations():
@@ -88,35 +94,58 @@ def test_griddevice_self_loop():
 def test_griddevice_json_load():
     qubits = cirq.GridQubit.rect(2, 3)
     qubit_pairs = [(a, b) for a in qubits for b in qubits if a != b and a.is_adjacent(b)]
-    gateset = cirq.Gateset(cirq.XPowGate, cirq.YPowGate, cirq.ZPowGate)
+    gateset = cirq.Gateset(cirq.XPowGate, cirq.YPowGate, cirq.ZPowGate, cirq.CZ)
     duration = {
         cirq.GateFamily(cirq.XPowGate): cirq.Duration(nanos=1),
         cirq.GateFamily(cirq.YPowGate): cirq.Duration(picos=2),
         cirq.GateFamily(cirq.ZPowGate): cirq.Duration(picos=3),
+        cirq.GateFamily(cirq.CZ): cirq.Duration(nanos=4),
     }
     isolated_qubits = [cirq.GridQubit(9, 9), cirq.GridQubit(10, 10)]
+    target_gatesets = [cirq.CZTargetGateset()]
     metadata = cirq.GridDeviceMetadata(
-        qubit_pairs, gateset, gate_durations=duration, all_qubits=qubits + isolated_qubits
+        qubit_pairs,
+        gateset,
+        gate_durations=duration,
+        all_qubits=qubits + isolated_qubits,
+        compilation_target_gatesets=target_gatesets,
     )
     rep_str = cirq.to_json(metadata)
     assert metadata == cirq.read_json(json_text=rep_str)
 
 
+def test_griddevice_json_load_with_defaults():
+    qubits = cirq.GridQubit.rect(2, 3)
+    qubit_pairs = [(a, b) for a in qubits for b in qubits if a != b and a.is_adjacent(b)]
+    gateset = cirq.Gateset(cirq.XPowGate, cirq.YPowGate, cirq.ZPowGate, cirq.CZ)
+
+    # Don't set parameters with default values
+    metadata = cirq.GridDeviceMetadata(qubit_pairs, gateset)
+    rep_str = cirq.to_json(metadata)
+
+    assert metadata == cirq.read_json(json_text=rep_str)
+
+
 def test_griddevice_metadata_equality():
     qubits = cirq.GridQubit.rect(2, 3)
     qubit_pairs = [(a, b) for a in qubits for b in qubits if a != b and a.is_adjacent(b)]
-    gateset = cirq.Gateset(cirq.XPowGate, cirq.YPowGate, cirq.ZPowGate)
+    gateset = cirq.Gateset(cirq.XPowGate, cirq.YPowGate, cirq.ZPowGate, cirq.CZ, cirq.SQRT_ISWAP)
     duration = {
         cirq.GateFamily(cirq.XPowGate): cirq.Duration(nanos=1),
         cirq.GateFamily(cirq.YPowGate): cirq.Duration(picos=3),
         cirq.GateFamily(cirq.ZPowGate): cirq.Duration(picos=2),
+        cirq.GateFamily(cirq.CZ): cirq.Duration(nanos=4),
+        cirq.GateFamily(cirq.SQRT_ISWAP): cirq.Duration(nanos=5),
     }
     duration2 = {
         cirq.GateFamily(cirq.XPowGate): cirq.Duration(nanos=10),
         cirq.GateFamily(cirq.YPowGate): cirq.Duration(picos=13),
         cirq.GateFamily(cirq.ZPowGate): cirq.Duration(picos=12),
+        cirq.GateFamily(cirq.CZ): cirq.Duration(nanos=14),
+        cirq.GateFamily(cirq.SQRT_ISWAP): cirq.Duration(nanos=15),
     }
     isolated_qubits = [cirq.GridQubit(9, 9)]
+    target_gatesets = [cirq.CZTargetGateset(), cirq.SqrtIswapTargetGateset()]
     metadata = cirq.GridDeviceMetadata(qubit_pairs, gateset, gate_durations=duration)
     metadata2 = cirq.GridDeviceMetadata(qubit_pairs[:2], gateset, gate_durations=duration)
     metadata3 = cirq.GridDeviceMetadata(qubit_pairs, gateset, gate_durations=None)
@@ -125,28 +154,47 @@ def test_griddevice_metadata_equality():
     metadata6 = cirq.GridDeviceMetadata(
         qubit_pairs, gateset, gate_durations=duration, all_qubits=qubits + isolated_qubits
     )
+    metadata7 = cirq.GridDeviceMetadata(
+        qubit_pairs, gateset, compilation_target_gatesets=target_gatesets
+    )
+    metadata8 = cirq.GridDeviceMetadata(
+        qubit_pairs, gateset, compilation_target_gatesets=target_gatesets[::-1]
+    )
+    metadata9 = cirq.GridDeviceMetadata(
+        qubit_pairs, gateset, compilation_target_gatesets=tuple(target_gatesets)
+    )
+    metadata10 = cirq.GridDeviceMetadata(
+        qubit_pairs, gateset, compilation_target_gatesets=set(target_gatesets)
+    )
 
     eq = cirq.testing.EqualsTester()
     eq.add_equality_group(metadata)
     eq.add_equality_group(metadata2)
     eq.add_equality_group(metadata3)
     eq.add_equality_group(metadata4)
     eq.add_equality_group(metadata6)
+    eq.add_equality_group(metadata7, metadata8, metadata9, metadata10)
 
     assert metadata == metadata5
 
 
 def test_repr():
     qubits = cirq.GridQubit.rect(2, 3)
     qubit_pairs = [(a, b) for a in qubits for b in qubits if a != b and a.is_adjacent(b)]
-    gateset = cirq.Gateset(cirq.XPowGate, cirq.YPowGate, cirq.ZPowGate)
+    gateset = cirq.Gateset(cirq.XPowGate, cirq.YPowGate, cirq.ZPowGate, cirq.CZ)
     duration = {
         cirq.GateFamily(cirq.XPowGate): cirq.Duration(nanos=1),
         cirq.GateFamily(cirq.YPowGate): cirq.Duration(picos=3),
         cirq.GateFamily(cirq.ZPowGate): cirq.Duration(picos=2),
+        cirq.GateFamily(cirq.CZ): cirq.Duration(nanos=4),
     }
     isolated_qubits = [cirq.GridQubit(9, 9)]
+    target_gatesets = [cirq.CZTargetGateset()]
     metadata = cirq.GridDeviceMetadata(
-        qubit_pairs, gateset, gate_durations=duration, all_qubits=qubits + isolated_qubits
+        qubit_pairs,
+        gateset,
+        gate_durations=duration,
+        all_qubits=qubits + isolated_qubits,
+        compilation_target_gatesets=target_gatesets,
     )
     cirq.testing.assert_equivalent_repr(metadata)

cirq-core/cirq/protocols/json_test_data/GridDeviceMetadata.json

@@ -115,12 +115,76 @@
         "ignore_global_phase": true,
         "tags_to_accept": [],
         "tags_to_ignore": []
+      },
+      {
+        "cirq_type": "GateFamily",
+        "gate": {
+          "cirq_type": "CZPowGate",
+          "exponent": 1.0,
+          "global_shift": 0.0
+        },
+        "name": "Instance GateFamily: CZ",
+        "description": "Accepts `cirq.Gate` instances `g` s.t. `g == CZ`",
+        "ignore_global_phase": true,
+        "tags_to_accept": [],
+        "tags_to_ignore": []
+      },
+      {
+        "cirq_type": "GateFamily",
+        "gate": {
+          "cirq_type": "ISwapPowGate",
+          "exponent": 0.5,
+          "global_shift": 0.0
+        },
+        "name": "Instance GateFamily: ISWAP**0.5",
+        "description": "Accepts `cirq.Gate` instances `g` s.t. `g == ISWAP**0.5`",
+        "ignore_global_phase": true,
+        "tags_to_accept": [],
+        "tags_to_ignore": []
       }
     ],
     "name": null,
     "unroll_circuit_op": true
   },
   "gate_durations": [
+    [
+      {
+        "cirq_type": "GateFamily",
+        "gate": {
+          "cirq_type": "ISwapPowGate",
+          "exponent": 0.5,
+          "global_shift": 0.0
+        },
+        "name": "Instance GateFamily: ISWAP**0.5",
+        "description": "Accepts `cirq.Gate` instances `g` s.t. `g == ISWAP**0.5`",
+        "ignore_global_phase": true,
+        "tags_to_accept": [],
+        "tags_to_ignore": []
+      },
+      {
+        "cirq_type": "Duration",
+        "picos": 600
+      }
+    ],
+    [
+      {
+        "cirq_type": "GateFamily",
+        "gate": {
+          "cirq_type": "CZPowGate",
+          "exponent": 1.0,
+          "global_shift": 0.0
+        },
+        "name": "Instance GateFamily: CZ",
+        "description": "Accepts `cirq.Gate` instances `g` s.t. `g == CZ`",
+        "ignore_global_phase": true,
+        "tags_to_accept": [],
+        "tags_to_ignore": []
+      },
+      {
+        "cirq_type": "Duration",
+        "picos": 500
+      }
+    ],
     [
       {
         "cirq_type": "GateFamily",
@@ -208,5 +272,18 @@
       "row": 10,
       "col": 10
     }
+  ],
+  "compilation_target_gatesets": [
+    {
+      "cirq_type": "CZTargetGateset",
+      "atol": 1e-08,
+      "allow_partial_czs": false
+    },
+    {
+      "cirq_type": "SqrtIswapTargetGateset",
+      "atol": 1e-08,
+      "required_sqrt_iswap_count": null,
+      "use_sqrt_iswap_inv": false
+    }
   ]
 }