§
    OŠtj¦  ã                   ó”  — 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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/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`maZambZbmcZcmdZdmeZemfZfmgZgmhZhmiZimjZjmkZkmlZlmmZmmnZnmoZompZpmqZqmrZrmsZsmtZtmuZumvZvmwZwmxZx ddlmyZymzZzm{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” ddlm•Z• g d	¢Z–d
„ Z—dS )zisort:skip_fileé   )Úmath)Úextra)ÚargmaxÚargminÚbitonic_mergeÚcdivÚcumprodÚcumsumÚflipÚ
interleaveÚmaxÚminÚravelÚ	reduce_orÚsigmoidÚsoftmaxÚsortÚsqueezeÚsumÚ	swizzle2dÚtopkÚ	unsqueezeÚxor_sumÚzerosÚ
zeros_like)\ÚPropagateNanÚTRITON_MAX_TENSOR_NUMELÚload_tensor_descriptorÚstore_tensor_descriptorÚmake_tensor_descriptorÚtensor_descriptorÚtensor_descriptor_typeÚaddÚadvanceÚarangeÚassociative_scanÚassumeÚ
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Ú	conditionÚconstÚ	constexprÚconstexpr_typeÚdebug_barrierÚdevice_assertÚdevice_printÚdotÚ
dot_scaledÚdtypeÚexpand_dimsÚfloat16Úfloat32Úfloat64Úfloat8e4b15Ú
float8e4nvÚ
float8e4b8Úfloat8e5Úfloat8e5b16ÚfullÚgatherÚ	histogramÚinline_asm_elementwiseÚint1Úint16Úint32Úint64Úint8ÚjoinÚloadÚmake_block_ptrÚmap_elementwiseÚmax_constancyÚmax_contiguousÚmaximumÚminimumÚmulÚmultiple_ofÚnum_programsÚpermuteÚpi32_tÚpointer_typeÚ
program_idÚrangeÚreduceÚreshapeÚsliceÚsplitÚstatic_assertÚstatic_printÚstatic_rangeÚstoreÚsubÚtensorÚ	to_tensorÚtransÚtupleÚ
tuple_typeÚ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_float)Útarget_info)�r   r   r   r   r    r!   rƒ   r#   r$   r%   r   r   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r   r1   r2   r3   r4   r5   r   rˆ   r6   r7   r8   r9   r:   r~   r	   r
   r;   r<   r=   r…   r>   r?   r@   r†   ry   rz   rA   r   r„   r   rB   rC   rD   rE   rF   rG   rH   rI   r‡   r{   rJ   rK   rL   rM   r   rN   rO   rP   rQ   rR   rS   rT   r|   r}   rU   rV   r   r   rW   rX   rY   r   rZ   r[   r\   r]   r‰   r^   rŠ   r‹   r_   r`   ra   rŒ   r�   rŽ   r�   r�   r‘   rb   r   rc   r   rd   r   re   r   r€   r   r   rf   r�   r‚   r   rg   rh   ri   rj   rk   r   r   r“   rl   r   rm   rn   ro   rq   rr   rs   rt   r’   rx   r   ru   rv   rw   r   r   r   c                 ó4  ‡— ddl m} t          | |¦  «        rDt          | ¦  «        j                             dd ¦  «        }t          ˆfd„| D ¦   «         |¦  «        S | d         dk    rE| dd …         } d}| d         dk    r| dd …         } d	}t          | ‰¦  «        }t          ||¬
¦  «        S |  	                    d¦  «        �rŒ|  
                    d¦  «        d                              d¦  «        }| 
                    dd¬¦  «        \  }}| 
                    dd¬¦  «        \  }	}d„ |	                     d¦  «         
                    d¦  «        D ¦   «         }	|                     d¦  «        }
