o 0 i1@sddlmZddlmZddlmZddlmZddlm Z m Z ddl m Z ddlm Z mZddlmZdd lmZmZdd lZd d ZGd dde ZGddde je ZddZd S))typeof)types) GUFuncBuilder)parse_signature) UfuncBaseUfuncLowererBase)ufunc_find_matching_loop) serializeerrors)npydecl) signatureAbstractTemplateNcs6ddlm}Gfddd|j}|jj7_|S)Nrnpyimplcs<eZdZdZZfddZfddZfddZZS)z(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()). cs6t||||j|j}|j|\|_|_dSN)super__init__dufunc_get_ewise_dtypesargsfind_ewise_functionZ inner_sigcres)selfcontextbuilderZ outer_sig ewise_types __class__a/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/numba/np/ufunc/gufunc.pyrs z1make_gufunc_kernel..GUFuncKernel.__init__cs:t|tjrt|tjst||j|St|||Sr) isinstancerArrayrcastdtype)rvalZfromtyZtotyrrrr"s  z-make_gufunc_kernel..GUFuncKernel.castcs4|jjr d}t||j|jjftj|S)NzCCalling a guvectorize function in object mode is not supported yet.) rZ objectmoder ZNumbaRuntimeErrorrZadd_linking_libsZlibraryrgenerate)rrmsgrrrr%%s   z1make_gufunc_kernel..GUFuncKernel.generate) __name__ __module__ __qualname____doc__rrr"r% __classcell__r_dufuncrr GUFuncKernels   r.)numba.nprZ_Kernelr')r-rr.rr,rmake_gufunc_kernel s r0cs eZdZdZfddZZS) GUFuncLowererzHCallable class responsible for lowering calls to a specific gufunc. cs"ddlm}t|t|jdS)Nrr)r/rrrr0Znumpy_gufunc_kernel)rZgufuncrrrrr4s zGUFuncLowerer.__init__)r'r(r)r*rr+rrrrr11sr1c@seZdZdZdddidfddZddZd d Zed d Zd dZ d'ddZ ddZ ddZ ddZ ddZd'ddZddZeddZdd Zd!d"Zd#d$Zd%d&ZdS)(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. NFrcCsjd|_d|_||_||_t|||||||_|jjj|_|jjj|_|jj |_ | |j t ||dS)NF)ufunc_frozen _is_dynamic _identityrgufunc_builderpy_funcr'r*Znb_func _dispatcher _initialize functoolsupdate_wrapper)rr8r identitycache is_dynamic targetoptions writable_argsrrrrCs     zGUFunc.__init__cCs&||t||_|dSr) build_ufunc _install_typer1Z _lower_meZ _install_cg)r dispatcherrrrr:Vs  zGUFunc._initializec Cs6|j}t|j|j|j|j|j|j|j|j |j d }|S)N) r8r r=r>r?r@rAtypesigsfrozen) r7dictr8r r6r>r5r@rAZ_sigsr4)rgbdctrrr_reduce_states\s zGUFunc._reduce_statesc Cs<||||||||d} |D]} | | q | | | _| S)N)r8r r=r>r?r@rA)addrBr4) clsr8r r=r>r?r@rArErFrsigrrr_rebuildks zGUFunc._rebuildcCsd|jdS)Nz)r'rrrr__repr__wszGUFunc.__repr__cCs@|dur |jjj}td|jtft||jd}|||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()). NZ GUFuncTyping_)keyZgeneric) r9Z targetdescrZtyping_contexttyper'r rG_type_meZinsert_user_function)rZ typingctxZ_ty_clsrrrrCzs   zGUFunc._install_typecCs|j|dSr)r7rK)rZftyrrrrKsz GUFunc.addcCs|j|_|Sr)r7rBr3rOrrrrBs zGUFunc.build_ufunccCs0t|jj}ttt|dttt|dfSNr)rr7r tuplemaplen)r parsed_sigrrrexpected_ndimss $zGUFunc.expected_ndimsc Cs|rJ|j}|jj}|\}}||}t|sJt|t|D]F\}} t| tjri| j ||kri|t|krsz*GUFunc._get_ewise_dtypes..)rWr rr!appendr#)rrrbZtysZargtyrrrrs  zGUFunc._get_ewise_dtypescGs,t|jj}t|t|dt|dkSrT)rr7r rX)rrrYrrr_num_args_matchs  zGUFunc._num_args_matchc Gst|jj}||}g}t|dD] \}}t|}|dkr'|||q|t|||dqt|d}t|dD]\}}|||} t|pOd} |t| | dq@tj |S)NrArU) rr7r rr^rXrnrr!ra) rrrYrlrdZsig_dimr_offsetZrettyZret_ndimrrrris      zGUFunc._get_function_typecOs|js|js%|rt|dr|dj|dg|Ri|S|j|i|Sd|vr1||df7}|j|durCd|jd}t|| |}|jrQt |j|sf| |dksb|j |}| |||j|i|S)Nr__call__outFz Too few arguments for function 'z'. Note that the pattern `out = gufunc(Arg1, Arg2, ..., ArgN)` is not allowed. Use `gufunc(Arg1, Arg2, ..., ArgN, out) instead.r])r4r?_is_array_wrapper__array_ufunc__r3popror' TypeErrorrrrrirKrB)rrkwargsr&ZewiserMrrrrss.     zGUFunc.__call__r)r'r(r)r*rr:rJ classmethodrNrPrCrKrBrZrSr`rkpropertyr?rrorirsrrrrr2;s.    /   r2cCs t|t ot|dot|dS)adReturn True if obj wraps around numpy or another numpy-like library and is likely going to apply the ufunc to the wrapped array; False otherwise. At the moment, this returns True for - dask.array.Array - dask.dataframe.DataFrame - dask.dataframe.Series - xarray.DataArray - xarray.Dataset - xarray.Variable - pint.Quantity - other potential wrappers around dask array or dask dataframe We may need to add other libraries that pickle ufuncs from their __array_ufunc__ method in the future. Note that the below test is a lot more naive than `dask.base.is_dask_collection` (https://github.com/dask/dask/blob/5949e54bc04158d215814586a44d51e0eb4a964d/dask/base.py#L209-L249), # noqa: E501 because it doesn't need to find out if we're actually dealing with a dask collection, only that we're dealing with a wrapper. Namely, it will return True for a pint.Quantity wrapping around a plain float, a numpy.ndarray, or a dask.array.Array, and it's OK because in all cases Quantity.__array_ufunc__ is going to forward the ufunc call inwards. Z__dask_graph__rv)r rRhasattr)objrrrru#s ru)ZnumbarZ numba.corerZnumba.np.ufunc.ufuncbuilderrZnumba.np.ufunc.sigparserZnumba.np.ufunc.ufunc_baserrZnumba.np.numpy_supportrr r Znumba.core.typingr Znumba.core.typing.templatesr r r;r0r1Z ReduceMixinr2rurrrrs     $  i