o
    ^iD                     @   sp  d dl mZ ejdoddZejdoddZejdoddZejdod	d
ZejdoddZejdoddZejdoddZ	ejdoddZ
ejdoddZejdoddZejdoddZejdoddZejdoddZejdoddZejdodd Zejdod!d"Zejdod#d$Zejdod%d&Zejdod'd(Zejdod)d*Zejdod+d,Zejdod-d.Zejdod/d0Zejdod1d2Zejdod3d4Zejdod5d6Zejdod7d8Zejdod9d:Zejdod;d<Zejdod=d>Z ejdod?d@Z!ejdodAdBZ"ejdodCdDZ#ejdodEdFZ$ejdodGdHZ%ejdodIdJZ&ejdodKdLZ'ejdodMdNZ(ejdodOdPZ)ejdodQdRZ*ejdodSdTZ+ejdodUdVZ,ejdodWdXZ-ejdodYdZZ.ejdod[d\Z/ejdod]d^Z0ejdod_d`Z1ejdodadbZ2ejdodcddZ3ejdodedfZ4ejdodgdhZ5ejdodidjZ6ejdodkdlZ7ejdodmdnZ8dS )p    )coreNc                 C   sp   t jdd| gt dfdt dft dfdt dft dfdt dft dfdt dfid|d	S )
N int32__triton_hip_iabsint64fp32__triton_hip_fabsfp64Tis_pure	_semanticr   extern_elementwisedtypearg0r    r   ^/var/www/html/RAG/RAG_venv/lib/python3.10/site-packages/triton/language/extra/hip/libdevice.pyabs   s   r   c                 C   D   t jdd| gt dfdt dft dfdt dfid|dS )Nr   r   __ocml_floor_f32r	   __ocml_floor_f64Tr
   r   r   r   r   r   floor      r   c                 C   r   )Nr   r   __ocml_rsqrt_f32r	   __ocml_rsqrt_f64Tr
   r   r   r   r   r   rsqrt   r   r   c                 C   r   )Nr   r   __ocml_ceil_f32r	   __ocml_ceil_f64Tr
   r   r   r   r   r   ceil!   r   r   c                 C   r   )Nr   r   __ocml_trunc_f32r	   __ocml_trunc_f64Tr
   r   r   r   r   r   trunc*   r   r"   c                 C   r   )Nr   r   __ocml_exp2_f32r	   __ocml_exp2_f64Tr
   r   r   r   r   r   exp23   r   r%   c                 C   r   )Nr   r   __ocml_exp_f32r	   __ocml_exp_f64Tr
   r   r   r   r   r   exp<   r   r(   c              	   C   .   t jdd| gt dfdt dfid|dS )Nr   r   __triton_hip_fast_expfTr
   r   r   r   r   r   	fast_expfE   
   r+   c              	   C   r)   )Nr   r   __triton_hip_fast_tanhfTr
   r   r   r   r   r   
fast_tanhfL   r,   r.   c              	   C   s8   t jdd| |gt dt dfdt dfid|dS )Nr   r   __triton_hip_fast_fdividefTr
   r   r   arg1r   r   r   r   fast_dividefS   s
   r2   c                 C   r   )Nr   r   __ocml_sqrt_f32r	   __ocml_sqrt_f64Tr
   r   r   r   r   r   sqrtZ   r   r5   c                 C   sD   t jdd| gt dfdt dft dfdt dfid|dS )Nr   r   __triton_hip_llrintr   r	   Tr
   r   r   r   r   r   llrintc   r   r7   c                 C   r   )Nr   r   __ocml_nearbyint_f32r	   __ocml_nearbyint_f64Tr
   r   r   r   r   r   	nearbyintl      r:   c                 C   P   t jdd| gt dfdt dft dfdt dfid|djt j|d	S )
Nr   r   __ocml_isnan_f32r   r	   __ocml_isnan_f64Tr
   r   r   r   r   toint1r   r   r   r   isnanw   s   rC   c                 C   D   t jdd| gt dfdt dft dfdt dfid|dS )	Nr   r   __ocml_signbit_f32r   r	   __ocml_signbit_f64Tr
   r   r   r   r   r   signbit   r;   rG   c                 C   V   t jdd| |gt dt dfdt dft dt dfdt dfid|dS )Nr   r   __ocml_copysign_f32r	   __ocml_copysign_f64Tr
   r   r0   r   r   r   copysign      
