J/Ph@dZddlZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z m Z mZddlmZmZdd lmZejeZd ZGd d Zd ZdZdZdZdZdZdZdZ dZ!dZ"dZ#GddZ$Gdde$Z%GddZ&Gdde$Z'Gd d!e$Z(d"Z)dS)#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_blocksc8t|j|_|S)znApply SSA reconstruction algorithm on the given IR. Produces minimal SSA using Choi et al algorithm. )_run_ssablocks)func_irs N/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/numba/core/ssa.pyreconstruct_ssars gn--GN NceZdZdZdZdS)_CacheListVarsci|_dSN)_savedselfs r__init__z_CacheListVars.__init__%s  rcz|j|}||x|j|<}|Sr)rget list_vars)rinstgots rrz_CacheListVars.get(s:kood## ;&*nn&6&6 6DK  rN)__name__ __module__ __qualname__rrrrrr$s2rrc |siSt|}t|}t||}t}|D]l}td|t ||\}}tdt|t||||||}mt|}||krtj d|S)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$VS11I$&&O 0 0 ('   %VW55 0-2G2GHHHvwW.00 'v..H3"#<=== Mrc8t|}||d<||d<ttx|d<}||d<t|||d<t ||t |}|D]\} } || } | | jz| _|S)z=Rewrite all uses to ``varname`` given the definition map r1r2phimapr- phi_locations) _make_statesrlist_compute_phi_locations_run_block_rewrite _FixSSAVarsitemsbody) rr1r2r-r.r0statesr5 newblockslabelphilistcurblks rr+r+Ss& ! !FF9F8 +D 1 11F8vF5M4WfEEF?"66;3O3OPPI ,,..,,w5! + rcJd|Dd}|rmd}D]T\}}ttjfd|Dt }||r||z}d}U|mS)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. c4i|]\}}|t|Sr$)set.0kvss r z'_iterated_domfronts..ks$JJJ2CGGJJJrTFc g|] }| Sr$r$)rGv domfrontss r z'_iterated_domfronts..ps)C)C)C1)A,)C)C)Cr)dominance_frontierr<roperatoror_rE difference)r- keep_goingrHrIinnerrMs @rr&r&es KJ)?)?)A)A)G)G)I)IJJJIJ " __&& " "EAr8<)C)C)C)C)C)C)CSUUKKE## "e ! " rcpt}|D]\}}|r |||z}|Sr)rEr<) iterated_dfr2r6deflabeldefstmtss rr9r9wsGEEM$llnn33(  3 [2 2M rct|}||d<ttx|d<}t||t }||fS)z(Rewrite to put fresh variable names r1r2)r7rr8r:_FreshVarHandler)rr1r>r2r?s rr*r*sU& ! !FF9 +D 1 11F8v"663C3E3EFFI f rc<|^}}|jSr)valuesscope)rfirst_s r _get_scoper`s IEA ;rctt}tt}t||}t ||t t dt|td| D}| }| D]T\}}||vrK|D]H} || } d||D} | | s| |nIUt dt||S)zm Returns ------- res : Set[str] The SSA violators in a dictionary of variable names. )defsuseszdefs %sc>g|]\}}t|dk|S))lenrFs rrNz(_find_defs_violators..s'GGG%!R3r77Q;;A;;;rch|]\}}|Sr$r$)rG_assignr@s r z'_find_defs_violators..sCCCeCCCrzSSA violators %s)rr8rEdict_run_block_analysis_GatherDefsHandlerr(r)r r r< dominators intersectionadd) rr-rbrcr>r/domsrH use_blocksr@dom def_labelss rr'r'sG t  D s  D t$ ' ' 'F(:(<(<=== MM)]400111GG4::<<GGGHHI >>  D : I  #  5kCC$q'CCC !..s33MM!$$$E MM$mI&>&>??? rc|D]9\}}td|||d<t|||D]}:dS)Nz"==== SSA block analysis pass on %sr@)r<r(r)_run_ssa_block_pass)rr>handlerr@blkr_s rrkrkshllnn s :EBBBw$VS'::  A  rc>i}|D]\}}td|tj|j|j}g}||d<||d<t|||D]}|J||||_ |||<|S)Nz!==== SSA block rewrite pass on %s)r]locr@block) r<r(r)rBlockr]ryruappendr=) rr>rvr?r@rwnewblknewbodystmts rr:r:sIllnn " " s 95AAA sw777ww'W== ! !D### NN4  ! % rc<tt|S)N)r])rjr`)rs rr7r7s$      rc#bKtd||jD]}td|t|tjr|||}n|||}||ur|td||VdS)Nz Running %sz on stmt: %szreplaced with: %s)r(r)r= isinstancerAssign on_assignon_other)r>rwrvrrets rrurus MM,((( mT*** dBI & & 1##FD11CC""6400C d??s MM-s 3 3 3 rceZdZdZdZdZdS) _BaseHandlerzGA base handler for all the passes used here for the SSA algorithm. cdS)a 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$)rr>assigns rrz_BaseHandler.on_assignrcdS)a 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$rr>rs rrz_BaseHandler.on_otherrrNr!r"r#__doc__rrr$rrrrs<   $     rrceZdZdZdZdZdS)rlaZFind 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]] c|d|jj||df|D]@}|j}||jjkr'|d||dAdS)Nrbr@rc)targetnamer|rro)rr>rvarrHs rrz_GatherDefsHandler.on_assign svv})*1166'?2KLLL##%% 7 7CAFM&&&vq!%%fWo666 7 7rc|D]0}|j}|d||d1dS)Nrcr@)rrro)rr>rrrHs rrz_GatherDefsHandler.on_othersP>>## 3 3CA 6N1  ! !&/ 2 2 2 2 3 3rNrr$rrrlrls<77733333rrlc"eZdZdZejZdS)UndefinedVariablec td)NzNot intended for instantiation)NotImplementedErrorrs rrzUndefinedVariable.__init__s!"BCCCrN)r!r"r#rr UNDEFINEDrr$rrrrs)DDD\FFFrrceZdZdZdZdZdS)rZz9Replaces assignment target with new fresh variables. c|jj|dkr|d}|d}t|dkri|j}td||j|jvr8d|jd}t jtj ||j n&| |jj|j }tj ||j|j }||d ||S) Nr1r]r2rzfirst assign: %sz variable z is not in scope.ryrvalueryr@)rrrfr(r) localvarswarningswarnr NumbaIRAssumptionWarningryredefinerrrr|)rr>rr]r2 newtargetwmsgs rrz_FreshVarHandler.on_assign&s  =  !2 2 27OEH%F6{{a"M  0)<<<>88Jy~JJJDM&"A$&,j#2#2#2333"NN6=+=6:NNN Y lJF 6'? # * *6 2 2 2 rc|Srr$rs rrz_FreshVarHandler.on_other<s rNrr$rrrZrZ#s<,rrZcDeZdZdZdZdZdZdZdZdZ dZ d d Z d S) r;aFReplace 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. c||_dSr)_cache_list_vars)rr0s rrz_FixSSAVars.__init__Is /rc|j}t|tjr||||j|j}|}|jtjurj|d|jj krT|d|ji}t|}tj ||tj |j||jSnt|tjri||||g}|O|jtjur<|d|jj kr&tj |j|j|jS|S)Nr1r)rrrInst_fix_varrrrrrrrreplace_vars_innerrryVar)rr>rrhsnewdefreplmaps rrz_FixSSAVars.on_assignLsPl c27 # # ]] 5 9 9&, G GF!fm2<&G&G)$ (:::%i0&-@Gs))C/W===9%}!"J RV $ $ ]]66C599F!fm2<&G&G)$ (:::9%}$m"J  rc ||||j|}|\|jtjurI|d|jjkr3|d|ji}t|}tj |||S)Nr1) rrrrrrrrrreplace_vars_stmt)rr>rrrs rrz_FixSSAVars.on_otherks D$/33D99    &-r|"C"Ci FM$666!),fm<Dzz*4999 rcbd|D}|d}||vr|||SdS)z0Fix all variable uses in ``used_vars``. cg|] }|j Sr$)r)rGrHs rrNz(_FixSSAVars._fix_var..ys...qAF...rr1N) _find_def)rr>r used_varsvarnamesphivars rrz_FixSSAVars._fix_varvsH/.I... " X  >>&$// /  rctd|d|d}|d}|d|}|d|}|d}|||}t|D]2} || ||} | |kr| }n| |vr |d }n3|||||j }|S) z?Find definition of ``stmt`` for the statement ``stmt`` zfind_def var=%r stmt=%sr1Nr@r2r5rz)stopr)r(r) _stmt_indexreversed_find_def_from_topry) rr>r selected_defr@ local_defs local_phisrzcur_posdefstmtdef_poss rrz_FixSSAVars._find_def~s  / 1BDIII wH%e, H%e, w""4// ++  G&&wG&DDG  & J&&)"~ '  22483Lrctd||d}|d}|d}|d}||vrM|d}|dj}||d| } t j| tj| | } td | |||d | || | | |D]{\} } | || | } td | | j j | j| j j | || S||}||krt#|d|t$Std||| ||| S)zFind definition reaching block of ``label``. This method would look at all dominance frontiers. Insert phi node if necessary. zfind_def_from_top label %rr-r2r5r6r]rzr1rrzinsert phi node %s at %srzincoming_def %szidom %s from label %s)r(r)ryrrrExprphiinsertr| predecessors_find_def_from_bottomrincoming_valuesrincoming_blocksimmediate_dominators"_warn_about_uninitialized_variabler)rr>r@ryr-r2r5r6r]freshvarphinodepredr_ incoming_defidoms rrz_FixSSAVars._find_def_from_tops  2E:::Um!!/ M ! !7OE/%C~~fY&7S~AAHigkkck**G MM4gu E E E 5M G , , , 5M  ) ) )++E22 ; ;a#99Dc :    />>> -44\5HIII -44T::::N++--e4Du}}36)3DcJJJ(( MM14 ? ? ?--fd-DD Drctd||d}||}|r |d}|S||||S)zr@ryr2rblastdefs rrz!_FixSSAVars._find_def_from_bottoms^  5u===!e}  C2hGN**65c*BB Brrctt|jd|D]}|j||ur|cSt|jS)zFind the positional index of the statement at ``block``. Assumptions: - no two statements can point to the same object. N)rangerfr=)rrrzris rrz_FixSSAVars._stmt_indexsYs5:''.  Az!}''(5:rN)r) r!r"r#rrrrrrrrrr$rrr;r;@s000>   0008,E,E,E\ C C C      rr;cxtjr-tjt jd||dSdS)Nz Detected uninitialized variable r)rALWAYS_WARN_UNINIT_VARrrr NumbaWarning)r1rys rrrsX $   <7<<          r)*rloggingrPr functoolsrr collectionsrnumbar numba.corerrr numba.core.utilsr r numba.core.analysisr getLoggerr!r(rrrr+r&r9r*r`r'rkr:r7rurrlrrZr;rr$rrrs######++++++++++66666666777777 ' H % %!!!H$$ <"   % % % % % % % % P333333332|:`````,```F     r