o
    iw%                    @   sV  d dl Z d dlZddlmZ e  dd ZejdddZejddd	Zejdd
dZ	ejdddZ
ejdddZejdddZejdddZejdddZejdddZejdddZejdddZejdddZejdddZejdd d!Zejdd"d#Zejdd$d%Zejdd&d'Zejdd(d)Zejdd*d+Zejdd,d-Zejdd.d/Zejdd0d1Zejdd2d3Zejdd4d5Zejdd6d7Zejdd8d9Z ejdd:d;Z!ejdd<d=Z"ejdd>d?Z#ejdd@dAZ$ejddBdCZ%ejddDdEZ&ejddFdGZ'ejddHdIZ(ejddJdKZ)ejddLdMZ*ejddNdOZ+ejddPdQZ,ejddRdSZ-ejddTdUZ.ejddVdWZ/ejddXdYZ0ejddZd[Z1ejdd\d]Z2ejdd^d_Z3ejdd`daZ4ejddbdcZ5ejddddeZ6ejddfdgZ7ejddhdiZ8ejddjdkZ9ejddldmZ:ejddndoZ;ejddpdqZ<ejddrdsZ=ejddtduZ>ejddvdwZ?ejddxdyZ@ejddzd{ZAejdd|d}ZBejdd~dZCejdddZDejdddZEejdddZFejdddZGejdddZHejdddZIejdddZJejdddZKejdddZLejdddZMejdddZNejdddZOejdddZPejdddZQejdddZRejdddZSejdddZTejdddZUejdddZVejdddZWejdddZXejdddZYejdddZZejdddZ[ejdddZ\ejdddZ]ejdddZ^ejdddZ_ejdddZ`ejdddZaejdddZbejdddZcejdddZdejdddÄZeejdddńZfejdddǄZgejdddɄZhejddd˄Ziejddd̈́ZjejdddτZkejdddфZlejdddӄZmejdddՄZnejdddׄZoejdddلZpejdddۄZqejddd݄Zrejddd߄ZsejdddZtejdddZuejdddZvejdddZwejdddZxejdddZyejdddZzejdddZ{ejdddZ|ejdddZ}ejdddZ~ejdddZejdddZejdddZejdddZejdddZejdd dZejdddZejdddZejdddZejddd	Zejdd
dZejdddZejdddZejdddZejdddZejdddZejdddZejdddZejdddZejdddZejdddZejdd d!Zejdd"d#Zejdd$d%Zejdd&d'Zejdd(d)Zejdd*d+Zejdd,d-Zejdd.d/Zejdd0d1Zejdd2d3Zejdd4d5Zejdd6d7Zejdd8d9Zejdd:d;Zejdd<d=Zejdd>d?Zejdd@dAZejddBdCZejddDdEZejddFdGZejddHdIZejddJdKZejddLdMZejddNdOZejddPdQZejddRdSZejddTdUZejddVdWZejddXdYZejddZd[Zejdd\d]Zejdd^d_Zejdd`daZejddbdcZejddddeZejddfdgZejddhdiZejddjdkZejddldmZejddndoZejddpdqZejddrdsZejddtduZejddvdwZejddxdyZejddzd{Zejdd|d}Zejdd~dZejdddZejdddZejdddZejdddZejdddZejdddZejdddZejdddZejdddZejdddZdS (      N   )corec                  C   sX   dd l } tjtjtjtdd}| jjd u r$tj|ddd}nd}t	d|S )	Nr   z..Zthird_partycudalibzlibdevice.10.bc ZTRITON_LIBDEVICE_PATH)
torchospathjoindirnameabspath__file__versionZhipgetenv)r   Zthird_party_dirdefault r   c/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/triton/language/math.pylibdevice_path   s    r   c                 C   F   t jdt | gt dfdt dft dfdt dfid|dS )N	libdeviceint32Z__nv_clzint64Z
__nv_clzllTZis_pure_builderr   Zextern_elementwiser   Zdtypearg0r   r   r   r   clz      r   c                 C   r   )Nr   r   Z	__nv_popcr   Z__nv_popcllTr   r   r   r   r   r   popc   r   r   c              	   C   sD   t jdt | ||gt dt dt dfdt dfid|dS )Nr   r   Z__nv_byte_permTr   r   r   arg1arg2r   r   r   r   	byte_perm"   s   (r#   c                 C      t jdt | |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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__nv_minuint32Z	__nv_uminr   Z
__nv_llminuint64Z__nv_ullminfp32Z
__nv_fminffp64Z	__nv_fminTr   r   r   r!   r   r   r   r   min)      r*   c                 C   r$   )Nr   r   Z__nv_maxr%   Z	__nv_umaxr   Z
__nv_llmaxr&   Z__nv_ullmaxr'   Z
__nv_fmaxfr(   Z	__nv_fmaxTr   r   r)   r   r   r   max5   r+   r,   c                 C   s   t jdt | |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   Z
__nv_mulhir%   Z__nv_umulhir   Z__nv_mul64hir&   Z__nv_umul64hiTr   r   r)   r   r   r   mulhiA      r-   c                 C   X   t jdt | |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
__nv_mul24r%   Z__nv_umul24Tr   r   r)   r   r   r   mul24K      r0   c                 C   F   t jdt | gt dfdt dft dfdt dfid|dS )Nr   r   Z	__nv_brevr   Z__nv_brevllTr   r   r   r   r   r   brevS   r   r3   c                 C   sj   t jdt | ||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   r%   Z__nv_sadZ	__nv_usadTr   r   r    r   r   r   sad[      &&r4   c                 C   sr   t jdt | 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 )Nr   r   Z__nv_absr   Z
__nv_llabsr'   Z
__nv_fabsfr(   Z	__nv_fabsTr   r   r   r   r   r   absc   s   r6   c                 C   r2   )Nr   r'   Z__nv_floorfr(   Z
__nv_floorTr   r   r   r   r   r   floorm   r   r7   c              	   C   0   t jdt | gt dfdt dfid|dS )Nr   r(   Z__nv_rcp64hTr   r   r   r   r   r   rcp64hu      r9   c                 C   r2   )Nr   r'   Z__nv_rsqrtfr(   Z
__nv_rsqrtTr   r   r   r   r   r   rsqrt|   r   r;   c                 C   r2   )Nr   r(   Z	__nv_ceilr'   Z
__nv_ceilfTr   r   r   r   r   r   ceil   r   r<   c                 C   r2   )Nr   r(   Z
__nv_truncr'   Z__nv_truncfTr   r   r   r   r   r   trunc   r   r=   c                 C   r2   )Nr   r'   Z
__nv_exp2fr(   Z	__nv_exp2Tr   r   r   r   r   r   exp2   r   r>   c              	   C   r8   )Nr   r'   Z__nv_saturatefTr   r   r   r   r   r   	saturatef   r:   r?   c                 C   j   t jdt | ||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__nv_fmaf_rnr(   Z__nv_fma_rnTr   r   r    r   r   r   fma_rn   r5   rA   c                 C   r@   )Nr   r'   Z__nv_fmaf_rzr(   Z__nv_fma_rzTr   r   r    r   r   r   fma_rz   r5   rB   c                 C   r@   )Nr   r'   Z__nv_fmaf_rdr(   Z__nv_fma_rdTr   r   r    r   r   r   fma_rd   r5   rC   c                 C   r@   )Nr   r'   Z__nv_fmaf_rur(   Z__nv_fma_ruTr   r   r    r   r   r   fma_ru   r5   rD   c              	   C   :   t jdt | |gt dt dfdt dfid|dS )Nr   r'   Z__nv_fast_fdividefTr   r   r)   r   r   r   fast_dividef       rF   c                 C   r/   )Nr   r'   Z__nv_fdiv_rnr(   Z__nv_ddiv_rnTr   r   r)   r   r   r   div_rn   r1   rH   c                 C   r/   )Nr   r'   Z__nv_fdiv_rzr(   Z__nv_ddiv_rzTr   r   r)   r   r   r   div_rz   r1   rI   c                 C   r/   )Nr   r'   Z__nv_fdiv_rdr(   Z__nv_ddiv_rdTr   r   r)   r   r   r   div_rd   r1   rJ   c                 C   r/   )Nr   r'   Z__nv_fdiv_rur(   Z__nv_ddiv_ruTr   r   r)   r   r   r   div_ru   r1   rK   c                 C   r2   )Nr   r'   Z__nv_frcp_rnr(   Z__nv_drcp_rnTr   r   r   r   r   r   rcp_rn   r   rL   c                 C   r2   )Nr   r'   Z__nv_frcp_rzr(   Z__nv_drcp_rzTr   r   r   r   r   r   rcp_rz   r   rM   c                 C   r2   )Nr   r'   Z__nv_frcp_rdr(   Z__nv_drcp_rdTr   r   r   r   r   r   rcp_rd   r   rN   c                 C   r2   )Nr   r'   Z__nv_frcp_rur(   Z__nv_drcp_ruTr   r   r   r   r   r   rcp_ru  r   rO   c                 C   r2   )Nr   r'   Z__nv_fsqrt_rnr(   Z__nv_dsqrt_rnTr   r   r   r   r   r   sqrt_rn
  r   rP   c                 C   r2   )Nr   r'   Z__nv_fsqrt_rzr(   Z__nv_dsqrt_rzTr   r   r   r   r   r   sqrt_rz  r   rQ   c                 C   r2   )Nr   r'   Z__nv_fsqrt_rdr(   Z__nv_dsqrt_rdTr   r   r   r   r   r   sqrt_rd  r   rR   c                 C   r2   )Nr   r'   Z__nv_fsqrt_rur(   Z__nv_dsqrt_ruTr   r   r   r   r   r   sqrt_ru"  r   rS   c                 C   r2   )Nr   r'   Z