rK   c                 C   r<   )
Nr   r   __ocml_isinf_f32r   r	   __ocml_isinf_f64Tr
   r?   r@   r   r   r   r   isinf   s   rO   c                 C   rH   )Nr   r   __ocml_nextafter_f32r	   __ocml_nextafter_f64Tr
   r   r0   r   r   r   	nextafter   rL   rR   c                 C   r   )Nr   r   __ocml_sin_f32r	   __ocml_sin_f64Tr
   r   r   r   r   r   sin   r   rU   c                 C   r   )Nr   r   __ocml_cos_f32r	   __ocml_cos_f64Tr
   r   r   r   r   r   cos   r   rX   c                 C   r   )Nr   r   __ocml_tan_f32r	   __ocml_tan_f64Tr
   r   r   r   r   r   tan   r   r[   c                 C   r   )Nr   r   __ocml_log2_f32r	   __ocml_log2_f64Tr
   r   r   r   r   r   log2   r   r^   c                 C   r   )Nr   r   __ocml_cosh_f32r	   __ocml_cosh_f64Tr
   r   r   r   r   r   cosh   r   ra   c                 C   r   )Nr   r   __ocml_sinh_f32r	   __ocml_sinh_f64Tr
   r   r   r   r   r   sinh   r   rd   c                 C   r   )Nr   r   __ocml_tanh_f32r	   __ocml_tanh_f64Tr
   r   r   r   r   r   tanh   r   rg   c                 C   rH   )Nr   r   __ocml_atan2_f32r	   __ocml_atan2_f64Tr
   r   r0   r   r   r   atan2   rL   rj   c                 C   r   )Nr   r   __ocml_atan_f32r	   __ocml_atan_f64Tr
   r   r   r   r   r   atan   r   rm   c                 C   r   )Nr   r   __ocml_asin_f32r	   __ocml_asin_f64Tr
   r   r   r   r   r   asin   r   rp   c                 C   r   )Nr   r   __ocml_acos_f32r	   __ocml_acos_f64Tr
   r   r   r   r   r   acos  r   rs   c                 C   r   )Nr   r   __ocml_log_f32r	   __ocml_log_f64Tr
   r   r   r   r   r   log  r   rv   c                 C   r   )Nr   r   __ocml_log10_f32r	   __ocml_log10_f64Tr
   r   r   r   r   r   log10  r   ry   c                 C   r   )Nr   r   __ocml_log1p_f32r	   __ocml_log1p_f64Tr
   r   r   r   r   r   log1p  r   r|   c                 C   r   )Nr   r   __ocml_acosh_f32r	   __ocml_acosh_f64Tr
   r   r   r   r   r   acosh&  r   r   c                 C   r   )Nr   r   __ocml_asinh_f32r	   __ocml_asinh_f64Tr
   r   r   r   r   r   asinh/  r   r   c                 C   r   )Nr   r   __ocml_atanh_f32r	   __ocml_atanh_f64Tr
   r   r   r   r   r   atanh8  r   r   c                 C   r   )Nr   r   __ocml_expm1_f32r	   __ocml_expm1_f64Tr
   r   r   r   r   r   expm1A  r   r   c                 C   rH   )Nr   r   __ocml_hypot_f32r	   __ocml_hypot_f64Tr
   r   r0   r   r   r   hypotJ  rL   r   c                 C   r   )Nr   r   __ocml_j0_f32r	   __ocml_j0_f64Tr
   r   r   r   r   r   j0S  r   r   c                 C   r   )Nr   r   __ocml_j1_f32r	   __ocml_j1_f64Tr
   r   r   r   r   r   j1\  r   r   c                 C   r   )Nr   r   __ocml_y0_f32r	   __ocml_y0_f64Tr
   r   r   r   r   r   y0e  r   r   c                 C   r   )Nr   r   __ocml_y1_f32r	   __ocml_y1_f64Tr
   r   r   r   r   r   y1n  r   r   c                 C   r   )Nr   r   __ocml_i0_f32r	   __ocml_i0_f64Tr
   r   r   r   r   r   cyl_bessel_i0w  r   r   c                 C   r   )Nr   r   __ocml_i1_f32r	   __ocml_i1_f64Tr
   r   r   r   r   r   cyl_bessel_i1  r   r   c                 C   r   )Nr   r   __ocml_erf_f32r	   __ocml_erf_f64Tr
   r   r   r   r   r   erf  r   r   c                 C   r   )Nr   r   __ocml_erfinv_f32r	   __ocml_erfinv_f64Tr
   r   r   r   r   r   erfinv  r   r   c                 C   r   )Nr   r   __ocml_erfc_f32r	   __ocml_erfc_f64Tr
   r   r   r   r   r   erfc  r   r   c                 C   r   )Nr   r   __ocml_erfcx_f32r	   __ocml_erfcx_f64Tr
   r   r   r   r   r   erfcx  r   r   c                 C   r   )Nr   r   __ocml_lgamma_f32r	   __ocml_lgamma_f64Tr
   r   r   r   r   r   lgamma  r   r   c                 C   sV   t jdd| |gt dt dfdt dft dt dfdt dfid|dS )	Nr   r   r   __ocml_ldexp_f32r	   __ocml_ldexp_f64Tr
   r   r0   r   r   r   ldexp  rL   r   c                 C   rH   )Nr   r   __ocml_fmod_f32r	   __ocml_fmod_f64Tr
   r   r0   r   r   r   fmod  rL   r   c                 C   sh   t jdd| ||gt dt dt dfdt dft dt dt dfdt dfid|dS )Nr   r   __ocml_fma_f32r	   __ocml_fma_f64Tr
   r   )r   r1   arg2r   r   r   r   fma  s   &&r   c                 C   s   t jdd| |gt dt dfdt dft dt dfdt dft dt dfdt dft dt dfdt dfid	|d
S )Nr   r   r   __ocml_pown_f32r	   __ocml_pown_f64__ocml_pow_f32__ocml_pow_f64Tr
   r   r0   r   r   r   pow  s   
r   c                 C   rD   )	Nr   r   __ocml_ilogb_f32r   r	   __ocml_ilogb_f64Tr
   r   r   r   r   r   ilogb  r   r   c                 C   r   )Nr   r   __ocml_round_f32r	   __ocml_round_f64Tr
   r   r   r   r   r   round  r   r   )N)9triton.languager   externr   r   r   r   r"   r%   r(   r+   r.   r2   r5   r7   r:   rC   rG   rK   rO   rR   rU   rX   r[   r^   ra   rd   rg   rj   rm   rp   rs   rv   ry   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>   s    




