o Y۷i):@sfdZddlZddlZddlZddlZddlmZddlmZm Z m Z m Z ddl m Z GdddeZGdd d e Zed ZejZejZejZejZejZejZejZd d Zd dZddZddZddZddZGdddeZ GdddeZ!d2ddZ"ddZ#dd Z$d!d"Z%e&d#d$Z'ej(d%d&Z)d'd(Z*d)d*Z+d+d,Z,ej(d3d.d/Z-d0d1Z.dS)4z< Utilities to simplify the boilerplate for native lowering. N)Enum)typingtypesutilscgutils)BaseRegistryLoaderc@sbeZdZdZdddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ dS)Registryz? A registry of function and attribute implementations. unspecifiedcCs(||_g|_g|_g|_g|_g|_dSN)name functionsgetattrssetattrscasts constants)selfr rI/home/ubuntu/vllm_env/lib/python3.10/site-packages/numba/core/imputils.py__init__s  zRegistry.__init__cfdd}|S)a1 Decorate an implementation of *func* for the given argument types. *func* may be an actual global function object, or any pseudo-function supported by Numba, such as "getitem". The decorated implementation has the signature (context, builder, sig, args). csj|f|Sr )r appendimplargtysfuncrrrdecorate%z Registry.lower..decorater)rrrrrrrlowers zRegistry.lowercCs"||||}||||jf|Sr )r signature)rrtyattr impl_list decorator real_implrrr_decorate_attr*s zRegistry._decorate_attrcr)z Decorate an implementation of __getattr__ for type *ty* and the attribute *attr*. The decorated implementation will have the signature (context, builder, typ, val). c|jtSr )r%r _decorate_getattrrr!rr rrr7z(Registry.lower_getattr..decoraterrr r!rrr(r lower_getattr/zRegistry.lower_getattrcC ||dS)a: Decorate the fallback implementation of __getattr__ for type *ty*. The decorated implementation will have the signature (context, builder, typ, val, attr). The implementation is called for attributes which haven't been explicitly registered with lower_getattr(). N)r+rr rrrlower_getattr_generic< zRegistry.lower_getattr_genericcr)z Decorate an implementation of __setattr__ for type *ty* and the attribute *attr*. The decorated implementation will have the signature (context, builder, sig, args). cr&r )r%r_decorate_setattrrr(rrrOr)z(Registry.lower_setattr..decoraterr*rr(r lower_setattrGr,zRegistry.lower_setattrcCr-)a; Decorate the fallback implementation of __setattr__ for type *ty*. The decorated implementation will have the signature (context, builder, sig, args, attr). The implementation is called for attributes which haven't been explicitly registered with lower_setattr(). N)r2r.rrrlower_setattr_genericTr0zRegistry.lower_setattr_genericcr)z Decorate the implementation of implicit conversion between *fromty* and *toty*. The decorated implementation will have the signature (context, builder, fromty, toty, val). csj|ff|Sr )rrrfromtyrtotyrrrgsz%Registry.lower_cast..decorater)rr5r6rrr4r lower_cast_r,zRegistry.lower_castcsfdd}|S)z Decorate the implementation for creating a constant of type *ty*. The decorated implementation will have the signature (context, builder, ty, pyval). csj|ff|Sr )rrrr.rrrsrz)Registry.lower_constant..decorater)rr rrr.rlower_constantlszRegistry.lower_constantcCsd|jdS)NzLowering Registry<>)r rrrr__repr__xzRegistry.__repr__N)r )__name__ __module__ __qualname____doc__rrr%r+r/r2r3r7r8r;rrrrrs      rc@seZdZdZdZdS)RegistryLoaderz6 An incremental loader for a target registry. )r r rrrN)r=r>r?r@registry_itemsrrrrrA|srAbuiltin_registrycs8||dur fdd}nfdd}|f|_||_|S)Nc||||Sr rcontextbuildertypvaluer!r$rrresr<z_decorate_getattr..resc|||||Sr rrErJrrrK)rr!rr r!rKrrJrr's r'cs<||dur fdd}nfdd}|tjf|_||_|S)NcrDr rrFrGsigargsr!rJrrrKr<z_decorate_setattr..rescrLr rrOrJrrrKrM)rAnyrr!rNrrJrr1s  r1c Cst|jtjrB|jj}|||}t||}|| |j | |||}| ||Wdn1s8wY| |}|Sr ) isinstance return_typerOptionaltypemake_optional_noneralloca_once_valueif_thennot_is_nonemake_optional_valuestoreload) rFrGrPstatusretval value_type optional_none retvalptroptional_valuerrrfix_returning_optionals   recs"fdd}j|_t||_|S)zC A wrapper inserting code calling Numba-compiled *fndesc*. cs||j}|j||jj|\}}t||j|j ||Wdn1s.wY|j jks;Jt |||||}|j | |j krXd}t||j |j t||j|S)Nz$function returned {0} but expect {1})declare_functionmodule call_conv call_functionrestypeargtypesr if_unlikelyis_errorreturn_status_propagaterTrerVget_value_type TypeErrorformatimpl_ret_new_ref)rFrGrPrQrr_r`msgfndescrrimpszuser_function..imp)rkrtuplelibs)rurxrvrrtr user_functions  rycsfdd}t||_|S)zD A wrapper inserting code calling Numba-compiled *gendesc*. cs2||j}|j||jj|\}}||fSr )rfrgrhrirjrk)rFrGrPrQrr_r`gendescrrrvs zuser_generator..imp)rwrx)r{rxrvrrzruser_generators  r|csfdd}|S)zl Decorator a given class as implementing *iterator_type* (by providing an `iternext()` method). cs0jttjfdd}td|S)Ncs |\}|||}||||Sr r)rFrGrPrQresultrIiterobjclsiternextrriternext_wrappers z8iterator_impl..wrapper..iternext_wrapperr)r iternext_implRefTypeBORROWED lower_builtin)rr iterator_typerrwrappers ziterator_impl..wrapperr) iterable_typerrrrr iterator_impls  rc@sFeZdZdZdZddZddZddd Zd d Zd d Z ddZ dS)_IternextResultzb A result wrapper for iteration, passed by iternext_impl() into the wrapped function. _context_builder_pairobjcCs||_||_||_dSr r)rrFrGpairobjrrrrs z_IternextResult.__init__cCs|jtjd|j_dS)z1 Mark the iterator as exhausted. FNr get_constantrbooleanrsecondr:rrr set_exhausted sz_IternextResult.set_exhaustedTcCs$|dvr |jtj|}||j_dS)z Mark the iterator as valid according to *is_valid* (which must be either a Python boolean or a LLVM inst). )FTNr)ris_validrrr set_valids z_IternextResult.set_validcCs ||j_dS)zP Mark the iterator as yielding the given *value* (a LLVM inst). Nrfirst)rrIrrryield_s z_IternextResult.yield_cCs|j|jtj|jjS)z> Return whether the iterator is marked valid. )rget_argument_valuerrrrrr:rrrrs z_IternextResult.is_validcCs|jjS)z> Return the iterator's yielded value, if any. rr:rrr yielded_value&z_IternextResult.yielded_valueN)T) r=r>r?r@ __slots__rrrrrrrrrrrs   rc@s"eZdZdZ dZ dZ dZdS)rz& Enumerate the reference type N)r=r>r?r@NEWr UNTRACKEDrrrrr,srcs*ddtDvr tdfdd}|S)a Wrap the given iternext() implementation so that it gets passed an _IternextResult() object easing the returning of the iternext() result pair. ref_type: a numba.targets.imputils.RefType value, the reference type used is that specified through the RefType enum. The wrapped function will be called with the following signature: (context, builder, sig, args, iternext_result) cSsg|]}|qSrr).0xrrr Isz!iternext_impl..z3ref_type must be an enum member of imputils.RefTypecsfdd}|S)Nc st|j}|||}||||t|||tjkrt}ntjkr%t}n tjkr-t }nt d||||| S)NzUnknown ref_type encountered) rT make_helperrrrrrrimpl_ret_borrowedrimpl_ret_untracked ValueError _getvalue)rFrGrPrQ pair_typerimpl_ret)rref_typerrrMs      z-iternext_impl..outer..wrapperr)rrr)rrouterLsziternext_impl..outer)rr)rrrrrr=s  rcCs&t|j|}|d|}|||fS)z Call the `getiter()` implementation for the given *iterable_type* of value *val*, and return the corresponding LLVM inst. getiter)rrr get_function)rFrGrval getiter_sig getiter_implrrr call_getiter`s  rc CsR|j}t|tj}t||}|d|}|||f}||||}t|||S)z Call the `iternext()` implementation for the given *iterator_type* of value *val*, and return a convenience _IternextResult() object reflecting the results. r) yield_typerPairrrrrrr) rFrGrritemtyr iternext_sigrrrrr call_iternextjs    rcCs<z |tttj|}Wn tyYdSw|||fS)z^ Call len() on the given value. Return None if len() isn't defined on this type. N)rlenrrrintpNotImplementedError)rFrGr rlen_implrrrcall_lenys   r _ForIterLoop)rIdo_breakc #s|j}t|||}d}dfdd}||=t|||}j|dd Wdn1sJwYt | |V|Wdn1sfwY |j r}|j ||dSdS)z Simulate a for loop on the given iterable. Yields a namedtuple with the given members: - `value` is the value being yielded - `do_break` is a callable to early out of the loop z for_iter.bodyz for_iter.endcsdSr )branchrbb_endrGrrrr<zfor_iter..do_breakF)likelyN)rrappend_basic_blockr goto_blockrrYrZrrrposition_at_end enable_nrtnrtdecref) rFrGrrritervalbb_bodyrrKrrrfor_iters&       rcC|S)z5 The implementation returns a new reference. rctxrGrettyretrrrrrrrcCs|jr |j||||S)z The implementation returns a borrowed reference. This function automatically incref so that the implementation is returning a new reference. )rrincrefrrrrrsrcCr)z. The return type is not a NRT object. rrrrrrrrnumpyccs>ddlm}|j}||||_z dVW||_dS||_w)z7 Temporarily change the context's error model. r)callconvN) numba.corer error_modelcreate_error_model)rF model_namerold_error_modelrrrforce_error_models rcOstd)zPA stub for use internally by Numba when a call is emitted on a TypeRef. z%This function should not be executed.)r)rQkwargsrrrnumba_typeref_ctorrrr )r)/r@ collections contextlibinspect functoolsenumrrrrrrnumba.core.typing.templatesrobjectrrArCrrr+r/r2r3r7r8r'r1reryr|rrrrrrr namedtuplercontextmanagerrrrrrrrrrrrsR  l / #