Ë
    ÄÓ^jk!  ã                  óZ  — d dl mZ d dlZd dlZd dlZd dlZd dlmZmZ d dl	m
Z
mZ d dlmZ d dlmZ d dlmZ erd dlmZ d d	lmZ ej,                   G d
„ d«      «       Z G d„ de«      Z G d„ dej2                  «      Z e«       Z G d„ dee   «      ZdZ G d„ dej<                  «      Zy)é    )ÚannotationsN)ÚAnyÚTYPE_CHECKING)ÚoverrideÚSelf)Úpatch)Úconfig)ÚRemoteCacheBackend)ÚCallable)ÚTracebackTypec                  óX   — e Zd ZU dZded<   dZded<   dZded<   dd„Zdd„Zdd„Z	dd	„Z
y
)ÚStatsr   ÚintÚnum_putÚnum_get_hitÚnum_get_missc                óÀ   — | xj                   |j                   z  c_         | xj                  |j                  z  c_        | xj                  |j                  z  c_        | S ©N©r   r   r   ©ÚselfÚothers     úe/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/torch/_inductor/mock_cache.pyÚ__iadd__zStats.__iadd__   sG   € Ø�Š˜Ÿ™Ñ%�Ø×Ò˜E×-Ñ-Ñ-ÕØ×Ò˜U×/Ñ/Ñ/ÕØˆó    c                ó.   — d| _         d| _        d| _        y )Nr   r   ©r   s    r   ÚresetzStats.reset"   s   € ØˆŒØˆÔØˆÕr   c                ó~   — dj                  d| j                  › d�d| j                  › d�d| j                  › d�f«      S )NÚ zputs: z, zmisses: zhits: )Újoinr   r   r   r   s    r   Ú__str__zStats.__str__'   sN   € Ø�w‰wà˜Ÿ™˜ bÐ)Ø˜4×,Ñ,Ð-¨RÐ0Ø˜×)Ñ)Ð*¨"Ð-ðó
ð 	
r   c                óÐ   — t        |t        t        f«      xrO | j                  |j                  k(  xr4 | j                  |j                  k(  xr | j
                  |j
                  k(  S r   )Ú
isinstancer   Ú_GlobalItemStatsr   r   r   r   s     r   Ú__eq__zStats.__eq__0   sb   € ô �uœuÔ&6Ð7Ó8ò 8Ø—‘ §¡Ñ-ò8à× Ñ  E×$5Ñ$5Ñ5ò8ð ×!Ñ! U×%7Ñ%7Ñ7ð		
r   N)r   r   Úreturnr   ©r'   ÚNone)r'   Ústr)r   Úobjectr'   Úbool)Ú__name__Ú
__module__Ú__qualname__r   Ú__annotations__r   r   r   r   r"   r&   © r   r   r   r      s4   … à€GˆSÓØ€K�ÓØ€L�#Óóóó

