o ZÛ·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ÚkernelÚlibs)ÚselfrÚmake_kernel_fnr©r úO/home/ubuntu/vllm_env/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__rrr 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 )Ú _lower_me)Ú.0rrr r Ú ]sz)UfuncBase._install_cg..) r(Ú targetdescrÚtarget_contextrÚAnyÚArrayrrrÚinsert_func_defn)r Ú targetctxÚ_anyÚ_arrÚsig0Úsig1r 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*rÚufunc_find_matching_loopÚtupleÚinputsÚoutputsr'r(r)ÚitemsÚmatch_signature)r Ú ewise_typesÚlooprÚcresr r r Úfind_ewise_function_s   ÿzUfuncBase.find_ewise_functionr)rrrÚpropertyrrrrrr r!r"r#r$r%r&r+r9rCr r r r rs8              rN)Únumba.nprÚ numba.corerrrr r r r Ús