o ۾i@sddlmZddlmZddlmZddlmZ ddl m Z ddl m Z ddlmZmZddlmZed gd ZGd d d eZejZe ed dZeeZeeej[dS)) namedtuple)finalize) nrtdynmod)binding)global_compiler_lock) typeof_impl)typesconfig) _nrt_python nrt_mstatsallocfreemi_allocmi_freec@sZeZdZddZeddZddZeddZe d d Z d d Z dddZ ddZ dS)_RuntimecCs d|_dS)NF)_initselfrJ/home/ubuntu/.local/lib/python3.10/site-packages/numba/core/runtime/nrt.py__init__s z_Runtime.__init__cCsh|jrdStjr ttjD]}|dr|}nd|}tj|}t||qt ||_ d|_dS)zInitializes the NRT Must be called before any actual call to the NRT API. Safe to be called multiple times. N_NRT_T) rr NRT_STATS_nrtmemsys_enable_stats c_helpers startswithll add_symbolrcompile_nrt_functions_library)rctxpy_namec_name c_addressrrr initializes     z_Runtime.initializecCs|js d}t|dS)Nz'Runtime must be initialized before use.)r RuntimeError)rmsgrrr _init_guard0sz_Runtime._init_guardcCs tdS)ze Shutdown the NRT Safe to be called without calling Runtime.initialize first N)rmemsys_shutdownrrrrshutdown5s z_Runtime.shutdowncCs||jS)zQ Return the Library object containing the various NRT functions. )r*r"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`. )r*r meminfo_newMemInfo)rdatapyobjmirrrr.Es 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.)r* ValueErrorrmeminfo_alloc_safe meminfo_alloc MemoryErrorr/)rsizesafer)r2rrrr6Os     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_statsfs z_Runtime.get_allocation_statsN)F)__name__ __module__ __qualname__rrr'r* staticmethodr,propertyr-r.r6r?rrrrrs    rcCs ttjS)N)rMemInfoPointervoidptr)valcrrrtypeof_meminfovs rIN) collectionsrweakrefr _finalizenumba.core.runtimerllvmliterrnumba.core.compiler_lockrnumba.core.typing.typeofr numba.corerr r rr:objectr_MemInfor/registerrImemsys_use_cpython_allocatorrtsysr,rrrrs"      d