o i82@sddlZddlmZmZddlmZmZmZmZddl m Z ddl m Z ddl mZmZddlmZddlmZdd lmZdd lmZd d ZGd ddeZGdddejejZejedS)N)jittypeof)cgutilstypes serializesigutils) is_jitted)npydecl)AbstractTemplate signature) _internal)array_analysis) ufuncbuilder) numpy_supportcs:ddlm}Gfddd|jjjj7_S)Nrnpyimplcs.eZdZdZZfddZddZZS)z(make_dufunc_kernel..DUFuncKernelz npyimpl._Kernel subclass responsible for lowering a DUFunc kernel (element-wise function) inside a broadcast loop (which is generated by npyimpl.numpy_ufunc_kernel()). cs.t|||||j|j\|_|_dSN)super__init__dufuncfind_ewise_functionargs inner_sigcres)selfcontextbuilder outer_sig) DUFuncKernel __class__d/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/numba/np/ufunc/dufunc.pyrsz1make_dufunc_kernel..DUFuncKernel.__init__c sj}j}fddt||j|jD}jjr*jjt j t j gt |j}n jj|j |j}j jjj}t||jjj}|jdjjj ||j |j|\}} | |j |j S)Ncs g|] \}}}|||qSr )cast).0valZintyZouttyrr r! !szEmake_dufunc_kernel..DUFuncKernel.generate..Z alwaysinline)rrziprrZ objectmoderZ call_convZget_function_typerZpyobjectlen return_typerblockfunctionmodulerZget_or_insert_functionZfndescZllvm_func_name attributesaddZ call_functionr") rrZisigZosigZ cast_args func_typer, entry_point_resr r%r!generates.   z1make_dufunc_kernel..DUFuncKernel.generate)__name__ __module__ __qualname____doc__rrr3 __classcell__r r_dufuncrr!rs r)numba.nprZ_Kernelr4ufunc)r:rr r9r!make_dufunc_kernels $r>c@s eZdZdZddZddZdS) DUFuncLowererzHCallable class responsible for lowering calls to a specific DUFunc. cCst||_g|_dSr)r>kernellibs)rrr r r!r<s  zDUFuncLowerer.__init__cCs(ddlm}||||||jjj|jS)Nrr)r<rZnumpy_ufunc_kernelr@rr=)rrrsigrrr r r!__call__@s  zDUFuncLowerer.__call__N)r4r5r6r7rrCr r r r!r?9s r?cseZdZdZedZddifddZfddZd d Ze d d Z d dZ e ddZ e ddZe ddZe ddZe ddZe ddZe ddZddZdd Zfd!d"Zd#d$Zd/d%d&Zd/d'd(Zd)d*Zd+d,Zd/d-d.ZZS)0DUFuncz Dynamic universal function (DUFunc) intended to act like a normal Numpy ufunc, but capable of call-time (just-in-time) compilation of fast loops specialized to inputs. )identity _keepaliveninnoutNFcCsjt|r|j}ttdd|d||}Wdn1s"wY|||t||dS)NZnpyufunc)_targetcacher )rpy_funcrZ3_suppress_deprecation_warning_nopython_not_suppliedr _initialize functoolsupdate_wrapper)rrKrErJ targetoptions dispatcherr r r!rQs  zDUFunc.__init__csPt|}tt|j||d|t||_||j j |_ |j j |_ dS)N)rE) rZparse_identityrrDr _install_typer? _lower_me _install_cgrKr4r7)rrPrEr;r r!rL[s   zDUFunc._initializecCs&t|jj}t|j|j|j|dS)4 NOTE: part of ReduceMixin protocol )rPrEfrozensiglist)list _dispatcher overloadskeysdictrE_frozen)rrVr r r!_reduce_statesfszDUFunc._reduce_statescCs<tj|}||||D]}||q|r||S)rT)r _DUFunc__new__rLr.disable_compile)clsrPrErUrVrrBr r r!_rebuildrs   zDUFunc._rebuildcCs|S)zK For compatibility with the various *UFuncBuilder classes. r r%r r r! build_ufuncszDUFunc.build_ufunccC|jjSr)rXrOr%r r r!rOzDUFunc.targetoptionscCrdr)r=rGr%r r r!rGrez DUFunc.nincCrdr)r=rHr%r r r!rHrez DUFunc.noutcCrdr)r=nargsr%r r r!rfrez DUFunc.nargscCrdr)r=ntypesr%r r r!rgrez DUFunc.ntypescCrdr)r=rr%r r r!rrez DUFunc.typescCrdr)r=rEr%r r r!rErezDUFunc.identitycCst|jjdks Jd|_dS)zI Disable the compilation of new signatures at call time. rTN)r(rXrYr\r%r r r!r`s zDUFunc.disable_compilecCst|\}}|||S)z= Compile the DUFunc for the given signature. )rZnormalize_signature_compile_for_argtys)rrBrr)r r r!r.s z DUFunc.addcsptjjj}|t|D]!}tt|d||ur.|j|dg|Ri|}|tur.|Sq t j |i|S)z} Allow any argument that has overridden __array_ufunc__ (NEP-18) to take control of DUFunc.__call__. __array_ufunc__rC) rnpZndarrayrituplevaluesgetattrtypeNotImplementedrrC)rrkwsdefaultargoutputr;r r!rCs zDUFunc.__call__c Os|jj}|r&d|vr|d}||f7}|r&tddddt|Dt|}||ks8|||jjks8J|rsz+DUFunc._compile_for_args..)r=rGpop TypeErrorjoinsortedr(rHr isinstancerArrayZdtyperZmap_arrayscalar_typeappendrhrk) rrrprGrtZargs_lenargtysrrZargtyr r r!_compile_for_argss(    zDUFunc._compile_for_argsc Cs|jr td|ft|tsJ|dur|}n||}t|j|j|\}}}t|||}t ||\}}}| t |||j ||j|f|jj |j|S)a/ Given a tuple of argument types (these should be the array dtypes, and not the array types themselves), compile the element-wise function for those inputs, generate a UFunc loop wrapper, and register the loop with the Numpy ufunc object for this DUFunc. zcompilation disabled for %sN)r\ RuntimeErrorr|rkrZ_compile_element_wise_functionrXrOZ_finalize_ufunc_signatureZ!_build_element_wise_ufunc_wrapperZ _add_loopintrFr~libraryrRrA) rrr)rBrZ actual_sigZ dtypenumsZptrenvr r r!rhs&   zDUFunc._compile_for_argtyscCsB|dur |jjj}td|jjtft||jd}| ||dS)a*Constructs and installs a typing class for a DUFunc 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()). NZ DUFuncTyping_)keyZgeneric) rX targetdescrZtyping_contextrnr=r4r r[_type_meZinsert_user_function)rZ typingctxZ_ty_clsr r r!rQs   zDUFunc._install_typecCsj|jrt||}|durdSt|j|jdt|}|jj D]\}}|j |kr2||fSq#dS)a Given a tuple of element-wise argument types, find a matching signature in the dispatcher. Return a 2-tuple containing the matching signature, and compilation result. Will return two None's if no matching signature was found. N)NN) r\rZufunc_find_matching_looprkZinputsZoutputsr(rXrYitemsr)r ewise_typeslooprBrr r r!rs   zDUFunc.find_ewise_functioncCs|rJ|j}tj|||}|\}}}}t|} | dkr)t|dt| } nt|} || \} } | durU|jrCtd||f| | || \} } | dusUJ| dkr^t |} n|j dkrv|dkrqt | j||g} n | jg} ntd| |t| S)z Implement AbstractTemplate.generic() for the typing class built by DUFunc._install_type(). Return the call-site signature after either validating the element-wise signature or compiling for it. rNzcannot call %s with types %sztyping gufuncs (nout > 1))r=r ZNumpy_rules_ufuncZ_handle_inputsr(rkrr\ryrhrWrHrr}r)NotImplementedErrorextendr )rrZkwtysr=Z_handle_inputs_resultZ base_typesZexplicit_outputsZndimsZlayoutZexplicit_output_countrrBrZouttysr r r!rs8       zDUFunc._type_mecsh|dur jjj}tj}tj}|fjj|fjj}|fjj}| fdd||fDdS)a- Install an implementation function for a DUFunc object in the given target context. If no target context is given, then _install_cg() installs into the target context of the dispatcher object (should be same default context used by jit() and njit()). Ncsg|]}j|fqSr )rR)r#rBr%r r!r&Ssz&DUFunc._install_cg..) rXrZtarget_contextrAnyr}r=rGrHZinsert_func_defn)rZ targetctxZ_anyZ_arrZsig0Zsig1r r%r!rSCs zDUFunc._install_cgr)r4r5r6r7setZ_DUFunc__base_kwargsrrLr] classmethodrbrcpropertyrOrGrHrfrgrrEr`r.rCrrhrQrrrSr8r r r;r!rDGs@              (rD)rMZnumbarrZ numba.corerrrrZnumba.core.extendingrZnumba.core.typingr Znumba.core.typing.templatesr r Znumba.np.ufuncr Z numba.parforsr rr<rr>objectr?Z ReduceMixinr^rDZ MAP_TYPESr~r r r r!s       +