o i @s8ddlmZddlmZGdddZGdddZdS)) numpy_support)typesc@s eZdZdZddZddZdS)UfuncLowererBasezHCallable class responsible for lowering calls to a specific gufunc. cCs ||_||_|||_g|_dSN)ufuncmake_ufunc_kernel_fnkernellibs)selfrmake_kernel_fnrr V/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/numba/np/ufunc/ufunc_base.py__init__s  zUfuncLowererBase.__init__cCs||||||j|jSr)rrr)r contextbuildersigargsr r r __call__s zUfuncLowererBase.__call__N)__name__ __module__ __qualname____doc__rrr r r r rs rc@seZdZeddZeddZeddZeddZed d Zed d Z ed dZ eddZ eddZ eddZ eddZeddZddZd ddZddZdS)! UfuncBasecC|jjSr)rninr r r r rz UfuncBase.nincCrr)rnoutrr r r rrzUfuncBase.noutcCrr)rnargsrr r r rrzUfuncBase.nargscCrr)rntypesrr r r r!rzUfuncBase.ntypescCrr)rrrr r r r%rzUfuncBase.typescCrr)ridentityrr r r r )rzUfuncBase.identitycCrr)r signaturerr r r r!-rzUfuncBase.signaturecCrr)r accumulaterr r r r"1rzUfuncBase.accumulatecCrr)ratrr r r r#5rz UfuncBase.atcCrr)routerrr r r r$9rzUfuncBase.outercCrr)rreducerr r r r%=rzUfuncBase.reducecCrr)rreduceatrr r r r&ArzUfuncBase.reduceatcCst|jjdks Jd|_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}|fjj|fjj}|fjj}| fdd||fDdS)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|fqSr ) _lower_me).0rrr r ]sz)UfuncBase._install_cg..) r( targetdescrtarget_contextrAnyArrayrrrinsert_func_defn)r targetctx_any_arrsig0sig1r rr _install_cgMs zUfuncBase._install_cgcCsl|jrt||}|durdSt|j|jdt|}|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*rufunc_find_matching_looptupleinputsoutputsr'r(r)itemsmatch_signature)r ewise_typeslooprcresr r r find_ewise_function_s   zUfuncBase.find_ewise_functionr)rrrpropertyrrrrrr r!r"r#r$r%r&r+r9rCr r r r rs8              rN)numba.npr numba.corerrrr r r r s