PhddlmZddlmZddlmZmZmZddlZeje Z gdZ dedefdZ ejfdedefd Zdd ed eed eefd ZdeeegefdeefdZdedefdZdedefdZGddZy)wrapsunwrap)CallableListOptionalN) PassManagerinplace_wrapperlog_hook loop_pass this_before_that_pass_constraint"these_before_those_pass_constraintfnreturnc.tfd}|S)z Convenience wrapper for passes which modify an object inplace. This wrapper makes them return the modified object instead. Args: fn (Callable[Object, Any]) Returns: wrapped_fn (Callable[Object, Object]) c|}|SN)gmvalrs gC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torch/fx/passes/pass_manager.py wrapped_fnz#inplace_wrapper..wrapped_fnsf r)rrs` rr r s# 2Y rc2tfd}|S)a Logs callable output. This is useful for logging output of passes. Note inplace_wrapper replaces the pass output with the modified object. If we want to log the original output, apply this wrapper before inplace_wrapper. ``` def my_pass(d: Dict) -> bool: changed = False if 'foo' in d: d['foo'] = 'bar' changed = True return changed pm = PassManager( passes=[ inplace_wrapper(log_hook(my_pass)) ] ) ``` Args: fn (Callable[Type1, Type2]) level: logging level (e.g. logging.INFO) Returns: wrapped_fn (Callable[Type1, Type2]) cH|}tjd||S)NzRan pass %s Return value: %s)loggerlog)rrrlevels rrzlog_hook..wrapped_fnEs$f 5:BD rr)rrrs`` rr r &s#> 2Y r base_passn_iter predicatecVduduz sJdtfd}|S)aW Convenience wrapper for passes which need to be applied multiple times. Exactly one of `n_iter`or `predicate` must be specified. Args: base_pass (Callable[Object, Object]): pass to be applied in loop n_iter (int, optional): number of times to loop pass predicate (Callable[Object, bool], optional): Nz8Exactly one of `n_iter`or `predicate` must be specified.c|}dkDrtD] }|} |S|r|}|r|Stddd)Nrz3loop_pass must be given positive int n_iter (given z) xor predicate (given ))range RuntimeError)sourceoutput_r r!r"s rnew_passzloop_pass..new_pass_s  &1*6]"6*#  "F#"6*F# E(1)A? rr)r r!r"r+s``` rr r OsN $  BAB  9   Or constraintpassesct|D]E\}}t||dzdD],\}}|||rtd|d|d|d|d|d|d Gy) Nz,pass schedule constraint violated. Expected z before z but found z at index z and z at indexz in pass list.) enumerater')r,r-iajbs r"_validate_pass_schedule_constraintr5vs&!1fQUWo.DAq!Q>qc!aS 1#U1#YqcB /"rthisthatc0dtdtffd }|S)zd Defines a partial order ('depends on' function) where `this` must occur before `that`. r2r4c|k(r|k(ryyNFTr)r2r4r7r6s r depends_onz4this_before_that_pass_constraint..depends_ons 9drr)r6r7r;s`` rrrs h8 rthesethosec0dtdtffd }|S)a_ Defines a partial order ('depends on' function) where `these` must occur before `those`. Where the inputs are 'unwrapped' before comparison. For example, the following pass list and constraint list would be invalid. ``` passes = [ loop_pass(pass_b, 3), loop_pass(pass_a, 5), ] constraints = [ these_before_those_pass_constraint(pass_a, pass_b) ] ``` Args: these (Callable): pass which should occur first those (Callable): pass which should occur later Returns: depends_on (Callable[[Object, Object], bool] r2r4c@t|k(rt|k(ryyr:r)r2r4r=r>s rr;z6these_before_those_pass_constraint..depends_ons !9 &)u"4rr<)r=r>r;s`` rrrs2h8 rceZdZUdZeeed<eeed<dZeed< ddZ e dZ d efd Z d Z d eefd ZdZdZy)r aT Construct a PassManager. Collects passes and constraints. This defines the pass schedule, manages pass constraints and pass execution. Args: passes (Optional[List[Callable]]): list of passes. A pass is a callable which modifies an object and returns modified object constraint (Optional[List[Callable]]): list of constraints. A constraint is a callable which takes two passes (A, B) and returns True if A depends on B and False otherwise. See implementation of `this_before_that_pass_constraint` for example. r- constraintsF _validatedNc0|xsg|_|xsg|_yr)r-rB)selfr-rBs r__init__zPassManager.__init__s l &,"rct|}|Sr)r )clsr-pms rbuild_from_passlistzPassManager.build_from_passlists   r_passcH|jj|d|_yNF)r-appendrC)rErKs radd_passzPassManager.add_passs 5!rcH|jj|d|_yrM)rBrNrCrEr,s radd_constraintzPassManager.add_constraints  +r_passesc|yg}|jD]"}|j|vs|j|$||_d|_yrM)r-__name__rNrC)rErS passes_leftpss r remove_passzPassManager.remove_passsG ?  ++B{{')""2&" rcz|jry|jD]}t||jd|_y)z Validates that current pass schedule defined by `self.passes` is valid according to all constraints in `self.constraints` NT)rCrBr5r-rQs rvalidatezPassManager.validates3 ?? **J .z4;; G+rc\|j|}|jD] }||} |Sr)rZr-)rEr(outrKs r__call__zPassManager.__call__s, [[E*C! rNN)rU __module__ __qualname____doc__rr__annotations__rCboolrF classmethodrJrOrRrXrZr]rrrr r ss  NhJ-  h   4>  rr r^) functoolsrinspectrtypingrrr logging getLoggerrUr__all__r INFOr intr rcr5rrr rrrrms++   8 $ X&")%%(%R  8C= HU]L^ N (H-t34 >B8n  8 8 hxBDDr