o i:1@sddlmZddlmZddlmZddlmZddlmZmZmZmZm Z m Z m Z ddl Z ddl Z ddlZddlZddlZddlZddlZddlZejGdd d eZe jd d Ze jd d Ze jdddZe jddZdS))patched_boxed_run)patched_lazy_format_graph_code)patched_load_by_key_path)patched__exec_with_source)ListTupleDictUnionCallableOptionalAnyNc@s<eZdZUeed<eed<ejeddZ e ed<ddZ dS) DebuggableHook dump_src_dir log_bytecodeF)default_factoryinitoptimized_code_and_modulec Cst}ddl} |j}|jj}|jj|jj d}|dkr$|dkr$nq |j d}|j|ks1J|j ||j gddlm}zddl}|jdd d d d d d d }|j|jd |d} ddlm} ddlm} | | | j||| d} | j} Wdn1swY|jr | | lddl}tt||t| ddWdn1swYtt||t| ddWdn1swYtt|| t| ddWdn1swYWdn 1swYddlm}ddl m!}|"| ||| <| WS|t#fyd}z2ddl$m%}|}ddl&}t'd|d|j&||dt'd|dt()|*t+,WYd}~dSd}~ww)NrT_compilezconvert_frame.pyframe)DecompilationError._<> z__transformed_code_%s_for_z.py) lock_on_file) Decompiler)code_to_decompilereference_codefilepath_templatez.original_bytecodewbz.transformed_bytecodez&.decompiled_and_compiled_back_bytecode) orig_code_map) output_codes)StringIOzEThere is a problem when decompiling and compiling the following code:)filezGPlease consider submitting an issue to https://github.com/thuml/depyf .)-sys _getframeosf_backf_codeco_name co_filenamesplitpathsepf_localsrappend f_globalsdepyf.decompilerrreplacejoinrdepyf.explain.utilsrrdecompile_and_compile_likerdill contextlibsuppress Exceptiondumpopentorch._dynamo.utilsr#torch._dynamo.convert_framer$add SyntaxErrorior%disprintwarningswarngetvalue traceback print_exc)selfcodenew_coderr) code_name file_namer func_namer!rr!decompiled_and_compiled_back_codefilenamer9r#r$er% string_iorDrU[/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/depyf/explain/enable_debugging.py__call__sp  &              zDebuggableHook.__call__N) __name__ __module__ __qualname__str__annotations__bool dataclassesfieldlistrrrWrUrUrUrVrs  rc cs`t||d}|durt|||zdVW|dur"t|||dSdS|dur/t|||ww)N)getattrsetattr)parentnamevalue old_valuerUrUrVpatchTs   rgccs8ddl}|jj|}z dVW|dS|w)Nr)torch_dynamo convert_frameregister_bytecode_hookremove)hookrhhandlerUrUrVenable_bytecode_hook`s roTFc#st|ts tdddl}ddl}tjd}t t d|dt ddddl m }|j|s7|||j|}dd lm}||d <|jjj|d <|jjjj|d <|jjjj|d <d|d<t||}t|jjdt t|jjjdt!t|jj"j#dt$j%t&|zdVWdd|j'D} |j'D]p\} j()drt*fdd| Drqj()drt*fdd| Drqddl+m,} ddl-m.} ddl/m0} m1}|}|stjd}t t d|ddt dd| | }|j2|dj(d }| ||}q|3|D]4}|4|jj5d!}|r(|)d"s.|6d#rGz |7|j2||Wqt8yFYqwqd$|d<ndd|j'D} |j'D]x\} j()drrt*fdd| DrrqYj()drt*fdd| DrqYddl+m,} ddl-m.} ddl/m0} m1}|}|stjd}t t d|ddt dd| | }|j2|dj(d }| ||}qY|3|D]4}|4|jj5d!}|r|)d"s|6d#r z |7|j2||Wqt8y Yqwqd$|d<wWdn 1swYWdn 1s,wYWdn1ssz prepare_debug.. resume_in_c3 |] }d|jvVqdS)rNr|r}rdrLrUrV z prepare_debug..torch_dynamo_resume_in_c3r)rNr|rrrUrVrr)dump_src)write_code_to_file_template) innermost_fn_debug_get_cache_entry_listz: Code object z is compiled but does not have any compiled cache entries. Probably some torch.nn.Module instances are destroyed too early. It is recommended to make sure the torch.nn.Module instances exist after `with depyf.prepare_debug`.full_code_for_z_%s.pyr fx_graph_codez.lockF)9 isinstancer[ RuntimeErrorr)rhinspect currentframef_linenorF warn_explicit__file__ UserWarning depyf.utilsrrr/existsabspathglobal_variablesrtfx graph_modulery _inductor codecache PyCodeCacherzri eval_frameOptimizeContextrWrrgrrutilslazy_format_graph_coderr{rorr, startswithany depyf.explainrr7rtorch._dynamo.eval_framerrr6listdirr.r0endswithrlOSError)rclean_wild_fx_coderr)rhcurrent_line_numberrrrt bytecode_hook code_namesmodulerrrrentriesfull_srcr!full_code_pathr&rdrUrrV prepare_debugjs 2      ""  "   &&  "  Trc csddlm}|drtd|d}ddl}|jjjd}t|j j j d t j -zd |d }t|tdVW|jjj|n |jjj|wWddS1sYwYdS) a A context manager to debug the compiled code. Essentially, it sets a breakpoint to pause the program and allows you to check the full source code in files with prefix ``full_code_for_`` in the ``dump_src_dir`` argument of :func:`depyf.prepare_debug`, and set breakpoints in their separate ``__transformed_code_`` files according to the function name. Then continue your debugging. rrsrxz:You cannot use `depyf.debug` inside `depyf.prepare_debug`.rrNFr{z`depyf` places a breakpoint here to pause the program. You can check the full source code in files with prefix `full_code_for_` in zq first, and set breakpoints in their separate files according to the function name. Then continue your debugging.)rrtrrh_Crirset_eval_framergr Interpreter boxed_runrr{rE breakpoint)rtrrhcallbackmsgrUrUrVdebugs  &"r)TF)rrrrtypingrrr r r r r r:rFrIr^ itertoolsr'r)r dataclassobjectrcontextmanagerrgrorrrUrUrUrVs.   $A   y