`i RddlZddlmZmZdejdejfdZdadZdZdS)N) decompileDecompilationErrorcodenew_codecddl}|jdd}ddl}||jr t |}|d||n:#t$r-}|dt|Yd}~nd}~wwxYw|ddS#|dwxYwdS)Nrztorch._dynamo.convert_framebytecodezpossible source code:zDecompilation fails due to: %sziIf you find the decompiled code is wrong,please submit an issue at https://github.com/thuml/depyf/issues.) torch_logginggetArtifactLoggerlogging isEnabledForDEBUGrdebugrstr)rrr bytecode_logr decompiled_srces q/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/depyf/explain/enhance_logging.pypytorch_bytecode_src_hookrs/LLL>33%zLNNN  //  &x00N   6 7 7 7   ~ . . . .! I I I   ?Q H H H H H H H H I   9     L  9      s/9A98C 9 B0#B+&C +B00C C!chddl}tdS|jjt adS)a Install the bytecode hook for PyTorch, integrate into PyTorch's logging system. Example: .. code-block:: python import torch import depyf depyf.install() # anything with torch.compile @torch.compile def f(a, b): return a + b f(torch.tensor(1), torch.tensor(2)) Turn on bytecode log by ``export TORCH_LOGS="+bytecode"``, and execute the script. We will see the decompiled source code in the log: .. code-block:: text ORIGINAL BYTECODE f test.py line 5 7 0 LOAD_FAST 0 (a) 2 LOAD_FAST 1 (b) 4 BINARY_ADD 6 RETURN_VALUE MODIFIED BYTECODE f test.py line 5 5 0 LOAD_GLOBAL 0 (__compiled_fn_1) 2 LOAD_FAST 0 (a) 4 LOAD_FAST 1 (b) 6 CALL_FUNCTION 2 8 UNPACK_SEQUENCE 1 10 RETURN_VALUE possible source code: def f(a, b): __temp_2, = __compiled_fn_1(a, b) return __temp_2 If you find the decompiled code is wrong,please submit an issue at https://github.com/thuml/depyf/issues. To uninstall the hook, use :func:`depyf.uninstall()`. rN)r _handle_dynamo convert_frameregister_bytecode_hookr)r s rinstallrs:^LLLm)@@!##GGGcNtdStdadS)zf Uninstall the bytecode hook for PyTorch. Should be called after :func:`depyf.install()`. N)rremoverr uninstallr Us'  NNGGGr) typesdepyf.decompilerrrCodeTyperrrr rrrr$s| ::::::::ENen, 4#4#4#n     r