From 9af08a2a3e9e3689975e6323e703de889cb391fb Mon Sep 17 00:00:00 2001 From: yangxinxian <947098055@qq.com> Date: Fri, 4 Jul 2025 17:32:27 +0800 Subject: [PATCH] Update script_wrapper.py --- .../msprobe/pytorch/hook_module/script_wrapper.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/debug/accuracy_tools/msprobe/pytorch/hook_module/script_wrapper.py b/debug/accuracy_tools/msprobe/pytorch/hook_module/script_wrapper.py index 7d4c3176dbf..136f0c99745 100644 --- a/debug/accuracy_tools/msprobe/pytorch/hook_module/script_wrapper.py +++ b/debug/accuracy_tools/msprobe/pytorch/hook_module/script_wrapper.py @@ -63,7 +63,8 @@ def wrap_compile_script_func(): hooks) # type: ignore[attr-defined] return _wrapped - torch._dynamo.convert_frame = _patched_convert_frame + import torch._dynamo.convert_frame as _cf_mod + _cf_mod.convert_frame = _patched_convert_frame def wrap_script_func(): -- Gitee