.@j\?ddlZddlmZddlmZddlmZddlmZddl m cm Z dZ GddZGd d eZejd ZGd d eeZGddeeZGddeZGddeZGddeZGddeZedkrejdSdS)N) defaultdict)ir)binding)TestCasec#Kfd}tjD],\}}|drd|||fV-dS)NcDfd}djdj|_|S)Nc|SN)selffn generate_tests MC:\Projects\Historical\venv\Lib\site-packages\llvmlite/tests/test_refprune.pywrappedz-_iterate_cases..wrap..wrapped s =r** *zgenerated test for .) __module____name____doc__)r rrs` rwrapz_iterate_cases..wrap sB + + + + + +M MM MMrcasetest_)proto__dict__items startswith)rrkcase_fns` r_iterate_casesr s n**,,-- 7 <<   -!++ttG}}, , , ,--rceZdZdZdS)PassManagerMixinctjtj}tjd}tj||S)Nr) speed_level)llvminitialize_native_targetTargetfrom_default_triplecreate_target_machinecreate_pipeline_tuning_optionscreate_pass_builder)r tmptos rpbzPassManagerMixin.pbsV %''' [ , , . . D D F F1a@@@'C000rN)rr __qualname__r-r rrr!r!s#11111rr!cTeZdZdZdZeeD]\ZZeee<dS)TestRefPrunePrototypez- Test that the prototype is working. c|\}}}tj||}|||dSr )rFanoutAlgorithmrun assertEqual)r case_gennodesedgesexpectedgots rrz#TestRefPrunePrototype.generate_test#sN!)uh#E5115577 3'''''rN) rrr.rrrnamerlocalsr rrr0r0s]((( %n]33 drr0c~eZdZdZdZdZdZdZdZdZ dZ d Z e e D]\Z Zeee <d S) 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. ctjtjtg}tj||dS)N NRT_increfr:r FunctionTypeVoidTypeptr_tyFunctionr mfntys r make_increfzTestRefPrunePass.make_incref92r{}}vh77{1d6666rctjtjtg}tj||dS)N NRT_decrefrArBrGs r make_decrefzTestRefPrunePass.make_decref=rKrc~tjtjdd}tj||dS)N r switcherrArrCIntTyperFrGs r make_switcherzTestRefPrunePass.make_switcherAs2rz"~~r22{1d4444rc~tjtjdd}tj||dS)Nr brancherrArRrGs r make_brancherzTestRefPrunePass.make_brancherEs2rz!}}b11{1d4444rcBtj}||}||}||}||}tjtjtg}tj ||d} | j \} d| _ i} |D]} | | | | <tj } |D]\} }| | | || D]L}|dkr| || g |dkr| || g>t#dt%|}|dkr| |dkr |\}| | ||d kr@|\}}| |d }| || || | |d kr}| |d }|^}}| || | }t/|D]4\}}|||| |5t#d|S) NmainrAmemincrefdecref unreachablerrVr )default)rModulerJrNrTrXrCrDrErFargsr:append_basic_block IRBuilderrposition_at_endcallAssertionErrorlenret_voidbranchcbranchswitch enumerateadd_casetype)r r6r7rH incref_fn decref_fn switcher_fn brancher_fnrIr ptrbbmapbbbuilder jump_targetsaction n_targetsdstleftrightselheadtailswis r generate_irzTestRefPrunePass.generate_irIs IKK$$Q'' $$Q'' ((++ ((++ r{}}vh77 [Dv . . . 2 2B--b11E"II,.. %  4 4 B  # #E"I . . .) 8 8X%%LLSE2222x''LLSE2222(777L))IA~~  """"a$uSz****a , ull;33U4[%,????Qll;33 , ^^Ct^=='oo99FAsKK U3Z88889%]333rctjt|}|}|}|||||Sr )r$parse_assemblystrr-getModulePassManageradd_refprune_passr3)r irmodmodr-pms rapply_refprunezTestRefPrunePass.apply_refprunes^!#e**-- WWYY  $ $ & &  sB rctd}|D]7\}}|d}|d}||d||<8|D]K\}} ||r1| dxxdzcc<||D]} || dxxdzcc<L|jD]} | jdkrn| jD]} || j} t| } | d}| d}|| d|d |  || d|d |  dS) Nc*ttSr )rintr rrz(TestRefPrunePass.check..s C 0 0rr\r])r\r]rVrZr@rMzBB )msg) rrcountget functionsr:blocksrr4)r rr8r6drvsn_increfn_decrefstatsdec_bbfrvtexts rcheckzTestRefPrunePass.checks 00 1 1[[]] < r>0s777777555555444lHHH>,,,%n]33 drr>ceZdZdZdZddZdS) BaseTestByIRrzG declare void @NRT_incref(i8* %ptr) declare void @NRT_decref(i8* %ptr) Nctj|jd|}|}|}|||jn||j|tj}|||tj}|||z fS)N subgraph_limit) r$rprologuer-rrrefprune_bitmaskdump_refprune_statsr3)r rrrr-rbeforeafters rrzBaseTestByIR.checks!T]"="=e"="=>> WWYY  $ $ & &  !  !6 7 7 7 7  !60> ! @ @ @)++ sB(**EFN""rr )rrr.rrrr rrrrs5H # # # # # #rrcNeZdZejjZdZdZdZ dZ dZ dZ dZ dZd S) TestPerBBzv define void @main(i8* %ptr) { call void @NRT_incref(i8* %ptr) call void @NRT_decref(i8* %ptr) ret void } cv||j\}}||jddSNr_)r per_bb_ir_1r4 basicblockr rrs r test_per_bb_1zTestPerBB.test_per_bb_1s9ZZ 011 U )1-----rz 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 } c||j\}}||jd|dt |dS)Nzcall void @NRT_incref(ptr %ptr))r per_bb_ir_2r4rassertInrrs r test_per_bb_2zTestPerBB.test_per_bb_2sTZZ 011 U )1--- 7SBBBBBrz 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 } c||j\}}||jd|dt |dS)Nr_!call void @NRT_decref(ptr %other))r per_bb_ir_3r4rrrrs r test_per_bb_3zTestPerBB.test_per_bb_3TZZ 011 U )1--- 93s88DDDDDrz ; 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 } c||j\}}||jd|dt |dS)Nrr)r per_bb_ir_4r4rrrrs r test_per_bb_4zTestPerBB.test_per_bb_4rrN)rrr.r$RefPruneSubpassesPER_BBrrrrrrrrrr rrrrs-4K... KCCC KEEE KEEEEErrcXeZdZejjZdZdZdZ dZ dZ dZ dZ dZd Zd Zd S) 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 } cv||j\}}||jddSr)r per_diamond_1r4diamondrs rtest_per_diamond_1zTestDiamond.test_per_diamond_18ZZ 233 U *****rz 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 } cv||j\}}||jddSr)r per_diamond_2r4rrs rtest_per_diamond_2zTestDiamond.test_per_diamond_2,rra3 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 } cv||j\}}||jddSNr)r per_diamond_3r4rrs rtest_per_diamond_3zTestDiamond.test_per_diamond_3@rra5 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 } cv||j\}}||jddSr)r per_diamond_4r4rrs rtest_per_diamond_4zTestDiamond.test_per_diamond_4Trra0 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 } cv||j\}}||jddS)Nr)r per_diamond_5r4rrs rtest_per_diamond_5zTestDiamond.test_per_diamond_5irrN)rrr.r$rDIAMONDrrrrrrrrrrrr rrrr s-5 M+++ M+++M +++M +++M"+++++rrcNeZdZdZejjZdZdZ dZ dZ dZ dZ dZd 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 } cv||j\}}||jddS)N)rfanout_1r4fanoutrs r test_fanout_1zTestFanout.test_fanout_17ZZ .. U q)))))ra6 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 } cv||j\}}||jddSr)rfanout_2r4rrs r test_fanout_2zTestFanout.test_fanout_2rra} 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 } cv||j\}}||jddS)Nrfanout_3r4rrs r test_fanout_3zTestFanout.test_fanout_3rrcz||jd\}}||jddS)NrVrrrrs rtest_fanout_3_limitedz TestFanout.test_fanout_3_limiteds=ZZ aZ@@ U q)))))rN)rrr.rr$rFANOUTrrrrrrrrr rrrrns}-4 H*** H***H$********rrcbeZdZejjZdZdZdZ dZ dZ dZ dZ dZd Zd Zd Zd Zd S)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} cv||j\}}||jddSr)rfanout_raise_1r4 fanout_raisers rtest_fanout_raise_1z#TestFanoutRaise.test_fanout_raise_19ZZ 344 U +Q/////ra: 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} cv||j\}}||jddSr)rfanout_raise_2r4rrs rtest_fanout_raise_2z#TestFanoutRaise.test_fanout_raise_2s;ZZ 344 U +Q/////ra: 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 cv||j\}}||jddSr)rfanout_raise_3r4rrs rtest_fanout_raise_3z#TestFanoutRaise.test_fanout_raise_3rra! 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} cv||j\}}||jddSr)rfanout_raise_4r4rrs rtest_fanout_raise_4z#TestFanoutRaise.test_fanout_raise_4rra 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} cv||j\}}||jddSr)rfanout_raise_5r4rrs rtest_fanout_raise_5z#TestFanoutRaise.test_fanout_raise_5rraf 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} cv||j\}}||jddS)N)rfanout_raise_6r4rrs rtest_fanout_raise_6z#TestFanoutRaise.test_fanout_raise_68rrN)rrr.r$r FANOUT_RAISErrrrrrrrrrrrrr rrrrs-: N000N 000 N 000 N000N$000 N400000rr__main__)unittest collectionsrllvmliterrr$llvmlite.testsrllvmlite.tests.refprune_prototestsrefprune_protorrr!r0rS as_pointerrEr>rrrrrrrZr rrr sL######$$$$$$######--------- - - -11111111     H    A ! ! # #~~~~~x!1~~~B#####8-###.@E@E@E@E@E @E@E@EF`+`+`+`+`+,`+`+`+FD*D*D*D*D*D*D*D*NE0E0E0E0E0lE0E0E0P zHMOOOOOr