ȅi[SrSSKrSSKJrJrJrJrJrJr SSK J r J r J r SSK JrJrJr SSKJr SSKJr SS KJrJrJrJrJrJrJrJrJrJrJrJrJ r J!r!J"r"J#r#J$r$J%r%J&r&J'r'J(r( SS KJ)r)J*r*J+r+J,r,J-r-J.r.J/r/J0r0J1r1 SS K2J3r3 SS K4J5r5 SS K6J7r7 SSK8J9r9 SSK:J;r;Jr>J?r?J@r@ SSKAJBrC /SQrD\RR\RR&R/5 \R\RRLaUSSKKr\R"\R5\lI\RRR\RS5 SSjrPSSjrQS\R4SjrSS\RSS4SjrTg)a TorchDynamo is a Python-level JIT compiler designed to make unmodified PyTorch programs faster. TorchDynamo hooks into the frame evaluation API in CPython (PEP 523) to dynamically modify Python bytecode right before it is executed. It rewrites Python bytecode in order to extract sequences of PyTorch operations into an FX Graph which is then just-in-time compiled with a customizable backend. It creates this FX Graph through bytecode analysis and is designed to mix Python execution with compiled backends to get the best of both worlds: usability and performance. This allows it to seamlessly optimize PyTorch programs, including those using modern Python features. N) aot_compileconfig convert_frame eval_framefunctional_exportresume_execution) list_backendslookup_backendregister_backend)callback_handleron_compile_endon_compile_start) code_context)replay)allow_in_graphassume_constant_resultdisabledisable_nested_graph_breaksdisallow_in_graphdont_skip_tracingerror_on_graph_breakforbid_in_graph graph_breakis_dynamo_disable_recursive mark_dynamic mark_staticmark_static_addressmaybe_mark_dynamicnonstrict_tracepatch_dynamo_configrun set_stance skip_framestep_unsupportedsubstitute_in_graph) _reset_guarded_backend_cacheexplainexportis_dynamo_supportedis_inductor_supportedoptimizeoptimize_assertOptimizedModule reset_code) is_compiling)GenerationTracker)reset_code_state)TensorifyState)graph_break_reasonsguard_failures orig_code_map(register_hook_for_recompile_user_contextreset_frame_countreset_recompile_user_contexts)loader)#rrrrrrrr)r(rrr0rr r rrrrr r,r-r.r!r rresetr9r"rset_recursion_limitr#r$r%r&zaten::manual_seedreturncxSSKnUR[5nURS5 [R  [ 5 [RR5 [5 [RR5 [R"5 [R"5 [R"5 [RR R5 [#5 [%5 [&R(R*R-5 S[l[R0R5 [2R"5 [4R"5 [6R"5 [&R(R8R:R5 [&R<R>RAS5 [&RBRE5(a SSK#J$n U"5 SSS5 g!,(df  g=f)a Clear all compile caches and restore initial state. This function is intended to reset Dynamo's state *as if* you had started a fresh process invocation, which makes it good for testing scenarios where you want to behave as if you started a new process. It does NOT affect any file system caches. NB: this does NOT reset logging state. Don't use this to test logging initialization/reinitialization. rNztorch._dynamo.resetF)reset_cudagraph_trees)%logging getLogger__name__infor compile_lockreset_code_caches input_codesclearr2 output_codesr6r5r4r ContinueExecutionCachecacher'r8torch_dynamocompiled_autogradr; FRAME_COUNTERFRAME_COMPILE_COUNTERr r1r3utilswarn_once_cache_C _autograd _saved_tensors_hooks_set_tracingcuda is_availabletorch._inductor.cudagraph_treesr?)r@logr?s P/home/james-whalen/.local/lib/python3.13/site-packages/torch/_dynamo/__init__.pyr;r;sW   H %CHH "#  # #!!'')""((*!!#//55;;=$& ''--/&' #++113 ! ++113 ;;EB :: " " $ $ M ! #3 $ # #s G'H++ H9cSSKnUR[5nURS5 [R  [ 5 [RR[RR-HnU"5nU(dM[U5 M [R"5 SSS5 g!,(df  g=f)z Clears in-memory code cache, which is what stores compiled products. This resets less state than :func:`reset` and is mostly only used for testing purposes. rNztorch._dynamo.reset_code_caches) r@rArBrCrrDr2rFseenrHr/rrG)r@rX weak_codecodes rYrErEs   H %CHH ./=  # #  % % * *]-G-G-L-L L ;Dt4 M  $ # #sAB<$B<< C cf[RRRR 5$)z Returns the internal dynamo recursion limit set by `torch._dynamo.set_recursion_limit`. Returns -1 if no c recursion limit has been set. )rKrRrLrget_c_recursion_limitrYget_recursion_limitrbs# 88   & & < < >>ralimitcj[RRRR U5 g)ao Sets an internal dynamo recursion limit. The limit must be >= 1, or -1 to reset to the default (unset) state. This is possibly needed in Python 3.12-3.13 since there is a separate C recursion limit that is not visible at the Python level. If you are getting RecursionErrors during Dynamo compilation and `sys.setrecursionlimit()` doesn't help, this function may alleviate the issue. NOTE: this function does NOT call `sys.setrecursionlimit()` - the user is expected to manually call this if required. This is because the 2 recursion limits are not sync'd up - e.g. in Python 3.12, functions can be inline-evaluated, which apparently doesn't use up the C stack. WARNING: increasing the recursion limit to an arbitrary large value may cause segfaults due to stack overflows! You can try also try to manually increase the stack size, e.g. with `$ ulimit -s ...` N)rKrRrLrset_c_recursion_limit)rcs rYr<r<s"$ HH55er}s* ONHH&!.   )-!,#$ N$$emm&>&>&H&H%IJ 000--e.?.?@E II))%*;*;=PQ($V.?S?=s=t=ra