+
    "Ý³j:   ã                   óÌ   € ^ RI t ^ RIt^ RIt^ RIt^ RIt^ RIt]P                  ! RR4      tRR lt]P                  ! 4       t
RsR R ltR tR tR tR	 tR
 tRR ltR tR tR# )é    NÚTORCHAO_AUTOTUNER_DATA_PATHc           	      ó’  € VR9   g   Q h^ RI p V ! 4        VP                  P                  4        V'       d*   VP                  ! \	        R	4      VP                  RR7      pM(VP                  ! \	        R4      VP
                  RR7      pVP                  P                  RR7      p	VP                  P                  RR7      p
V	P                  4        \        ^4       F  pVP                  4        V ! 4        K  	  V
P                  4        VP                  P                  4        V	P                  V
4      ^,          p\        ^\	        W,          4      4      p\        ^\	        W,,          4      4      p\        V4       Uu. uF  q÷P                  P                  RR7      NK   	  p	p\        V4       Uu. uF  q÷P                  P                  RR7      NK   	  p
p\        V4       F
  pV ! 4        K  	  \        V4       FZ  pVe   V F
  pRVn        K  	  VP                  4        WŸ,          P                  4        V ! 4        W¯,          P                  4        K\  	  VP                  P                  4        VP                  ! \        Wš4       UUu. uF  w  ppVP                  V4      NK  	  uppVP                  R7      pVeX   VP                   ! VVP                  ! WGP                  R7      4      P#                  4       p\%        V4      ^8X  d
   V^ ,          pV# \'        Wv4      ! V4      P)                  4       # u upi u upi u uppi )
ÚminNç    €„®AÚcuda©ÚdtypeÚdeviceT©Úenable_timing)r	   )r   ÚmaxÚmeanÚmedianç    €„ŽA)Útorchr   ÚsynchronizeÚemptyÚintÚint8ÚEventÚrecordÚrangeÚzero_Úelapsed_timer   ÚgradÚtensorÚzipÚfloatÚquantileÚtolistÚlenÚgetattrÚitem)ÚfnÚwarmupÚrepÚgrad_to_noneÚ	quantilesÚ
fast_flushÚreturn_moder   ÚcacheÚstart_eventÚ	end_eventÚ_Úestimate_msÚn_warmupÚn_repeatÚiÚxÚsÚeÚtimesÚrets   &&&&&&&              ÚY/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torchao/kernel/autotuner.pyÚdo_bench_tritonr9      s“  € ð Ð:Ô:Ð:Ð:Ûðñ$ „DØ	‡J�J×ÑÔ÷
 Ø—’œC 
›O°5·9±9ÀVÔL‰à—’œC ›J¨e¯j©jÀÔHˆð —*‘*×"Ñ"°Ð"Ó6€KØ—
‘
× Ñ ¨tÐ Ó4€IØ×ÑÔÜ�1ŽXˆØ�‰ŒÙ
Žñ ð ×ÑÔØ	‡J�J×ÑÔØ×*Ñ*¨9Ó5¸Õ9€Kô �1”c˜&Õ.Ó/Ó0€HÜ�1”c˜#Õ+Ó,Ó-€HÜAFÀxÄÓQÁ¸A—:‘:×#Ñ#°$Ð#Ö7Á€KÐQÜ?DÀX¼ÓO¹¸!—‘×!Ñ!°Ð!Ö5¹€IÐOä�8Ž_ˆÙ
Žñ ô �8Ž_ˆð Ò#Û!�Ø�–ñ "ð 	�‰Œà�×ÑÔÙ
ŒØ�×ÑÖñ ð 
‡J�J×ÑÔØ�LŠLÜ'*¨;Ô'BÔCÑ'B™t˜q !ˆ�‰˜Ö	Ñ'BÒCÈ5Ï;É;ô€Eð ÒØ�nŠn˜U E§L¢L°Ç+Á+Ô$NÓO×VÑVÓXˆÜˆs‹8�qŒ=Ø�a•&ˆCØˆ
Ü�5Ô& uÓ-×2Ñ2Ó4Ð4ùò; RùÚOùó* 	Ds   Å?$L9Æ2$L>ÊM
c                ó$   € V ^8„  d   QhR\         /# )é   Úreturn)Údict)Úformats   "r8   Ú__annotate__r?   n   s   € ÷ ñ œ4ñ ó    c                  óœ   € \         ;_uu_ 4        \        f   \        4       p V e   T M/ s\        uuR R R 4       #   + '       g   i     R # ; i©N)Ú_best_configs_lockÚ_best_configsÚ_load_best_configs)Úloadeds    r8   Ú_get_best_configsrG   n   s4   € ç	Ö	ÜÒ Ü'Ó)ˆFØ&,Ò&8™F¸bˆMÜ÷	 
×	×	Ó	ús	   � :ºA	c                 óÈ  € \         P                  P                  4       p\        f'   \        P
                  P                  4       R,          pM\        P
                  ! \        4      p\        P                  ! RV RV 24       \        VR4      ;_uu_ 4       p^ R I
