create_device_from_processor_id

cirq-google/cirq_google/engine/__init__.py

@@ -71,6 +71,7 @@
 from cirq_google.engine.validating_sampler import ValidatingSampler
 
 from cirq_google.engine.virtual_engine_factory import (
+    create_device_from_processor_id,
     create_noiseless_virtual_engine_from_device,
     create_noiseless_virtual_engine_from_proto,
     create_noiseless_virtual_engine_from_templates,

cirq-google/cirq_google/engine/virtual_engine_factory.py

@@ -206,6 +206,30 @@ def _create_device_spec_from_template(template_name: str) -> v2.device_pb2.Devic
     return device_spec
 
 
+def create_device_from_processor_id(
+    processor_id: str,
+    gate_sets: Optional[Iterable[serializable_gate_set.SerializableGateSet]] = None,
+) -> serializable_device.SerializableDevice:
+    """Generates a `cirq_google.SerializableDevice` for a given processor ID.
+
+    Args:
+         processor_id: name of the processor to simulate.
+         gate_sets: Iterable of serializers to use in the processor.  Defaults
+             to the FSIM_GATESET.
+
+    Raises:
+        ValueError: if processor_id is not a supported QCS processor.
+    """
+    if gate_sets is None:
+        gate_sets = [FSIM_GATESET]
+
+    template_name = MOST_RECENT_TEMPLATES.get(processor_id, None)
+    if template_name is None:
+        raise ValueError(f"Got processor_id={processor_id}, but no such processor is defined.")
+    device_specification = _create_device_spec_from_template(template_name)
+    return serializable_device.SerializableDevice.from_proto(device_specification, gate_sets)
+
+
 def create_noiseless_virtual_processor_from_template(
     processor_id: str,
     template_name: str,

cirq-google/cirq_google/engine/virtual_engine_factory_test.py

@@ -38,6 +38,14 @@ def _test_processor(processor: cg.engine.abstract_processor.AbstractProcessor):
         _ = processor.run(circuit, repetitions=100)
 
 
+def test_create_device_from_processor_id():
+    device = factory.create_device_from_processor_id('rainbow')
+    assert device is not None
+
+    with pytest.raises(ValueError, match='no such processor is defined'):
+        _ = factory.create_device_from_processor_id('bad_processor')
+
+
 def test_create_from_device():
     engine = factory.create_noiseless_virtual_engine_from_device('sycamore', cg.Sycamore)
     _test_processor(engine.get_processor('sycamore'))