__nv_sqrtfr(   Z	__nv_sqrtTr   r   r   r   r   r   sqrt*  r   rT   c                 C   r/   )Nr   r(   Z__nv_dadd_rnr'   Z__nv_fadd_rnTr   r   r)   r   r   r   add_rn2  r1   rU   c                 C   r/   )Nr   r(   Z__nv_dadd_rzr'   Z__nv_fadd_rzTr   r   r)   r   r   r   add_rz:  r1   rV   c                 C   r/   )Nr   r(   Z__nv_dadd_rdr'   Z__nv_fadd_rdTr   r   r)   r   r   r   add_rdB  r1   rW   c                 C   r/   )Nr   r(   Z__nv_dadd_rur'   Z__nv_fadd_ruTr   r   r)   r   r   r   add_ruJ  r1   rX   c                 C   r/   )Nr   r(   Z__nv_dmul_rnr'   Z__nv_fmul_rnTr   r   r)   r   r   r   mul_rnR  r1   rY   c                 C   r/   )Nr   r(   Z__nv_dmul_rzr'   Z__nv_fmul_rzTr   r   r)   r   r   r   mul_rzZ  r1   rZ   c                 C   r/   )Nr   r(   Z__nv_dmul_rdr'   Z__nv_fmul_rdTr   r   r)   r   r   r   mul_rdb  r1   r[   c                 C   r/   )Nr   r(   Z__nv_dmul_rur'   Z__nv_fmul_ruTr   r   r)   r   r   r   mul_ruj  r1   r\   c              	   C   0   t jdt | gt dfdt dfid|dS )Nr   r(   Z__nv_double2float_rnr'   Tr   r   r   r   r   r   double2float_rnr  r:   r^   c              	   C   r]   )Nr   r(   Z__nv_double2float_rzr'   Tr   r   r   r   r   r   double2float_rzy  r:   r_   c              	   C   r]   )Nr   r(   Z__nv_double2float_rdr'   Tr   r   r   r   r   r   double2float_rd  r:   r`   c              	   C   r]   )Nr   r(   Z__nv_double2float_rur'   Tr   r   r   r   r   r   double2float_ru  r:   ra   c              	   C   r]   )Nr   r(   Z__nv_double2int_rnr   Tr   r   r   r   r   r   double2int_rn  r:   rb   c              	   C   r]   )Nr   r(   Z__nv_double2int_rzr   Tr   r   r   r   r   r   double2int_rz  r:   rc   c              	   C   r]   )Nr   r(   Z__nv_double2int_rdr   Tr   r   r   r   r   r   double2int_rd  r:   rd   c              	   C   r]   )Nr   r(   Z__nv_double2int_rur   Tr   r   r   r   r   r   double2int_ru  r:   re   c              	   C   r]   )Nr   r(   Z__nv_double2uint_rnr   Tr   r   r   r   r   r   double2uint_rn  r:   rf   c              	   C   r]   )Nr   r(   Z__nv_double2uint_rzr   Tr   r   r   r   r   r   double2uint_rz  r:   rg   c              	   C   r]   )Nr   r(   Z__nv_double2uint_rdr   Tr   r   r   r   r   r   double2uint_rd  r:   rh   c              	   C   r]   )Nr   r(   Z__nv_double2uint_rur   Tr   r   r   r   r   r   double2uint_ru  r:   ri   c              	   C   r]   )Nr   r   Z__nv_int2double_rnr(   Tr   r   r   r   r   r   int2double_rn  r:   rj   c              	   C   r]   )Nr   r%   Z__nv_uint2double_rnr(   Tr   r   r   r   r   r   uint2double_rn  r:   rk   c              	   C   r]   )Nr   r'   Z__nv_float2int_rnr   Tr   r   r   r   r   r   float2int_rn  r:   rl   c              	   C   r]   )Nr   r'   Z__nv_float2int_rzr   Tr   r   r   r   r   r   float2int_rz  r:   rm   c              	   C   r]   )Nr   r'   Z__nv_float2int_rdr   Tr   r   r   r   r   r   float2int_rd  r:   rn   c              	   C   r]   )Nr   r'   Z__nv_float2int_rur   Tr   r   r   r   r   r   float2int_ru  r:   ro   c              	   C   r]   )Nr   r'   Z__nv_float2uint_rnr   Tr   r   r   r   r   r   float2uint_rn  r:   rp   c              	   C   r]   )Nr   r'   Z__nv_float2uint_rzr   Tr   r   r   r   r   r   float2uint_rz  r:   rq   c              	   C   r]   )Nr   r'   Z__nv_float2uint_rdr   Tr   r   r   r   r   r   float2uint_rd  r:   rr   c              	   C   r]   )Nr   r'   Z__nv_float2uint_rur   Tr   r   r   r   r   r   float2uint_ru  r:   rs   c              	   C   r]   )Nr   r   Z__nv_int2float_rnr'   Tr   r   r   r   r   r   int2float_rn  r:   rt   c              	   C   r]   )Nr   r   Z__nv_int2float_rzr'   Tr   r   r   r   r   r   int2float_rz  r:   ru   c              	   C   r]   )Nr   r   Z__nv_int2float_rdr'   Tr   r   r   r   r   r   int2float_rd  r:   rv   c              	   C   r]   )Nr   r   Z__nv_int2float_rur'   Tr   r   r   r   r   r   int2float_ru!  r:   rw   c              	   C   r]   )Nr   r%   Z__nv_uint2float_rnr'   Tr   r   r   r   r   r   uint2float_rn(  r:   rx   c              	   C   r]   )Nr   r%   Z__nv_uint2float_rzr'   Tr   r   r   r   r   r   uint2float_rz/  r:   ry   c              	   C   r]   )Nr   r%   Z__nv_uint2float_rdr'   Tr   r   r   r   r   r   uint2float_rd6  r:   rz   c              	   C   r]   )Nr   r%   Z__nv_uint2float_rur'   Tr   r   r   r   r   r   uint2float_ru=  r:   r{   c              	   C   s:   t jdt | |gt dt dfdt dfid|dS )Nr   r   Z__nv_hiloint2doubler(   Tr   r   r)   r   r   r   hiloint2doubleD  rG   r|   c              	   C   r]   )Nr   r(   Z__nv_double2lointr   Tr   r   r   r   r   r   double2lointK  r:   r}   c              	   C   r]   )Nr   r(   Z__nv_double2hiintr   Tr   r   r   r   r   r   double2hiintR  r:   r~   c              	   C   r]   )Nr   r'   Z__nv_float2ll_rnr   Tr   r   r   r   r   r   float2ll_rnY  r:   r   c              	   C   r]   )Nr   r'   Z__nv_float2ll_rzr   Tr   r   r   r   r   r   float2ll_rz`  r:   r   c              	   C   r]   )Nr   r'   Z__nv_float2ll_rdr   Tr   r   r   r   r   r   float2ll_rdg  r:   r   c              	   C   r]   )Nr   r'   Z__nv_float2ll_rur   Tr   r   r   r   r   r   float2ll_run  r:   r   c              	   C   r]   )Nr   r'   Z__nv_float2ull_rnr   Tr   r   r   r   r   r   float2ull_rnu  r:   r   c              	   C   r]   )Nr   r'   Z__nv_float2ull_rzr   Tr   r   r   r   r   r   float2ull_rz|  r:   r   c              	   C   r]   )Nr   r'   Z__nv_float2ull_rdr   Tr   r   r   r   r   r   float2ull_rd  r:   r   c              	   C   r]   )Nr   r'   Z__nv_float2ull_rur   Tr   r   r   r   r   r   float2ull_ru  r:   r   c              	   C   r]   )Nr   r(   Z__nv_double2ll_rnr   Tr   r   r   r   r   r   double2ll_rn  r:   r   c              	   C   r]   )Nr   r(   Z__nv_double2ll_rzr   Tr   r   r   r   r   r   double2ll_rz  r:   r   c              	   C   r]   )Nr   r(   Z__nv_double2ll_rdr   Tr   r   r   r   r   r   double2ll_rd  r:   r   c              	   C   r]   )Nr   r(   Z__nv_double2ll_rur   Tr   r   r   r   r   r   double2ll_ru  r:   r   c              	   C   r]   )Nr   r(   Z__nv_double2ull_rnr   Tr   r   r   r   r   r   double2ull_rn  r:   r   c              	   C   r]   )Nr   r(   Z__nv_double2ull_rzr   Tr   r   r   r   r   r   double2ull_rz  r:   r   c              	   C   r]   )Nr   r(   Z__nv_double2ull_rdr   Tr   r   r   r   r   r   double2ull_rd  r:   r   c              	   C   r]   )Nr   r(   Z__nv_double2ull_rur   Tr   r   r   r   r   r   double2ull_ru  r:   r   c              	   C   r]   )Nr   r   Z__nv_ll2float_rnr'   Tr   r   r   r   r   r   ll2float_rn  r:   r   c              	   C   r]   )Nr   r   Z__nv_ll2float_rzr'   Tr   r   r   r   r   r   ll2float_rz  r:   r   c              	   C   r]   )Nr   r   Z__nv_ll2float_rdr'   Tr   r   r   r   r   r   ll2float_rd  r:   r   c              	   C   r]   )Nr   r   Z__nv_ll2float_rur'   Tr   r   r   r   r   r   ll2float_ru  r:   r   c              	   C   r]   )Nr   r&   Z__nv_ull2float_rnr'   Tr   r   r   r   r   r   ull2float_rn  r:   r   c              	   C   r]   )Nr   r&   Z__nv_ull2float_rzr'   Tr   r   r   r   r   r   ull2float_rz  r:   r   c              	   C   r]   )Nr   r&   Z__nv_ull2float_rdr'   Tr   r   r   r   r   r   ull2float_rd  r:   r   c              	   C   r]   )Nr   r&   Z__nv_ull2float_rur'   Tr   r   r   r   r   r   ull2float_ru  r:   r   c              	   C   r]   )Nr   r   Z__nv_ll2double_rnr(   Tr   r   r   r   r   r   ll2double_rn  r:   r   c              	   C   r]   )Nr   r   Z__nv_ll2double_rzr(   Tr   r   r   r   r   r   ll2double_rz  r:   r   c              	   C   r]   )Nr   r   Z__nv_ll2double_rdr(   Tr   r   r   r   r   r   ll2double_rd  r:   r   c              	   C   r]   )Nr   r   Z__nv_ll2double_rur(   Tr   r   r   r   r   r   ll2double_ru  r:   r   c              	   C   r]   )Nr   r&   Z__nv_ull2double_rnr(   Tr   r   r   r   r   r   ull2double_rn  r:   r   c              	   C   r]   )Nr   r&   Z__nv_ull2double_rzr(   Tr   r   r   r   r   r   ull2double_rz$  r:   r   c              	   C   r]   )Nr   r&   Z__nv_ull2double_rdr(   Tr   r   r   r   r   r   ull2double_rd+  r:   r   c              	   C   r]   )Nr   r&   Z__nv_ull2double_rur(   Tr   r   r   r   r   r   ull2double_ru2  r:   r   c              	   C   r]   )Nr   r   Z__nv_int_as_floatr'   Tr   r   r   r   r   r   int_as_float9  r:   r   c              	   C   r]   )Nr   r'   Z__nv_float_as_intr   Tr   r   r   r   r   r   float_as_int@  r:   r   c              	   C   r]   )Nr   r%   Z__nv_uint_as_floatr'   Tr   r   r   r   r   r   uint_as_floatG  r:   r   c              	   C   r]   )Nr   r'   Z__nv_float_as_uintr   Tr   r   r   r   r   r   float_as_uintN  r:   r   c              	   C   r]   )Nr   r   Z__nv_longlong_as_doubler(   Tr   r   r   r   r   r   longlong_as_doubleU  r:   r   c              	   C   r]   )Nr   r(   Z__nv_double_as_longlongr   Tr   r   r   r   r   r   double_as_longlong\  r:   r   c              	   C   r8   )Nr   r'   Z__nv_fast_sinfTr   r   r   r   r   r   	fast_sinfc  r:   r   c              	   C   r8   )Nr   r'   Z__nv_fast_cosfTr   r   r   r   r   r   	fast_cosfj  r:   r   c              	   C   r8   )Nr   r'   Z__nv_fast_log2fTr   r   r   r   r   r   
fast_log2fq  r:   r   c              	   C   r8   )Nr   r'   Z__nv_fast_logfTr   r   r   r   r   r   	fast_logfx  r:   r   c              	   C   r8   )Nr   r'   Z__nv_fast_expfTr   r   r   r   r   r   	fast_expf  r:   r   c              	   C   r8   )Nr   r'   Z__nv_fast_tanfTr   r   r   r   r   r   	fast_tanf  r:   r   c              	   C   r8   )Nr   r'   Z__nv_fast_exp10fTr   r   r   r   r   r   fast_exp10f  r:   r   c              	   C   r8   )Nr   r'   Z__nv_fast_log10fTr   r   r   r   r   r   fast_log10f  r:   r   c              	   C   rE   )Nr   r'   Z__nv_fast_powfTr   r   r)   r   r   r   	fast_powf  rG   r   c                 C   r/   )Nr   r   Z	__nv_haddr%   Z
__nv_uhaddTr   r   r)   r   r   r   hadd  r1   r   c                 C   r/   )Nr   r   Z
__nv_rhaddr%   Z__nv_urhaddTr   r   r)   r   r   r   rhadd  r1   r   c                 C   r/   )Nr   r'   Z__nv_fsub_rnr(   Z__nv_dsub_rnTr   r   r)   r   r   r   sub_rn  r1   r   c                 C   r/   )Nr   r'   Z__nv_fsub_rzr(   Z__nv_dsub_rzTr   r   r)   r   r   r   sub_rz  r1   r   c                 C   r/   )Nr   r'   Z__nv_fsub_rdr(   Z__nv_dsub_rdTr   r   r)   r   r   r   sub_rd  r1   r   c                 C   r/   )Nr   r'   Z__nv_fsub_rur(   Z__nv_dsub_ruTr   r   r)   r   r   r   sub_ru  r1   r   c              	   C   r8   )Nr   r'   Z__nv_frsqrt_rnTr   r   r   r   r   r   rsqrt_rn  r:   r   c                 C   r   )Nr   r   Z__nv_ffsr   Z
__nv_ffsllTr   r   r   r   r   r   ffs  r   r   c                 C   r2   )Nr   r'   Z
__nv_rintfr(   Z	__nv_rintTr   r   r   r   r   r   rint  r   r   c                 C   F   t jdt | gt dfdt dft dfdt dfid|dS )	Nr   r'   Z__nv_llrintfr   r(   Z__nv_llrintTr   r   r   r   r   r   llrint  r   r   c                 C   r2   )Nr   r'   Z__nv_nearbyintfr(   Z__nv_nearbyintTr   r   r   r   r   r   	nearbyint  r   r   c                 C   r   )	Nr   r'   Z__nv_isnanfr   r(   Z__nv_isnandTr   r   r   r   r   r   isnan  r   r   c                 C   r   )	Nr   r'   Z__nv_signbitfr   r(   Z__nv_signbitdTr   r   r   r   r   r   signbit  r   r   c                 C   r/   )Nr   r'   Z__nv_copysignfr(   Z__nv_copysignTr   r   r)   r   r   r   copysign	  r1   r   c              	   C   r]   )Nr   r'   Z__nv_finitefr   Tr   r   r   r   r   r   finitef  r:   r   c                 C   r   )	Nr   r'   Z__nv_isinffr   r(   Z__nv_isinfdTr   r   r   r   r   r   isinf  r   r   c                 C   r/   )Nr   r'   Z__nv_nextafterfr(   Z__nv_nextafterTr   r   r)   r   r   r   	nextafter   r1   r   c                 C   r2   )Nr   r'   Z	__nv_sinfr(   Z__nv_sinTr   r   r   r   r   r   sin(  r   r   c                 C   r2   )Nr   r'   Z	__nv_cosfr(   Z__nv_cosTr   r   r   r   r   r   cos0  r   r   c                 C   r2   )Nr   r'   Z__nv_sinpifr(   Z
__nv_sinpiTr   r   r   r   r   r   sinpi8  r   r   c                 C   r2   )Nr   r'   Z__nv_cospifr(   Z
__nv_cospiTr   r   r   r   r   r   cospi@  r   r   c                 C   r2   )Nr   r'   Z	__nv_tanfr(   Z__nv_tanTr   r   r   r   r   r   tanH  r   r   c                 C   r2   )Nr   r'   Z
__nv_log2fr(   Z	__nv_log2Tr   r   r   r   r   r   log2P  r   r   c                 C   r2   )Nr   r'   Z	__nv_expfr(   Z__nv_expTr   r   r   r   r   r   expX  r   r   c                 C   r2   )Nr   r'   Z__nv_exp10fr(   Z
__nv_exp10Tr   r   r   r   r   r   exp10`  r   r   c                 C   r2   )Nr   r'   Z
__nv_coshfr(   Z	__nv_coshTr   r   r   r   r   r   coshh  r   r   c                 C   r2   )Nr   r'   Z
__nv_sinhfr(   Z	__nv_sinhTr   r   r   r   r   r   sinhp  r   r   c                 C   r2   )Nr   r'   Z
__nv_tanhfr(   Z	__nv_tanhTr   r   r   r   r   r   tanhx  r   r   c                 C   r/   )Nr   r'   Z__nv_atan2fr(   Z
__nv_atan2Tr   r   r)   r   r   r   atan2  r1   r   c                 C   r2   )Nr   r'   Z
__nv_atanfr(   Z	__nv_atanTr   r   r   r   r   r   atan  r   r   c                 C   r2   )Nr   r'   Z
__nv_asinfr(   Z	__nv_asinTr   r   r   r   r   r   asin  r   r   c                 C   r2   )Nr   r'   Z
__nv_acosfr(   Z	__nv_acosTr   r   r   r   r   r   acos  r   r   c                 C   r2   )Nr   r'   Z	__nv_logfr(   Z__nv_logTr   r   r   r   r   r   log  r   r   c                 C   r2   )Nr   r'   Z__nv_log10fr(   Z
__nv_log10Tr   r   r   r   r   r   log10  r   r   c                 C   r2   )Nr   r'   Z__nv_log1pfr(   Z
__nv_log1pTr   r   r   r   r   r   log1p  r   r   c                 C   r2   )Nr   r'   Z__nv_acoshfr(   Z
__nv_acoshTr   r   r   r   r   r   acosh  r   r   c                 C   r2   )Nr   r'   Z__nv_asinhfr(   Z
__nv_asinhTr   r   r   r   r   r   asinh  r   r   c                 C   r2   )Nr   r'   Z__nv_atanhfr(   Z
__nv_atanhTr   r   r   r   r   r   atanh  r   r   c                 C   r2   )Nr   r'   Z__nv_expm1fr(   Z
__nv_expm1Tr   r   r   r   r   r   expm1  r   r   c                 C   r/   )Nr   r'   Z__nv_hypotfr(   Z
__nv_hypotTr   r   r)   r   r   r   hypot  r1   r   c                 C   r/   )Nr   r'   Z__nv_rhypotfr(   Z__nv_rhypotTr   r   r)   r   r   r   rhypot  r1   r   c                 C   r@   )Nr   r'   Z__nv_norm3dfr(   Z__nv_norm3dTr   r   r    r   r   r   norm3d  r5   r   c                 C   r@   )Nr   r'   Z__nv_rnorm3dfr(   Z__nv_rnorm3dTr   r   r    r   r   r   rnorm3d  r5   r   c                 C   |   t jdt | |||gt dt dt dt dfdt dft dt dt dt dfdt dfid|dS )Nr   r'   Z__nv_norm4dfr(   Z__nv_norm4dTr   r   r   r!   r"   Zarg3r   r   r   r   norm4d     ..r   c                 C   r   )Nr   r'   Z__nv_rnorm4dfr(   Z__nv_rnorm4dTr   r   r   r   r   r   rnorm4d   r   r   c                 C   r2   )Nr   r'   Z
__nv_cbrtfr(   Z	__nv_cbrtTr   r   r   r   r   r   cbrt  r   r   c                 C   r2   )Nr   r'   Z__nv_rcbrtfr(   Z
__nv_rcbrtTr   r   r   r   r   r   rcbrt  r   r   c                 C   r2   )Nr   r'   Z__nv_j0fr(   Z__nv_j0Tr   r   r   r   r   r   j0  r   r   c                 C   r2   )Nr   r'   Z__nv_j1fr(   Z__nv_j1Tr   r   r   r   r   r   j1   r   r   c                 C   r2   )Nr   r'   Z__nv_y0fr(   Z__nv_y0Tr   r   r   r   r   r   y0(  r   r   c                 C   r2   )Nr   r'   Z__nv_y1fr(   Z__nv_y1Tr   r   r   r   r   r   y10  r   r   c                 C   X   t jdt | |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__nv_ynfr(   Z__nv_ynTr   r   r)   r   r   r   yn8  r1   r   c                 C   r   )	Nr   r   r'   Z__nv_jnfr(   Z__nv_jnTr   r   r)   r   r   r   jn@  r1   r   c                 C   r2   )Nr   r'   Z__nv_cyl_bessel_i0fr(   Z__nv_cyl_bessel_i0Tr   r   r   r   r   r   cyl_bessel_i0H  r   r   c                 C   r2   )Nr   r'   Z__nv_cyl_bessel_i1fr(   Z__nv_cyl_bessel_i1Tr   r   r   r   r   r   cyl_bessel_i1P  r   r   c                 C   r2   )Nr   r'   Z	__nv_erffr(   Z__nv_erfTr   r   r   r   r   r   erfX  r   r   c                 C   r2   )Nr   r'   Z__nv_erfinvfr(   Z__nv_erfinvTr   r   r   r   r   r   erfinv`  r   r   c                 C   r2   )Nr   r'   Z
__nv_erfcfr(   Z	__nv_erfcTr   r   r   r   r   r   erfch  r   r   c                 C   r2   )Nr   r'   Z__nv_erfcxfr(   Z
__nv_erfcxTr   r   r   r   r   r   erfcxp  r   r   c                 C   r2   )Nr   r'   Z__nv_erfcinvfr(   Z__nv_erfcinvTr   r   r   r   r   r   erfcinvx  r   r   c                 C   r2   )Nr   r'   Z__nv_normcdfinvfr(   Z__nv_normcdfinvTr   r   r   r   r   r   
normcdfinv  r   r   c                 C   r2   )Nr   r'   Z__nv_normcdffr(   Z__nv_normcdfTr   r   r   r   r   r   normcdf  r   r   c                 C   r2   )Nr   r'   Z__nv_lgammafr(   Z__nv_lgammaTr   r   r   r   r   r   lgamma  r   r   c                 C   X   t jdt | |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__nv_ldexpfr(   Z
__nv_ldexpTr   r   r)   r   r   r   ldexp  r1   r   c                 C   r   )	Nr   r'   r   Z__nv_scalbnfr(   Z__nv_scalbnTr   r   r)   r   r   r   scalbn  r1   r   c                 C   r/   )Nr   r'   Z
__nv_fmodfr(   Z	__nv_fmodTr   r   r)   r   r   r   fmod  r1   r   c                 C   r/   )Nr   r'   Z__nv_remainderfr(   Z__nv_remainderTr   r   r)   r   r   r   	remainder  r1   r   c                 C   r@   )Nr   r'   Z	__nv_fmafr(   Z__nv_fmaTr   r   r    r   r   r   fma  r5   r   c                 C   s   t jdt | |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
__nv_powifr(   Z	__nv_powiZ	__nv_powfZ__nv_powTr   r   r)   r   r   r   pow  r.   r   c                 C   r2   )Nr   r'   Z__nv_tgammafr(   Z__nv_tgammaTr   r   r   r   r   r   tgamma  r   r   c                 C   r2   )Nr   r'   Z__nv_roundfr(   Z
__nv_roundTr   r   r   r   r   r   round  r   r   c                 C   r   )	Nr   r'   Z__nv_llroundfr   r(   Z__nv_llroundTr   r   r   r   r   r   llround  r   r   c                 C   r/   )Nr   r'   Z
__nv_fdimfr(   Z	__nv_fdimTr   r   r)   r   r   r   fdim  r1   r   c                 C   r   )	Nr   r'   Z__nv_ilogbfr   r(   Z
__nv_ilogbTr   r   r   r   r   r   ilogb  r   r   c                 C   r2   )Nr   r'   Z
__nv_logbfr(   Z	__nv_logbTr   r   r   r   r   r   logb  r   r   c              	   C   r]   )Nr   r(   Z__nv_isfinitedr   Tr   r   r   r   r   r   	isfinited  r:   r   )N)	functoolsr   r   r   	lru_cacher   Zexternr   r   r#   r*   r,   r-   r0   r3   r4   r6   r7   r9   r;   r<   r=   r>   r?   rA   rB   rC   rD   rF   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   rR   rS   rT   rU   rV   rW   rX   rY   rZ   r[   r\   r^   r_   r`   ra   rb   rc   rd   re   rf   rg   rh   ri   rj   rk   rl   rm   rn   ro   rp   rq   rr   rs   rt   ru   rv   rw   rx   ry   rz   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   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   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   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   r   r   r   <module>   s&   

			