
    ɯwgO                        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     [/home/mcse/projects/flask/flask-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	II))%*;*;=PQC,	rA   