o V۷itT@sddlZddlmZddlmZddlmZddlmm Z ddlm Z ddZ Gdd d Z Gd d d eZed ZGd ddee ZGdddee ZGdddeZGdddeZGdddeZGdddeZedkr{edSdS)N)ir)binding)TestCase)opaque_pointers_enabledc#sDfdd}tjD]\}}|drd|||fVq dS)Ncs(fdd}djdj|_|S)Ncs |SN)self)fn generate_testrR/home/ubuntu/vllm_env/lib/python3.10/site-packages/llvmlite/tests/test_refprune.pywrappeds z-_iterate_cases..wrap..wrappedzgenerated test for .) __module____name____doc__)r r r )r r wrapsz_iterate_cases..wrapcasetest_)proto__dict__items startswith)r rkcase_fnrrr _iterate_casess  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@seZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ee D] \ZZeee<q2ee D] \ZZeeed<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_incref=zTestRefPrunePass.make_increfcCr7)N NRT_decrefr9r:r?rrr make_decrefArCzTestRefPrunePass.make_decrefcC"ttdd}tj||ddS)N rswitcherr9rr;IntTyper>r?rrr make_switcherErCzTestRefPrunePass.make_switchercCrF)Nrbrancherr9rIr?rrr make_brancherIrCzTestRefPrunePass.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_irMsP        zTestRefPrunePass.generate_ircCs6tt|}|}|}|||||Sr)rparse_assemblystrr(getModulePassManageradd_refprune_passr,)rirmodmodr(pmrrr apply_refprunes  zTestRefPrunePass.apply_refprunecCs,tt|}t}||||Sr)rryrzModulePassManagerr|r,)rr}r~rrrr apply_refprune_legacys  z&TestRefPrunePass.apply_refprune_legacycCs 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) NrQrR)rQrRrLrOr8rDzBB )msg)rcountget functionsr3blocksrzr-)rr~r1r/drvsn_increfn_decrefstatsdec_bbfrktextrrr checks.          zTestRefPrunePass.checkcC4|\}}}|||}||}||||dSr)rxrrrr.r/r0r1r}outmodrrr r    zTestRefPrunePass.generate_testcCrr)rxrrrrrr generate_test_legacyrz%TestRefPrunePass.generate_test_legacy_legacyN)rrr)rrBrErKrNrxrrrr rrr3rr4rrrr r64s"6 r6c@s(eZdZdZdZdddZdddZdS) BaseTestByIRrzG declare void @NRT_incref(i8* %ptr) declare void @NRT_decref(i8* %ptr) NcCstt|jd|}|}|}|dur||jn|j|j|dt}|||t}|||fSN subgraph_limit) rryprologuer(r{r|refprune_bitmaskdump_refprune_statsr,)rr}rr~r(rbeforeafterrrr rs  zBaseTestByIR.checkcCsjt|jd|}t}|dur||jn|j|j|dt}||t}|||fSr)rryrrr|rrr,)rr}rr~rrrrrr check_legacys  zBaseTestByIR.check_legacyr)rrr)rrrrrrrr rs  rc@steZdZejjZdZddZddZ dZ ddZ d d Z e r d nd Zd dZddZe r.dndZddZddZdS) 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_1cCrr)rrr-rrrrr test_per_bb_1_legacyrzTestPerBB.test_per_bb_1_legacyz 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 } cCJ||j\}}||jdtr|dt|dS|dt|dSNzcall void @NRT_incref(ptr %ptr)zcall void @NRT_incref(i8* %ptr))r per_bb_ir_2r-rrassertInrzrrrr test_per_bb_2 zTestPerBB.test_per_bb_2cCrr)rrr-rrrrzrrrr test_per_bb_2_legacy rzTestPerBB.test_per_bb_2_legacyz 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 } z 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 } cCrNrT!call void @NRT_decref(ptr %other)!call void @NRT_decref(i8* %other))r per_bb_ir_3r-rrrrzrrrr test_per_bb_3(rzTestPerBB.test_per_bb_3cCrr)rrr-rrrrzrrrr test_per_bb_3_legacy2rzTestPerBB.test_per_bb_3_legacyz ; 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 } z ; 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 } cCrNrrr)r per_bb_ir_4r-rrrrzrrrr test_per_bb_4RrzTestPerBB.test_per_bb_4cCrr)rrr-rrrrzrrrr test_per_bb_4_legacy\rzTestPerBB.test_per_bb_4_legacyN)rrr)rRefPruneSubpassesPER_BBrrrrrrrrrrrrrrrrrr rs(     rc@sxeZdZejjZdZddZddZ dZ ddZ d d Z d Z d d ZddZdZddZddZdZddZddZdS) 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_1urzTestDiamond.test_per_diamond_1cCrr)rrr-rrrrr test_per_diamond_1_legacyyrz%TestDiamond.test_per_diamond_1_legacyz 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_2cCrr)rrr-rrrrr test_per_diamond_2_legacyrz%TestDiamond.test_per_diamond_2_legacya3 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_3rzTestDiamond.test_per_diamond_3cCrr)rrr-rrrrr test_per_diamond_3_legacyrz%TestDiamond.test_per_diamond_3_legacya5 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_4rzTestDiamond.test_per_diamond_4cCrr)rrr-rrrrr test_per_diamond_4_legacyrz%TestDiamond.test_per_diamond_4_legacya0 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 } cCrNr)r per_diamond_5r-rrrrr test_per_diamond_5rzTestDiamond.test_per_diamond_5cCrr)rrr-rrrrr test_per_diamond_5_legacyrz%TestDiamond.test_per_diamond_5_legacyN)rrr)rrDIAMONDrrrrrrrrrrrrrrrrrrrr rgs"  rc@sdeZdZdZejjZdZddZ ddZ dZ dd Z d d Z d Zd dZddZddZddZdS) 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 } cCrN)rfanout_1r-fanoutrrrr test_fanout_1rzTestFanout.test_fanout_1cCrr)rrr-rrrrr test_fanout_1_legacyrzTestFanout.test_fanout_1_legacya6 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_2cCrr)rrr-rrrrr test_fanout_2_legacy rzTestFanout.test_fanout_2_legacya} 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 } cCrNrfanout_3r-rrrrr test_fanout_3"rzTestFanout.test_fanout_3cC&|j|jdd\}}||jddSNrLrrrrrrr test_fanout_3_limited&z TestFanout.test_fanout_3_limitedcCrrrrr-rrrrr test_fanout_3_legacy,rzTestFanout.test_fanout_3_legacycCrrrrrrr test_fanout_3_limited_legacy0rz'TestFanout.test_fanout_3_limited_legacyN)rrr)rrrFANOUTrrrrrrrrrrrrrrrr rs rc@seZdZejjZdZddZddZ dZ ddZ d d Z d Z d d ZddZdZddZddZdZddZddZdZddZddZdS) 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_1Irz#TestFanoutRaise.test_fanout_raise_1cCrr)rrr-rrrrr test_fanout_raise_1_legacyMrz*TestFanoutRaise.test_fanout_raise_1_legacya: 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_2az#TestFanoutRaise.test_fanout_raise_2cCrr)rrr-rrrrr test_fanout_raise_2_legacygrz*TestFanoutRaise.test_fanout_raise_2_legacya: 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_3}rz#TestFanoutRaise.test_fanout_raise_3cCrr)rrr-rrrrr test_fanout_raise_3_legacyrz*TestFanoutRaise.test_fanout_raise_3_legacya! 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_4cCrr)rrr-rrrrr test_fanout_raise_4_legacyrz*TestFanoutRaise.test_fanout_raise_4_legacya 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_5cCrr)rrr-rrrrr test_fanout_raise_5_legacyrz*TestFanoutRaise.test_fanout_raise_5_legacyaf 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} cCrN)rfanout_raise_6r-rrrrr test_fanout_raise_6rz#TestFanoutRaise.test_fanout_raise_6cCrr)rrr-rrrrr test_fanout_raise_6_legacyrz*TestFanoutRaise.test_fanout_raise_6_legacyN)rrr)rr FANOUT_RAISErrrrrrrrrrrrrrrrrrrrrrr r7s( r__main__)unittestllvmliterrrllvmlite.testsrllvmlite.tests.refprune_prototestsrefprune_protorrrrr*rJ as_pointerr=r6rrrrrrrOrrrr s,       $wY#