diff --git a/debug/accuracy_tools/msprobe/core/common/const.py b/debug/accuracy_tools/msprobe/core/common/const.py index d1e40f280eff36b43008513f5f1df0c06fee672e..4ec1834ba913741993bed58c9d087dac6a7de29f 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" + 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 c2a5efee6c20aad0e403c9d0ed09b62f651db919..b108d731178702a15fcc547a355ea670f5de6ade 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(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: