U zh< @stddlZddlmZmZmZddlmZmZmZddl m Z m Z m Z ddl m Z ddlmZddlmZmZmZmZmZmZmZmZmZmZmZdd lmZmZmZmZmZm Z m!Z!m"Z"m#Z#dd l$m%Z%dd l&m'Z'dd l(m)Z)m*Z*m+Z+m,Z,d ddddddddddddddddddd d!d"gZ-ej.ej/j.krTddl0Ze1ej.e_.ej2j34ej.d#dd$d%dZ5dd$d&d'Z6dS)(N) convert_frame eval_frameresume_execution) list_backendslookup_backendregister_backend)callback_handleron_compile_endon_compile_start) code_context)replay) allow_in_graphassume_constant_resultdisabledisallow_in_graphforbid_in_graph graph_break mark_dynamic mark_staticmark_static_addressmaybe_mark_dynamicrun) _reset_guarded_backend_cacheexplainexportis_dynamo_supportedis_inductor_supportedoptimizeoptimize_assertOptimizedModule reset_code) is_compiling)GenerationTracker)graph_break_reasonsguard_failures orig_code_mapreset_frame_countrrrrrrrrrrrrrrr rresetr r"rrrzaten::manual_seed)returnc Cstjttjtjtttt j j t t tjjjdt_tjtttjjjW5QRXdS)z2Clear all compile caches and restore initial staterN)r compile_lockreset_code_caches input_codesclear output_codesr&r%r$rZContinueExecutionCachecacherr'torchZ_CZ_dynamoZcompiled_autograd clear_cacheZ FRAME_COUNTERZFRAME_COMPILE_COUNTERr r#utilsZwarn_once_cacher3r3H/var/www/html/venv/lib/python3.8/site-packages/torch/_dynamo/__init__.pyr(Fs     c CsFtj6tjjtjjD]}|}|rt|qtW5QRXdS)z3Clear compile caches that are keyed by code objectsN)rr*r,seenr.r!r r-)Z weak_codecoder3r3r4r+Zs r+)7r0rrrZbackends.registryrrrcallbackr r r r r Z decoratorsrrrrrrrrrrrrrrrrrrr r!Zexternal_utilsr"Zmutation_guardr#r2r$r%r&r'__all__Z manual_seedrandomZtorch.jit._builtinsZ_disable_dynamoZjitZ _builtinsZ_register_builtinr(r+r3r3r3r4sN  4 ,