J/Ph RddlmZddlmZGddZGddZdS)) numpy_support)typesceZdZdZdZdZdS)UfuncLowererBasezHCallable class responsible for lowering calls to a specific gufunc. cP||_||_|||_g|_dSN)ufuncmake_ufunc_kernel_fnkernellibs)selfr make_kernel_fnr s Y/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/numba/np/ufunc/ufunc_base.py__init__zUfuncLowererBase.__init__s- $8!$nU++  cJ||||||j|jSr)r r r )r contextbuildersigargss r__call__zUfuncLowererBase.__call__s.(('3)-T[BB BrN)__name__ __module__ __qualname____doc__rrrrrrsA BBBBBrrc*eZdZedZedZedZedZedZedZ edZ edZ ed Z ed Z ed Zed Zd ZddZdZdS) UfuncBasec|jjSr)r ninr s rr z UfuncBase.nins z~rc|jjSr)r noutr!s rr#zUfuncBase.nouts zrc|jjSr)r nargsr!s rr%zUfuncBase.nargs zrc|jjSr)r ntypesr!s rr(zUfuncBase.ntypes! z  rc|jjSr)r rr!s rrzUfuncBase.types%r&rc|jjSr)r identityr!s rr,zUfuncBase.identity) z""rc|jjSr)r signaturer!s rr/zUfuncBase.signature-s z##rc|jjSr)r accumulater!s rr1zUfuncBase.accumulate1s z$$rc|jjSr)r atr!s rr3z UfuncBase.at5s z}rc|jjSr)r outerr!s rr5zUfuncBase.outer9r&rc|jjSr)r reducer!s rr7zUfuncBase.reduce=r)rc|jjSr)r reduceatr!s rr9zUfuncBase.reduceatAr-rcRt|jjdksJd|_dS)zI Disable the compilation of new signatures at call time. rTN)len _dispatcher overloads_frozenr!s rdisable_compilezUfuncBase.disable_compileEs. 4#-..2222 rNc|jjj}tj}tj}|fjjz|fjjzz}|fjjz}| fd||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()). Nc$g|] }j|f Sr) _lower_me).0rr s r z)UfuncBase._install_cg..]s" A A ASdndC ( A A Ar) r< targetdescrtarget_contextrAnyArrayr r r#insert_func_defn)r targetctx_any_arrsig0sig1s` r _install_cgzUfuncBase._install_cgMs  (4CIy{w'4'DJO*CCw'"" A A A AT4L A A A C C C C Crc(|jrJtj||}|dSt|j|jzdt |}|jj D]!\}}| ||r||fcS"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_typeslooprcress rfind_ewise_functionzUfuncBase.find_ewise_function_s < O 9$ LLD|!z dl :;;r^s"""""" B B B B B B B B^^^^^^^^^^r