o [Û·i¹ ã@sFddlZddlmZmZdejdejfdd„Zdadd„Zd d „ZdS) éN)Ú decompileÚDecompilationErrorÚcodeÚnew_codec Cs²ddl}|j dd¡}ddl}| |j¡rWz;zt|ƒ}| d¡| |¡Wnty@}z| dt |ƒ¡WYd}~n d}~wwW| d¡dSW| d¡dS| d¡wdS)Nrztorch._dynamo.convert_frameÚbytecodezpossible 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Ú_loggingÚgetArtifactLoggerÚloggingÚ isEnabledForÚDEBUGrÚdebugrÚstr)rrrÚ bytecode_logr Údecompiled_srcÚe©rúS/home/ubuntu/vllm_env/lib/python3.10/site-packages/depyf/explain/enhance_logging.pyÚpytorch_bytecode_src_hooks2ÿ  €ÿÿÿþ ÿÿørcCs&ddl}tdur dS|jj t¡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_frameÚregister_bytecode_hookr)rrrrÚinstalls /ÿrcCstdurdSt ¡dadS)zf Uninstall the bytecode hook for PyTorch. Should be called after :func:`depyf.install()`. N)rÚremoverrrrÚ uninstallUsr) ÚtypesÚdepyf.decompilerrrÚCodeTyperrrrrrrrÚs  7