U ?h@sddlmZddlmZddlmZddlmZ ddl m Z ddl m Z ddlmZmZddlmZed d d d d gZGdddeZejZe eddZeeZeeej[dS)) namedtuple)finalize) nrtdynmod)binding)global_compiler_lock) typeof_impl)typesconfig) _nrt_pythonZ nrt_mstatsallocfreemi_allocmi_freec@sZeZdZddZeddZddZeddZe d d Z d d Z dddZ ddZ dS)_RuntimecCs d|_dS)NF)_initselfrH/var/www/html/venv/lib/python3.8/site-packages/numba/core/runtime/nrt.py__init__sz_Runtime.__init__cCsh|jr dStjrttjD]2}|dr2|}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_ZNRT_T) rr Z NRT_STATS_nrtZmemsys_enable_statsZ c_helpers startswithllZ add_symbolrZcompile_nrt_functions_library)rctxZpy_nameZc_nameZ c_addressrrr initializes    z_Runtime.initializecCs|jsd}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)rZmemsys_shutdownrrrrshutdown5sz_Runtime.shutdowncCs||jS)zQ Return the Library object containing the various NRT functions. )rrrrrrlibrary=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"Es z_Runtime.meminfo_newFcCsb||dkr$d|d}t||r4t|}n t|}|dkrZd|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 ValueErrorrZmeminfo_alloc_safe meminfo_alloc MemoryErrorr#)rsizesaferr%rrrr(Os     z_Runtime.meminfo_alloccCs tttttdS)zw Returns a namedtuple of (alloc, free, mi_alloc, mi_free) for count of each memory operations. )r r r r) _nrt_mstatsrZmemsys_get_stats_allocZmemsys_get_stats_freeZmemsys_get_stats_mi_allocZmemsys_get_stats_mi_freerrrrget_allocation_statsfs z_Runtime.get_allocation_statsN)F)__name__ __module__ __qualname__rrrr staticmethodr propertyr!r"r(r-rrrrrs    rcCs ttjS)N)rZMemInfoPointerZvoidptr)valcrrrtypeof_meminfovsr5N) collectionsrweakrefrZ _finalizeZnumba.core.runtimerZllvmliterrZnumba.core.compiler_lockrZnumba.core.typing.typeofrZ numba.corerr r rr,objectrZ_MemInfor#registerr5Zmemsys_use_cpython_allocatorZrtsysr rrrrs       d