pi4ddlmZddlmZddlmZddlmZ ddl m Z ddl m Z ddlmZmZddlmZed gd ZGd d eZej,Ze j0ed Zej4eZeeej8[y)) namedtuple)finalize) nrtdynmod)binding)global_compiler_lock) typeof_impl)typesconfig) _nrt_python nrt_mstatsallocfreemi_allocmi_freec\eZdZdZedZdZedZe dZ dZ d dZ dZ y ) _Runtimecd|_y)NF)_initselfs X/opt/services/ai/voice_agent/venv/lib/python3.12/site-packages/numba/core/runtime/nrt.py__init__z_Runtime.__init__s  cV|jrytjrtjtj D]D}|j dr|}nd|z}tj |}tj||Ftj||_ d|_y)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_addresss r initializez_Runtime.initializes ::      $ $ &~~ -G!!#& ')w/I MM&) , -"77<  rc6|js d}t|y)Nz'Runtime must be initialized before use.)r RuntimeError)rmsgs r _init_guardz_Runtime._init_guard0szz;Cs# #rc,tjy)ze Shutdown the NRT Safe to be called without calling Runtime.initialize first N)rmemsys_shutdownrrshutdownz_Runtime.shutdown5s rc:|j|jS)zQ Return the Library object containing the various NRT functions. )r/r&rs rlibraryz_Runtime.library=s }}rcd|jtj||}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)rdatapyobjmis rr7z_Runtime.meminfo_newEs,    dE *r{rc|j|dkrd|d}t||rtj|}ntj|}|dk(rd|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 MemoryErrorr8)rsizesafer.r;s rr@z_Runtime.meminfo_allocOs|  !8@aHCS/ ! ((.B##D)B 7,TF.ACc" "r{rcttjtjtjtj S)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_freers rget_allocation_statsz_Runtime.get_allocation_statsfsB ! $ : : <$($B$B$D#'#@#@#BD DrN)F)__name__ __module__ __qualname__rrr+r/ staticmethodr3propertyr5r7r@rJr2rrrrsX8$ . DrrcHtjtjS)N)r MemInfoPointervoidptr)valcs rtypeof_meminforUvs    ..rN) collectionsrweakrefr _finalizenumba.core.runtimerllvmliterr#numba.core.compiler_lockrnumba.core.typing.typeofr numba.corer r r rrEobjectr_MemInfor8registerrUmemsys_use_cpython_allocatorrtsysr3r2rrrcs")("90$2'OP `Dv`DH --g// "!!#  %""# r