U 9%eF @sLdZddlmZddlmZGdddeZddZdZd d Zd d Z d S)zW Implement a rewrite pass on a LLVM module to remove unnecessary refcount operations. ) CallVisitor)typesc@s eZdZdZddZddZdS)_MarkNrtCallVisitorz7 A pass to mark all NRT_incref and NRT_decref. cCs t|_dSN)setmarked)selfr Y/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/numba/core/removerefctpass.py__init__sz_MarkNrtCallVisitor.__init__cCs"t|jddtkr|j|dS)Nname)getattrZcallee_accepted_nrtfnsradd)rinstrr r r visit_Callsz_MarkNrtCallVisitor.visit_CallN)__name__ __module__ __qualname____doc__r rr r r r r srcCsFt}|||jD]*}t|jD]}||jkr$|j|q$qdSr)rZvisit_FunctionZ basic_blockslistZ instructionsrremove)functionZmarkpassZbbinstr r r _rewrite_functions    r)Z NRT_increfZ NRT_decrefc sfddfdd}z|d}Wntk r:YnXt|jdkrNdS|j}|j}|j}|D]}||sddSqd|sdS|D]} | dk r| jsdSq|j D] } | j d r| j t krdSqd S) z Legalize the code in the module. Returns True if the module is legal for the rewrite pass that removes unnecessary refcounts. cs|}| S)zG Valid output are any type that does not need refcount )Zcontains_nrt_meminfo)tymodel)dmmr r valid_output,sz_legalize..valid_outputcs|pt|tjS)zT Valid input are any type that does not need refcount except Array. ) isinstancerZArray)r)rr r valid_input3sz_legalize..valid_inputZnumba_args_may_always_need_nrtrFNZNRT_T) Zget_named_metadataKeyErrorlenZoperandsargtypesrestype calltypesvalues return_typeZ functionsr startswithr) modulerfndescr!Znmdr$r%r&ZargtyZcalltyfnr )rrr _legalize%s0      r-cCs(|j}t|j||r t|dSdSdS)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. TFN)Zdata_model_managerr-r*r)rcontextr+rr r r remove_unnecessary_nrt_usagecs r/N) rZllvmlite.ir.transformsrZ numba.corerrrrr-r/r r r r s    >