gudZddlmZddlZddlZddlmZmZddlm Z m Z m Z ddl m Z ddlmZmZmZejeZe ZGd d eZGd d eeZGd deeZdS)z:Baseclasses for the Qiskit passmanager optimization tasks.) annotationsN)abstractmethodABC)IterableCallable Generator)Any)RunStatePassManagerState PropertySetc.eZdZdZe d d d ZdS)TaskzAn interface of the pass manager task. The task takes a Qiskit IR, and outputs new Qiskit IR after some operation on it. A task can rely on the :class:`.PropertySet` to communicate intermediate data among tasks. Npassmanager_ir PassManagerIRstater callbackrreturn&tuple[PassManagerIR, PassManagerState]cdS)a|Execute optimization task for input Qiskit IR. Args: passmanager_ir: Qiskit IR to optimize. state: State associated with workflow execution by the pass manager itself. callback: A callback function which is caller per execution of optimization task. Returns: Optimized Qiskit IR and state of the workflow. N)selfrrrs S/granary/edev/quantum/lib/python3.11/site-packages/qiskit/passmanager/base_tasks.pyexecutez Task.execute%s " Nrrrr rrrr)__name__ __module__ __qualname____doc__rrrrrrrsH  "     ^   rrcLeZdZdZdZddZ dddZddZeddZ dS) GenericPasszBase class of a single pass manager task. A pass instance can read and write to the provided :class:`.PropertySet`, and may modify the input pass manager IR. c:t|_g|_dSr)r property_setrequiresrs r__init__zGenericPass.__init__@s'MM(* rrstrc|jjS)zName of the pass.) __class__rr's rnamezGenericPass.nameDs ~&&rNrrrr rrrc |j|_|jr/ddlm}||j|||\}}d}d}t j} ||jjvr"||}tj }n tj }n#t$rtj }wxYw|p|}|tj krht j|z }td||dz|||||j||jjn#|p|}|tj krit j|z }td||dz|!||||j||jjwwwxYw||||fS)Nr )FlowControllerLinearrrrzPass: %s - %.5f (ms)i)taskrr% running_timecount)r%r&flow_controllersr.rtimeworkflow_statuscompleted_passesrunr SUCCESSSKIP ExceptionFAILloggerinfor,r2 update_status) rrrrr. run_stateret start_timer1s rrzGenericPass.executeHs". =  > > > > > >$8$8$G$G$O$O-!%P%% !NE  Y[[  50AAAhh~..$, $M     I  'CHM))#y{{Z7  2DIIKKPTATUUU'H!'*%*%7%1#39  'CHM))#y{{Z7  2DIIKKPTATUUU'H!'*%*%7%1#39  *(D&&ui8888szGenericPass.update_statusxsS.7* ( ( (  ! ' '1 , ' '  ! 2 6 6t < < < rcdS)zRun optimization task. Args: passmanager_ir: Qiskit IR to optimize. Returns: Optimized Qiskit IR. Nr)rrs rr7zGenericPass.runs  r)rr)rr)rr r?r rr )rrrr) rrr r!r(r,rr>rr7rrrr#r#9s +++''''" .9.9.9.9.9`(    ^    rr#cBeZdZdZ dddZedd Z dddZdS)BaseControllera9Base class of controller. A controller is built with a collection of pass manager tasks, and a subclass provides a custom logic to choose next task to run. Note a controller can be nested into another controller, and a controller itself doesn't provide any subroutine to modify the input IR. Noptionsdict[str, Any] | Nonec|pi|_dS)ziCreate new flow controller. Args: options: Option for this flow controller. N)_options)rrHs rr(zBaseController.__init__s 2 rrr r'Generator[Task, PassManagerState, None]cdS)aVA custom logic to choose a next task to run. Controller subclass can consume the state to build a proper task pipeline. The updated state after a task execution will be fed back in as the "return" value of any ``yield`` statements. This indicates the order of task execution is only determined at running time. This method is not allowed to mutate the given state object. Args: state: The state of the passmanager workflow at the beginning of this flow controller's execution. Receives: state: the state of pass manager after the execution of the last task that was yielded. The generator does not need to inspect this if it is irrelevant to its logic, nor update it. Yields: Task: Next task to run. Nr)rrs r iter_taskszBaseController.iter_taskss 0 rrrrrrc||} |d}n#t$r||fcYSwxYw ||||\}} ||}n#t$rYnwxYwC||fS)NTr/)rNsend StopIterationr)rrrrtask_generator next_tasks rrzBaseController.executes// )&++D11II ) ) )!5( ( ( ( ) $-$5$5-!%6%% !NE  +//66       u$$s->>A44 BBr)rHrI)rr rrLr)rrr r!r(rrNrrrrrGrGsz*. & & & & &   ^ :" %%%%%%%rrG)r! __future__rloggingr4abcrrcollections.abcrrrtypingr compilation_statusr r r getLoggerrr<rrr#rGrrrr[sPA@"""""" ########9999999999GGGGGGGGGG  8 $ $      3   6` ` ` ` ` $` ` ` FJ%J%J%J%J%T3J%J%J%J%J%r