sh@pSrSSKrSSKrSSKrSSKJr SSKJr SSKJr SSK J r SSK J r J r Jr SSKJrJr SS KJr \R("\5rS r"S S 5rS rSrSrSrSrSrSrSr Sr!Sr"Sr#"SS5r$"SS\$5r%"SS5r&"SS\$5r'"S S!\$5r(S"r)g)#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:[UR5UlU$)zfApply SSA reconstruction algorithm on the given IR. Produces minimal SSA using Choi et al algorithm. )_run_ssablocks)func_irs b/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/numba/core/ssa.pyreconstruct_ssars gnn-GN Nc \rSrSrSrSrSrg)_CacheListVars$c0UlgN_savedselfs r__init___CacheListVars.__init__%s  rcURRU5nUcUR5=URU'nU$r)rget list_vars)rinstgots rr _CacheListVars.get(s8kkood# ;&*nn&6 6DKK  rrN)__name__ __module__ __qualname____firstlineno__rr __static_attributes__rrrr$s rrc lU(d0$[U5n[U5n[X5n[5nUHSn[R SU5 [ X5up[R S[U55 [XXaUU5nMU [U5nXq:wa[R"S5eU$)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 rrr/s  !& )C!#&G$V1I$&O (' %V5 0-2GHvW.0'v.H""#<== Mrc[U5nXS'X&S'[[5=US'nX6S'[XB5US'[ X[ U55nUR 5HupXn XR-U lM U$)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 UR5R5VVs0sHupU[U5_M nnnSnU(axSnUR5HYup[[R UVs/sHoSUPM sn[55nUR U5(dMSX&-nSnM[ U(aMxU$s snnfs snf)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 cl[5nUR5Hup4U(dMX U-nM U$r)rKrB) iterated_dfr8r<deflabeldefstmtss rr?r?ws8EM$lln 8 2 2M- rcv[U5nXS'[[5=US'n[X[ 55nXC4$)z(Rewrite to put fresh variable names r7r8)r=rr>r@_FreshVarHandler)rr7rDr8rEs rr0r0sC& !F9 +D 11F8v"63C3EFI  rc<UR5tpUR$r)valuesscope)rfirst_s r _get_scoper`s IE ;;rc [[5n[[5n[X#S9n[ X[ 55 [ RS[U55 [UR5VVs/sHupV[U5S:dMUPM snn5nUR5nUR5H[upYXW;dM U HIn Xn X%V V s1sHupU iM n n n U RU 5(aM7URU5 MY M] [ RS[U55 U$s snnfs sn n f)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;A<GHI >> D  #k:>'C'e' C!..s33MM!$ $& MM$mI&>? HDs3D: D: E cUR5H2up4[RSU5 X1S'[XU5HnM M4 g)Nz"==== SSA block analysis pass on %srF)rBr.r/_run_ssa_block_pass)rrDhandlerrFblkr_s rrfrfs>lln  :EBw$V':A ;%rc40nUR5HupE[RSU5 [R"UR UR S9n/nXAS'XQS'[XU5HnUceURU5 M Xvl XcU'M U$)Nz!==== SSA block rewrite pass on %s)r]locrFblock) rBr.r/rBlockr]rvrrappendrC) rrDrsrErFrtnewblknewbodystmts rr@r@sIlln  95A sww7ww'W=D# ## NN4 > !%% rc&[[U5S9$)N)r])rer`)rs rr=r=s   rc#P# [RSU5 URH|n[RSU5 [U[R 5(aUR X5nOURX5nXCLaUb[RSU5 Uv M~ g7f)Nz Running %sz on stmt: %szreplaced with: %s)r.r/rC isinstancerAssign on_assignon_other)rDrtrsr|rets rrrrrs} MM,( mT* dBII & &##F1C""60C ?s MM-s 3 sB$B&c$\rSrSrSrSrSrSrg) _BaseHandlerzGA base handler for all the passes used here for the SSA algorithm. cg)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. Nr*)rrDassigns rr_BaseHandler.on_assignrcg)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. Nr*rrDr|s rr_BaseHandler.on_otherrrr*Nr%r&r'r(__doc__rrr)r*rrrrs $ rrc$\rSrSrSrSrSrSrg)rgiaBFind 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]] cUSURRRX!S45 UR5HDnURnXBRR:wdM*USUR US5 MF g)NrbrFrc)targetnameryr!rk)rrDrvarrOs rr_GatherDefsHandler.on_assign sqvv}}))*116'?2KL##%CAMM&&&vq!%%fWo6&rc~UR5H)nURnUSURUS5 M+ g)NrcrF)r!rrk)rrDr|rrOs rr_GatherDefsHandler.on_others7>>#CA 6N1  ! !&/ 2$rr*Nrr*rrrgrgs73rrgc2\rSrSrSr\R rSrg)UndefinedVariableic[S5e)NzNot intended for instantiation)NotImplementedErrorrs rrUndefinedVariable.__init__s!"BCCrr*N) r%r&r'r(rr UNDEFINEDrr)r*rrrrsD\\Frrc$\rSrSrSrSrSrSrg)rZi#z9Replaces assignment target with new fresh variables. cRURRUS:XGaUSnUSn[U5S:XaURn[R SU5 URUR ;aDSUR<S3n[ R"[R"UURS95 O.URURRURS9n[R"UURURS 9nXAS RU5 U$) Nr7r]r8rzfirst assign: %sz variable z is not in scope.rvrvaluervrF)rrrhr.r/ localvarswarningswarnr NumbaIRAssumptionWarningrvredefinerrrry)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 rcU$rr*rs rr_FreshVarHandler.on_other<s rr*Nrr*rrrZrZ#s,rrZcL\rSrSrSrSrSrSrSrSr Sr S r S S jr S r g )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. cXlgr_cache_list_vars)rr6s rr_FixSSAVars.__init__Is /rcBURn[U[R5(aUR XUR R UR55nUbUR[RLayUSURR:wa\USUR0n[U5n[R"X55 [R"URUURS9$U$[U[R5(aUR XU/5nUbnUR[RLaQUSURR:wa4[R"URURURS9$U$)Nr7r)rrrInst_fix_varrr rrrrrreplace_vars_innerrrvVar)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  rc:URXURRU55nUblUR[R LaOUSURR :wa2USUR0n[U5n[R"X$5 U$)Nr7) rrr rrrrrrreplace_vars_stmt)rrDr|rrs rr_FixSSAVars.on_otherks $//33D9   &--r||"Ci FMM$6$66!),fmm<Dz**49 rczUVs/sHoDRPM nnUSnXe;aURX5$gs snf)z0Fix all variable uses in ``used_vars``. r7N)r _find_def)rrDr| used_varsrOvarnamesphivars rr_FixSSAVars._fix_varvsC%..IqFFI. "  >>&/ / /s8c<[RSUSU5 SnUSnUSUnUSUnUSnURX'5n[U5H(n URXUS9n X:aU n OX;dM#US n O UcUR XUR S 9nU$) z?Find definition of ``stmt`` for the statement ``stmt`` zfind_def var=%r stmt=%sr7NrFr8r;rw)stopr)r.r/ _stmt_indexreversed_find_def_from_toprv) rrDr| selected_defrF local_defs local_phisrwcur_posdefstmtdef_poss rr_FixSSAVars._find_def~s  / 1BDI wH%e, H%e, w""4/ +G&&wG&DG & &)"~ ,  224883Lrc^[RSU5 USnUSnUSnUSnX';Ga+USnUSRnURUSUS 9n [R "U [R RUS 9US 9n [RS X5 XRRS U 5 XbRU 5 URU5HupURXUS 9n [RS U 5 U RRRU R5 U RRRU 5 M U $UR!5UnX:Xa[#USU5 [$$[RSX5 URXUS 9$)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]rwr7rrzinsert phi node %s at %srzincoming_def %szidom %s from label %s)r.r/rvrrrExprphiinsertry predecessors_find_def_from_bottomrincoming_valuesrincoming_blocksimmediate_dominators"_warn_about_uninitialized_variabler)rrDrFrvr3r8r;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|[RSU5 USnXBnU(aUSnU$URXUS9$)z 08,E\ C rrAc[R(a-[R"[R "SU3US95 gg)Nz Detected uninitialized variable r)rALWAYS_WARN_UNINIT_VARrrr NumbaWarning)r7rvs rrrs7 $$   27)<  %r)*rloggingrLr functoolsrr collectionsrnumbar numba.corerrr numba.core.utilsr r numba.core.analysisr getLoggerr%r.rrrr1r,r?r0r`r-rfr@r=rrrrgrrZrArr*rrrs#++67   H %!H$$ <" % % P332|:`,`F r