i,=tddlZddlmZddlmZddlmZddlmZddl m cm Z dZ GddZGd d eZej d j#ZGd d eeZGddeeZGddeZGddeZGddeZGddeZedk(rej4yy)N) defaultdict)ir)binding)TestCasec#Kfd}tjjD]&\}}|jdsd|||f(yw)NcXfd}djdj|_|S)Nc|SN)selffn generate_tests q/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/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 s) +/ a }Mrcasetest_)proto__dict__items startswith)rrkcase_fns` r_iterate_casesr sJ nn**, 7 << !+tG}, ,-s ;AAceZdZdZy)PassManagerMixinctjtjjj }tj dd}tj ||S)Nr) speed_level size_level)llvminitialize_native_targetTargetfrom_default_triplecreate_target_machinecreate_pipeline_tuning_optionscreate_pass_builder)r tmptos rpbzPassManagerMixin.pbsN %%' [[ , , . D D F11aAN''C00rN)rr __qualname__r.r rrr!r!s1rr!cJeZdZdZdZeeD]\ZZeee<y)TestRefPrunePrototypez- Test that the prototype is working. c|\}}}tj||j}|j||yr )rFanoutAlgorithmrun assertEqual)r case_gennodesedgesexpectedgots rrz#TestRefPrunePrototype.generate_test#s;!)uh##E51557 3'rN) rrr/rrrnamerlocalsr rrr1r1s,( %]3 d4rr1cteZdZdZdZdZdZdZdZdZ dZ d Z e e D]\Z Zeee <y ) 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_incref9-r{{}vh7{{1d66rctjtjtg}tj||dS)N NRT_decrefrBrCrHs r make_decrefzTestRefPrunePass.make_decref=rLrctjtjdd}tj||dS)N r switcherrBrrDIntTyperGrHs r make_switcherzTestRefPrunePass.make_switcherAs-rzz"~r2{{1d44rctjtjdd}tj||dS)Nr brancherrBrSrHs r make_brancherzTestRefPrunePass.make_brancherEs-rzz!}b1{{1d44rctj}|j|}|j|}|j |}|j |}tj tjtg}tj||d} | j\} d| _ i} |D]} | j| | | <tj} |jD]H\} }| j| | || D]>}|dk(r| j!|| g|dk(r| j!|| g5t#dt%|}|dk(r| j'|dk(r|\}| j)| ||d k(r1|\}}| j!|d }| j+|| || ||d kDre| j!|d }|^}}| j-|| | }t/|D])\}}|j1|j3|| |+@t#d|S) NmainrBmemincrefdecref unreachablerrWr )default)rModulerKrOrUrYrDrErFrGargsr;append_basic_block IRBuilderrposition_at_endcallAssertionErrorlenret_voidbranchcbranchswitch enumerateadd_casetype)r r7r8rI incref_fn decref_fn switcher_fn brancher_fnrJr ptrbbmapbbbuilder jump_targetsaction n_targetsdstleftrightselheadtailswis r generate_irzTestRefPrunePass.generate_irIs IIK$$Q' $$Q' ((+ ((+ r{{}vh7 [[Dv .B--b1E"I,,. % B  # #E"I .)X%LLSE2x'LLSE2(77 $L)IA~  "a$uSz*a , ull;3U4[%,?Qll;3 , ^^Ct^='oFAsKK U3Z8.%]33?!.Brctjt|}|j}|j }|j |j |||Sr )r%parse_assemblystrr.getModulePassManageradd_refprune_passr4)r irmodmodr.pms rapply_refprunezTestRefPrunePass.apply_refprunesM!!#e*- WWY  $ $ &  sB rctd}|jD]/\}}|jd}|jd}||d||<1|jD]>\}} |j|s| dxxdzcc<||D]} || dxxdzcc<@|jD]} | j dk(sn j D]r} || j } t| } | jd}| jd}|j| d|d |  |j| d|d |  ty) Nc ttSr )rintr rrz(TestRefPrunePass.check..s C 0rr]r^)r]r^rWr[rArNzBB )msg) rrcountget functionsr;blocksrr5)r rr9r7drvsn_increfn_decrefstatsdec_bbfrwtexts rcheckzTestRefPrunePass.checks= 0 1[[]EArxx)Hxx)H&(;AaD# HAu||Ah1$&qkFfIh'1,'*"Avv ((BbggJEr7Dzz,/Hzz,/H   U8_hc"J  G   U8_hc"J  G rc|\}}}|j||}|j|}|j|||yr )rrr)r r6r7r8r9routmods rrzTestRefPrunePass.generate_testsB!)uh  .$$U+ 68U+rN)rrr/rrKrOrUrYrrrrrr;rr<r rrr?r?0sQ77554lH>,%]3 d4rr?ceZdZdZdZddZy) BaseTestByIRrzG declare void @NRT_incref(i8* %ptr) declare void @NRT_decref(i8* %ptr) Nctj|jd|}|j}|j }||j |j n|j |j |tj}|j||tj}|||z fS)N subgraph_limit) r%rprologuer.rrrefprune_bitmaskdump_refprune_statsr4)r rrrr.rbeforeafters rrzBaseTestByIR.checks!!T]]O2eW"=> WWY  $ $ &  !  !6!6 7  !6!60> ! @))+ sB((*EFN""rr )rrr/rrrr rrrrsH #rrc`eZdZejj ZdZdZdZ dZ dZ dZ dZ dZy ) TestPerBBzv define void @main(i8* %ptr) { call void @NRT_incref(i8* %ptr) call void @NRT_decref(i8* %ptr) ret void } cx|j|j\}}|j|jdyNr`)r per_bb_ir_1r5 basicblockr rrs r test_per_bb_1zTestPerBB.test_per_bb_1s0ZZ 0 01 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|j\}}|j|jd|j dt |y)Nzcall void @NRT_incref(ptr %ptr))r per_bb_ir_2r5rassertInrrs r test_per_bb_2zTestPerBB.test_per_bb_2sCZZ 0 01 U ))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 } c|j|j\}}|j|jd|j dt |y)Nr`!call void @NRT_decref(ptr %other))r per_bb_ir_3r5rrrrs r test_per_bb_3zTestPerBB.test_per_bb_3CZZ 0 01 U ))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 } c|j|j\}}|j|jd|j dt |y)Nrr)r per_bb_ir_4r5rrrrs r test_per_bb_4zTestPerBB.test_per_bb_4rrN)rrr/r%RefPruneSubpassesPER_BBrrrrrrrrrr rrrrsG--44K. KC KE KErrcjeZdZejj ZdZdZdZ dZ dZ dZ dZ dZd Zd Zy ) 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 } cx|j|j\}}|j|jdyr)r per_diamond_1r5diamondrs rtest_per_diamond_1zTestDiamond.test_per_diamond_1.ZZ 2 23 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 } cx|j|j\}}|j|jdyr)r per_diamond_2r5rrs 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 } cx|j|j\}}|j|jdyNr)r per_diamond_3r5rrs 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 } cx|j|j\}}|j|jdyr)r per_diamond_4r5rrs 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 } cx|j|j\}}|j|jdy)Nr)r per_diamond_5r5rrs rtest_per_diamond_5zTestDiamond.test_per_diamond_5irrN)rrr/r%rDIAMONDrrrrrrrrrrrr rrrr sP--55 M+ M+M +M +M"+rrc`eZdZdZej j ZdZdZ dZ dZ dZ dZ dZy ) 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 } cx|j|j\}}|j|jdy)N)rfanout_1r5fanoutrs r test_fanout_1zTestFanout.test_fanout_1,ZZ . 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 } cx|j|j\}}|j|jdyr)rfanout_2r5rrs 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 } cx|j|j\}}|j|jdy)Nrfanout_3r5rrs r test_fanout_3zTestFanout.test_fanout_3rrc||j|jd\}}|j|jdy)NrWrrrrs rtest_fanout_3_limitedz TestFanout.test_fanout_3_limiteds2ZZ aZ@ U q)rN)rrr/rr%rFANOUTrrrrrrrrr rrrrnsD--44 H* H*H$**rrcteZdZejj ZdZdZdZ dZ dZ dZ dZ dZd Zd Zd Zd Zy )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} cx|j|j\}}|j|jdyr)rfanout_raise_1r5 fanout_raisers rtest_fanout_raise_1z#TestFanoutRaise.test_fanout_raise_10ZZ 3 34 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} cx|j|j\}}|j|jdyr)rfanout_raise_2r5rrs rtest_fanout_raise_2z#TestFanoutRaise.test_fanout_raise_2s2ZZ 3 34 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 cx|j|j\}}|j|jdyr)rfanout_raise_3r5rrs 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} cx|j|j\}}|j|jdyr)rfanout_raise_4r5rrs 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} cx|j|j\}}|j|jdyr)rfanout_raise_5r5rrs 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} cx|j|j\}}|j|jdy)N)rfanout_raise_6r5rrs rtest_fanout_raise_6z#TestFanoutRaise.test_fanout_raise_68rrN)rrr/r%r FANOUT_RAISErrrrrrrrrrrrrr rrrrs\--:: N0N 0 N 0 N0N$0 N40rr__main__)unittest collectionsrllvmliterrr%llvmlite.testsrllvmlite.tests.refprune_prototestsrefprune_protorrr!r1rT as_pointerrFr?rrrrrrr[r rrr s#$#-- -11 H  A ! ! #~x!1~B#8-#.@E @EF`+,`+FD*D*NE0lE0P zHMMOr