o 8ήc9@s<ddlZddlmZddlmZmZmZGdddeZdS)N)ir)typescgutilserrorsc@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||_||_dS)N)_context_enabled)selfcontextenabledr A/tmp/pip-target-vg8gfxp4/lib/python/numba/core/runtime/context.py__init__ s zNRTContext.__init__cCs|jstddS)NzNRT required but not enabled)rrNumbaRuntimeError)r r r r _require_nrts zNRTContext._require_nrtcstfdd}|S)Ncs4||g|Ri|}d}tj|j|||d|S)Nz'Allocation failed (probably too large).)msg)rguard_memory_errorr)r builderargskwargsmemptrrfuncr r wrapsz+NRTContext._check_null_result..wrap) functoolswraps)rrr rr _check_null_resultszNRTContext._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_uncheckedr rsizer r r allocates 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 rmoduler FunctionTyper voidptr_tintp_tget_or_insert_function return_value add_attributecallr rr modfntyfnr r r r's  zNRTContext.allocate_uncheckedcCs>||j}tttjg}t||d}|||gS)zI Low-level free a memory area allocated with allocate(). NRT_Free rr&rr'VoidTyperr(r*r-)r rptrr/r0r1r r r free4s 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_uncheckedrr r r meminfo_alloc?s zNRTContext.meminfo_alloccCr")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. NRT_MemInfo_alloc_safer$r%r.r r r r7Ks  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_uncheckedr rr dtorr r r meminfo_alloc_dtor[ zNRTContext.meminfo_alloc_dtorcCsX||j}ttjtjtjg}t||d}|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. NRT_MemInfo_alloc_dtor_safer$) rr&rr'rr(r)r*r+r,r-bitcastr rr r=r/r0r1r r r r;hs    z'NRTContext.meminfo_alloc_dtor_uncheckedcCr:)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)r rr alignr r r meminfo_alloc_aligned|s z NRTContext.meminfo_alloc_alignedcCs||j}td}ttjtj|g}t||d}|j dt |t r1|j tj|}n |j|ks:Jd||||gS)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. Returns NULL to indicate error/failure to allocate. NRT_MemInfo_alloc_safe_alignedr$zalign must be a uint32)rr&rIntTyper'rr(r)r*r+r, isinstanceintr get_constantruint32typer-)r rr rDr/u32r0r1r r r rCs    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_uncheckedrr r r meminfo_new_varsizes zNRTContext.meminfo_new_varsizecCr")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_varsizer$r%r.r r r rOs  z(NRTContext.meminfo_new_varsize_uncheckedcCr:)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_uncheckedr<r r r meminfo_new_varsize_dtorr?z#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) rr&rr'rr(r)r*r-rBr r r rRs  z-NRTContext.meminfo_new_varsize_dtor_uncheckedcCr:)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_uncheckedr rmeminfor r r r meminfo_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_allocrVr r r rUs z*NRTContext.meminfo_varsize_alloc_uncheckedcCr:)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_uncheckedrVr r r meminfo_varsize_reallocs z"NRTContext.meminfo_varsize_realloccCrY)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_reallocr[rVr r r r]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_freer3)r rrWr5r/r0r1r r r meminfo_varsize_frees  zNRTContext.meminfo_varsize_freecCsN||j}ttjtjtjg}t|||}|j d| |||gS)Nr$r%)r rrWr funcnamer/r0r1r r r r\&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)rnumba.core.runtime.nrtdynmodrcr&rr*r-)r rrWrcr/r1r r r meminfo_data0s 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) r rtyval datamodelmembersmeminfosmimtypgetterfieldinner_meminfosr r r rl?s     zNRTContext.get_meminfosc Csv|ddlm}||||}|D]%\}}|j} t| ||} | jdd| jdd| | |gqdS)zGCall function of *funcname* on every meminfo found in *value*. r)incref_decref_tyr$ nocaptureN) rrerxrlr&rr*rr,r-) r rtypvaluerbrxrr_rsr/r1r r r _call_incref_decrefPs  zNRTContext._call_incref_decrefcC||||ddS)zG Recursively incref the given *value* and its members. NRT_increfNr}r rrzr{r r r increfbzNRTContext.increfcCr~)zG Recursively decref the given *value* and its members. NRT_decrefNrrr r r decrefhrzNRTContext.decrefcCs6|ttjd}|j}t||d}||dS)zGCalls NRT_get_api(), which returns the NRT API function table. r NRT_get_api)rrr'rr(r&r*r-)r rr0r/r1r r r get_nrt_apins  zNRTContext.get_nrt_apicCsh|j}|j}||}|j}|t||}||||Wd|S1s-wY|S)z(Check if an exception is raised N) r call_convcheck_try_statusexcinfonot_ris_nullif_then eh_end_try)r rctxcc trystatusr has_raisedr r r eh_checkxs    zNRTContext.eh_checkcC|j}|j}||dS)zBegin a try-block. N)rrset_try_statusr rrrr r r eh_tryzNRTContext.eh_trycCr)zEnd a try-block N)rrunset_try_statusrr r r rrzNRTContext.eh_end_tryN)#__name__ __module__ __qualname____doc__rrrr!rr6r8r7r>r;rErCrPrOrSrRrXrUr^r]rar\rfrlr}rrrrrrr r r r rsR                   r) rllvmliter numba.corerrrobjectrr r r r s