diff --git a/debug/accuracy_tools/msprobe/mindspore/compare/ms_compare.py b/debug/accuracy_tools/msprobe/mindspore/compare/ms_compare.py
index e76ac16af4dd04dc7c3b1af06d5b6f44ffd5174c..5a0d7a97fc69079ca666ee2226bc298b75c4bebd 100644
--- a/debug/accuracy_tools/msprobe/mindspore/compare/ms_compare.py
+++ b/debug/accuracy_tools/msprobe/mindspore/compare/ms_compare.py
@@ -313,11 +313,10 @@ def generate_file_mapping(npu_json_path, bench_json_path, mapping_list):
 
 
 def check_cross_framework(bench_json_path):
-    pattern = r'"data_name":\s*"[^"]+\.pt"'
-    with FileOpen(bench_json_path, 'r') as file:
-        for line in file:
-            if re.search(pattern, line):
-                return True
+    json_data = load_json(bench_json_path)
+    for key, value in json_data.items():
+        if key == 'data_name' and value.endswith(Const.PT_SUFFIX):
+            return True
     return False