From a344c5356a4a91f48bbfbdc96cfdc1f608e1931c Mon Sep 17 00:00:00 2001 From: gitee Date: Tue, 25 Mar 2025 11:03:13 +0800 Subject: [PATCH 1/2] delete tool's stack --- debug/accuracy_tools/msprobe/core/common/const.py | 1 + .../msprobe/core/data_dump/data_processor/base.py | 2 ++ 2 files changed, 3 insertions(+) diff --git a/debug/accuracy_tools/msprobe/core/common/const.py b/debug/accuracy_tools/msprobe/core/common/const.py index d1e40f280ef..9e309c5da78 100644 --- a/debug/accuracy_tools/msprobe/core/common/const.py +++ b/debug/accuracy_tools/msprobe/core/common/const.py @@ -142,6 +142,7 @@ class Const: "Aten", "VF", "NPU", "Jit"] MODULE_PREFIX = ["Module", "Cell"] FORWARD_NAME_SUFFIX = ".forward" + MSPROBE_PATHS = ["msprobe/core", "msprobe/pytorch", "msprobe/mindspore"] # struct json param ORIGIN_DATA = "origin_data" diff --git a/debug/accuracy_tools/msprobe/core/data_dump/data_processor/base.py b/debug/accuracy_tools/msprobe/core/data_dump/data_processor/base.py index c2a5efee6c2..d3bd39c7cb4 100644 --- a/debug/accuracy_tools/msprobe/core/data_dump/data_processor/base.py +++ b/debug/accuracy_tools/msprobe/core/data_dump/data_processor/base.py @@ -128,6 +128,8 @@ class BaseDataProcessor: for (_, path, line, func, code, _) in api_stack: if not code: continue + if any(msprobe_path in path for msprobe_path in Const.MSPROBE_PATHS): + continue stack_line = f"File {path}, line {str(line)}, in {func}, \n {code[0].strip()}" stack_str.append(stack_line) else: -- Gitee From b47853d93a43e89732dbecd3be311fbf305c0566 Mon Sep 17 00:00:00 2001 From: gitee Date: Tue, 25 Mar 2025 15:39:23 +0800 Subject: [PATCH 2/2] fix review --- debug/accuracy_tools/msprobe/core/common/const.py | 2 +- .../msprobe/core/data_dump/data_processor/base.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/debug/accuracy_tools/msprobe/core/common/const.py b/debug/accuracy_tools/msprobe/core/common/const.py index 9e309c5da78..4ec1834ba91 100644 --- a/debug/accuracy_tools/msprobe/core/common/const.py +++ b/debug/accuracy_tools/msprobe/core/common/const.py @@ -142,7 +142,7 @@ class Const: "Aten", "VF", "NPU", "Jit"] MODULE_PREFIX = ["Module", "Cell"] FORWARD_NAME_SUFFIX = ".forward" - MSPROBE_PATHS = ["msprobe/core", "msprobe/pytorch", "msprobe/mindspore"] + STACK_FILTER_PATHS = ["msprobe/core", "msprobe/pytorch", "msprobe/mindspore", "torch/nn/modules/module.py"] # struct json param ORIGIN_DATA = "origin_data" diff --git a/debug/accuracy_tools/msprobe/core/data_dump/data_processor/base.py b/debug/accuracy_tools/msprobe/core/data_dump/data_processor/base.py index d3bd39c7cb4..b108d731178 100644 --- a/debug/accuracy_tools/msprobe/core/data_dump/data_processor/base.py +++ b/debug/accuracy_tools/msprobe/core/data_dump/data_processor/base.py @@ -128,7 +128,7 @@ class BaseDataProcessor: for (_, path, line, func, code, _) in api_stack: if not code: continue - if any(msprobe_path in path for msprobe_path in Const.MSPROBE_PATHS): + if any(filter_path in path for filter_path in Const.STACK_FILTER_PATHS): continue stack_line = f"File {path}, line {str(line)}, in {func}, \n {code[0].strip()}" stack_str.append(stack_line) -- Gitee