Source code for quantify_scheduler.device_under_test.mock_setup

# Repository: https://gitlab.com/quantify-os/quantify-scheduler
# Licensed according to the LICENCE file on the main branch

"""Code to set up a mock setup for use in tutorials and testing."""

from typing import Any, Dict
import numpy as np

from quantify_core.measurement.control import MeasurementControl
from quantify_scheduler.device_under_test.quantum_device import QuantumDevice
from quantify_scheduler.device_under_test.transmon_element import BasicTransmonElement

from quantify_scheduler.device_under_test.nv_element import BasicElectronicNVElement
from quantify_scheduler.device_under_test.composite_square_edge import (
    CompositeSquareEdge,
)
from quantify_scheduler.instrument_coordinator import InstrumentCoordinator


[docs] def set_up_mock_transmon_setup() -> Dict: """ Set up a system containing 5 transmon qubits connected in a star shape. .. code-block:: q0 q1 \ / q2 / \ q3 q4 Returns a dictionary containing the instruments that are instantiated as part of this setup. The keys corresponds to the names of the instruments. """ # noqa: D301 meas_ctrl = MeasurementControl("meas_ctrl") instrument_coordinator = InstrumentCoordinator( name="instrument_coordinator", add_default_generic_icc=False ) q0 = BasicTransmonElement("q0") q1 = BasicTransmonElement("q1") q2 = BasicTransmonElement("q2") q3 = BasicTransmonElement("q3") q4 = BasicTransmonElement("q4") edge_q0_q2 = CompositeSquareEdge( parent_element_name=q0.name, child_element_name=q2.name ) edge_q1_q2 = CompositeSquareEdge( parent_element_name=q1.name, child_element_name=q2.name ) edge_q2_q3 = CompositeSquareEdge( parent_element_name=q2.name, child_element_name=q3.name ) edge_q2_q4 = CompositeSquareEdge( parent_element_name=q2.name, child_element_name=q4.name ) quantum_device = QuantumDevice(name="quantum_device") quantum_device.add_element(q0) quantum_device.add_element(q1) quantum_device.add_element(q2) quantum_device.add_element(q3) quantum_device.add_element(q4) quantum_device.add_edge(edge_q0_q2) quantum_device.add_edge(edge_q1_q2) quantum_device.add_edge(edge_q2_q3) quantum_device.add_edge(edge_q2_q4) quantum_device.instr_measurement_control(meas_ctrl.name) quantum_device.instr_instrument_coordinator(instrument_coordinator.name) # Rationale of the dict format is that all instruments can be cleaned up by # iterating over the values and calling close. It also avoids that the instruments # get garbage-collected while their name is still recorded and used to refer to the # instruments. return { "meas_ctrl": meas_ctrl, "instrument_coordinator": instrument_coordinator, "q0": q0, "q1": q1, "q2": q2, "q3": q3, "q4": q4, "q0_q2": edge_q0_q2, "q1_q2": edge_q1_q2, "q2_q3": edge_q2_q3, "q2_q4": edge_q2_q4, "quantum_device": quantum_device, }
[docs] def set_standard_params_transmon(mock_setup): """ Set somewhat standard parameters to the mock setup generated above. These parameters serve so that the quantum-device is capable of generating a configuration that can be used for compiling schedules. In normal use, unknown parameters are set as 'nan' values, forcing the user to set these. However for testing purposes it can be useful to set some semi-random values. The values here are chosen to reflect typical values as used in practical experiments. """ q0 = mock_setup["q0"] q0.rxy.amp180(0.115) q0.rxy.motzoi(0.1) q0.clock_freqs.f01(7.3e9) q0.clock_freqs.f12(7.0e9) q0.clock_freqs.readout(8.0e9) q0.measure.acq_delay(100e-9) q1 = mock_setup["q1"] q1.rxy.amp180(0.325) q1.rxy.motzoi(0.1) q1.clock_freqs.f01(7.25e9) q1.clock_freqs.f12(6.89e9) q1.clock_freqs.readout(8.3e9) q1.measure.acq_delay(100e-9) q2 = mock_setup["q2"] # controlled by a QCM-RF max output amp is 0.25V q2.rxy.amp180(0.213) q2.rxy.motzoi(0.1) q2.clock_freqs.f01(6.33e9) q2.clock_freqs.f12(6.09e9) q2.clock_freqs.readout(8.5e9) q2.measure.acq_delay(100e-9) q3 = mock_setup["q3"] q3.rxy.amp180(0.215) q3.rxy.motzoi(0.1) q3.clock_freqs.f01(5.71e9) q3.clock_freqs.f12(5.48e9) q3.clock_freqs.readout(8.7e9) q3.measure.acq_delay(100e-9) q4 = mock_setup["q4"] q4.rxy.amp180(0.208) q4.rxy.motzoi(0.1) q4.clock_freqs.f01(5.68e9) q4.clock_freqs.f12(5.41e9) q4.clock_freqs.readout(9.1e9) q4.measure.acq_delay(100e-9) for i in range(5): qi: BasicTransmonElement = mock_setup[f"q{i}"] sample_rate_MHz = 500 acq_duration_us = 2 qi.measure.acq_weights_a(np.ones(sample_rate_MHz * acq_duration_us) * 0.6) qi.measure.acq_weights_b(np.ones(sample_rate_MHz * acq_duration_us) * 0.4) qi.measure.acq_weights_sampling_rate(sample_rate_MHz * 1e6) qi.measure.acq_weight_type("Numerical")
[docs] def set_up_mock_basic_nv_setup() -> Dict: """ Set up a system containing 1 electronic qubit in an NV center. After usage, close all instruments. Returns ------- All instruments created. Containing a "quantum_device", electronic qubit "qe0", "meas_ctrl" and "instrument_coordinator". """ meas_ctrl = MeasurementControl("meas_ctrl") instrument_coordinator = InstrumentCoordinator( name="instrument_coordinator", add_default_generic_icc=False ) qe0 = BasicElectronicNVElement("qe0") qe1 = BasicElectronicNVElement("qe1") quantum_device = QuantumDevice(name="quantum_device") quantum_device.add_element(qe0) quantum_device.add_element(qe1) quantum_device.instr_measurement_control(meas_ctrl.name) quantum_device.instr_instrument_coordinator(instrument_coordinator.name) # Rationale of the dict format is that all instruments can be cleaned up by # iterating over the values and calling close. It also avoids that the instruments # get garbage-collected while their name is still recorded and used to refer to the # instruments. return { "meas_ctrl": meas_ctrl, "instrument_coordinator": instrument_coordinator, "qe0": qe0, "qe1": qe1, "quantum_device": quantum_device, }
[docs] def set_standard_params_basic_nv(mock_nv_device: Dict[str, Any]) -> None: """ Set somewhat standard parameters to the mock setup generated above. These parameters serve so that the quantum-device is capable of generating a configuration that can be used for compiling schedules. In normal use, unknown parameters are set as 'nan' values, forcing the user to set these. However for testing purposes it can be useful to set some semi-random values. The values here are chosen to reflect typical values as used in practical experiments. All amplitudes for pulses are set to 1e-3. """ quantum_device = mock_nv_device["quantum_device"] qe0: BasicElectronicNVElement = quantum_device.get_element("qe0") qe0.clock_freqs.f01.set(3.592e9) qe0.clock_freqs.spec.set(2.2e9) qe0.clock_freqs.ionization.set(564e12) qe0.clock_freqs.ge0.set(470.4e12) qe0.clock_freqs.ge1.set(470.4e12 - 5e9) qe0.charge_reset.amplitude(1e-3) qe0.cr_count.readout_pulse_amplitude(1e-3) qe0.cr_count.spinpump_pulse_amplitude(1e-3) qe0.reset.amplitude(1e-3) qe0.measure.pulse_amplitude(1e-3) qe0.spectroscopy_operation.amplitude.set(1e-3) qe1: BasicElectronicNVElement = quantum_device.get_element("qe1") qe1.clock_freqs.f01.set(4.874e9) qe1.clock_freqs.spec.set(1.4e9) qe1.clock_freqs.ionization.set(420e12) qe1.clock_freqs.ge0.set(470.4e12) qe1.clock_freqs.ge1.set(470.4e12 - 5e9) qe1.charge_reset.amplitude(1e-3) qe1.cr_count.readout_pulse_amplitude(1e-3) qe1.cr_count.spinpump_pulse_amplitude(1e-3) qe1.reset.amplitude(1e-3) qe1.measure.pulse_amplitude(1e-3) qe1.spectroscopy_operation.amplitude.set(1e-3)