wgH>ddlZddlmZddlmZddlmZmZmZm Z ddl m Z eddZ e dd d Z e d d d ZGddeZy)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_alignedceZdZdZdZdZdZedZdZdZ edZ d Z ed Z d Z ed Zd ZedZdZedZdZedZdZedZdZdZdZdZdZdZdZdZdZdZ dZ!d Z"y!)" NRTContextzF An object providing access to NRT APIs in the lowering pass. cn||_||_tjr t|_yt |_y)N)_context_enabledr DEBUG_NRT_NRT_MEMINFO_SAFE_API _meminfo_api_NRT_MEMINFO_DEFAULT_API)selfcontextenableds _/home/mcse/projects/flask/flask-venv/lib/python3.12/site-packages/numba/core/runtime/context.py__init__zNRTContext.__init__s+      5D  8D cF|jstjdy)NzNRT required but not enabled)rrNumbaRuntimeError)rs r _require_nrtzNRTContext._require_nrt(s }}**+IJ Jr"cBtjfd}|S)Ncl||g|i|}d}tj|j||||S)Nz'Allocation failed (probably too large).)msg)rguard_memory_errorr)rbuilderargskwargsmemptrr(funcs r wrapz+NRTContext._check_null_result..wrap-s;$9$9&9F;C  & &t}}gv3 OMr") functoolswraps)r.r/s` r _check_null_resultzNRTContext._check_null_result,s%       r"c&|j||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*sizes r allocatezNRTContext.allocate5s&&w55r"c*|j|j}tjtj tj g}t j||d}|jjd|j||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*r6modfntyfns r r4zNRTContext.allocate_unchecked>so nnw007>>2BC  + +C~ F %%i0||B''r"c|j|j}tjtjt j g}t j||d}|j||gS)zI Low-level free a memory area allocated with allocate(). NRT_Free r%r<rr=VoidTyperr>r@rC)rr*ptrrErFrGs r freezNRTContext.freeKs_ nnr{{}w/@/@.AB  + +Cz B||B&&r"c&|j||S)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_uncheckedr5s r meminfo_alloczNRTContext.meminfo_allocVs++GT::r"cR|j|j}tjtj tj g}t j|||jj}|jjd|j||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. r:) r%r<rr=rr>r?r@rr rArBrCrDs r rOz"NRTContext.meminfo_alloc_uncheckedbs nnw007>>2BC  + +C,0,=,=,C,CE %%i0||B''r"c(|j|||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*r6dtors r meminfo_alloc_dtorzNRTContext.meminfo_alloc_dtorss00$EEr"c|j|j}tjtj tj tj g}t j|||jj}|jjd|j|||j|tj gS)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. r:)r%r<rr=rr>r?r@rr rArBrCbitcastrr*r6rUrErFrGs r rSz'NRTContext.meminfo_alloc_dtor_uncheckeds nnw00 '0A0ABD  + +C,0,=,=,H,HJ %%i0||B!(w7H7H!I!KL Lr"c(|j|||S)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*r6aligns r meminfo_alloc_alignedz NRTContext.meminfo_alloc_aligneds33GT5IIr"c"|j|j}tjd}tjt j t j|g}t j|||jj}|jjdt|tr+|jj!t"j$|}n|j&|k(sJd|j)|||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. r:zalign must be a uint32)r%r<rIntTyper=rr>r?r@rr rArB isinstanceintr get_constantruint32typerC)rr*r6r\rEu32rFrGs r r[z*NRTContext.meminfo_alloc_aligned_uncheckeds nnjjnw007>>32GH  + +C,0,=,=,K,KM %%i0 eS !MM..u||UCE::$ >&> >$||Bu ..r"c&|j||S)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_uncheckedr5s r meminfo_new_varsizezNRTContext.meminfo_new_varsizes11'4@@r"c*|j|j}tjtj tj g}t j||d}|jjd|j||gS)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;rDs r rhz(NRTContext.meminfo_new_varsize_uncheckedss nnw007>>2BC  + +C,EG %%i0||B''r"c(|j|||S)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_uncheckedrTs r meminfo_new_varsize_dtorz#NRTContext.meminfo_new_varsize_dtors66wdKKr"c|j|j}tjtj tj tj g}t j||d}|j|||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%r<rr=rr>r?r@rCrYs r rmz-NRTContext.meminfo_new_varsize_dtor_uncheckedso nnw00 '0A0ABD  + + 57||Bt --r"c(|j|||S)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*meminfor6s r meminfo_varsize_allocz NRTContext.meminfo_varsize_allocs33GWdKKr"c*|j|||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_allocrss r rrz*NRTContext.meminfo_varsize_alloc_unchecked s"''$(CE Er"c(|j|||S)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_uncheckedrss r meminfo_varsize_reallocz"NRTContext.meminfo_varsize_reallocs55gwMMr"c*|j|||dS)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_reallocrxrss r r{z,NRTContext.meminfo_varsize_realloc_unchecked&s"''$(EG Gr"c|j|j}tjtjt j t j g}t j||d}|j|||fS)z Free a memory area allocated for a NRT varsize object. Note this does *not* free the NRT object itself! NRT_MemInfo_varsize_freerJ)rr*rtrLrErFrGs r meminfo_varsize_freezNRTContext.meminfo_varsize_free0sq nnr{{} ' 1 173D3DEG  + +C,FH||B#//r"cJ|j|j}tjtj tj tj g}t j|||}|jjd|j|||gS)Nr:r;)rr*rtr6funcnamerErFrGs r ryzNRTContext._call_varsize_alloc>s{ nnw00 ' 1 17>>BD  + +Cx @ %%i0||B$00r"c|jddlm}|j}t j ||d}|j ||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.nrtdynmodrr<rr@rC)rr*rtrrErGs r meminfo_datazNRTContext.meminfo_dataHsG @nn  + +C,CE||B **r"c4|jj|}|j|}g}|jr%|j ||}|j ||f|D]1\}} | |} |j ||| } |j| 3|S)zCReturn a list of *(type, meminfo)* inside the given value. )rdata_model_managertraversehas_nrt_meminfoget_nrt_meminfoappend get_meminfosextend) rr*tyval datamodelmembersmeminfosmimtypgetterfieldinner_meminfoss r rzNRTContext.get_meminfosWsMM44R8 $$W-  $ $ &**7C8B OORH %# ,LD&3KE!..weDN OON + ,r"cN|jddlm}|j|||}|D]w\}}|j} t j | ||} | jdjd| jdjd|j| |gyy)zGCall function of *funcname* on every meminfo found in *value*. r)incref_decref_tyr: nocaptureN) r%rrrr<rr@r+rBrC) rr*typvaluerrr_rrErGs r _call_incref_decrefzNRTContext._call_incref_decrefhs A$$Wc59 #EAr..C//5E08:B GGAJ $ $Y / GGAJ $ $[ 1 LLbT " #r"c,|j|||dy)zG Recursively incref the given *value* and its members. NRT_increfNrrr*rrs r increfzNRTContext.increfz   #ulCr"c,|j|||dy)zG Recursively decref the given *value* and its members. NRT_decrefNrrs r decrefzNRTContext.decrefrr"c|jtjtjd}|j }tj ||d}|j|dS)zGCalls NRT_get_api(), which returns the NRT API function table. NRT_get_api)r%rr=rr>r<r@rC)rr*rFrErGs r get_nrt_apizNRTContext.get_nrt_apisT w00"5nn  + +C} E||B##r"c@|j}|j}|j|}|j}|j t j ||}tdkr-|j|5|j|ddd|S|S#1swY|SxYw)z(Check if an exception is raised ) N) r call_convcheck_try_statusexcinfonot_ris_nullr if_then eh_end_try)rr*ctxcc trystatusr has_raiseds r eh_checkzNRTContext.eh_checksmm ]]''0 ##\\'//'7"CD w , )( )z )s 5BBcV|j}|j}|j|y)zBegin a try-block. N)rrset_try_statusrr*rrs r eh_tryzNRTContext.eh_trys$mm ]] '"r"cV|j}|j}|j|y)zEnd a try-block N)rrunset_try_statusrs r rzNRTContext.eh_end_trys$mm ]] G$r"N)#__name__ __module__ __qualname____doc__r!r%r2r7r4rMrPrOrVrSr]r[rirhrnrmrurrr|r{rryrrrrrrrrrrr"r rrs,9K66 ( ' ; ;(" F FL( J J/0 A A(& L L.$LL E NNG 01 +"#$D D $ #%r"r)r0 collectionsrllvmliter numba.corerrrrnumba.core.utilsr r rrobjectrrr"r rsh"55&$$<%67 //G/L/OQ 22E2J2MO Q%Q%r"