U ?hBã@sÈdZddlmZddlmZmZddlmZmZe ej ¡Z e  e d¡¡Z e  e e e e e g¡Ze e ¡e g¡Ze e e g¡Zdd„Zdd „Zd d „ZdZd d „Zdd„Zdd„Zdd„Zdd„ZdS)z% Dynamically generate the NRT module é)Úconfig)ÚtypesÚcgutils)ÚirÚbindingécCsXt |td¡}t | ¡¡}|j\}| |t  ¡¡}|  t  ||dd¡¡}|  |¡dS)zt Implement NRT_MemInfo_data_fast in the module. This allows LLVM to inline lookup of the data pointer. ZNRT_MemInfo_data_fastréN) rÚget_or_insert_functionÚmeminfo_data_tyrÚ IRBuilderÚappend_basic_blockÚargsÚbitcastÚ_meminfo_struct_typeÚ as_pointerÚloadZgepÚret)ÚmoduleÚfnÚbuilderÚptrZ struct_ptrZdata_ptr©rúN/var/www/html/venv/lib/python3.8/site-packages/numba/core/runtime/nrtdynmod.pyÚ_define_nrt_meminfo_datasÿrc Cs°t |td¡}|j d¡t | ¡¡}|j\}|  d|t  |j ¡¡}t  ||¡|  ¡W5QRX| ||jdj ¡}tjr–t |d| |¡|¡| ||g¡|  ¡dS)z, Implement NRT_incref in the module Z NRT_increfÚnoinlineú==rz*** NRT_Incref %zu [%p] N)rr Úincref_decref_tyÚ attributesÚaddrr r r Ú icmp_unsignedÚget_null_valueÚtypeÚ if_unlikelyÚret_voidrrÚ DEBUG_NRTÚprintfrÚcall)rZ atomic_incrZ fn_increfrrÚis_nullÚword_ptrrrrÚ_define_nrt_incref)s ÿ ÿr)c Cs t |td¡}|j d¡tj|t t ¡t g¡dd}t  |  ¡¡}|j \}|  d|t |j¡¡}t ||¡| ¡W5QRX| d¡| ||j dj¡}tjr¾t |d| |¡|¡| ||g¡}|  d|t |jd¡¡} t || ¡| d ¡| ||g¡W5QRX| ¡d S) z, Implement NRT_decref in the module Z NRT_decrefrZNRT_MemInfo_call_dtor©ÚnamerÚreleaserz*** NRT_Decref %zu [%p] ÚacquireN)rr rrrrÚFunctionÚ FunctionTypeÚVoidTypeÚ _pointer_typer r r rr r!r"r#Zfencerrr$r%rr&ÚConstant) rZ atomic_decrZ fn_decrefZcalldtorrrr'r(ZnewrefctZ refct_eq_0rrrÚ_define_nrt_decref?s:ÿ þ ÿÿ ÿ r3c Cs´t tt ¡g¡}tj||d |¡d}|j\}| ¡}t |¡}t  td¡}t s€|j ||||d} t ||ƒ| |ƒ} |  | ¡n0| |¡} t ||ƒ| |ƒ} | | |¡|  | ¡|S)zâDefine a llvm function for atomic increment/decrement to the given module Argument ``op`` is the operation "add"/"sub". Argument ``ordering`` is the memory ordering. The generated function returns the new value. znrt_atomic_{0}r*é©Úordering)rr/Ú _word_typerr.Úformatr r r r2Ú_disable_atomicityZ atomic_rmwÚgetattrrrÚstore) rÚopr6ÚftypeZ fn_atomicrÚbbrZONEZoldvalÚresZnewvalrrrÚ_define_atomic_inc_decos      r@c Csšt t d¡t ¡ttt ¡g¡}tj||dd}|j\}}}}| ¡}t |¡} | j ||||d} t   | | d¡\} } |   | |¡|   |  | |j¡¡|S)a¨Define a llvm function for atomic compare-and-swap. The generated function is a direct wrapper of the LLVM cmpxchg with the difference that the a int indicate success (1) or failure (0) is returned and the last argument is a output pointer for storing the old value. Note ---- On failure, the generated function behaves like an atomic load. The loaded value is stored to the last argument. é Znrt_atomic_casr*r5é)rr/ÚIntTyper7rr.r r r ZcmpxchgrZ unpack_tupler;rZzextÚ return_type) rr6r=Zfn_casrÚcmpÚreplZoldptrr>rZouttupÚoldÚokrrrÚ_define_atomic_casŠs þ  rIcCsL|j tjd¡}tj||dd}| ¡}t |¡}d}|j |t |f¡|S)zÑ Defines an abort function due to unresolved symbol. The function takes no args and will always raise an exception. It should be safe to call this function with incorrect number of arguments. rZnrt_unresolved_abortr*z6numba jitted function aborted due to unresolved symbol) Z call_convZget_function_typerÚnonerr.r r Zreturn_user_excÚ RuntimeError)ÚctxrZfntyrr>rÚmsgrrrÚ_define_nrt_unresolved_abort¥s rNcCsr| ¡}| d¡}| d¡}t|ddd}t|ddd}t|ddt|ƒt||ƒt||ƒt||ƒ||fS)zl Create an IR module defining the LLVM NRT functions. A (IR module, library) tuple is returned. ZnrtZ nrt_modulerÚ monotonicr5Úsub) ÚcodegenZcreate_libraryZcreate_ir_moduler@rIrr)r3rN)rLrQÚlibraryÚir_modZ atomic_incZ atomic_decrrrÚcreate_nrt_moduleµs      rTcCs"t|ƒ\}}| |¡| ¡|S)z‰ Compile all LLVM NRT functions and return a library containing them. The library is created using the given target context. )rTZ add_ir_moduleÚfinalize)rLrSrRrrrÚcompile_nrt_functionsÍs  rVN)Ú__doc__Z numba.corerrrZllvmliterrrCZ MACHINE_BITSr7Z PointerTyper1ZLiteralStructTyperr/r0rr rr)r3r9r@rIrNrTrVrrrrÚs.  û -