ô
r   r   c                  ó8   ‡ — e Zd ZU ded<   dˆ fd„Zdˆ fd„Zˆ xZS )r%   zdict[str, object]Úcachec                ó0   •— t         ‰| �  «        i | _        y r   )ÚsuperÚ__init__r3   ©r   Ú	__class__s    €r   r6   z_GlobalItemStats.__init__>   s   ø€ Ü‰ÑÔØˆ�
r   c                ó0   •— t         ‰| �  «        i | _        y r   )r5   r   r3   r7   s    €r   r   z_GlobalItemStats.resetB   s   ø€ Ü‰‰ŒØˆ�
r   r(   )r-   r.   r/   r0   r6   r   Ú__classcell__)r8   s   @r   r%   r%   ;   s   ø… ØÓõ÷ñ r   r%   c                  ó*   — e Zd Zdd„Zdd„Zdd„Zd„ Zy)Ú_GlobalStatsc                óÖ   — t        «       | _        t        «       | _        t        «       | _        t        «       | _        t        «       | _        t        «       | _        t        «       | _        y r   )r%   Úautotune_localÚautotune_remoteÚbundled_autotuneÚfx_graphÚtritonÚaot_autogradÚ
dynamo_pgor   s    r   r6   z_GlobalStats.__init__N   sL   € Ü.Ó0ˆÔÜ/Ó1ˆÔÜ 0Ó 2ˆÔÜ(Ó*ˆŒÜ&Ó(ˆŒÜ,Ó.ˆÔÜ*Ó,ˆ�r   c                óp  — | j                   j                  «        | j                  j                  «        | j                  j                  «        | j                  j                  «        | j
                  j                  «        | j                  j                  «        | j                  j                  «        y r   )r>   r   r?   r@   rA   rB   rC   rD   r   s    r   r   z_GlobalStats.resetW   sz   € Ø×Ñ×!Ñ!Ô#Ø×Ñ×"Ñ"Ô$Ø×Ñ×#Ñ#Ô%Ø�‰×ÑÔØ�‰×ÑÔØ×Ñ×ÑÔ!Ø�‰×ÑÕr   c                ó   — t        | |«      S r   )Úgetattr©r   Únames     r   Úget_statz_GlobalStats.get_stat`   s   € Ü�t˜TÓ"Ð"r   c                óÆ  — d| j                   fd| j                  fd| j                  fd| j                  fd| j                  fd| j
                  fd| j                  ff}t        dt        j                  ¬	«       |D ]&  \  }}t        d
