diff --git a/debug/accuracy_tools/msprobe/core/common/const.py b/debug/accuracy_tools/msprobe/core/common/const.py index 0c8689ed78248494ecddb3f94e5f643b00c63256..fc610b8e2909e9799a19c99e4e169fb12d181665 100644 --- a/debug/accuracy_tools/msprobe/core/common/const.py +++ b/debug/accuracy_tools/msprobe/core/common/const.py @@ -140,6 +140,7 @@ class Const: "Aten", "VF", "NPU", "Jit"] MODULE_PREFIX = ["Module", "Cell"] FORWARD_NAME_SUFFIX = ".forward" + 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 282ff5946cb5cbaf03bc68bd6b643bf24558af1b..34ca29afb2e1a24eef8d21fd764adfddac6c0523 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 @@ -129,6 +129,8 @@ class BaseDataProcessor: for (_, path, line, func, code, _) in api_stack: if not code: continue + 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) else: