def test_floquet_to_calibration_layer_with_measure_qubits():
qubits = tuple(cirq.GridQubit(0, index) for index in range(5))
q_00, q_01, q_02, q_03, _ = qubits
gate = cirq.FSimGate(theta=np.pi / 4, phi=0.0)
request = FloquetPhasedFSimCalibrationRequest(
gate=gate,
pairs=((q_00, q_01), (q_02, q_03)),
options=FloquetPhasedFSimCalibrationOptions(
characterize_theta=True,
characterize_zeta=True,
characterize_chi=False,
characterize_gamma=False,
characterize_phi=True,
measure_qubits=qubits,
),
)
assert request.to_calibration_layer() == cirq_google.CalibrationLayer(
calibration_type='floquet_phased_fsim_characterization',
program=cirq.Circuit(
[gate.on(q_00, q_01),
gate.on(q_02, q_03),
cirq.measure(*qubits)], ),
args={
'est_theta': True,
'est_zeta': True,
'est_chi': False,
'est_gamma': False,
'est_phi': True,
'readout_corrections': True,
'version': 2,
},
)
def test_floquet_parse_result_bad_metric():
q_00, q_01, q_02, q_03 = [cirq.GridQubit(0, index) for index in range(4)]
gate = cirq.FSimGate(theta=np.pi / 4, phi=0.0)
request = FloquetPhasedFSimCalibrationRequest(
gate=gate,
pairs=((q_00, q_01), (q_02, q_03)),
options=FloquetPhasedFSimCalibrationOptions(
characterize_theta=True,
characterize_zeta=True,
characterize_chi=False,
characterize_gamma=False,
characterize_phi=True,
),
)
result = cirq_google.CalibrationResult(
code=cirq_google.api.v2.calibration_pb2.SUCCESS,
error_message=None,
token=None,
valid_until=None,
metrics=cirq_google.Calibration(
cirq_google.api.v2.metrics_pb2.MetricsSnapshot(metrics=[
cirq_google.api.v2.metrics_pb2.Metric(
name='angles',
targets=[
'1000gerbils',
],
values=[
cirq_google.api.v2.metrics_pb2.Value(str_val='100_10'),
],
)
])),
)
with pytest.raises(ValueError, match='Unknown metric name 1000gerbils'):
_ = request.parse_result(result)
def test_floquet_to_calibration_layer_readout_thresholds():
q_00, q_01, q_02, q_03 = [cirq.GridQubit(0, index) for index in range(4)]
gate = cirq.FSimGate(theta=np.pi / 4, phi=0.0)
request = FloquetPhasedFSimCalibrationRequest(
gate=gate,
pairs=((q_00, q_01), (q_02, q_03)),
options=FloquetPhasedFSimCalibrationOptions(
characterize_theta=True,
characterize_zeta=True,
characterize_chi=False,
characterize_gamma=False,
characterize_phi=True,
readout_error_tolerance=0.4,
),
)
assert request.to_calibration_layer() == cirq_google.CalibrationLayer(
calibration_type='floquet_phased_fsim_characterization',
program=cirq.Circuit([gate.on(q_00, q_01),
gate.on(q_02, q_03)]),
args={
'est_theta': True,
'est_zeta': True,
'est_chi': False,
'est_gamma': False,
'est_phi': True,
'readout_corrections': True,
'readout_error_tolerance': 0.4,
'correlated_readout_error_tolerance': 7 / 6 * 0.4 - 1 / 6,
'version': 2,
},
)
def test_run_characterization_fails_when_invalid_arguments():
with pytest.raises(ValueError):
assert workflow.run_calibrations([],
None,
'qproc',
cirq_google.FSIM_GATESET,
max_layers_per_request=0)
request = FloquetPhasedFSimCalibrationRequest(
gate=SQRT_ISWAP_GATE,
pairs=(),
options=WITHOUT_CHI_FLOQUET_PHASED_FSIM_CHARACTERIZATION,
)
engine = mock.MagicMock(spec=cirq_google.Engine)
with pytest.raises(ValueError):
assert workflow.run_calibrations([request], engine, None,
cirq_google.FSIM_GATESET)
with pytest.raises(ValueError):
assert workflow.run_calibrations([request], engine, 'qproc', None)
with pytest.raises(ValueError):
assert workflow.run_calibrations([request], 0, 'qproc',
cirq_google.FSIM_GATESET)
def _merge_into_calibrations(
calibration: FloquetPhasedFSimCalibrationRequest,
calibrations: List[FloquetPhasedFSimCalibrationRequest],
pairs_map: Dict[Tuple[Tuple[Qid, Qid], ...], int],
options: FloquetPhasedFSimCalibrationOptions,
) -> int:
"""Merges a calibration into list of calibrations.
If calibrations contains an item of which pairs could be expanded to include a new calibration
pairs, without breaking a moment structure, then those two calibrations will be merged together
and used as a calibration for both old and newly added calibration.
If no calibration like that exists, the list will be expanded by calibration item.
Args:
calibration: Calibration to be added.
calibrations: List of calibrations to be mutated.
pairs_map: Map from pairs parameter of each calibration on the calibrations list to the
index on that list. This map will be updated if the calibrations list us updated.
options: Calibrations options to use when creating a new requests.
Returns:
Index of the calibration on the updated calibrations list. If the calibration was added, it
points to the last element of a list. If not, it points to already existing element.
"""
new_pairs = set(calibration.pairs)
for index in pairs_map.values():
assert calibration.gate == calibrations[index].gate
assert calibration.options == calibrations[index].options
existing_pairs = calibrations[index].pairs
if new_pairs.issubset(existing_pairs):
return index
elif new_pairs.issuperset(existing_pairs):
calibrations[index] = calibration
return index
else:
new_qubit_pairs = calibration.qubit_to_pair
existing_qubit_pairs = calibrations[index].qubit_to_pair
if all((new_qubit_pairs[q] == existing_qubit_pairs[q]
for q in set(new_qubit_pairs.keys()).intersection(
existing_qubit_pairs.keys()))):
calibrations[index] = FloquetPhasedFSimCalibrationRequest(
gate=calibration.gate,
pairs=tuple(sorted(new_pairs.union(existing_pairs))),
options=options,
)
return index
index = len(calibrations)
calibrations.append(calibration)
pairs_map[calibration.pairs] = index
return index
def test_floquet_parse_result_failure():
gate = cirq.FSimGate(theta=np.pi / 4, phi=0.0)
request = FloquetPhasedFSimCalibrationRequest(
gate=gate,
pairs=(),
options=FloquetPhasedFSimCalibrationOptions(
characterize_theta=True,
characterize_zeta=True,
characterize_chi=False,
characterize_gamma=False,
characterize_phi=True,
),
)
result = cirq_google.CalibrationResult(
code=cirq_google.api.v2.calibration_pb2.ERROR_CALIBRATION_FAILED,
error_message="Test message",
token=None,
valid_until=None,
metrics=cirq_google.Calibration(),
)
with pytest.raises(PhasedFSimCalibrationError, match='Test message'):
request.parse_result(result)
def test_run_characterization_with_simulator():
q_00, q_01, q_02, q_03 = [cirq.GridQubit(0, index) for index in range(4)]
gate = SQRT_ISWAP_GATE
request = FloquetPhasedFSimCalibrationRequest(
gate=gate,
pairs=((q_00, q_01), (q_02, q_03)),
options=FloquetPhasedFSimCalibrationOptions(
characterize_theta=True,
characterize_zeta=True,
characterize_chi=False,
characterize_gamma=False,
characterize_phi=True,
),
)
simulator = PhasedFSimEngineSimulator.create_with_ideal_sqrt_iswap()
actual = workflow.run_calibrations([request], simulator)
assert actual == [
PhasedFSimCalibrationResult(
parameters={
(q_00, q_01):
PhasedFSimCharacterization(theta=np.pi / 4,
zeta=0.0,
chi=None,
gamma=None,
phi=0.0),
(q_02, q_03):
PhasedFSimCharacterization(theta=np.pi / 4,
zeta=0.0,
chi=None,
gamma=None,
phi=0.0),
},
gate=SQRT_ISWAP_GATE,
options=FloquetPhasedFSimCalibrationOptions(
characterize_theta=True,
characterize_zeta=True,
characterize_chi=False,
characterize_gamma=False,
characterize_phi=True,
),
)
]
def prepare_floquet_characterization_for_moment(
moment: Moment,
options: FloquetPhasedFSimCalibrationOptions,
gates_translator: Callable[
[Gate],
Optional[PhaseCalibratedFSimGate]] = try_convert_sqrt_iswap_to_fsim,
canonicalize_pairs: bool = False,
sort_pairs: bool = False,
) -> Optional[FloquetPhasedFSimCalibrationRequest]:
"""Describes a given moment in terms of a Floquet characterization request.
Args:
moment: Moment to characterize.
options: Options that are applied to each characterized gate within a moment.
gates_translator: Function that translates a gate to a supported FSimGate which will undergo
characterization. Defaults to sqrt_iswap_gates_translator.
canonicalize_pairs: Whether to sort each of the qubit pair so that the first qubit
is always lower than the second.
sort_pairs: Whether to sort all the qutibt pairs extracted from the moment which will
undergo characterization.
Returns:
Instance of FloquetPhasedFSimCalibrationRequest that characterizes a given moment, or None
when it is an empty, measurement or single-qubit gates only moment.
Raises:
IncompatibleMomentError when a moment contains operations other than the operations matched
by gates_translator, or it mixes a single qubit and two qubit gates.
"""
pairs_and_gate = _list_moment_pairs_to_characterize(
moment,
gates_translator,
canonicalize_pairs=canonicalize_pairs,
permit_mixed_moments=False)
if pairs_and_gate is None:
return None
pairs, gate = pairs_and_gate
return FloquetPhasedFSimCalibrationRequest(
pairs=tuple(sorted(pairs) if sort_pairs else pairs),
gate=gate,
options=options)
def test_from_moment():
q_00, q_01, q_02, q_03 = [cirq.GridQubit(0, index) for index in range(4)]
m = cirq.Moment(cirq.ISWAP(q_00, q_01)**0.5, cirq.ISWAP(q_02, q_03)**0.5)
options = FloquetPhasedFSimCalibrationOptions(
characterize_theta=True,
characterize_zeta=True,
characterize_chi=False,
characterize_gamma=False,
characterize_phi=True,
)
request = FloquetPhasedFSimCalibrationRequest.from_moment(m, options)
assert request == FloquetPhasedFSimCalibrationRequest(gate=cirq.ISWAP**0.5,
pairs=((q_00, q_01),
(q_02, q_03)),
options=options)
non_identical = cirq.Moment(
cirq.ISWAP(q_00, q_01)**0.5, cirq.ISWAP(q_02, q_03))
with pytest.raises(ValueError, match='must be identical'):
_ = FloquetPhasedFSimCalibrationRequest.from_moment(
non_identical, options)
sq = cirq.Moment(cirq.X(q_00))
with pytest.raises(ValueError, match='must be two qubit gates'):
_ = FloquetPhasedFSimCalibrationRequest.from_moment(sq, options)
threeq = cirq.Moment(cirq.TOFFOLI(q_00, q_01, q_02))
with pytest.raises(ValueError, match='must be two qubit gates'):
_ = FloquetPhasedFSimCalibrationRequest.from_moment(threeq, options)
not_gate = cirq.Moment(cirq.CircuitOperation(cirq.FrozenCircuit()))
with pytest.raises(ValueError, match='must be two qubit gates'):
_ = FloquetPhasedFSimCalibrationRequest.from_moment(not_gate, options)
empty = cirq.Moment()
with pytest.raises(ValueError, match='No gates found'):
_ = FloquetPhasedFSimCalibrationRequest.from_moment(empty, options)
def test_run_characterization():
q_00, q_01, q_02, q_03 = [cirq.GridQubit(0, index) for index in range(4)]
gate = cirq.FSimGate(theta=np.pi / 4, phi=0.0)
request = FloquetPhasedFSimCalibrationRequest(
gate=gate,
pairs=((q_00, q_01), (q_02, q_03)),
options=FloquetPhasedFSimCalibrationOptions(
characterize_theta=True,
characterize_zeta=True,
characterize_chi=False,
characterize_gamma=False,
characterize_phi=True,
),
)
result = cirq_google.CalibrationResult(
code=cirq_google.api.v2.calibration_pb2.SUCCESS,
error_message=None,
token=None,
valid_until=None,
metrics=cirq_google.Calibration(
cirq_google.api.v2.metrics_pb2.MetricsSnapshot(metrics=[
cirq_google.api.v2.metrics_pb2.Metric(
name='angles',
targets=[
'0_qubit_a',
'0_qubit_b',
'0_theta_est',
'0_zeta_est',
'0_phi_est',
'1_qubit_a',
'1_qubit_b',
'1_theta_est',
'1_zeta_est',
'1_phi_est',
],
values=[
cirq_google.api.v2.metrics_pb2.Value(str_val='0_0'),
cirq_google.api.v2.metrics_pb2.Value(str_val='0_1'),
cirq_google.api.v2.metrics_pb2.Value(double_val=0.1),
cirq_google.api.v2.metrics_pb2.Value(double_val=0.2),
cirq_google.api.v2.metrics_pb2.Value(double_val=0.3),
cirq_google.api.v2.metrics_pb2.Value(str_val='0_2'),
cirq_google.api.v2.metrics_pb2.Value(str_val='0_3'),
cirq_google.api.v2.metrics_pb2.Value(double_val=0.4),
cirq_google.api.v2.metrics_pb2.Value(double_val=0.5),
cirq_google.api.v2.metrics_pb2.Value(double_val=0.6),
],
)
])),
)
job = cirq_google.engine.EngineJob('', '', '', None)
job._calibration_results = [result]
engine = mock.MagicMock(spec=cirq_google.Engine)
engine.run_calibration.return_value = job
progress_calls = []
def progress(step: int, steps: int) -> None:
progress_calls.append((step, steps))
actual = workflow.run_calibrations([request],
engine,
'qproc',
cirq_google.FSIM_GATESET,
progress_func=progress)
expected = [
PhasedFSimCalibrationResult(
parameters={
(q_00, q_01):
PhasedFSimCharacterization(theta=0.1,
zeta=0.2,
chi=None,
gamma=None,
phi=0.3),
(q_02, q_03):
PhasedFSimCharacterization(theta=0.4,
zeta=0.5,
chi=None,
gamma=None,
phi=0.6),
},
gate=gate,
options=FloquetPhasedFSimCalibrationOptions(
characterize_theta=True,
characterize_zeta=True,
characterize_chi=False,
characterize_gamma=False,
characterize_phi=True,
),
)
]
assert actual == expected
assert progress_calls == [(1, 1)]
def test_floquet_parse_result():
q_00, q_01, q_02, q_03 = [cirq.GridQubit(0, index) for index in range(4)]
gate = cirq.FSimGate(theta=np.pi / 4, phi=0.0)
request = FloquetPhasedFSimCalibrationRequest(
gate=gate,
pairs=((q_00, q_01), (q_02, q_03)),
options=FloquetPhasedFSimCalibrationOptions(
characterize_theta=True,
characterize_zeta=True,
characterize_chi=False,
characterize_gamma=False,
characterize_phi=True,
),
)
result = cirq_google.CalibrationResult(
code=cirq_google.api.v2.calibration_pb2.SUCCESS,
error_message=None,
token=None,
valid_until=None,
metrics=cirq_google.Calibration(
cirq_google.api.v2.metrics_pb2.MetricsSnapshot(metrics=[
cirq_google.api.v2.metrics_pb2.Metric(
name='angles',
targets=[
'0_qubit_a',
'0_qubit_b',
'0_theta_est',
'0_zeta_est',
'0_phi_est',
'1_qubit_a',
'1_qubit_b',
'1_theta_est',
'1_zeta_est',
'1_phi_est',
],
values=[
cirq_google.api.v2.metrics_pb2.Value(str_val='0_0'),
cirq_google.api.v2.metrics_pb2.Value(str_val='0_1'),
cirq_google.api.v2.metrics_pb2.Value(double_val=0.1),
cirq_google.api.v2.metrics_pb2.Value(double_val=0.2),
cirq_google.api.v2.metrics_pb2.Value(double_val=0.3),
cirq_google.api.v2.metrics_pb2.Value(str_val='0_2'),
cirq_google.api.v2.metrics_pb2.Value(str_val='0_3'),
cirq_google.api.v2.metrics_pb2.Value(double_val=0.4),
cirq_google.api.v2.metrics_pb2.Value(double_val=0.5),
cirq_google.api.v2.metrics_pb2.Value(double_val=0.6),
],
)
])),
)
assert request.parse_result(result) == PhasedFSimCalibrationResult(
parameters={
(q_00, q_01):
PhasedFSimCharacterization(theta=0.1,
zeta=0.2,
chi=None,
gamma=None,
phi=0.3),
(q_02, q_03):
PhasedFSimCharacterization(theta=0.4,
zeta=0.5,
chi=None,
gamma=None,
phi=0.6),
},
gate=gate,
options=FloquetPhasedFSimCalibrationOptions(
characterize_theta=True,
characterize_zeta=True,
characterize_chi=False,
characterize_gamma=False,
characterize_phi=True,
),
)
def prepare_floquet_characterization_for_operations(
circuit: Union[Circuit, Iterable[Circuit]],
options:
FloquetPhasedFSimCalibrationOptions = WITHOUT_CHI_FLOQUET_PHASED_FSIM_CHARACTERIZATION,
gates_translator: Callable[
[Gate],
Optional[PhaseCalibratedFSimGate]] = try_convert_sqrt_iswap_to_fsim,
permit_mixed_moments: bool = False,
) -> List[FloquetPhasedFSimCalibrationRequest]:
"""Extracts a minimal set of Floquet characterization requests necessary to characterize all the
operations within a circuit(s).
This variant of prepare method works on two-qubit operations of the circuit. The method extracts
all the operations and groups them in a way to minimize the number of characterizations
requested, depending on the connectivity.
Contrary to prepare_floquet_characterization_for_moments, this method ignores moments structure
and is less accurate because certain errors caused by cross-talks are ignored.
The major advantage of this method is that the number of generated characterization requests is
bounded by four for grid-like devices, where for the
prepare_floquet_characterization_for_moments the number of characterizations is bounded by
number of moments in a circuit.
The circuit can only be composed of single qubit operations, wait operations, measurement
operations and operations supported by gates_translator.
Args:
circuit: Circuit or circuits to characterize. Only circuits with qubits of type GridQubit
that can be covered by HALF_GRID_STAGGERED_PATTERN are supported
options: Options that are applied to each characterized gate within a moment. Defaults
to all_except_for_chi_options which is the broadest currently supported choice.
gates_translator: Function that translates a gate to a supported FSimGate which will undergo
characterization. Defaults to sqrt_iswap_gates_translator.
permit_mixed_moments: Whether to allow a mix of two-qubit gates with other irrelevant
single-qubit gates.
Returns:
List of PhasedFSimCalibrationRequest for each group of operations to characterize.
Raises:
IncompatibleMomentError when circuit contains a moment with operations other than the
operations matched by gates_translator, or it mixes a single qubit and two qubit gates.
"""
circuits = [circuit] if isinstance(circuit, Circuit) else circuit
pairs, gate = _extract_all_pairs_to_characterize(circuits,
gates_translator,
permit_mixed_moments)
if gate is None:
return []
characterizations = []
for pattern in HALF_GRID_STAGGERED_PATTERN:
pattern_pairs = [pair for pair in pairs if pair in pattern]
if pattern_pairs:
characterizations.append(
FloquetPhasedFSimCalibrationRequest(pairs=tuple(
sorted(pattern_pairs)),
gate=gate,
options=options))
if sum((len(characterization.pairs)
for characterization in characterizations)) != len(pairs):
raise ValueError(
'Unable to cover all interactions with HALF_GRID_STAGGERED_PATTERN'
)
return characterizations