:j1ddlmZddlmZddlmZddlmZddlm Z m Z ddl m Z ddlm Z mZddlmZdd lmZmZdd lZd ZGd d e ZGdde j.e ZdZy ))typeof)types) GUFuncBuilder)parse_signature) UfuncBaseUfuncLowererBase)ufunc_find_matching_loop) serializeerrors)npydecl) signatureAbstractTemplateNcddlm}Gfdd|j}|xjjz c_|S)Nrnpyimplc>eZdZdZWZfdZfdZfdZxZS)(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|||||jj|j}|jj |\|_|_yN)super__init__dufunc_get_ewise_dtypesargsfind_ewise_function inner_sigcres)selfcontextbuilder outer_sig ewise_types __class__s j/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/numba/np/ufunc/gufunc.pyrz1make_gufunc_kernel..GUFuncKernel.__init__sL G Wgy 9++77 GK(, (G(G) %DNDIct|tjr5t|tjst|||j |St||||Sr) isinstancerArrayrcastdtype)rvalfromtytotyr#s r$r)z-make_gufunc_kernel..GUFuncKernel.castsI&%++.tU[[1w|Ct<<7<VT2 2r%c|jjrd}tj||jj |jj ft| |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%sWyy##(..s33 LL ) )499+<+<*> ?7#T* *r%) __name__ __module__ __qualname____doc__rrr)r3 __classcell__)r#_dufuncs@r$ GUFuncKernelrs#    3 + +r%r;)numba.npr_Kernelr5)r:rr;s` r$make_gufunc_kernelr> s5 +w+:W--- r%c"eZdZdZfdZxZS) GUFuncLowererzHCallable class responsible for lowering calls to a specific gufunc. cPddlm}t| |t|j y)Nrr)r<rrrr>numpy_gufunc_kernel)rgufuncrr#s r$rzGUFuncLowerer.__init__4s!$ + 44 6r%)r5r6r7r8rr9)r#s@r$r@r@1s66r%r@ceZdZdZ ddZdZdZedZdZ ddZ d Z d Z d Z d Zdd ZdZedZdZdZdZdZy)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. Nc|i}d|_d|_||_||_t |||||||_|j j j|_|j j j|_|j j|_ |j|jtj||y)NF)ufunc_frozen _is_dynamic _identityrgufunc_builderpy_funcr5r8nb_func _dispatcher _initialize functoolsupdate_wrapper)rrLr identitycache is_dynamic targetoptions writable_argss r$rzGUFunc.__init__Cs  M  %! , Y% O++33<< **22:: ..66 ))*  w/r%c|j|jt||_|j yr) build_ufunc _install_typer@ _lower_me _install_cg)r dispatchers r$rOzGUFunc._initializeXs2  &t, r%c |j}t|j|j|j|j |j |j|j|j|j }|S)N) rLr rRrSrTrUrVtypesigsfrozen) rKdictrLr rJrSrIrUrV_sigsrH)rgbdcts r$_reduce_stateszGUFunc._reduce_states^sc  JJll^^((''****XX<<   r%c ||||||||} |D]} | j| | j| | _| S)N)rLr rRrSrTrUrV)addrXrH) clsrLr rRrSrTrUrVr^r_rsigs r$_rebuildzGUFunc._rebuildmsS7i(:!.mM C HHSM    r%c"d|jdS)Nz)r5rs r$__repr__zGUFunc.__repr__ys!$--33r%c| |jjj}td|jzt ft ||j}|j||y)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) rN targetdescrtyping_contexttyper5rr`_type_meinsert_user_function)r typingctx_ty_clss r$rYzGUFunc._install_type|s\  ((44CCI6(*dmm<> &&tW5r%c:|jj|yr)rKrf)rftys r$rfz GUFunc.adds $r%cD|jj|_|Sr)rKrXrGrks r$rXzGUFunc.build_ufuncs((446  r%ct|jj}tt t |dtt t |dfSNr)rrKr tuplemaplen)r parsed_sigs r$expected_ndimszGUFunc.expected_ndimssC$T%8%8%B%BC c#z!}-.c#z!}6M0NOOr%c |rJ|j}|jj}|j\}}||z}t |sJt |t |D]\}} t | tjs!| j||ks4|t |krdnd} |t |kr|n |t |z } |jd| d| d| jd|d||d } tj| tjj|||} | \}}}}|j!|\}}|~|j"rd |d |} tj| |j%||j!|\}}|d k(r'd |jd |} tj| |Jttj&g|S) 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 )z cannot call z with types NNzFail to compile )rGrKr rr enumerater'rr(ndimr5r TypingErrorr Numpy_rules_ufunc_handle_inputsrrH_compile_for_argtysnone)rargtyskwsrGrh inp_ndims out_ndimsndimsidxargkindir4_handle_inputs_resultr"_s r$rtzGUFunc._type_mesw !!++#224 9I%6{&CJ&{!&) .HC#u{{+5:0E"%I"6wHY/CS3y>5I}}oRvYqc:..1hhZ8!!$Zc |1>((-- .!( 9 9 H H 63! 4 Q1))+6Q ;||$TF,vh?((--  $ $[ 1--k:FCl"(|F8L((--? "?-f--r%cX|j|}|jj|yr)_get_function_typerKrf)rr return_typefntys r$rzGUFunc._compile_for_argtyss*'t&&/ %r%cf|j|j}t|t|k(Sr)rrr~)rr"rhdtypess r$match_signaturezGUFunc.match_signatures*''1V}k 222r%c|jSr)rIrks r$rTzGUFunc.is_dynamicsr%ctd|}g}|D]I}t|tjr|j |j 9|j |K|S)NcPt|tjr|St|Sr)r'rTyper)rs r$z*GUFunc._get_ewise_dtypes..s 3 (CC[r%)rr'rr(appendr*)rrrtysargtys r$rzGUFunc._get_ewise_dtypessY!"&( "E%- 5;;' 5!  "  r%ct|jj}t|t|dt|dzk(Sr|)rrKr r)rrrs r$_num_args_matchzGUFunc._num_args_matchs>$T%8%8%B%BC 4yC 1 .Z]1CCCCr%ct|jj}|j|}g}t |dD]S\}}t |}|dk(r|j ||+|j tj|||dUt |d}t |dD]B\}}|||z} t |xsd} |j tj| | dDtj|S)NrAr}) rrKr rrrrrr(r) rrrr"lrsig_dimroffsetrettyret_ndims r$rzGUFunc._get_function_types$T%8%8%B%BC ,,T2  %jm4 CLCw <r@ ReduceMixinrErrr%r$rsR53A;(%C!H6$6g+Y " "Ig+T r%