g):dZddlZddlZddlZddlZddlmZddlmZm Z m Z m Z ddl m Z GddeZGdd e Zed ZejZejZejZejZejZejZejZd Zd Zd ZdZdZdZGddeZ GddeZ!d"dZ"dZ#dZ$dZ%ej&ddZ'ej(dZ)dZ*dZ+dZ,ej(d#d Z-d!Z.dS)$z< Utilities to simplify the boilerplate for native lowering. N)Enum)typingtypesutilscgutils)BaseRegistryLoadercPeZdZdZddZdZdZdZdZdZ d Z d Z d Z d Z d S)Registryz? A registry of function and attribute implementations. unspecifiedcZ||_g|_g|_g|_g|_g|_dSN)name functionsgetattrssetattrscasts constants)selfrs I/granary/edev/quantum/lib/python3.11/site-packages/numba/core/imputils.py__init__zRegistry.__init__s0    cfd}|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). cBj|f|Sr )rappend)implargtysfuncrs rdecoratez Registry.lower..decorate%s% N ! !4v"6 7 7 7Kr)rrrrs``` rlowerzRegistry.lowers/       rcZ||||}||||jf|Sr )r signature)rrtyattr impl_list decorator real_impls r_decorate_attrzRegistry._decorate_attr*s8IdB-- )T9+>?@@@ rcfd}|S)z Decorate an implementation of __getattr__ for type *ty* and the attribute *attr*. The decorated implementation will have the signature (context, builder, typ, val). cJ|jtSr )r(r_decorate_getattrrr$rr#s rrz(Registry.lower_getattr..decorate7(&&tRt}'8:: :rrrr#r$rs``` r lower_getattrzRegistry.lower_getattr// : : : : : : :rc.||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#s rlower_getattr_genericzRegistry.lower_getattr_generic<!!"d+++rcfd}|S)z Decorate an implementation of __setattr__ for type *ty* and the attribute *attr*. The decorated implementation will have the signature (context, builder, sig, args). cJ|jtSr )r(r_decorate_setattrr,s rrz(Registry.lower_setattr..decorateOr-rrr.s``` r lower_setattrzRegistry.lower_setattrGr0rc.||dS)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)r8r2s rlower_setattr_genericzRegistry.lower_setattr_genericTr4rcfd}|S)z Decorate the implementation of implicit conversion between *fromty* and *toty*. The decorated implementation will have the signature (context, builder, fromty, toty, val). cDj|ff|Sr )rr)rfromtyrtotys rrz%Registry.lower_cast..decorategs' J  tfd^4 5 5 5Krr)rr=r>rs``` r lower_castzRegistry.lower_cast_s/       rcfd}|S)z Decorate the implementation for creating a constant of type *ty*. The decorated implementation will have the signature (context, builder, ty, pyval). cBj|ff|Sr )rr)rrr#s rrz)Registry.lower_constant..decoratess$ N ! !4"- 0 0 0Krr)rr#rs`` rlower_constantzRegistry.lower_constantls)      rcd|jdS)NzLowering Registry<>)rrs r__repr__zRegistry.__repr__xs0DI0000rN)r )__name__ __module__ __qualname____doc__rr r(r/r3r8r:r?rBrFrrrr r s       , , ,    , , ,      11111rr ceZdZdZdZdS)RegistryLoaderz6 An incremental loader for a target registry. )rrrrrN)rGrHrIrJregistry_itemsrrrrLrL|s QNNNrrLbuiltin_registrycD||fd}nfd}|f|_||_|S)Nc ||||Sr rcontextbuildertypvaluer$r's rresz_decorate_getattr..ress9WgsE:: :rc"|||||Sr rrQs rrVz_decorate_getattr..ress9WgsE4@@ @r)r"r$rr#r$rVr's @rr+r+saI  ; ; ; ; ; ; A A A A AECMCH JrcZ||fd}nfd}|tjf|_||_|S)Nc ||||Sr rrRrSsigargsr$r's rrVz_decorate_setattr..ress9WgsD99 9rc"|||||Sr rr[s rrVz_decorate_setattr..ress9WgsD$?? ?r)rAnyr"r$rXs @rr7r7seI  : : : : : : @ @ @ @ @OCMCH Jrct|jtjr|jj}|||}t j||}|| |j 5| |||}| ||dddn #1swxYwY| |}|Sr ) isinstance return_typerOptionaltypemake_optional_noneralloca_once_valueif_thennot_is_nonemake_optional_valuestoreload) rRrSr\statusretval value_type optional_none retvalptroptional_values rfix_returning_optionalrss#/5>22 )_) 227JGG -g}EE __W\\&.99 : : 5 5$88VN MM.) 4 4 4  5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 i(( Ms.B>>CCcRfd}j|_t||_|S)zC A wrapper inserting code calling Numba-compiled *fndesc*. cF||j}|j||jj|\}}t j||j5|j ||dddn #1swxYwY|j jksJt|||||}|j | |j kr/d}t||j |j t!||j|S)Nz$function returned {0} but expect {1})declare_functionmodule call_conv call_functionrestypeargtypesr if_unlikelyis_errorreturn_status_propagaterbrsrdget_value_type TypeErrorformatimpl_ret_new_ref) rRrSr\r]rrmrnmsgfndescs rimpzuser_function..imps]''?? *88 T6>6?DBB  &/ : : G G   5 5gv F F F G G G G G G G G G G G G G G G&.0000'#vvNN ;'00AA A A8CCJJv{CODDEE E&.&IIIs"B  BB)r{r"tuplelibs)rrrs` r user_functionrs> JJJJJ"OCMT{{CH Jrc:fd}t||_|S)zD A wrapper inserting code calling Numba-compiled *gendesc*. c||j}|j||jj|\}}||fSr )rvrwrxryrzr{)rRrSr\r]rrmrngendescs rrzuser_generator..impsT''@@ *88 T7?G,.wrapper..iternext_wrappers3HUc'7E22G8GWgv>> >rr)r iternext_implRefTypeBORROWED lower_builtin)rrr iterator_types` @rwrapperziterator_impl..wrappersb< w' ( ( ? ? ? ? ? ) ( ? 1 j-001ABBB rr) iterable_typerrs ` r iterator_implrs#      Nrc<eZdZdZdZdZdZd dZdZdZ d Z d S) _IternextResultzb A result wrapper for iteration, passed by iternext_impl() into the wrapped function. _context_builder_pairobjc0||_||_||_dSr r)rrRrSpairobjs rrz_IternextResult.__init__s   rcd|jtjd|j_dS)z1 Mark the iterator as exhausted. FNr get_constantrbooleanrsecondrEs r set_exhaustedz_IternextResult.set_exhausted s( $}99%-OO rTcp|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_valids r set_validz_IternextResult.set_valids8 } $ $}11%-JJH' rc||j_dS)zP Mark the iterator as yielding the given *value* (a LLVM inst). Nrfirst)rrUs ryield_z_IternextResult.yield_s$ rcl|j|jtj|jjS)z> Return whether the iterator is marked valid. )rget_argument_valuerrrrrrEs rrz_IternextResult.is_valids2}// 05 04 0DFF Frc|jjS)z> Return the iterator's yielded value, if any. rrEs r yielded_valuez_IternextResult.yielded_value&s}""rN)T) rGrHrIrJ __slots__rrrrrrrrrrrs5I   PPP (((($$$ FFF#####rrc$eZdZdZ dZ dZ dZdS)rz& Enumerate the reference type N)rGrHrIrJNEWr UNTRACKEDrrrrr,s< CHIIIrrcVdtDvrtdfd}|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) cg|]}|Srr).0xs r z!iternext_impl..Is+++a+++rz3ref_type must be an enum member of imputils.RefTypecfd}|S)Nc h|j}|||}||||t|||tjkrt }n?tjkrt}n'tjkrt}ntd||||| S)NzUnknown ref_type encountered) rb make_helperrrrrrimpl_ret_borrowedrimpl_ret_untracked ValueError _getvalue) rRrSr\r] pair_typerimpl_retrref_types rrz-iternext_impl..outer..wrapperMsI))'9==G D'3':: < < <7;&&+W---,W...- !?@@@8GW$-w/@/@/B/BDD Drr)rrrs` routerziternext_impl..outerLs/ D D D D D Dr)rr)rrs` rrr=sN++7+++++NOOO" Lrc|tj|j|}|d|}|||fS)z Call the `getiter()` implementation for the given *iterable_type* of value *val*, and return the corresponding LLVM inst. getiter)rr"r get_function)rRrSrval getiter_sig getiter_impls r call_getiterr`sB "=#> NNK'' ;??L <# ( ((rc|j}tj|tj}t j||}|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_typerPairrrr"rrr) rRrSrritemtyr iternext_sigrrs r call_iternextrjs  %F 65=11I#I}==L((\BBM -# ( (C!!'9c::G 7GW 5 55rc |ttjtj|}|||fS#t $rYdSwxYw)z^ Call len() on the given value. Return None if len() isn't defined on this type. N)rlenrr"rintpNotImplementedError)rRrSr#rlen_impls rcall_lenryse )''V-=ej"-N-NOOx#((( tts8A AA _ForIterLoop)rUdo_breakc# K|j}t|||}d}d fd}||5t |||}|d5 dddn #1swxYwYt| |V|dddn #1swxYwY |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.endc2dSr )branch)bb_endrSsrrzfor_iter..do_breaksvrF)likelyN)rrappend_basic_blockr goto_blockrrgrhrrrposition_at_end enable_nrtnrtdecref) rRrSrrritervalbb_bodyrrVrs ` @rfor_iterrs "/M7G]C@@G((99G  ' ' 7 7F NN7   G $ $  GWmWEE __W\\#,,..99%_ H H # # NN6 " " " # # # # # # # # # # # # # # #3,,..99999w                 F###< 7M7;;;;;<rs 444444444444::::::i1i1i1i1i1vi1i1i1XQQQQQ'QQQ8.// &  . (> . (>  ( !0         6",-#-#-#-#-#f-#-#-#^d"    F))) 6 6 6 ) ) )&{%n&;==  <<<<  . . . .GGGGGr