:j9TddlZddlZddlmZddlmZmZddlmZddl m Z ddl m Z ddl mZddlmZmZdd lmZdd lmZmZdd lmZmZmZmZmZdd lmZdd lmZddl m!Z!ddl"m#Z#m$Z$ddl%m&Z&m'Z'ddl(m)Z)edddddZ*Gdde*eZ+Gdde Z,e,Z-Gddej\Z/e/eed<dZ0dZ1d Z2ejfejhejjejld!Z7d"Z8ed#Z9Gd$d%e:Z;Gd&d'e;Z<Gd(d)e;Z=d*Z>y)+N)contextmanager)config targetconfig)jit)TargetDescriptor) is_jitted)NumbaDeprecationWarning) TargetOptionsinclude_default_options) cpu_target)dispatcher_registrytarget_registry)utilstypes serializecompilersigutils)as_dtype) _internal)parse_signature)build_ufunc_wrapperbuild_gufunc_wrapper) FunctionCache NullCache)global_compiler_locknopythonforceobj boundscheckfastmath writable_argsceZdZdZy)UFuncTargetOptionscJ|jdsd|_|jdsd|_|jdd|jdstj |_|jds|j |_d|_|jdy) Nenable_pyobjectTenable_loopliftnrt)default debuginforr) is_setr$r%inherit_if_not_setrDEBUGINFO_DEFAULTr(renable_pyobject_looplift)selfflagsoptionss p/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/numba/np/ufunc/ufuncbuilder.pyfinalizezUFuncTargetOptions.finalize#s||-.$(E !||-.$(E !    5||K($66EO||M* %E )-&   ,N)__name__ __module__ __qualname__r1r2r0r"r"!s-r2r"cBeZdZeZfdZedZedZxZ S) UFuncTargetc$t|dy)Nufunc)super__init__)r- __class__s r0r<zUFuncTarget.__init__:s !r2c"tjSN)r typing_contextr-s r0r@zUFuncTarget.typing_context=(((r2c"tjSr?)r target_contextrAs r0rDzUFuncTarget.target_contextArBr2) r3r4r5r"r/r<propertyr@rD __classcell__)r=s@r0r8r87s5 G"))))r2r8cFeZdZdZeZd dZdZedZ dZ d dZ dZ y) UFuncDispatcherzK An object handling compilation of various signatures for a ufunc. Nc|i}|i}||_tj|_||_||_t |_yr?)py_funcr UniqueDict overloads targetoptionslocalsrcache)r-rJrNrMs r0r<zUFuncDispatcher.__init__OsF >F  M ))+* [ r2cZt|j|j|jS)4 NOTE: part of ReduceMixin protocol )pyfuncrNrM)dictrJrNrMrAs r0_reduce_stateszUFuncDispatcher._reduce_statesZs(<<;;,,  r2c||||S)rQ)rJrNrMr6)clsrRrNrMs r0_rebuildzUFuncDispatcher._rebuildds 6& NNr2c8t|j|_yr?)rrJrOrAs r0enable_cachingzUFuncDispatcher.enable_cachingks"4<<0 r2c z|i}|jj}|j||jj}|j|t j }|j jj||d|_ d|_ d|_ |j|||S)NTnumpyF) rNcopyupdaterMrFlags targetdescrr/parse_as_flagsno_cpython_wrapper error_modelr% _compile_core)r-sigrNrMlocstoptr.s r0compilezUFuncDispatcher.compilens >F{{! F!!&&( M"    //t<#' #!&!!#uf55r2c  jj}jj}t fd}t5t j j|j5|5jj||  cdddcdddcdddStj|\}}tj||j|||| jj!|  cdddcdddcdddS#1swYnxYw dddn #1swYnxYwdddy#1swYyxYw)z Trigger the compiler on the core function or load a previously compiled version from the cache. Returns the CompileResult. c3K djjj}|jj<yy#t$rwxYwwr?)rLget signature Exception)existscresr-s r0store_overloads_on_successzAUFuncDispatcher._compile_core..store_overloads_on_successsX :++DNN;>59DNN4>>2"   sAA AA AAN)args return_typer.rN)r_r@rDrrr ConfigStackenterr\rO load_overloadrnormalize_signaturer compile_extrarJ save_overload) r-rdr.rN typingctx targetctxrorprqrns ` @r0rczUFuncDispatcher._compile_coresQ $$33 $$33  :  :" ))+11%**,? /1 ::33CCD'#       )1(D(DS(I%D+#11)Y26,,T>I8=fND JJ,,S$7!               sU2E6E> D1 E' E:AD1 E E1D: 6E> EE EE%)NNr?) r3r4r5__doc__ ufunc_targetr_r<rT classmethodrWrYrgrcr6r2r0rHrHIs=K ! OO 16*' r2rHnpyufunccb|j|fi|}tj|\}}|||fSr?)rgrru)nb_funcrMrdrnrprqs r0_compile_element_wise_functionrs; 7??3 0- 0D 44S9D+ { ""r2c|-|jr td|jj}|tj k7sJ||S)zGiven a compilation result, argument types, and a return type, build a valid Numba signature after validating that it doesn't violate the constraints for the compilation mode. z-return type must be specified for object mode) objectmode TypeErrorrkrqrpyobject)rnrprqs r0_finalize_ufunc_signaturersI  ??KL L..44K %.. (( (  r2c |j}|j}|jj}t5t |||||j |}|jj|j}ddd|jDcgc]}t|j}}|jt|jj||jfS#1swYrxYwcc}w)zBuild a wrapper for the ufunc loop entry point given by the compilation result object, using the element-wise signature. N)rDlibraryfndescllvm_func_namerrrget_pointer_to_functionnamerprnumappendrq environment) rnrkctxrfnameinfoptra dtypenumss r0!_build_element_wise_ufunc_wrapperrs   CllG KK & &E >"7C #'??D:ll22499=> +4..9Q!9I9 Xi334889 c4++ ++>> :sAC& C2&C/)rN reorderablecP t|}|S#t$rtd|wxYw)z] Parse an identity value and return the corresponding low-level value for Numpy. zInvalid identity value ) _identitiesKeyError ValueError)identitys r0parse_identityrs: Dx( O DBCCDs %c#Ktj5tjdtdddddy#1swYyxYww)zThis suppresses the NumbaDeprecationWarning that occurs through the use of `jit` without the `nopython` kwarg. This use of `jit` occurs in a few places in the `{g,}ufunc` mechanism in Numba, predominantly to wrap the "kernel" function.ignorez3.*The 'nopython' keyword argument was not supplied*)categorymessageN)warningscatch_warningsfilterwarningsr r6r2r03_suppress_deprecation_warning_nopython_not_suppliedrsF  ")@*= @  sA !A A A A ceZdZddZdZy)_BaseUFuncBuilderNct|dr |j}n|jj}t|j||\}}}|j |||}|j j |||j|<|S)NrM)hasattrrMrr_finalize_signature_sigsr_cres)r-rdrMrnrprqs r0addz_BaseUFuncBuilder.addsz 4 ) ..M LL66M"@ LL-#.dK&&tT;? # 3 r2cy)zI Disable the compilation of new signatures at call time. Nr6rAs r0disable_compilez!_BaseUFuncBuilder.disable_compilesr2r?)r3r4r5rrr6r2r0rrs   r2rc&eZdZddZdZdZdZy) UFuncBuilderNc|i}t|r |j}||_t||_t 5t dd|d|||_dddg|_i|_y#1swYxYw)Nr}_targetrOr6) rrJrrrrrrr)r-rJrrOrMs r0r<zUFuncBuilder.__init__s  M W ooG &x0 @ B 9030z%*0!.0079DL 9   9 9s A..A7ct|||S)z^Slated for deprecation, use ufuncbuilder._finalize_ufunc_signature() instead. )rr-rnrprqs r0rz UFuncBuilder._finalize_signature%s)t[AAr2c t5g}g}|js tdg}d}|jD]o}|j|}|j ||\}}}|j ||j t||j |j|fqdgt|z} |5tj|j} t| j} nt|jj} d} tj |jj"|jj$||| | | ||j& } | cdddS#1swYyxYw)N No definitionr)rrrrrbuildrintrleninspectgetfullargspecrJrprkrfromfuncr3rzr)r- dtypelistptrlist keepalivernrdrrenvdatlistargspecinctoutctr:s r0 build_ufunczUFuncBuilder.build_ufunc+sE !$ IG<<00IDzz 6zz#&*jjs&;# 3  +s3x(  $,,!45  6fs7|+G|!00>7<<(4>>../E&& %%t||';';D%4==E I$ $ $ s EE&&E/ct||S)z^Slated for deprecation, use ufuncbuilder._build_element_wise_ufunc_wrapper(). )r)r-rnrks r0rzUFuncBuilder.buildRs1yAAr2)NFN)r3r4r5r<rrrr6r2r0rrs B %NBr2rc4eZdZ ddZdZedZdZy) GUFuncBuilderNc|i}||_t||_t5t d|||_ddd||_t|\|_|_ ||_ ||_ g|_ i|_ t|}t|Dcgc] }|| c}|_y#1swYtxYwcc}w)Nr}r)rJrrrrrrkrsinsoutrMrOrr_get_transform_argtupler ) r-rJrkrrOrMr transform_argrs r0r<zGUFuncBuilder.__init__\s  M &x0 @ B I?3z?HDL I"-i8$)*   *73 "m#LM!$4#LM I I$MsB2B>2B;c|js2|jjtjk7r t d|tj}||S)Nz(gufunc kernel must have void return type)rrkrqrvoidrrs r0rz!GUFuncBuilder._finalize_signaturensD4>>#=#=#KFG G  **KD!!r2c vg}g}|js tdg}|jD]n}|j|}|j |\}}}|j ||j t ||j |j|fpdgt|z} t|j} t|j} tj|jj|jj||| | | ||j |j"|j$ } | S)Nr)rrrrrrrrrrrrrrJr3rzrrkr ) r- type_list func_listrrdrnrrrdatalistninnoutr:s r0rzGUFuncBuilder.build_ufuncws  ||O, , :: 2C::c?D"&**T"2 IsC   Y '   SX &   dllC0 1  26C N*$((m499~"" LL ! !4<<#7#7 y#tX t}}dnnd6H6H   r2c|j}t|j||j|j|j d}|j }|jj|j}g}|jD]O}t|tjr |j}n|}|jt!|j"Q|||fS)zJ Returns (dtype numbers, function ptr, EnvironmentObject) F)rO is_parfors)rkrrJrrrOrrrrrp isinstancerArraydtyperrr) r-rnrkrrrrrtys r0rzGUFuncBuilder.builds NN # LL$$))**  hhll22499=  /A!U[[)WW   Xb\-- .  / #s""r2)NFNr6)r3r4r5r<rrrrr6r2r0rrYs0AF35N$"8#r2rctjjtDcic]\}}|| c}}fd}|Scc}}w)z-Return function that transform arg into indexct|tr|S |S#t$r!d|ddj}t |wxYw)NzSpecified writable arg z not found in arg list z for function )rrrr5 RuntimeError)argmsgrp pos_by_argrJs r0rz)_get_transform_arg..transform_args_ c3 J $c? " $,SE1HV>'*>*>)?ACs# # $s *A)rrrp enumerate)rJirrrprs` @@r0rrsG  ! !' * / /D'07VQ#q&7J $ 8s A )?rr contextlibr numba.corerrnumba.core.decoratorsrnumba.core.descriptorsrnumba.core.extendingrnumba.core.errorsr numba.core.optionsr r numba.core.registryr numba.core.target_extensionr rrrrrrnumba.np.numpy_supportrnumba.np.ufuncrnumba.np.ufunc.sigparsernumba.np.ufunc.wrappersrrnumba.core.cachingrrnumba.core.compiler_lockr_options_mixinr"r8r{ ReduceMixinrHrrr PyUFunc_Zero PyUFunc_One PyUFunc_NonePyUFunc_ReorderableNonerrrobjectrrrrr6r2r0rs/%+%3*5E*LBB+$3M79) --, )" )} a i++a H4COJ/0 #  ,&  44       *AB$ABHP#%P#fr2