gd<^ dZddlZddlZddlZddlZddlZddlmZddl m Z m Z ddl m Z ddlmZmZmZmZddlmZddlmZdd lmZe d ZejZejejd ZejZej Z!ejejd Z"e"jZ#e"j Z$d Z%dZ&dZ'dZ(dZ)dZ*dZ+dZ,dZ-dZ.dZ/dZ0dZ1dZ2dZ3dZ4dedZ5e4ej6de4ej7d Z8e4ej9d!Z:e4ej;d"Z<e4ej=d#Z>e4ej?d$Z@e5ejAd%d&ZBe5ejCd'd(ZDe5ejEd)d*ZFe5ejGd+d,ZHe5ejId-d.ZJe5ejKd/d0ZLe5ejMd1d2ZNe5ejOd3d4ZPe5ejQd5d6ZRe5ejSd7d8ZTe5ejUd9d:ZVe5ejWd;d<ZXe5ejYd=d>ZZe5ej[d?d@Z\e5ej]dAdBZ^e5ej_dCdDdEZ`e5ejadFdGdEZbe5ejcdHdIZde5ejedJdKZfe5ejgdLdMdEZhe5ejidNdOZjeejkejldPZmeejkejndQZoeejpejldRZqeejpejndSZreejsejldTZteejsejndUZueejvejlejldVZweejxejldWZyeejzejlej{dXZ|eej}ej~ej~dYZeej}ejejdZZeej}ejlejld[Zeejej~ej~d\Zeejejejd]Zeejejlejld^Zeejejld_Ze3ejeeejejld`Ze3ejeeejejlejleejejlejndaZdbZeedcZddZeejejnejnedS)fzA Provide math calls that uses intrinsics or libc math functions. N)Constant)Registryimpl_ret_untracked)typeof)typesutilsconfigcgutils)overload) signature)trailing_zerosmathimplfloat32float64illlc0|d||S)z< Return a condition testing whether *val* is a NaN. uno)fcmp_unorderedbuildervals L/granary/edev/quantum/lib/python3.11/site-packages/numba/cpython/mathimpl.pyis_nanr(s  ! !%c 2 22ct|jtd}t|jtd}|d||}|d||}|||S)zB Return a condition testing whether *val* is an infinite. z+infz-infz==)rtypefloat fcmp_orderedor_)rrpos_infneg_infisposinfisneginfs ris_infr#.stsxv//Gsxv//G##D#w77H##D#w77H ;;x * **rc\|||}|d||S)z? Return a condition testing whether *val* is a finite. ord)fsubr)rr val_minus_vals r is_finiter(8s/ LLc**M   }m D DDrc|jtjksJ||tjdS)z1 Bitcast a double into a 64-bit integer. @)rllvmliteir DoubleTypebitcastIntTypers r f64_as_int64r0@sI 8x{--// / / / / ??3 3 3B 7 7 8 88rc|jtjdksJ||tjS)z1 Bitcast a 64-bit integer into a double. r*)rr+r,r/r.r-rs r int64_as_f64r2GsI 8x{**2.. . . . . ??3 6 6 8 8 9 99rc|jtjksJ||tjdS)z0 Bitcast a float into a 32-bit integer. )rr+r, FloatTyper.r/rs r f32_as_int32r6NsI 8x{,,.. . . . . ??3 3 3B 7 7 8 88rc|jtjdksJ||tjS)z0 Bitcast a 32-bit integer into a float. r4)rr+r,r/r.r5rs r int32_as_f32r8UsI 8x{**2.. . . . . ??3 5 5 7 7 8 88rcT|t|jd|S)zB Negate real number *val*, with proper handling of zeros. g)r&rrrs r negate_realr:\s$ <<400# 6 66rc||j}||d|D}|||S)z9 Call a LLVM intrinsic floating-point operation. cg|] }|j S)r).0as r z%call_fp_intrinsic..hs'='='=1'='='=r)moduledeclare_intrinsiccall)rnameargsmodintrs rcall_fp_intrinsicrHcsC .C  '='='='='= > >D <<d # ##rcfd}|S)z Return an implementation factory to convert the single integral input argument to a float64, then defer to the *wrapped_impl*. c|\}|jd}||||tj}t tjtj} ||||f}|||tj|jS)Nr)rEcastrrr return_type) contextrsigrEr input_typefpval inner_sigres wrapped_impls r implementerz2_unary_int_input_wrapper_impl..implementerqstXa[  Wc:u}EEemU];; l7GYAA||GS%-IIIrr=)rSrTs` r_unary_int_input_wrapper_implrUls) JJJJJ rcjt|}t|tj|dSN)rUlowerrInteger)fn float_implimpls runary_math_int_implr]{s1 ( 4 4DE"emT"""""rcrt|tjfd}t|||S)zO Implement the math function *fn* using the LLVM intrinsic *intrcode*. cTt||}t|||j|SrW)rHrrL)rMrrNrErRintrcodes rr[z#unary_math_intr..float_impls*488!'7COSIIIr)rXrFloatr])rZr`r[s ` runary_math_intrrbsR 2u{JJJJJJ''' rFc|r tjnd}fd}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|\}|j}|jd}||}tj tj i|}t j||g} tj |j| |} | | |f} | || ||j } t|||j | S)z9 Implement *fn* for a types.Float input. rrD)rArEget_value_typerrrr+r, FunctionTyper insert_pure_functionrCrKrLr)rMrrNrErrFrOlty func_namefntyrZrR f32extern f64externs rr[z%unary_math_extern..float_implsnXa[ $$Z00 M9 M9 {''cU33  )'.$Y O O Oll2v&&ll7CS_EE!'7COSIIIr)rint64rXrar])rZrlrm int_restype f_restyper[s `` runary_math_externrqsp +4 IJJJJJJ$E"ek:&&&J''' rz llvm.fabszllvm.expzllvm.logz llvm.log10zllvm.sinzllvm.coslog1pflog1pexpm1fexpm1erfferferfcferfctanftanasinfasinacosfacosatanfatanasinhfasinhacoshfacoshatanhfatanhsinhfsinhcoshfcoshtanhftanhlog2flog2ceilfceilTfloorffloor numba_gammaf numba_gammasqrtfsqrttruncftrunclgammaflgammacX|\}t||}t|||j|SrW)rrrLrMrrNrErrRs risnan_float_implr. ES #  C gw E EErcHtj}t|||j|SrWr false_bitrrLrMrrNrErRs risnan_int_implr  C gw E EErcX|\}t||}t|||j|SrW)r#rrLrs risinf_float_implrrrcHtj}t|||j|SrWrrs risinf_int_implrrrcX|\}t||}t|||j|SrW)r(rrLrs risfinite_float_implrs. ES GS ! !C gw E EErcHtj}t|||j|SrW)r true_bitrrLrs risfinite_int_implrs  C gw E EErc|dj}|j}tj|tj|||fd|jz}|||}t|||j |S)Nrzllvm.copysign.%s) rrAr get_or_insert_functionr+r,rgintrinsic_namerCrrL)rMrrNrErirFrZrRs rcopysign_float_implrsw q',C .C  'X[-E-EcCQT:-V-V(:S=O(O Q QB ,,r4 C gw E EErcN|\}||jd}||jd}tj||d}t j||t j|f}dddt|} tj |j || } | | ||f} tj || ||f} t|||j| S)Nrexpre numba_frexpf numba_frexprdouble) get_data_typerErLr alloca_oncer+r,rg PointerTypestrrrArCmake_anonymous_structloadr) rMrrNrErflttyinttyexpptrrkfnamerZrRs r frexp_implrs  DC  ! !#(1+ . .E  ! !#/!"4 5 5E  %e < < : :D   e** E  %gnd G G GB ,,rC: & &C gw E EErcZ|\}}||tj}||tj}t t jt jt j}t|||||fSrW)sitofpr+r,r-r rratan2_float_implrMrrNrEyxfsigs ratan2_s64_implr%| FQq(+002233Aq(+002233A U]EM5= A AD GWdQF ; ;;rcZ|\}}||tj}||tj}t t jt jt j}t|||||fSrW)uitofpr+r,r-r rrrrs ratan2_u64_implr-rrct|dksJ|j}|jd}||}tjdtjdi|}tj |||f}tj |j||} | | |} t|||j| S)Nratan2fatan2re)lenrArErfrrrr+r,rgr rhrCrrL) rMrrNrErFtyrirjrkrZrRs rrr5s t99>>>> .C !B   $ $C x w  I ; # #C#s 4 4D  %gnd K K KB ,,r4 C gw E EErc|\}}||tj}||tj}t t jt jt j}t|||||f}t|||j |SrW rr+r,r-r rrhypot_float_implrrLrMrrNrErrrrRs rhypot_s64_implrH FQq(+002233Aq(+002233A U]EM5= A AD 7GTAq6 : :C gw E EErc|\}}||tj}||tj}t t jt jt j}t|||||f}t|||j |SrWrrs rhypot_u64_implrRrrc |j\}}||cxkr |jksnJ|\}}tjtjdkrdndtjtjdkrdndi|}tj|| tjdkr/tj dkr|td fd} n fd } | || ||} t|||j| S) Nwin32_hypotfhypotf_hypothypotr4infcptj|stj|rS||SrW)mathisinf)rrr plat_hypots r hypot_implz$hypot_float_impl..hypot_implms9z!}}  1   :a## #rc||SrWr=)rrrs rrz$hypot_float_impl..hypot_implrs:a## #r) rErLrrsysplatformrExternalFunctionr MACHINE_BITSrcompile_internalr) rMrrNrExtyytyrrrrrRrrs @@rrr\s0xHC # ( ( ( ( ( ( ( ( ( ( DAq  CLG$;$;yy 3<7#:#:xx   E's33J |w6#6"#<#<c%,, $ $ $ $ $ $ $  $ $ $ $ $  " "7JT B BC gw E EErc|\}||jtjdz }|||}t |||j|SN get_constantrLrpifmulrrMrrNrErcoefrRs rradians_float_implr{sO CQ   3 ? ?D ,,q$  C gw E EErc|\}||jdtjz }|||}t |||j|Srrrs rdegrees_float_implrsO CQ   tw ? ?D ,,q$  C gw E EErcZ|tj|}|||SrW) get_functionoperatorpow)rMrrNrEr\s rpow_implrs-    c 2 2D 4  rcdS)z8Convert integer to unsigned integer of equivalent width.Nr=Ts r _unsignedr sDrc|tjvrdS|tjvr2ttd|jfdSdS)Nc|SrWr=r s rz _unsigned_impl..srzuint{}c|SrWr=r newTs rrz _unsigned_impl..sar)runsigned_domain signed_domaingetattrformatbitwidthrs @r_unsigned_implrs[E !!!{ e! ! !uhooaj99::      " !rc|j\}}||cxkr |jksnJ|\}}d}|||||} t|||j| S)NcFt|}|dkrt|S|dkrt|St|}t|}t||}t tt j||}t tt j||}||kr7||kr||}}||z}t j|t|}||k7t j|||}|S)zO Stein's algorithm, heavily cribbed from Julia implementation. r)rabsr minr np right_shift left_shift) r?br zazbkuvrs rgcdzgcd_impl..gcds GG 66#a&&= 66#a&&= A   A   BKK c".B//00 1 1 c".B//00 1 11ff1uu!1 FAq."3"344A 1ff M!!A$$ " "r)rErLrr) rMrrNrErrrrr'rRs rgcd_implr(s}xHC # ( ( ( ( ( ( ( ( ( ( DAq*  " "7Cd ; ;C gw E EEr)F)__doc__rrrnumpyr llvmlite.irr+rnumba.core.imputilsrrnumbar numba.corerrr r numba.core.extendingr numba.core.typingr numba.cpython.unsafe.numbersr registryrXfinfodtype _NP_FLT_FINFOmaxFLT_MAXtinyFLT_MIN _NP_DBL_FINFODBL_MAXDBL_MINFLOAT_ABS_MASKFLOAT_SIGN_MASKDOUBLE_ABS_MASKDOUBLE_SIGN_MASKrr#r(r0r2r6r8r:rHrUr]rbrqfabsrexp_implloglog_impllog10 log10_implsinsin_implcoscos_implrs log1p_implru expm1_implrwerf_implry erfc_implr{tan_implr} asin_implr acos_implr atan_implr asinh_implr acosh_implr atanh_implr sinh_implr cosh_implr tanh_implr log2_implr ceil_implr floor_implgamma gamma_implr sqrt_implr trunc_implr lgamma_implisnanrarrYrrrrisfiniterrcopysignrfrexprldexpintcrrrnruint64rrrrrrradiansrdegreesrrrr rr(r'r=rrrjs   <<<<<<<<444444444444))))))''''''777777 8J  ),,--    ),,--    $%333 +++EEE999:::999999777$$$   ###   !!!!H ;''' ?48Z 0 0 ?48Z 0 0 _TZ 6 6 ?48Z 0 0 ?48Z 0 0  tz8W = =  tz8W = =  TXvu 5 5  di& 9 9  TXvu 5 5  di& 9 9  di& 9 9  di& 9 9  tz8W = =  tz8W = =  tz8W = =  di& 9 9  di& 9 9  di& 9 9  di& 9 9  di&$ ? ?  tz8Wd C C  tz>= I I  di& 9 9  tz8Wd C C  YAA tz5;FF F tz5=!!FF"!F tz5;FF F tz5=!!FF"!F t}ek""FF#"F t}em$$FF%$F t}ek5;//FF0/Ftz5; F F  F tz5; ++ F F,+ F tz5; ,,<<-,<tz5<..<</.<tz5; ,, F F-, F$tz5; ,,FF-,Ftz5<..FF/.Ftz5; ,,FF-,F<t|U[!!FF"!F DL"4555t|U[!!FF"!F DL"4555txek**txem,,-,+*    )!!!FFF<.dh u}--h77777r