σ \ K]c@`s+dZddlmZmZmZddlmZddlmZm Z ddl m Z m Z e Ze jeƒZe je jdƒƒZe jeeeeegƒZe je jƒegƒZe jeegƒZd„Zd„Zd „ZdZd „Zd „Zd „Zd „Z d„Z!dS(s% Dynamically generate the NRT module i(tprint_functiontabsolute_importtdivision(t MACHINE_BITS(tcgutilsttypes(tirtbindingicC`s€|jtddƒ}tj|jƒƒ}|j\}|j|tjƒƒ}|j t j ||ddƒƒ}|j |ƒdS(st Implement NRT_MemInfo_data_fast in the module. This allows LLVM to inline lookup of the data pointer. tnametNRT_MemInfo_data_fastiiN( tget_or_insert_functiontmeminfo_data_tyRt IRBuildertappend_basic_blocktargstbitcastt_meminfo_struct_typet as_pointertloadRtgeptret(tmoduletfntbuildertptrt struct_ptrtdata_ptr((s6lib/python2.7/site-packages/numba/runtime/nrtdynmod.pyt_define_nrt_meminfo_datas   !cC`sι|jtddƒ}|jjdƒtj|jƒƒ}|j\}|jd|t j |j ƒƒ}t j ||ƒ|j ƒWdQXtr²t j|d|j|ƒ|ƒn|j||j||jdj ƒgƒ|j ƒdS(s, Implement NRT_incref in the module Rt NRT_increftnoinlines==Ns*** NRT_Incref %zu [%p] i(R tincref_decref_tyt attributestaddRR R Rt icmp_unsignedRtget_null_valuettypet if_unlikelytret_voidt _debug_printtprintfRtcallR(Rt atomic_incrt fn_increfRRtis_null((s6lib/python2.7/site-packages/numba/runtime/nrtdynmod.pyt_define_nrt_incref,s   ! )c C`s|jtddƒ}|jjdƒ|jtjtjƒtgƒddƒ}tj |j ƒƒ}|j \}|j d|t j|jƒƒ}t j||ƒ|jƒWdQXtrάt j|d|j|ƒ|ƒn|jdƒ|j||j||j d jƒgƒ}|j d|tj|jd ƒƒ}t j||ƒ%|jd ƒ|j||gƒWdQX|jƒdS( s, Implement NRT_decref in the module Rt NRT_decrefRtNRT_MemInfo_call_dtors==Ns*** NRT_Decref %zu [%p] treleaseitacquire(R RRR t add_functionRt FunctionTypetVoidTypet _pointer_typeR R RR!RR"R#R$R%R&R'RtfenceR(RtConstant( Rt atomic_decrt fn_decreftcalldtorRRR+tnewrefctt refct_eq_0((s6lib/python2.7/site-packages/numba/runtime/nrtdynmod.pyt_define_nrt_decrefAs,  !  !   "  c C`stjttjƒgƒ}tj||ddj|ƒƒ}|j\}|jƒ}tj|ƒ}tj tdƒ}t sΎ|j |||d|ƒ} t ||ƒ| |ƒ} |j | ƒnD|j|ƒ} t ||ƒ| |ƒ} |j| |ƒ|j | ƒ|S(sβ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. Rsnrt_atomic_{0}itordering(RR2t _word_typeRtFunctiontformatRR R R6t_disable_atomicityt atomic_rmwtgetattrRRtstore( RtopR=tftypet fn_atomicRtbbRtONEtoldvaltrestnewval((s6lib/python2.7/site-packages/numba/runtime/nrtdynmod.pyt_define_atomic_inc_decms!   c C`sαtjtjdƒtjƒtttjƒgƒ}tj||ddƒ}|j\}}}}|jƒ}tj|ƒ} | j |||d|ƒ} t j | | dƒ\} } | j | |ƒ| j | j| |jƒƒ|S(s¨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. i Rtnrt_atomic_casR=i(RR2tIntTypeR>RR?RR R tcmpxchgRt unpack_tupleRDRtzextt return_type( RR=RFtfn_casRtcmptrepltoldptrRHRtouttuptoldtok((s6lib/python2.7/site-packages/numba/runtime/nrtdynmod.pyt_define_atomic_casˆs  cC`sn|jjtjdƒ}tj||ddƒ}|jƒ}tj|ƒ}d}|jj|t |fƒ|S(sΡ 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. Rtnrt_unresolved_aborts6numba jitted function aborted due to unresolved symbol(( t call_convtget_function_typeRtnoneRR?R R treturn_user_exct RuntimeError(tctxRtfntyRRHRtmsg((s6lib/python2.7/site-packages/numba/runtime/nrtdynmod.pyt_define_nrt_unresolved_abort£s cC`sŸ|jƒ}|jdƒ}|jdƒ}t|dddƒ}t|dddƒ}t|ddƒt|ƒt||ƒt||ƒt||ƒ||fS(sl Create an IR module defining the LLVM NRT functions. A (IR module, library) tuple is returned. tnrtt nrt_moduleR R=t monotonictsub( tcodegentcreate_librarytcreate_ir_moduleRMR[RR,R<Re(RbRjtlibrarytir_modt atomic_inct atomic_dec((s6lib/python2.7/site-packages/numba/runtime/nrtdynmod.pytcreate_nrt_module³s     cC`s-t|ƒ\}}|j|ƒ|jƒ|S(s‰ Compile all LLVM NRT functions and return a library containing them. The library is created using the given target context. (Rqt add_ir_moduletfinalize(RbRnRm((s6lib/python2.7/site-packages/numba/runtime/nrtdynmod.pytcompile_nrt_functionsΛs  N("t__doc__t __future__RRRt numba.configRtnumbaRRtllvmliteRRtFalseR&ROR>t PointerTypeR4tLiteralStructTypeRR2R3RR RR,R<RARMR[ReRqRt(((s6lib/python2.7/site-packages/numba/runtime/nrtdynmod.pyts0    )