we7zddlZddlmZddlmZddlmZddlmZ dZ Gdd eZ ej d  ZGd d eZGd deZGddeZGddeZGddeZGddeZedkrejdSdS)N)ir)binding)TestCase)refprune_protoc#Kfd}tjD],\}}|drd|||fV-dS)NcDfd}djdj|_|S)Nc|SN)selffn generate_tests .wrap..wrapped s =r** *zgenerated test for .) __module____name____doc__)rrrs` rwrapz_iterate_cases..wrap sB + + + + + +M MM MMrcasetest_)proto__dict__items startswith)rrkcase_fns` r_iterate_casesr s n**,,-- 7 <<   -!++ttG}}, , , ,--rcTeZdZdZdZeeD]\ZZeee<dS)TestRefPrunePrototypez- Test that the prototype is working. c|\}}}tj||}|||dSr )rFanoutAlgorithmrun assertEqual)r case_gennodesedgesexpectedgots rrz#TestRefPrunePrototype.generate_testsN!)uh#E5115577 3'''''rN) rr __qualname__rrr namerlocalsr rrr"r"s]((( %n]33 drr"c~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_incref/2r{}}vh77{1d6666rctjtjtg}tj||dS)N NRT_decrefr4r5r:s r make_decrefzTestRefPrunePass.make_decref3r>rc~tjtjdd}tj||dS)N r switcherr4rr6IntTyper9r:s r make_switcherzTestRefPrunePass.make_switcher7s2rz"~~r22{1d4444rc~tjtjdd}tj||dS)Nrr brancherr4rEr:s r make_brancherzTestRefPrunePass.make_brancher;s2rz!}}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) Nmainr4memincrefdecref unreachablerrr )default)rModuler=rArGrJr6r7r8r9argsr-append_basic_block IRBuilderrposition_at_endcallAssertionErrorlenret_voidbranchcbranchswitch enumerateadd_casetype)r r(r)r; incref_fn decref_fn switcher_fn brancher_fnr<rptrbbmapbbbuilder jump_targetsaction n_targetsdstleftrightselheadtailswis r generate_irzTestRefPrunePass.generate_ir?s 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|}tj}||||Sr )llvmparse_assemblystrModulePassManageradd_refprune_passr%)r irmodmodpms rapply_refprunezTestRefPrunePass.apply_refpruneusL!#e**--  # % %  s  rci}|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) NrNrO)rNrOrrLr3r@zBB )msg)rcountget functionsr-blocksryr&)r r}r*r(drvsn_increfn_decrefstatsdec_bbfrhtexts rcheckzTestRefPrunePass.check|s [[]] < >  # % %  !  !6 7 7 7 7  !60> ! @ @ @)++ s (**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\}}||jddSNrQ)r per_bb_ir_1r& basicblockr r}rs 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(i8* %ptr))r per_bb_ir_2r&rassertInryrs r test_per_bb_2zTestPerBB.test_per_bb_2sTZZ 011 U )1--- 7SBBBBBrz 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 } c||j\}}||jd|dt |dS)NrQ!call void @NRT_decref(i8* %other))r per_bb_ir_3r&rrryrs r test_per_bb_3zTestPerBB.test_per_bb_3TZZ 011 U )1--- 93s88DDDDDrz ; 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 } c||j\}}||jd|dt |dS)Nrr)r per_bb_ir_4r&rrryrs r test_per_bb_4zTestPerBB.test_per_bb_4rrN)rrr,rwRefPruneSubpassesPER_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_1r&diamondrs 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_2r&rrs rtest_per_diamond_2zTestDiamond.test_per_diamond_2rra3 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_3r&rrs 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_4r&rrs rtest_per_diamond_4zTestDiamond.test_per_diamond_4Crra0 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_5r&rrs rtest_per_diamond_5zTestDiamond.test_per_diamond_5XrrN)rrr,rwrDIAMONDrrrrrrrrrrrr rrrrs-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_1r&fanoutrs r test_fanout_1zTestFanout.test_fanout_1q7ZZ .. 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_2r&rrs 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_3r&rrs r test_fanout_3zTestFanout.test_fanout_3rrcz||jd\}}||jddS)Nrrrrrs rtest_fanout_3_limitedz TestFanout.test_fanout_3_limiteds=ZZ aZ@@ U q)))))rN)rrr,rrwrFANOUTrrrrrrrrr rrrr]s}-4 H*** H***H$********rrcXeZdZejjZdZdZdZ dZ dZ dZ dZ dZd 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_1r& 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_2r&rrs 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_3r&rrs 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_4r&rrs 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_5r&rrs rtest_fanout_raise_5z#TestFanoutRaise.test_fanout_raise_5rrN)rrr,rwr FANOUT_RAISErrrrrrrrrrrr rrrrs-: N000N 000 N 000 N000N$00000rr__main__)unittestllvmliterrrwllvmlite.testsrrrr r"rF as_pointerr8r1rrrrrrrLr rrrs$$$$$$######%%%%%% - - -     H    A ! ! # #xxxxxxxxxv#####8###,@E@E@E@E@E @E@E@EF`+`+`+`+`+,`+`+`+FD*D*D*D*D*D*D*D*Nf0f0f0f0f0lf0f0f0R zHMOOOOOr