o ç0 i ã@s8ddlmZddlmZGdd„dƒZGdd„dƒZdS)é)Ú numpy_support)Útypesc@s eZdZdZdd„Zdd„ZdS)ÚUfuncLowererBasezHCallable class responsible for lowering calls to a specific gufunc. cCs ||_||_||ƒ|_g|_dS©N)ÚufuncÚmake_ufunc_kernel_fnÚkernelZlibs)ÚselfrZmake_kernel_fnr©r úe/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/numba/np/ufunc/ufunc_base.pyÚ__init__s  zUfuncLowererBase.__init__cCs| |||||j|j¡Sr)rrr)r ÚcontextÚbuilderÚsigÚargsr r r Ú__call__s ÿzUfuncLowererBase.__call__N)Ú__name__Ú __module__Ú __qualname__Ú__doc__r rr r r r rs rc@s¶eZdZedd„ƒZedd„ƒZedd„ƒZedd„ƒZed d „ƒZed d „ƒZ ed d„ƒZ edd„ƒZ edd„ƒZ edd„ƒZ edd„ƒZedd„ƒZdd„Zd dd„Zdd„ZdS)!Ú UfuncBasecCó|jjSr)rÚnin©r r r r róz UfuncBase.nincCrr)rÚnoutrr r r rrzUfuncBase.noutcCrr)rÚnargsrr r r rrzUfuncBase.nargscCrr)rÚntypesrr r r r!rzUfuncBase.ntypescCrr)rrrr r r r%rzUfuncBase.typescCrr)rÚidentityrr r r r)rzUfuncBase.identitycCrr)rÚ signaturerr r r r-rzUfuncBase.signaturecCrr)rÚ accumulaterr r r r 1rzUfuncBase.accumulatecCrr)rÚatrr r r r!5rz UfuncBase.atcCrr)rÚouterrr r r r"9rzUfuncBase.outercCrr)rÚreducerr r r r#=rzUfuncBase.reducecCrr)rÚreduceatrr r r r$ArzUfuncBase.reduceatcCst|jjƒdks J‚d|_dS)zI Disable the compilation of new signatures at call time. rTN)ÚlenÚ _dispatcherÚ overloadsÚ_frozenrr r r Údisable_compileEs zUfuncBase.disable_compileNcsh|dur ˆjjj}tj}tj}|fˆjj|fˆjj}|fˆjj}|  ‡fdd„||fDƒ¡dS)a4 Install an implementation function for a GUFunc/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ˆ|f‘qSr )Z _lower_me)Ú.0rrr r Ú ]sz)UfuncBase._install_cg..) r&Z targetdescrZtarget_contextrÚAnyZArrayrrrZinsert_func_defn)r Z targetctxZ_anyZ_arrZsig0Zsig1r rr Ú _install_cgMs ÿzUfuncBase._install_cgcCsl|jrt ||¡}|durdSt|j|jƒdt|ƒ…}|jj  ¡D]\}}|  ||¡r3||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_loopÚtupleZinputsZoutputsr%r&r'ÚitemsZmatch_signature)r Z ewise_typesÚlooprZcresr r r Úfind_ewise_function_s   ÿzUfuncBase.find_ewise_functionr)rrrÚpropertyrrrrrrrr r!r"r#r$r)r-r1r r r r rs8              rN)Znumba.nprZ numba.corerrrr r r r Ús