[Th&SSKrSSKJr SSKJr SSKJrJr \R"\ 5r /SQr S\S\4Sjr \R4S\S\4S jjrSS \S \\S \\4S jjrS\\\/\4S\\4SjrS\S\4SjrS\S\4Sjr"SS5rg)Nwrapsunwrap)CallableOptional) PassManagerinplace_wrapperlog_hook loop_pass this_before_that_pass_constraint"these_before_those_pass_constraintfnreturnc0^[T5U4Sj5nU$)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>T"U5 U$N)gmrs T/var/www/auris/envauris/lib/python3.13/site-packages/torch/fx/passes/pass_manager.py wrapped_fn#inplace_wrapper..wrapped_fn"s 2 r)rrs` rr r s# 2Y rc4^^[T5UU4Sj5nU$)a7 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>T"U5n[RTSTU5 U$)NzRan pass %s Return value: %s)loggerlog)rvalrlevels rrlog_hook..wrapped_fnGs$f 5:BD rr)rrrs`` rr r *s#: 2Y r base_passn_iter predicatecb^^^TSLTSL- (dS5e[T5UUU4Sj5nU$)a; 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>UnTb"TS:a[T5H nT"U5nM U$Tb&T"U5(aT"U5nT"U5(aMU$[STSTS35e)Nrz3loop_pass must be given positive int n_iter (given z) xor predicate (given ))range RuntimeError)sourceoutput_r!r"r#s rnew_passloop_pass..new_passds  &1*6]"6*#  "F##"6*F## E(1)A? rr)r!r"r#r,s``` rr r PsN $  BAB  9   Or constraintpassesc[U5HIup#[XS-S5H0upEU"X55(aM[SUSUSUSUSUSUS3 5e MK g) 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_constraintr7{sz&!fUWo.DA!>qc!aS 1#U1#YqcB /"rthisthatc4^^S[S[4UU4SjjnU$)zX Defines a partial order ('depends on' function) where `this` must occur before `that`. r4r6c&>UT:g=(d UT:g$rr)r4r6r9r8s r depends_on4this_before_that_pass_constraint..depends_onsDy%AI%rr)r8r9r<s`` rr r s# &h&8&& rthesethosec4^^S[S[4UU4SjjnU$)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] r4r6cJ>[U5T:g=(d [U5T:g$rr)r4r6r?r@s rr<6these_before_those_pass_constraint..depends_ons ayE!7VAY%%77rr>)r?r@r<s`` rrrs#.8h8888 rc\rSrSr%Sr\\\S'\\\S'Sr\ \S'SSjr \ S 5r S \4S jr S rS \\4SjrSrSrSrSrg)r a( 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 _validatedNcDU=(d /UlU=(d /Ulgr)r/rF)selfr/rFs r__init__PassManager.__init__s l &,"rc[U5nU$r)r )clsr/pms rbuild_from_passlistPassManager.build_from_passlists   r_passcHURRU5 SUlgNF)r/appendrG)rIrQs radd_passPassManager.add_passs 5!rcHURRU5 SUlgrS)rFrTrGrIr.s radd_constraintPassManager.add_constraints  +r_passescUcgURVs/sHo"RU;dMUPM nnX0lSUlgs snfrS)r/__name__rG)rIr[ps passes_lefts r remove_passPassManager.remove_passs= ? $(KKNKb;;g3MrK N! Os AAc/nURHAnURUR:XaURU5 M0URU5 MC X0lSUlgrS)r/r]rTrG)rI_target _replacementr_r^s r replace_passPassManager.replace_passsQ ++B{{g...""<0""2&  " rcUR(agURHn[XR5 M SUlg)z{ Validates that current pass schedule defined by `self.passes` is valid according to all constraints in `self.constraints` NT)rGrFr7r/rXs rvalidatePassManager.validates1 ?? **J .z;; G+rc`UR5 UnURH nU"U5nM U$r)rhr/)rIr)outrQs r__call__PassManager.__call__s, [[E*C! r)rGrFr/NN)r] __module__ __qualname____firstlineno____doc__listr__annotations__rGboolrJ classmethodrOrUrYstrr`rerhrl__static_attributes__rrrr r sx  NhJ-  h   49   rr rn)logging functoolsrinspectrtypingrr getLoggerr]r__all__r INFOr intr rursr7r rr rrrrs%   8 $ X(")##(#P!$($$ SM$!$V (H-t34 >B8n  8 8 hx:KKr