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