σ \ K]c@`sώddlmZmZmZddlZddlmZmZm Z m Z m Z m Z ddl mZddlmZmZddlmZmZdd lmZdd lmZd „Zd efd „ƒYZdejfd„ƒYZejjeƒdS(i(tabsolute_importtprint_functiontdivisionNi(tjitttypeoftutilsttypest numpy_supporttsigutils(tnpydecl(tAbstractTemplatet signaturei(t _internalt ufuncbuilder(t Dispatcher(tarray_analysisc`sKddlm}d|jf‡‡fd†ƒY‰ˆjˆjj7_ˆS(Ni(tnpyimplt DUFuncKernelc`s,eZdZˆZ‡fd†Zd„ZRS(sΙ npyimpl._Kernel subclass responsible for lowering a DUFunc kernel (element-wise function) inside a broadcast loop (which is generated by npyimpl.numpy_ufunc_kernel()). c`sAtˆ|ƒj|||ƒ|jj|jƒ\|_|_dS(N(tsupert__init__tdufunctfind_ewise_functiontargst inner_sigtcres(tselftcontexttbuildert outer_sig(R(s4lib/python2.7/site-packages/numba/npyufunc/dufunc.pyRs c W`s9|j}|j}gt||j|jƒD]$\}}}|j|||ƒ^q+}|jjr’|jjj t j t j gt |jƒƒ}n|jjj |j |jƒ}|jjjj} | j|d|jjjƒ} | jjdƒ|jjj|j| |j |j|ƒ\} } |j| |j |j ƒS(Ntnamet alwaysinline(RRtzipRtcastRt objectmodeRt call_convtget_function_typeRtpyobjecttlent return_typeRtblocktfunctiontmoduletget_or_insert_functiontfndesctllvm_func_namet attributestaddt call_function( RRtisigtosigtvaltintytouttyt cast_argst func_typeR)t entry_pointt_tres((s4lib/python2.7/site-packages/numba/npyufunc/dufunc.pytgenerates"  @  %  (t__name__t __module__t__doc__RRR:((Rt_dufunc(s4lib/python2.7/site-packages/numba/npyufunc/dufunc.pyRs(ttargetsRt_KernelR;tufunc(R>R((RR>s4lib/python2.7/site-packages/numba/npyufunc/dufunc.pytmake_dufunc_kernel s""t DUFuncLowerercB`s eZdZd„Zd„ZRS(sHCallable class responsible for lowering calls to a specific DUFunc. cC`st|ƒ|_g|_dS(N(RBtkerneltlibs(RR((s4lib/python2.7/site-packages/numba/npyufunc/dufunc.pyR8scC`sPddlm}t|ƒ|jjjjk}|j|||||jd|ƒS(Ni(Rtexplicit_output(R?RR%RDRRAtnintnumpy_ufunc_kernel(RRRtsigRRRF((s4lib/python2.7/site-packages/numba/npyufunc/dufunc.pyt__call__<s  (R;R<R=RRJ(((s4lib/python2.7/site-packages/numba/npyufunc/dufunc.pyRC5s tDUFunccB`sΰeZdZedƒZdeid„Zd„Ze d„ƒZ e d„ƒZ e d „ƒZ e d „ƒZ e d „ƒZe d „ƒZd „Zd„Zd„Zdd„Zdd„Zd„Zd„Zdd„ZRS(sΉ Dynamic universal function (DUFunc) intended to act like a normal Numpy ufunc, but capable of call-time (just-in-time) compilation of fast loops specialized to inputs. tidentityt _keepaliveRGtnoutcC`sΆt|tƒr|j}n|jƒ|_i}tj|ƒ|d”s(RARGtpopt TypeErrortjointsortedR%RNRaRtis_arrayscalartappendtmap_arrayscalar_typeRRRRtArraytdtypeRfttuple( RRRZRGRgtargs_lentargtystargtargty((s4lib/python2.7/site-packages/numba/npyufunc/dufunc.pyt_compile_for_argsŒs(  ) (   c C`sψ|jrtd|fƒ‚nt|tƒs4t‚|dkrI|}n ||Œ}tj|j|j |ƒ\}}}tj |||ƒ}tj ||ƒ\}}}|j t j|ƒ|ƒ|jj||j|fƒ|jjj|jƒ|S(s/ Given a tuple of argument types (these should be the array dtypes, and not the array types themselves), compile the element-wise function for those inputs, generate a UFunc loop wrapper, and register the loop with the Numpy ufunc object for this DUFunc. scompilation disabled for %sN(Rct RuntimeErrorRRRtRatNoneR t_compile_element_wise_functionR_RUt_finalize_ufunc_signaturet!_build_element_wise_ufunc_wrappert _add_loopRtlongintRMRptlibraryRXRE( RRvR&RIRt actual_sigt dtypenumstptrtenv((s4lib/python2.7/site-packages/numba/npyufunc/dufunc.pyRf¨s     cC`sc|dkr|jjj}ntd|jjtftd|d|j ƒƒ}|j ||ƒdS(s*Constructs and installs a typing class for a DUFunc object in the input typing context. If no typing context is given, then _install_type() installs into the typing context of the dispatcher object (should be same default context used by jit() and njit()). t DUFuncTyping_tkeytgenericN( R{R_t targetdescrttyping_contextttypeRAR;R tdictt_type_metinsert_user_function(Rt typingctxt_ty_cls((s4lib/python2.7/site-packages/numba/npyufunc/dufunc.pyRWΒs  cC`sŽ|jrNtj||ƒ}|dkr+dSt|j|jƒt|ƒ }nx9|jj j ƒD]%\}}|j |kra||fSqaWdS(s 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(NN( RcRtufunc_find_matching_loopR{RttinputstoutputsR%R_R`titemsR(Rt ewise_typestloopRIR((s4lib/python2.7/site-packages/numba/npyufunc/dufunc.pyRΠs  #cC`s~| s t‚|j}tjj|||ƒ}|\}}}}t|ƒ} | dkrrt|t|ƒ ƒ} n t|ƒ} |j| ƒ\} } | dkrψ|j rΑt d||fƒ‚n|j | ƒ|j| ƒ\} } | dk sψt‚n| dkrt |ƒ} nT|j dkr[|dkrLtj| j||ƒg} qg| jg} n tdƒ‚| j|ƒt| ŒS(sμ Implement AbstractTemplate.generic() for the typing class built by DUFunc._install_type(). Return the call-site signature after either validating the element-wise signature or compiling for it. iscannot call %s with types %sistyping gufuncs (nout > 1)N(RaRAR tNumpy_rules_ufunct_handle_inputsR%RtRR{RcRlRftlistRNRRrR&tNotImplementedErrortextendR (RRvtkwtysRAt_handle_inputs_resultt base_typestexplicit_outputstndimstlayouttexplicit_output_countR•RIRtouttys((s4lib/python2.7/site-packages/numba/npyufunc/dufunc.pyRδs4             cC`s|dkr|jjj}ntj}tj}|f|jj|f|jj }|f|jj}|j g||fD]}|j ||f^qzƒdS(s- Install an implementation function for a 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{R_R‰ttarget_contextRtAnyRrRARGRNtinsert_func_defnRX(Rt targetctxt_anyt_arrtsig0tsig1RI((s4lib/python2.7/site-packages/numba/npyufunc/dufunc.pyRY s   $(RLRMRGRNN(R;R<R=tsett_DUFunc__base_kwargsR{tFalseRR\tpropertyRGRNR]R^RRLRdR.RyRfRWRRRY(((s4lib/python2.7/site-packages/numba/npyufunc/dufunc.pyRKDs$         ((t __future__RRRtnumpytnptRRRRRRttypingR ttyping.templatesR R R R R[RRRBtobjectRCt_DUFuncRKt MAP_TYPESRp(((s4lib/python2.7/site-packages/numba/npyufunc/dufunc.pyts . )Ϊ