o 8ήc'@s<dZddlZddlZddlZddlmZddlm Z m Z m Z m Z m Z ddlmZddlmZddlmZmZmZdZd Zd Zdd d Ze jfd dZddZddZddZddZddZ ddZ!ddZ"e!Z#ddZ$dd Z%d!d"Z&d#d$Z'd%d&Z(d'd(Z)d)d*Z*d+d,Z+d-d.Z,d/d0Z-d1d2Z.d3d4Z/d5d6Z0d7d8Z1d9d:Z2d;d<Z3d=d>Z4d?d@Z5dAdBZ6dCdDZ7dEdFZ8dGdHZ9dIdJZ:dKdLZ;dMdNZdSdTZ?dUdVZ@dWdXZAdYdZZBd[d\ZCd]d^ZDd_d`ZEdadbZFdcddZGdedfZHdgdhZIdidjZJdkdlZKdmdnZLdodpZMdqdrZNdsdtZOdudvZPdwdxZQdydzZRd{d|ZSd}d~ZTddZUddZVddZWddZXddZYddZZddZ[ddZ\ddZ]ddZ^ddZ_ddZ`ddZaddZbddZcddZdddZeddZfddZgddZhddZiddZjddZkddZlddZmddZnddZoddZpddZqddZrddZsddZtddZudd„ZvddĄZwddƄZxddȄZyddʄZzdd̄Z{dd΄Z|ddЄZ}dd҄Z~ddԄZddքZdd؄ZddڄZdd܄ZddބZddZddZddZddZddZddZddZddZddZddZddZddZdS)zCodegen for functions used as kernels in NumPy functions Typically, the kernels of several ufuncs that can't map directly to Python builtins N)impl_ret_untracked)typingtypeserrorsloweringcgutils)register_jitable) npdatetime) cmathimplmathimplnumbersg+eG?g&{?g9B.?cst||ksJt|j|ksJ|jd|dur}tfdd|jDr-|j|ksBddl}|jjj}d ||}J|dS)zchecks that the following are true: - args and sig.args have arg_count elements - all input types are homogeneous - return type is 'return_type' if provided, otherwise it must be homogeneous with the input types. rNc3s|]}|kVqdSN.0argtyr8/tmp/pip-target-vg8gfxp4/lib/python/numba/np/npyfuncs.py (sz/_check_arity_and_homogeneity..z"{0} called with invalid types: {1}) lenargsall return_typeinspect currentframef_backf_codeco_nameformat)sigrarityrrfnamemsgrrr_check_arity_and_homogeneitys " r$c sxj}}tj||gt|j}tj|||d} fddt ||jD} | | }  | t j |jS)Nnamecs g|] \}}||qSr)cast)rrargtybuildercontextrrr 8sz0_call_func_by_name_with_cast..)moduleget_argument_typellvmliteir FunctionTyperrrinsert_pure_functionzipcallr'rfloat64r) r+r*r r func_namermodltyfntyfn cast_argsresultrr)r_call_func_by_name_with_cast/s   r=c s<|jd}z||}Wnty$}z d|t|} t| d}~wwj} |tjvrt |} fdd|D} | g| } |gt |j}fdd|D}t j t j |}t| ||}|| | d}|Sfdd|jD}|j}t j ||}tj| ||d}||j|}|S)Nrz!No {0} function for real type {1}csg|]}t|qSr)ralloca_once_valuer)r*rrr,Xsz/_dispatch_func_by_name_type..csg|] }|qSr)get_value_type as_pointer)rrr+rrr,`scsg|]}|qSr)r.)ratyrArrr,hr%)rKeyErrorrstrr LoweringErrorr-rcomplex_domain make_complex _getpointerlistr/r0r1VoidTyperget_or_insert_functionr4loadr.rr2call_external_function)r+r*r rtable user_namerr6er#r7outptrargs call_args call_argtys call_argltysr9r:retvalargtypesrestyper)r*r+r_dispatch_func_by_name_type?s<         rZc Cst||d|\}}|jd}||d}||d}||d|jjd>} |d||} |d||} |d| |} || | } || | }|j|ddn\}}| |j }Wdn1sbwY|E|j }| ||}| ||}| d||}| d||}| ||}|d ||}|||}||||}|||}Wdn1swYWdn1swY||j}|||||||S) Nr==Flikely>!=)r$r get_constanttypewidth icmp_unsignedand_or_if_else basic_blocksdivsrem icmp_signedxorselectaddphi add_incoming)r+r*r rnumdenrZERO MINUS_ONEMIN_INT den_is_zeroden_is_minus_onenum_is_min_intcould_cause_sigfpe force_zerothen otherwisebb_then bb_otherwisedivr7 num_gt_zero den_gt_zero not_same_sign mod_not_zero needs_fixing fix_valueresult_otherwiser<rrrnp_int_sdiv_impl}sB             rcCst||d|\}}|jd}||d}|d||}|j} t||?|j} |||} |d||} |d||} | | | }|d| |}| ||}| |||}| || }Wdn1sgwY| |j}||| ||| |S)Nr[rrbra)r$rrcrfrjr if_unlikelyrlrmrnrgrorprqrdrr)r+r*r rrsrtrru den_not_zerobb_no_ifbb_ifr7rrrrrr final_modr<rrrnp_int_srem_impls*        rcCHt|||jd|j|}t|||jd|j|}|||j||gSNrr])rrrr make_tupler+r*r rrremrrrnp_int_sdivrem_implrc Cst||d|\}}|jd}||d}|d||}|j|dd:\} } | |j} Wdn1s6wY| |||} |j} Wdn1sQwYWdn1s`wY||j}| || | | | |S)Nr[rr^Fr_) r$rrcrfrirjudivrqrdrr)r+r*r rrsrtrru div_by_zeror}r~rrrr<rrrnp_int_udiv_impls&      rc Cst||d|\}}|jd}||d}|d||}|j} t|||j} |||} Wdn1s9wY||j } | || | | | | S)Nr[rrb) r$rrcrfrjrruremrqrdrr) r+r*r rrsrtrrurrrr7r<rrrnp_int_urem_impls      rcCrr)rrrrrrrrrnp_int_udivrem_implrrcCt||d|j|SNr[)r$fdivr+r*r rrrrnp_real_div_impls  rcCst||d|\}}|jd}||d}|||}|d||} |d||} |d||} || || | } || ||} ||| S)Nr[rrb<) r$rrcfrem fcmp_orderedrgrnrofadd)r+r*r rin1in2rrures res_ne_zero den_lt_zero res_lt_zerorrrrrnp_real_mod_impls      rcCrr)r$rrrrrnp_real_fmod_impls  rcCs8tj|jd}|||}|d||}||||S)Nrr)r/r0Constantrdfsubrro)r+r*rru arg_negatedarg_is_negativerrr_fabss rc sfdd|D\}}|j}|j}|j}|j} |jtfdd|||| fDs/Jdj} tjd} tjd} t |} t | } d| |} |\}}| d| | } d|| } ||} |z\}}| || | _ ||| _Wdn1swY|F | |}| |}||} | |}||}||}||}||}||| _||| _Wdn1swYWdn1swYWdn 1s wY|F || }||}| |} | |}||}||}||}||}||| _||| _Wdn1sZwYWd| SWd| S1swwY| S) Nc"g|] }jjd|dqSrvaluerHrrr*r+r rrr,)z'np_complex_div_impl..cg|]}|jkqSrrdriftyperrr,1rCmismatched typesr?>=r^)realimagrdr make_helperrr/r0rrrrirgrfmulrr _getvalue)r+r*r rrrin1rin1iin2rin2irRruONEin2r_absin2i_absin2r_abs_ge_in2i_absr}r~ in2r_is_zero in2i_is_zero in2_is_zeroinn_then inn_otherwiserattmp1tmp2scltmp3tmp4tmp5tmp6rr*r+rr rnp_complex_div_impl!sx &                    - --rcC.t||dtjdtjdi}t|||||dS)Nr[npy_logaddexpf npy_logaddexp logaddexpr$rfloat32r5rZr+r*r rdispatch_tablerrrnp_real_logaddexp_impln  rcCr)Nr[npy_logaddexp2fnpy_logaddexp2 logaddexp2rrrrrnp_real_logaddexp2_impl|rrcsf|\}}|jtfdd|DsJd|j\}}||||tj}||||tj}|||S)Nc3s|]}|jkVqdSr rrlltyperrrsz&np_int_truediv_impl..zmust have homogeneous types)rdrrr'rr5r)r+r*r rrsrtnumtydentyrrrnp_int_truediv_impls  rcCs.t||||}t|j|j}t||||fSr )rr signaturernp_real_floor_impl)r+r*r rrsrrrnp_real_floor_div_implsrcCrr)rrrrrrrrrnp_real_divmod_implrrc s jdj}t||}fdd|D\}}|j}|j} |j} |j} |jtfdd|| | | fDs;Jdtj d}  j } | | _t | }t | }d||}|\}}|5| | }| |}| |}||}| |}||}t||f| _Wdn1swY|5| | }||}| |}| |}| |}||}t||f| _Wdn1swYWd| SWd| S1swY| S)Nrcrrrrrrrr,rz-np_complex_floor_div_impl..crrrrrrrr,rCrrr)runderlying_floatrrrrrdrr/r0rrrrrrirrrrr)r+r*r r float_kind floor_sigrrrrrrrurRrrrr}r~rrrrrrrrrnp_complex_floor_div_implsV   &                rcCt||dt||||Srr$r complex_power_implrrrrnp_complex_power_impl rcCrr)r$r real_power_implrrrrreal_float_power_implrrcCrrrrrrrnp_complex_float_power_implrrcCrr)r$r gcd_implrrrr np_gcd_impl rc CsX|j\}}||kr|jksJJ|\}}dd}|||||} t|||j| S)NcSs$|dkrdSt||t||S)z7 Like gcd, heavily cribbed from Julia. r)absnpgcd)abrrrlcms$znp_lcm_impl..lcm)rrcompile_internalr) r+r*r rxtyytyxyr rrrr np_lcm_impls rcCst||d|d}|jd}|j}||d}||d}||d} ||td} |||} || _|| _tj t j g|gdR} || g} t ||| | }t||| | }t||| | }||||}||| | }||||| _| S)Nr]rrrnanr[)r$rrrcfloatrHrrrrrbooleanrnp_complex_ge_implnp_complex_eq_implnp_complex_lt_implro)r+r*r roprfloat_tyrurrvNANr<cmp_sigcmp_args arg1_ge_arg2 arg1_eq_arg2 arg1_lt_arg2 real_when_ge real_when_ngerrrnp_complex_sign_impls(       r$cCt||dt|d|S)Nr]z llvm.rintr$r call_fp_intrinsicrrrrnp_real_rint_impl2 r(c Cs|t||d|jd}|j}|j|||dd}|||}tj|gd}t||||jg|_t||||jg|_| S)Nr]rrr[) r$rrrHrrr(rrr) r+r*r rrrrrR inner_sigrrrnp_complex_rint_impl8s   r+cCrNr])r$r exp_implrrrrnp_real_exp_implKrr.cCrr,)r$r r-rrrrnp_complex_exp_implPrr/cCr)Nr] npy_exp2fnpy_exp2exp2rrrrrnp_real_exp2_implWrr3c Cs|t||d|jd}|j}|j|||dd}|||}||t}|||j|_|||j|_t |||| gSNr]rr) r$rrrHrc _NPY_LOGE2rrrr/r) r+r*r rrrrtmploge2rrrnp_complex_exp2_implcs    r8cCrr,)r$r log_implrrrrnp_real_log_implrrr:cCrr,)r$r r9rrrrnp_complex_log_implwrr;cCr)Nr] npy_log2fnpy_log2log2rrrrrnp_real_log2_impl~rr?cCnt||d|jd}|j}t||||}|j|||d}||t}|||j|_|||j |_ | Sr4) r$rrr;rHrc _NPY_LOG2Errrr)r+r*r rrrr6log2errrnp_complex_log2_impl   rCcCrr,)r$r log10_implrrrrnp_real_log10_implrrFcCr@r4) r$rrr;rHrc _NPY_LOG10Errrr)r+r*r rrrr6log10errrnp_complex_log10_implrDrIcCrr,)r$r expm1_implrrrrnp_real_expm1_implrrKcCst||d|jd}|j}tj|gd}||d}|j|||dd}t||||jg} |||} t ||||j g} t ||||j g} | | | } | | | | _ | | || _| S)Nr]rr[rr)r$rrrrrcrHr.rnp_real_cos_implrnp_real_sin_implrrr)r+r*r rrrfloat_unary_sigrvrr rRcos_imagsin_imagr6rrrnp_complex_expm1_impls     rQcCrr,)r$r log1p_implrrrrnp_real_log1p_implrrSc Cst||d|jd}|j}tj|gd}tj|gd}||d}|j|||dd} |||} || j|} t |||| | j g} t |||| j | g| _ t |||| g| _| S)Nr]rr[rr)r$rrrrrcrHrrnp_real_hypot_implrnp_real_atan2_implr:r) r+r*r rrrrNfloat_binary_sigrrrR real_plus_onelrrrnp_complex_log1p_impls"    rZcCrr,)r$r sqrt_implrrrrnp_real_sqrt_implrr\cCrr,)r$r r[rrrrnp_complex_sqrt_implrr]cC t||d||d|dSNr]r)r$mulrrrrnp_int_square_impl racCr^r_)r$rrrrrnp_real_square_implrbrccCs:t||dtj|jgd}t||||d|dgSNr]rTr)r$rrrr complex_mul_impl)r+r*r r binary_sigrrrnp_complex_square_impls  rgcs:t||dtddddfdd}|||||S)Nr]T)fastmathcSs$|dkr t| d St|dS)NrgUUUUUU?)rpowerrrrrcbrt s znp_real_cbrt_impl..cbrtcst|rtjS|Sr )risnanrrjrkrr_cbrts z np_real_cbrt_impl.._cbrt)r$rr )r+r*r rrnrrmrnp_real_cbrt_impls   roc Csdt||d|j}tj|gd}|||d|tj}|tjd}|||}|||tj|Srd) r$rrrr'rr5rcr) r+r*r rrrf in_as_floatrresult_as_floatrrrnp_int_reciprocal_impls  rrcCs*t||d||jd}|||dS)Nr]rr)r$rcrr)r+r*r rrrrrnp_real_reciprocal_impl,s rsc Cst||d|jd}|j}||d}||d}|j|||dd}|||} |j} |j} t||| } t||| } |d| | }| |\}}|0| | | }| | |}| | |}| ||}| ||}|| _| ||| _Wdn1swY|.| | | }| | |}| || }| ||}| ||| _| ||| _Wdn1swYWd| SWd| S1swY| S)Nr]rrrr<=)r$rrrcrHrrrrrirrrrr)r+r*r rrrrurrrRrrin1r_absin1i_absin1i_abs_le_in1r_absr}r~rtmp0dinv_dminus_rrrrnp_complex_reciprocal_impl2sJ                  r}cCrr,)r$r sin_implrrrrrM\rrMcCrr,)r$r r~rrrrnp_complex_sin_implarrcCrr,)r$r cos_implrrrrrLirrLcCrr,)r$r rrrrrnp_complex_cos_implnrrcCrr,)r$r tan_implrrrrnp_real_tan_implvrrcCrr,)r$r asin_implrrrrnp_real_asin_impl~rrcCrr,)r$r acos_implrrrrnp_real_acos_implrrcCrr,)r$r atan_implrrrrnp_real_atan_implrrcCrr)r$r atan2_float_implrrrrrVrrVcCrr)r$r hypot_float_implrrrrrUrrUcCrr,)r$r sinh_implrrrrnp_real_sinh_implrrcCst||d|jd}|j}tj|gd}||||d}|||}|j} |j} t|||| g} t |||| g} t |||| g} t |||| g}| | | |_| | ||_| SNr]rr[)r$rrrrrHrrrMrrLnp_real_cosh_implrr)r+r*r rrftyfsig1rrRxrxisxishxrcxichxrrrrnp_complex_sinh_impls   rcCrr,)r$r cosh_implrrrrrrrcCst||d|jd}|j}tj|gd}||||d}|||}|j} |j} t|||| g} t |||| g} t |||| g} t |||| g}| | | |_| | ||_| Sr)r$rrrrrHrrrLrrMrrr)r+r*r rrrrrrRrrrrrrrrrnp_complex_cosh_impls   rcCrr,)r$r tanh_implrrrrnp_real_tanh_implrrcCsnt||d|jd}|j}tj|gd}||d}||||d}|||} |j} |j} t |||| g} t |||| g} t |||| g}t |||| g}| | |}| | |}| | |}| | |}| ||}| ||}|||}|||}| ||}| ||}| ||}| ||}|||}|||}| ||| _| ||| _| S)Nr]rr[r)r$rrrrrcrHrrrMrLrrrrrrr)r+r*r rrrrrrrRrrsicishrchr_rsis_rcicsqr_rcsqr_icrzr{rs_rcis_icis_rcrs_icnumrnumirrrnp_complex_tanh_impls<                  rcCrr,)r$r asinh_implrrrrnp_real_asinh_implrrcCrr,)r$r acosh_implrrrrnp_real_acosh_implrrcCst||d|jd}tj|gd}|||d}|d}t|||||g}t|||||g} t||||g} t|||| g} t |||| | g} t||||| g} t |||| gS)Nr]rrTy?) r$rrrget_constant_genericr complex_add_implcomplex_sub_implr]rer;)r+r*r rrcsig2rr x_plus_one x_minus_onesqrt_x_plus_onesqrt_x_minus_one prod_sqrtlog_argrrrnp_complex_acosh_impl"s,      rcCrr,)r$r atanh_implrrrrnp_real_atanh_impl@rrcCr%)Nr]z llvm.floorr&rrrrrHr)rcCr%)Nr]z llvm.ceilr&rrrrnp_real_ceil_implQr)rcCr%)Nr]z llvm.truncr&rrrrnp_real_trunc_implZr)rcCr%)Nr]z llvm.fabsr&rrrrnp_real_fabs_implcr)rct||dtjd|jdfdd|D\}}|j}|j}|j}|j} d||} d|| } d||} d || } | | }| | }||S) Nr[rrcg|] }j|dqSrrHrr)rrr,tz&np_complex_ge_impl..raordr^r r$rrrrrrrgrh)r+r*r rrrrryryixr_gt_yr no_nan_xi_yixr_eq_yrxi_ge_yi first_term second_termrr)rrn    rcr) Nr[rrcrrrrr)rrr,rz&np_complex_le_impl..rrr^rtr)r+r*r rrrrrrrxr_lt_yrrrxi_le_yirrrr)rnp_complex_le_implrrct||dtjd|jdfdd|D\}}|j}|j}|j}|j} d||} d|| } d||} d|| } | | }| | }||S) Nr[rrcrrrrr)rrr,rz&np_complex_gt_impl..rarr^r)r+r*r rrrrrrrrrrxi_gt_yirrrr)rnp_complex_gt_implrrcr) Nr[rrcrrrrr)rrr,rz&np_complex_lt_impl..rrr^r)r+r*r rrrrrrrrrrxi_lt_yirrrr)rrrrc vt||dtjd|jdfdd|D\}}|j}|j}|j}|j} d||} d|| } | | S)Nr[rrcrrrrr)rrr,rz&np_complex_eq_impl..r^)r$rrrrrrrg) r+r*r rrrrrrrrxi_eq_yirr)rr  rc r)Nr[rrcrrrrr)rrr,rz&np_complex_ne_impl..rb)r$rrrrrfcmp_unorderedrh) r+r*r rrrrrrrxr_ne_yrxi_ne_yirr)rnp_complex_ne_implrrcCs8|j|||d}t||j}t||j}|||S)Nr)rHris_truerrrh)r+r*rval complex_valre_trueim_truerrr_complex_is_trues rcC>t||dtjdt||d}t||d}|||SNr[rrr])r$rrrrrgr+r*r rr r rrrnp_logical_and_impl rcCNt||dtjdt|||jd|d}t|||jd|d}|||Sr)r$rrrrrgrrrrnp_complex_logical_and_impl rcCrr)r$rrrrrhrrrrnp_logical_or_implrrcCrr)r$rrrrrhrrrrnp_complex_logical_or_implrrcCrr)r$rrrrrnrrrrnp_logical_xor_impl rrcCrr)r$rrrrrnrrrrnp_complex_logical_xor_implrrcC"t||dtjdt||dSNr]rr)r$rrris_falserrrrnp_logical_not_implrcCs4t||dtjdt|||jd|d}||Sr)r$rrrrnot_)r+r*r rr rrrnp_complex_logical_not_impl s rcC0t||d|\}}|d||}||||SNr[rr$rmro)r+r*r rarg1arg2 arg1_sge_arg2rrrnp_int_smax_impl/ rcCrrr$rfro)r+r*r rrr arg1_uge_arg2rrrnp_int_umax_impl6rrc Cht||d|\}}|d||}|d||}||||}|d||} || ||} |||| SNr[unorr$rror) r+r*r rrrarg1_nanany_nan nan_resultrnon_nan_resultrrrnp_real_maximum_impl= r c Csht||d|\}}|d||}|d||}||||}|d||} || ||} |||| Srr) r+r*r rrrarg2_nanr r rr rrrnp_real_fmax_implLrrcCt||d|jd}ttj|}tjtjg|gdR}|\}}t||||g} t||||g} || | } || ||} t ||||} || ||}|| | |SNr[r r$rrrrrnp_complex_isnan_implrhrorr+r*r rrbc_sigbcc_sigrrr rr r rr rrrnp_complex_maximum_impl[   rcCt||d|jd}ttj|}tjtjg|gdR}|\}}t||||g} t||||g} || | } || ||} t ||||} || ||}|| | |Srrrrrrnp_complex_fmax_implq   rcCrNr[rtr)r+r*r rrr arg1_sle_arg2rrrnp_int_smin_implrrcCrrr)r+r*r rrr arg1_ule_arg2rrrnp_int_umin_implrr!c CrNr[rrtr r+r*r rrrr r r  arg1_le_arg2r rrrnp_real_minimum_implrr%c Csht||d|\}}|d||}|d||}||||}|d||} || ||} |||| Sr"rr#rrrnp_real_fmin_implrr&cCrr r$rrrrrrrhrorr+r*r rrrrrrr rr r r$r rrrnp_complex_minimum_implrr)cCrrr'r(rrrnp_complex_fmin_implrr*cCt||dtjdtjSNr]rr$rrr false_bitrrrrnp_int_isnan_implr/cCrr)r$rrr is_nanrrrrnp_real_isnan_implrr2cC<t||dtjd|\}|j\}|j|||d}t||SNr]rr)r$rrrrHr r1r+r*r rrrrrrrrs  rcCr+r,)r$rrrtrue_bitrrrrnp_int_isfinite_implr0r7cC&t||dtjd|d|dtjS)Nr]rrbr)r$rrrfr NATrrrrnp_datetime_isfinite_implr:cCr8)Nr]rr^r)r$rrrmr r9rrrrnp_datetime_isnat_implr;r<cCrr)r$rrr is_finiterrrrnp_real_isfinite_implrr>cCr3r4)r$rrrrHr r=r5rrrnp_complex_isfinite_impl  r?cCr+r,r-rrrrnp_int_isinf_implr0rAcCrr)r$rrr is_infrrrrnp_real_isinf_implrrCcCr3r4)r$rrrrHr rBr5rrrnp_complex_isinf_implr@rDcCsbt||dtjdtjdtjdi}tjtjg|jR}t |||||d}| d|| d}|S)Nr]rnumba_signbitf numba_signbitsignbitrbr) r$rrrr5rrintcrrZrfrd)r+r*r rrr*int_resbool_resrrrnp_real_signbit_impl!s rKcCrr)r$r copysign_float_implrrrrnp_real_copysign_impl0rrMcCr)Nr[npy_nextafterf npy_nextafter nextafterrrrrrnp_real_nextafter_impl5rrQcCr)Nr] npy_spacingf npy_spacingspacingrrrrrnp_real_spacing_impl@rrUc CsH|\}}|j\}}||||tj}t||tj}t|||||fSr )rr'rrHrrr ldexp_impl) r+r*r rx1x2ty1ty2f_fi_sigrrrnp_real_ldexp_implLs  r\r )__doc__math llvmlite.irr/numpyrnumba.core.imputilsr numba.corerrrrrnumba.core.extendingrnumba.npr numba.cpythonr r r rArGr5r$r5r=rZrrrrrrnp_int_fmod_implrrrrrrrrrrrrrrrrr$r(r+r.r/r3r8r:r;r?rCrFrIrKrQrSrZr\r]rarcrgrorrrsr}rMrrLrrrrrrVrUrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr rrrrr!r%r&r)r*r/r2rr7r:r<r>r?rArCrDrKrMrQrUr\rrrrs     >%M6      *)