gOdZddlmZddlZddlZddlZddlmZmZm Z ddl m Z ddl m Z mZmZmZddlmZddlmZmZdd lmZdd lmZdd lmZdd lmZdd lmZddl m!Z!ddl"m#Z#ddl$m%Z%m&Z&ede eeefZ'GddeZ(Gdde(Z)dZ*ej+e(ej,D]3\Z-Z.e-/dre*e.Z0e1e(e-e04ddZ2dS)FManager for a set of Passes and their scheduling during transpilation.) annotationsN)IteratorIterableCallablewraps)UnionListAnyTypeVar)QuantumCircuit)circuit_to_dagdag_to_circuit) DAGCircuit)BasePassManager)Task)FlowControllerLinear)PassManagerError)BasePass)TranspilerError)TranspileLayoutLayout _CircuitsT)boundcteZdZdZ d!d"fd Zd#d Zd$dZdZd%fd Zd&fd Z d'd(fd Z d)d Z xZ S)* PassManagerrpassesTask | list[Task] max_iterationintcNt||dS)aInitialize an empty pass manager object. Args: passes: A pass set to be added to the pass manager schedule. max_iteration: The maximum number of iterations the schedule will be looped if the condition is not met. )tasksr#N)super__init__)selfr!r# __class__s S/granary/edev/quantum/lib/python3.11/site-packages/qiskit/transpiler/passmanager.pyr(zPassManager.__init__(s6 '      input_programrreturnrc $t|dS)NTcopy_operations)r)r)r-kwargss r+_passmanager_frontendz!PassManager._passmanager_frontend9s mTBBBBr,passmanager_ir in_programc >t|d}|||dd}|||_|jdQt |jd|jd|jdt |j|j|_|jd|_ |jd |_ |jd rHg}|jd }| D]}| ||||_ |S) NFr0 output_namelayoutoriginal_qubit_indices final_layout)initial_layoutinput_qubit_mappingr:_input_qubit_count_output_qubit_listclbit_write_latencyconditional_latencynode_start_time)r_finalize_layoutsgetname property_setrlenqubits_layout_clbit_write_latency_conditional_latencytopological_op_nodesappend_op_start_times) r)r4r5r2 out_programout_nametopological_start_times start_timesdag_nodes r+_passmanager_backendz PassManager._passmanager_backend@s9 %^UKKK  ~...::mT22  'K   X & 2"1#0:$($56N$O!.~>#&z'8#9#9#.#5 ###K ,0+<=R+S (+/+<=R+S (  . / B') #+,=>K*??AA F F'..{8/DEEEE*AK 'r,c|jdx}dS|jd|jdd}|D]}||vr ||||<|j}|jddx})t t t |}|jddx})t t t |}d|D}g} tt|D]} | || | | | } | | |} | } | | | } | || } | ||} ||jd<| |jd<dS)Nvirtual_permutation_layoutr9r8r:ci|]\}}|| Srr).0kvs r+ z1PassManager._finalize_layouts..zs"S"S"SDAq1a"S"S"Sr,) rEpoprCrGrdict enumerateitemsrangerFrLinversecompose)r)dagrUr9oqt_qubitst_initial_layoutt_final_layoutoriginal_qubits_reverseoriginal_qubitsivirtual_permutation_layout_invt_initial_layout_invnew_final_layouts r+rBzPassManager._finalize_layoutsds*.*;+>&?&?@@ "/33NDII IN R#D8)<)<$=$=>>N#T"S4J4P4P4R4R"S"S"Ss23344 ? ?A  " "#:1#= > > > >)C)K)K _* * & 077RR0+334RTcdd+334DoVV+33NHMM&6(#,<.)))r,NonecLt|dS)zAppend a Pass Set to the schedule of passes. Args: passes: A set of transpiler passes to be added to schedule. Raises: TranspilerError: if a pass in passes is not a proper pass. r&N)r'rL)r)r!r*s r+rLzPassManager.appends# V$$$$$r,indexcNt||dS)zReplace a particular pass in the scheduler. Args: index: Pass index to replace, based on the position in passes(). passes: A pass set to be added to the pass manager schedule. roN)r'replace)r)rpr!r*s r+rrzPassManager.replaces% V,,,,,r,Ncircuitsrr7 str | Nonecallbackr num_processescp|t|}t||||S)aRun all the passes on the specified ``circuits``. Args: circuits: Circuit(s) to transform via all the registered passes. output_name: The output circuit name. If ``None``, it will be set to the same as the input circuit name. callback: A callback function that will be called after each pass execution. The function will be called with 5 keyword arguments:: pass_ (Pass): the pass being run dag (DAGCircuit): the dag output of the pass time (float): the time to execute the pass property_set (PropertySet): the property set count (int): the index for the pass execution .. note:: Beware that the keyword arguments here are different to those used by the generic :class:`.BasePassManager`. This pass manager will translate those arguments into the form described above. The exact arguments pass expose the internals of the pass manager and are subject to change as the pass manager internals change. If you intend to reuse a callback function over multiple releases be sure to check that the arguments being passed are the same. To use the callback feature you define a function that will take in kwargs dict and access the variables. For example:: def callback_func(**kwargs): pass_ = kwargs['pass_'] dag = kwargs['dag'] time = kwargs['time'] property_set = kwargs['property_set'] count = kwargs['count'] ... num_processes: The maximum number of parallel processes to launch if parallel execution is enabled. This argument overrides ``num_processes`` in the user configuration file, and the ``QISKIT_NUM_PROCS`` environment variable. If set to ``None`` the system default or local user configuration will be used. Returns: The transformed circuit(s). N) in_programsrur7rv)_legacy_style_callbackr'runr)rsr7rurvr*s r+rzzPassManager.runsDh  -h77Hww{{ #'    r,Fc,ddlm}|||||S)aDraw the pass manager. This function needs `pydot `__, which in turn needs `Graphviz `__ to be installed. Args: filename (str): file path to save image to. style (dict): keys are the pass classes and the values are the colors to make them. An example can be seen in the DEFAULT_STYLE. An ordered dict can be used to ensure a priority coloring when pass falls into multiple categories. Any values not included in the provided dict will be filled in from the default dict. raw (bool): If ``True``, save the raw Dot output instead of the image. Returns: Optional[PassManager]: an in-memory representation of the pass manager, or ``None`` if no image was generated or `Pillow `__ is not installed. Raises: ImportError: when nxpd or pydot not installed. r)pass_manager_drawerfilenamestyleraw)qiskit.visualizationr})r)rrrr}s r+drawzPassManager.draws0, =<<<<<""4(%SQQQQr,)rr )r!r"r#r$)r-rr.r)r4rr5rr.rr!r"r.rm)rpr$r!r"r.rmNNN) rsrr7rtrurrvr$r.rNNF) __name__ __module__ __qualname____doc__r(r3rSrBrLrrrzr __classcell__r*s@r+rr%sPP%'!       "CCCC""""H)=)=)=V % % % % % % - - - - - -"#'!! < < < < < < < |RRRRRRRRr,rceZdZdZejdZd-d.fd Zd/d Zd0d Z e d1dZ e d1dZ d2dZ d3dZfdZd4dZd5dZd6dZdZfdZdZd Z d7d8fd( Zd9fd* Zd:d,ZxZS);StagedPassManagerac A pass manager pipeline built from individual stages. This class enables building a compilation pipeline out of fixed stages. Each ``StagedPassManager`` defines a list of stages which are executed in a fixed order, and each stage is defined as a standalone :class:`~.PassManager` instance. There are also ``pre_`` and ``post_`` stages for each defined stage. This enables easily composing and replacing different stages and also adding hook points to enable programmatic modifications to a pipeline. When using a staged pass manager you are not able to modify the individual passes and are only able to modify stages. By default, instances of ``StagedPassManager`` define a typical full compilation pipeline from an abstract virtual circuit to one that is optimized and capable of running on the specified backend. The default pre-defined stages are: #. ``init`` - Initial passes to run before embedding the circuit to the backend. #. ``layout`` - Maps the virtual qubits in the circuit to the physical qubits on the backend. #. ``routing`` - Inserts gates as needed to move the qubit states around until the circuit can be run with the chosen layout on the backend's coupling map. #. ``translation`` - Translates the gates in the circuit to the target backend's basis gate set. #. ``optimization`` - Optimizes the circuit to reduce the cost of executing it. These passes will typically run in a loop until a convergence criteria is met. For example, the convergence criteria might be that the circuit depth does not decrease in successive iterations. #. ``scheduling`` - Hardware-aware passes that schedule the operations in the circuit. .. note:: For backwards compatibility the relative positioning of these default stages will remain stable moving forward. However, new stages may be added to the default stage list in between current stages. For example, in a future release a new phase, something like ``logical_optimization``, could be added immediately after the existing ``init`` stage in the default stage list. This would preserve compatibility for pre-existing ``StagedPassManager`` users as the relative positions of the stage are preserved so the behavior will not change between releases. These stages will be executed in order and any stage set to ``None`` will be skipped. If a stage is provided multiple times (i.e. at diferent relative positions), the associated passes, including pre and post, will run once per declaration. If a :class:`~qiskit.transpiler.PassManager` input is being used for more than 1 stage here (for example in the case of a :class:`~.Pass` that covers both Layout and Routing) you will want to set that to the earliest stage in sequence that it covers. z>\s|\+|\-|\*|\/|\\|\%|\<|\>|\@|\!|\~|\^|\&|\:|\[|\]|\{|\}|\(|\)NstagesIterable[str] | Noner.rmc |pgd}||tdt|tdt|t||t|jD])}| |d}t|||*dS)aInitialize a new StagedPassManager object Args: stages (Iterable[str]): An optional list of stages to use for this instance. If this is not specified the default stages list ``['init', 'layout', 'routing', 'translation', 'optimization', 'scheduling']`` is used. After instantiation, the final list will be immutable and stored as tuple. If a stage is provided multiple times (i.e. at diferent relative positions), the associated passes, including pre and post, will run once per declaration. kwargs: The initial :class:`~.PassManager` values for any stages defined in ``stages``. If a argument is not defined the stages will default to ``None`` indicating an empty/undefined stage. Raises: AttributeError: If a stage in the input keyword arguments is not defined. ValueError: If an invalid stage name is specified. )initr8routing translation optimization scheduling_stages_expanded_stagesN) _validate_stagesr' __setattr__tuple_generate_expanded_stagesr(_validate_init_kwargssetexpanded_stagesrCsetattr)r)rr2stagepmr*s r+r(zStagedPassManager.__init__9s&     f%%% IuV}}555 .d6T6T6V6V0W0WXXX  ""6***-.. % %EE4((B D% $ $ $ $ % %r, Iterable[str]c2fd|D}|rtj5}|d|d|ddD]}|d|t|#1swxYwYdSdS)NcHg|]}j||SN)invalid_stage_regexsearch)rWrr)s r+ z6StagedPassManager._validate_stages.._s6   )A)H)H)O)O)[E)[)[)[r,z)The following stage names are not valid: rrz, )ioStringIOwrite ValueErrorgetvalue)r)rinvalid_stagesmsg invalid_stages` r+rz"StagedPassManager._validate_stages^s    %     1 1# YnUVFWYYZZZ%3ABB%744MII2=223333 000  1 1 1 1 1 1 1 1 1 1 1 1sA%B  BBr2dict[str, Any]ct|j}|D]}||vrt|ddS)Nz is not a valid stage.)rrkeysAttributeError)r)r2rrs r+rz'StagedPassManager._validate_init_kwargsis^d233[[]] G GEO++$%E%E%EFFF, G Gr,tuple[str, ...]c|jS)zPass manager stages)rr)s r+rzStagedPassManager.stagesos |r,c|jS)zEExpanded Pass manager stages including ``pre_`` and ``post_`` phases.)rrs r+rz!StagedPassManager.expanded_stagests $$r, Iterator[str]c#BK|jD]}d|zV|Vd|zVdS)Npre_post_)r)r)rs r+rz+StagedPassManager._generate_expanded_stagesysM[ " "E5. KKKE/ ! ! ! ! " "r,cxg|_|jD]*}t||d}||xj|jz c_+dSr)_tasksrgetattr)r)rrs r+_update_passmanagerz%StagedPassManager._update_passmanagersN ) ) )Eud++B~ ry(  ) )r,c||kr||jvrtdt||||jvr|dSdS)Nz5Recursive definition of StagedPassManager disallowed.)rrr'rr)r)attrvaluer*s r+rzStagedPassManager.__setattr__sp D==TT%999!"YZZ Z D%((( 4' ' '  $ $ & & & & & ( 'r,r!r"ctrNotImplementedError)r)r!s r+rLzStagedPassManager.appends "!r,rpr$BasePass | list[BasePass]ctrr)r)rpr!s r+rrzStagedPassManager.replaces "!r,ctrr)r)rps r+removezStagedPassManager.remove!!r,c||t|j}|j||_|S)N)r#)rrr#r)r)rpnew_passmanagers r+ __getitem__zStagedPassManager.__getitem__s>   """&D4FGGG!%U!3r,cl|tSr)rr'__len__r)r*s r+rzStagedPassManager.__len__s)   """ww   r,ctrr)r)rpitems r+ __setitem__zStagedPassManager.__setitem__rr,ctrr)r)others r+__add__zStagedPassManager.__add__rr,rsrr7rtruCallable | Nonervcv|t||||S)N)rv)rr'rzr{s r+rzzStagedPassManager.runs5   """ww{{8[(-{XXXr,rcl|tSr)rr'to_flow_controllerrs r+rz$StagedPassManager.to_flow_controllers+   """ww))+++r,Fc,ddlm}|||||S)zDraw the staged pass manager.r)staged_pass_manager_drawerr~)rr)r)rrrrs r+rzStagedPassManager.draws/CCCCCC))$TWXXXXr,r)rrr.rm)rrr.rm)r2rr.rm)r.r)r.r)r.rmr)rpr$r!rr.rm)rpr$r.rmr) rsrr7rtrurrvr$r.r)r.rr)rrrrrecompilerr(rrpropertyrrrrrrLrrrrrrrrzrrrrs@r+rrs..`%"*I#%#%#%#%#%#%#%J 1 1 1 1GGGG X%%%X%"""" ))))'''''"""" """"""""!!!!!"""""" #'$(! YYYYYYY,,,,,,YYYYYYYYr,rc<tfd}|S)Ncd |i|S#t$r}t|j|d}~wwxYwr)rrmessage) meth_args meth_kwargsexmeths r+wrapperz_replace_error..wrappersN 642k22 2 6 6 6!"*--2 5 6s /*/r)rrs` r+_replace_errorrs3 4[[6666[6 Nr,) predicate_rurcfd}|S)Nc(|||||dS)N)pass_rbtimerEcountr)taskr4rE running_timerrus r+_wrapped_callablez1_legacy_style_callback.._wrapped_callables5%       r,r)rurs` r+ryrys$      r,)rur)3r __future__rinspectrrcollections.abcrrr functoolsr typingr r r r qiskit.circuitrqiskit.convertersrrqiskit.dagcircuitrqiskit.passmanager.passmanagerrqiskit.passmanager.base_tasksr#qiskit.passmanager.flow_controllersrqiskit.passmanager.exceptionsr basepassesr exceptionsrr8rrrrrr getmembers isfunction_name_method startswith_wrappedrryrr,r+r sML"""""" 8888888888,,,,,,,,,,,,))))))<<<<<<<<((((((::::::......DDDDDD:::::: ''''''++++++++ W\tN/C^/S)T U U U \R\R\R\R\R/\R\R\R~YYYYY YYYP)g(@RSSS**NE7  ~g&&H GK))))      r,