ó
    “~iJ  ã                   ó†  • S r SSKJr  SSKJr  SSKJrJrJrJrJ	r	J
r
JrJrJrJrJrJrJrJrJrJrJrJrJrJrJr  SSKJrJrJrJrJrJ r J!r!J"r"J#r#J$r$J%r%J&r&J'r'J(r(J)r)J*r*J+r+J,r,J-r-J.r.J/r/J0r0J1r1J2r2J3r3J4r4J5r5J6r6J7r7J8r8J9r9J:r:J;r;J<r<J=r=J>r>J?r?J@r@JArAJBrBJCrCJDrDJErEJFrFJGrGJHrHJIrIJJrJJKrKJLrLJMrMJNrNJOrOJPrPJQrQJRrRJSrSJTrTJUrUJVrVJWrWJXrXJYrYJZrZJ[r[J\r\J]r]J^r^J_r_J`r`JaraJbrbJcrcJdrdJereJfrfJgrgJhrhJiriJjrjJkrkJlrlJmrmJnrnJoroJprpJqrqJrrrJsrsJtrtJuru  SSKJvrvJwrwJxrxJyryJzrzJ{r{J|r|J}r}J~r~JrJ€r€J�r�J‚r‚JƒrƒJ„r„J…r…J†r†  SSK‡JˆrˆJ‰r‰JŠrŠJ‹r‹JŒrŒJ�r�JŽrŽJ�r�J�r�J‘r‘  SSKJ’r’  / S	Qr“S
 r”g)zisort:skip_fileé   )Úmath)Úextra)ÚargmaxÚargminÚbitonic_mergeÚcdivÚcumprodÚcumsumÚflipÚ
