0;jiH>ddlZddlmZddlmZddlmZmZmZm Z ddl m Z eddZ e dd d Z e d d d ZGddeZdS)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_alignedcLeZdZdZdZdZdZedZdZdZ edZ d Z ed Z d Z ed Zd ZedZdZedZdZedZdZedZdZdZdZdZdZdZdZdZdZdZ dZ!d Z"d!S)" NRTContextzF An object providing access to NRT APIs in the lowering pass. cn||_||_tjrt|_dSt |_dS)N)_context_enabledr DEBUG_NRT_NRT_MEMINFO_SAFE_API _meminfo_api_NRT_MEMINFO_DEFAULT_API)selfcontextenableds T/root/voice-cloning/.venv/lib/python3.11/site-packages/numba/core/runtime/context.py__init__zNRTContext.__init__s9     9 5D    8D   c<|jstjddS)NzNRT required but not enabled)rrNumbaRuntimeError)rs r _require_nrtzNRTContext._require_nrt(s+} K*+IJJ J K Kr"cFtjfd}|S)Nc`||g|Ri|}d}tj|j||||S)Nz'Allocation failed (probably too large).)msg)rguard_memory_errorr)rbuilderargskwargsmemptrr(funcs r wrapz+NRTContext._check_null_result..wrap-sJT$9$999&99F;C  &t}gv3 O O O OMr") functoolswraps)r.r/s` r _check_null_resultzNRTContext._check_null_result,s8            r"c.|||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&&w555r"c||j}tjtjtjg}t j||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*r6modfntyfns r r4zNRTContext.allocate_unchecked>sv nw07>2BCC  +C~ F F %%i000||B'''r"c||j}tjtjt jg}t j||d}|||gS)zI Low-level free a memory area allocated with allocate(). NRT_Free r%r<rr=VoidTyperr>r@rC)rr*ptrrErFrGs r freezNRTContext.freeKsd nr{}}w/@.ABB  +Cz B B||B&&&r"c.|||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"c.||j}tjtjtjg}t j|||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. r:) r%r<rr=rr>r?r@rr rArBrCrDs r rOz"NRTContext.meminfo_alloc_uncheckedbs nw07>2BCC  +C,0,=,CEE %%i000||B'''r"c0||||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$EEEr"c||j}tjtjtjtjg}t j|||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. r:)r%r<rr=rr>r?r@rr rArBrCbitcastrr*r6rUrErFrGs r rSz'NRTContext.meminfo_alloc_dtor_uncheckeds nw0 '0ABDD  +C,0,=,HJJ %%i000||B!(w7H!I!I!KLL Lr"c0||||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_aligneds33GT5IIIr"c||j}tjd}tjt jt j|g}t j|||j j }|j dt|tr&|jt"j|}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. 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 njnnw07>32GHH  +C,0,=,KMM %%i000 eS ! ! ?M..u|UCCEE:$$$&>$$$||Bu ...r"c.|||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}tjtjtjg}t j||d}|j d| ||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_uncheckeds} nw07>2BCC  +C,EGG %%i000||B'''r"c0||||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_dtors66wdKKKr"c||j}tjtjtjtjg}t j||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%r<rr=rr>r?r@rCrYs r rmz-NRTContext.meminfo_new_varsize_dtor_uncheckedst nw0 '0ABDD  + 577||Bt ---r"c0||||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_allocs33GWdKKKr"c2||||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(''$(CEE Er"c0||||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_reallocs55gwMMMr"c2||||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(''$(EGG Gr"c ||j}tjtjt jt jg}t j||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_freerJ)rr*rtrLrErFrGs r meminfo_varsize_freezNRTContext.meminfo_varsize_free0sy nr{}} ' 173DEGG  +C,FHH||B#///r"c2||j}tjtjtjtjg}t j|||}|j d| |||gS)Nr:r;)rr*rtr6funcnamerErFrGs r ryzNRTContext._call_varsize_alloc>s nw0 ' 17>BDD  +Cx @ @ %%i000||B$000r"c|ddlm}|j}t j||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.nrtdynmodrr<rr@rC)rr*rtrrErGs r meminfo_datazNRTContext.meminfo_dataHs` @@@@@@n  +C,CEE||B ***r"cX|jj|}||}g}|r-|||}|||f|D]<\}} | |} |||| } || =|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_meminfosWsM4R8 $$W--  $ $ & & &**7C88B OORH % % %# , ,LD&F3KKE!..weDDN OON + + + +r"c`|ddlm}||||}|D]y\}}|j} t j| ||} | jdd| jdd| | |gzdS)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 AAAAAA$$Wc599 # #EAr.C/5E08::B GAJ $ $Y / / / GAJ $ $[ 1 1 1 LLbT " " " " # #r"c6||||ddS)zG Recursively incref the given *value* and its members. NRT_increfNrrr*rrs r increfzNRTContext.increfz$   #ulCCCCCr"c6||||ddS)zG Recursively decref the given *value* and its members. NRT_decrefNrrs r decrefzNRTContext.decrefrr"c|tjtjd}|j}tj||d}||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_apisY w0"55n  +C} E E||B###r"cD|j}|j}||}|j}|t j||}tdkrB||5| |dddn #1swxYwY|S)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_checksm ]''00 #\\'/'7"C"CDD w  ,, ) )((( ) ) ) ) ) ) ) ) ) ) ) ) ) ) )s3BBBcL|j}|j}||dS)zBegin a try-block. N)rrset_try_statusrr*rrs r eh_tryzNRTContext.eh_trys,m ] '"""""r"cL|j}|j}||dS)zEnd a try-block N)rrunset_try_statusrs r rzNRTContext.eh_end_trys,m ] G$$$$$r"N)#__name__ __module__ __qualname____doc__r!r%r2r7r4rMrPrOrVrSr]r[rirhrnrmrurrr|r{rryrrrrrrrrrrr"r rrs{999KKK666 ( ( ( ' ' ' ; ; ;(((" F F FLLL( J J J///0 A A A(((& L L L...$LLL EEE NNNGGG 0 0 0111 + + +"###$DDD DDD $$$   ###%%%%%r"r)r0 collectionsrllvmliter numba.corerrrrnumba.core.utilsr r rrobjectrrr"r rs""""""555555555555&&&&&&$$<%677 /./G/L/OQQ 212E2J2MOO Q%Q%Q%Q%Q%Q%Q%Q%Q%Q%r"