U ?hH>ã@szddlZddlmZddlmZddlmZmZmZm Z ddl m Z eddƒZ e dd d ƒZ e d d d ƒZGdd„deƒZdS)éN)Ú namedtuple)Úir)ÚtypesÚcgutilsÚerrorsÚconfig)Ú PYVERSIONÚ_NRT_Meminfo_Functions)ÚallocÚ alloc_dtorÚ alloc_alignedZNRT_MemInfo_alloc_safeZNRT_MemInfo_alloc_dtor_safeZNRT_MemInfo_alloc_safe_alignedZNRT_MemInfo_allocZNRT_MemInfo_alloc_dtorZNRT_MemInfo_alloc_alignedc@s(eZdZdZdd„Zdd„Zdd„Zedd „ƒZd d „Zd d „Z edd„ƒZ dd„Z edd„ƒZ dd„Z edd„ƒZdd„Zedd„ƒZdd„Zedd„ƒZd d!„Zed"d#„ƒZd$d%„Zed&d'„ƒZd(d)„Zd*d+„Zd,d-„Zd.d/„Zd0d1„Zd2d3„Zd4d5„Zd6d7„Zd8d9„Zd:d;„Z dd?„Z"d@S)AÚ NRTContextzF An object providing access to NRT APIs in the lowering pass. cCs$||_||_tjrt|_nt|_dS)N)Ú_contextÚ_enabledrZ DEBUG_NRTÚ_NRT_MEMINFO_SAFE_APIÚ _meminfo_apiÚ_NRT_MEMINFO_DEFAULT_API)ÚselfÚcontextZenabled©rúL/var/www/html/venv/lib/python3.8/site-packages/numba/core/runtime/context.pyÚ__init__s zNRTContext.__init__cCs|jst d¡‚dS)NzNRT required but not enabled)rrZNumbaRuntimeError)rrrrÚ _require_nrt(szNRTContext._require_nrtcst ˆ¡‡fdd„ƒ}|S)Ncs.ˆ||f|ž|Ž}d}tj|j|||d|S)Nz'Allocation failed (probably too large).)Úmsg)rZguard_memory_errorr)rÚbuilderÚargsÚkwargsZmemptrr©ÚfuncrrÚwrap-sz+NRTContext._check_null_result..wrap)Ú functoolsÚwraps)rrrrrÚ_check_null_result,szNRTContext._check_null_resultcCs | ||¡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_unchecked©rrÚsizerrrÚallocate5szNRTContext.allocatecCsH| ¡|j}t tjtjg¡}t ||d¡}|j  d¡|  ||g¡S)zƒ Low-level allocate a new memory area of `size` bytes. Returns NULL to indicate error/failure to allocate. Z NRT_AllocateÚnoalias© rÚmodulerÚ FunctionTyperÚ voidptr_tÚintp_tÚget_or_insert_functionÚ return_valueÚ add_attributeÚcall©rrr%ÚmodÚfntyÚfnrrrr#>s  zNRTContext.allocate_uncheckedcCs>| ¡|j}t t ¡tjg¡}t ||d¡}| ||g¡S)zI Low-level free a memory area allocated with allocate(). ZNRT_Free© rr)rr*ZVoidTyperr+r-r0)rrÚptrr2r3r4rrrÚfreeKs zNRTContext.freecCs | ||¡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_uncheckedr$rrrÚ meminfo_allocVs zNRTContext.meminfo_alloccCsL| ¡|j}t tjtjg¡}t |||jj ¡}|j   d¡|  ||g¡S)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') rr)rr*rr+r,r-rr r.r/r0r1rrrr8bsÿ z"NRTContext.meminfo_alloc_uncheckedcCs| |||¡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_unchecked©rrr%ÚdtorrrrÚmeminfo_alloc_dtorss zNRTContext.meminfo_alloc_dtorcCs\| ¡|j}t tjtjtjg¡}t |||jj ¡}|j   d¡|  |||  |tj¡g¡S)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')rr)rr*rr+r,r-rr r.r/r0Zbitcast©rrr%r<r2r3r4rrrr:€s  ÿÿ  ÿz'NRTContext.meminfo_alloc_dtor_uncheckedcCs| |||¡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)rrr%ÚalignrrrÚmeminfo_alloc_aligned”s z NRTContext.meminfo_alloc_alignedcCsˆ| ¡|j}t d¡}t tjtj|g¡}t |||j j ¡}|j   d¡t |tƒrf|j tj|¡}n|j|ksxtdƒ‚| |||g¡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. Returns NULL to indicate error/failure to allocate. é r'zalign must be a uint32)rr)rZIntTyper*rr+r,r-rr r.r/Ú isinstanceÚintrZ get_constantrZuint32ÚtypeÚAssertionErrorr0)rrr%r@r2Úu32r3r4rrrr?¢s  ÿ  z*NRTContext.meminfo_alloc_aligned_uncheckedcCs | ||¡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_uncheckedr$rrrÚmeminfo_new_varsizeºs zNRTContext.meminfo_new_varsizecCsH| ¡|j}t tjtjg¡}t ||d¡}|j  d¡|  ||g¡S)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. ZNRT_MemInfo_new_varsizer'r(r1rrrrHÈs ÿ z(NRTContext.meminfo_new_varsize_uncheckedcCs| |||¡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_uncheckedr;rrrÚmeminfo_new_varsize_dtorÛs z#NRTContext.meminfo_new_varsize_dtorcCsB| ¡|j}t tjtjtjg¡}t ||d¡}| |||g¡S)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. ZNRT_MemInfo_new_varsize_dtor) rr)rr*rr+r,r-r0r>rrrrJès  ÿÿz-NRTContext.meminfo_new_varsize_dtor_uncheckedcCs| |||¡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_unchecked©rrÚmeminfor%rrrÚmeminfo_varsize_allocúsz NRTContext.meminfo_varsize_alloccCs| |||d¡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(). Returns NULL to indicate error/failure to allocate. ZNRT_MemInfo_varsize_alloc©Ú_call_varsize_allocrMrrrrL s ÿz*NRTContext.meminfo_varsize_alloc_uncheckedcCs| |||¡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_uncheckedrMrrrÚmeminfo_varsize_reallocs z"NRTContext.meminfo_varsize_realloccCs| |||d¡S)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. ZNRT_MemInfo_varsize_reallocrPrMrrrrR&s ÿz,NRTContext.meminfo_varsize_realloc_uncheckedcCsD| ¡|j}t t ¡tjtjg¡}t ||d¡}| |||f¡S)z Free a memory area allocated for a NRT varsize object. Note this does *not* free the NRT object itself! ZNRT_MemInfo_varsize_freer5)rrrNr6r2r3r4rrrÚmeminfo_varsize_free0s  ÿÿzNRTContext.meminfo_varsize_freecCsN| ¡|j}t tjtjtjg¡}t |||¡}|j  d¡|  |||g¡S)Nr'r()rrrNr%Úfuncnamer2r3r4rrrrQ>s ÿ zNRTContext._call_varsize_alloccCs6| ¡ddlm}|j}t ||d¡}| ||g¡S)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_tyZNRT_MemInfo_data_fast)rÚnumba.core.runtime.nrtdynmodrVr)rr-r0)rrrNrVr2r4rrrÚ meminfo_dataHs ÿzNRTContext.meminfo_datac Csn|jj|}| |¡}g}| ¡r<| ||¡}| ||f¡|D](\}} | |ƒ} | ||| ¡} | | ¡q@|S)zCReturn a list of *(type, meminfo)* inside the given value. )rZdata_model_managerÚtraverseZhas_nrt_meminfoZget_nrt_meminfoÚappendÚ get_meminfosÚextend) rrÚtyÚvalZ datamodelÚmembersÚmeminfosÚmiZmtypÚgetterÚfieldZinner_meminfosrrrr[Ws     zNRTContext.get_meminfosc Csv| ¡ddlm}| |||¡}|D]J\}}|j} t | ||¡} | jd d¡| jd d¡|  | |g¡q&dS)zGCall function of *funcname* on every meminfo found in *value*. r)Úincref_decref_tyr'Z nocaptureN) rrWrdr[r)rr-rr/r0) rrÚtypÚvaluerUrdr`Ú_rar2r4rrrÚ_call_incref_decrefhs  ÿzNRTContext._call_incref_decrefcCs| |||d¡dS)zG Recursively incref the given *value* and its members. Z NRT_increfN©rh©rrrerfrrrÚincrefzszNRTContext.increfcCs| |||d¡dS)zG Recursively decref the given *value* and its members. Z NRT_decrefNrirjrrrÚdecref€szNRTContext.decrefcCs6| ¡t tjd¡}|j}t ||d¡}| |d¡S)zGCalls NRT_get_api(), which returns the NRT API function table. rZ NRT_get_api)rrr*rr+r)r-r0)rrr3r2r4rrrÚ get_nrt_api†s zNRTContext.get_nrt_apic CsZ|j}|j}| |¡}|j}| t ||¡¡}tdkrV| |¡|  |¡W5QRX|S)z(Check if an exception is raised )éé ) rÚ call_convZcheck_try_statusÚexcinfoÚnot_rZis_nullrZif_thenÚ eh_end_try)rrÚctxÚccZ trystatusrqZ has_raisedrrrÚeh_checks  zNRTContext.eh_checkcCs|j}|j}| |¡dS)zBegin a try-block. N)rrpZset_try_status©rrrtrurrrÚeh_tryžszNRTContext.eh_trycCs|j}|j}| |¡dS)zEnd a try-block N)rrpZunset_try_statusrwrrrrs¥szNRTContext.eh_end_tryN)#Ú__name__Ú __module__Ú __qualname__Ú__doc__rrr"r&r#r7r9r8r=r:rAr?rIrHrKrJrOrLrSrRrTrQrXr[rhrkrlrmrvrxrsrrrrr sP               r )r Ú collectionsrZllvmliterZ numba.corerrrrZnumba.core.utilsrr rrÚobjectr rrrrÚs    ÿþþ