U |ed<@sdZddlZddlZddlZddlZddlZddlmZddl m Z m Z ddl m Z ddlmZmZmZmZddlmZddlmZdd lmZe d ZejZeed ZejZej Z!eed Z"e"jZ#e"j Z$d Z%dZ&dZ'dZ(ddZ)ddZ*ddZ+ddZ,ddZ-ddZ.ddZ/dd Z0d!d"Z1d#d$Z2d%d&Z3d'd(Z4dd*d+Z5e4ej6d,e4ej7d-Z8e4ej9d.Z:e4ej;d/Ze4ej?d1Z@e5ejAd2d3ZBe5ejCd4d5ZDe5ejEd6d7ZFe5ejGd8d9ZHe5ejId:d;ZJe5ejKdd?ZNe5ejOd@dAZPe5ejQdBdCZRe5ejSdDdEZTe5ejUdFdGZVe5ejWdHdIZXe5ejYdJdKZZe5ej[dLdMZ\e5ej]dNdOZ^e5ej_dPdQdRZ`e5ejadSdTdRZbe5ejcdUdVZde5ejedWdXZfe5ejgdYdZdRZhe5ejid[d\Zjeejkejld]d^Zmeejkejnd_d`ZoeejpejldadbZqeejpejndcddZreejsejldedfZteejsejndgdhZueejvejlejldidjZweejxejldkdlZyeejzejlej{dmdnZ|eej}ej~ej~dodpZeej}ejejdqdrZeej}ejlejldsdtZeejej~ej~dudvZeejejejdwdxZeejejlejldydzZeejejld{d|Ze3ejeeejejld}d~Ze3ejeeejejlejleejejlejnddZddZeeddZddZeejejnejnedS)zA Provide math calls that uses intrinsics or libc math functions. N)Constant)Registryimpl_ret_untracked)typeof)typesutilsconfigcgutils)overload) signature)trailing_zerosmathimplfloat32float64illlcCs|d||S)z< Return a condition testing whether *val* is a NaN. uno)fcmp_unorderedbuildervalrS/var/www/website-v5/atlas_env/lib/python3.8/site-packages/numba/cpython/mathimpl.pyis_nan(srcCsHt|jtd}t|jtd}|d||}|d||}|||S)zB Return a condition testing whether *val* is an infinite. z+infz-infz==)rtypefloat fcmp_orderedor_)rrZpos_infZneg_infisposinfisneginfrrris_inf.s rcCs|||}|d||S)z? Return a condition testing whether *val* is a finite. ord)fsubr)rrZ val_minus_valrrr is_finite8s r!cCs(|jtjkst||tjdS)z1 Bitcast a double into a 64-bit integer. @)rllvmliteir DoubleTypeAssertionErrorbitcastIntTyperrrr f64_as_int64@sr)cCs(|jtjdkst||tjS)z1 Bitcast a 64-bit integer into a double. r")rr#r$r(r&r'r%rrrr int64_as_f64Gsr*cCs(|jtjkst||tjdS)z0 Bitcast a float into a 32-bit integer. )rr#r$ FloatTyper&r'r(rrrr f32_as_int32Nsr-cCs(|jtjdkst||tjS)z0 Bitcast a 32-bit integer into a float. r+)rr#r$r(r&r'r,rrrr int32_as_f32Usr.cCs|t|jd|S)zB Negate real number *val*, with proper handling of zeros. g)r rrrrrr negate_real\sr/cCs(|j}||dd|D}|||S)z9 Call a LLVM intrinsic floating-point operation. cSsg|] }|jqSr)r).0arrr hsz%call_fp_intrinsic..)moduledeclare_intrinsiccall)rnameargsmodZintrrrrcall_fp_intrinsiccsr9csfdd}|S)z Return an implementation factory to convert the single integral input argument to a float64, then defer to the *wrapped_impl*. c sT|\}|jd}||||tj}ttjtj}||||f}|||tj|jS)Nr)r7castrrr return_type) contextrsigr7r input_typeZfpval inner_sigres wrapped_implrr implementerqs  z2_unary_int_input_wrapper_impl..implementerr)rBrCrrAr_unary_int_input_wrapper_implls rDcCst|}t|tj|dSN)rDlowerrInteger)fn float_implimplrrrunary_math_int_impl{srKcs&t|tjfdd}t|||S)zO Implement the math function *fn* using the LLVM intrinsic *intrcode*. cst||}t|||j|SrE)r9rr;r<rr=r7r@intrcoderrrIs z#unary_math_intr..float_impl)rFrFloatrK)rHrNrIrrMrunary_math_intrs  rPFcs:|r tjnd}fdd}t|tj|t|||S)a! Register implementations of Python function *fn* using the external function named *f32extern* and *f64extern* (for float32 and float64 inputs, respectively). If *int_restype* is true, then the function's return value should be integral, otherwise floating-point. Nc s|\}|j}|jd}||}tjtji|}tj||g} t j |j| |d} | | |f} | || ||j } t|||j | S)z9 Implement *fn* for a types.Float input. rr6)r3r7get_value_typerrrr#r$ FunctionTyper insert_pure_functionr5r:r;r) r<rr=r7rr8r>lty func_namefntyrHr@ f32extern f64externrrrIs   z%unary_math_extern..float_impl)rint64rFrOrK)rHrYrZZ int_restypeZ f_restyperIrrXrunary_math_externs  r\z llvm.fabszllvm.expzllvm.logz llvm.log10zllvm.sinzllvm.coslog1pflog1pexpm1fexpm1erfferferfcferfctanftanasinfasinacosfacosatanfatanasinhfasinhacoshfacoshatanhfatanhsinhfsinhcoshfcoshtanhftanhlog2flog2ceilfceilTfloorffloorZ numba_gammafZ numba_gammasqrtfsqrttruncftrunclgammaflgammacCs |\}t||}t|||j|SrE)rrr;r<rr=r7rr@rrrisnan_float_impls rcCstj}t|||j|SrEr false_bitrr;rLrrrisnan_int_implsrcCs |\}t||}t|||j|SrE)rrr;rrrrisinf_float_impls rcCstj}t|||j|SrErrLrrrisinf_int_implsrcCs |\}t||}t|||j|SrE)r!rr;rrrrisfinite_float_impls rcCstj}t|||j|SrE)r true_bitrr;rLrrrisfinite_int_implsrcCsN|dj}|j}t|tj|||fd|j}|||}t |||j |S)Nrzllvm.copysign.%s) rr3r get_or_insert_functionr#r$rSintrinsic_namer5rr;)r<rr=r7rUr8rHr@rrrcopysign_float_impls  rc Cs|\}||jd}||jd}tj||dd}tj||tj|f}dddt |} t |j || } | | ||f} t || ||f} t|||j| S)NrexprQZ numba_frexpfZ numba_frexprdouble) get_data_typer7r;r alloca_oncer#r$rS PointerTypestrrr3r5make_anonymous_structloadr) r<rr=r7rflttyinttyZexpptrrWfnamerHr@rrr frexp_implsrc Csp|\}}t|j|j\}}tj|||f}dddt|} tj|j || d} | | ||f} t |||j | S)NZ numba_ldexpfZ numba_ldexprrQ) maprr7r#r$rSrr rTr3r5rr;) r<rr=r7rrrrrWrrHr@rrr ldexp_implsrcCsP|\}}||tj}||tj}ttjtjtj}t|||||fSrE)sitofpr#r$r%r rratan2_float_implr<rr=r7yxfsigrrratan2_s64_impl%s rcCsP|\}}||tj}||tj}ttjtjtj}t|||||fSrE)uitofpr#r$r%r rrrrrrratan2_u64_impl-s rc Cs~t|dkst|j}|jd}||}tjdtjdi|}tj |||f}t j |j||d} | | |} t|||j| S)Nratan2fatan2rQ)lenr&r3r7rRrrrr#r$rSr rTr5rr;) r<rr=r7r8tyrUrVrWrHr@rrrr5s   rcCs`|\}}||tj}||tj}ttjtjtj}t|||||f}t|||j |SrE rr#r$r%r rrhypot_float_implrr;r<rr=r7rrrr@rrrhypot_s64_implHs rcCs`|\}}||tj}||tj}ttjtjtj}t|||||f}t|||j |SrErrrrrhypot_u64_implRs rc s|j\}}||kr |jks&nt|\}}tjtjdkr@dndtjtjdkrTdndi|}t||tjdkrt j dkr|t dfdd } n fd d } | || ||} t |||j| S) Nwin32Z_hypotfZhypotfZ_hypothypotr+infcs"t|st|rS||SrE)mathisinfrrr plat_hypotrr hypot_implmsz$hypot_float_impl..hypot_implcs ||SrErr)rrrrrs)r7r;r&rrsysplatformrExternalFunctionr MACHINE_BITSrcompile_internalr) r<rr=r7xtyytyrrrrr@rrrr\s"    rcCs6|\}||jtjd}|||}t|||j|SN get_constantr;rpifmulrr<rr=r7rcoefr@rrrradians_float_impl{s rcCs6|\}||jdtj}|||}t|||j|Srrrrrrdegrees_float_impls rcCs|tj|}|||SrE) get_functionoperatorpow)r<rr=r7rJrrrpow_implsrcCsdS)z8Convert integer to unsigned integer of equivalent width.NrTrrr _unsignedsrcs>|tjkrddS|tjkr:ttd|jfddSdS)NcSs|SrErrrrrz _unsigned_impl..zuint{}cs|SrErrZnewTrrrr)runsigned_domain signed_domaingetattrformatbitwidthrrrr_unsigned_impls   rc CsV|j\}}||kr |jks&nt|\}}dd}|||||} t|||j| S)Nc Sst|}|dkrt|S|dkr(t|St|}t|}t||}ttt||}ttt||}||kr||kr||}}||8}t|t|}qjt|||}|S)zO Stein's algorithm, heavily cribbed from Julia implementation. r)rabsr minrnp right_shift left_shift) r1brZzaZzbkuvrrrrgcds"  zgcd_impl..gcd)r7r;r&rr) r<rr=r7rrrrrr@rrrgcd_impls  r)F)__doc__rrrnumpyr llvmlite.irr#rnumba.core.imputilsrrnumbar numba.corerrrr numba.core.extendingr numba.core.typingr numba.cpython.unsafe.numbersr registryrFfinfodtypeZ _NP_FLT_FINFOmaxFLT_MAXtinyFLT_MINZ _NP_DBL_FINFODBL_MAXZDBL_MINZFLOAT_ABS_MASKZFLOAT_SIGN_MASKZDOUBLE_ABS_MASKZDOUBLE_SIGN_MASKrrr!r)r*r-r.r/r9rDrKrPr\fabsrexp_implloglog_impllog10 log10_implsinsin_implcoscos_implr^ log1p_implr` expm1_implrbZerf_implrdZ erfc_implrftan_implrh asin_implrj acos_implrl atan_implrn asinh_implrp acosh_implrr atanh_implrt sinh_implrv cosh_implrx tanh_implrzZ log2_implr|Z ceil_implr~Z floor_implgamma gamma_implr sqrt_implrZ trunc_implrZ lgamma_implisnanrOrrGrrrrisfiniterrcopysignrfrexprldexpintcrrr[ruint64rrrrrrradiansrdegreesrrrrrrrrrrrs        $