g+1dZddlmZddlZddlZddlmZmZddlm Z m Z ddl m Z ddl mZddlZddlmZmZd d lmZmZd d lmZd d lmZd d lmZmZmZejeZ e!Z"GddeZ#ddZ$ddZ%dS)zFManager for a set of Passes and their scheduling during transpilation.) annotationsN)ABCabstractmethod)CallableIterable)chain)Any) parallel_mapshould_run_in_parallel)Task PassManagerIR)PassManagerError)FlowControllerLinear) PropertySetWorkflowStatusPassManagerStateceZdZdZ d*d+dZd,d Zd-d Zd.dZdZdZ dZ dZ e d/dZ e d0dZ d1edd2d#Zd3d%Zd4d)ZdS)5BasePassManagerzPass manager base class.tasksTask | list[Task] max_iterationintczg|_||_t|_|r||dSdS)a Initialize an empty pass manager object. Args: tasks: 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. N)_tasksrr property_setappend)selfrrs T/granary/edev/quantum/lib/python3.11/site-packages/qiskit/passmanager/passmanager.py__init__zBasePassManager.__init__'sK *(MM   KK       returnNonect|tr|g}td|Drtd|j|dS)zAppend tasks to the schedule of passes. Args: tasks: A set of pass manager tasks to be added to schedule. Raises: TypeError: When any element of tasks is not a subclass of passmanager Task. c3BK|]}t|t VdSN) isinstancer ).0ts r! z)BasePassManager.append..Js/661:a&&&666666r#z6Added tasks are not all valid pass manager task types.N)r)r any TypeErrorrr)r rs r!rzBasePassManager.append<sh eT " " GE 66666 6 6 VTUU U 5!!!!!r#indexch ||j|<dS#t$r}td|d|d}~wwxYw)a~Replace a particular pass in the scheduler. Args: index: Task index to replace, based on the position in :meth:`tasks` tasks: A set of pass manager tasks to be added to schedule. Raises: TypeError: When any element of tasks is not a subclass of passmanager Task. PassManagerError: If the index is not found. Index to replace  does not existsNr IndexErrorr)r r/rexs r!replacezBasePassManager.replaceOsZ X!&DK    X X X"#Nu#N#N#NOOUW W Xs  1,1cd |j|=dS#t$r}td|d|d}~wwxYw)zRemoves a particular pass in the scheduler. Args: index: Pass index to remove, based on the position in :meth:`passes`. Raises: PassManagerError: If the index is not found. r1r2Nr3)r r/r5s r!removezBasePassManager.removecsX X E""" X X X"#Nu#N#N#NOOUW W Xs /*/c2|||dSr()r6)r r/items r! __setitem__zBasePassManager.__setitem__qs UD!!!!!r#c*t|jSr()lenr)r s r!__len__zBasePassManager.__len__ts4;r#c`||j}|j||_|S)Nr) __class__rr)r r/new_passmanagers r! __getitem__zBasePassManager.__getitem__ws-..t7I.JJ!%U!3r#c<||j}|j|_t||jr|xj|jz c_|S |||S#t $r%}t d|jd|j|d}~wwxYw)Nr@zunsupported operand type + for z and )rArrr)rrr.)r otherrBr5s r!__add__zBasePassManager.__add__|s..t7I.JJ!% eT^ , ,   " "el 2 " "" " &&u---&&#   \dn\\5?\\ sA,, B6 BB input_programr rc dS)zConvert input program into pass manager IR. Args: in_program: Input program. Returns: Pass manager IR. Nr)r rGkwargss r!_passmanager_frontendz%BasePassManager._passmanager_frontends  r#passmanager_ir in_programc dS)ajConvert pass manager IR into output program. Args: passmanager_ir: Pass manager IR after optimization. in_program: The input program, this can be used if you need any metadata about the original input for the output. It should not be mutated. Returns: Output program. Nr)r rKrLrIs r!_passmanager_backendz$BasePassManager._passmanager_backends $ r#N)r in_programsAny | list[Any]callbackr num_processesrobjectc |tur!tjdtd|d<jss|Sd}t |t s|g}d}t|dkst|s/fd |D}t|dkr |s|d S|Stt|d tj i| S) a Run all the passes on the specified ``in_programs``. Args: in_programs: Input programs to transform via all the registered passes. A single input object cannot be a Python builtin list object. A list object is considered as multiple input objects to optimize. callback: A callback function that will be called after each pass execution. The function will be called with 4 keyword arguments:: task (GenericPass): the pass being run passmanager_ir (Any): depending on pass manager subclass property_set (PropertySet): the property set running_time (float): the time to execute the pass count (int): the index for the pass execution 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): task = kwargs['task'] passmanager_ir = kwargs['passmanager_ir'] property_set = kwargs['property_set'] running_time = kwargs['running_time'] 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. property_set: Currently a key that will be interpreted as all other arbitrary ``kwargs``. In Qiskit 2.0, this will instead seed the :class:`.PropertySet` of the compilation (but does not do so in this version). kwargs: Arbitrary arguments passed to the compiler frontend and backend. Returns: The transformed program(s). zFrom Qiskit 2.0, 'property_set' will be a reserved keyword argument of 'BasePassManager.run', and not passed on to the conversion functions. This subclass of 'BasePassManager' needs to use a different keyword argument for passing on this data.) stacklevelrNTFr c 0g|]}td|dS))program pass_managerrQr) _run_workflow)r*rXrQrIr s r! z'BasePassManager.run..sA^gD8^^W]^^r#rpass_manager_bin)values task_kwargsrR) _MISSINGwarningswarn FutureWarningrr)listr=r r _run_workflow_in_new_processdilldumps)r rOrQrRrrIis_listouts` ` ` r!runzBasePassManager.runs;h x ' ' M-     &2F> "{ 6 h.> +t,, &-KG {  q (>}(M(M *C;1$$W$1v J    (+TZ-=-=>'     r#rcnt||j}t|S)zLinearize this manager into a single :class:`.FlowControllerLinear`, so that it can be nested inside another pass manager. Returns: A linearized pass manager. )rc_flatten_tasksrr)r flatten_taskss r!to_flow_controllerz"BasePassManager.to_flow_controllers/T00==>> #M222r#elementsIterable | Taskrcjt|ts|gStt|j|S)z;A helper method to recursively flatten a nested task chain.)r)rrmaprk)r rns r!rkzBasePassManager._flatten_taskss4(H-- : c$-x8899r#)rr)rrrr)rrr$r%)r/rrrr$r%)r/rr$r%)rGr r$r)rKrrLr r$r )NN) rOrPrQrrRrrrSr$r )r$r)rnror$r)__name__ __module__ __qualname____doc__r"rr6r8r;r>rCrFrrJrNr_rirmrkrr#r!rr$s[""$&!*""""&XXXX( X X X X"""          ^     ^ ,"! ^ ( ^ ^ ^ ^ ^ ^ @3333::::::r#rrXr rYr$c 4|}t}|jdd|i|}||t |t |dd\}}|j|_|jd||d|}|S)aRun single program optimization with a pass manager. Args: program: Arbitrary program to optimize. pass_manager: Pass manager with scheduled passes. **kwargs: Keyword arguments for IR conversion. Returns: Optimized program. rG)workflow_statusrrQN)rKstaterQ)rKrLr) rmrrJexecuterrgetrrN)rXrYrIflow_controllerinitial_statusrK final_state out_programs r!rZrZ s#5577O#%%N7\7 N#2"9"9%*$   J-- #:##NK!, 8L3,3% K r#r\bytescHt|tj|S)zRun single program optimization in new process. Args: program: Arbitrary program to optimize. pass_manager_bin: Binary of the pass manager with scheduled passes. Returns: Optimized program. )rXrY)rZreloads)rXr\s r!rdrdOs, Z 011   r#)rXr rYrr$r )rXr r\rr$r )&ru __future__rloggingr`abcrrcollections.abcrr itertoolsrtypingr reqiskit.utils.parallelr r base_tasksr r exceptionsrflow_controllersrcompilation_statusrrr getLoggerrrloggerrSr_rrZrdrr#r!rsML""""""########........ FFFFFFFF++++++++((((((222222MMMMMMMMMM  8 $ $ 688y:y:y:y:y:cy:y:y:x,,,,^r#