shH>SSKrSSKJr SSKJr SSKJrJrJrJ r SSK J r \"SS5r \ "SS S 5r \ "S S S 5r"SS\5rg)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"\rSrSrSrSrSrSr\S5rSr Sr \S 5r S r \S 5r S r\S 5rSr\S5rSr\S5rSr\S5rSr\S5rSrSrSrSrSrSrSrSrSr Sr!S r"S!r#S"r$g#)$ NRTContextz> An object providing access to NRT APIs in the lowering pass. ctXlX l[R(a [Ulg[ Ulg)N)_context_enabledr DEBUG_NRT_NRT_MEMINFO_SAFE_API _meminfo_api_NRT_MEMINFO_DEFAULT_API)selfcontextenableds n/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/numba/core/runtime/context.py__init__NRTContext.__init__s'      5D  8D cRUR(d[R"S5eg)NzNRT required but not enabled)rrNumbaRuntimeError)rs r! _require_nrtNRTContext._require_nrt(s }}**+IJ Jr$cF^[R"T5U4Sj5nU$)Ncd>T"X/UQ70UD6nSn[R"URXUS9 U$)Nz'Allocation failed (probably too large).)msg)rguard_memory_errorr)rbuilderargskwargsmemptrr+funcs r!wrap+NRTContext._check_null_result..wrap-s7$9$9&9F;C  & &t}}g3 OMr$) functoolswraps)r1r2s` r!_check_null_resultNRTContext._check_null_result,s%       r$c$URX5$)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!allocateNRTContext.allocate5s&&w55r$c*UR5 URn[R"[R [R /5n[R"X4S5nURRS5 URXR/5$)zk 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-r;modfntyfns r!r9NRTContext.allocate_unchecked>sk nnw007>>2BC  + +C~ F %%i0||B''r$cUR5 URn[R"[R"5[ R /5n[ R"X4S5nURXR/5$)z9 Low-level free a memory area allocated with allocate(). NRT_Free r'rBrrCVoidTyperrDrFrI)rr-ptrrKrLrMs r!freeNRTContext.freeKs[ nnr{{}w/@/@.AB  + +Cz B||B&&r$c$URX5$)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_uncheckedr:s r! meminfo_allocNRTContext.meminfo_allocVs++G::r$cRUR5 URn[R"[R [R /5n[R"X4URR5nURRS5 URXR/5$)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'rBrrCrrDrErFrr rGrHrIrJs r!rW"NRTContext.meminfo_alloc_uncheckedbs{ nnw007>>2BC  + +C,0,=,=,C,CE %%i0||B''r$c&URXU5$)z 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-r;dtors r!meminfo_alloc_dtorNRTContext.meminfo_alloc_dtorss00EEr$cUR5 URn[R"[R [R [R /5n[R"XEURR5nURRS5 URXbURU[R 5/5$)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'rBrrCrrDrErFrr rGrHrIbitcastrr-r;r_rKrLrMs r!r]'NRTContext.meminfo_alloc_dtor_uncheckeds nnw00 '0A0ABD  + +C,0,=,=,H,HJ %%i0||B!(w7H7H!I!KL Lr$c&URXU5$)aC 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-r;aligns r!meminfo_alloc_aligned NRTContext.meminfo_alloc_aligneds33G5IIr$c0UR5 URn[R"S5n[R"[ R [ RU/5n[ R"XFURR5nURRS5 [U[5(a+URR!["R$U5nOUR&U:XdS5eUR)XrU/5$)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'rBrIntTyperCrrDrErFrr rGrH isinstanceintr get_constantruint32typerI)rr-r;rhrKu32rLrMs r!rg*NRTContext.meminfo_alloc_aligned_uncheckeds nnjjnw007>>32GH  + +C,0,=,=,K,KM %%i0 eS ! !MM..u||UCE::$ >&> >$||Bu ..r$c$URX5$)aP 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_uncheckedr:s r!meminfo_new_varsizeNRTContext.meminfo_new_varsizes11'@@r$c*UR5 URn[R"[R [R /5n[R"X4S5nURRS5 URXR/5$)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. Returns NULL to indicate error/failure to allocate. NRT_MemInfo_new_varsizer@rArJs r!rv(NRTContext.meminfo_new_varsize_uncheckedso nnw007>>2BC  + +C,EG %%i0||B''r$c&URXU5$)a 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^s r!meminfo_new_varsize_dtor#NRTContext.meminfo_new_varsize_dtors66wdKKr$cUR5 URn[R"[R [R [R /5n[R"XES5nURXbU/5$)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'rBrrCrrDrErFrIrds r!r}-NRTContext.meminfo_new_varsize_dtor_uncheckedsk nnw00 '0A0ABD  + + 57||Bt --r$c&URXU5$)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-meminfor;s r!meminfo_varsize_alloc NRTContext.meminfo_varsize_allocs33GdKKr$c(URXUS5$)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_allocrs r!r*NRTContext.meminfo_varsize_alloc_unchecked s ''$(CE Er$c&URXU5$)z 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_uncheckedrs r!meminfo_varsize_realloc"NRTContext.meminfo_varsize_reallocs55gMMr$c(URXUS5$)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_reallocrrs r!r,NRTContext.meminfo_varsize_realloc_unchecked&s ''$(EG Gr$cUR5 URn[R"[R"5[ R [ R /5n[ R"XES5nURXbU45$)zi Free a memory area allocated for a NRT varsize object. Note this does *not* free the NRT object itself! NRT_MemInfo_varsize_freerQ)rr-rrSrKrLrMs r!meminfo_varsize_freeNRTContext.meminfo_varsize_free0sm nnr{{} ' 1 173D3DEG  + +C,FH||B#//r$cJUR5 URn[R"[R [R [R /5n[R"XVU5nURRS5 URXrU/5$)Nr@rA)rr-rr;funcnamerKrLrMs r!rNRTContext._call_varsize_alloc>sw nnw00 ' 1 17>>BD  + +Cx @ %%i0||B$00r$cUR5 SSKJn URn[R "XCS5nUR XR/5$)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.nrtdynmodrrBrrFrI)rr-rrrKrMs r! meminfo_dataNRTContext.meminfo_dataHsC @nn  + +C,CE||B **r$c:URRUnURU5n/nUR5(a#UR X5nUR X'45 UH0upU "U5n UR XU 5n URU 5 M2 U$)zCReturn a list of *(type, meminfo)* inside the given value. )rdata_model_managertraversehas_nrt_meminfoget_nrt_meminfoappend get_meminfosextend) rr-tyval datamodelmembersmeminfosmimtypgetterfieldinner_meminfoss r!rNRTContext.get_meminfosWsMM44R8 $$W-  $ $ & &**78B OORH %#LD3KE!..weDN OON +$r$cLUR5 SSKJn URXU5nUHvupxURn [ R "XU5n U RSRS5 U RSRS5 URX/5 Mx g)zGCall function of *funcname* on every meminfo found in *value*. r)incref_decref_tyr@ nocaptureN) r'rrrrBrrFr.rHrI) rr-typvaluerrr_rrKrMs r!_call_incref_decrefNRTContext._call_incref_decrefhs A$$W59EA..C//08:B GGAJ $ $Y / GGAJ $ $[ 1 LLT "r$c*URXUS5 g)z7 Recursively incref the given *value* and its members. NRT_increfNrrr-rrs r!increfNRTContext.increfz   ulCr$c*URXUS5 g)z7 Recursively decref the given *value* and its members. NRT_decrefNrrs r!decrefNRTContext.decrefrr$cUR5 [R"[RS5nUR n[R "X2S5nURUS5$)zGCalls NRT_get_api(), which returns the NRT API function table. NRT_get_api)r'rrCrrDrBrFrI)rr-rLrKrMs r! get_nrt_apiNRTContext.get_nrt_apisR w00"5nn  + +C} E||B##r$cLURnURnURU5nURnUR [ R "X55n[S:a-URU5 URU5 SSS5 U$U$!,(df  U$=f)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_checkNRTContext.eh_checksmm ]]''0 ##\\'//'"CD w ,(-z-,s 6B B#cVURnURnURU5 g)zBegin a try-block. N)rrset_try_statusrr-rrs r!eh_tryNRTContext.eh_trys$mm ]] '"r$cVURnURnURU5 g)zEnd a try-block N)rrunset_try_statusrs r!rNRTContext.eh_end_trys$mm ]] G$r$)rrrN)%__name__ __module__ __qualname____firstlineno____doc__r"r'r6r<r9rTrXrWr`r]rirgrwrvr~r}rrrrrrrrrrrrrrr__static_attributes__rr$r!rrs,9K66 ( ' ; ;(" F FL( J J/0 A A(& L L.$LL E NNG 01 +"#$D D $ #%r$r)r4 collectionsrllvmliter numba.corerrrrnumba.core.utilsr r rrobjectrrr$r!rsh"55&$$<%67 //G/L/OQ 22E2J2MO Q%Q%r$