o Td7@sddlZddlmZddlmZddlmZddlmZ ddZ Gd d d eZ e d  ZGd d d eZGdddeZGdddeZGdddeZGdddeZGdddeZedkriedSdS)N)ir)binding)TestCase)refprune_protoc#sDfdd}tjD]\}}|drd|||fVq dS)Ncs(fdd}djdj|_|S)Ncs |SN)self)fn generate_testro/home/ncw/WWW/www-new/content/articles/pi-bbp/venv/lib/python3.10/site-packages/llvmlite/tests/test_refprune.pywrapped s z-_iterate_cases..wrap..wrappedzgenerated test for .) __module____name____doc__)r r r )r r wrap sz_iterate_cases..wrapcasetest_)proto__dict__items startswith)r rkcase_fnrrr _iterate_cases s  rc@s4eZdZdZddZeeD] \ZZeee<qdS)TestRefPrunePrototypez- Test that the prototype is working. cCs,|\}}}t||}|||dSr)rFanoutAlgorithmrun assertEqual)r case_gennodesedgesexpectedgotrrr r s z#TestRefPrunePrototype.generate_testN) rr __qualname__rr rnamerlocalsrrrr rs  rc@sleZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ e e D] \Z Zeee <q*dS)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. cC"tttg}tj||ddS)N NRT_increfr'r FunctionTypeVoidTypeptr_tyFunctionr mfntyrrr make_incref/zTestRefPrunePass.make_increfcCr+)N NRT_decrefr-r.r3rrr make_decref3r7zTestRefPrunePass.make_decrefcC"ttdd}tj||ddS)N rswitcherr-rr/IntTyper2r3rrr make_switcher7r7zTestRefPrunePass.make_switchercCr:)Nrrbrancherr-r=r3rrr make_brancher;r7zTestRefPrunePass.make_branchercCst}||}||}||}||}tttg}tj ||dd} | j \} d| _ i} |D] } | | | | <q4t } |D]\} }| | | || D]}|dkrc| || gqU|dkro| || gqUtdt|}|dkr| qF|dkr|\}| | |qF|d kr|\}}| |d }| || || |qF|d kr| |d }|^}}| j|| |d }t|D]\}}|||| |qqFtd|S) Nmainr-memincrefdecref unreachablerrr)default)rModuler6r9r?rAr/r0r1r2argsr'append_basic_block IRBuilderrposition_at_endcallAssertionErrorlenret_voidbranchcbranchswitch enumerateadd_casetype)r r"r#r4 incref_fn decref_fn switcher_fn brancher_fnr5r ptrbbmapbbbuilder jump_targetsaction n_targetsdstleftrightselheadtailswirrr generate_ir?sP        zTestRefPrunePass.generate_ircCs,tt|}t}||||Sr)llvmparse_assemblystrModulePassManageradd_refprune_passr)r irmodmodpmrrr apply_refpruneus  zTestRefPrunePass.apply_refprunecCs i}|D]\}}|d}|d}||d||<q|D]"\}} ||rB| dd8<||D] } || dd8<q5q |jD] } | jdkrOnqF| jD]/} || j} t| } | d}| d}|j| d|d| d |j| d|d| d qSdS) NrDrE)rDrErrBr,r8zBB )msg)rcountget functionsr'blocksrnr )r rrr$r"drvsn_increfn_decrefstatsdec_bbfr^textrrr check|s.          zTestRefPrunePass.checkcCs4|\}}}|||}||}||||dSr)rkrtr)r r!r"r#r$rqoutmodrrr r s   zTestRefPrunePass.generate_testN)rrr&rr6r9r?rArkrtrr rr'rr(rrrr r*&s6 r*c@seZdZdZdZdddZdS) BaseTestByIRrzG declare void @NRT_incref(i8* %ptr) declare void @NRT_decref(i8* %ptr) NcCsjt|jd|}t}|dur||jn|j|j|dt}||t}|||fS)N subgraph_limit)rlrmprologuerorprefprune_bitmaskdump_refprune_statsr)r rqrrrrsbeforeafterrrr rs  zBaseTestByIR.checkr)rrr&rrrrrrr rsrc@sDeZdZejjZdZddZdZ ddZ dZ dd Z d Z d d Zd S) TestPerBBzv define void @main(i8* %ptr) { call void @NRT_incref(i8* %ptr) call void @NRT_decref(i8* %ptr) ret void } cC"||j\}}||jddSNrG)r per_bb_ir_1r basicblockr rrr~rrr test_per_bb_1zTestPerBB.test_per_bb_1z 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 } cC2||j\}}||jd|dt|dS)Nzcall void @NRT_incref(i8* %ptr))r per_bb_ir_2r rassertInrnrrrr test_per_bb_2zTestPerBB.test_per_bb_2z 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 } cCr)NrG!call void @NRT_decref(i8* %other))r per_bb_ir_3r rrrnrrrr test_per_bb_3rzTestPerBB.test_per_bb_3z ; 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 } cCr)Nrr)r per_bb_ir_4r rrrnrrrr test_per_bb_4rzTestPerBB.test_per_bb_4N)rrr&rlRefPruneSubpassesPER_BBrrrrrrrrrrrrr rs   rc@PeZdZejjZdZddZdZ ddZ dZ dd Z d Z d d Zd ZddZdS) 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 } cCrr)r per_diamond_1r diamondrrrr test_per_diamond_1rzTestDiamond.test_per_diamond_1z 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 } cCrr)r per_diamond_2r rrrrr test_per_diamond_2rzTestDiamond.test_per_diamond_2a3 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 } cCrNr)r per_diamond_3r rrrrr test_per_diamond_3/rzTestDiamond.test_per_diamond_3a5 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 } cCrr)r per_diamond_4r rrrrr test_per_diamond_4CrzTestDiamond.test_per_diamond_4a0 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 } cCr)Nr)r per_diamond_5r rrrrr test_per_diamond_5XrzTestDiamond.test_per_diamond_5N)rrr&rlrDIAMONDrrrrrrrrrrrrrrr rs  rc@sDeZdZdZejjZdZddZ dZ ddZ dZ d d Z d d Zd 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 } cCr)N)rfanout_1r fanoutrrrr test_fanout_1qrzTestFanout.test_fanout_1a6 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 } cCrr)rfanout_2r rrrrr test_fanout_2rzTestFanout.test_fanout_2a} 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 } cCr)Nrfanout_3r rrrrr test_fanout_3rzTestFanout.test_fanout_3cCs&|j|jdd\}}||jddS)Nrrrrrrrr test_fanout_3_limitedsz TestFanout.test_fanout_3_limitedN)rrr&rrlrFANOUTrrrrrrrrrrrr r]s rc@r)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} cCrr)rfanout_raise_1r fanout_raiserrrr test_fanout_raise_1rz#TestFanoutRaise.test_fanout_raise_1a: 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} cCrr)rfanout_raise_2r rrrrr test_fanout_raise_2sz#TestFanoutRaise.test_fanout_raise_2a: 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 cCrr)rfanout_raise_3r rrrrr test_fanout_raise_3rz#TestFanoutRaise.test_fanout_raise_3a! 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} cCrr)rfanout_raise_4r rrrrr test_fanout_raise_4rz#TestFanoutRaise.test_fanout_raise_4a 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} cCrr)rfanout_raise_5r rrrrr test_fanout_raise_5rz#TestFanoutRaise.test_fanout_raise_5N)rrr&rlr FANOUT_RAISErrrrrrrrrrrrrrr rs r__main__)unittestllvmliterrrlllvmlite.testsrrrrrr> as_pointerr1r*rrrrrrrBrrrr s"     {CcGi