o 8ήc:@sHdZddlZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z ddl mZmZmZdd lmZeeZd d ZGd d d ZddZddZddZddZddZddZddZddZddZ d d!Z!d"d#Z"Gd$d%d%Z#Gd&d'd'e#Z$Gd(d)d)Z%Gd*d+d+e#Z&Gd,d-d-e#Z'd.d/Z(dS)0a 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)pformat) defaultdict)config)irir_utilserrors)compute_cfg_from_blockscCst|j|_|S)znApply SSA reconstruction algorithm on the given IR. Produces minimal SSA using Choi et al algorithm. )_run_ssablocks)func_irr5/tmp/pip-target-vg8gfxp4/lib/python/numba/core/ssa.pyreconstruct_ssas rc@seZdZddZddZdS)_CacheListVarscCs i|_dSN)_savedselfrrr__init__% z_CacheListVars.__init__cCs*|j|}|dur||j|<}|Sr)rget list_vars)rinstgotrrrr(s z_CacheListVars.getN)__name__ __module__ __qualname__rrrrrrr$s rcCs|siSt|}t|}t|}t}|D] }td|t||\}}tdt|t||||||}qt|}||krCt 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)r cfgdf_plus violatorscache_list_varsvarnamedefmapcfg_postrrrr /s&  r c Csxt|}||d<||d<tt|d<}||d<t|||d<t||t|}|D]\} } || } | | j| _q+|S)z=Rewrite all uses to ``varname`` given the definition map r*r+phimapr& phi_locations) _make_statesrlist_compute_phi_locations_run_block_rewrite _FixSSAVarsitemsbody) r r*r+r&r'r)statesr- newblockslabelphilistcurblkrrrr$Ssr$cspdd|Dd}|r6d}D]\}}ttjfdd|Dt}||r3||O}d}q|sS)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. cSsi|] \}}|t|qSr)set.0kvsrrr ksz'_iterated_domfronts..TFcsg|]}|qSrr)r=v domfrontsrr psz'_iterated_domfronts..)dominance_frontierr4roperatoror_r; difference)r& keep_goingr>r?innerrrBrres rcCs,t}|D] \}}|r|||O}q|Sr)r;r4) iterated_dfr+r.deflabeldefstmtsrrrr1ws  r1cCs6t|}||d<tt|d<}t||t}||fS)z(Rewrite to put fresh variable names r*r+)r/rr0r2_FreshVarHandler)r r*r6r+r7rrrr#s r#cCs|^}}|jSr)valuesscope)r first_rrr _get_scopes rScCsLtt}t||ttdt|dd|D}tdt||S)zm Returns ------- res : Set[str] The SSA violators in a dictionary of variable names. zdefs %scSs h|] \}}t|dkr|qS))lenr<rrr s z'_find_defs_violators..zSSA violators %s)rr0_run_block_analysis_GatherDefsHandlerr!r"rr4)r defsr(rrrr s r cCs4|D]\}}td|t|||D]}qqdS)Nz"==== SSA block analysis pass on %s)r4r!r"_run_ssa_block_pass)r r6handlerr8blkrRrrrrWs  rWc Csi}|D]8\}}td|tj|j|jd}g}||d<||d<t|||D] }|dus1J||q)||_ |||<q|S)Nz!==== SSA block rewrite pass on %s)rPlocr8block) r4r!r"rBlockrPr]rZappendr5) r r6r[r7r8r\newblknewbodystmtrrrr2s    r2cCstt|dS)N)rP)dictrS)r rrrr/sr/ccsrtd||jD],}td|t|tjr|||}n|||}||ur3|dur3td||Vq dS)Nz Running %sz on stmt: %szreplaced with: %s)r!r"r5 isinstancerAssign on_assignon_other)r6r\r[rcretrrrrZs      rZc@ eZdZdZddZddZdS) _BaseHandlerzGA base handler for all the passes used here for the SSA algorithm. cCdS)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. Nrrr6assignrrrrgz_BaseHandler.on_assigncCrl)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. Nrrr6rcrrrrhroz_BaseHandler.on_otherNrrr__doc__rgrhrrrrrks rkc@seZdZdZddZdS)rXzEFind all defs ``states`` is a Mapping[str, List[ir.Assign]] cCs||jj|dSr)targetnamer`rmrrrrgsz_GatherDefsHandler.on_assignN)rrrrrrgrrrrrXs rXc@seZdZddZejZdS)UndefinedVariablecCstd)NzNot intended for instantiation)NotImplementedErrorrrrrrszUndefinedVariable.__init__N)rrrrr UNDEFINEDrsrrrrrus ruc@rj)rNz9Replaces assignment target with new fresh variables. cCs|jj|dkrU|d}|d}t|dkr8|j}td||j|jvr7d|jd}ttj ||j dn |j |jj|j d}t j ||j|j d }||d ||S) Nr*rPr+rzfirst assign: %sz variable z is not in scope.r]rsvaluer]r8)rsrtrUr!r" localvarswarningswarnr NumbaIRAssumptionWarningr]redefinerrfrzr`)rr6rnrPr+ newtargetwmsgrrrrgs(    z_FreshVarHandler.on_assigncCs|Srrrprrrrhsz_FreshVarHandler.on_otherNrqrrrrrNs rNc@sReZdZdZddZddZddZdd Zd d Zd d Z ddZ dddZ dS)r3aFReplace 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. cCs ||_dSr)_cache_list_vars)rr)rrrr)rz_FixSSAVars.__init__cCs|j}t|tjrD||||j|j}|durB|jtjurB|d|jj krB|d|ji}t |}t ||tj |j||jdS|St|tjro||||g}|duro|jtjuro|d|jj krotj |j|j|jdS|S)Nr*ry)rzrerInst_fix_varrrrsrwrtrrreplace_vars_innerrfr]Var)rr6rnrhsnewdefreplmaprrrrg,s4   z_FixSSAVars.on_assigncCs`||||j|}|dur.|jtjur.|d|jjkr.|d|ji}t|}t |||S)Nr*) rrrrsrrwrtrrreplace_vars_stmt)rr6rcrrrrrrhKs z_FixSSAVars.on_othercCs.dd|D}|d}||vr|||SdS)z0Fix all variable uses in ``used_vars``. cSsg|]}|jqSr)rt)r=r>rrrrDYsz(_FixSSAVars._fix_var..r*N) _find_def)rr6rc used_varsvarnamesphivarrrrrVs  z_FixSSAVars._fix_varc Cstd|d|d}|d}|d|}|d|}|d}|||}t|D]} |j| ||d} | |kr;| }n | |vrE|d }nq)|durS|j|||jd }|S) z?Find definition of ``stmt`` for the statement ``stmt`` zfind_def var=%r stmt=%sr*Nr8r+r-r^)stoprx)r!r" _stmt_indexreversed_find_def_from_topr]) rr6rc selected_defr8 local_defs local_phisr^cur_posdefstmtdef_posrrrr^s*    z_FixSSAVars._find_defcCs4td||d}|d}|d}|d}||vrx|d}|dj}|j|d|d } tj| tjj|d |d } td | |||d | || | | |D]!\} } |j || |d } td | | j j | j| j j | qT| S||}||krt|d|tStd|||j |||d 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&r+r-r.rPr^r*rxryzinsert phi node %s at %srzincoming_def %szidom %s from label %s)r!r"r]rrrfExprphiinsertr` predecessors_find_def_from_bottomrzincoming_valuesrsincoming_blocksimmediate_dominators"_warn_about_uninitialized_variableru)rr6r8r]r&r+r-r.rPfreshvarphinodepredrR incoming_defidomrrrrzs>     z_FixSSAVars._find_def_from_topcCs<td||d}||}|r|d}|S|j|||dS)zs@          $   (  $