:j-(dZddlZddlZddlZddlmZddlm Z ddl m Z m Z m Z mZmZmZddlmZddlmZddlmZmZmZdd lmZd Zd Zd Zdd Ze j<ZefdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'e&Z(dZ)dZ*dZ+dZ,dZ-dZ.dZ/dZ.e/e.eej`ejbdZ2dZ3d Z4ee4d!"d#Z5e/e3d$e4ejld%Z7d&Z8d'Z9d(Z:d)Z;d*Zd-Z?d.Z@d/ZAd0ZBd1ZCd2ZDd3ZEd4ZFd5ZGd6ZHd7ZId8ZJd9ZKd:ZLd;ZMd<ZNd=ZOd>ZPd?ZQd@ZRdAZSdBZTdCZUdDZVdEZWdFZXdGZYdHZZdIZ[dJZ\dKZ]dLZ^dMZ_dNZ`dOZadPZbdQZcdRZddSZedTZfdUZgdVZhdWZidXZjdYZkdZZld[Zmd\Znd]Zod^Zpd_Zqd`ZrdaZsdbZtdcZuddZvdeZwdfZxdgZydhZzdiZ{djZ|dkZ}dlZ~dmZdnZdoZdpZdqZdrZdsZdtZduZdvZdwZdxZdyZdzZd{Zd|Zd}Zd~ZdZdZdZdZdZdZdZdZdZdZdZdZdZy)zCodegen for functions used as kernels in NumPy functions Typically, the kernels of several ufuncs that can't map directly to Python builtins Noverload)impl_ret_untracked)typingtypeserrorsloweringcgutilsconfig)register_jitable) npdatetime) cmathimplmathimplnumbers) numpy_versiong+eG?g&{?g9B.?cpt||k(sJt|j|k(sJ|jd|}tfd|jDr|j|k(sKddl}|j j jj}dj||}J|y)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. rNc3(K|] }|k( ywN).0argtys f/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/numba/np/npyfuncs.py z/_check_arity_and_homogeneity..*s,CR,sz"{0} called with invalid types: {1}) lenargsall return_typeinspect currentframef_backf_codeco_nameformat)sigrarityrrfnamemsgrs @r_check_arity_and_homogeneityr)s t9   sxx=E !! ! !B ,388, ,K1O$$&--44<<299%Ecu 2Pc |j}|j|}tjj ||gt |j z}tj|||} t||j D cgc]\} } |j|| | |} } } |j| | } |j|| tj|jScc} } w)Nname)moduleget_argument_typellvmliteir FunctionTyperrr insert_pure_functionzipcastcallrfloat64r)contextbuilderr%r func_namermodltyfntyfnrargty cast_argsresults r_call_func_by_name_with_castrB2s ..C  # #B 'C ;; # #C#s388})< =D  % %c4i @B#&tSXX#69S%gsE269I9\\"i (F << HH 9sC%cR|jd} ||}|j } |tjvr |j||} |D cgc]} tj|| } } | jg| z}|gt|jz}|Dcgc]!}|j|j#}}t j"j%t j"j'|}tj(| ||}|j+|||j-|d}|S|jDcgc]}|j/|}}|j/|j0}t j"j%||}tj2| ||}|j5|||j|}|S#t$r5}dj|t|} t j | d}~wwxYwcc} wcc}wcc}w)Nrz!No {0} function for real type {1}r,)rKeyErrorr$strr LoweringErrorr.rcomplex_domain make_complexr alloca_once_value _getpointerlistget_value_type as_pointerr0r1r2VoidTypeget_or_insert_functionr6loadr/rr3call_external_function)r8r9r%rtable user_namerr:er(r;outrptrargs call_args call_argtys call_argltysr=r>retvalatyargtypesrestypes r_dispatch_func_by_name_typer^Bs !B("I ..C U ! !!""7B/"$,,Wc:$$__&''1 dT#((^+ "-/ ..r2==?/ /{{'' (<(<(> M  + +Cy A R#il+ M ?BhhGsG--c2GG++COO<{{'':  ) )#t) D//SXXtL ME (188CFK""3''($/Hs)G H&H H$ H"0HHcPt||d|\}}|jd}|j|d}|j|d}|j|d|jjdz z} |j d||} |j d||} |j d| |} |j | | } |j| | }|j|d5\}}|5|j}ddd|5|j}|j||}|j||}|jd||}|jd||}|j||}|j d ||}|j ||}|j|||}|j||}dddddd|j!|j}|j#||j#|S#1swYxYw#1swY_xYw#1swYcxYw) Nr==Flikely>!=)r)r get_constanttypewidth icmp_unsignedand_or_if_else basic_blocksdivsrem icmp_signedxorselectaddphi add_incoming)r8r9r%rnumdenrZERO MINUS_ONEMIN_INT den_is_zeroden_is_minus_onenum_is_min_intcould_cause_sigfpe force_zerothen otherwisebb_then bb_otherwisedivr; num_gt_zero den_gt_zero not_same_sign mod_not_zero needs_fixing fix_valueresult_otherwiserAs rnp_int_sdiv_implrs dA.HC !B   A &D$$R,I""2qSXX^^A-='>?G''dC8K,,T9cB**4#>N &6G[*<=J E 2 ;6GtY  *))G *  ;"..L,,sC(C,,sC(C!--c3=K!--c3=K#KK [AM"00sDAL"<< |DL|YEI&{{3 :  ; ;[[ #F g& (,7 M% * * ; ; ; ;s="H( H5 H?B3H2HH HH HH%ct||d|\}}|jd}|j|d}|jd||}|j} t j ||5|j} |j||} |jd||} |jd||} |j| | }|jd| |}|j||}|j|||}|j|| }ddd|j|j}|j|| |j |S#1swYJxYw)Nr`rrgrf)r)rrhrkror if_unlikelyrqrrrsrlrtrurvrirw)r8r9r%rrxryrrz den_not_zerobb_no_ifbb_ifr;rrrrrr final_modrAs rnp_int_srem_implrsL dA.HC !B   A &D((tS9L""H   Wl 3 0##ll3s#))#sD9 ))#sD9  K= ,,T3= ||M<@ NN<d; KK 3/  0[[ #F h'  5) M 0 0s )B!EEct|||jd|j|}t|||jd|j|}|j ||j||gSNrrb)rrrr make_tupler8r9r%rrrems rnp_int_sdivrem_implri 7G-?S__Q-?-JD QC 7G-?S__Q-?-JD QC   gsc CCr*ct||d|\}}|jd}|j|d}|jd||}|j |d5\} } | 5|j } ddd| 5|j ||} |j } dddddd|j|j}|j| |j  |S#1swY{xYw#1swY^xYw#1swYbxYw)Nr`rrcFrd) r)rrhrkrnroudivrvrirw)r8r9r%rrxryrrz div_by_zerorrrrrrAs rnp_int_udiv_implrs dA.HC !B   A &D''dC8K U 3/7Hi  *))G * /,,sC(C"..L / /[[ #F g& \* M * * / / //s<C? C'- C?7C3C?'C0 ,C?3C< 8C??Dct||d|\}}|jd}|j|d}|jd||}|j} t j ||5|j} |j||} ddd|j|j} | j|| | j  | S#1swYJxYw)Nr`rrg) r)rrhrkror ruremrvrirw) r8r9r%rrxryrrzrrrr;rAs rnp_int_urem_implrs dA.HC !B   A &D((tS9L""H   Wl 3$##ll3s#$[[ #F h' U# M$$s )CCct|||jd|j|}t|||jd|j|}|j ||j||gSr)rrrrrrs rnp_int_udivrem_implrrr*c:t||d|j|SNr`)r)fdivr8r9r%rs rnp_real_div_implrs!!dA. 7<< r*ct||d|\}}|jd}|j|d}|j||}|j d||} |j d||} |j d||} |j | |j | | } |j| ||} |j|| S)Nr`rrg<) r)rrhfrem fcmp_orderedrlrsrtfadd)r8r9r%rin1in2rrzres res_ne_zero den_lt_zero res_lt_zerorrs rnp_real_mod_implrs!dA.HC !B   C (D ,,sC C&&tS$7K&&sC6K&&sC6K<< ' K EGL|S$7I <<Y ''r*c:t||d|j|Sr)r)rrs rnp_real_fmod_implrs dA. 7<< r*ctjj|jd}|j ||}|j d||}|j |||S)Nrr)r0r1Constantrifsubrrt)r8r9rrz arg_negatedarg_is_negatives r_fabsrsS ;;  # .D,,tS)K**3T:O >>/; <=rc)rHrrealimagrir make_helperrr0r1rrrrnrlrfmulrr _getvalue)"r8r9r%rrrrin1rin1iin2rin2iftypeirUrzONEin2r_absin2i_absin2r_abs_ge_in2i_absrr in2r_is_zero in2i_is_zero in2_is_zeroinn_then inn_otherwiserattmp1tmp2scltmp3tmp4tmp5tmp6s" rnp_complex_div_implr$s !$$Wchhqk$E!HC 88D 88D 88D 88D IIE dD$'?@! @ AUCUU A   gs 7C ;;  s +D ++  uc *CWgt,HWgt,H"//hI - .+/2C4  7"//hEL"//hEL!,,|\BK- 71J(M< '||D(;CH&||D(;CH < #7",,tT2C"<<c2D"<<d3D!,,sD1C#<<c2D"<<c2D"<<d3D"<<d3D&||D#6CH&||D#6CH7  7  76 /,,tT*C<<c*D<<d+D,,sD)C<<c*D<<c*D<<d+D<<d+D||D#.CH||D#.CH /9+/Z ==?!A"<< 77  7 7  7 76 / /9+/Z ==?s(N<N:O A N2 N%/N ? N% B?N N%N2 O "B?N?!O  N N%N" N%%N/ *N22N< 7O ?O O  O#cyrrx1x2s r_npy_logaddexprqr*c8t|dfd}y)Ngenerictargetc4||k7ry|fd}|S)Nc||}}||k(r|zS||z }|dkDr|| zS|dkr||zS|S)Nrr)rrxytmpexpfnlog1pfnshifts rimplz;_generate_logaddexp..ol_npy_logaddexp..impl}serqAQ5y !e!GwucT{333QhwuSz222Jr*r)rrrrconstrrs @rol_npy_logaddexpz-_generate_logaddexp..ol_npy_logaddexpxs" 8 5   r*r) fnoverloadrrrrs ``` r_generate_logaddexprtsj+,r*cyrrrs rrrrr*ct||d|jjt}|j |jg|j i}|j ||}|||Sr)r)typing_contextresolve_value_typer get_call_typer get_functionr8r9r%rr=rs rnp_real_logaddexp_implrse dA.  ! ! 4 4^ DD   W33\sxx\2 FC   c *D  r*cyrrrs r_npy_logaddexp2rrr*cyrrrs r npy_log2_1prrr*rrc*|tfd}|S)Nc4tj|zSr)nplog1p)rLOG2Es rrzol_npy_log2_1p..implsrxx{""r*) _NPY_LOG2E)rrr s @rol_npy_log2_1pr s jME# Kr*rct||d|jjt}|j |jg|j i}|j ||}|||Sr)r)rrrrrrrs rnp_real_logaddexp2_implrse dA.  ! ! 4 4_ ED   W33\sxx\2 FC   c *D  r*c&|\}}|jtfd|DsJd|j\}}|j|||tj }|j|||tj }|j ||S)Nc3<K|]}|jk(ywr)ri)rrlltypes rrz&np_int_truediv_impl..s,!qvvv~,szmust have homogeneous types)rirrr5rr7r) r8r9r%rrxrynumtydentyrs @rnp_int_truediv_implrsHC XXF ,t, ,K.KK ,88LE5 ,,wUEMM :C ,,wUEMM :C <<C  r*ct||||}tj|j|j}t ||||fSr)rr signaturernp_real_floor_impl)r8r9r%rrss rnp_real_floor_div_implrs? 7GS$ 7C#//:A gwC6 ::r*ct|||jd|j|}t|||jd|j|}|j ||j||gSr)rrrrrrs rnp_real_divmod_implrsj '3E3??13Esxx3PRV WC 7G-?S__Q-?-JD QC   gsc CCr*c~|jdj}tj||}|Dcgc]#}|j ||jd|%c}\}}|j } |j } |j } |j } | j} t| | | | fDcgc]}|j| k(c}sJdtjj| d}|j||j}||_t||| }t||| }|jd||}|j!|5\}}|5|j#| | }|j%| |}|j%| |}|j'| |}|j'| |}|j#||}t)||||f|_ddd|5|j#| | }|j%| |}|j%| |}|j'| |}|j'| |}|j#||}t)||||f|_dddddd|j+Scc}wcc}w#1swYxYw#1swY7xYw#1swY|j+SxYw)Nrrrrr)runderlying_floatrrrHrrrirr0r1rrrrrrnrrrrr)r8r9r%r float_kind floor_sigrrrrrrrrrrzrUrrrrrrrrrrrs rnp_complex_floor_div_implr!s!--J  Z8I !$$Wchhqk$E!HC 88D 88D 88D 88D IIE dD$'?@! @ AUCUU A ;;  s +D   gs 7CCHWgt,HWgt,H"//hI - .P2C4  P,,tT*C<<c*D<<c*D<<d+D<<d+D<<d+D)'7IwOCH P P,,tT*C<<c*D<<c*D<<d+D<<d+D<<d+D)'7IwOCH PP& ==?M!A P P P PP& ==?sI(J+JJ$BJ  J$!BJ"J$ J J$J! J$$J<cLt||dtj||||Srr)rcomplex_power_implrs rnp_complex_power_implr%% dA.  % %gwT BBr*cLt||dtj||||Sr)r)rreal_power_implrs rreal_float_power_implr)s% dA.  " "7GS$ ??r*cLt||dtj||||Srr#rs rnp_complex_float_power_implr+!r&r*cLt||dtj||||Sr)r)rgcd_implrs r np_gcd_implr.*% dA.   WgsD 99r*c|j\}}||cxk(r|jk(sJJ|\}}d}|j||||} t|||j| S)NcZ|dk(rdSt||tj||zzS)z7 Like gcd, heavily cribbed from Julia. r)absr gcd)abs rlcmznp_lcm_impl..lcm8s.Fq<AbffQl):$; <.complex_signLs-**QVVQVV,Cax3wr*rrrnanr`)r)rr7rrrrHrrhfloatrrrrrbooleanrnp_complex_ge_implnp_complex_eq_implnp_complex_lt_implrt)r8r9r%rr@roprrAfloat_tyrzrr{NANcmp_sigcmp_args arg1_ge_arg2 arg1_eq_arg2 arg1_lt_arg2 real_when_ge real_when_nges rnp_complex_sign_implrREs!dA. &&w c4H!'7COOSII !W XXa[%%gr2&&##Hc2##Hc2((48 ""8U5\:  ""5== |YD nn\<O !!r*cJt||dtj|d|S)Nrbz llvm.rintr)rcall_fp_intrinsicrs rnp_real_rint_implrVr# dA.  % %g{D AAr*cht||d|jd}|j}|j|||d}|j||}t j |gdz}t ||||jg|_t ||||jg|_|jS)Nrbrrr`) r)rrrHrrrVrrr) r8r9r%rrrIrrU inner_sigs rnp_complex_rint_implrZxs!dA. !B""H   w$q'  :C   w +C  8*Q,/I '9sxxjICH '9sxxjICH ==?r*cLt||dtj||||SNrb)r)rexp_implrs rnp_real_exp_implr^r/r*cLt||dtj||||Sr\)r)rr]rs rnp_complex_exp_implr`% dA.   gwT ::r*ct||d|dj}tjj ||g}t j |j|d}|j||dgS)Nrbrz llvm.exp2r, r)rir0r1r2r r3r.r6r8r9r%rll_tyr=r>s rnp_real_exp2_implrfe dA. GLLE ;; # #EE8 4D  % %gnnd+6 8B <<T!WI &&r*ct||d|jd}|j}|j|||d}|j||}|j |t }|j ||j|_|j ||j|_t||||jgSNrbrr) r)rrrHrh _NPY_LOGE2rrrr`r) r8r9r%rrrIrrloge2s rnp_complex_exp2_implrls dA. !B""H   w$q'  :C   w +C  : 6E||E388,CH||E388,CH ws}}6G HHr*cLt||dtj||||Sr\)r)rlog_implrs rnp_real_log_implror/r*cLt||dtj||||Sr\)r)rrnrs rnp_complex_log_implrqrar*ct||d|dj}tjj ||g}t j |j|d}|j||dgS)Nrbrz llvm.log2r,rcrds rnp_real_log2_implrsrgr*cft||d|jd}|j}t||||}|j |||}|j |t }|j||j|_|j||j|_ |jSri) r)rrrqrHrhr rrrr)r8r9r%rrrIrlog2es rnp_complex_log2_implrvs dA. !B""H gwT :C   w#  6C  : 6E||E388,CH||E388,CH ==?r*cLt||dtj||||Sr\)r)r log10_implrs rnp_real_log10_implry% dA.   wd ;;r*cft||d|jd}|j}t||||}|j |||}|j |t }|j||j|_|j||j|_ |jSri) r)rrrqrHrh _NPY_LOG10Errrr)r8r9r%rrrIrlog10es rnp_complex_log10_implr~s dA. !B""H gwT :C   w#  6C  ! !(K 8F||FCHH-CH||FCHH-CH ==?r*cLt||dtj||||Sr\)r)r expm1_implrs rnp_real_expm1_implrrzr*c*t||d|jd}|j}tj|gdz}|j |d}|j |||d}t||||jg} |j ||} t||||jg} t||||jg} |j| | } |j| | | _ |j| || _| jS)Nrbrr`rAr)r)rrrrrhrHr^rnp_real_cos_implrnp_real_sin_implrrr)r8r9r%rrrIfloat_unary_sigr{rr4rUcos_imagsin_imagrs rnp_complex_expm1_implrs dA. !B""H&& 1 5O$$Xt4I   w$q'  :C'?SXXJGA   w +C/CHH:NH/CHH:NH ,,q( #C||Ax(CH||C+CH ==?r*cLt||dtj||||Sr\)r)r log1p_implrs rnp_real_log1p_implrrzr*ct||d|jd}|j}tj|gdz}tj|gdz}|j |d}|j |||d} |j ||} |j| j|} t|||| | jg} t|||| j| g| _ t|||| g| _| jS)Nrbrr`rr)r)rrrrrhrHrrnp_real_hypot_implrnp_real_atan2_implror) r8r9r%rrrIrfloat_binary_sigrrrU real_plus_onels rnp_complex_log1p_implr s dA. !B""H&& 1 5O''(A6   x -C   w$q'  :C   w +CLL3/M7G-=)3884 6A!'74D#&88]";=CH/A3GCH ==?r*cLt||dtj||||Sr\)r)r sqrt_implrs rnp_real_sqrt_implr#% dA.   gwT ::r*cLt||dtj||||Sr\)r)rrrs rnp_complex_sqrt_implr(s% dA.   wd ;;r*cLt||d|j|d|dSNrbr)r)mulrs rnp_int_square_implr0s' dA. ;;tAwQ ((r*cLt||d|j|d|dSr)r)rrs rnp_real_square_implr5s' dA. <<Qa ))r*ct||dtj|jgdz}t j ||||d|dgSNrbrr)r)rrrrcomplex_mul_impl)r8r9r%r binary_sigs rnp_complex_square_implr9sS dA.!!COO#4Q#67J  # #GWj&*1gtAw%7 99r*ctt||dtddfd}|j||||S)NrbT)fastmathch|dkrtj| d Stj|dS)NrgUUUUUU?)r powerrs rcbrtznp_real_cbrt_impl..cbrtHs0 q5HHaR++ +88Ay) )r*c^tj|rtjS|Sr)r isnanrB)rrs r_cbrtz np_real_cbrt_impl.._cbrtOs 88A;66MAwr*)r)r r7)r8r9r%rrrs @rnp_real_cbrt_implrCsF dA.t$*%*   # #GUC >>r*cRt||d|j}tj|gdz}|j ||d|t j }|jt j d}|j||}|j ||t j |Sr) r)rrrr5rr7rhr) r8r9r%rrr in_as_floatrresult_as_floats rnp_int_reciprocal_implrZs!dA. B!!B46*J,,wQU]]CK   u}}a 0Cll3 4O <<%-- DDr*c~t||d|j|jd}|j||dS)Nrbrr)r)rhrr)r8r9r%rrs rnp_real_reciprocal_implrhs9 dA.   s 4C <<T!W %%r*c>t||d|jd}|j}|j|d}|j|d}|j |||d}|j ||} |j } |j } t||| } t||| } |jd| | }|j|5\}}|5|j| | }|j| |}|j| |}|j||}|j||}|| _|j||| _ddd|5|j| | }|j| |}|j|| }|j||}|j||| _|j||| _dddddd| jS#1swYxYw#1swY-xYw#1swY| jSxYw)Nrbrrrr<=)r)rrrhrHrrrrrnrrrrr)r8r9r%rrrIrzrrrUrrin1r_absin1i_absin1i_abs_le_in1r_absrrrtmp0dinv_dminus_rs rnp_complex_reciprocal_implrns!dA. !B""H   # .D   x -C   w$q'  :C   w +C 88D 88DWgt,HWgt,H"//hI - .12C4  4 T4(A<<a(D T4(ALLa(Ell4+GCH||GU3CH 4 1 T4(A<<a(D T4(ALLa(E||Au-CH||D%0CH  11" ==?! 4 4 1 11" ==?s> HA9G, HA7G8 H,G5 1H8H =HHcLt||dtj||||Sr\)r)rsin_implrs rrrr/r*cLt||dtj||||Sr\)r)rrrs rnp_complex_sin_implrrar*cLt||dtj||||Sr\)r)rcos_implrs rrrr/r*cLt||dtj||||Sr\)r)rrrs rnp_complex_cos_implrrar*cLt||dtj||||Sr\)r)rtan_implrs rnp_real_tan_implrr/r*cLt||dtj||||Sr\)r)r asin_implrs rnp_real_asin_implrrr*cLt||dtj||||Sr\)r)r acos_implrs rnp_real_acos_implrrr*cLt||dtj||||Sr\)r)r atan_implrs rnp_real_atan_implrrr*cLt||dtj||||Sr)r)ratan2_float_implrs rrr% dA.  $ $WgsD AAr*cLt||dtj||||Sr)r)rhypot_float_implrs rrrrr*cLt||dtj||||Sr\)r)r sinh_implrs rnp_real_sinh_implrrr*ct||d|jd}|j}tj|gdz}|j |||d}|j ||}|j } |j} t|||| g} t|||| g} t|||| g} t|||| g}|j| | |_|j| ||_|jSNrbrr`)r)rrrrrHrrrrrnp_real_cosh_implrr)r8r9r%rrftyfsig1rrUxrxisxishxrcxichxrs rnp_complex_sinh_implrs!dA. !B  C   seAg &EWb$q'2A   w +C B B 7GURD 9C Wgurd ;D 7GURD 9C Wgurd ;D||C&CH||C&CH ==?r*cLt||dtj||||Sr\)r)r cosh_implrs rrrrr*ct||d|jd}|j}tj|gdz}|j |||d}|j ||}|j } |j} t|||| g} t|||| g} t|||| g} t|||| g}|j| | |_|j| ||_|jSr)r)rrrrrHrrrrrrrr)r8r9r%rrrrrrUrrrrrrs rnp_complex_cosh_implrs!dA. !B  C   seAg &EWb$q'2A   w +C B B 7GURD 9C Wgurd ;D 7GURD 9C Wgurd ;D||C&CH||C&CH ==?r*cLt||dtj||||Sr\)r)r tanh_implrs rnp_real_tanh_implr#rr*ct||d|jd}|j}tj|gdz}|j |d}|j |||d}|j ||} |j} |j} t|||| g} t|||| g} t|||| g}t|||| g}|j| |}|j| |}|j| |}|j| |}|j||}|j||}|j||}|j||}|j||}|j||}|j||}|j||}|j||}|j!||}|j||| _|j||| _| j#S)Nrbrr`r)r)rrrrrhrHrrrrrrrrrrr)r8r9r%rrrrrrrUrrsicishrchr_rsis_rcicsqr_rcsqr_icrrrs_rcis_icis_rcrs_icnumrnumis rnp_complex_tanh_implr(s!dA. !B  C   seAg &E   sC (CWb$q'2A   w +C B B '7EB4 8B '7EB4 8B GWebT :C Wgurd ;D b# B ,,r4 C b$ B b# B \\"b !F \\"b !F VV$A LLa E LLR E LLb !E LLb !E LLR E <<u %D <<u %D||D%(CH||D%(CH ==?r*cLt||dtj||||Sr\)r)r asinh_implrs rnp_real_asinh_implrQrzr*cLt||dtj||||Sr\)r)r acosh_implrs rnp_real_acosh_implrYrzr*ct||d|jd}tj|gdz}|j ||d}|d}t j |||||g}t j|||||g} t||||g} t|||| g} t j|||| | g} t j ||||| g} t|||| gS)Nrbrry?) r)rrrget_constant_genericrcomplex_add_implcomplex_sub_implrrrq)r8r9r%rrcsig2rr x_plus_one x_minus_onesqrt_x_plus_onesqrt_x_minus_one prod_sqrtlog_args rnp_complex_acosh_implr^s!dA. !B   rd1f %E  & &wJ ?C QA))'7EAEHDJKJ**7GUQFIEKLK*7GS:,OO+GWcK=Q(('5+:+;+=>I&&wBKAMNG wwi @@r*cLt||dtj||||Sr\)r)r atanh_implrs rnp_real_atanh_implr|rzr*cJt||dtj|d|S)Nrbz llvm.floorrTrs rrr# dA.  % %g|T BBr*cJt||dtj|d|S)Nrbz llvm.ceilrTrs rnp_real_ceil_implrrWr*cJt||dtj|d|S)Nrbz llvm.truncrTrs rnp_real_trunc_implrrr*cJt||dtj|d|S)Nrbz llvm.fabsrTrs rnp_real_fabs_implrrWr*ct||dtj|jd}|Dcgc]}|j |||c}\}}|j }|j } |j } |j } |jd|| } |jd| | } |jd|| }|jd| | }|j| | }|j||}|j||Scc}w) Nr`rrrrfordrcr r)rrDrrHrrrrlrm)r8r9r%rrrrrrryryixr_gt_yr no_nan_xi_yixr_eq_yrxi_ge_yi first_term second_terms rrErE!dA5==I !BHLM$$Wb$<MHC B B B B##CR0H''r26L##D"b1H##D"b1Hh 5J,,x2K ;;z; //NDct||dtj|jd}|Dcgc]}|j |||c}\}}|j }|j } |j } |j } |jd|| } |jd| | } |jd|| }|jd| | }|j| | }|j||}|j||Scc}w) Nr`rrrrrrcrr )r8r9r%rrrrrrrr!r"xr_lt_yrr$r%xi_le_yir'r(s rnp_complex_le_implr.r)r*ct||dtj|jd}|Dcgc]}|j |||c}\}}|j }|j } |j } |j } |jd|| } |jd| | } |jd|| }|jd| | }|j| | }|j||}|j||Scc}w)Nr`rrrrfrrcr )r8r9r%rrrrrrrr!r"r#r$r%xi_gt_yir'r(s rnp_complex_gt_implr1!dA5==I !BHLM$$Wb$<MHC B B B B##CR0H''r26L##D"b1H##CR0Hh 5J,,x2K ;;z; //Nr*ct||dtj|jd}|Dcgc]}|j |||c}\}}|j }|j } |j } |j } |jd|| } |jd| | } |jd|| }|jd| | }|j| | }|j||}|j||Scc}w)Nr`rrrrrrcr )r8r9r%rrrrrrrr!r"r,r$r%xi_lt_yir'r(s rrGrGr2r*c~t||dtj|jd}|Dcgc]}|j |||c}\}}|j }|j } |j } |j } |jd|| } |jd| | } |j| | Scc}w)Nr`rrrrc) r)rrDrrHrrrrl)r8r9r%rrrrrrrr!r"r%xi_eq_yis rrFrFs!dA5==I !BHLM$$Wb$<MHC B B B B##D"b1H##D"b1H <<( ++NB:c~t||dtj|jd}|Dcgc]}|j |||c}\}}|j }|j } |j } |j } |jd|| } |jd| | } |j| | Scc}w)Nr`rrrrg) r)rrDrrHrrfcmp_unorderedrm)r8r9r%rrrrrrrr!r"xr_ne_yrxi_ne_yis rnp_complex_ne_implr<s!dA5==I !BHLM$$Wb$<MHC B B B B%%dB3H%%dB3H ;;x **Nr7c|j|||}tj||j}tj||j}|j ||S)Nr)rHr is_truerrrm)r8r9rval complex_valre_trueim_trues r_complex_is_truerC&sV&&w#&>Koog{'7'78Goog{'7'78G ;;w ((r*ct||dtjtj||d}tj||d}|j ||SNr`rrrb)r)rrDr r>rlr8r9r%rr4r5s rnp_logical_and_implrG-sM dA5==Ia)Aa)A <<1 r*ct||dtjt|||jd|d}t|||jd|d}|j ||SrE)r)rrDrCrrlrFs rnp_complex_logical_and_implrI4s] dA5==I'388A;Q@A'388A;Q@A <<1 r*ct||dtjtj||d}tj||d}|j ||SrE)r)rrDr r>rmrFs rnp_logical_or_implrK;M dA5==Ia)Aa)A ;;q! r*ct||dtjt|||jd|d}t|||jd|d}|j ||SrE)r)rrDrCrrmrFs rnp_complex_logical_or_implrNB] dA5==I'388A;Q@A'388A;Q@A ;;q! r*ct||dtjtj||d}tj||d}|j ||SrE)r)rrDr r>rsrFs rnp_logical_xor_implrQIrLr*ct||dtjt|||jd|d}t|||jd|d}|j ||SrE)r)rrDrCrrsrFs rnp_complex_logical_xor_implrSPrOr*cnt||dtjtj||dSNrbrr)r)rrDr is_falsers rnp_logical_not_implrWWs+ dA5==I   GT!W --r*ct||dtjt|||jd|d}|j |SrU)r)rrDrCrnot_)r8r9r%rr4s rnp_complex_logical_not_implrZ\s> dA5==I'388A;Q@A <<?r*crt||d|\}}|jd||}|j|||SNr`rr)rrrt)r8r9r%rarg1arg2 arg1_sge_arg2s rnp_int_smax_implrak> dA.JD$''dD9M >>-t 44r*crt||d|\}}|jd||}|j|||Sr\r)rkrt)r8r9r%rr^r_ arg1_uge_arg2s rnp_int_umax_implrfr> dA.JD$))$d;M >>-t 44r*c t||d|\}}|jd||}|jd||}|j|||}|jd||} |j| ||} |j||| SNr`unorr)r9rtr) r8r9r%rr^r_arg1_nanany_nan nan_resultrMnon_nan_results rnp_real_maximum_implrpy dA.JD$%%eT48H$$UD$7G$5J''dD9L^^L$=N >>':~ >>r*c t||d|\}}|jd||}|jd||}|j|||}|jd||} |j| ||} |j||| Srirk) r8r9r%rr^r_arg2_nanrmrnrMros rnp_real_fmax_implrtrqr*ct||d|jd}tjtj |}tjtj g|gdz}|\}}t ||||g} t ||||g} |j| | } |j| ||} t||||} |j| ||}|j| | |SNr`r r)rrrrrDnp_complex_isnan_implrmrtrEr8r9r%rrbc_sigbcc_sigr^r_rlrsrmrnrMros rnp_complex_maximum_implr| !dA. !B   emmR 0Fu}}6tAv6GJD$$WgvvFH$WgvvFHkk(H-G$5J%gwFL^^L$=N >>':~ >>r*ct||d|jd}tjtj |}tjtj g|gdz}|\}}t ||||g} t ||||g} |j| | } |j| ||} t||||} |j| ||}|j| | |Srvrwrys rnp_complex_fmax_implr !dA. !B   emmR 0Fu}}6tAv6GJD$$WgvvFH$WgvvFHkk(H-G$5J%gwFL^^L$=N >>':~ >>r*crt||d|\}}|jd||}|j|||SNr`rr])r8r9r%rr^r_ arg1_sle_arg2s rnp_int_smin_implrrbr*crt||d|\}}|jd||}|j|||Srrd)r8r9r%rr^r_ arg1_ule_arg2s rnp_int_umin_implrrgr*c t||d|\}}|jd||}|jd||}|j|||}|jd||} |j| ||} |j||| SNr`rjrrk r8r9r%rr^r_rlrmrn arg1_le_arg2ros rnp_real_minimum_implrrqr*c t||d|\}}|jd||}|jd||}|j|||}|jd||} |j| ||} |j||| Srrkrs rnp_real_fmin_implrrqr*ct||d|jd}tjtj |}tjtj g|gdz}|\}}t ||||g} t ||||g} |j| | } |j| ||} t||||} |j| ||}|j| | |Srv r)rrrrrDrxrmrtr.r8r9r%rrrzr{r^r_rlrsrmrnrros rnp_complex_minimum_implrr}r*ct||d|jd}tjtj |}tjtj g|gdz}|\}}t ||||g} t ||||g} |j| | } |j| ||} t||||} |j| ||}|j| | |Srvrrs rnp_complex_fmin_implrrr*c\t||dtjtjSNrbrr)rrDr false_bitrs rnp_int_isnan_implr  dA5==I   r*cnt||dtjtj||dSrU)r)rrDris_nanrs rnp_real_isnan_implr!) dA5==I ??7DG ,,r*ct||dtj|\}|j\}|j |||}t j ||SNrbrr)r)rrDrrHrrr8r9r%rrrr@s rrxrx&sP dA5==I BA ((CB&&w!&s+ dA5==I   gtAw //r*ct||dtj|\}|j\}|j |||}t j ||Sr)r)rrDrrHrrrs rnp_complex_isfinite_implrCsP dA5==I BA ((CB&&w!&ll_sinf inner_argsrs rnp_real_spacing_implrs!dA.  ) (N 88DR  "b9I GLLE 266]F ;; # #EE5> :D  % %gnnd+: rs% )2GG1660 5 4 4 &mm 0; I .|"J8D .*D$(& =GZ 2 NJ"&&A   +i()OS+rww? !$; D 0lC@ C: F&'"ZB &: ;' I: ;'  <  <0<2; <) *9?. E& $T: ;: ;:;;;BB; 8; 6; #R<< A<<CBCB0*0*0*0*,"+.). 55 ? ??,?*55 ? ??,?0 - 2 @ > 0 5 - 2$E D,: Er*