p\        P                  ! RV 24       VP                  W4       R R R 4       R #   + '       g   i     R # ; i)Nzdata.pklzTrying to store configs for z locally under ÚwbzSaving best configs to file )r   r   Úget_device_nameÚAUTOTUNER_DATA_PATHÚpathlibÚPathÚcwdÚloggingÚinfoÚopenÚpickleÚdump)Úbest_configsÚdevice_nameÚsaved_configsÚfrR   s   &    r8   Ú_save_best_configsrX   w   sŸ   € Ü—*‘*×,Ñ,Ó.€KÜÒ"ÜŸ™×(Ñ(Ó*¨ZÕ7‰äŸšÔ%8Ó9ˆÜ‡L‚LØ
& { m°?À=À/ÐRôô 
ˆm˜T×	"Ô	" aÛä�ŠÐ3°M°?ÐCÔDØ�‰�LÔ$÷	 
#×	"×	"Ò	"ús   Â/CÃC!	c                  ót  € \         P                  P                  4       p ^ RIp\        fa   VP
                  P                  R4      pVR,          R,          R,          pV P                  R4      '       g   \        P                  ! R4       M\        P                  ! \        4      p\        P                  ! RV  R	V 24       VP                  4       '       dN   ^ RIp\        VR
4      ;_uu_ 4       p\        P                  ! RV 24       VP                  V4      uuRRR4       # R#   + '       g   i     R# ; i)r   NÚtorchaoÚkernelÚconfigszdata_a100.pklzNVIDIA A100z6Warning! Loaded configurations are optimized for A100!zTrying to load configs for z from ÚrbzLoading best configs from file )r   r   rJ   Ú	importlibrK   Ú	resourcesÚfilesÚ
startswithrO   rP   rL   rM   Úis_filerR   rQ   Úload)rU   r^   rV   rR   rW   s        r8   rE   rE   ‡   sè   € Ü—*‘*×,Ñ,Ó.€KÛäÒ"Ø!×+Ñ+×1Ñ1°)Ó<ˆØ%¨Õ0°9Õ<¸ÕNˆØ×%Ñ% m×4Ò4Ü�LŠLÐQÔRøäŸšÔ%8Ó9ˆÜ‡L‚LÐ.¨{¨m¸6À-ÀÐQÔRØ×Ñ×ÒÛä�- ×&Ô&¨!Ü�LŠLÐ:¸=¸/ÐJÔKØ—;‘;˜q“>÷ 'Ò&ñ ÷ '×&Ð&ús   Ã/*D&Ä&D7	c                 ó–   € \         P                  ! V 4      '       d,   V P                  V P                  4       V P	                  4       3# V 3# rB   )r   Ú	is_tensorr	   ÚsizeÚstride)Úas   &r8   Úget_arg_keyri   ›   s5   € Ü‡‚�q×ÒØ—‘˜Ÿ™› 1§8¡8£:Ð.Ð.Øˆ4€Kr@   c                 óx   € \        \        ;QJ d    . R  V  4       F  NK  	  5R4      # ! R  V  4       4      R4      # )c              3   ó8   "  € T F  p\        V4      x € K  	  R # 5irB   )ri   )Ú.0rh   s   & r8   Ú	<genexpr>Úget_args_key.<locals>.<genexpr>¢   s   é € Ð2©T¨”[ —^�^«Tùs   ‚© )ÚsumÚtuple)Úargss   &r8   Úget_args_keyrs   ¡   s0   € Ü�uŒuÑ2©TÓ2�u°BÓ7Ð7ˆuÑ2©TÓ2Ó2°BÓ7Ð7r@   c                 ó.  € V ! 4        \         P                  P                  4        \         P                  ! \	        R4      \         P                  RR7      pVP                  4        \         P                  P                  RR7      p\         P                  P                  RR7      pVP                  4        \        V4       F
  pV ! 4        K  	  VP                  4        \         P                  P                  4        VP                  V4      V,          pV# )r   r   r   Tr   r   )
