xKg DddlmZddlmZGddZGddZy)) numpy_support)typesceZdZdZdZdZy)UfuncLowererBasezHCallable class responsible for lowering calls to a specific gufunc. cH||_||_|||_g|_yN)ufuncmake_ufunc_kernel_fnkernellibs)selfr make_kernel_fnr s ]/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/numba/np/ufunc/ufunc_base.py__init__zUfuncLowererBase.__init__s$ $8!$U+  cV|j|||||j|jSr)r r r )r contextbuildersigargss r__call__zUfuncLowererBase.__call__s,(('3)-T[[B BrN)__name__ __module__ __qualname____doc__rrrrrrs BrrceZdZedZedZedZedZedZedZ edZ edZ ed Z ed Z ed Zed Zd ZddZdZy) UfuncBasec.|jjSr)r ninr s rr z UfuncBase.ninszz~~rc.|jjSr)r noutr!s rr#zUfuncBase.noutszzrc.|jjSr)r nargsr!s rr%zUfuncBase.nargszzrc.|jjSr)r ntypesr!s rr(zUfuncBase.ntypes!zz   rc.|jjSr)r rr!s rrzUfuncBase.types%r&rc.|jjSr)r identityr!s rr,zUfuncBase.identity)zz"""rc.|jjSr)r signaturer!s rr/zUfuncBase.signature-szz###rc.|jjSr)r accumulater!s rr1zUfuncBase.accumulate1szz$$$rc.|jjSr)r atr!s rr3z UfuncBase.at5szz}}rc.|jjSr)r outerr!s rr5zUfuncBase.outer9r&rc.|jjSr)r reducer!s rr7zUfuncBase.reduce=r)rc.|jjSr)r reduceatr!s rr9zUfuncBase.reduceatAr-rcZt|jjdkDsJd|_y)zI Disable the compilation of new signatures at call time. rTN)len _dispatcher overloads_frozenr!s rdisable_compilezUfuncBase.disable_compileEs* 4##--.222 rNc| |jjj}tj}tj }|f|j jz|f|j jzz}|f|j jz}|j||fDcgc]}|j||fc}ycc}w)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()). N) r< targetdescrtarget_contextrAnyArrayr r r#insert_func_defn _lower_me)r targetctx_any_arrsig0sig1rs r _install_cgzUfuncBase._install_cgMs  ((44CCIyy{{w'4'DJJOO*CCw'""594L ALSdnndC (L A C As#Cc2|jrGtj||}|yt|j|j zdt |}|jjjD]\}}|j||s||fcSy)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 << 99$ LD|! dll :;r[s#" B B^^r