
    XhC                        d dl mZ ej        d8d            Zej        d8d            Zej        d8d            Zej        d8d            Zej        d8d            Zej        d8d            Zej        d8d	            Z	ej        d8d
            Z
ej        d8d            Zej        d8d            Zej        d8d            Zej        d8d            Zej        d8d            Zej        d8d            Zej        d8d            Zej        d8d            Zej        d8d            Zej        d8d            Zej        d8d            Zej        d8d            Zej        d8d            Zej        d8d            Zej        d8d            Zej        d8d            Zej        d8d            Zej        d8d            Zej        d8d            Zej        d8d            Zej        d8d            Zej        d8d             Z ej        d8d!            Z!ej        d8d"            Z"ej        d8d#            Z#ej        d8d$            Z$ej        d8d%            Z%ej        d8d&            Z&ej        d8d'            Z'ej        d8d(            Z(ej        d8d)            Z)ej        d8d*            Z*ej        d8d+            Z+ej        d8d,            Z,ej        d8d-            Z-ej        d8d.            Z.ej        d8d/            Z/ej        d8d0            Z0ej        d8d1            Z1ej        d8d2            Z2ej        d8d3            Z3ej        d8d4            Z4ej        d8d5            Z5ej        d8d6            Z6ej        d8d7            Z7dS )9    )coreNc                    t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          ft          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|	          S )
N int32__triton_hip_iabsint64fp32__triton_hip_fabsfp64Tis_pure	_semanticr   extern_elementwisedtypearg0r   s     u/var/www/tools.fuzzalab.pt/emblema-extractor/venv/lib/python3.11/site-packages/triton/language/extra/hip/libdevice.pyabsr      s    "
BZ  #&94:g;N;N%OZ  #&94:g;N;N%OZ"%8$*V:L:L$MZ"%8$*V:L:L$M	

 9. . . .    c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_floor_f32r   __ocml_floor_f64Tr   r   r   s     r   floorr      u    "
BZ"%7F9K9K$LZ"%7F9K9K$L
 9	. . . .r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_rsqrt_f32r   __ocml_rsqrt_f64Tr   r   r   s     r   rsqrtr      r   r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_ceil_f32r   __ocml_ceil_f64Tr   r   r   s     r   ceilr#   !   u    "
BZ"%6
68J8J$KZ"%6
68J8J$K
 9	. . . .r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_trunc_f32r   __ocml_trunc_f64Tr   r   r   s     r   truncr(   *   r   r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_exp2_f32r   __ocml_exp2_f64Tr   r   r   s     r   exp2r,   3   r$   r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_exp_f32r   __ocml_exp_f64Tr   r   r   s     r   expr0   <   u    "
BZ"%5tz&7I7I$JZ"%5tz&7I7I$J
 9	. . . .r   c           
          t          j        dd| gt          j        d          fdt          j        d          fid|          S )Nr   r	   __triton_hip_fast_expfTr   r   r   s     r   	fast_expfr4   E   sP    "2rD6	F		!94:f;M;M N4y* * * *r   c           
          t          j        dd| |gt          j        d          t          j        d          fdt          j        d          fid|          S )Nr   r	   __triton_hip_fast_fdividefTr   r   r   arg1r   s      r   fast_dividefr9   L   sa    "2rD$<	F		TZ//03OQUQ[\bQcQc2d:y* * * *r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_sqrt_f32r   __ocml_sqrt_f64Tr   r   r   s     r   sqrtr=   S   r$   r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __triton_hip_llrintr   r   Tr   r   r   s     r   llrintr@   \   su    "
BZ"%:DJw<O<O$PZ"%:DJw<O<O$P
 9	. . . .r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_nearbyint_f32r   __ocml_nearbyint_f64Tr   r   r   s     r   	nearbyintrD   e   s{    "
B
 Z"%;TZ=O=O$PZ"%;TZ=O=O$P
 9. . . .r   c                    t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|                              t           j        |	          S )