interleaveÚmaxÚminÚravelÚ	reduce_orÚsigmoidÚsoftmaxÚsortÚsumÚ	swizzle2dÚtopkÚ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Ú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-   r.   r   r/   r0   r1   r2   r3   r   r…   r4   r5   r6   r7   r8   r{   r	   r
   r9   r:   r;   r‚   r<   r=   r>   rƒ   rv   rw   r?   r   r�   r   r@   rA   rB   rC   rD   rE   rF   rG   r„   rx   rH   rI   rJ   rK   r   rL   rM   rN   rO   rP   rQ   rR   ry   rz   rS   rT   r   r   rU   rV   rW   r   rX   rY   rZ   r[   r†   r\   r‡   rˆ   r]   r^   r_   r‰   rŠ   r‹   rŒ   r�   rŽ   r`   r   ra   r   rb   r|   rc   r   r}   r   r   rd   r~   r   re   rf   rg   rh   ri   r   r   r�   rj   r   rk   rl   rn   ro   rp   rq   r�   ru   rr   rs   rt   r   r   r   c           	      ó€  • SSK Jn  [        X5      (       aK  [        U 5      R                  R                  SS 5      n[        U  Vs/ s H  n[        XA5      PM     snU5      $ U S   S:X  a+  U SS  n SnU S   S:X  a  U SS  n Sn[        X5      n[        XeS	9$ U R                  S
5      (       GaX  U R                  S5      S   R                  S5      nUR                  SSS9u  p‰U	R                  SSS9u  p©U
R                  S5      R                  S5       Vs/ s H  n[        UR                  5       5      PM     n
nU	R                  S5      n[        U5      n[        US 5      n[        U
5      n[        [         /U-  5      n[        ["        /U-  5      n[%        XŠ5      nU(       aR  SSKJnJnJn  SSKJn  SSKJn  [5        U[7        UUUS95      n[        UU5      (       a  U" UUUU5      $ U" UUUU5      $ [1        UUU5      $ U R                  S5      (       a  [9        U5      $ 0 S[:        _S[<        _S[>        _S[@        _S[B        _S[D        _S[F        _S[H        _S[J        _S[L        _S[N        _S[P        _S [         _S!["        _S"[L        _S#[R        _S$[T        _[V        [X        [L        S%.EnUU    $ s  snf s  snf )&Né    )rl   Ú_fieldsÚ*r   FÚkT)Ú
element_tyr6   Ú
tensordescÚ<Ú>Ú[)ÚmaxsplitÚ]Ú,)ÚNVMMASharedLayoutÚPaddedSharedLayoutÚSwizzledSharedLayout)r    r7   Úfp8e4nvÚfp8e4b8Úfp8e5Úfp8e5b16Úfp8e4b15Úfp16Úbf16Úfp32Úfp64Úi1Úi8Úi16Úi32Úi64Úu1Úu8Úu16)Úu32Úu64ÚB)-Úbuiltinsrl   Ú
isinstanceÚtypeÚ__dict__Úgetrm   Ú	str_to_tyr^   Ú
startswithrd   ÚrstripÚintÚstripÚlstripÚlenrN   rO   r/   Ú+triton.experimental.gluon.language._layoutsrž   rŸ   r    Ú4triton.experimental.gluon.language.nvidia.hopper.tmar    Ú2triton.experimental.gluon.language.amd.gfx1250.tdmÚevalÚdictr8   rD   rE   rF   rG   rC   r@   r.   rA   rB   rL   rP   rM   rq   rn   ro   rp   )ÚnameÚcrl   ÚfieldsÚxr6   ÚtyÚinnerr>   ÚrestÚblock_shapeÚsÚlayoutÚis_gluonÚndimÚ
shape_typeÚstride_typeÚblockrž   rŸ   r    Únvidia_tensor_descriptor_typeÚamd_tensor_descriptor_typeÚtyss                           ÚU/home/mande/repo/quber/.venv/lib/python3.13/site-packages/triton/language/__init__.pyrº   rº     sÿ  € Ýä�$×ÑÜ�d“×$Ñ$×(Ñ(¨°DÓ9ˆÜ±DÓ9²D¨qœ9 Qž?±DÑ9¸6ÓBÐBàˆA�w�#ƒ~Ø�A�BˆxˆØˆØ�‰7�c‹>Ø˜˜�8ˆDØˆEÜ�tÓˆÜ rÑ7Ð7à‡��|×$Ò$Ø—
‘
˜3“ Ñ"×)Ñ)¨#Ó.ˆØ—k‘k #°�kÐ2‰ˆØ ŸJ™J s°Q˜JÐ7ÑˆØ/:×/AÑ/AÀ#Ó/F×/LÑ/LÈSÔ/QÓRÒ/Q¨!”s˜1Ÿ7™7›9–~Ñ/QˆÐRØ—‘˜SÓ!ˆÜ�v“;ˆÜ˜% Ó&ˆÜ�;ÓˆÜ¤ ¨$¡Ó/ˆ
ä ¤5 '¨D¡.Ó2ˆÜ˜5Ó.ˆÞßÑõ EÝÜØÜÐ'8ÐM_Ø*>ñ@óAˆFô ˜&Ð"3×4Ñ4Ù4°U¸JÈÐU[Ó\Ð\á1°%¸À[ÐRXÓYÐYÜ% e¨Z¸ÓEÐEà‡��{×#Ñ#Ü˜aÓ Ð ðØ”:ðà”:ðð 	”ðð 	”Kð	ð
 	”Kðð 	”ðð 	”ðð 	”ðð 	”ðð 	Œdðð 	Œdðð 	Œuðð 	Œuðð 	Œuðð 	Œdðð  	Œeð!ð" 	Œvð#ô$ ÜÜò)€Cð, ˆt‰9Ðùò :ùò Ss   ÁJ6Ä#J;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   Úcorer   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   ru   rv   rw   rx   ry   rz   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Ø   Ú<module>rá      sû   ðÙ õ Ý ÷÷ ÷ ÷ ÷ õ ÷.\÷ \÷ \÷ \÷ \÷ \÷ \÷ \÷ \÷ \÷ \÷ \÷ \÷ \÷ \÷ \÷ \÷ \÷ \÷ \÷ \÷ \÷ \ñ \÷z÷ ÷ ÷ õ ÷÷ ÷ õ òM€ó`Drà   