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)Zcode_to_decompileZreference_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_globalsZdepyf.decompilerrreplacejoinrdepyf.explain.utilsrrZdecompile_and_compile_likerdill contextlibsuppress ExceptiondumpopenZtorch._dynamo.utilsr!Ztorch._dynamo.convert_framer"add SyntaxErrorior#disprintwarningswarngetvalue traceback print_exc)selfcodeZnew_coderr'Z code_name file_namer func_namerrrZ!decompiled_and_compiled_back_codefilenamer5r!r"er#Z string_ior>rKj/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/depyf/explain/enable_debugging.py__call__sp  &              zDebuggableHook.__call__N) __name__ __module__ __qualname__str__annotations__bool dataclassesfieldlistrrrMrKrKrKrLrs  rc cs`t||d}|durt|||zdVW|dur"t|||dSdS|dur/t|||ww)N)getattrsetattr)parentnamevalue old_valuerKrKrLpatchTs   r]ccs8ddl}|jj|}z dVW|dS|w)Nr)torch_dynamoZ convert_frameZregister_bytecode_hookremove)hookr^handlerKrKrLenable_bytecode_hook`s rcTFc#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)rqNrmrnrZrFrKrL z prepare_debug..torch_dynamo_resume_in_c3rr)rwNrmrsrtrKrLrurv)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`.Zfull_code_for_z_%s.pyrZ fx_graph_codez.lockF)9 isinstancerQ RuntimeErrorr'r^inspect currentframef_linenor@ warn_explicit__file__ UserWarningZ depyf.utilsrfr-existsabspathglobal_variablesrhfxZ graph_modulerjZ _inductorZ codecacheZ PyCodeCacherkr_ eval_frameZOptimizeContextrMrr]rrutilsZlazy_format_graph_coderrlrcrr* startswithanyZ depyf.explainrxr4ryZtorch._dynamo.eval_framerzr{r3listdirr,r.endswithr`OSError)rZclean_wild_fx_coderr'r^Zcurrent_line_numberrfrhZ bytecode_hookZ code_namesmodulerxryrzr{entriesZfull_srcrZfull_code_pathr$rZrKrtrL 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. rrgriz:You cannot use `depyf.debug` inside `depyf.prepare_debug`.rrNFrlz`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.)rrhr}r^Z_Cr_rZset_eval_framer]rZ InterpreterZ boxed_runrrlr? breakpoint)rhrr^callbackmsgrKrKrLdebugs  &"r)TF)rrrrtypingrrr r r r r r6r@rCrT itertoolsr%r'r~ dataclassobjectrcontextmanagerr]rcrrrKrKrKrLs.   $A   y