a
    khC                     @   s`  d dl m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'd(Zejdd)d*Zejdd+d,Zejdd-d.Zejdd/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Z5ejddidjZ6ejddkdlZ7d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	_semanticr   extern_elementwisedtypearg0r
    r   Q/var/www/auris/lib/python3.9/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   s.   t jdd| gt dfdt dfid|dS )Nr   r   Z__triton_hip_fast_expfTr   r   r   r   r   r   	fast_expfE   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_dividefL   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   sqrtS   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   llrint\   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   	nearbyinte   s    r   c                 C   sP   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   Z__ocml_isnan_f32r   r   Z__ocml_isnan_f64Tr   r
   r   r   r   toZint1r   r   r   r   isnanp   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   signbit{   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   sP   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   Z__ocml_isinf_f32r   r   Z__ocml_isinf_f64Tr   r    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    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_asin_f32r   Z__ocml_asin_f64Tr   r   r   r   r   r   asin   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_acos_f32r   Z__ocml_acos_f64Tr   r   r   r   r   r   acos   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_log_f32r   Z__ocml_log_f64Tr   r   r   r   r   r   log  s    r3   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    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_log1p_f32r   Z__ocml_log1p_f64Tr   r   r   r   r   r   log1p  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_acosh_f32r   Z__ocml_acosh_f64Tr   r   r   r   r   r   acosh  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_asinh_f32r   Z__ocml_asinh_f64Tr   r   r   r   r   r   asinh(  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_atanh_f32r   Z__ocml_atanh_f64Tr   r   r   r   r   r   atanh1  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_expm1_f32r   Z__ocml_expm1_f64Tr   r   r   r   r   r   expm1:  s    r9   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   hypotC  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_j0_f32r   Z__ocml_j0_f64Tr   r   r   r   r   r   j0L  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_j1_f32r   Z__ocml_j1_f64Tr   r   r   r   r   r   j1U  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_y0_f32r   Z__ocml_y0_f64Tr   r   r   r   r   r   y0^  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_y1_f32r   Z__ocml_y1_f64Tr   r   r   r   r   r   y1g  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_i0_f32r   Z__ocml_i0_f64Tr   r   r   r   r   r   cyl_bessel_i0p  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_i1_f32r   Z__ocml_i1_f64Tr   r   r   r   r   r   cyl_bessel_i1y  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    rA   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    rB   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    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_erfcx_f32r   Z__ocml_erfcx_f64Tr   r   r   r   r   r   erfcx  s    rD   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    rE   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    
rF   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    
rG   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    &&rH   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    
rI   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    rJ   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_round_f32r   Z__ocml_round_f64Tr   r   r   r   r   r   round  s    rK   )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)N)N)8Ztriton.languager   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/   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   r   r   r   r   <module>   s   