r   r   r   r   r   r   r   r   r   r   )r$   r&   r+   r,   r-   r.   r/   s   &&     r8   Údo_bench_basicru   ¥   s¾   € á„DÜ	‡J�J×ÑÔä�KŠKœ˜J›¬u¯y©yÀÔH€EØ	‡K�K„MÜ—*‘*×"Ñ"°Ð"Ó6€KÜ—
‘
× Ñ ¨tÐ Ó4€IØ×ÑÔÜ�3ŽZˆÙ
Žñ à×ÑÔÜ	‡J�J×ÑÔØ×*Ñ*¨9Ó5¸Õ;€KØÐr@   c                 óB  a aa€ VVV 3R  lp \        V^4      pVe   Ve   WS^d,          8”  d   \        R4      # \        V^4      pVe   WS^
,          8”  d   \        R4      # \        V4      #   \         d    Rp Lb\        P                  P                   d    Rp L„i ; i)c                  ó    <€ S! S S.,           !  # rB   ro   )rr   Úconfigr$   s   €€€r8   Ú
wrapped_fnÚdo_bench.<locals>.wrapped_fn¹   s   ø€ Ù�D˜F˜8•OÑ%Ð%r@   NÚinf)ru   ÚRuntimeErrorÚtritonÚruntimeÚOutOfResourcesr   r9   )r$   rr   rx   Ú	best_timery   Útimes   fff&  r8   Údo_benchr‚   ·   s™   ú€ ÷&ðÜ˜j¨!Ó,ˆð
 ‚|˜	Ò-°$ÀS½Ô2HÜ�U‹|Ðô ˜* aÓ(€DØÒ °B­Ô!6Ü�U‹|Ðô ˜:Ó&Ð&øô ô ØŠÜ�>‰>×(Ñ(ô ØŠðús   �A, Á,BÁ;BÂBÂBc                 óD   € \        4       pW9   d   W,          ^ ,          # R# )r   N)rG   )ÚkeyÚ
cache_dicts   & r8   Úget_best_config_by_keyr†   Ñ   s"   € Ü"Ó$€JØ
ÔØ�˜qÕ!Ð!ñ r@   c           
      óR  € \        4       p\        V4      ^ 8X  d   R# \        V4      p\        V4      pVe   V# \        P
                  ! RV R24       V^ ,          p\        WV^ ,          4      p\        P
                  ! RP                  \        \        WFV.4      4      4       ^pVR,           Fe  p\        WW†4      p	\        P
                  ! RP                  VR R\        V4      R 2V	R \        V4      .4      4       W–8  d   T	pTpV^,          pKg  	  \        P
                  ! R	4       \        P
                  ! RP                  \        \        We.4      4      4       \        ;_uu_ 4        WV3W4&   \        V4      p
RRR4       \        X
4       V#   + '       g   i     L; i)
r   Nz2Starting autotune search. No config found for key Ú.Ú :é   NNÚ4dÚ/z6.3fz-- perfetto --)rG   r!   rs   r†   rO   rP   r‚   ÚjoinÚmapÚstrrC   r=   rX   )r$   rr   r\   r…   r„   Úbest_configr€   r2   rx   r�   Úsnapshots   &&&        r8   Úget_best_config_fnr’   ×   sl  € Ü"Ó$€Jä
ˆ7ƒ|�qÔÙä
�tÓ
€CÜ(¨Ó-€KØÒØÐä‡L‚LÐEÀcÀUÈ!ÐLÔMð
 ˜!•*€KÜ˜ 7¨1¥:Ó.€IÜ‡L‚L�—‘œ#œc C°KÐ#@ÓAÓBÔCØ	€Að ˜"—+�+ˆÜ˜ &Ó4ˆÜ�ŠØ�H‰H˜˜2˜˜a¤ G£¨RÐ0Ð1°d¸4°[Ä3ÀvÃ;ÐOÓPô	
ð ÔØˆIØ ˆKØ	ˆQ�Šñ ô ‡L‚LÐ!Ô"Ü‡L‚L�—‘œ#œc IÐ#;Ó<Ó=Ô>÷ 
Ö	Ø&Ð2ˆ
‰Ü˜
Ó#ˆ÷ 
ô �xÔ ØÐ÷	 
×	ús   Å0FÆF&	)é   éd   NNTr   rB   )rO   ÚosrL   Ú	threadingr   r}   ÚgetenvrK   r9   ÚLockrC   rD   rG   rX   rE   ri   rs   ru   r‚   r†   r’   ro   r@   r8   Ú<module>r™      ss   ðó Û 	Û Û ã Û à—i’iÐ =¸tÓDÐ ôT5ðr —^’^Ó%Ð Ø€õò%ò "ò(ò8òô$'ò4"ô(r@   