o 8ήc@sddlmZddlmZddlmZddlmZ ddl m Z ddl m Z ddlmZddlmZed gd ZGd d d eZejZe ed dZeeZeeej[dS)) namedtuple)finalize) nrtdynmod)binding)global_compiler_lock) typeof_impl)types) _nrt_python nrt_mstatsallocfreemi_allocmi_freec@sZeZdZddZeddZddZeddZe d d Z d d Z dddZ ddZ dS)_RuntimecCs d|_dS)NF)_initselfr=/tmp/pip-target-vg8gfxp4/lib/python/numba/core/runtime/nrt.py__init__s z_Runtime.__init__cCs|jrdStjD]}|dr|}nd|}tj|}t||qt||_|j d|_ |j d|_ |j d|_ 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. N_NRT_nrt_atomic_addnrt_atomic_subnrt_atomic_casT)r_nrt c_helpers startswithll add_symbolrcompile_nrt_functions_libraryget_pointer_to_function_ptr_inc_ptr_dec_ptr_casmemsys_set_atomic_inc_decmemsys_set_atomic_cas)rctxpy_namec_name c_addressrrr initializes      z_Runtime.initializecCs|js d}t|dS)Nz'Runtime must be initialized before use.)r RuntimeError)rmsgrrr _init_guard5sz_Runtime._init_guardcCs tdS)ze Shutdown the NRT Safe to be called without calling Runtime.initialize first N)rmemsys_shutdownrrrrshutdown:s z_Runtime.shutdowncCs||jS)zQ Return the Library object containing the various NRT functions. )r0r"rrrrlibraryBsz_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`. )r0r meminfo_newMemInfo)rdatapyobjmirrrr4Js z_Runtime.meminfo_newFcCsb||dkrd|d}t||rt|}nt|}|dkr-d|d}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. rz,Cannot allocate a negative number of bytes: .zRequested allocation of z bytes failed.)r0 ValueErrorrmeminfo_alloc_safe meminfo_alloc MemoryErrorr5)rsizesafer/r8rrrr<Ts     z_Runtime.meminfo_alloccCs tttttdS)zw Returns a namedtuple of (alloc, free, mi_alloc, mi_free) for count of each memory operations. r ) _nrt_mstatsrmemsys_get_stats_allocmemsys_get_stats_freememsys_get_stats_mi_allocmemsys_get_stats_mi_freerrrrget_allocation_statsks z_Runtime.get_allocation_statsN)F)__name__ __module__ __qualname__rrr-r0 staticmethodr2propertyr3r4r<rErrrrrs !   rcCs ttjS)N)rMemInfoPointervoidptr)valcrrrtypeof_meminfo{s rON) collectionsrweakrefr _finalizenumba.core.runtimerllvmliterrnumba.core.compiler_lockrnumba.core.typing.typeofr numba.corerr rr@objectr_MemInfor5registerrOmemsys_use_cpython_allocatorrtsysr2rrrrs"       i