o ٴi,=@sddlZddlmZddlmZddlmZddlmZddl m m Z ddZ Gdd d ZGd d d eZed ZGd ddeeZGdddeeZGdddeZGdddeZGdddeZGdddeZedkr{edSdS)N) defaultdict)ir)binding)TestCasec#sDfdd}tjD]\}}|drd|||fVq dS)Ncs(fdd}djdj|_|S)Ncs |SN)self)fn generate_testrO/home/kuhnn/.local/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@seZdZddZdS)PassManagerMixincCs0ttj}tjddd}t||S)Nr) speed_level size_level)llvminitialize_native_targetTargetfrom_default_triplecreate_target_machinecreate_pipeline_tuning_optionscreate_pass_builder)rtmptorrr pbs zPassManagerMixin.pbN)rr __qualname__r(rrrr rs rc@s4eZdZdZddZeeD] \ZZeee<qdS)TestRefPrunePrototypez- Test that the prototype is working. cCs,|\}}}t||}|||dSr)rFanoutAlgorithmrun assertEqual)rcase_gennodesedgesexpectedgotrrr r #s z#TestRefPrunePrototype.generate_testN) rrr)rr rnamerlocalsrrrr r*s  r*c@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_increfr3r FunctionTypeVoidTypeptr_tyFunctionrmfntyrrr make_incref9zTestRefPrunePass.make_increfcCr7)N NRT_decrefr9r:r?rrr make_decref=rCzTestRefPrunePass.make_decrefcC"ttdd}tj||ddS)N rswitcherr9rr;IntTyper>r?rrr make_switcherArCzTestRefPrunePass.make_switchercCrF)Nrbrancherr9rIr?rrr make_brancherErCzTestRefPrunePass.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) Nmainr9memincrefdecref unreachablerrLr)default)rModulerBrErKrNr;r<r=r>argsr3append_basic_block IRBuilderrposition_at_endcallAssertionErrorlenret_voidbranchcbranchswitch enumerateadd_casetype)rr/r0r@ incref_fn decref_fn switcher_fn brancher_fnrAr ptrbbmapbbbuilder jump_targetsaction n_targetsdstleftrightselheadtailswirrr generate_irIsP        zTestRefPrunePass.generate_ircCs6tt|}|}|}|||||Sr)rparse_assemblystrr(getModulePassManageradd_refprune_passr,)rirmodmodr(pmrrr apply_refprunes  zTestRefPrunePass.apply_refprunecCstdd}|D]\}}|d}|d}||d||<q |D]"\}} ||rF| dd8<||D] } || dd8<q9q$|jD] } | jdkrSnqJ| jD]/} || j} t| } | d}| d }|j| d|d | d |j| d|d | d qWdS) NcSsttSr)rintrrrr sz(TestRefPrunePass.check..rQrR)rQrRrLrOr8rDzBB )msg) rrcountget functionsr3blocksrzr-)rr~r1r/drvsn_increfn_decrefstatsdec_bbfrktextrrr checks.           zTestRefPrunePass.checkcCs4|\}}}|||}||}||||dSr)rxrr)rr.r/r0r1r}outmodrrr r s   zTestRefPrunePass.generate_testN)rrr)rrBrErKrNrxrrr rr3rr4rrrr r60s6 r6c@seZdZdZdZdddZdS) BaseTestByIRrzG declare void @NRT_incref(i8* %ptr) declare void @NRT_decref(i8* %ptr) NcCstt|jd|}|}|}|dur||jn|j|j|dt}|||t}|||fS)N subgraph_limit) rryprologuer(r{r|refprune_bitmaskdump_refprune_statsr,)rr}rr~r(rbeforeafterrrr 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\}}||jddSNrT)r per_bb_ir_1r- basicblockrr~rrrr 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(ptr %ptr))r per_bb_ir_2r-rassertInrzrrrr test_per_bb_2zTestPerBB.test_per_bb_2z 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 } cCr)NrT!call void @NRT_decref(ptr %other))r per_bb_ir_3r-rrrzrrrr test_per_bb_3rzTestPerBB.test_per_bb_3z ; 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 } cCr)Nrr)r per_bb_ir_4r-rrrzrrrr test_per_bb_4rzTestPerBB.test_per_bb_4N)rrr)rRefPruneSubpassesPER_BBrrrrrrrrrrrrr rs   rc@sPeZdZejjZdZddZdZ 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_2,rzTestDiamond.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_4TrzTestDiamond.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_5irzTestDiamond.test_per_diamond_5N)rrr)rrDIAMONDrrrrrrrrrrrrrrr r s  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_1rzTestFanout.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)NrLrrrrrrr test_fanout_3_limitedsz TestFanout.test_fanout_3_limitedN)rrr)rrrFANOUTrrrrrrrrrrrr rns rc@s\eZdZejjZdZddZdZ ddZ dZ dd Z d Z d d Zd ZddZdZddZdS)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_5af 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} cCr)N)rfanout_raise_6r-rrrrr test_fanout_raise_68rz#TestFanoutRaise.test_fanout_raise_6N)rrr)rr FANOUT_RAISErrrrrrrrrrrrrrrrr rs r__main__)unittest collectionsrllvmliterrrllvmlite.testsrllvmlite.tests.refprune_prototestsrefprune_protorrrr*rJ as_pointerr=r6rrrrrrrOrrrr s*      CcG