o "Æ&ij?ã@súddlZddlmZddlmZddlmZddlmZddl m m Z dd„Z Gdd „d ƒZGd d „d eƒZe d ¡ ¡ZGd d„deeƒZGdd„deeƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZedkr{e ¡dSdS)éN)Ú defaultdict)Úir)Úbinding)ÚTestCasec#sD‡fdd„}tj ¡D]\}}| d¡rd|›||ƒfVq dS)Ncs(‡‡fdd„}dˆj›dˆj›|_|S)Ncs ˆ|ˆƒS©N©)Úself)ÚfnÚ generate_testrúGC:\wamp64\www\opt\env\Lib\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@seZdZdd„ZdS)ÚPassManagerMixincCs0t ¡tj ¡ ¡}tjddd}t ||¡S)Nr)Z speed_levelZ size_level)ÚllvmZinitialize_native_targetZTargetZfrom_default_tripleZcreate_target_machineZcreate_pipeline_tuning_optionsZcreate_pass_builder)rÚtmZptorrr Úpbs zPassManagerMixin.pbN)rrÚ __qualname__rrrrr rs 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) rrrrr 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_incref9ózTestRefPrunePass.make_increfcCr+)NÚ NRT_decrefr-r.r3rrr Ú make_decref=r7zTestRefPrunePass.make_decrefcCó"t t d¡d¡}tj||ddS)Né rZswitcherr-©rr/ÚIntTyper2r3rrr Ú make_switcherAr7zTestRefPrunePass.make_switchercCr:)NérZbrancherr-r<r3rrr Ú make_brancherEr7zTestRefPrunePass.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ÚModuler6r9r>r@r/r0r1r2Úargsr'Zappend_basic_blockZ IRBuilderrZposition_at_endÚcallÚAssertionErrorÚlenZret_voidÚbranchZcbranchÚswitchÚ enumerateZadd_caseÚtype)rr#r$r4Z incref_fnZ decref_fnZ switcher_fnZ brancher_fnr5r ÚptrZbbmapÚbbZbuilderZ jump_targetsÚactionZ n_targetsÚdstÚleftÚrightÚselÚheadÚtailÚswÚirrr Ú generate_irIsP        ÿzTestRefPrunePass.generate_ircCs6t t|ƒ¡}| ¡}| ¡}| ¡| ||¡|Sr)rÚparse_assemblyÚstrrÚgetModulePassManagerÚadd_refprune_passr )rÚirmodÚmodrÚpmrrr Úapply_refprunes  zTestRefPrunePass.apply_refprunecCstdd„ƒ}| ¡D]\}}| d¡}| d¡}||dœ||<q | ¡D]"\}} | |¡rF| dd8<||D] } || dd8<q9q$|jD] } | jdkrSnqJ| jD]/} || j} t| ƒ} |  d¡}|  d ¡}|j| d|d | ›d |j| d|d | ›d qWdS) NcSsttƒSr)rÚintrrrr ÚŒsz(TestRefPrunePass.check..rBrC)rBrCr?rAr,r8zBB )Úmsg) rrÚcountÚgetZ functionsr'Úblocksr\r!)rr`r%r#ÚdrÚvsZn_increfZn_decrefÚstatsZdec_bbÚfrPÚtextrrr Úcheck‡s.     €  ÿ    úzTestRefPrunePass.checkcCs4|ƒ\}}}| ||¡}| |¡}| |||¡dSr)rZrbrn)rr"r#r$r%r_Zoutmodrrr r ¦s   zTestRefPrunePass.generate_testN)rrrrr6r9r>r@rZrbrnr rr'rr(rrrr r*0s6 ÿr*c@seZdZdZdZddd„ZdS)Ú BaseTestByIRrzG declare void @NRT_incref(i8* %ptr) declare void @NRT_decref(i8* %ptr) NcCstt |j›d|›¡}| ¡}| ¡}|dur| |j¡n|j|j|dt ¡}| ||¡t ¡}|||fS)NÚ ©Úsubgraph_limit) rr[Úprologuerr]r^Úrefprune_bitmaskZdump_refprune_statsr )rr_rrr`rraÚbeforeÚafterrrr rn¹sÿ  zBaseTestByIR.checkr)rrrrtrsrnrrrr ro±sroc@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©NrD)rnÚ per_bb_ir_1r!Ú basicblock©rr`rkrrr Ú 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(ptr %ptr))rnÚ per_bb_ir_2r!r{ÚassertInr\r|rrr Ú test_per_bb_2âózTestPerBB.test_per_bb_2zÌ define void @main(ptr %ptr, ptr %other) { call void @NRT_incref(ptr %ptr) call void @NRT_incref(ptr %ptr) call void @NRT_decref(ptr %ptr) call void @NRT_decref(ptr %other) ret void } cCr)NrDú!call void @NRT_decref(ptr %other))rnÚ per_bb_ir_3r!r{r‚r\r|rrr Ú test_per_bb_3òr„zTestPerBB.test_per_bb_3zü ; reordered define void @main(ptr %ptr, ptr %other) { call void @NRT_incref(ptr %ptr) call void @NRT_decref(ptr %ptr) call void @NRT_decref(ptr %ptr) call void @NRT_decref(ptr %other) call void @NRT_incref(ptr %ptr) ret void } cCr)Nr€r…)rnÚ per_bb_ir_4r!r{r‚r\r|rrr Ú test_per_bb_4r„zTestPerBB.test_per_bb_4N)rrrrÚRefPruneSubpassesZPER_BBrtrzr}rrƒr†r‡rˆr‰rrrr rwÈs   rwc@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 } cCrxry)rnÚ per_diamond_1r!Údiamondr|rrr Útest_per_diamond_1r~zTestDiamond.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 } cCrxry)rnÚ per_diamond_2r!rr|rrr Útest_per_diamond_2,r~zTestDiamond.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 } cCrx©Nr)rnÚ per_diamond_3r!rr|rrr Útest_per_diamond_3@r~zTestDiamond.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 } cCrxry)rnÚ per_diamond_4r!rr|rrr Útest_per_diamond_4Tr~zTestDiamond.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 } cCrx)Nr€)rnÚ per_diamond_5r!rr|rrr Útest_per_diamond_5ir~zTestDiamond.test_per_diamond_5N)rrrrrŠZDIAMONDrtrŒrŽrrr’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 } cCrx)Né)rnÚfanout_1r!Úfanoutr|rrr Ú test_fanout_1‚r~zTestFanout.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 } cCrxr‘)rnÚfanout_2r!r›r|rrr Ú test_fanout_2”r~zTestFanout.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 } cCrx)Né©rnÚfanout_3r!r›r|rrr Ú test_fanout_3ªr~zTestFanout.test_fanout_3cCs&|j|jdd\}}| |jd¡dS)Nr?rqrr r|rrr Útest_fanout_3_limited®sz TestFanout.test_fanout_3_limitedN)rrrrrrŠZFANOUTrtršrœrržr¡r¢r£rrrr r˜ns r˜c@s\eZdZejjZdZdd„ZdZ dd„Z dZ dd „Z d Z d d „Zd Zdd„ZdZdd„ZdS)Ú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} cCrxry)rnÚfanout_raise_1r!Ú fanout_raiser|rrr Útest_fanout_raise_1Çr~z#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} cCrxr‘)rnÚfanout_raise_2r!r¦r|rrr Ú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 cCrxry)rnÚfanout_raise_3r!r¦r|rrr Útest_fanout_raise_3ñr~z#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} cCrxr‘)rnÚfanout_raise_4r!r¦r|rrr Útest_fanout_raise_4r~z#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} cCrxry)rnÚfanout_raise_5r!r¦r|rrr Útest_fanout_raise_5r~z#TestFanoutRaise.test_fanout_raise_5af define i32 @main(i8* %ptr, i1 %cond1, i1 %cond2, i1 %cond3, i8** %excinfo) { bb_A: call void @NRT_incref(i8* %ptr) call void @NRT_incref(i8* %ptr) br i1 %cond1, label %bb_B, label %bb_C bb_B: call void @NRT_decref(i8* %ptr) br i1 %cond2, label %bb_D, label %bb_E bb_C: store i8* null, i8** %excinfo, !numba_exception_output !0 ret i32 1 bb_D: call void @NRT_decref(i8* %ptr) ret i32 0 bb_E: call void @NRT_incref(i8* %ptr) br i1 %cond3, label %bb_F, label %bb_C bb_F: call void @NRT_decref(i8* %ptr) call void @NRT_decref(i8* %ptr) ret i32 0 } !0 = !{i1 1} cCrx)Né)rnÚfanout_raise_6r!r¦r|rrr Útest_fanout_raise_68r~z#TestFanoutRaise.test_fanout_raise_6N)rrrrrŠZ FANOUT_RAISErtr¥r§r¨r©rªr«r¬r­r®r¯r±r²rrrr r¤µs r¤Ú__main__)ZunittestÚ collectionsrZllvmliterrrZllvmlite.testsrZllvmlite.tests.refprune_protoÚtestsZrefprune_protorrrrr=Z as_pointerr1r*rorwr‹r˜r¤rrArrrr Ús*      CcG ÿ