
    ciD                        d dl mZ ej                  d9d       Zej                  d9d       Zej                  d9d       Zej                  d9d       Zej                  d9d       Zej                  d9d       Zej                  d9d	       Z	ej                  d9d
       Z
ej                  d9d       Zej                  d9d       Zej                  d9d       Zej                  d9d       Zej                  d9d       Zej                  d9d       Zej                  d9d       Zej                  d9d       Zej                  d9d       Zej                  d9d       Zej                  d9d       Zej                  d9d       Zej                  d9d       Zej                  d9d       Zej                  d9d       Zej                  d9d       Zej                  d9d       Zej                  d9d       Zej                  d9d       Zej                  d9d       Zej                  d9d       Zej                  d9d        Z ej                  d9d!       Z!ej                  d9d"       Z"ej                  d9d#       Z#ej                  d9d$       Z$ej                  d9d%       Z%ej                  d9d&       Z&ej                  d9d'       Z'ej                  d9d(       Z(ej                  d9d)       Z)ej                  d9d*       Z*ej                  d9d+       Z+ej                  d9d,       Z,ej                  d9d-       Z-ej                  d9d.       Z.ej                  d9d/       Z/ej                  d9d0       Z0ej                  d9d1       Z1ej                  d9d2       Z2ej                  d9d3       Z3ej                  d9d4       Z4ej                  d9d5       Z5ej                  d9d6       Z6ej                  d9d7       Z7ej                  d9d8       Z8y):    )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     ]/var/www/html/engine/venv/lib/python3.12/site-packages/triton/language/extra/hip/libdevice.pyabsr      s    ""
BZZ #&94::g;N%OZZ #&94::g;N%OZZ"%8$**V:L$MZZ"%8$**V: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      j    ""
BZZ"%7F9K$LZZ"%7F9K$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#   !   j    ""
BZZ"%6

68J$KZZ"%6

68J$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   <   j    ""
BZZ"%5tzz&7I$JZZ"%5tzz&7I$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   sG    ""2rD6	F	!94::f;M N4y* *r   c           
          t        j                  dd| gt        j                  d      fdt        j                  d      fid|      S )Nr   r	   __triton_hip_fast_tanhfTr   r   r   s     r   
fast_tanhfr7   L   sG    ""2rD6	F	!:DJJv<N O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_dividefr<   S   sX    ""2rD$<	F	TZZ/03OQUQ[Q[\bQc2d: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@   Z   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   llrintrC   c   sj    ""
BZZ"%:DJJw<O$PZZ"%:DJJw<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   	nearbyintrG   l   sp    ""
B
 ZZ"%;TZZ=O$PZZ"%;TZZ=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|      j                  t         j                  |	      S )
Nr   r	   __ocml_isnan_f32r   r   __ocml_isnan_f64Tr   r   r   r   r   toint1r   s     r   isnanrO   w   s    ""
B
 ZZ"%7G9L$MZZ"%7G9L$M
 9. /1bib.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   signbitrS      sp    ""
B
 ZZ"%94::g;N$OZZ"%94::g;N$O
 9. .r   c                 8   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   r:   s      r   copysignrW      s    ""
BtZZF!347LdjjY_N`6aZZF!347LdjjY_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|      j                  t         j                  |	      S )
Nr   r	   __ocml_isinf_f32r   r   __ocml_isinf_f64Tr   rK   rL   r   s     r   isinfr[      s    ""
BZZ"%7G9L$MZZ"%7G9L$M
 9	. /1bib.P	Qr   c                 8   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   r:   s      r   	nextafterr_      s    ""
BtZZF!347MtzzZ`Oa6bZZF!347MtzzZ`Oa6b
 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   sinrc      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   cosrg      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   tanrk      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   log2ro      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   coshrs      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   sinhrw      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   tanhr{      r$   r   c                 8   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   r:   s      r   atan2r          ""
BtZZF!347I4::V\K]6^ZZF!347I4::V\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   8  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   A  r   r   c                 8   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   r:   s      r   hypotr   J  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   S  h    ""
BZZ"_djj6H$IZZ"_djj6H$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   \  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   e  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   n  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   w  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     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     j    ""
BZZ"%8$**V:L$MZZ"%8$**V: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                 8   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   r:   s      r   ldexpr     s    ""
BtZZG!458JDJJW]L^7_ZZG!458JDJJW]L^7_
 9	. .r   c                 8   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   r:   s      r   fmodr     s    ""
BtZZF!347H$**U[J\6]ZZF!347H$**U[J\6]
 9	. .r   c                    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   r;   arg2r   s       r   fmar     s    ""
BtT"ZZF!3TZZ5GHK[]a]g]ghn]oJpZZF!3TZZ5GHK[]a]g]ghn]oJp%
 9	. .r   c                 4   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   r:   s      r   powr     s    ""
BtZZG!458I4::V\K]7^ZZG!458I4::V\K]7^ZZF!347GTZI[6\ZZF!347GTZI[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     sj    ""
BZZ"%7G9L$MZZ"%7G9L$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)9triton.languager   externr   r   r   r#   r(   r,   r0   r4   r7   r<   r@   rC   rG   rO   rS   rW   r[   r_   rc   rg   rk   ro   rs   rw   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   r   <module>r      s     . . . . . . . . . . . . . . * * * * * * . . . . . . Q Q . . . . Q Q . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . .r   