mindstudio-probe 8.3.3__py3-none-any.whl → 26.0.0a1__py3-none-any.whl

msprobe/mindspore/common/const.py

@@ -1,20 +1,19 @@
-# Copyright (c) 2024-2025, Huawei Technologies Co., Ltd.
-# All rights reserved.
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
 #
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
 #
-# http://www.apache.org/licenses/LICENSE-2.0
+#          http://license.coscl.org.cn/MulanPSL2
 #
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-import numpy as np
-import mindspore as ms
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
+
 from mindspore import dtype as mstype
 
 from msprobe.core.common.const import Const as CoreConst
@@ -84,6 +83,8 @@ class Const:
         mstype.int64, mstype.intp, mstype.uint64, mstype.uintp
     )
 
+    FP8_TYPE_LIST = ["Float8E5M2", "Float8E4M3FN", "HiFloat8"]
+
 
 class MsCompareConst:
     # api_info field
@@ -139,63 +140,3 @@ class MsCompareConst:
         SUCCESS = "success"
         API_NOT_FOUND = "api_not_found"
         EXCEPTION_SKIP = "exception_skip"
-
-
-class FreeBenchmarkConst:
-    ADD_NOISE = "add_noise"
-    BIT_NOISE = "bit_noise"
-    NO_CHANGE = "no_change"
-    EXCHANGE_VALUE = "change_value"
-    IMPROVE_PRECISION = "improve_precision"
-    CHECK = "check"
-    FIX = "fix"
-    DEFAULT_DEVICE = "npu"
-    DEFAULT_STAGE = CoreConst.FORWARD
-    DEFAULT_DUMP_LEVEL = "L1"
-    DEFAULT_PERT_TYPE = IMPROVE_PRECISION
-    DEFAULT_HANDLER_TYPE = CHECK
-    DEVICE_LIST = [DEFAULT_DEVICE]
-    STAGE_LIST = [CoreConst.FORWARD, CoreConst.BACKWARD]
-    DUMP_LEVEL_LIST = [DEFAULT_DUMP_LEVEL]
-    PERT_TYPE_LIST = [IMPROVE_PRECISION, ADD_NOISE, BIT_NOISE, NO_CHANGE, EXCHANGE_VALUE]
-    HANDLER_TYPE_LIST = [CHECK, FIX]
-    NO_CHANGE_ERROR_THRESHOLD = 1.0
-    SYMBOL_FLIPPING_RATIO = 8.0
-
-    SUPPORTED_CHECK_API_FILE = "support_wrap_ops.yaml"
-    CHECK_RESULT_FILE = "free_benchmark.csv"
-
-    API_PREFIX_DICT = {
-        "ops": Const.OPS_PREFIX,
-        "Tensor": Const.TENSOR_PREFIX,
-        "mint": Const.MINT_PREFIX,
-        "mint.nn.functional": Const.MINT_NN_FUNC_PREFIX
-    }
-
-    PERT_VALUE_DICT = {
-        ms.bfloat16: 1e-4,
-        ms.float16: 1e-6,
-        ms.float32: 1e-8,
-        ms.float64: 1e-16
-    }
-
-    ERROR_THRESHOLD = {
-        ms.bfloat16: 1.004,
-        ms.float16: 1.002,
-        ms.float32: 1.0002
-    }
-
-    PERT_BIT_DICT = {
-        ms.float16: np.int16,
-        ms.float32: np.int32,
-        ms.float64: np.int64
-    }
-
-    MS_NUMPY_DTYPE_DICT = {
-        ms.int16: np.int16,
-        ms.int32: np.int32,
-        ms.int64: np.int64,
-        ms.float16: np.float16,
-        ms.float32: np.float32,
-        ms.float64: np.float64
-    }

msprobe/mindspore/common/log.py

@@ -1,17 +1,18 @@
-# Copyright (c) 2024-2024, Huawei Technologies Co., Ltd.
-# All rights reserved.
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
 #
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
 #
