+
    ‹°³j|  ã                   óØ  € R t ^RIHt ^RIHt ^RIHtHtHtHtH	t	H
t
HtHtHtHtHtHtHtHtHtHtHtHtHtHtHtHtHt ^RIHtHtHtH t H!t!H"t"H#t#H$t$H%t%H&t&H't'H(t(H)t)H*t*H+t+H,t,H-t-H.t.H/t/H0t0H1t1H2t2H3t3H4t4H5t5H6t6H7t7H8t8H9t9H:t:H;t;H<t<H=t=H>t>H?t?H@t@HAtAHBtBHCtCHDtDHEtEHFtFHGtGHHtHHItIHJtJHKtKHLtLHMtMHNtNHOtOHPtPHQtQHRtRHStSHTtTHUtUHVtVHWtWHXtXHYtYHZtZH[t[H\t\H]t]H^t^H_t_H`t`HataHbtbHctcHdtdHeteHftfHgtgHhthHitiHjtjHktkHltlHmtmHntnHotoHptpHqtqHrtrHstsHtttHutuHvtvHwtwHxtxHytyHztz ^RIH{t{H|t|H}t}H~t~HtH€t€H�t�H‚t‚HƒtƒH„t„H…t…H†t†H‡t‡HˆtˆH‰t‰HŠtŠH‹t‹ ^RIŒH�t�HŽtŽH�t�H�t�H‘t‘H’t’H“t“H”t”H•t•H–t– ^RIH—t— . RNR	NR
NRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNR NR!NR"NR#NR$NR%NR&NR'NR(NR)NR*NR+NR,NR-NR.NR/NR0NR1NR2NR3NR4NR5NR6NR7NR8NR9NR:NR;NR<NR=NR>NR?NR@NRANRBNRCNRDNRENRFNRGNRHNRINRJNRKNRLNRMNRNNRONRPNRQNRRNRSNRTNRUNRVNRWNRXNRYNRZNR[NR\NR]NR^NR_NR`NRaNRbNRcNRdNReNRfNRgNRhNRiNRjNRkNRlNRmNRnNRoNRpNRqNRrNRsNRtNRuNRvNRwNRxNRyNRzNR{NR|NR}NR~NRNR€NR�NR‚NRƒNR„NR…NR†NR‡NRˆNR‰NRŠNR‹NRŒNR�NRŽNR�NR�NR‘NR’NR“NR”NR•NR–NR—Nt˜R˜ t™R™# )š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Úaggregate_replaceÚ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Úexpect_zeroÚ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_infor   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‡   rz   r{   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   ru   r“   ry   r   rv   rw   rx   r   r   r   c           	      ó  € ^ RI Hp \        W4      '       dK   \        V 4      P                  P                  RR4      p\        V  Uu. uF  p\        WA4      NK  	  upV4      # V ^ ,          R8X  d<   V R,          p RpV ^ ,          R8X  d   V R,          p Rp\        W4      p\        WeR	7      # V P                  R
4      '       Ed   V P                  R4      pV P                  R4      ^,          P                  R4      pVP                  R^R7      w  ršV
P                  R^R7      w  rºVP                  R4      P                  R4       Uu. uF  p\        VP                  4       4      NK  	  ppRp^ RIpVP                  RV
4      pV'       dE   \        VP!                  ^4      4      pV
RVP#                  4        W¯P%                  4       R ,           p
V
P'                  R4      p\)        V4      p\        V	R4      p	V'       d   Ve   TM
\)        V4      p\        \*        .V,          4      p\        \,        .V,          4      p\/        W›4      pV'       dp   ^ RIHpHpHp ^ RIHp ^ RIHp ^ RIHp \A        V\C        VVVR7      4      p\        VV4      '       d   V'       d   V! VVVV4      # V! VVVV4      # V! VVVV4      # \;        VVV4      # V P                  R4      '       d   \E        V4      # / R\F        bR\H        bR\J        bR\L        bR\N        bR\P        bR\R        bR\T        bR\V        bR \X        bR!\Z        bR"\\        bR#\*        bR$\,        bR%\X        bR&\^        bR'\`        bR(\b        R)\d        R*\X        /CpVV ,          # u upi u upi )+é    )ro   Ú_fieldsNÚ*:é   NNFÚkT)Ú
element_tyr8   Ú
tensordescÚtensordesc_im2colÚ<Ú>Ú[)ÚmaxsplitÚ]Ú,z,input_rank=(\d+))ÚNVMMASharedLayoutÚPaddedSharedLayoutÚSwizzledSharedLayout)r"   )Útensor_descriptor_im2col_typer9   Úfp8e4nvÚfp8e4b8Úfp8e5Úfp8e5b16Úfp8e4b15Úfp16Úbf16Úfp32Úfp64Úi1Úi8Úi16Úi32Úi64Úu1Úu8Úu16Úu32Úu64ÚB)3Úbuiltinsro   Ú
isinstanceÚtypeÚ__dict__Úgetrp   Ú	str_to_tyr`   Ú
startswithrf   ÚrstripÚintÚstripÚreÚsearchÚgroupÚstartÚendÚlstripÚlenrP   rQ   r1   Ú+triton.experimental.gluon.language._layoutsr¤   r¥   r¦   Ú4triton.experimental.gluon.language.nvidia.hopper.tmar"   r§   Ú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   )ÚnameÚcro   ÚfieldsÚxr8   ÚtyÚ	is_im2colÚinnerr@   ÚrestÚblock_shapeÚsÚtensor_rankÚ_reÚ
rank_matchÚ
layout_strÚis_gluonÚndimÚ
shape_typeÚstride_typeÚblockr¤   r¥   r¦   Únvidia_tensor_descriptor_typeÚ$nvidia_tensor_descriptor_im2col_typeÚamd_tensor_descriptor_typeÚlayoutÚtyss   &&                            ÚY/home/luca/screen-checking/.venv/lib/python3.14/site-packages/triton/language/__init__.pyrÁ   rÁ   #  sŽ  € Ýä�$×ÒÜ�d“×$Ñ$×(Ñ(¨°DÓ9ˆÜ±DÓ9±D¨qœ9 Qž?±DÑ9¸6ÓBÐBàˆA…w�#„~Ø�B�xˆØˆØ��7�cŒ>Ø˜•8ˆDØˆEÜ�tÓˆÜ rÔ7Ð7à‡��|×$Ó$à—O‘OÐ$7Ó8ˆ	à—
‘
˜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àˆÛØ—Z‘ZÐ 4°dÓ;ˆ
ßÜ˜j×.Ñ.¨qÓ1Ó2ˆKØÐ+˜×)Ñ)Ó+Ð,¨t·N±NÓ4DÐ4EÐ/FÕFˆDØ—[‘[ Ó%ˆ
Ü�z“?ˆÜ˜% Ó&ˆç(¨[Ò-D‰{Ì3È{ÓK[ˆÜ¤ ¨$¥Ó/ˆ
ä ¤5 '¨D¥.Ó2ˆÜ˜5Ó.ˆßßÑõ Eõ SÝÜØÜÐ'8ÐM_Ø*>ô@óAˆFô ˜&Ð"3×4Ò4ßÙ?ÀÀzÐS^Ð`fÓgÐgÙ4°U¸JÈÐU[Ó\Ð\á1°%¸À[ÐRXÓYÐYÜ% e¨Z¸ÓEÐEà‡��{×#Ò#Ü˜aÓ Ð ðØ”:ðà”:ðð 	”ðð 	”Kð	ð
 	”Kðð 	”ðð 	”ðð 	”ðð 	”ðð 	Œdðð 	Œdðð 	Œuðð 	Œuðð 	Œuðð 	Œdðð  	Œeð!ð" 	Œvð#ð$ 	ŒvØŒvØŒTñ)€Cð, ˆt�9Ðùò[ :ùò$ Ss   ÁNÅ"N
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.   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ˆ   r‰   ÚrandomrŠ   r‹   rŒ   r�   rŽ   r�   r�   r‘   r’   r“   r”   Ú__all__rÁ   © ó    rê   Ú<module>ró      s  ðÙ õ Ý ÷÷ ÷ ÷ ÷ ÷ ñ ÷2_÷ _÷ _÷ _÷ _÷ _÷ _÷ _÷ _÷ _÷ _÷ _÷ _÷ _÷ _÷ _÷ _÷ _÷ _÷ _÷ _÷ _÷ _÷ _÷@÷ ÷ ÷ õ ÷÷ ÷ õ ðQØðQàðQð ðQð ð	Qð
 ðQð ðQð 
ðQð 
ðQð ðQð ðQð ðQð ðQð ðQð ðQð ðQð  ð!Qð" ð#Qð$ ð%Qð& ð'Qð( ð)Qð* ð+Qð, ð-Qð. ð/Qð0 ð1Qð2 ð3Qð4 ð5Qð6 ð7Qð8 
ð9Qð: ð;Qð< ð=Qð> ð?Qð@ ðAQðB ðCQðD ðEQðF ðGQðH ðIQðJ 
ðKQðL ðMQðN ðOQðP ðQQðR ðSQðT ðUQðV ðWQðX 
ðYQðZ ð[Qð\ ð]Qð^ 
ð_Qð` 
ðaQðb ðcQðd ðeQðf ðgQðh ðiQðj ðkQðl ðmQðn ðoQðp ðqQðr ðsQðt ðuQðv ðwQðx ðyQðz ð{Qð| ð}Qð~ 
ðQð@ ðAQðB ðCQðD ðEQðF ðGQðH ðIQðJ ðKQðL ðMQðN ðOQðP ðQQðR ðSQðT ðUQðV ðWQðX 
ðYQðZ ð[Qð\ ð]Qð^ ð_Qð` ðaQðb 
ðcQðd ðeQðf ðgQðh ðiQðj 
ðkQðl ðmQðn 
ðoQðp ðqQðr ðsQðt ðuQðv ðwQðx ðyQðz ð{Qð| ð}Qð~ ðQð@ ðAQðB ðCQðD ðEQðF ðGQðH ðIQðJ ðKQðL ðMQðN ðOQðP ðQQðR ðSQðT ðUQðV ðWQðX ðYQðZ ð[Qð\ ð]Qð^ 
ð_Qð` ðaQðb ðcQðd ðeQðf ðgQðh ðiQðj ðkQðl ðmQðn ðoQðp ðqQðr ðsQðt 
ðuQðv 
ðwQðx ðyQðz ð{Qð| ð}Qð~ ðQð@ ðAQðB ðCQðD ðEQðF ðGQðH ðIQðJ ðKQðL ðMQðN ðOQðP ðQQðR ðSQðT ðUQðV ðWQðX ðYQðZ ð[Qð\ ð]Qð^ ð_Qð` ðaQ€ôhRrò   