o £¸¢i,=ã@súddlZddlmZddlmZddlmZddlmZddl m m Z dd„Z Gdd „d ƒZGd d „d eƒZe d ¡ ¡ZGd d„deeƒZGdd„deeƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZedkr{e ¡dSdS)éN)Ú defaultdict)Úir)Úbinding)ÚTestCasec#sD‡fdd„}tj ¡D]\}}| d¡rd|›||ƒfVq dS)Ncs(‡‡fdd„}dˆj›dˆj›|_|S)Ncs ˆ|ˆƒS©N©)Úself)ÚfnÚ generate_testrúZ/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/llvmlite/tests/test_refprune.pyÚwrapped s z-_iterate_cases..wrap..wrappedzgenerated test for Ú.)Ú __module__Ú__name__Ú__doc__)r r ©r )r r Úwrap sz_iterate_cases..wrapÚcaseÚtest_)ÚprotoÚ__dict__ÚitemsÚ startswith)r rÚkÚcase_fnrrr Ú_iterate_cases s€  €þrc@seZdZdd„ZdS)ÚPassManagerMixincCs0t ¡tj ¡ ¡}tjddd}t ||¡S)Nr)Ú speed_levelÚ size_level)ÚllvmÚinitialize_native_targetÚTargetÚfrom_default_tripleÚcreate_target_machineÚcreate_pipeline_tuning_optionsÚcreate_pass_builder)rÚtmÚptorrr Úpbs zPassManagerMixin.pbN)rrÚ __qualname__r(rrrr rs rc@s4eZdZdZdd„ZeeƒD] \ZZeeƒe<qdS)ÚTestRefPrunePrototypez- Test that the prototype is working. cCs,|ƒ\}}}t ||¡ ¡}| ||¡dSr)rÚFanoutAlgorithmÚrunÚ assertEqual)rÚcase_genÚnodesÚedgesÚexpectedÚgotrrr r #s z#TestRefPrunePrototype.generate_testN) rrr)rr rÚnamerÚlocalsrrrr r*s  ÿr*éc@sleZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z e e ƒD] \Z Zeeƒe <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ó"t t ¡tg¡}tj||ddS)NÚ NRT_incref©r3©rÚ FunctionTypeÚVoidTypeÚptr_tyÚFunction©rÚmÚfntyrrr Ú make_incref9ózTestRefPrunePass.make_increfcCr7)NÚ NRT_decrefr9r:r?rrr Ú make_decref=rCzTestRefPrunePass.make_decrefcCó"t t d¡d¡}tj||ddS)Né rÚswitcherr9©rr;ÚIntTyper>r?rrr Ú make_switcherArCzTestRefPrunePass.make_switchercCrF)NérÚbrancherr9rIr?rrr Ú make_brancherErCzTestRefPrunePass.make_branchercCs¶t ¡}| |¡}| |¡}| |¡}| |¡}t t ¡tg¡}tj ||dd} | j \} d| _ i} |D] } |   | ¡| | <q4t  ¡} | ¡D]’\} }|  | | ¡|| D]}|dkrc|  || g¡qU|dkro|  || g¡qUtdƒ‚t|ƒ}|dkr€|  ¡qF|dkr|\}|  | |¡qF|d kr©|\}}|  |d ¡}|  || || |¡qF|d krÕ|  |d ¡}|^}}| j|| |d }t|ƒD]\}}| | |¡| |¡qÄqFtdƒ‚|S) NÚmainr9ÚmemÚincrefÚdecrefÚ unreachablerrLér)Údefault)rÚModulerBrErKrNr;r<r=r>Úargsr3Úappend_basic_blockÚ IRBuilderrÚposition_at_endÚcallÚAssertionErrorÚlenÚret_voidÚbranchÚcbranchÚswitchÚ enumerateÚadd_caseÚtype)rr/r0r@Ú incref_fnÚ decref_fnÚ switcher_fnÚ brancher_fnrAr ÚptrÚbbmapÚbbÚbuilderÚ jump_targetsÚactionÚ n_targetsÚdstÚleftÚrightÚselÚheadÚtailÚswÚirrr Ú generate_irIsP        ÿzTestRefPrunePass.generate_ircCs6t t|ƒ¡}| ¡}| ¡}| ¡| ||¡|Sr)rÚparse_assemblyÚstrr(ÚgetModulePassManagerÚadd_refprune_passr,)rÚirmodÚmodr(Ú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) NcSsttƒSr)rÚintrrrr ÚŒsz(TestRefPrunePass.check..rQrR)rQrRrLrOr8rDzBB )Úmsg) rrÚcountÚgetÚ functionsr3Úblocksrzr-)rr~r1r/ÚdrÚvsÚn_increfÚn_decrefÚstatsÚdec_bbÚfrkÚtextrrr Úcheck‡s.     €  ÿ    úzTestRefPrunePass.checkcCs4|ƒ\}}}| ||¡}| |¡}| |||¡dSr)rxr€r)rr.r/r0r1r}Úoutmodrrr r ¦s   zTestRefPrunePass.generate_testN)rrr)rrBrErKrNrxr€rr rr3rr4rrrr r60s6 ÿr6c@seZdZdZdZddd„ZdS)Ú BaseTestByIRrzG declare void @NRT_incref(i8* %ptr) declare void @NRT_decref(i8* %ptr) NcCstt |j›d|›¡}| ¡}| ¡}|dur| |j¡n|j|j|dt ¡}| ||¡t ¡}|||fS)NÚ ©Úsubgraph_limit) rryÚprologuer(r{r|Úrefprune_bitmaskÚdump_refprune_statsr,)rr}r•r~r(rÚbeforeÚafterrrr r¹sÿ  zBaseTestByIR.checkr)rrr)r—r–rrrrr r’±sr’c@sDeZdZejjZdZdd„ZdZ dd„Z 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¡\}}| |jd¡dS©NrT)rÚ per_bb_ir_1r-Ú basicblock©rr~rŒrrr Ú test_per_bb_1ÓózTestPerBB.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 } cCó2| |j¡\}}| |jd¡| dt|ƒ¡dS)Nézcall void @NRT_incref(ptr %ptr))rÚ per_bb_ir_2r-rŸÚassertInrzr rrr Ú test_per_bb_2âózTestPerBB.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-rŸr¦rzr rrr Ú test_per_bb_3òr¨zTestPerBB.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£)Nr¤r©)rÚ per_bb_ir_4r-rŸr¦rzr rrr Ú test_per_bb_4r¨zTestPerBB.test_per_bb_4N)rrr)rÚRefPruneSubpassesÚPER_BBr—ržr¡r¥r§rªr«r¬r­rrrr r›Ès   r›c@sPeZdZejjZdZdd„ZdZ dd„Z dZ dd „Z d Z d d „Zd Zdd„ZdS)Ú 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 } cCrœr)rÚ per_diamond_1r-Údiamondr rrr Útest_per_diamond_1r¢zTestDiamond.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 } cCrœr)rÚ per_diamond_2r-r²r rrr Útest_per_diamond_2,r¢zTestDiamond.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 } cCrœ©Nr)rÚ per_diamond_3r-r²r rrr Útest_per_diamond_3@r¢zTestDiamond.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 } cCrœr)rÚ per_diamond_4r-r²r rrr Útest_per_diamond_4Tr¢zTestDiamond.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-r²r rrr Útest_per_diamond_5ir¢zTestDiamond.test_per_diamond_5N)rrr)rr®ÚDIAMONDr—r±r³r´rµr·r¸r¹rºr»r¼rrrr r° s  r°c@sDeZdZdZejjZdZdd„Z dZ dd„Z 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é)rÚfanout_1r-Úfanoutr rrr Ú test_fanout_1‚r¢zTestFanout.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 } cCrœr¶)rÚfanout_2r-rÁr rrr Ú test_fanout_2”r¢zTestFanout.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œ)Né©rÚfanout_3r-rÁr rrr Ú test_fanout_3ªr¢zTestFanout.test_fanout_3cCs&|j|jdd\}}| |jd¡dS)NrLr”rrÆr rrr Útest_fanout_3_limited®sz TestFanout.test_fanout_3_limitedN)rrr)rrr®ÚFANOUTr—rÀrÂrÃrÄrÇrÈrÉrrrr r¾ns r¾c@s\eZdZejjZdZdd„ZdZ dd„Z dZ dd „Z d Z d d „Zd Zdd„ZdZdd„ZdS)Ú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} cCrœr)rÚfanout_raise_1r-Ú fanout_raiser rrr Útest_fanout_raise_1Çr¢z#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} cCrœr¶)rÚfanout_raise_2r-rÍr rrr Útest_fanout_raise_2Ûsz#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 cCrœr)rÚfanout_raise_3r-rÍr rrr Útest_fanout_raise_3ñr¢z#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} cCrœr¶)rÚfanout_raise_4r-rÍr rrr Útest_fanout_raise_4r¢z#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} cCrœr)rÚfanout_raise_5r-rÍr rrr Útest_fanout_raise_5r¢z#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é)rÚfanout_raise_6r-rÍr rrr Útest_fanout_raise_68r¢z#TestFanoutRaise.test_fanout_raise_6N)rrr)rr®Ú FANOUT_RAISEr—rÌrÎrÏrÐrÑrÒrÓrÔrÕrÖrØrÙrrrr r˵s rËÚ__main__)ÚunittestÚ collectionsrÚllvmliterrrÚllvmlite.testsrÚllvmlite.tests.refprune_protoÚtestsÚrefprune_protorrrr*rJÚ as_pointerr=r6r’r›r°r¾rËrrOrrrr Ús*      CcG ÿ