|› d|› �t        j                  ¬	«       Œ( t        dt        j                  ¬	«       |D ]ž  \  }}|j                  sŒt        d
|› d�t        j                  ¬	«       t        |j                  j                  «       «      D ]G  \  }}t        |«      }t        |«      dkD  r|d d dz   }t        d|›d|› �t        j                  ¬	«       ŒI Œ  y )Nr>   r?   r@   rA   rB   rC   rD   zCache Stats:)Úfilez  z: zCache Entries:ú:éd   z...z    )r>   r?   r@   rA   rB   rC   rD   ÚprintÚsysÚstderrr3   ÚsortedÚitemsÚreprÚlen)r   ÚsubsrI   ÚsubÚkÚvs         r   Úreportz_GlobalStats.reportc   sE  € à˜t×2Ñ2Ð3Ø × 4Ñ 4Ð5Ø ×!6Ñ!6Ð7Ø˜Ÿ™Ð'Ø�t—{‘{Ð#Ø˜T×.Ñ.Ð/Ø˜4Ÿ?™?Ð+ð
ˆô 	ˆn¤3§:¡:Õ.Øò 	7‰IˆD�#Ü�B�t�f˜B˜s˜eÐ$¬3¯:©:Ö6ð	7ô 	Ð¤S§Z¡ZÕ0Øò 	>‰IˆD�#Ø�y‹yÜ˜˜4˜& �l¬¯©Õ4Ü" 3§9¡9§?¡?Ó#4Ó5ò >‘D�A�qÜ˜Q›�AÜ˜1“v ’|Ø˜d˜s˜G e™O˜Ü˜D   R¨ sÐ+´#·*±*Ö=ñ	>ñ	>r   Nr(   )rI   r*   r'   r%   )r-   r.   r/   r6   r   rJ   rZ   r1   r   r   r<   r<   M   s   „ ó-ó ó#ó>r   r<   c                  óJ   — e Zd Zdd„Zedd„«       Zedd„«       Zed	d„«       Zy)
ÚMockBackendc                ó   — || _         y r   )Ú_namerH   s     r   r6   zMockBackend.__init__�   s	   € Øˆ�
r   c                ó   ‡ — dˆ fd„}|S )Nc                 ó   •— t        ‰ «      S r   )r\   )rI   s   €r   Úwrapperz&MockBackend.with_name.<locals>.wrapper†   s   ø€ Ü˜tÓ$Ð$r   )r'   r\   r1   )rI   ra   s   ` r   Ú	with_namezMockBackend.with_name„   s   ø€ õ	%ð ˆr   c                óÐ   — t         j                  | j                  «      }||j                  v r*|t	        d¬«      z  }|j                  j                  |«      S |t	        d¬«      z  }y )Né   )r   )r   )Úglobal_statsrJ   r^   r3   r   Úget)r   ÚkeyÚstats      r   Ú_getzMockBackend._get‹   sV   € ä×$Ñ$ T§Z¡ZÓ0ˆØ�$—*‘*ÑØ”E aÔ(Ñ(ˆDØ—:‘:—>‘> #Ó&Ð&à”E qÔ)Ñ)ˆDØr   c                ó~   — t         j                  | j                  «      }|t        d¬«      z  }||j                  |<   y )Nrd   )r   )re   rJ   r^   r   r3   )r   rg   Údatarh   s       r   Ú_putzMockBackend._put•   s2   € ä×$Ñ$ T§Z¡ZÓ0ˆØ”˜aÔ Ñ ˆØˆ�
‰
�3Šr   N)rI   r*   r'   r)   )rI   r*   r'   zCallable[[], MockBackend])rg   r*   r'   z
Any | None)rg   r*   rk   r   r'   r)   )	r-   r.   r/   r6   Ústaticmethodrb   r   ri   rl   r1   r   r   r\   r\   €   sC   „ óð òó ðð òó ðð òó ñr   r\   )Úfx_graph_cacheÚfx_graph_remote_cacheÚautotune_local_cacheÚautotune_remote_cacheÚbundled_autotune_remote_cachec                  óT   — e Zd Zed„ «       Zed„ «       Zdd„Zdd„Z	 	 	 	 	 	 	 	 d	d„Zy)
ÚPatchCachesc                ó¤   — i | _         t        D ]@  }t        t        |«      rt	        t        |«      | j                   |<   t        t        |d«       ŒB y )NF)Ú_savedCacheStateÚ_CACHE_CONFIG_ENÚhasattrr	   rG   Úsetattr©ÚclsrI   s     r   ÚsetUpzPatchCaches.setUp§   sH   € ð
  "ˆÔÜ$ò 	)ˆDÜ”v˜tÔ$Ü-4´V¸TÓ-B�×$Ñ$ TÑ*Ü”F˜D %Õ(ñ	)r   c                ó”   — t         D ]?  }t        t        |«       || j                  v sŒ"t	        t        || j                  |   «       ŒA y r   )rw   Údelattrr	   rv   ry   rz   s     r   ÚtearDownzPatchCaches.tearDown²   sD   € ô %ò 	BˆDÜ”F˜DÔ!Ø�s×+Ñ+Ò+Üœ  c×&:Ñ&:¸4Ñ&@ÕAñ	Br   c                ó6   — t        j                  «       | _        y r   )Ú
contextlibÚ	ExitStackÚ_stackr   s    r   r6   zPatchCaches.__init__º   s   € Ü ×*Ñ*Ó,ˆ�r   c                óü  — t         j                  «        | j                  j                  «        t	        dt
        j                  d«      «      }| j                  j                  |«       t	        dt
        j                  d«      «      }| j                  j                  |«       t	        dt
        j                  d«      «      }| j                  j                  |«       t	        dt
        j                  d«      «      }| j                  j                  |«       t	        d	t
        j                  d
«      «      }| j                  j                  |«       t	        dt
        j                  d«      «      }| j                  j                  |«       t        j                  «       �r\t	        dt
        j                  d«      «      }| j                  j                  |«       t	        dt
        j                  d«      «      }| j                  j                  |«       t	        dt
        j                  d«      «      }| j                  j                  |«       t	        dt
        j                  d«      «      }| j                  j                  |«       t	        dt
        j                  d
