J/PhF TdZddlmZddlmZGddeZdZdZdZd Z d S) zW Implement a rewrite pass on a LLVM module to remove unnecessary refcount operations. ) CallVisitor)typesceZdZdZdZdZdS)_MarkNrtCallVisitorz7 A pass to mark all NRT_incref and NRT_decref. c,t|_dSN)setmarked)selfs Z/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/numba/core/removerefctpass.py__init__z_MarkNrtCallVisitor.__init__see cxt|jddtvr|j|dSdS)Nname)getattrcallee_accepted_nrtfnsr add)r instrs r visit_Callz_MarkNrtCallVisitor.visit_Calls? 5< , ,0@ @ @ KOOE " " " " " A @rN)__name__ __module__ __qualname____doc__r rrr rr s<#####rrct}|||jD]<}t|jD]%}||jvr|j|&=dSr)rvisit_Function basic_blockslist instructionsr remove)functionmarkpassbbinsts r _rewrite_functionr's"$$H H%%%#--)) - -Dx&&&&t,,, ---r) NRT_incref NRT_decrefc fd fd} |d}t|jdkrdSn#t$rYnwxYw|j}|j}|j}|D]}||sdS |sdS|D]} |  | jsdS|j D]-} | j dr| j tvrdS.dS) z Legalize the code in the module. Returns True if the module is legal for the rewrite pass that removes unnecessary refcounts. c>|}| S)zG Valid output are any type that does not need refcount )contains_nrt_meminfo)tymodeldmms r valid_outputz_legalize..valid_output,s#B--////rcN|pt|tjS)zT Valid input are any type that does not need refcount except Array. ) isinstancerArray)r-r0s r valid_inputz_legalize..valid_input3s&|B>:b%+#>#>>rnumba_args_may_always_need_nrtrFNNRT_T) get_named_metadatalenoperandsKeyErrorargtypesrestype calltypesvalues return_type functionsr startswithr) moduler/fndescr4nmdr;r<r=argtycalltyfnr0s ` @r _legalizerH%sv00000????? ''(HII s|  q 5 !      HnG I{5!! 55  < u""$$  ll63E&F&F 55 7  f % % w...uu 4s> A  A cb|j}t|j||rt|dSdS)ac 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(s); - take arguments that need refcounting except array; - call function(s) that return refcounted object. In effect, the function will not capture or create references that extend the lifetime of any refcounted objects beyond the lifetime of the function. The rewrite is performed in place. If rewrite has happened, this function returns True, otherwise, it returns False. TF)data_model_managerrHrBr')r#contextrCr/s r remove_unnecessary_nrt_usagerLcs:  $C#v..(###turN) rllvmlite.ir.transformsr numba.corerrr'rrHrLrrr rOs /..... # # # # #+ # # #---.;;;|r