J/Phf$ddlmZddlmZmZmZmZGddeZGddeZ GddeZ iZ Gd d eZ d S) )cached_property)iranalysis transformsir_utilsceZdZdZdS) YieldPointct|tjsJt|tjsJ||_||_d|_d|_dSN) isinstancerBlockYieldblockinst live_varsweak_live_vars)selfrrs S/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/numba/core/postproc.py__init__zYieldPoint.__init__sU%*****$)))))  "N)__name__ __module__ __qualname__rrrr r s######rr ceZdZdZdZdS) GeneratorInfoc"i|_g|_dSr ) yield_points state_varsrs rrzGeneratorInfo.__init__src4|jS)z= Return an iterable of YieldPoint instances. )rvaluesr s rget_yield_pointszGeneratorInfo.get_yield_pointss '')))rN)rrrrr#rrrrrs2 *****rrcpeZdZdZdZedZedZedZedZ dS)VariableLifetimez> For lazily building information of variable lifetime c||_dSr )_blocks)rblockss rrzVariableLifetime.__init__#s  rc4tj|jSr )rcompute_cfg_from_blocksr'r s rcfgzVariableLifetime.cfg&s/ ===rc4tj|jSr )rcompute_use_defsr'r s rusedefszVariableLifetime.usedefs*s(666rcltj|j|j|jj|jjSr )rcompute_live_mapr+r'r.usemapdefmapr s rlivemapzVariableLifetime.livemap.s/(4<)-)<)-)<>> >rcbtj|j|j|j|jjSr )rcompute_dead_mapsr+r'r3r.r2r s rdeadmapszVariableLifetime.deadmaps4s+)$(DL$,*.,*=?? ?rN) rrr__doc__rrr+r.r3r6rrrr%r%s>>_>77_7>>_> ??_???rr%cNeZdZdZdZd dedefdZdZdZdd Z dd Z d Z d S) PostProcessorz( A post-processor for Numba IR. c||_dSr )func_ir)rr;s rrzPostProcessor.__init__Ds  rF emit_delsextend_lifetimescDtj|jj|j_t |jj}||j_t j|j|jj|j j |j j }|jj D]\}}|||jj|<|jjr-t!|j_|n d|j_|r||dSdS)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) Nr=)rcanonicalize_cfgr;r(r%variable_lifetimercompute_live_variablesr+r.r2r6combineditemsblock_entry_vars is_generatorrgenerator_info_compute_generator_info_insert_var_dels)rr<r=vltbevoffsetir_blocks rrunzPostProcessor.runGs)9$,:MNN t|233), &-cgt|7J.1k.@.1l.CEE!% 3 9 9 ; ; B B FH69&kDL )( 3 3 < $ /*7//DL '  ( ( * * * **.DL '  E  ! !3C ! D D D D D E Ercr|jjj}|r Jd|jjD]z}|jD]p}t |tjrT|j }t |tj r3t|dz}||_ t||}|||j <q{dS)zP Fill `index` for the Yield instruction and create YieldPoints. z"rerunning _populate_generator_infoN)r;rGrr(r"bodyr rAssignvaluerlenindexr )rdctrr yieldinstrUyps r_populate_generator_infoz&PostProcessor._populate_generator_infodsl)6<<<<<<\(//11 2 2E  2 2dBI..2 $ I!)RX662 #C1 */ 'y99/1IO, 2 2 2rc|||jj}|D]F}t |j|j}t }t|jj }|D]}t|tj r0|j |jurn_||jjLt|tjr||j Jd|D]R}t|tjr6|j }||vr*||||R||_||_Ht }|D]}||jz}||jz}t-||_|dS)zu Compute the generator's state variables as the union of live variables at all yield points. rzcouldn't find yield pointN)rIrYr;rGr#setget_block_entry_varsriterrQr rrRrSraddtargetnameDelremoverrsortedr remove_dels) rgirXrrstmtsstmtr`sts rrHz%PostProcessor._compute_generator_infots  %%''' \ (%%'' / /BDL==bhGGHHI UUN''E 6 6dBI..1zRW,,MM$+"23333bf--1$$TZ00055555  dBF++:Dy((!((...&**4000$BL .B   UU%%'' $ $B ", B "# #BBr   rcv|jj}||jj|jj|dS)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;rA_patch_var_delsr6internalescaping)rr=rJs rrIzPostProcessor._insert_var_delssK"l, S\2CL4I.>  @ @ @ @ @rc |jjD]%\}}||}g}t |jddD]}t d|D} | |z} tD]+\} } t|| r| || } | | z} || z},| || f|| z}g}|j }g}t |D]\}}|r|jdj }n|j }t|tj s| |t|dD]E}tj ||}|r| |0| |F|r||| |jd||_||}t|D];}|tj ||jdj <'dS)z- Insert delete in each block Nc3$K|] }|jV dSr )r`).0vs r z0PostProcessor._patch_var_dels..s$@@!qv@@@@@@rT)reverse)locr)r;r(rDcopyreversedrQr[ list_varsir_extension_insert_delsr appendrtrrarcextendprepend)rinternal_dead_mapescaping_dead_mapr=rLrMinternal_dead_set delete_ptsrglive_setdead_setTdef_func done_delsrQlastloc del_store delete_setvar_namedelnodeescape_dead_sets rrjzPostProcessor._patch_var_delss !% 3 9 9 ; ;1 M1 M FH 1& 9 > > @ @ J ss!344 . .@@t~~/?/?@@@@@#&77#;#A#A#C#C77KAx!$**7$,HT8$<$<  I-)Y6)!!4"2333!X-!!DlGI$,Z$8$8 - - j$'&mB/3GG"hG!$//&KK%%%!'z4 @ @ @--H fX7;;;G'-!((1111 G,,,, -   ' I&&& KK b) * * * HM07O"?33 M M  hmA6F6J!K!K!KLLLL Ma1 M1 MrcBtj|jjdS)z, Strips the IR of Del nodes N)rrdr;r(r s rrdzPostProcessor.remove_delss  T\011111rN)FF)F) rrrr7rboolrNrYrHrIrjrdrrrr9r9?sEETETEEEE:222 (((T@@@@,*/6M6M6M6Mp22222rr9N) functoolsr numba.corerrrrobjectr rr%rxr9rrrrs%%%%%%999999999999######## * * * * *F * * *?????v???:p2p2p2p2p2Fp2p2p2p2p2r