«      «      }| j                  j                  |«       t	        dt
        j                  d«      «      }| j                  j                  |«       | S )NzDtorch._inductor.remote_cache.LocalAutotuneCache.backend_override_clsr>   zEtorch._inductor.remote_cache.RemoteAutotuneCache.backend_override_clsr?   zLtorch._inductor.remote_cache.RemoteBundledAutotuneCache.backend_override_clsr@   zDtorch._inductor.remote_cache.RemoteFxGraphCache.backend_override_clsrA   zHtorch._inductor.remote_cache.RemoteAOTAutogradCache.backend_override_clsrC   zFtorch._inductor.remote_cache.RemoteDynamoPGOCache.backend_override_clsrD   zJtorch._inductor.fb.remote_cache.FbRemoteAutotuneCache.backend_override_clszQtorch._inductor.fb.remote_cache.FbRemoteBundledAutotuneCache.backend_override_clszItorch._inductor.fb.remote_cache.FbRemoteFxGraphCache.backend_override_clszFtriton.fb.fb_memcache.FbMemcacheRemoteKernelCache.backend_override_clsrB   zMtorch._inductor.fb.remote_cache.FbRemoteAOTAutogradCache.backend_override_clszKtorch._inductor.fb.remote_cache.FbRemoteDynamoPGOCache.backend_override_cls)
re   r   rƒ   Ú	__enter__r   r\   rb   Úenter_contextr	   Ú	is_fbcode)r   Úctxs     r   r…   zPatchCaches.__enter__½   sh  € Ü×ÑÔØ�‰×ÑÔäØRÜ×!Ñ!Ð"2Ó3ó
ˆð 	�‰×!Ñ! #Ô&äØSÜ×!Ñ!Ð"3Ó4ó
ˆð 	�‰×!Ñ! #Ô&äØZÜ×!Ñ!Ð"4Ó5ó
ˆð 	�‰×!Ñ! #Ô&äØRÜ×!Ñ! *Ó-ó
ˆð 	�‰×!Ñ! #Ô&äØVÜ×!Ñ! .Ó1ó
ˆð 	�‰×!Ñ! #Ô&äØTÜ×!Ñ! ,Ó/ó
ˆð 	�‰×!Ñ! #Ô&ä×ÑÕÜØ\Ü×%Ñ%Ð&7Ó8óˆCð �K‰K×%Ñ% cÔ*äØcÜ×%Ñ%Ð&8Ó9óˆCð �K‰K×%Ñ% cÔ*äØ[Ü×%Ñ% jÓ1óˆCð �K‰K×%Ñ% cÔ*äØXÜ×%Ñ% hÓ/óˆCð �K‰K×%Ñ% cÔ*äØ_Ü×%Ñ% nÓ5óˆCð �K‰K×%Ñ% cÔ*äØ]Ü×%Ñ% lÓ3óˆCð �K‰K×%Ñ% cÔ*àˆr   c                ó>   — | j                   j                  |||«       y r   )rƒ   Ú__exit__)r   Úexc_typeÚ	exc_valueÚ	tracebacks       r   rŠ   zPatchCaches.__exit__  s   € ð 	�‰×Ñ˜X y°)Õ<r   Nr(   )r'   r   )r‹   ztype[BaseException] | NonerŒ   zBaseException | Noner�   zTracebackType | Noner'   r)   )	r-   r.   r/   Úclassmethodr|   r   r6   r…   rŠ   r1   r   r   rt   rt   ¦   sa   „ Øñ)ó ð)ð ñBó ðBó-óMð^=à,ð=ð (ð=ð (ð	=ð
 
ô=r   rt   ) Ú
__future__r   r�   ÚdataclassesrP   Ú	threadingÚtypingr   r   Útyping_extensionsr   r   Úunittest.mockr   Útorch._inductorr	   Útorch._inductor.remote_cacher
   Úcollections.abcr   Útypesr   Ú	dataclassr   r%   Úlocalr<   re   r\   rw   ÚAbstractContextManagerrt   r1   r   r   ú<module>rœ      s¤   ðõ #ã Û Û 
Û ß %ß ,Ý å "Ý ;ñ Ý(Ý#ð ×Ñ÷!
ð !
ó ð!
ôH	�uô 	ô$->�9—?‘?ô ->ñ` ‹~€ôÐ$ SÑ)ô ð:Ð ôl=�*×3Ñ3õ l=r   