/
base_transforms.py
71 lines (56 loc) · 2.34 KB
/
base_transforms.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
# This code is part of Qiskit.
#
# (C) Copyright IBM 2021.
#
# This code is licensed under the Apache License, Version 2.0. You may
# obtain a copy of this license in the LICENSE.txt file in the root directory
# of this source tree or at http://www.apache.org/licenses/LICENSE-2.0.
#
# Any modifications or derivative works of this code must retain this
# copyright notice, and modified files need to carry a notice indicating
# that they have been altered from the originals.
"""A collection of set of transforms."""
# TODO: replace this with proper pulse transformation passes. Qiskit-terra/#6121
from typing import Union, Iterable, Tuple
from qiskit.pulse.instructions import Instruction
from qiskit.pulse.schedule import ScheduleBlock, Schedule
from qiskit.pulse.transforms import canonicalization
InstructionSched = Union[Tuple[int, Instruction], Instruction]
def target_qobj_transform(
sched: Union[ScheduleBlock, Schedule, InstructionSched, Iterable[InstructionSched]],
remove_directives: bool = True,
) -> Schedule:
"""A basic pulse program transformation for OpenPulse API execution.
Args:
sched: Input program to transform.
remove_directives: Set `True` to remove compiler directives.
Returns:
Transformed program for execution.
"""
if not isinstance(sched, Schedule):
# convert into schedule representation
if isinstance(sched, ScheduleBlock):
sched = canonicalization.block_to_schedule(sched)
else:
sched = Schedule(*_format_schedule_component(sched))
# remove subroutines, i.e. Call instructions
sched = canonicalization.inline_subroutines(sched)
# inline nested schedules
sched = canonicalization.flatten(sched)
# remove directives, e.g. barriers
if remove_directives:
sched = canonicalization.remove_directives(sched)
return sched
def _format_schedule_component(sched: Union[InstructionSched, Iterable[InstructionSched]]):
"""A helper function to convert instructions into list of instructions."""
# TODO remove schedule initialization with *args, Qiskit-terra/#5093
try:
sched = list(sched)
# (t0, inst), or list of it
if isinstance(sched[0], int):
# (t0, inst) tuple
return [tuple(sched)]
else:
return sched
except TypeError:
return [sched]