J/Ph1ddlmZddlmZddlmZddlmZddlm Z m Z ddl m Z ddlm Z mZddlmZdd lmZmZdd lZd ZGd d e ZGdde je ZdZd S))typeof)types) GUFuncBuilder)parse_signature) UfuncBaseUfuncLowererBase)ufunc_find_matching_loop) serializeerrors)npydecl) signatureAbstractTemplateNchddlm}Gfdd|j}|xjjz c_|S)Nrnpyimplc<eZdZdZZfdZfdZfdZxZS)(make_gufunc_kernel..GUFuncKernelz npyimpl._Kernel subclass responsible for lowering a gufunc kernel (element-wise function) inside a broadcast loop (which is generated by npyimpl.numpy_gufunc_kernel()). ct||||j|j}|j|\|_|_dSN)super__init__dufunc_get_ewise_dtypesargsfind_ewise_function inner_sigcres)selfcontextbuilder outer_sig ewise_types __class__s U/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/numba/np/ufunc/gufunc.pyrz1make_gufunc_kernel..GUFuncKernel.__init__s] GG  Wgy 9 9 9+77 GGK(, (G(G)) %DNDIIIct|tjrBt|tjs(t||j|St|||Sr) isinstancerArrayrcastdtype)rvalfromtytotyr#s r$r)z-make_gufunc_kernel..GUFuncKernel.castsa&%+.. =tU[11 =ww||Ct<<<77<<VT22 2r%c|jjrd}tj||j|jjftj|S)NzCCalling a guvectorize function in object mode is not supported yet.) r objectmoder NumbaRuntimeErrorradd_linking_libslibraryrgenerate)rrmsgr#s r$r3z1make_gufunc_kernel..GUFuncKernel.generate%sZy# 4(.s333 L ) )49+<*> ? ? ?#577#T* *r%) __name__ __module__ __qualname____doc__rrr)r3 __classcell__)r#_dufuncs@r$ GUFuncKernelrs|         3 3 3 3 3 + + + + + + + + +r%r;)numba.npr_Kernelr5)r:rr;s` r$make_gufunc_kernelr> sj      +++++++w+++:W-- r%c"eZdZdZfdZxZS) GUFuncLowererzHCallable class responsible for lowering calls to a specific gufunc. cnddlm}t|t|jdS)Nrr)r<rrrr>numpy_gufunc_kernel)rgufuncrr#s r$rzGUFuncLowerer.__init__4sF$$$$$$ + 4 6 6 6 6 6r%)r5r6r7r8rr9)r#s@r$r@r@1sB666666666r%r@ceZdZdZdddidfdZdZdZedZd Z dd Z d Z d Z d Z dZddZdZedZdZdZdZdZdS)GUFuncz Dynamic generalized universal function (GUFunc) intended to act like a normal Numpy gufunc, but capable of call-time (just-in-time) compilation of fast loops specialized to inputs. NFcHd|_d|_||_||_t |||||||_|jjj|_|jjj|_|jj |_ | |j tj ||dS)NF)ufunc_frozen _is_dynamic _identityrgufunc_builderpy_funcr5r8nb_func _dispatcher _initialize functoolsupdate_wrapper)rrMr identitycache is_dynamic targetoptions writable_argss r$rzGUFunc.__init__Cs  %! , Y% OO+3< *2: .6 )*** w/////r%c||t||_|dSr) build_ufunc _install_typer@ _lower_me _install_cg)r dispatchers r$rPzGUFunc._initializeVsM  &t,, r%c |j}t|j|j|j|j|j|j|j|j |j  }|S)N) rMr rSrTrUrVrWtypesigsfrozen) rLdictrMr rKrTrJrVrW_sigsrI)rgbdcts r$_reduce_stateszGUFunc._reduce_states\sU  Jl^('**X<     r%c ||||||||} |D]} | | | | | _| S)N)rMr rSrTrUrVrW)addrYrI) clsrMr rSrTrUrVrWr_r`rsigs r$_rebuildzGUFunc._rebuildksms7i(:!.mMMM  C HHSMMMM   r%cd|jdS)Nz)r5rs r$__repr__zGUFunc.__repr__ws3$-3333r%c||jjj}td|jzt ft ||j}|||dS)a*Constructs and installs a typing class for a gufunc object in the input typing context. If no typing context is given, then _install_type() installs into the typing context of the dispatcher object (should be same default context used by jit() and njit()). N GUFuncTyping_)keygeneric) rO targetdescrtyping_contexttyper5rra_type_meinsert_user_function)r typingctx_ty_clss r$rZzGUFunc._install_typezsf  (4CI6(*dm<<<>> &&tW55555r%c:|j|dSr)rLrg)rftys r$rgz GUFunc.adds $$$$$r%cB|j|_|Sr)rLrYrHrls r$rYzGUFunc.build_ufuncs(4466  r%ct|jj}tt t |dtt t |dfSNr)rrLr tuplemaplen)r parsed_sigs r$expected_ndimszGUFunc.expected_ndimssL$T%8%BCC c#z!}--..c#z!}6M6M0N0NOOr%c |rJ|j}|jj}|\}}||z}t |sJt |t |D]\}} t | tjr| j ||krv|t |krdnd} |t |kr|n|t |z } |j d| d| d| j d|d||d } tj | tj|||} | \}}}}||\}}|{|jrd |d |} tj | ||||\}}|d kr!d |j d |} tj | |Jttjg|RS)z Implement AbstractTemplate.generic() for the typing class built by gufunc._install_type(). Return the call-site signature after either validating the element-wise signature or compiling for it. InputOutputz: z operand z& does not have enough dimensions (has z, gufunc core with signature z requires )Nz cannot call z with types NNzFail to compile )rHrLr rr enumerater'rr(ndimr5r TypingErrorr Numpy_rules_ufunc_handle_inputsrrI_compile_for_argtysnone)rargtyskwsrHri inp_ndims out_ndimsndimsidxargkindir4_handle_inputs_resultr"_s r$ruzGUFunc._type_mesH !+#2244 9I%6{{&&CJJ&&&!&)) . .HC#u{++ .5:0E0E"%I"6"6wwHY//CCS3y>>5I}>>>>q>>.1h>>!$>>05c >>>(--- ' 9 H H 63! ! 4 Q1))+66Q ;| .?T??v??(---  $ $[ 1 1 1--k::FCl""LLLFLL(---???-f----r%cN|j|}|j|dSr)_get_function_typerLrg)rr return_typefntys r$rzGUFunc._compile_for_argtyss0't&/ %%%%%r%cv||j}t|t|kSr)rrr)rr"ridtypess r$match_signaturezGUFunc.match_signatures0''11V}}k 2 222r%c|jSr)rJrls r$rUzGUFunc.is_dynamics r%ctd|}g}|D]L}t|tjr||j7||M|S)NcXt|tjr|nt|Sr)r'rTyper)rs r$z*GUFunc._get_ewise_dtypes..s' 3 (C(C"!C[[r%)rr'rr(appendr*)rrrtysargtys r$rzGUFunc._get_ewise_dtypessw!!"&(( " "E%-- " 5;'''' 5!!!! r%ct|jj}t|t|dt|dzkSr})rrLr r)rrrs r$_num_args_matchzGUFunc._num_args_matchsC$T%8%BCC 4yyC 1 ..Z]1C1CCCCr%c^t|jj}||}g}t |dD]e\}}t |}|dkr|||6|tj|||dft |d}t |dD]J\}}|||z} t |pd} |tj| | dKtj |S)NrAr~) rrLr rrrrrr(r) rrrr"lrsig_dimroffsetrettyret_ndims r$rzGUFunc._get_function_types#$T%8%BCC ,,T22  %jm44 C CLCw<r@ ReduceMixinrErrFr%r$rsQ555555333333AAAAAAAA;;;;;;((((((((%%%%%%CCCCCCCC!!!H66666$666e+e+e+e+e+Y "Ie+e+e+P     r%