U +‰dM4ã@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ƒZedkrÎe ¡dS)éN)Úir)Úbinding)ÚTestCaseé)Úrefprune_protoc#sB‡fdd„}tj ¡D]&\}}| d¡rd|›||ƒfVqdS)Ncs(‡‡fdd„}dˆj›dˆj›|_|S)Ncs ˆ|ˆƒS©N©)Úself)ÚfnÚ generate_testrú@/tmp/pip-unpacked-wheel-stw2luzp/llvmlite/tests/test_refprune.pyÚwrapped sz-_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 rsréc@sleZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z e e ƒD]\Z Zeeƒe <qTdS)Ú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. cCs"t t ¡tg¡}tj||ddS)NÚ NRT_incref©r$©rÚ FunctionTypeÚVoidTypeÚptr_tyÚFunction©r ÚmÚfntyrrr Ú make_incref/szTestRefPrunePass.make_increfcCs"t t ¡tg¡}tj||ddS)NÚ NRT_decrefr)r*r/rrr Ú make_decref3szTestRefPrunePass.make_decrefcCs"t t d¡d¡}tj||ddS)Né rZswitcherr)©rr+ÚIntTyper.r/rrr Ú make_switcher7szTestRefPrunePass.make_switchercCs"t t d¡d¡}tj||ddS)NrrZbrancherr)r6r/rrr Ú make_brancher;szTestRefPrunePass.make_branchercCsÆt ¡}| |¡}| |¡}| |¡}| |¡}t t ¡tg¡}tj ||dd} | j \} d| _ i} |D]} |   | ¡| | <qht  ¡} | ¡D]2\} }|  | | ¡|| D]<}|dkrÈ|  || g¡q¬|dkrà|  || g¡q¬tdƒ‚q¬t|ƒ}|dkr|  ¡qŒ|dkr&|\}|  | |¡qŒ|d kr\|\}}|  |d ¡}|  || || |¡qŒ|d kr¸|  |d ¡}|^}}| j|| |d }t|ƒD] \}}| | |¡| |¡q”qŒtdƒ‚qŒ|S) NÚmainr)ZmemÚincrefÚdecrefZ unreachablerrér)Údefault)rÚModuler2r4r8r9r+r,r-r.Úargsr$Zappend_basic_blockZ IRBuilderrZposition_at_endÚcallÚAssertionErrorÚlenZret_voidÚbranchZcbranchÚswitchÚ enumerateZadd_caseÚtype)r rr r0Z incref_fnZ decref_fnZ switcher_fnZ brancher_fnr1r ÚptrZbbmapÚbbZbuilderZ jump_targetsÚactionÚ n_targetsÚdstÚleftÚrightÚselÚheadÚtailÚswÚirrr Ú generate_ir?sN              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]D\}} | |¡r@| dd8<||D]} || dd8<qjq@|jD]} | jdkrŒq qŒ| jD]^} || j} t| ƒ} |  d¡}|  d¡}|j| d|d| ›d |j| d|d| ›d q¦dS) Nr;r<)r;r<rr:r(r3zBB )Úmsg)rÚcountÚgetZ functionsr$ÚblocksrWr)r r[r!rÚdrÚvsZn_increfZn_decrefÚstatsZdec_bbÚfrIÚtextrrr Úcheck|s(          zTestRefPrunePass.checkcCs4|ƒ\}}}| ||¡}| |¡}| |||¡dSr)rTr]rg)r rrr r!rZZoutmodrrr r –s   zTestRefPrunePass.generate_testN)rrr#rr2r4r8r9rTr]rgr rr$rr%rrrr r'&s6r'c@seZdZdZdZddd„ZdS)Ú BaseTestByIRrzG declare void @NRT_incref(i8* %ptr) declare void @NRT_decref(i8* %ptr) NcCsjt |j›d|›¡}t ¡}|dkr4| |j¡n|j|j|dt ¡}| |¡t ¡}|||fS)NÚ ©Úsubgraph_limit)rUrVÚprologuerXrYÚrefprune_bitmaskZdump_refprune_statsr)r rZrkr[r\ÚbeforeÚafterrrr rg©sÿ zBaseTestByIR.check)N)rrr#rmrlrgrrrr rh¡srhc@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 } cCs"| |j¡\}}| |jd¡dS©Nr=)rgÚ per_bb_ir_1rÚ basicblock©r r[rdrrr Ú test_per_bb_1ÂszTestPerBB.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 } cCs2| |j¡\}}| |jd¡| dt|ƒ¡dS)Nézcall void @NRT_incref(i8* %ptr))rgÚ per_bb_ir_2rrsÚassertInrWrtrrr Ú test_per_bb_2ÑszTestPerBB.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 } cCs2| |j¡\}}| |jd¡| dt|ƒ¡dS)Nr=ú!call void @NRT_decref(i8* %other))rgÚ per_bb_ir_3rrsrxrWrtrrr Ú test_per_bb_3ászTestPerBB.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 } cCs2| |j¡\}}| |jd¡| dt|ƒ¡dS)Nrvrz)rgÚ per_bb_ir_4rrsrxrWrtrrr Ú test_per_bb_4ószTestPerBB.test_per_bb_4N)rrr#rUÚRefPruneSubpassesZPER_BBrmrrrurwryr{r|r}r~rrrr rp·s   rpc@sPeZdZejjZdZdd„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 } cCs"| |j¡\}}| |jd¡dSrq)rgÚ per_diamond_1rÚdiamondrtrrr Útest_per_diamond_1szTestDiamond.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 } cCs"| |j¡\}}| |jd¡dSrq)rgÚ per_diamond_2rr‚rtrrr Útest_per_diamond_2szTestDiamond.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 } cCs"| |j¡\}}| |jd¡dS©Nr)rgÚ per_diamond_3rr‚rtrrr Útest_per_diamond_3/szTestDiamond.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 } cCs"| |j¡\}}| |jd¡dSrq)rgÚ per_diamond_4rr‚rtrrr Útest_per_diamond_4CszTestDiamond.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 } cCs"| |j¡\}}| |jd¡dS)Nrv)rgÚ per_diamond_5rr‚rtrrr Útest_per_diamond_5XszTestDiamond.test_per_diamond_5N)rrr#rUrZDIAMONDrmrrƒr„r…r‡rˆr‰rŠr‹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 } cCs"| |j¡\}}| |jd¡dS)Né)rgÚfanout_1rÚfanoutrtrrr Ú test_fanout_1qszTestFanout.test_fanout_1a' 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) br label %bb_B ; illegal jump to other decref } cCs"| |j¡\}}| |jd¡dSr†)rgÚfanout_2rrrtrrr Ú test_fanout_2ƒszTestFanout.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 } cCs"| |j¡\}}| |jd¡dS)Né©rgÚfanout_3rrrtrrr Ú test_fanout_3™szTestFanout.test_fanout_3cCs&|j|jdd\}}| |jd¡dS)Nrrjrr•rtrrr Útest_fanout_3_limitedsz TestFanout.test_fanout_3_limitedN)rrr#rrUrZFANOUTrmrr‘r’r“r–r—r˜rrrr r]src@sDeZdZejjZdZdd„ZdZ dd„Z dZ dd „Z d Z d d „Zd S)ÚTestFanoutRaisezí define i32 @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 i32 0 bb_C: ret i32 1, !ret_is_raise !0 } !0 = !{i1 true} cCs"| |j¡\}}| |jd¡dSrq)rgÚfanout_raise_1rÚ fanout_raisertrrr Útest_fanout_raise_1¶sz#TestFanoutRaise.test_fanout_raise_1a define i32 @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 i32 0 bb_C: ret i32 1, !ret_is_a_raise !0 ; bad metadata } !0 = !{i1 true} cCs"| |j¡\}}| |jd¡dSr†)rgÚfanout_raise_2rr›rtrrr Útest_fanout_raise_2Ész#TestFanoutRaise.test_fanout_raise_2zÿ define i32 @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 i32 0 bb_C: ret i32 1, !ret_is_raise !0 } !0 = !{i32 1} ; ok; use i32 cCs"| |j¡\}}| |jd¡dSrq)rgÚfanout_raise_3rr›rtrrr Útest_fanout_raise_3Üsz#TestFanoutRaise.test_fanout_raise_3a7 define i32 @main(i8* %ptr, i1 %cond) { bb_A: call void @NRT_incref(i8* %ptr) br i1 %cond, label %bb_B, label %bb_C bb_B: ret i32 1, !ret_is_raise !0 ; BAD; all tails are raising without decref bb_C: ret i32 1, !ret_is_raise !0 ; BAD; all tails are raising without decref } !0 = !{i32 1} cCs"| |j¡\}}| |jd¡dSr†)rgÚfanout_raise_4rr›rtrrr Útest_fanout_raise_4îsz#TestFanoutRaise.test_fanout_raise_4N)rrr#rUrZ FANOUT_RAISErmršrœrržrŸr r¡r¢rrrr r™¤sr™Ú__main__)ZunittestZllvmliterrrUZllvmlite.testsrÚrrrrr7Z as_pointerr-r'rhrpr€rr™rr:rrrr Ús     {CcGO