Nr   r	   __ocml_isnan_f32r   r   __ocml_isnan_f64Tr   r   r   r   r   toint1r   s     r   isnanrL   p   s    "
B
 Z"%7G9L9L$MZ"%7G9L9L$M
 9. . . /1bib.P.PQr   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )	Nr   r	   __ocml_signbit_f32r   r   __ocml_signbit_f64Tr   r   r   s     r   signbitrP   {   s{    "
B
 Z"%94:g;N;N$OZ"%94:g;N;N$O
 9. . . .r   c                 *   t          j        dd| |gt          j        d          t          j        d          fdt          j        d          ft          j        d          t          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_copysign_f32r   __ocml_copysign_f64Tr   r   r7   s      r   copysignrT      s    "
BtZF!3!347LdjY_N`N`6aZF!3!347LdjY_N`N`6a
 9	. . . .r   c                    t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|                              t           j        |	          S )
Nr   r	   __ocml_isinf_f32r   r   __ocml_isinf_f64Tr   rH   rI   r   s     r   isinfrX      s    "
BZ"%7G9L9L$MZ"%7G9L9L$M
 9	. . . /1bib.P.P	Qr   c                 *   t          j        dd| |gt          j        d          t          j        d          fdt          j        d          ft          j        d          t          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_nextafter_f32r   __ocml_nextafter_f64Tr   r   r7   s      r   	nextafterr\      s    "
BtZF!3!347MtzZ`OaOa6bZF!3!347MtzZ`OaOa6b
 9	. . . .r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_sin_f32r   __ocml_sin_f64Tr   r   r   s     r   sinr`      r1   r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_cos_f32r   __ocml_cos_f64Tr   r   r   s     r   cosrd      r1   r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_tan_f32r   __ocml_tan_f64Tr   r   r   s     r   tanrh      r1   r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_log2_f32r   __ocml_log2_f64Tr   r   r   s     r   log2rl      r$   r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_cosh_f32r   __ocml_cosh_f64Tr   r   r   s     r   coshrp      r$   r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_sinh_f32r   __ocml_sinh_f64Tr   r   r   s     r   sinhrt      r$   r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_tanh_f32r   __ocml_tanh_f64Tr   r   r   s     r   tanhrx      r$   r   c                 *   t          j        dd| |gt          j        d          t          j        d          fdt          j        d          ft          j        d          t          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_atan2_f32r   __ocml_atan2_f64Tr   r   r7   s      r   atan2r|          "
BtZF!3!347I4:V\K]K]6^ZF!3!347I4:V\K]K]6^
 9	. . . .r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_atan_f32r   __ocml_atan_f64Tr   r   r   s     r   atanr      r$   r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_asin_f32r   __ocml_asin_f64Tr   r   r   s     r   asinr      r$   r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_acos_f32r   __ocml_acos_f64Tr   r   r   s     r   acosr      r$   r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_log_f32r   __ocml_log_f64Tr   r   r   s     r   logr     r1   r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_log10_f32r   __ocml_log10_f64Tr   r   r   s     r   log10r     r   r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_log1p_f32r   __ocml_log1p_f64Tr   r   r   s     r   log1pr     r   r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_acosh_f32r   __ocml_acosh_f64Tr   r   r   s     r   acoshr     r   r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_asinh_f32r   __ocml_asinh_f64Tr   r   r   s     r   asinhr   (  r   r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_atanh_f32r   __ocml_atanh_f64Tr   r   r   s     r   atanhr   1  r   r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_expm1_f32r   __ocml_expm1_f64Tr   r   r   s     r   expm1r   :  r   r   c                 *   t          j        dd| |gt          j        d          t          j        d          fdt          j        d          ft          j        d          t          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_hypot_f32r   __ocml_hypot_f64Tr   r   r7   s      r   hypotr   C  r}   r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_j0_f32r   __ocml_j0_f64Tr   r   r   s     r   j0r   L  s    "
BZ"_dj6H6H$IZ"_dj6H6H$I
 9	. . . .r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_j1_f32r   __ocml_j1_f64Tr   r   r   s     r   j1r   U  r   r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_y0_f32r   __ocml_y0_f64Tr   r   r   s     r   y0r   ^  r   r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_y1_f32r   __ocml_y1_f64Tr   r   r   s     r   y1r   g  r   r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_i0_f32r   __ocml_i0_f64Tr   r   r   s     r   cyl_bessel_i0r   p  r   r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_i1_f32r   __ocml_i1_f64Tr   r   r   s     r   cyl_bessel_i1r   y  r   r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_erf_f32r   __ocml_erf_f64Tr   r   r   s     r   erfr     r1   r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_erfinv_f32r   __ocml_erfinv_f64Tr   r   r   s     r   erfinvr     u    "
BZ"%8$*V:L:L$MZ"%8$*V:L:L$M
 9	. . . .r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_erfc_f32r   __ocml_erfc_f64Tr   r   r   s     r   erfcr     r$   r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_erfcx_f32r   __ocml_erfcx_f64Tr   r   r   s     r   erfcxr     r   r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_lgamma_f32r   __ocml_lgamma_f64Tr   r   r   s     r   lgammar     r   r   c                 *   t          j        dd| |gt          j        d          t          j        d          fdt          j        d          ft          j        d          t          j        d          fdt          j        d          fid|          S )	Nr   r	   r   __ocml_ldexp_f32r   __ocml_ldexp_f64Tr   r   r7   s      r   ldexpr     s    "
BtZG!4!458JDJW]L^L^7_ZG!4!458JDJW]L^L^7_
 9	. . . .r   c                 *   t          j        dd| |gt          j        d          t          j        d          fdt          j        d          ft          j        d          t          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_fmod_f32r   __ocml_fmod_f64Tr   r   r7   s      r   fmodr     s    "
BtZF!3!347H$*U[J\J\6]ZF!3!347H$*U[J\J\6]
 9	. . . .r   c                 x   t          j        dd| ||gt          j        d          t          j        d          t          j        d          fdt          j        d          ft          j        d          t          j        d          t          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_fma_f32r   __ocml_fma_f64Tr   r   )r   r8   arg2r   s       r   fmar     s    "
BtT"ZF!3!3TZ5G5GHK[]a]ghn]o]oJpZF!3!3TZ5G5GHK[]a]ghn]o]oJp%
 9	. . . .r   c                    t          j        dd| |gt          j        d          t          j        d          fdt          j        d          ft          j        d          t          j        d          fdt          j        d          ft          j        d          t          j        d          fdt          j        d          ft          j        d          t          j        d          fdt          j        d          fid	|
          S )Nr   r	   r   __ocml_pown_f32r   __ocml_pown_f64__ocml_pow_f32__ocml_pow_f64Tr   r   r7   s      r   powr     s    "
BtZG!4!458I4:V\K]K]7^ZG!4!458I4:V\K]K]7^ZF!3!347GTZI[I[6\ZF!3!347GTZI[I[6\	

 9. . . .r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )	Nr   r	   __ocml_ilogb_f32r   r   __ocml_ilogb_f64Tr   r   r   s     r   ilogbr     su    "
BZ"%7G9L9L$MZ"%7G9L9L$M
 9	. . . .r   c                     t          j        dd| gt          j        d          fdt          j        d          ft          j        d          fdt          j        d          fid|          S )Nr   r	   __ocml_round_f32r   __ocml_round_f64Tr   r   r   s     r   roundr     r   r   )N)8triton.languager   externr   r   r   r#   r(   r,   r0   r4   r9   r=   r@   rD   rL   rP   rT   rX   r\   r`   rd   rh   rl   rp   rt   rx   r|   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r   r   <module>r      s3               . . . . . . . . . . . . . . . . . . . . . . . . . . . . * * * * * * * * . . . . . . . . . . . . Q Q Q Q . . . . . . . . Q Q Q Q . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . .r   