o ài›7ã@sÖddlZddlmZddlmZddlmZddlmZ dd„Z Gd d „d eƒZ e  d ¡  ¡ZGd d „d eƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZedkrie ¡dSdS)éN)Úir)Úbinding)ÚTestCaseé)Úrefprune_protoc#sD‡fdd„}tj ¡D]\}}| d¡rd|›||ƒfVq dS)Ncs(‡‡fdd„}dˆj›dˆj›|_|S)Ncs ˆ|ˆƒS©N©)Úself)ÚfnÚ generate_testrúk/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/llvmlite/tests/test_refprune.pyÚwrapped s z-_iterate_cases..wrap..wrappedzgenerated test for Ú.)Ú __module__Ú__name__Ú__doc__)r r ©r )r r Úwrap sz_iterate_cases..wrapÚcaseZtest_)ÚprotoÚ__dict__ÚitemsÚ startswith)r rÚkZcase_fnrrr Ú_iterate_cases s€  €þrc@s4eZdZdZdd„ZeeƒD] \ZZeeƒe<qdS)ÚTestRefPrunePrototypez- Test that the prototype is working. cCs,|ƒ\}}}t ||¡ ¡}| ||¡dSr)rZFanoutAlgorithmÚrunÚ assertEqual)r Úcase_genÚnodesÚedgesÚexpectedÚgotrrr r s z#TestRefPrunePrototype.generate_testN) rrÚ __qualname__rr rÚnamerÚlocalsrrrr rs  ÿréc@sleZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z e e ƒD] \Z Zeeƒe <q*dS)ÚTestRefPrunePasszß Test that the C++ implementation matches the expected behavior as for the prototype. This generates a LLVM module for each test case, runs the pruner and checks that the expected results are achieved. cCó"t t ¡tg¡}tj||ddS)NÚ NRT_incref©r$©rÚ FunctionTypeÚVoidTypeÚptr_tyÚFunction©r ÚmÚfntyrrr Ú make_incref/ózTestRefPrunePass.make_increfcCr()NÚ NRT_decrefr*r+r0rrr Ú make_decref3r4zTestRefPrunePass.make_decrefcCó"t t d¡d¡}tj||ddS)Né rZswitcherr*©rr,ÚIntTyper/r0rrr Ú make_switcher7r4zTestRefPrunePass.make_switchercCr7)NrrZbrancherr*r9r0rrr Ú make_brancher;r4zTestRefPrunePass.make_branchercCs¶t ¡}| |¡}| |¡}| |¡}| |¡}t t ¡tg¡}tj ||dd} | j \} d| _ i} |D] } |   | ¡| | <q4t  ¡} | ¡D]’\} }|  | | ¡|| D]}|dkrc|  || g¡qU|dkro|  || g¡qUtdƒ‚t|ƒ}|dkr€|  ¡qF|dkr|\}|  | |¡qF|d kr©|\}}|  |d ¡}|  || || |¡qF|d krÕ|  |d ¡}|^}}| j|| |d }t|ƒD]\}}| | |¡| |¡qÄqFtdƒ‚|S) NÚmainr*ZmemÚincrefÚdecrefZ unreachablerrér)Údefault)rÚModuler3r6r;r<r,r-r.r/Úargsr$Zappend_basic_blockZ IRBuilderrZposition_at_endÚcallÚAssertionErrorÚlenZret_voidÚbranchZcbranchÚswitchÚ enumerateZadd_caseÚtype)r rr r1Z incref_fnZ decref_fnZ switcher_fnZ brancher_fnr2r ZptrZbbmapÚbbZbuilderZ jump_targetsÚactionZ n_targetsÚdstÚleftÚrightÚselÚheadÚtailÚswÚirrr Ú generate_ir?sP        ÿzTestRefPrunePass.generate_ircCs,t t|ƒ¡}t ¡}| ¡| |¡|Sr)ÚllvmÚparse_assemblyÚstrÚModulePassManagerÚadd_refprune_passr)r ÚirmodÚmodÚpmrrr Úapply_refpruneus  zTestRefPrunePass.apply_refprunecCs i}| ¡D]\}}| d¡}| d¡}||dœ||<q| ¡D]"\}} | |¡rB| dd8<||D] } || dd8<q5q |jD] } | jdkrOnqF| jD]/} || j} t| ƒ} |  d¡}|  d¡}|j| d|d| ›d |j| d|d| ›d qSdS) Nr>r?)r>r?rr=r)r5zBB )Úmsg)rÚcountÚgetZ functionsr$ÚblocksrXr)r r\r!rÚdrÚvsZn_increfZn_decrefÚstatsZdec_bbÚfrKÚtextrrr Úcheck|s.    €  ÿ    úzTestRefPrunePass.checkcCs4|ƒ\}}}| ||¡}| |¡}| |||¡dSr)rUr^rh)r rrr r!r[Zoutmodrrr r –s   zTestRefPrunePass.generate_testN)rrr#rr3r6r;r<rUr^rhr rr$rr%rrrr r'&s6 ÿr'c@seZdZdZdZddd„ZdS)Ú BaseTestByIRrzG declare void @NRT_incref(i8* %ptr) declare void @NRT_decref(i8* %ptr) NcCsjt |j›d|›¡}t ¡}|dur| |j¡n|j|j|dt ¡}| |¡t ¡}|||fS)NÚ ©Úsubgraph_limit)rVrWÚprologuerYrZÚrefprune_bitmaskZdump_refprune_statsr)r r[rlr\r]ÚbeforeÚafterrrr rh©sÿ  zBaseTestByIR.checkr)rrr#rnrmrhrrrr ri¡sric@sDeZdZejjZdZdd„ZdZ dd„Z dZ dd „Z d Z d d „Zd S)Ú TestPerBBzv define void @main(i8* %ptr) { call void @NRT_incref(i8* %ptr) call void @NRT_decref(i8* %ptr) ret void } cCó"| |j¡\}}| |jd¡dS©Nr@)rhÚ per_bb_ir_1rÚ basicblock©r r\rerrr Ú test_per_bb_1ÂózTestPerBB.test_per_bb_1zâ define void @main(i8* %ptr) { call void @NRT_incref(i8* %ptr) call void @NRT_incref(i8* %ptr) call void @NRT_incref(i8* %ptr) call void @NRT_decref(i8* %ptr) call void @NRT_decref(i8* %ptr) ret void } cCó2| |j¡\}}| |jd¡| dt|ƒ¡dS)Nézcall void @NRT_incref(i8* %ptr))rhÚ per_bb_ir_2rruÚassertInrXrvrrr Ú test_per_bb_2ÑózTestPerBB.test_per_bb_2zÌ define void @main(i8* %ptr, i8* %other) { call void @NRT_incref(i8* %ptr) call void @NRT_incref(i8* %ptr) call void @NRT_decref(i8* %ptr) call void @NRT_decref(i8* %other) ret void } cCry)Nr@ú!call void @NRT_decref(i8* %other))rhÚ per_bb_ir_3rrur|rXrvrrr Ú test_per_bb_3ár~zTestPerBB.test_per_bb_3zü ; reordered define void @main(i8* %ptr, i8* %other) { call void @NRT_incref(i8* %ptr) call void @NRT_decref(i8* %ptr) call void @NRT_decref(i8* %ptr) call void @NRT_decref(i8* %other) call void @NRT_incref(i8* %ptr) ret void } cCry)Nrzr)rhÚ per_bb_ir_4rrur|rXrvrrr Ú test_per_bb_4ór~zTestPerBB.test_per_bb_4N)rrr#rVÚRefPruneSubpassesZPER_BBrnrtrwr{r}r€rr‚rƒrrrr rq·s   rqc@óPeZdZejjZdZdd„ZdZ dd„Z dZ dd „Z d Z d d „Zd Zdd„ZdS)Ú TestDiamondz• define void @main(i8* %ptr) { bb_A: call void @NRT_incref(i8* %ptr) br label %bb_B bb_B: call void @NRT_decref(i8* %ptr) ret void } cCrrrs)rhÚ per_diamond_1rÚdiamondrvrrr Útest_per_diamond_1rxzTestDiamond.test_per_diamond_1zè define void @main(i8* %ptr, i1 %cond) { bb_A: call void @NRT_incref(i8* %ptr) br i1 %cond, label %bb_B, label %bb_C bb_B: br label %bb_D bb_C: br label %bb_D bb_D: call void @NRT_decref(i8* %ptr) ret void } cCrrrs)rhÚ per_diamond_2rrˆrvrrr Útest_per_diamond_2rxzTestDiamond.test_per_diamond_2a3 define void @main(i8* %ptr, i1 %cond) { bb_A: call void @NRT_incref(i8* %ptr) br i1 %cond, label %bb_B, label %bb_C bb_B: br label %bb_D bb_C: call void @NRT_decref(i8* %ptr) ; reject because of decref in diamond br label %bb_D bb_D: call void @NRT_decref(i8* %ptr) ret void } cCrr©Nr)rhÚ per_diamond_3rrˆrvrrr Útest_per_diamond_3/rxzTestDiamond.test_per_diamond_3a5 define void @main(i8* %ptr, i1 %cond) { bb_A: call void @NRT_incref(i8* %ptr) br i1 %cond, label %bb_B, label %bb_C bb_B: call void @NRT_incref(i8* %ptr) ; extra incref will not affect prune br label %bb_D bb_C: br label %bb_D bb_D: call void @NRT_decref(i8* %ptr) ret void } cCrrrs)rhÚ per_diamond_4rrˆrvrrr Útest_per_diamond_4CrxzTestDiamond.test_per_diamond_4a0 define void @main(i8* %ptr, i1 %cond) { bb_A: call void @NRT_incref(i8* %ptr) call void @NRT_incref(i8* %ptr) br i1 %cond, label %bb_B, label %bb_C bb_B: br label %bb_D bb_C: br label %bb_D bb_D: call void @NRT_decref(i8* %ptr) call void @NRT_decref(i8* %ptr) ret void } cCrr)Nrz)rhÚ per_diamond_5rrˆrvrrr Útest_per_diamond_5XrxzTestDiamond.test_per_diamond_5N)rrr#rVr„ZDIAMONDrnr‡r‰rŠr‹rrŽrrr‘r’rrrr r†ús  r†c@sDeZdZdZejjZdZdd„Z dZ dd„Z dZ d d „Z d d „Zd S)Ú TestFanoutz6More complex cases are tested in TestRefPrunePass zí define void @main(i8* %ptr, i1 %cond) { bb_A: call void @NRT_incref(i8* %ptr) br i1 %cond, label %bb_B, label %bb_C bb_B: call void @NRT_decref(i8* %ptr) ret void bb_C: call void @NRT_decref(i8* %ptr) ret void } cCrr)Né)rhÚfanout_1rÚfanoutrvrrr Ú test_fanout_1qrxzTestFanout.test_fanout_1a6 define void @main(i8* %ptr, i1 %cond, i8** %excinfo) { bb_A: call void @NRT_incref(i8* %ptr) br i1 %cond, label %bb_B, label %bb_C bb_B: call void @NRT_decref(i8* %ptr) ret void bb_C: call void @NRT_decref(i8* %ptr) br label %bb_B ; illegal jump to other decref } cCrrrŒ)rhÚfanout_2rr–rvrrr Ú test_fanout_2ƒrxzTestFanout.test_fanout_2a} define void @main(i8* %ptr, i1 %cond) { bb_A: call void @NRT_incref(i8* %ptr) call void @NRT_incref(i8* %ptr) br i1 %cond, label %bb_B, label %bb_C bb_B: call void @NRT_decref(i8* %ptr) call void @NRT_decref(i8* %ptr) call void @NRT_decref(i8* %ptr) ret void bb_C: call void @NRT_decref(i8* %ptr) call void @NRT_decref(i8* %ptr) ret void } cCrr)Né©rhÚfanout_3rr–rvrrr Ú test_fanout_3™rxzTestFanout.test_fanout_3cCs&|j|jdd\}}| |jd¡dS)Nrrkrr›rvrrr Útest_fanout_3_limitedsz TestFanout.test_fanout_3_limitedN)rrr#rrVr„ZFANOUTrnr•r—r˜r™rœrržrrrr r“]s r“c@r…)ÚTestFanoutRaisea' define i32 @main(i8* %ptr, i1 %cond, i8** %excinfo) { bb_A: call void @NRT_incref(i8* %ptr) br i1 %cond, label %bb_B, label %bb_C bb_B: call void @NRT_decref(i8* %ptr) ret i32 0 bb_C: store i8* null, i8** %excinfo, !numba_exception_output !0 ret i32 1 } !0 = !{i1 true} cCrrrs)rhÚfanout_raise_1rÚ fanout_raiservrrr Útest_fanout_raise_1¶rxz#TestFanoutRaise.test_fanout_raise_1a: define i32 @main(i8* %ptr, i1 %cond, i8** %excinfo) { bb_A: call void @NRT_incref(i8* %ptr) br i1 %cond, label %bb_B, label %bb_C bb_B: call void @NRT_decref(i8* %ptr) ret i32 0 bb_C: store i8* null, i8** %excinfo, !numba_exception_typo !0 ; bad metadata ret i32 1 } !0 = !{i1 true} cCrrrŒ)rhÚfanout_raise_2rr¡rvrrr Útest_fanout_raise_2Êsz#TestFanoutRaise.test_fanout_raise_2a: define i32 @main(i8* %ptr, i1 %cond, i8** %excinfo) { bb_A: call void @NRT_incref(i8* %ptr) br i1 %cond, label %bb_B, label %bb_C bb_B: call void @NRT_decref(i8* %ptr) ret i32 0 bb_C: store i8* null, i8** %excinfo, !numba_exception_output !0 ret i32 1 } !0 = !{i32 1} ; ok; use i32 cCrrrs)rhÚfanout_raise_3rr¡rvrrr Útest_fanout_raise_3àrxz#TestFanoutRaise.test_fanout_raise_3a! define i32 @main(i8* %ptr, i1 %cond, i8** %excinfo) { bb_A: call void @NRT_incref(i8* %ptr) br i1 %cond, label %bb_B, label %bb_C bb_B: ret i32 1 ; BAD; all tails are raising without decref bb_C: ret i32 1 ; BAD; all tails are raising without decref } !0 = !{i1 1} cCrrrŒ)rhÚfanout_raise_4rr¡rvrrr Útest_fanout_raise_4òrxz#TestFanoutRaise.test_fanout_raise_4a™ define i32 @main(i8* %ptr, i1 %cond, i8** %excinfo) { bb_A: call void @NRT_incref(i8* %ptr) br i1 %cond, label %bb_B, label %bb_C bb_B: call void @NRT_decref(i8* %ptr) br label %common.ret bb_C: store i8* null, i8** %excinfo, !numba_exception_output !0 br label %common.ret common.ret: %common.ret.op = phi i32 [ 0, %bb_B ], [ 1, %bb_C ] ret i32 %common.ret.op } !0 = !{i1 1} cCrrrs)rhÚfanout_raise_5rr¡rvrrr Útest_fanout_raise_5rxz#TestFanoutRaise.test_fanout_raise_5N)rrr#rVr„Z FANOUT_RAISErnr r¢r£r¤r¥r¦r§r¨r©rªrrrr rŸ¤s rŸÚ__main__)ZunittestZllvmliterrrVZllvmlite.testsrÚrrrrr:Z as_pointerr.r'rirqr†r“rŸrr=rrrr Ús"     {CcGi ÿ