o ۾iH>@szddlZddlmZddlmZddlmZmZmZm Z ddl m Z eddZ e dd d Z e d d d ZGdddeZdS)N) namedtuple)ir)typescgutilserrorsconfig) PYVERSION_NRT_Meminfo_Functions)alloc alloc_dtor alloc_alignedNRT_MemInfo_alloc_safeNRT_MemInfo_alloc_dtor_safeNRT_MemInfo_alloc_safe_alignedNRT_MemInfo_allocNRT_MemInfo_alloc_dtorNRT_MemInfo_alloc_alignedc@s(eZdZdZddZddZddZedd Zd d Zd d Z eddZ ddZ eddZ ddZ eddZddZeddZddZeddZd d!Zed"d#Zd$d%Zed&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3Zd4d5Zd6d7Zd8d9Zd:d;Z dd?Z"d@S)A NRTContextzF An object providing access to NRT APIs in the lowering pass. cCs&||_||_tjrt|_dSt|_dS)N)_context_enabledr DEBUG_NRT_NRT_MEMINFO_SAFE_API _meminfo_api_NRT_MEMINFO_DEFAULT_API)selfcontextenabledrN/home/ubuntu/.local/lib/python3.10/site-packages/numba/core/runtime/context.py__init__s   zNRTContext.__init__cCs|jstddS)NzNRT required but not enabled)rrNumbaRuntimeError)rrrr _require_nrt(s zNRTContext._require_nrtcstfdd}|S)Ncs4||g|Ri|}d}tj|j|||d|S)Nz'Allocation failed (probably too large).)msg)rguard_memory_errorr)rbuilderargskwargsmemptrr"funcrrwrap-sz+NRTContext._check_null_result..wrap) functoolswraps)r)r*rr(r_check_null_result,szNRTContext._check_null_resultcC |||S)z Low-level allocate a new memory area of `size` bytes. The result of the call is checked and if it is NULL, i.e. allocation failed, then a MemoryError is raised. )allocate_uncheckedrr$sizerrrallocate5s zNRTContext.allocatecCH||j}ttjtjg}t||d}|j d| ||gS)z Low-level allocate a new memory area of `size` bytes. Returns NULL to indicate error/failure to allocate. NRT_Allocatenoalias r!moduler FunctionTyper voidptr_tintp_tget_or_insert_function return_value add_attributecallrr$r1modfntyfnrrrr/>s  zNRTContext.allocate_uncheckedcCs>||j}tttjg}t||d}|||gS)zI Low-level free a memory area allocated with allocate(). NRT_Free r!r7rr8VoidTyperr9r;r>)rr$ptrr@rArBrrrfreeKs zNRTContext.freecCr.)z Allocate a new MemInfo with a data payload of `size` bytes. A pointer to the MemInfo is returned. The result of the call is checked and if it is NULL, i.e. allocation failed, then a MemoryError is raised. )meminfo_alloc_uncheckedr0rrr meminfo_allocVs zNRTContext.meminfo_alloccCsL||j}ttjtjg}t|||jj }|j d| ||gS)z Allocate a new MemInfo with a data payload of `size` bytes. A pointer to the MemInfo is returned. Returns NULL to indicate error/failure to allocate. r5) r!r7rr8rr9r:r;rr r<r=r>r?rrrrHbs z"NRTContext.meminfo_alloc_uncheckedcC||||S)a Allocate a new MemInfo with a data payload of `size` bytes and a destructor `dtor`. A pointer to the MemInfo is returned. The result of the call is checked and if it is NULL, i.e. allocation failed, then a MemoryError is raised. )meminfo_alloc_dtor_uncheckedrr$r1dtorrrrmeminfo_alloc_dtors zNRTContext.meminfo_alloc_dtorcCs\||j}ttjtjtjg}t|||jj }|j d| ||| |tjgS)z Allocate a new MemInfo with a data payload of `size` bytes and a destructor `dtor`. A pointer to the MemInfo is returned. Returns NULL to indicate error/failure to allocate. r5)r!r7rr8rr9r:r;rr r<r=r>bitcastrr$r1rMr@rArBrrrrKs    z'NRTContext.meminfo_alloc_dtor_uncheckedcCrJ)a{ Allocate a new MemInfo with an aligned data payload of `size` bytes. The data pointer is aligned to `align` bytes. `align` can be either a Python int or a LLVM uint32 value. A pointer to the MemInfo is returned. The result of the call is checked and if it is NULL, i.e. allocation failed, then a MemoryError is raised. )meminfo_alloc_aligned_unchecked)rr$r1alignrrrmeminfo_alloc_aligneds z NRTContext.meminfo_alloc_alignedcCs||j}td}ttjtj|g}t|||j j }|j dt |tr3|jtj|}n |j|ks)rr$r1rSr@u32rArBrrrrRs    z*NRTContext.meminfo_alloc_aligned_uncheckedcCr.)a Allocate a MemInfo pointing to a variable-sized data area. The area is separately allocated (i.e. two allocations are made) so that re-allocating it doesn't change the MemInfo's address. A pointer to the MemInfo is returned. The result of the call is checked and if it is NULL, i.e. allocation failed, then a MemoryError is raised. )meminfo_new_varsize_uncheckedr0rrrmeminfo_new_varsizes zNRTContext.meminfo_new_varsizecCr3)aI Allocate a MemInfo pointing to a variable-sized data area. The area is separately allocated (i.e. two allocations are made) so that re-allocating it doesn't change the MemInfo's address. A pointer to the MemInfo is returned. Returns NULL to indicate error/failure to allocate. NRT_MemInfo_new_varsizer5r6r?rrrr]s  z(NRTContext.meminfo_new_varsize_uncheckedcCrJ)a8 Like meminfo_new_varsize() but also set the destructor for cleaning up references to objects inside the allocation. A pointer to the MemInfo is returned. The result of the call is checked and if it is NULL, i.e. allocation failed, then a MemoryError is raised. )"meminfo_new_varsize_dtor_uncheckedrLrrrmeminfo_new_varsize_dtorrOz#NRTContext.meminfo_new_varsize_dtorcCsB||j}ttjtjtjg}t||d}||||gS)z Like meminfo_new_varsize() but also set the destructor for cleaning up references to objects inside the allocation. A pointer to the MemInfo is returned. Returns NULL to indicate error/failure to allocate. NRT_MemInfo_new_varsize_dtor) r!r7rr8rr9r:r;r>rQrrrr`s  z-NRTContext.meminfo_new_varsize_dtor_uncheckedcCrJ)a: Allocate a new data area for a MemInfo created by meminfo_new_varsize(). The new data pointer is returned, for convenience. Contrary to realloc(), this always allocates a new area and doesn't copy the old data. This is useful if resizing a container needs more than simply copying the data area (e.g. for hash tables). The old pointer will have to be freed with meminfo_varsize_free(). The result of the call is checked and if it is NULL, i.e. allocation failed, then a MemoryError is raised. )meminfo_varsize_alloc_uncheckedrr$meminfor1rrrmeminfo_varsize_allocsz NRTContext.meminfo_varsize_alloccC||||dS)a Allocate a new data area for a MemInfo created by meminfo_new_varsize(). The new data pointer is returned, for convenience. Contrary to realloc(), this always allocates a new area and doesn't copy the old data. This is useful if resizing a container needs more than simply copying the data area (e.g. for hash tables). The old pointer will have to be freed with meminfo_varsize_free(). Returns NULL to indicate error/failure to allocate. NRT_MemInfo_varsize_alloc_call_varsize_allocrdrrrrc s z*NRTContext.meminfo_varsize_alloc_uncheckedcCrJ)a Reallocate a data area allocated by meminfo_new_varsize(). The new data pointer is returned, for convenience. The result of the call is checked and if it is NULL, i.e. allocation failed, then a MemoryError is raised. )!meminfo_varsize_realloc_uncheckedrdrrrmeminfo_varsize_reallocs z"NRTContext.meminfo_varsize_realloccCrg)z Reallocate a data area allocated by meminfo_new_varsize(). The new data pointer is returned, for convenience. Returns NULL to indicate error/failure to allocate. NRT_MemInfo_varsize_reallocrirdrrrrk&s z,NRTContext.meminfo_varsize_realloc_uncheckedcCsD||j}tttjtjg}t||d}||||fS)z Free a memory area allocated for a NRT varsize object. Note this does *not* free the NRT object itself! NRT_MemInfo_varsize_freerD)rr$rerFr@rArBrrrmeminfo_varsize_free0s  zNRTContext.meminfo_varsize_freecCsN||j}ttjtjtjg}t|||}|j d| |||gS)Nr5r6)rr$rer1funcnamer@rArBrrrrj>s  zNRTContext._call_varsize_alloccCs6|ddlm}|j}t||d}|||gS)z Given a MemInfo pointer, return a pointer to the allocated data managed by it. This works for MemInfos allocated with all the above methods. r)meminfo_data_tyNRT_MemInfo_data_fast)r!numba.core.runtime.nrtdynmodrqr7rr;r>)rr$rerqr@rBrrr meminfo_dataHs zNRTContext.meminfo_datac Csn|jj|}||}g}|r|||}|||f|D]\}} | |} |||| } || q |S)zCReturn a list of *(type, meminfo)* inside the given value. )rdata_model_managertraversehas_nrt_meminfoget_nrt_meminfoappend get_meminfosextend) rr$tyval datamodelmembersmeminfosmimtypgetterfieldinner_meminfosrrrrzWs     zNRTContext.get_meminfosc Csv|ddlm}||||}|D]%\}}|j} t| ||} | jdd| jdd| | |gqdS)zGCall function of *funcname* on every meminfo found in *value*. r)incref_decref_tyr5 nocaptureN) r!rsrrzr7rr;r%r=r>) rr$typvaluerprr_rr@rBrrr_call_incref_decrefhs  zNRTContext._call_incref_decrefcC||||ddS)zG Recursively incref the given *value* and its members. NRT_increfNrrr$rrrrrincrefzzNRTContext.increfcCr)zG Recursively decref the given *value* and its members. NRT_decrefNrrrrrdecrefrzNRTContext.decrefcCs6|ttjd}|j}t||d}||dS)zGCalls NRT_get_api(), which returns the NRT API function table. r NRT_get_api)r!rr8rr9r7r;r>)rr$rAr@rBrrr get_nrt_apis  zNRTContext.get_nrt_apicCsp|j}|j}||}|j}|t||}tdkr6||| |Wd|S1s1wY|S)z(Check if an exception is raised ) N) r call_convcheck_try_statusexcinfonot_ris_nullrif_then eh_end_try)rr$ctxcc trystatusr has_raisedrrreh_checks    zNRTContext.eh_checkcC|j}|j}||dS)zBegin a try-block. N)rrset_try_statusrr$rrrrreh_tryzNRTContext.eh_trycCr)zEnd a try-block N)rrunset_try_statusrrrrrrzNRTContext.eh_end_tryN)#__name__ __module__ __qualname____doc__rr!r-r2r/rGrIrHrNrKrTrRr^r]rar`rfrcrlrkrorjrtrzrrrrrrrrrrrrsR                    r)r+ collectionsrllvmliter numba.corerrrrnumba.core.utilsrr rrobjectrrrrrs"