-# http://www.apache.org/licenses/LICENSE-2.0
+#          http://license.coscl.org.cn/MulanPSL2
 #
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
 
 from msprobe.core.common.exceptions import DistributedNotInitializedError
 from msprobe.core.common.log import BaseLogger

msprobe/mindspore/common/utils.py

@@ -1,17 +1,18 @@
-# Copyright (c) 2024-2025, Huawei Technologies Co., Ltd.
-# All rights reserved.
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
 #
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
 #
-# http://www.apache.org/licenses/LICENSE-2.0
+#          http://license.coscl.org.cn/MulanPSL2
 #
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
 
 import inspect
 import os
@@ -123,7 +124,7 @@ def list_lowest_level_directories(root_dir):
             full_path = os.path.join(current_dir, entry)
             if os.path.isdir(full_path):
                 if any(os.path.isdir(os.path.join(full_path, subentry)) for subentry in os.listdir(full_path)):
-                    recurse_dirs(full_path, depth=depth+1)
+                    recurse_dirs(full_path, depth=depth + 1)
                 else:
                     lowest_level_dirs.append(full_path)
 
@@ -200,9 +201,11 @@ def set_register_backward_hook_functions():
 
     if is_mindtorch():
         import torch
-        from msprobe.mindspore.mindtorch import (_call_impl,
-                                                 register_full_backward_pre_hook,
-                                                 register_full_backward_hook)
+        from msprobe.mindspore.dump.mindtorch import (
+            _call_impl,
+            register_full_backward_pre_hook,
+            register_full_backward_hook
+        )
         if not hasattr(torch.nn.Module, "register_full_backward_hook"):
             setattr(torch.nn.Module, "_call_impl", _call_impl)
             setattr(torch.nn.Module, "register_full_backward_pre_hook", register_full_backward_pre_hook)
@@ -370,6 +373,18 @@ def wrap_backward_hook_call_func(call_func):
                 if isinstance(item, ms.Tensor):
                     executor.set_creation_type(item, CreationType.DEFAULT)
         return outputs
+
     new_call.__name__ = '__call__'
 
     return new_call
+
+
+def cast_to_float_if_fp8(tensor):
+    dtype = str(tensor.dtype)
+    if dtype in MsConst.FP8_TYPE_LIST:
+        logger.debug(
+            f"The {dtype} tensor analyzing/saving is unsupported in dump function."
+            f"Casting to a float for processing"
+        )
+        tensor = tensor.to(ms.float32)
+    return tensor

msprobe/mindspore/compare/common_dir_compare.py

@@ -1,17 +1,18 @@
-# Copyright (c) 2025-2025, Huawei Technologies Co., Ltd.
-# All rights reserved.
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
 #
-# Licensed under the Apache License, Version 2.0  (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
 #
-# http://www.apache.org/licenses/LICENSE-2.0
+#          http://license.coscl.org.cn/MulanPSL2
 #
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.import functools
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
 
 import os
 import multiprocessing
@@ -30,25 +31,22 @@ from msprobe.core.common.utils import CompareException
 from msprobe.core.common.exceptions import FileCheckException
 from msprobe.core.common.file_utils import check_file_or_directory_path, write_df_to_csv, create_directory, \
                                            check_path_before_create, load_npy
-from msprobe.core.common.const import CompareConst
+from msprobe.core.common.const import Const, CompareConst
 from msprobe.core.compare.npy_compare import compare_ops_apply
+from msprobe.core.compare.utils import check_input_param_path_and_framework
 from msprobe.core.compare.multiprocessing_compute import check_accuracy
 from msprobe.mindspore.compare.utils import check_name_map_dict
 
 
