B \Y@sddlmZmZmZddlmZddlmZddlm Z ddl m Z ddlmZddlmZdd lmZdd lmZed d d ddgZGdddeZejZeeddZeeZe eej[dS))print_functionabsolute_importdivision) namedtuple) nrtdynmod)binding)finalize)global_compiler_lock) typeof_impl)types) _nrt_pythonZ nrt_mstatsallocfreemi_allocmi_freec@sZeZdZddZeddZddZeddZe d d Z d d Z dddZ ddZ dS)_RuntimecCs d|_dS)NF)_init)selfr0lib/python3.7/site-packages/numba/runtime/nrt.py__init__sz_Runtime.__init__cCs|jr dSx,tjD]"}d|}tj|}t||qWt||_|jd|_ |jd|_ |jd|_ t |j |j t |j d|_dS)zInitializes the NRT Must be called before any actual call to the NRT API. Safe to be called multiple times. NZNRT_Znrt_atomic_addZnrt_atomic_subZnrt_atomic_casT)r_nrtZ c_helpersllZ add_symbolrZcompile_nrt_functions_libraryZget_pointer_to_functionZ_ptr_incZ_ptr_decZ_ptr_casZmemsys_set_atomic_inc_decZmemsys_set_atomic_cas)rZctxZpy_nameZc_nameZ c_addressrrr initializes    z_Runtime.initializecCs|jsd}t|dS)Nz'Runtime must be initialized before use.)r RuntimeError)rmsgrrr _init_guard3sz_Runtime._init_guardcCs tdS)ze Shutdown the NRT Safe to be called without calling Runtime.initialize first N)rZmemsys_shutdownrrrrshutdown8sz_Runtime.shutdowncCs||jS)zQ Return the Library object containing the various NRT functions. )rr)rrrrlibrary@sz_Runtime.librarycCs|t||}t|S)z Returns a MemInfo object that tracks memory at `data` owned by `pyobj`. MemInfo will acquire a reference on `pyobj`. The release of MemInfo will release a reference on `pyobj`. )rr meminfo_newMemInfo)rdataZpyobjmirrrr!Hs z_Runtime.meminfo_newFcCs*||rt|}n t|}t|S)a Allocate a new memory of `size` bytes and returns a MemInfo object that tracks the allocation. When there is no more reference to the MemInfo object, the underlying memory will be deallocated. If `safe` flag is True, the memory is allocated using the `safe` scheme. This is used for debugging and testing purposes. See `NRT_MemInfo_alloc_safe()` in "nrt.h" for details. )rrZmeminfo_alloc_safe meminfo_allocr")rsizeZsafer$rrrr%Rs   z_Runtime.meminfo_alloccCs tttttdS)zw Returns a namedtuple of (alloc, free, mi_alloc, mi_free) for count of each memory operations. )rrrr) _nrt_mstatsrZmemsys_get_stats_allocZmemsys_get_stats_freeZmemsys_get_stats_mi_allocZmemsys_get_stats_mi_free)rrrrget_allocation_statscsz_Runtime.get_allocation_statsN)F)__name__ __module__ __qualname__rr rr staticmethodrpropertyr r!r%r(rrrrrs    rcCs ttjS)N)r ZMemInfoPointerZvoidptr)valcrrrtypeof_meminfossr0N) Z __future__rrr collectionsrrZllvmliterrZ numba.utilsr Z _finalizeZnumba.compiler_lockr Znumba.typing.typeofr Znumbar r rr'objectrZ_MemInfor"registerr0Zmemsys_use_cpython_allocatorZrtsysrrrrrs         _