U
    h@                     @   s@  d dl mZ ejdiddZejdjddZejdkddZejdld	d
ZejdmddZejdnddZejdoddZ	ejdpddZ
ejdqddZejdrddZejdsddZejdtddZejduddZejdvddZejdwdd Zejdxd!d"Zejdyd#d$Zejdzd%d&Zejd{d'd(Zejd|d)d*Zejd}d+d,Zejd~d-d.Zejdd/d0Zejdd1d2Zejdd3d4Zejdd5d6Zejdd7d8Zejdd9d:Zejdd;d<Zejdd=d>Z ejdd?d@Z!ejddAdBZ"ejddCdDZ#ejddEdFZ$ejddGdHZ%ejddIdJZ&ejddKdLZ'ejddMdNZ(ejddOdPZ)ejddQdRZ*ejddSdTZ+ejddUdVZ,ejddWdXZ-ejddYdZZ.ejdd[d\Z/ejdd]d^Z0ejdd_d`Z1ejddadbZ2ejddcddZ3ejddedfZ4ejddgdhZ5dS )    )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 int32Z__triton_hip_iabsint64fp32Z__triton_hip_fabsfp64Tis_pure_builderr   Zextern_elementwiseZdtypearg0r
    r   U/var/www/html/venv/lib/python3.8/site-packages/triton/language/extra/hip/libdevice.pyabs   s       
 
 
 
  r   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   Z__ocml_floor_f32r   Z__ocml_floor_f64Tr   r   r   r   r   r   floor   s      
 
  r   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   Z__ocml_rsqrt_f32r   Z__ocml_rsqrt_f64Tr   r   r   r   r   r   rsqrt   s      
 
  r   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   Z__ocml_ceil_f32r   Z__ocml_ceil_f64Tr   r   r   r   r   r   ceil!   s      
 
  r   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   Z__ocml_trunc_f32r   Z__ocml_trunc_f64Tr   r   r   r   r   r   trunc*   s      
 
  r   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   Z__ocml_exp2_f32r   Z__ocml_exp2_f64Tr   r   r   r   r   r   exp23   s      
 
  r   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   Z__ocml_exp_f32r   Z__ocml_exp_f64Tr   r   r   r   r   r   exp<   s      
 
  r   c              	   C   s8   t jdd| |gt dt dfdt dfid|dS )Nr   r   Z__triton_hip_fast_fdividefTr   r   r   arg1r
   r   r   r   fast_dividefE   s      r   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   Z__ocml_sqrt_f32r   Z__ocml_sqrt_f64Tr   r   r   r   r   r   sqrtL   s      
 
  r   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   Z__triton_hip_llrintr   r   Tr   r   r   r   r   r   llrintU   s      
 
  r   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   Z__ocml_nearbyint_f32r   Z__ocml_nearbyint_f64Tr   r   r   r   r   r   	nearbyint^   s     
 
  r   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   Z__ocml_isnan_f32r   r   Z__ocml_isnan_f64Tr   r   r   r   r   r   isnani   s     
 
  r   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   Z__ocml_signbit_f32r   r   Z__ocml_signbit_f64Tr   r   r   r   r   r   signbitt   s     
 
  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   Z__ocml_copysign_f32r   Z__ocml_copysign_f64Tr   r   r   r   r   r   copysign   s         r   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   Z__ocml_isinf_f32r   r   Z__ocml_isinf_f64Tr   r   r   r   r   r   isinf   s      
 
  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   Z__ocml_nextafter_f32r   Z__ocml_nextafter_f64Tr   r   r   r   r   r   	nextafter   s         r!   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   Z__ocml_sin_f32r   Z__ocml_sin_f64Tr   r   r   r   r   r   sin   s      
 
  r"   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   Z__ocml_cos_f32r   Z__ocml_cos_f64Tr   r   r   r   r   r   cos   s      
 
  r#   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   Z__ocml_tan_f32r   Z__ocml_tan_f64Tr   r   r   r   r   r   tan   s      
 
  r$   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   Z__ocml_log2_f32r   Z__ocml_log2_f64Tr   r   r   r   r   r   log2   s      
 
  r%   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   Z__ocml_cosh_f32r   Z__ocml_cosh_f64Tr   r   r   r   r   r   cosh   s      
 
  r&   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   Z__ocml_sinh_f32r   Z__ocml_sinh_f64Tr   r   r   r   r   r   sinh   s      
 
  r'   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   Z__ocml_tanh_f32r   Z__ocml_tanh_f64Tr   r   r   r   r   r   tanh   s      
 
  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   Z__ocml_atan2_f32r   Z__ocml_atan2_f64Tr   r   r   r   r   r   atan2   s         r)   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   Z__ocml_atan_f32r   Z__ocml_atan_f64Tr   r   r   r   r   r   atan   s      
 
  r*   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   Z__ocml_asin_f32r   Z__ocml_asin_f64Tr   r   r   r   r   r   asin   s      
 
  r+   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   Z__ocml_acos_f32r   Z__ocml_acos_f64Tr   r   r   r   r   r   acos   s      
 
  r,   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   Z__ocml_log_f32r   Z__ocml_log_f64Tr   r   r   r   r   r   log   s      
 
  r-   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   Z__ocml_log10_f32r   Z__ocml_log10_f64Tr   r   r   r   r   r   log10  s      
 
  r.   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   Z__ocml_log1p_f32r   Z__ocml_log1p_f64Tr   r   r   r   r   r   log1p  s      
 
  r/   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   Z__ocml_acosh_f32r   Z__ocml_acosh_f64Tr   r   r   r   r   r   acosh  s      
 
  r0   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   Z__ocml_asinh_f32r   Z__ocml_asinh_f64Tr   r   r   r   r   r   asinh!  s      
 
  r1   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   Z__ocml_atanh_f32r   Z__ocml_atanh_f64Tr   r   r   r   r   r   atanh*  s      
 
  r2   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   Z__ocml_expm1_f32r   Z__ocml_expm1_f64Tr   r   r   r   r   r   expm13  s      
 
  r3   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   Z__ocml_hypot_f32r   Z__ocml_hypot_f64Tr   r   r   r   r   r   hypot<  s         r4   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   Z__ocml_j0_f32r   Z__ocml_j0_f64Tr   r   r   r   r   r   j0E  s      
 
  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   Z__ocml_j1_f32r   Z__ocml_j1_f64Tr   r   r   r   r   r   j1N  s      
 
  r6   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   Z__ocml_y0_f32r   Z__ocml_y0_f64Tr   r   r   r   r   r   y0W  s      
 
  r7   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   Z__ocml_y1_f32r   Z__ocml_y1_f64Tr   r   r   r   r   r   y1`  s      
 
  r8   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   Z__ocml_i0_f32r   Z__ocml_i0_f64Tr   r   r   r   r   r   cyl_bessel_i0i  s      
 
  r9   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   Z__ocml_i1_f32r   Z__ocml_i1_f64Tr   r   r   r   r   r   cyl_bessel_i1r  s      
 
  r:   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   Z__ocml_erf_f32r   Z__ocml_erf_f64Tr   r   r   r   r   r   erf{  s      
 
  r;   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   Z__ocml_erfinv_f32r   Z__ocml_erfinv_f64Tr   r   r   r   r   r   erfinv  s      
 
  r<   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   Z__ocml_erfc_f32r   Z__ocml_erfc_f64Tr   r   r   r   r   r   erfc  s      
 
  r=   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   Z__ocml_erfcx_f32r   Z__ocml_erfcx_f64Tr   r   r   r   r   r   erfcx  s      
 
  r>   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   Z__ocml_lgamma_f32r   Z__ocml_lgamma_f64Tr   r   r   r   r   r   lgamma  s      
 
  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   Z__ocml_ldexp_f32r   Z__ocml_ldexp_f64Tr   r   r   r   r   r   ldexp  s         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   Z__ocml_fmod_f32r   Z__ocml_fmod_f64Tr   r   r   r   r   r   fmod  s         rA   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   Z__ocml_fma_f32r   Z__ocml_fma_f64Tr   r   )r   r   Zarg2r
   r   r   r   fma  s         rB   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   Z__ocml_pown_f32r   Z__ocml_pown_f64Z__ocml_pow_f32Z__ocml_pow_f64Tr   r   r   r   r   r   pow  s            rC   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   Z__ocml_ilogb_f32r   r   Z__ocml_ilogb_f64Tr   r   r   r   r   r   ilogb  s      
 
  rD   )N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)6Ztriton.languager   Zexternr   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/   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   r   r   r   r   <module>   s   




