U ,‰d_$ã@sdddlmZmZmZmZmZGdd„deƒZGdd„deƒZGdd„deƒZ iZ Gdd „d eƒZ d S) é)ÚutilsÚirÚanalysisÚ transformsÚir_utilsc@seZdZdd„ZdS)Ú YieldPointcCs<t|tjƒst‚t|tjƒs t‚||_||_d|_d|_dS©N) Ú isinstancerZBlockÚAssertionErrorÚYieldÚblockÚinstÚ live_varsÚweak_live_vars)Úselfr r ©rú7/tmp/pip-unpacked-wheel-eu7e0c37/numba/core/postproc.pyÚ__init__s zYieldPoint.__init__N)Ú__name__Ú __module__Ú __qualname__rrrrrrsrc@seZdZdd„Zdd„ZdS)Ú GeneratorInfocCsi|_g|_dSr)Ú yield_pointsÚ state_vars©rrrrrszGeneratorInfo.__init__cCs |j ¡S)z= Return an iterable of YieldPoint instances. )rÚvaluesrrrrÚget_yield_pointsszGeneratorInfo.get_yield_pointsN)rrrrrrrrrrsrc@sPeZdZdZdd„Zejdd„ƒZejdd„ƒZejdd „ƒZ ejd d „ƒZ d S) ÚVariableLifetimez> For lazily building information of variable lifetime cCs ||_dSr)Ú_blocks)rÚblocksrrrr"szVariableLifetime.__init__cCs t |j¡Sr)rZcompute_cfg_from_blocksrrrrrÚcfg%szVariableLifetime.cfgcCs t |j¡Sr)rZcompute_use_defsrrrrrÚusedefs)szVariableLifetime.usedefscCst |j|j|jj|jj¡Sr)rZcompute_live_mapr rr!ZusemapÚdefmaprrrrÚlivemap-s þzVariableLifetime.livemapcCst |j|j|j|jj¡Sr)rZcompute_dead_mapsr rr#r!r"rrrrÚdeadmaps3sÿzVariableLifetime.deadmapsN) rrrÚ__doc__rrÚcached_propertyr r!r#r$rrrrrs   rc@sVeZdZdZdd„Zdeedœdd„Zdd „Zd d „Zdd d „Z ddd„Z dd„Z dS)Ú PostProcessorz( A post-processor for Numba IR. cCs ||_dSr)Úfunc_ir)rr(rrrrCszPostProcessor.__init__F)Ú emit_delsÚextend_lifetimescCs¢t |jj¡|j_t|jjƒ}||j_t |j|jj|j j |j j ¡}|jj  ¡D]\}}|||jj|<qP|jjr†tƒ|j_| ¡nd|j_|rž|j|ddS)a Run the following passes over Numba IR: - canonicalize the CFG - emit explicit `del` instructions for variables - compute lifetime of variables - compute generator info (if function is a generator function) N©r*)rZcanonicalize_cfgr(rrÚvariable_lifetimerZcompute_live_variablesr r!r"r$ZcombinedÚitemsZblock_entry_varsZ is_generatorrÚgenerator_infoÚ_compute_generator_infoÚ_insert_var_dels)rr)r*ÚvltZbevÚoffsetÚir_blockrrrÚrunFs þ  zPostProcessor.runcCs||jjj}|rtdƒ‚|jj ¡D]T}|jD]H}t|tj ƒr,|j }t|tj ƒr,t |ƒd}||_ t||ƒ}|||j <q,q"dS)zP Fill `index` for the Yield instruction and create YieldPoints. z"rerunning _populate_generator_infoéN)r(r.rr rrÚbodyr rÚAssignÚvaluer ÚlenÚindexr)rÚdctr r Z yieldinstr:ÚyprrrÚ_populate_generator_infocs       z&PostProcessor._populate_generator_infoc Cs,| ¡| ¡|jj}| ¡D]Ì}t|j |j¡ƒ}tƒ}t|jj ƒ}|D]H}t |t j ƒr||j |jkrlq¢| |jj¡qLt |t jƒrL| |j ¡qLds¢tdƒ‚|D]8}t |t jƒrÚ|j }||krÞ| |¡| |¡q¦qàq¦||_||_q tƒ}| ¡D]}||jO}||jO}qüt|ƒ|_| ¡dS)zu Compute the generator's state variables as the union of live variables at all yield points. rzcouldn't find yield pointN)r0r=r(r.rÚsetZget_block_entry_varsr Úiterr6r rr7r8r ÚaddÚtargetÚnameÚDelÚremover rrÚsortedrÚ remove_dels) rZgir<rrZstmtsÚstmtrBÚstrrrr/ss<             z%PostProcessor._compute_generator_infocCs$|jj}|j|jj|jj|ddS)a® Insert del statements for each variable. Returns a 2-tuple of (variable definition map, variable deletion map) which indicates variables defined and deleted in each block. The algorithm avoids relying on explicit knowledge on loops and distinguish between variables that are defined locally vs variables that come from incoming blocks. We start with simple usage (variable reference) and definition (variable creation) maps on each block. Propagate the liveness info to predecessor blocks until it stabilize, at which point we know which variables must exist before entering each block. Then, we compute the end of variable lives and insert del statements accordingly. Variables are deleted after the last use. Variable referenced by terminators (e.g. conditional branch and return) are deleted by the successors or the caller. r+N)r(r,Ú_patch_var_delsr$ZinternalZescaping)rr*r1rrrr0sÿzPostProcessor._insert_var_delscCs”|jj ¡D]€\}}|| ¡}g}t|jdd…ƒD]n}tdd„| ¡Dƒƒ} | |@} t ¡D],\} } t || ƒrb| || ƒ} | | 8} || 8}qb|  || f¡|| 8}q8g}|j }g}t|ƒD]t\}}|rØ|jdj }n|j }t |t j ƒsô|  |¡t|ddD]0}t j ||d}|r$|  |¡n |  |¡qq¾|rD| |¡|  |jd¡||_||}t|ƒD]"}| t j ||jdj d¡qjq dS) z- Insert delete in each block Néÿÿÿÿcss|] }|jVqdSr)rB)Ú.0ÚvrrrÚ ¾sz0PostProcessor._patch_var_dels..T)Úreverse)Úlocr)r(rr-ÚcopyÚreversedr6r>Z list_varsÚir_extension_insert_delsr ÚappendrOrrCrEÚextendÚprepend)rZinternal_dead_mapZescaping_dead_mapr*r2r3Zinternal_dead_setZ delete_ptsrGZlive_setZdead_setÚTZdef_funcZ done_delsr6ZlastlocZ del_storeZ delete_setÚvar_nameZdelnodeZescape_dead_setrrrrI²sD          zPostProcessor._patch_var_delscCst |jj¡dS)z, Strips the IR of Del nodes N)rrFr(rrrrrrFêszPostProcessor.remove_delsN)FF)F)F) rrrr%rÚboolr4r=r/r0rIrFrrrrr'>s* ÿ 8r'N) Z numba.corerrrrrÚobjectrrrrRr'rrrrÚs