Ë
    P/æjO  ã                   ó  — d dl Z ddlmZmZmZ ddlmZmZmZ ddl	m
Z
mZm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 dd	lmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$ dd
l%m&Z& ddl'm(Z( ddl)m*Z*m+Z+m,Z,m-Z- ddl.m/Z0 g d¢Z1e jd                  e jf                  jd                  u rUd dl4Z  e jj                  e jd                  «      e _2        e jl                  jn                  jq                  e jd                  d«       dd„Z9dd„Z:y)é    Né   )Úconvert_frameÚ
eval_frameÚresume_execution)Úlist_backendsÚlookup_backendÚregister_backend)Úcallback_handlerÚon_compile_endÚon_compile_start)Úcode_context)Úreplay)Úallow_in_graphÚassume_constant_resultÚdisableÚdisallow_in_graphÚforbid_in_graphÚgraph_breakÚmark_dynamicÚmark_staticÚmark_static_addressÚmaybe_mark_dynamicÚrunÚsubstitute_in_graph)	Ú_reset_guarded_backend_cacheÚexplainÚexportÚis_dynamo_supportedÚis_inductor_supportedÚoptimizeÚoptimize_assertÚOptimizedModuleÚ
reset_code)Úis_compiling)ÚGenerationTracker)Úgraph_break_reasonsÚguard_failuresÚorig_code_mapÚreset_frame_count)Úloader)r   r   r   r   r   r   r   r   r   r   r    r!   r   r   r   r   r   Úresetr"   r$   r	   r   r   zaten::manual_seedc                  óê  — t         j                  5  t        «        t         j                  j	                  «        t         j
                  j	                  «        t        j                  «        t        j                  «        t        j                  «        t        j                  j                  j	                  «        t        «        t        «        t        j                  j                   j"                  j%                  «        dt         _        t         j(                  j	                  «        t+        j                  «        t-        j                  «        t        j                   j.                  j0                  j	                  «        t        j                   j.                  j2                  j	                  «        t        j                  j4                  j7                  d«       ddd«       y# 1 sw Y   yxY w)z2Clear all compile caches and restore initial stater   FN)r   Úcompile_lockÚreset_code_cachesÚinput_codesÚclearÚoutput_codesr(   r'   r&   r   ÚContinueExecutionCacheÚcacher   r)   ÚtorchÚ_CÚ_dynamoÚcompiled_autogradÚclear_cacheÚFRAME_COUNTERÚFRAME_COMPILE_COUNTERr
   r%   ÚutilsÚwarn_once_cacheÚuser_obj_id_to_weakrefÚ	_autogradÚ _saved_tensors_hooks_set_tracing© ó    úY/root/repos/pocs/video-evals/.venv/lib/python3.12/site-packages/torch/_dynamo/__init__.pyr+   r+   N   s4  € ä	×	#Ñ	#ñ CÜÔÜ×!Ñ!×'Ñ'Ô)Ü×"Ñ"×(Ñ(Ô*Ü×ÑÔÜ×ÑÔÜ×!Ñ!Ô#Ü×/Ñ/×5Ñ5×;Ñ;Ô=Ü$Ô&ÜÔÜ�‰×Ñ×*Ñ*×6Ñ6Ô8Ø&'ŒÔ#Ü×+Ñ+×1Ñ1Ô3Ü×ÑÔ Ü×ÑÔ!Ü�‰×Ñ×+Ñ+×1Ñ1Ô3Ü�‰×Ñ×2Ñ2×8Ñ8Ô:Ü�‰×Ñ×;Ñ;¸EÔB÷#C÷ Cñ Cús   ‘GG)Ç)G2c                  ó  — t         j                  5  t         j                  j                  t         j                  j                  z   D ]  }  | «       }|sŒt        |«       Œ t        j                  «        ddd«       y# 1 sw Y   yxY w)z3Clear compile caches that are keyed by code objectsN)r   r-   r/   Úseenr1   r#   r   r0   )Ú	weak_codeÚcodes     rB   r.   r.   d   sn   € ä	×	#Ñ	#ñ ä×%Ñ%×*Ñ*¬]×-GÑ-G×-LÑ-LÑLò	!ˆIñ “;ˆDÚÜ˜4Õ ð	!ô 	×ÑÔ÷÷ ñ ús   ‘AA>Á!A>Á>B)ÚreturnN);r4   Ú r   r   r   Úbackends.registryr   r   r	   Úcallbackr
   r   r   r   r   Ú
decoratorsr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   Úexternal_utilsr$   Úmutation_guardr%   r;   r&   r'   r(   r)   Ú	polyfillsr*   Ú_Ú__all__Úmanual_seedÚrandomÚtorch.jit._builtinsÚ_disable_dynamoÚjitÚ	_builtinsÚ_register_builtinr+   r.   r@   rA   rB   ú<module>rX      sÃ   ðÛ ç 9Ñ 9ß NÑ Nß HÑ HÝ &Ý !÷÷ ÷ ó ÷
÷ 
õ 
õ )Ý -ß XÓ Xõ #ò€ð4 	×Ñ˜Ÿ™×0Ñ0Ñ0Ûð .˜×-Ñ-¨e×.?Ñ.?Ó@€EÔà	‡I�I×Ñ×)Ñ)¨%×*;Ñ*;Ð=PÔQóCô,	rA   