sKg:JddlZddlmZddlmZddlmZddlmZ dZ Gdd eZ ejd jZGd d eZGd deZGddeZGddeZGddeZGddeZedk(rej,yy)N)ir)binding)TestCase)refprune_protoc#Kfd}tjjD]&\}}|jdsd|||f(yw)NcXfd}djdj|_|S)Nc|SN)selffn generate_tests `/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/llvmlite/tests/test_refprune.pywrappedz-_iterate_cases..wrap..wrapped s r* *zgenerated test for .) __module____name____doc__)rrrs` rwrapz_iterate_cases..wrap s) +/ a }Mrcasetest_)proto__dict__items startswith)rrkcase_fns` r_iterate_casesr sJ nn**, 7 << !+tG}, ,-s ;AAcJeZdZdZdZeeD]\ZZeee<y)TestRefPrunePrototypez- Test that the prototype is working. c|\}}}tj||j}|j||yr )rFanoutAlgorithmrun assertEqual)r case_gennodesedgesexpectedgots rrz#TestRefPrunePrototype.generate_tests;!)uh##E51557 3'rN) rr __qualname__rrr namerlocalsr rrr"r"s,( %]3 d4rr"cteZdZdZdZdZdZdZdZdZ 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_incref/-r{{}vh7{{1d66rctjtjtg}tj||dS)N NRT_decrefr4r5r:s r make_decrefzTestRefPrunePass.make_decref3r>rctjtjdd}tj||dS)N r switcherr4rr6IntTyper9r:s r make_switcherzTestRefPrunePass.make_switcher7s-rzz"~r2{{1d44rctjtjdd}tj||dS)Nrr brancherr4rEr:s r make_brancherzTestRefPrunePass.make_brancher;s-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) 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 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|}tj}|j |j ||Sr )llvmparse_assemblystrModulePassManageradd_refprune_passr%)r irmodmodpms rapply_refprunezTestRefPrunePass.apply_refpruneus@!!#e*-  # # %  s  rcti}|jD]/\}}|jd}|jd}||d||<1|jD]>\}} |j|s| dxxdzcc<||D]} || dxxdzcc<@|jD]} | jdk(sn j D]r} || j} t | } | jd}| jd}|j| d|d|  |j| d|d|  ty) NrNrO)rNrOrrLr3r@zBB )msg)rcountget functionsr-blocksryr&)r r}r*r(drvsn_increfn_decrefstatsdec_bbfrhtexts rcheckzTestRefPrunePass.check|s5 [[]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 )rurr)r r'r(r)r*r|outmods rrzTestRefPrunePass.generate_testsB!)uh  .$$U+ 68U+rN)rrr,rr=rArGrJrurrrr r-rr.r rrr1r1&sQ77554lH4,%]3 d4rr1ceZdZdZdZddZy) BaseTestByIRrzG declare void @NRT_incref(i8* %ptr) declare void @NRT_decref(i8* %ptr) Nchtj|jd|}tj}||j |j n|j |j |tj }|j|tj }|||z fS)N subgraph_limit)rwrxprologuerzr{refprune_bitmaskdump_refprune_statsr%)r r|rr}r~beforeafters rrzBaseTestByIR.checks!!T]]O2eW"=>  # # %  !  !6!6 7  !6!60> ! @))+ s ((*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|jdyNrQ)r per_bb_ir_1r& basicblockr r}rs 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(i8* %ptr))r per_bb_ir_2r&rassertInryrs r test_per_bb_2zTestPerBB.test_per_bb_2sCZZ 0 01 U ))1- 7SBrz 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|j\}}|j|jd|j dt |y)NrQ!call void @NRT_decref(i8* %other))r per_bb_ir_3r&rrryrs r test_per_bb_3zTestPerBB.test_per_bb_3CZZ 0 01 U ))1- 93s8Drz ; 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|j\}}|j|jd|j dt |y)Nrr)r per_bb_ir_4r&rrryrs r test_per_bb_4zTestPerBB.test_per_bb_4rrN)rrr,rwRefPruneSubpassesPER_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_1r&diamondrs 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_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 } cx|j|j\}}|j|jdyNr)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 } cx|j|j\}}|j|jdyr)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 } cx|j|j\}}|j|jdy)Nr)r per_diamond_5r&rrs rtest_per_diamond_5zTestDiamond.test_per_diamond_5XrrN)rrr,rwrDIAMONDrrrrrrrrrrrr rrrrsP--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_1r&fanoutrs r test_fanout_1zTestFanout.test_fanout_1q,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_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 } cx|j|j\}}|j|jdy)Nrfanout_3r&rrs r test_fanout_3zTestFanout.test_fanout_3rrc||j|jd\}}|j|jdy)Nrrrrrs rtest_fanout_3_limitedz TestFanout.test_fanout_3_limiteds2ZZ aZ@ U q)rN)rrr,rrwrFANOUTrrrrrrrrr rrrr]sD--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_1r& 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_2r&rrs 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_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} cx|j|j\}}|j|jdyr)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} cx|j|j\}}|j|jdyr)rfanout_raise_5r&rrs 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_6r&rrs rtest_fanout_raise_6z#TestFanoutRaise.test_fanout_raise_6'rrN)rrr,rwr FANOUT_RAISErrrrrrrrrrrrrr rrrrs\--:: N0N 0 N 0 N0N$0 N40rr__main__)unittestllvmliterrrwllvmlite.testsrrrr r"rF as_pointerr8r1rrrrrrrLr rrrs$#% - H  A ! ! #xxxv#8#,@E @EF`+,`+FD*D*NE0lE0P zHMMOr