U
    hl                     @   s8  d Z ddlmZ ddlmZ ddlmZmZmZmZm	Z	m
Z
mZmZmZmZmZmZmZmZmZmZmZmZ ddlmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZYmZZZm[Z[m\Z\m]Z]m^Z^m_Z_m`Z`maZambZbmcZcmdZdmeZe ddlmfZfmgZgmhZhmiZimjZjmkZkmlZlmmZmmnZnmoZompZpmqZqmrZrmsZsmtZtmuZumvZv ddlwmxZxmyZymzZzm{Z{m|Z|m}Z}m~Z~mZmZmZ dd	d
dddddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgdhdidjdkdldmdndodpdqdrdsdtdudvdwdxdydzd{d|d}d~dddddddddgZdd ZdS )zisort:skip_file   )math)extra)argmaxargmincdivcumprodcumsumflip
interleavemaxminravelsigmoidsoftmaxsortsum	swizzle2dxor_sumzeros
zeros_like)NPropagateNanTRITON_MAX_TENSOR_NUMEL_experimental_descriptor_load_experimental_descriptor_storeadvancearangeassociative_scan
atomic_add
atomic_and
atomic_cas
atomic_max
atomic_min	atomic_oratomic_xchg
atomic_xorbfloat16
block_type	broadcastbroadcast_tocatcastclampconstconst_pointer_type	constexprdebug_barrierdevice_assertdevice_printdotdtypeexpand_dimsfloat16float32float64float8e4b15
float8e4nv
float8e4b8float8e5float8e5b16fullfunction_type	histograminline_asm_elementwiseint1int16int32int64int8joinloadmake_block_ptrmax_constancymax_contiguousmaximumminimummultiple_ofnum_programspermutepi32_tpointer_type
program_idrangereducereshapesplitstatic_assertstatic_printstatic_rangestoretensortransuint16uint32uint64uint8viewvoidwhere)umulhiexpexp2fmaloglog2cosrsqrtsinsqrtsqrt_rnabsfdivdiv_rnerffloorceil)
pair_uniform_to_normalphiloxphilox_implrandrand4xrandint	randint4xrandnrandn4xuint_to_uniform_floatr   r   r   r   ro   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   builtinr)   r*   r   rt   r+   r,   r-   r.   rj   r   r   r/   r0   r1   rq   r2   r3   rr   re   rf   r4   r   rp   r	   r5   r6   r7   r8   r9   r:   r;   r<   rs   rg   r=   r>   r?   r@   r
   rA   rB   rC   rD   rE   ZirrF   rG   rh   ri   rH   r   r   rI   rJ   rK   r   rL   rM   rN   ru   rO   rv   rw   rP   rQ   rR   rx   ry   rz   r{   r|   r}   rS   r   rT   rU   rk   r   rl   r   r   rV   rm   rn   rW   rX   rY   rZ   r   r   r[   r\   Ztritonr]   r^   r_   r`   r~   rd   ra   rb   rc   r   r   r   c                 C   s   | d dkrP| dd  } | d dkr@| dd  } t | }t|S t | }t|S ttttttt	t
ttttttttttttd}||  S )N    *r   k)Zfp8e4nvZfp8e4b8Zfp8e5Zfp8e5b16Zfp8e4b15Zfp16Zbf16Zfp32Zfp64i1i8Zi16Zi32Zi64u1u8u16u32Zu64B)	str_to_tyr-   rQ   r9   r:   r;   r<   r8   r5   r%   r6   r7   rA   rE   rB   rC   rD   r`   r]   r^   r_   )nametyZtys r   J/var/www/html/venv/lib/python3.8/site-packages/triton/language/__init__.pyr      s<    r   N)__doc__ r   r   standardr   r   r   r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   corer   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   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`   ra   rb   rc   rd   re   rf   rg   rh   ri   rj   rk   rl   rm   rn   ro   rp   rq   rr   rs   rt   randomru   rv   rw   rx   ry   rz   r{   r|   r}   r~   __all__r   r   r   r   r   <module>   s  P APL0 