|h,=vSSKrSSKJr SSKJr SSKJr SSKJr SSK J s J r Sr "SS5r"S S \5r\R "S 5R#5r"S S \\5r"SS\\5r"SS\5r"SS\5r"SS\5r"SS\5r\S:Xa\R4"5 gg)N) defaultdict)ir)binding)TestCasec#^# U4Sjn[RR5H,up#URS5(dMSU3U"U54v M. g7f)NcZ>^UU4SjnSTRSTR3UlU$)Nc>T"UT5$N)selffn generate_tests V/home/james-whalen/.local/lib/python3.13/site-packages/llvmlite/tests/test_refprune.pywrapped-_iterate_cases..wrap..wrapped s r* *zgenerated test for .) __module____name____doc__)r rrs` rwrap_iterate_cases..wrap s) +/ a }Mrcasetest_)proto__dict__items startswith)rrkcase_fns` r_iterate_casesr! sJ nn**,  <<  !+tG}, ,-s >AAc\rSrSrSrSrg)PassManagerMixinc[R"5 [RR5R 5n[R "SSS9n[R "X5$)Nr) speed_level size_level)llvminitialize_native_targetTargetfrom_default_triplecreate_target_machinecreate_pipeline_tuning_optionscreate_pass_builder)r tmptos rpbPassManagerMixin.pbsL %%' [[ , , . D D F11aAN''00rr N)rr __qualname____firstlineno__r1__static_attributes__r rrr#r#s1rr#cV\rSrSrSrSr\"\5Hurr\\ "5\'M Sr g)TestRefPrunePrototypez% Test that the prototype is working. cU"5up#n[R"X#5R5nURXE5 gr )rFanoutAlgorithmrun assertEqual)r case_gennodesedgesexpectedgots rr#TestRefPrunePrototype.generate_test#s5!)h##E1557 'rr N) rrr3r4rrr!namerlocalsr5r rrr7r7s,( %]3 d4rr7c\rSrSrSrSrSrSrSrSr Sr S r S r \ "\ 5Hurr\\"5\'M S rg ) TestRefPrunePass0z 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. c[R"[R"5[/5n[R"XSS9$)N NRT_increfrCr FunctionTypeVoidTypeptr_tyFunctionr mfntys r make_increfTestRefPrunePass.make_incref9+r{{}vh7{{166rc[R"[R"5[/5n[R"XSS9$)N NRT_decrefrKrLrQs r make_decrefTestRefPrunePass.make_decref=rVrc[R"[R"S5S5n[R"XSS9$)N r switcherrKrrMIntTyperPrQs r make_switcherTestRefPrunePass.make_switcherAs+rzz"~r2{{144rc[R"[R"S5S5n[R"XSS9$)Nr brancherrKr^rQs r make_brancherTestRefPrunePass.make_brancherEs+rzz!}b1{{144rc[R"5nURU5nURU5nUR U5nUR U5n[R "[R"5[/5n[R"X8SS9n U Run SU l 0n UHn U RU 5X'M [R"5n UR5GHOupU RX5 XH@nUS:XaU R!XJ/5 MUS:XaU R!XZ/5 M7[#S5e [%U5nUS:XaU R'5 MUS:XaUunU R)U U5 MUS :Xa2UunnU R!US 5nU R+UU UU U5 MUS :aeU R!US 5nUtnnU R-UU US 9n[/U5H*unnUR1UR3U5U U5 M, GMG[#S5e U$) NmainrKmemincrefdecref unreachablerrcr )default)rModulerTrYr`rerMrNrOrPargsrCappend_basic_block IRBuilderrposition_at_endcallAssertionErrorlenret_voidbranchcbranchswitch enumerateadd_casetype)r r>r?rR incref_fn decref_fn switcher_fn brancher_fnrSr ptrbbmapbbbuilder jump_targetsaction n_targetsdstleftrightselheadtailswis r generate_irTestRefPrunePass.generate_irIs IIK$$Q' $$Q' ((+ ((+ r{{}vh7 [[v .B--b1EI,,. % B  # #EI .)X%LLE2x'LLE2(77 $L)IA~  "a$uSz*a , ull;3U4[%,?Qll;3 , ^^Ct^='oFAsKK U3Z8.%]33?!.Brc[R"[U55nUR5nUR 5nUR 5 UR X#5 U$r )r(parse_assemblystrr1getModulePassManageradd_refprune_passr;)r irmodmodr1pms rapply_refpruneTestRefPrunePass.apply_refprunesK!!#e*- WWY  $ $ &  s rc[S5nUR5H-upVURS5nURS5nXxS.XE'M/ UR5HDupYURU5(dMU S==S-ss'X%Hn XJS==S-ss'M MF URHn U R S:XdM O W R Hln XLR n [U 5n U RS5nU RS5nURU SUS U 3S 9 URU SUS U 3S 9 Mn g) Nc [[5$r )rintr rr(TestRefPrunePass.check..s C 0rrjrk)rjrkrcrhrJrXzBB )msg) rrcountget functionsrCblocksrr<)r rr@r>drvsn_increfn_decrefstatsdec_bbfrtexts rcheckTestRefPrunePass.checks1 0 1[[]EAxx)Hxx)H&;AD# HA||Ah1$&kFIh'1,'*"Avv ((BggJEr7Dzz,/Hzz,/H   U8_hc"J  G   U8_hc"J  G rcU"5up#nURX#5nURU5nURXdU5 gr )rrr)r r=r>r?r@routmods rrTestRefPrunePass.generate_tests<!)h  .$$U+ 6U+rr N)rrr3r4rrTrYr`rerrrrr!rCrrDr5r rrrGrG0sQ77554lH>,%]3 d4rrGc&\rSrSrSrSrSSjrSrg) BaseTestByIRrzG declare void @NRT_incref(i8* %ptr) declare void @NRT_decref(i8* %ptr) Nc[R"URSU35nUR5nUR 5nUcUR UR 5 OUR UR US9 [R"5nURX45 [R"5nX7U- 4$)N subgraph_limit) r(rprologuer1rrrefprune_bitmaskdump_refprune_statsr;)r rrrr1rbeforeafters rrBaseTestByIR.checks!!T]]O2eW"=> WWY  $ $ &  !  !6!6 7  !6!60> ! @))+ s((*FN""rr r )rrr3r4rrrr5r rrrrsH #rrch\rSrSr\R R rSrSr Sr Sr Sr Sr SrS rS rg ) TestPerBBzv define void @main(i8* %ptr) { call void @NRT_incref(i8* %ptr) call void @NRT_decref(i8* %ptr) ret void } cvURUR5upURURS5 gNrm)r per_bb_ir_1r< basicblockr rrs r test_per_bb_1TestPerBB.test_per_bb_1s.ZZ 0 01  ))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 } cURUR5upURURS5 UR S[ U55 g)Nzcall void @NRT_incref(ptr %ptr))r per_bb_ir_2r<rassertInrrs r test_per_bb_2TestPerBB.test_per_bb_2sAZZ 0 01  ))1- 7SBrz 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 } cURUR5upURURS5 UR S[ U55 g)Nrm!call void @NRT_decref(ptr %other))r per_bb_ir_3r<rrrrs r test_per_bb_3TestPerBB.test_per_bb_3AZZ 0 01  ))1- 93s8Drz ; 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 } cURUR5upURURS5 UR S[ U55 g)Nrr)r per_bb_ir_4r<rrrrs r test_per_bb_4TestPerBB.test_per_bb_4rrr N)rrr3r4r(RefPruneSubpassesPER_BBrrrrrrrrrr5r rrrrsG--44K. KC KE KErrcr\rSrSr\R R rSrSr Sr Sr Sr Sr SrS rS rS rS rg ) TestDiamondi z 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 } cvURUR5upURURS5 gr)r per_diamond_1r<diamondrs rtest_per_diamond_1TestDiamond.test_per_diamond_1,ZZ 2 23  *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 } cvURUR5upURURS5 gr)r per_diamond_2r<rrs rtest_per_diamond_2TestDiamond.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 } cvURUR5upURURS5 gNr)r per_diamond_3r<rrs rtest_per_diamond_3TestDiamond.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 } cvURUR5upURURS5 gr)r per_diamond_4r<rrs rtest_per_diamond_4TestDiamond.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 } cvURUR5upURURS5 g)Nr)r per_diamond_5r<rrs rtest_per_diamond_5TestDiamond.test_per_diamond_5irrr N)rrr3r4r(rDIAMONDrrrrrrrrrrrr5r rrrr sP--55 M+ M+M +M +M"+rrch\rSrSrSr\R RrSr Sr Sr Sr Sr SrS rS rg ) TestFanoutinz6More 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 } cvURUR5upURURS5 g)N)rfanout_1r<fanoutrs r test_fanout_1TestFanout.test_fanout_1*ZZ .  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 } cvURUR5upURURS5 gr)rfanout_2r<rrs r test_fanout_2TestFanout.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 } cvURUR5upURURS5 g)Nrfanout_3r<rrs r test_fanout_3TestFanout.test_fanout_3rrctURURSS9upURURS5 g)Nrcrrrrs rtest_fanout_3_limited TestFanout.test_fanout_3_limiteds0ZZ aZ@  q)rr N)rrr3r4rr(rFANOUTrrrrrrrr r5r rrrrnsD--44 H* H*H$**rrc|\rSrSr\R R rSrSr Sr Sr Sr Sr SrS rS rS rS rS rSrg)TestFanoutRaiseia' 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} cvURUR5upURURS5 gr)rfanout_raise_1r< fanout_raisers rtest_fanout_raise_1#TestFanoutRaise.test_fanout_raise_1.ZZ 3 34  ++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} cvURUR5upURURS5 gr)rfanout_raise_2r<rrs rtest_fanout_raise_2#TestFanoutRaise.test_fanout_raise_2s0ZZ 3 34  ++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 cvURUR5upURURS5 gr)rfanout_raise_3r<rrs rtest_fanout_raise_3#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} cvURUR5upURURS5 gr)rfanout_raise_4r<rrs rtest_fanout_raise_4#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} cvURUR5upURURS5 gr)rfanout_raise_5r<rrs rtest_fanout_raise_5#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} cvURUR5upURURS5 g)N)rfanout_raise_6r<rrs rtest_fanout_raise_6#TestFanoutRaise.test_fanout_raise_68rrr N)rrr3r4r(r FANOUT_RAISErrrrrrrrr r#r$r(r)r5r rrrrs\--:: N0N 0 N 0 N0N$0 N40rr__main__)unittest collectionsrllvmliterrr(llvmlite.testsrllvmlite.tests.refprune_prototestsrefprune_protorr!r#r7r_ as_pointerrOrGrrrrrrrhr rrr5s#$#-- -11 H  A ! ! #~x!1~B#8-#.@E @EF`+,`+FD*D*NE0lE0P z MMOr