U ,dd<@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_zerosZmathimplfloat32float64illlcCs|d||S)z< Return a condition testing whether *val* is a NaN. Zuno)Zfcmp_unorderedbuildervalr:/tmp/pip-unpacked-wheel-eu7e0c37/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_infZisposinfZisneginfrrris_inf.s rcCs|||}|d||S)z? Return a condition testing whether *val* is a finite. ord)fsubr)rrZ val_minus_valrrr is_finite8s rcCs(|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)rrrr#r!r"r rrrr int64_as_f64Gsr%cCs(|jtjkst||tjdS)z0 Bitcast a float into a 32-bit integer. )rrr FloatTyper!r"r#rrrr f32_as_int32Nsr(cCs(|jtjdkst||tjS)z0 Bitcast a 32-bit integer into a float. r&)rrrr#r!r"r'rrrr int32_as_f32Usr)cCs|t|jd|S)zB Negate real number *val*, with proper handling of zeros. g)rrrrrrr negate_real\sr*cCs(|j}||dd|D}|||S)z9 Call a LLVM intrinsic floating-point operation. cSsg|] }|jqSr)r).0arrr hsz%call_fp_intrinsic..)moduleZdeclare_intrinsiccall)rnameargsmodZintrrrrcall_fp_intrinsiccsr3csfdd}|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)r1castrrr return_type) contextrsigr1r input_typeZfpvalZ inner_sigres wrapped_implrr implementerqs  z2_unary_int_input_wrapper_impl..implementerr)r;r<rr:r_unary_int_input_wrapper_implls r=cCst|}t|tj|dSN)r=lowerrInteger)fn float_implimplrrrunary_math_int_impl{srDcs&t|tjfdd}t|||S)zO Implement the math function *fn* using the LLVM intrinsic *intrcode*. cst||}t|||j|Sr>)r3rr5r6rr7r1r9intrcoderrrBs z#unary_math_intr..float_impl)r?rFloatrD)rArGrBrrFrunary_math_intrs  rIFcs:|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. rr0)r.r1get_value_typerr rrr FunctionTyper insert_pure_functionr/r4r5r) r6rr7r1rr2r8lty func_namefntyrAr9 f32extern f64externrrrBs   z%unary_math_extern..float_impl)rint64r?rHrD)rArRrSZ int_restypeZ f_restyperBrrQrunary_math_externs  rUz llvm.fabszllvm.expzllvm.logz llvm.log10zllvm.sinzllvm.cosZlog1pflog1pZexpm1fexpm1ZerfferfZerfcferfcZtanftanZasinfasinZacosfacosZatanfatanZasinhfasinhZacoshfacoshZatanhfatanhZsinhfsinhZcoshfcoshZtanhftanhZlog2flog2ZceilfceilTZfloorffloorZ numba_gammafZ numba_gammaZsqrtfsqrtZtruncftruncZlgammaflgammacCs |\}t||}t|||j|Sr>)rrr5r6rr7r1rr9rrrisnan_float_impls rkcCstj}t|||j|Sr>r Z false_bitrr5rErrrisnan_int_implsrmcCs |\}t||}t|||j|Sr>)rrr5rjrrrisinf_float_impls rncCstj}t|||j|Sr>rlrErrrisinf_int_implsrocCs |\}t||}t|||j|Sr>)rrr5rjrrrisfinite_float_impls rpcCstj}t|||j|Sr>)r Ztrue_bitrr5rErrrisfinite_int_implsrqcCsN|dj}|j}t|tj|||fd|j}|||}t |||j |S)Nrzllvm.copysign.%s) rr.r get_or_insert_functionrrrLZintrinsic_namer/rr5)r6rr7r1rNr2rAr9rrrcopysign_float_impls  rsc Cs|\}||jd}||jd}tj||dd}tj||tj|f}dddt |} t |j || } | | ||f} t || ||f} t|||j| S)NrexprJZ numba_frexpfZ numba_frexprdouble) get_data_typer1r5r Z alloca_oncerrrLZ PointerTypestrrrr.r/Zmake_anonymous_structloadr) r6rr7r1rflttyinttyZexpptrrPfnamerAr9rrr frexp_implsr~c Csp|\}}t|j|j\}}tj|||f}dddt|} tj|j || d} | | ||f} t |||j | S)NZ numba_ldexpfZ numba_ldexprvrJ) maprxr1rrrLryr rMr.r/rr5) r6rr7r1rrur{r|rPr}rAr9rrr ldexp_implsrcCsP|\}}||tj}||tj}ttjtjtj}t|||||fSr>)sitofprrr r rratan2_float_implr6rr7r1yxfsigrrratan2_s64_impl%s rcCsP|\}}||tj}||tj}ttjtjtj}t|||||fSr>)Zuitofprrr r rrrrrrratan2_u64_impl-s rc Cs~t|dkst|j}|jd}||}tjdtjdi|}tj |||f}t j |j||d} | | |} t|||j| S)NrZatan2fatan2rJ)lenr!r.r1rKrr rrrrLr rMr/rr5) r6rr7r1r2tyrNrOrPrAr9rrrr5s   rcCs`|\}}||tj}||tj}ttjtjtj}t|||||f}t|||j |Sr> rrrr r rrhypot_float_implrr5r6rr7r1rrrr9rrrhypot_s64_implHs rcCs`|\}}||tj}||tj}ttjtjtj}t|||||f}t|||j |Sr>rrrrrhypot_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||Sr>)mathisinfrrr plat_hypotrr hypot_implmsz$hypot_float_impl..hypot_implcs ||Sr>rr)rrrrrs)r1r5r!rr sysplatformrZExternalFunctionrZ MACHINE_BITSrcompile_internalr) r6rr7r1xtyytyrrr}rr9rrrr\s"    rcCs6|\}||jtjd}|||}t|||j|SNZ get_constantr5rpiZfmulrr6rr7r1rZcoefr9rrrradians_float_impl{s rcCs6|\}||jdtj}|||}t|||j|Srrrrrrdegrees_float_impls rcCs|tj|}|||Sr>)Z get_functionoperatorpow)r6rr7r1rCrrrpow_implsrcCsdS)z8Convert integer to unsigned integer of equivalent width.NrTrrr _unsignedsrcs>|tjkrddS|tjkr:ttd|jfddSdS)NcSs|Sr>rrrrrz _unsigned_impl..zuint{}cs|Sr>rrZnewTrrrr)rZunsigned_domainZ signed_domaingetattrformatZbitwidthrrrr_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 minrnpZ right_shiftZ left_shift) r,brZzaZzbkuvrrrrgcds"  zgcd_impl..gcd)r1r5r!rr) r6rr7r1rrrrrr9rrrgcd_impls  r)F)__doc__rrrZnumpyrZ llvmlite.irrrZnumba.core.imputilsrrZnumbarZ numba.corerrrr Znumba.core.extendingr Znumba.core.typingr Znumba.cpython.unsafe.numbersr registryr?ZfinfoZdtypeZ _NP_FLT_FINFOmaxZFLT_MAXZtinyZFLT_MINZ _NP_DBL_FINFOZDBL_MAXZDBL_MINZFLOAT_ABS_MASKZFLOAT_SIGN_MASKZDOUBLE_ABS_MASKZDOUBLE_SIGN_MASKrrrr$r%r(r)r*r3r=rDrIrUfabsruZexp_impllogZlog_impllog10Z log10_implsinZsin_implcosZcos_implrVZ log1p_implrWZ expm1_implrXZerf_implrYZ erfc_implrZZtan_implr[Z asin_implr\Z acos_implr]Z atan_implr^Z asinh_implr_Z acosh_implr`Z atanh_implraZ sinh_implrbZ cosh_implrcZ tanh_implrdZ log2_implreZ ceil_implrfZ floor_implgammaZ gamma_implrgZ sqrt_implrhZ trunc_implriZ lgamma_implisnanrHrkr@rmrrnroisfiniterprqcopysignrsfrexpr~ldexpZintcrrrTrZuint64rrrrrrradiansrdegreesrrrrrrrrrrrs        $