+ :i@rRt^RIt^RIt^RIt^RIHt^RIHt^RIHt^RI H t ^RI H t H t Ht^RIHtHt^RIHt]P(!]4tR t!R R 4tR tR tRtRtRtRtRtRt Rt!Rt"Rt#!RR4t$!RR]$4t%!RR4t&!RR]$4t'!RR ]$4t(R!t)R#)"a Implement Dominance-Fronter-based SSA by Choi et al described in Inria SSA book References: - Static Single Assignment Book by Inria http://ssabook.gforge.inria.fr/latest/book.pdf - Choi et al. Incremental computation of static single assignment form. N)reduce)copy) defaultdict)config)irir_utilserrors) OrderedSet _lazy_pformat)compute_cfg_from_blocksc:\VP4VnV#)zfApply SSA reconstruction algorithm on the given IR. Produces minimal SSA using Choi et al algorithm. )_run_ssablocks)func_irs&L/var/www/html/photoedit/myenv/lib/python3.14/site-packages/numba/core/ssa.pyreconstruct_ssars gnn-GN Nc,a]tRt^$toRtRtRtVtR#)_CacheListVarsc/VnR#N_savedselfs&r__init___CacheListVars.__init__%s  rcVPPV4pVf VP4;VPV&pV#r)rget list_vars)rinstgots&& rr_CacheListVars.get(s8kkood# ;&*nn&6 6DKK  rrN)__name__ __module__ __qualname____firstlineno__rr__static_attributes____classdictcell__ __classdict__s@rrr$srrc pV'g/#\V4p\V4p\W4p\4pVFSp\P RV4\ W4wr\P R\V44\WWaVV4pKU \V4pWq8wd\P!R4hV#)z7Run SSA reconstruction on IR blocks of a function. zFix SSA violator on var %szReplaced assignments: %szCFG mutated in SSA pass) r _iterated_domfronts_find_defs_violatorsr_loggerdebug _fresh_varsr _fix_ssa_varsr CompilerError)rcfgdf_plus violatorscache_list_varsvarnamedefmapcfg_posts& rr r /s  !& )C!#&G$V1I$&O (' %V5 0-2GHvW.0'v.H""#<== Mrc\V4pWR&W&R&\\4;VR&pW6R&\WB4VR&\ W\ V44pVP 4F$wrW,p WP,V nK& V#)z=Rewrite all uses to ``varname`` given the definition map r7r8phimapr3 phi_locations) _make_statesrlist_compute_phi_locations_run_block_rewrite _FixSSAVarsitemsbody) rr7r8r3r4r6statesr; newblockslabelphilistcurblks &&&&&& rr1r1Ss& !F98 +D 11F8v5M4WEF?"6;3OPI ,,.! + ) rc VP4P4UUu/uFwrV\V4bK pppRpV'dzRpVP4Fawr\\P VUu.uF qSV,NK up\44pVP V4'gKWW&,pRpKc KV#uuppiuupi)zCompute the iterated dominance frontiers (DF+ in literatures). Returns a dictionary which maps block label to the set of labels of its iterated dominance frontiers. TF)dominance_frontierrBsetroperatoror_ difference)r3kvs domfronts keep_goingvinners& rr,r,es *-)?)?)A)G)G)IJ)ICG)IIJJ  __&EA8<<)C1A,,)CSUKE## ! ' K *Ds C5C c~\4pVP4Fwr4V'gKW V,,pK V#r)rKrB) iterated_dfr8r<deflabeldefstmtss&& rr?r?ws8EM$lln 8 2 2M- rcv\V4pWR&\\4;VR&p\W\ 44pWC3#)z(Rewrite to put fresh variable names r7r8)r=rr>r@_FreshVarHandler)rr7rDr8rEs&& rr0r0sC& !F9 +D 11F8v"63C3EFI  rc<VP4vrVP#r)valuesscope)rfirst_s& r _get_scoper`s IE ;;rc \\4p\\4p\W#R7p\ W\ 44\ PR\V44\VP4UUu.uFwrV\V4^8gKVNK upp4pVP4pVP4FdwrYWW9gK V FQp W,p W%,U U u0uFwrV kK p p p V PV 4'dK?VPV4Kb Kf \ PR\V44V#uuppiuup p i)zY Returns ------- res : Set[str] The SSA violators in a dictionary of variable names. )defsuseszdefs %szSSA violators %s)rr>rKdict_run_block_analysis_GatherDefsHandlerr.r/r r rBlen dominators intersectionadd)rr3rbrcrDrOrPr5doms use_blocksrFdom_assign def_labelss&& rr-r-s t D s D t 'F(:(<= MM)]4014::<G<%!3r7Q;AA<GHI >> D  #k:>'C'e' C!..s33MM!$ $& MM$mI&>? HDs6E E " E cVP4F2wr4\PRV4W1R&\WV4FpK K4 R#)z"==== SSA block analysis pass on %srFN)rBr.r/_run_ssa_block_pass)rrDhandlerrFblkr_s&&& rreres>lln  :EBw$V':A ;%rc</pVP4FwrE\PRV4\P!VP VP R7p.pWAR&WQR&\WV4FpVfQhVPV4K Wvn WcV&K V#)z!==== SSA block rewrite pass on %s)r]locrFblock) rBr.r/rBlockr]rurqappendrC) rrDrrrErFrsnewblknewbodystmts &&& rr@r@sIlln  95A sww7ww'W=D# ## NN4 > !%% rc,\\V4R7#))r])rdr`)rs&rr=r=s   rc#X"\PRV4VPFp\PRV4\V\P 4'dVP W4pMVPW4pWCJdVe\PRV4VxK R#5i)z Running %sz on stmt: %sNzreplaced with: %s)r.r/rC isinstancerAssign on_assignon_other)rDrsrrr{rets&&& rrqrqs} MM,( mT* dBII & &##F1C""60C ?s MM-s 3 sB(B*c0a]tRt^toRtRtRtRtVtR#) _BaseHandlerzGA base handler for all the passes used here for the SSA algorithm. c R#)a2 Called when the pass sees an ``ir.Assign``. Subclasses should override this for custom behavior Parameters ----------- states : dict assign : numba.ir.Assign Returns ------- stmt : numba.ir.Assign or None For rewrite passes, the return value is used as the replacement for the given statement. N)rrDassigns&&&rr_BaseHandler.on_assignrc R#)aE Called when the pass sees an ``ir.Stmt`` that's not an assignment. Subclasses should override this for custom behavior Parameters ----------- states : dict assign : numba.ir.Stmt Returns ------- stmt : numba.ir.Stmt or None For rewrite passes, the return value is used as the replacement for the given statement. NrrrDr{s&&&rr_BaseHandler.on_otherrrrN r#r$r%r&__doc__rrr'r(r)s@rrrs $  rrc0a]tRtRtoRtRtRtRtVtR#)rfiaBFind all defs and uses of variable in each block ``states["label"]`` is a int; label of the current block ``states["defs"]`` is a Mapping[str, List[Tuple[ir.Assign, int]]]: - a mapping of the name of the assignee variable to the assignment IR node and the block label. ``states["uses"]`` is a Mapping[Set[int]] cFVR,VPP,PW!R,34VP4FQpVPpWBPP8wgK+VR,V,P VR,4KS R#)rbrFrcN)targetnamerxrrj)rrDrvarrOs&&& rr_GatherDefsHandler.on_assign sqvv}}))*116'?2KL##%CAMM&&&vq!%%fWo6&rcVP4F5pVPpVR,V,PVR,4K7 R#)rcrFN)rrrj)rrDr{rrOs&&& rr_GatherDefsHandler.on_others7>>#CA 6N1  ! !&/ 2$rrNrr)s@rrfrfs733rrfc>a]tRtRtoRt]P tRtVt R#)UndefinedVariableic\R4h)zNot intended for instantiation)NotImplementedErrorrs&rrUndefinedVariable.__init__s!"BCCrrN) r#r$r%r&rr UNDEFINEDrr'r(r)s@rrrsD\\Frrc0a]tRtRtoRtRtRtRtVtR#)rZi#z9Replaces assignment target with new fresh variables. cVPPVR,8XEd$VR,pVR,p\V4^8XdVPp\P RV4VPVP 9dHRVP: R2p\ P!\P!VVPR74M1VPVPPVPR7p\P!VVPVPR7pWAR,,PV4V#) r7r]r8zfirst assign: %sz variable z is not in scope.rurvaluerurF)rrrgr.r/ localvarswarningswarnrNumbaIRAssumptionWarningruredefinerrrrx)rrDrr]r8 newtargetwmsgs&&& rr_FreshVarHandler.on_assign&s ==   !2 27OEH%F6{a"MM  0)<>>8&y~~&88IJDMM&"A"A$&,jj#23"NN6==+=+=6::NN YY llJJF '? # * *6 2 rcV#rrrs&&&rr_FreshVarHandler.on_other<s rrNrr)s@rrZrZ#s,rrZcXa]tRtRtoRtRtRtRtRtRt Rt R t R R lt R t VtR #)rAi@a2Replace variable uses in IR nodes to the correct reaching variable and introduce Phi nodes if necessary. This class contains the core of the SSA reconstruction algorithm. See Ch 5 of the Inria SSA book for reference. The method names used here are similar to the names used in the pseudocode in the book. cWnR#r_cache_list_vars)rr6s&&rr_FixSSAVars.__init__Is /rcvVPp\V\P4'dVP WVP P VP44pVeVP\PJdVR,VPP8wddVR,VP/p\V4p\P!W54\P!VPVVPR7#V#\V\P4'dVP WV.4pVexVP\PJdZVR,VPP8wd8\P!VPVPVPR7#V#)Nr7r)rr~rInst_fix_varrrrrrrrreplace_vars_innerrruVar)rrDrrhsnewdefreplmaps&&& rr_FixSSAVars.on_assignLsCll c277 # #]] 5 5 9 9&,, GF!fmm2<<&G)$ (:(::%i0&--@Gs)C//=99%}}!"JJ  RVV $ $]]6C59F!fmm2<<&G)$ (:(::99%}}$mm"JJ  rcPVPWVPPV44pVewVP\P JdYVR,VPP 8wd7VR,VP/p\V4p\P!W$4V#)Nr7) rrrrrrrrrreplace_vars_stmt)rrDr{rrs&&& rr_FixSSAVars.on_otherks $//33D9   &--r||"Ci FMM$6$66!),fmm<Dz**49 rc VUu.uFqDPNK ppVR,pWe9dVPW4#R#uupi)z0Fix all variable uses in ``used_vars``. r7N)r _find_def)rrDr{ used_varsrOvarnamesphivars&&&& rr_FixSSAVars._fix_varvsC%..IqFFI. "  >>&/ / /s=c \PRVR,V4RpVR,pVR,V,pVR,V,pVR,pVPW'4p\V4F1p VPWVR7p W8dT pMW9gK(VR ,pM VfVP WVP R 7pV#) z?Find definition of ``stmt`` for the statement ``stmt`` zfind_def var=%r stmt=%sr7NrFr8r;rv)stopr)r.r/ _stmt_indexreversed_find_def_from_topru) rrDr{ selected_defrF local_defs local_phisrvcur_posdefstmtdef_poss &&& rr_FixSSAVars._find_def~s  / 1BDI wH%e, H%e, w""4/ +G&&wG&DG & &)"~ ,  224883Lrc \PRV4VR,pVR,pVR,pVR,pW'9EdLVR,pVR,PpVPVR,VR 7p \P !V \P PVR 7VR 7p \PR W4WR,P^V 4Wb,PV 4VPV4FwrVPWVR 7p \PR V 4V PPPV P4V PPPV 4K V #VP!4V,pW8Xd\#VR,V4\$#\PR W4VPWVR 7#)zFind definition reaching block of ``label``. This method would look at all dominance frontiers. Insert phi node if necessary. zfind_def_from_top label %rr3r8r;r<r]rvr7rrzinsert phi node %s at %szincoming_def %szidom %s from label %s)r.r/rurrrExprphiinsertrx predecessors_find_def_from_bottomrincoming_valuesrincoming_blocksimmediate_dominators"_warn_about_uninitialized_variabler)rrDrFrur3r8r;r<r]freshvarphinodepredr_ incoming_defidoms&&&& rr_FixSSAVars._find_def_from_tops  2E:Um!!/  !7OE/%%C~~fY&7S~AHiiggkkck*G MM4g E M G , M  )++E2#99c :   /> --44\5H5HI --44T: 3N++-e4D}36)3DcJ(( MM14 ?--f-D Drc \PRV4VR,pWB,pV'd VR,pV#VPWVR7#)z 08,E\ C  rrAc\P'd2\P!\P !RV 2VR74R#R#)z Detected uninitialized variable rN)rALWAYS_WARN_UNINIT_VARrrr NumbaWarning)r7rus&&rrrs: $$$   27)<  %r)*rloggingrLr functoolsrr collectionsrnumbar numba.corerrrnumba.core.utilsr r numba.core.analysisr getLoggerr#r.rrr r1r,r?r0r`r-rer@r=rqrrfrrZrArrrrrs#++67   H %!H$$ <" % % P332|:`,`F r