ó \ K]c@@swdZddlmZmZddlmZddlmZdefd„ƒYZd„Z d Z d „Z d „Z d S(sS Implement a rewrite pass on LLVM module to remove unnecessary refcount operation. i(tabsolute_importtprint_function(t CallVisitor(ttypest_MarkNrtCallVisitorcB@s eZdZd„Zd„ZRS(s7 A pass to mark all NRT_incref and NRT_decref. cC@stƒ|_dS(N(tsettmarked(tself((s<lib/python2.7/site-packages/numba/targets/removerefctpass.pyt__init__scC@s2t|jddƒdkr.|jj|ƒndS(Ntnamett NRT_increft NRT_decref(R R (tgetattrtcalleeRtadd(Rtinstr((s<lib/python2.7/site-packages/numba/targets/removerefctpass.pyt visit_Calls(t__name__t __module__t__doc__RR(((s<lib/python2.7/site-packages/numba/targets/removerefctpass.pyR s cC@sptƒ}|j|ƒ|j}xJ|jD]?}x6t|jƒD]%}||kr?|jj|ƒq?q?Wq)WdS(N(Rtvisit_FunctionRt basic_blockstlistt instructionstremove(tfunctiontmarkpassRtbbtinst((s<lib/python2.7/site-packages/numba/targets/removerefctpass.pyt_rewrite_functions    R R c @sá‡fd†‰‡fd†}|j}|j}|j}x|D]}||ƒs@tSq@Wˆ|ƒsjtSx4|jƒD]&}|dk rwˆ|jƒ rwtSqwWx9|jD].} | jj dƒr«| jt krÙtSq«q«Wt S(s Legalize the code in the module. Returns True if the module is legal for the rewrite pass that remove unnecessary refcount. c@sˆ|}|jƒ S(sG Valid output are any type that does not need refcount (tcontains_nrt_meminfo(ttytmodel(tdmm(s<lib/python2.7/site-packages/numba/targets/removerefctpass.pyt valid_output.s c@sˆ|ƒpt|tjƒS(sT Valid input are any type that does not need refcount except Array. (t isinstanceRtArray(R (R#(s<lib/python2.7/site-packages/numba/targets/removerefctpass.pyt valid_input5stNRT_N( targtypestrestypet calltypestFalsetvaluestNonet return_typet functionsR t startswitht_accepted_nrtfnstTrue( tmoduleR"tfndescR&R(R)R*targtytcalltytfn((R"R#s<lib/python2.7/site-packages/numba/targets/removerefctpass.pyt _legalize's$       cC@s4|j}t|j||ƒr,t|ƒtStSdS(sS Remove unnecessary NRT incref/decref in the given LLVM function. It uses highlevel type info to determine if the function does not need NRT. Such a function does not: - return array object; - take arguments that need refcount except array; - call function that return refcounted object. In effect, the function will not capture or create references that extend the lifetime of any refcounted objects beyound the lifetime of the function. The rewrite performs inplace. If rewrite has happen, this function return True. Otherwise, return False. N(tdata_model_managerR8R3RR2R+(RtcontextR4R"((s<lib/python2.7/site-packages/numba/targets/removerefctpass.pytremove_unnecessary_nrt_usageVs   N(R R ( Rt __future__RRtllvmlite.ir.transformsRtnumbaRRRR1R8R;(((s<lib/python2.7/site-packages/numba/targets/removerefctpass.pyts  /