t          |
¦  «        }t          |d ¦  «        }t          |	¦  «        }t          t          g|z  ¦  «        }t          t          g|z  ¦  «        }t!          ||	¦  «        }|rbddlm}m}m} ddlm} ddlm} t1          |
t3          |||¬¦  «        ¦  «        }
t          |
|¦  «        r |||||
¦  «        S  |||||
¦  «        S t-          |||¦  «        S |  	                    d¦  «        rt5          ‰¦  «        S i dt6          “dt8          “dt:          “dt<          “dt>          “dt@          “dtB          “dtD          “dtF          “dtH          “d tJ          “d!tL          “d"t          “d#t          “d$tH          “d%tN          “d&tP          “tR          tT          tH          d'œ¥}||          S )(Né    )ro   Ú_fieldsc                 ó0   •— g | ]}t          |‰¦  «        ‘ŒS © )Ú	str_to_ty)Ú.0ÚxÚcs     €úV/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/triton/language/__init__.pyú
<listcomp>zstr_to_ty.<locals>.<listcomp>%  s!   ø€ Ð9Ð9Ð9¨q�9 Q¨™?œ?Ð9Ð9Ð9ó    Ú*r   FÚkT)Ú
element_tyr8   Ú
tensordescú<ú>ú[)Úmaxsplitú]c                 óP   — g | ]#}t          |                     ¦   «         ¦  «        ‘Œ$S r˜   )ÚintÚstrip)rš   Úss     r�   rž   zstr_to_ty.<locals>.<listcomp>4  s&   € ÐRÐRÐR¨!•s˜1Ÿ7š7™9œ9‘~”~ÐRÐRÐRrŸ   ú,)ÚNVMMASharedLayoutÚPaddedSharedLayoutÚSwizzledSharedLayout)r"   r9   Úfp8e4nvÚfp8e4b8Úfp8e5Úfp8e5b16Úfp8e4b15Úfp16Úbf16Úfp32Úfp64Úi1Úi8Úi16Úi32Úi64Úu1Úu8Úu16)Úu32Úu64ÚB)+Úbuiltinsro   Ú
isinstanceÚtypeÚ__dict__Úgetrp   r™   r`   Ú
startswithrf   ÚrstripÚlstripÚlenrP   rQ   r1   Ú+triton.experimental.gluon.language._layoutsr®   r¯   r°   Ú4triton.experimental.gluon.language.nvidia.hopper.tmar"   Ú2triton.experimental.gluon.language.amd.gfx1250.tdmÚevalÚdictr:   rF   rG   rH   rI   rE   rB   r0   rC   rD   rN   rR   rO   rt   rq   rr   rs   )Únamerœ   ro   Úfieldsr8   ÚtyÚinnerr@   ÚrestÚblock_shapeÚlayoutÚis_gluonÚndimÚ
shape_typeÚstride_typeÚblockr®   r¯   r°   Únvidia_tensor_descriptor_typeÚamd_tensor_descriptor_typeÚtyss    `                    r�   r™   r™      sÏ  ø€ ØÐÐÐÐÐå�$˜ÑÔð CÝ�d‘”Ô$×(Ò(¨°DÑ9Ô9ˆÝÐ9Ð9Ð9Ð9°DÐ9Ñ9Ô9¸6ÑBÔBÐBàˆA„w�#‚~€~Ø�A�B�BŒxˆØˆØ�Œ7�cŠ>ˆ>Ø˜˜˜”8ˆDØˆEÝ�t˜QÑÔˆÝ r°Ð7Ñ7Ô7Ð7à‡‚�|Ñ$Ô$ñ FØ—
’
˜3‘” Ô"×)Ò)¨#Ñ.Ô.ˆØ—k’k #°�kÑ2Ô2‰ˆˆtØ ŸJšJ s°Q˜JÑ7Ô7Ñˆ�TØRÐR¨{×/AÒ/AÀ#Ñ/FÔ/F×/LÒ/LÈSÑ/QÔ/QÐRÑRÔRˆØ—’˜SÑ!Ô!ˆÝ�v‘;”;ˆÝ˜% Ñ&Ô&ˆÝ�;ÑÔˆÝ¥ ¨$¡Ñ/Ô/ˆ
å ¥5 '¨D¡.Ñ2Ô2ˆÝ˜5 +Ñ.Ô.ˆØð 	ZØÐÐÐÐÐÐÐÐÐð Eð  Eð  Eð  Eð  Eð  EØÐÐÐÐÐÝØÝÐ'8ÐM_Ø*>ð@ñ @ô @ñAô AˆFõ ˜&Ð"3Ñ4Ô4ð ZØ4Ð4°U¸JÈÐU[Ñ\Ô\Ð\à1Ð1°%¸À[ÐRXÑYÔYÐYÝ% e¨Z¸ÑEÔEÐEà‡‚�{Ñ#Ô#ð !Ý˜aÑ Ô Ð ðØ•:ðà•:ðð 	•ðð 	•Kð	ð
 	•Kðð 	•ðð 	•ðð 	•ðð 	•ðð 	�dðð 	�dðð 	�uðð 	�uðð 	�uðð 	�dðð  	�eð!ð" 	�vð#õ$ ÝÝð)ð ð €Cð, ˆtŒ9ÐrŸ   N)˜Ú__doc__Ú r   r   Ústandardr   r   r   r   r	   r
   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/   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   ru   rv   rw   rx   ry   rz   r{   r|   r}   r~   r   r€   r�   r‚   rƒ   r„   r…   r†   r‡   rˆ   Úrandomr‰   rŠ   r‹   rŒ   r�   rŽ   r�   r�   r‘   r’   r“   Ú__all__r™   r˜   rŸ   r�   ú<module>rè      s  ðØ Ð ð Ð Ð Ð Ð Ð Ø Ð Ð Ð Ð Ð ðð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð2]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð ]ð|ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ðð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð Ð Ð Ð Ð Ð ðPð Pð P€ðfDð Dð Dð Dð DrŸ   