-def common_dir_compare(input_params: Dict, output_dir: str) -> Optional[pd.DataFrame]:
+def common_dir_compare(args) -> Optional[pd.DataFrame]:
     """
     高级目录比对函数，完全镜像输入目录结构
-    
-    Args:
-        input_params: 包含npu_path和bench_path的字典
-        output_dir: 输出根目录
-        
-    Returns:
-        当输入目录是平铺npy文件时返回DataFrame，否则返回None
     """
-    npu_root = Path(input_params.get('npu_path'))
-    bench_root = Path(input_params.get('bench_path'))
+
+    check_input_param_path_and_framework(args, target_framework=Const.MS_FRAMEWORK)
+
+    npu_root = Path(args.target_path)
+    bench_root = Path(args.golden_path)
     name_map_dict = input_params.get('map_dict', {})
     check_name_map_dict(name_map_dict)
     file_tree = build_mirror_file_tree(npu_root, bench_root)
@@ -57,7 +55,7 @@ def common_dir_compare(input_params: Dict, output_dir: str) -> Optional[pd.DataF
     with ProcessPoolExecutor() as executor:
         results = list(tqdm(
             executor.map(
-                partial(process_directory_pair, name_map_dict=name_map_dict, output_dir=output_dir),
+                partial(process_directory_pair, name_map_dict=name_map_dict, output_dir=args.output_path),
                 file_tree.items()
             ),
             total=len(file_tree),

msprobe/mindspore/compare/distributed_compare.py

@@ -1,36 +1,38 @@
-# Copyright (c) 2024-2024, Huawei Technologies Co., Ltd.
-# All rights reserved.
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
 #
-# Licensed under the Apache License, Version 2.0  (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
 #
-# http://www.apache.org/licenses/LICENSE-2.0
+#          http://license.coscl.org.cn/MulanPSL2
 #
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
 
 from msprobe.core.common.utils import CompareException
 from msprobe.core.common.file_utils import create_directory
 from msprobe.core.common.exceptions import FileCheckException
 from msprobe.mindspore.common.log import logger
-from msprobe.mindspore.compare.ms_compare import ms_compare
 from msprobe.core.compare.utils import compare_distributed_inner
 from msprobe.mindspore.compare.ms_graph_compare import GraphMSComparator
+from msprobe.core.compare.utils import print_compare_ends_info
 
 
 def ms_compare_distributed(npu_dump_dir, bench_dump_dir, output_path, **kwargs):
-    compare_distributed_inner(npu_dump_dir, bench_dump_dir, output_path, ms_compare, **kwargs)
+    compare_distributed_inner(npu_dump_dir, bench_dump_dir, output_path, **kwargs)
 
 
-def ms_graph_compare(inputs, outputs):
+def ms_graph_compare(args):
     try:
-        create_directory(outputs)
+        create_directory(args.output_path)
     except (CompareException, FileCheckException) as error:
         logger.error('Compare failed. Please check the arguments and do it again!')
         return
-    ms_comparator = GraphMSComparator(inputs, outputs)
+    ms_comparator = GraphMSComparator(args)
     ms_comparator.compare_core()
+    print_compare_ends_info()

msprobe/mindspore/compare/ms_compare.py

@@ -1,17 +1,19 @@
-# Copyright (c) 2024-2025, Huawei Technologies Co., Ltd.
-# All rights reserved.
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
 #
-# Licensed under the Apache License, Version 2.0  (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
 #
-# http://www.apache.org/licenses/LICENSE-2.0
+#          http://license.coscl.org.cn/MulanPSL2
 #
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
+
 
 from msprobe.core.compare.acc_compare import Comparator, ModeConfig, MappingConfig, setup_comparison
 from msprobe.core.compare.layer_mapping import generate_data_mapping_by_layer_mapping
@@ -34,13 +36,11 @@ def ms_compare(input_param, output_path, **kwargs):
     if config.layer_mapping:
         config.data_mapping = generate_data_mapping_by_layer_mapping(input_param, config.layer_mapping, output_path)
 
-    is_cross_framework = check_cross_framework(input_param.get('bench_json_path'))
+    is_cross_framework = check_cross_framework(input_param.get('bench_path'))
 
     config_dict = {
         'stack_mode': config.stack_mode,
-        'auto_analyze': config.auto_analyze,
         'fuzzy_match': config.fuzzy_match,
-        'highlight': config.highlight,
         'dump_mode': config.dump_mode,
         'compared_file_type': config.compared_file_type
     }

msprobe/mindspore/compare/ms_graph_compare.py

@@ -1,17 +1,18 @@
-# Copyright (c) 2024-2024, Huawei Technologies Co., Ltd.
-# All rights reserved.
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
 #
-# Licensed under the Apache License, Version 2.0  (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
 #
-# http://www.apache.org/licenses/LICENSE-2.0
+#          http://license.coscl.org.cn/MulanPSL2
 #
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
 
 import copy
 import glob
@@ -167,15 +168,20 @@ def transform_special_string_into_float(data_frame):
 
 
 class GraphMSComparator:
-    def __init__(self, input_param, output_path):
-        self.output_path = output_path
-        self.base_npu_path = input_param.get('npu_path', None)
-        self.base_bench_path = input_param.get('bench_path', None)
-        rank_id_list = input_param.get('rank_id', [])
-        step_id_list = input_param.get('step_id', [])
-        if not isinstance(rank_id_list, list) or not isinstance(step_id_list, list):
-            logger.error("'rank_id' and 'step_id' should both be lists, please check!")
-            raise CompareException(CompareException.INVALID_OBJECT_TYPE_ERROR)
+    def __init__(self, args):
+        self.output_path = args.output_path
+        self.base_npu_path = args.target_path
+        self.base_bench_path = args.golden_path
+        if args.rank:
+            rank_id_list = args.rank.split(',')
+            rank_id_list = [item.strip() for item in rank_id_list]
+        else:
+            rank_id_list = []
+        if args.step:
+            step_id_list = args.step.split(',')
+            step_id_list = [item.strip() for item in step_id_list]
+        else:
+            step_id_list = []
         self.rank_list = [convert_to_int(rank_id) for rank_id in rank_id_list]
         self.step_list = [convert_to_int(step_id) for step_id in step_id_list]
         # split by rank and step, generate rank step path

msprobe/mindspore/compare/utils.py

@@ -1,17 +1,18 @@
-# Copyright (c) 2025-2025, Huawei Technologies Co., Ltd.
-# All rights reserved.
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
 #
-# Licensed under the Apache License, Version 2.0  (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
 #
-# http://www.apache.org/licenses/LICENSE-2.0
+#          http://license.coscl.org.cn/MulanPSL2
 #
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
 
 import os
 
@@ -26,7 +27,8 @@ def read_npy_data(dir_path, file_name):
         return None
 
     data_path = os.path.join(dir_path, file_name)
-    path_checker = FileChecker(data_path, FileCheckConst.FILE, FileCheckConst.READ_ABLE, FileCheckConst.NUMPY_SUFFIX)
+    path_checker = FileChecker(data_path, FileCheckConst.FILE, FileCheckConst.READ_ABLE,
+                               FileCheckConst.NUMPY_SUFFIX)
     data_path = path_checker.common_check()
     data_value = load_npy(data_path)
     return data_value

msprobe/mindspore/{cell_processor.py → dump/cell_processor.py}

@@ -1,17 +1,18 @@
-# Copyright (c) 2024-2025, Huawei Technologies Co., Ltd.
-# All rights reserved.
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
 #
-# Licensed under the Apache License, Version 2.0  (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
 #
-# http://www.apache.org/licenses/LICENSE-2.0
+#          http://license.coscl.org.cn/MulanPSL2
 #
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
 
 import threading
 from collections import OrderedDict
@@ -24,7 +25,7 @@ from msprobe.core.common.const import Const
 from msprobe.core.common.exceptions import MsprobeException
 from msprobe.core.common.runtime import Runtime
 from msprobe.core.common.utils import ModuleQueue, ThreadSafe
-from msprobe.core.data_dump.scope import ModuleRangeScope, MixRangeScope, BaseScope
+from msprobe.core.dump.data_dump.scope import ModuleRangeScope, MixRangeScope, BaseScope
 from msprobe.core.common.megatron_utils import wrap_megatron_step, get_micro_step, is_megatron
 from msprobe.mindspore.common.const import Const as MsConst
 from msprobe.mindspore.common.log import logger
@@ -35,8 +36,8 @@ from msprobe.mindspore.common.utils import (
     is_graph_mode_cell_dump_allowed,
     is_backward_hook_output_a_view
 )
-from msprobe.mindspore.debugger.debugger_config import DebuggerConfig
-from msprobe.mindspore.dump.graph_mode_cell_dump import GraphModeCellDump
+from msprobe.mindspore.dump.debugger.debugger_config import DebuggerConfig
+from msprobe.mindspore.dump.dump_processor.graph_mode_cell_dump import GraphModeCellDump
 
 
 def get_cell_construct(construct):

msprobe/mindspore/{debugger → dump/debugger}/debugger_config.py

@@ -1,17 +1,18 @@
-# Copyright (c) 2024-2025, Huawei Technologies Co., Ltd.
-# All rights reserved.
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
 #
-# Licensed under the Apache License, Version 2.0  (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
 #
-# http://www.apache.org/licenses/LICENSE-2.0
+#          http://license.coscl.org.cn/MulanPSL2
 #
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
 
 import os
 
@@ -22,7 +23,6 @@ from msprobe.core.common.exceptions import MsprobeException
 from msprobe.core.common.file_utils import create_directory
 from msprobe.core.common.log import logger
 from msprobe.mindspore.common.const import Const as MsConst
-from msprobe.mindspore.common.const import FreeBenchmarkConst
 from msprobe.mindspore.common.utils import is_mindtorch
 
 if is_mindtorch():
@@ -43,7 +43,6 @@ class DebuggerConfig:
         self.scope = [] if not task_config.scope else task_config.scope
         self.data_mode = [Const.ALL] if not task_config.data_mode else task_config.data_mode
         self.file_format = task_config.file_format
-        self.overflow_nums = 1 if not task_config.overflow_nums else task_config.overflow_nums
         self.check_mode = task_config.check_mode
         self.framework = Const.MS_FRAMEWORK
         self.summary_mode = task_config.summary_mode
@@ -56,23 +55,6 @@ class DebuggerConfig:
         self._check_statistics_config(task_config)
         create_directory(self.dump_path)
 
-        if self.task == Const.FREE_BENCHMARK:
-            self.pert_type = (FreeBenchmarkConst.DEFAULT_PERT_TYPE
-                              if not task_config.pert_mode else task_config.pert_mode)
-            self.handler_type = (FreeBenchmarkConst.DEFAULT_HANDLER_TYPE
-                                 if not task_config.handler_type else task_config.handler_type)
-            self.stage = FreeBenchmarkConst.DEFAULT_STAGE if not task_config.fuzz_stage else task_config.fuzz_stage
-            if self.handler_type == FreeBenchmarkConst.FIX and \
-                    self.pert_type != FreeBenchmarkConst.DEFAULT_PERT_TYPE:
-                logger.error("pert_mode must be improve_precision or empty when handler_type is fix, "
-                             f"but got {self.pert_type}.")
-                raise ValueError
-            if self.stage == Const.BACKWARD and self.handler_type == FreeBenchmarkConst.FIX:
-                logger.error("handler_type must be check or empty when fuzz_stage is backward, "
-                             f"but got {self.handler_type}.")
-                raise ValueError
-            self.dump_level = FreeBenchmarkConst.DEFAULT_DUMP_LEVEL
-
     @staticmethod
     def check_model(models, token_range=None):
         if token_range and not models: