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

msprobe/mindspore/dump/{graph_tensor_dump.py → dump_processor/graph_tensor_dump.py}

@@ -1,134 +1,135 @@
-# Copyright (c) 2024-2025, Huawei Technologies Co., Ltd.
-# All rights reserved.
-#
-# 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
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# 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 os
-from collections import OrderedDict
-import mindspore as ms
-from mindspore import hal, ops, Tensor
-from mindspore.ops.primitive import _run_op
-
-
-def _iterate_items(data):
-    if isinstance(data, (dict, OrderedDict)):
-        return data.items()
-    elif isinstance(data, (list, tuple)):
-        return enumerate(data)
-    else:
-        raise TypeError("Unsupported data type")
-
-
-class _SaveBase:
-    def __init__(self, save_dir):
-        super(_SaveBase, self).__init__()
-        self.path = save_dir
-        self.save_func = _npy_save
-
-    def get_save_func(self):
-        return self.save_func
-
-
-@ms.jit_class
-class _SaveCell(_SaveBase):
-    def __call__(self, name, data):
-        return self.get_save_func()(self.path, name, data)
-
-
-class _SaveGradBase:
-    def __init__(self, save_dir, name):
-        super(_SaveGradBase, self).__init__()
-        self.file = save_dir + name
-
-
-@ms.jit_class
-class _SaveGradCell(_SaveGradBase):
-    def __init__(self, save_dir, name):
-        super(_SaveGradCell, self).__init__(save_dir, name)
-        self.ms_save_grad = ms.ops.InsertGradientOf(
-            _wrapper_save_grad_func(self.file))
-
-    def __call__(self, x):
-        if isinstance(x, ms.Tensor):
-            return self.ms_save_grad(x)
-        else:
-            raise TypeError(f"For 'save_grad', the type of argument 'data' must be mindspore.Tensor or torch.tensor, "
-                            f"but got {type(x)}")
-
-
-def _npy_save_ops(file, data):
-    if isinstance(data, ms.Tensor):
-        if data.dtype == ms.bfloat16:
-            data = data.float()
-        ms.ops.TensorDump()(file, data)
-    else:
-        raise TypeError(f"For 'save', the type of argument 'data' must be mindspore.Tensor or torch.tensor, "
-                        f"but got {type(data)}")
-
-
-def _wrapper_save_grad_func(file):
-    def _save_grad_func(grad):
-        data = grad
-        if data.dtype == ms.bfloat16:
-            data = data.float()
-        ms.ops.TensorDump()(file, data)
-        return grad
-    return _save_grad_func
-
-
-def _npy_save(save_dir, item_name, data):
-    if isinstance(data, (list, tuple, dict, OrderedDict)):
-        for key, val in _iterate_items(data):
-            _npy_save(save_dir, f"{item_name}.{key}", val)
-    else:
-        if data is None:
-            return
-        _npy_save_ops(f"{save_dir}{item_name}", data)
-
-
-def generate_dump_dir(save_dir, sep=os.sep):
-    """
-    usage: generate dump directory path str in mindspore graph mode
-    """
-    full_suffix = '{step}' + sep + '{rank}' + sep
-    if save_dir and save_dir[-1] != sep:
-        result_dir = save_dir + sep + full_suffix
-    else:
-        result_dir = save_dir + full_suffix
-    return result_dir
-
-
-def save(save_dir, name, data):
-    """
-    save tensor.
-    """
-    dump_dir = generate_dump_dir(save_dir)
-    _SaveCell(dump_dir)(name, data)
-
-
-def save_grad(save_dir, name, data):
-    """
-    save grad.
-    """
-    dump_dir = generate_dump_dir(save_dir)
-    suffix_name = name + '_grad'
-    return _SaveGradCell(dump_dir, suffix_name)(data)
-
-
-def step():
-    hal.synchronize()
-    temp_tensor = Tensor([1], dtype=ms.float32)
-    step_flag = "<tensordump-update-step>"
-    _run_op(ops.TensorDump(), "TensorDump", (step_flag, temp_tensor))
-    ops.tensordump(step_flag, temp_tensor)
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
+#
+# 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://license.coscl.org.cn/MulanPSL2
+#
+# 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
+from collections import OrderedDict
+import mindspore as ms
+from mindspore import hal, ops, Tensor
+from mindspore.ops.primitive import _run_op
+
+
+def _iterate_items(data):
+    if isinstance(data, (dict, OrderedDict)):
+        return data.items()
+    elif isinstance(data, (list, tuple)):
+        return enumerate(data)
+    else:
+        raise TypeError("Unsupported data type")
+
+
+class _SaveBase:
+    def __init__(self, save_dir):
+        super(_SaveBase, self).__init__()
+        self.path = save_dir
+        self.save_func = _npy_save
+
+    def get_save_func(self):
+        return self.save_func
+
+
+@ms.jit_class
+class _SaveCell(_SaveBase):
+    def __call__(self, name, data):
+        return self.get_save_func()(self.path, name, data)
+
+
+class _SaveGradBase:
+    def __init__(self, save_dir, name):
+        super(_SaveGradBase, self).__init__()
+        self.file = save_dir + name
+
+
+@ms.jit_class
+class _SaveGradCell(_SaveGradBase):
+    def __init__(self, save_dir, name):
+        super(_SaveGradCell, self).__init__(save_dir, name)
+        self.ms_save_grad = ms.ops.InsertGradientOf(
+            _wrapper_save_grad_func(self.file))
+
+    def __call__(self, x):
+        if isinstance(x, ms.Tensor):
+            return self.ms_save_grad(x)
+        else:
+            raise TypeError(f"For 'save_grad', the type of argument 'data' must be mindspore.Tensor or torch.tensor, "
+                            f"but got {type(x)}")
+
+
+def _npy_save_ops(file, data):
+    if isinstance(data, ms.Tensor):
+        if data.dtype == ms.bfloat16:
+            data = data.float()
+        ms.ops.TensorDump()(file, data)
+    else:
+        raise TypeError(f"For 'save', the type of argument 'data' must be mindspore.Tensor or torch.tensor, "
+                        f"but got {type(data)}")
+
+
+def _wrapper_save_grad_func(file):
+    def _save_grad_func(grad):
+        data = grad
+        if data.dtype == ms.bfloat16:
+            data = data.float()
+        ms.ops.TensorDump()(file, data)
+        return grad
+    return _save_grad_func
+
+
+def _npy_save(save_dir, item_name, data):
+    if isinstance(data, (list, tuple, dict, OrderedDict)):
+        for key, val in _iterate_items(data):
+            _npy_save(save_dir, f"{item_name}.{key}", val)
+    else:
+        if data is None:
+            return
+        _npy_save_ops(f"{save_dir}{item_name}", data)
+
+
+def generate_dump_dir(save_dir, sep=os.sep):
+    """
+    usage: generate dump directory path str in mindspore graph mode
+    """
+    full_suffix = '{step}' + sep + '{rank}' + sep
+    if save_dir and save_dir[-1] != sep:
+        result_dir = save_dir + sep + full_suffix
+    else:
+        result_dir = save_dir + full_suffix
+    return result_dir
+
+
+def save(save_dir, name, data):
+    """
+    save tensor.
+    """
+    dump_dir = generate_dump_dir(save_dir)
+    _SaveCell(dump_dir)(name, data)
+
+
+def save_grad(save_dir, name, data):
+    """
+    save grad.
+    """
+    dump_dir = generate_dump_dir(save_dir)
+    suffix_name = name + '_grad'
+    return _SaveGradCell(dump_dir, suffix_name)(data)
+
+
+def step():
+    hal.synchronize()
+    temp_tensor = Tensor([1], dtype=ms.float32)
+    step_flag = "<tensordump-update-step>"
+    _run_op(ops.TensorDump(), "TensorDump", (step_flag, temp_tensor))
+    ops.tensordump(step_flag, temp_tensor)
     hal.synchronize()

msprobe/mindspore/dump/{hook_cell → dump_processor/hook_cell}/api_register.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
 import inspect
@@ -23,11 +24,11 @@ from mindspore.communication import comm_func
 
 from msprobe.core.common.file_utils import load_yaml
 from msprobe.core.common.utils import Const
-from msprobe.core.data_dump.api_registry import ApiRegistry
+from msprobe.core.dump.api_dump.api_registry import ApiRegistry
 from msprobe.mindspore.common.log import logger
 from msprobe.mindspore.common.const import Const as MsConst
 from msprobe.mindspore.common.utils import is_mindtorch
-from msprobe.mindspore.dump.hook_cell.hook_cell import HOOKCell
+from msprobe.mindspore.dump.dump_processor.hook_cell.hook_cell import HOOKCell
 
 
 stub_tensor_existed = True
@@ -67,7 +68,7 @@ else:
             Const.PT_API_TYPE_DIST: ((torch.distributed,), (torch.distributed, torch.distributed.distributed_c10d))
         }
     }
-    _supported_api_list_path = (os.path.join(cur_path, '../../../pytorch/hook_module',
+    _supported_api_list_path = (os.path.join(cur_path, '../../../../pytorch/dump/api_dump',
                                              MsConst.SUPPORTED_API_LIST_FILE),)
     _blacklist = []
 

msprobe/mindspore/dump/{hook_cell → dump_processor/hook_cell}/hook_cell.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 defaultdict

msprobe/mindspore/dump/{hook_cell → dump_processor/hook_cell}/ms_hook_manager.py

@@ -1,17 +1,19 @@
-# Copyright (c) 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,15 +26,15 @@ from mindspore.ops.operations import _inner_ops as inner
 from msprobe.core.common.const import Const
 from msprobe.core.common.log import logger
 from msprobe.core.common.utils import replace_last_occurrence, ThreadSafe
-from msprobe.core.data_dump.data_processor.base import ModuleBackwardInputs
-from msprobe.core.hook_manager import BaseHookManager, HookSet
+from msprobe.core.dump.data_dump.data_processor.base import ModuleBackwardInputs
+from msprobe.core.dump.hook_manager import BaseHookManager, HookSet
 from msprobe.mindspore.common.const import Const as MsConst
 from msprobe.mindspore.common.utils import (
     has_kwargs_in_forward_hook,
     is_mindtorch,
     is_backward_hook_output_a_view
 )
-from msprobe.mindspore.dump.hook_cell.hook_cell import HOOKCell
+from msprobe.mindspore.dump.dump_processor.hook_cell.hook_cell import HOOKCell
 
 ms_version = ms.__version__
 
@@ -42,10 +44,6 @@ class MindsporeHookManager(BaseHookManager):
     cell_backward_pre_hook = []
     cell_backward_hook = []
 
-    @property
-    def _is_recompute(self):
-        return None
-
     @staticmethod
     def reset_status():
         BaseHookManager.reset_status()
@@ -116,7 +114,7 @@ class MindsporeHookManager(BaseHookManager):
             args = bw_hook(args) if is_backward_hook_output_a_view() else bw_hook(*args)
         return args
 
-    def _register_backward_pre_hook(self, module, full_backward_name, output):
+    def _register_backward_pre_hook(self, module, full_backward_name, args, kwargs, output):
         if not _pynative_executor.requires_grad():
             return output
 
@@ -199,7 +197,7 @@ class MindsporeHookManager(BaseHookManager):
             if self.config.level != Const.LEVEL_L2:
                 return
             tid = threading.get_ident()
-            if not self._should_execute_hook(tid):
+            if not self._should_execute_hook(hook_type, tid):
                 return
 
             with ThreadSafe():
@@ -212,3 +210,32 @@ class MindsporeHookManager(BaseHookManager):
                 self.restore_gc_state(original_state)
 
         return backward_pre_hook
+
+    def _register_param_hook(self, name, module, params_dict):
+        ori_name = name.rsplit(Const.SEP, 2)[0]
+        # data_mode为forward时，不注册参数hook
+        if not (Const.FORWARD in self.config.data_mode and Const.BACKWARD not in self.config.data_mode):
+            for param_name, param in params_dict.items():
+                if param.requires_grad:
+                    name = ori_name + Const.SEP + param_name
+                    old_handle = BaseHookManager.hook_handle_dict.get(name)
+                    if old_handle and hasattr(old_handle, "remove"):
+                        old_handle.remove()
+                    handle = param.register_hook(self._build_grad_hook(ori_name, param_name))
+                    BaseHookManager.hook_handle_dict[name] = handle
+
+    def _build_grad_hook(self, ori_name, param_name):
+        def hook_fn(grad):
+            tid = threading.get_ident()
+            if not self._should_execute_hook(Const.MODULE, tid):
+                return
+            with ThreadSafe():
+                original_state = self.ensure_gc_enabled()
+                BaseHookManager.inner_switch[tid] = True
+                index = self._get_grad_hook_call_index(ori_name, param_name)
+                self.data_collector.params_data_collect(ori_name, param_name, self._pid, grad, str(index))
+                BaseHookManager.inner_switch[tid] = False
+                self.restore_gc_state(original_state)
+            return
+
+        return hook_fn

msprobe/mindspore/dump/{hook_cell → dump_processor/hook_cell}/primitive_hooks.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 os
 import threading
@@ -19,12 +20,12 @@ import threading
 from mindspore import ops
 from mindspore.common.tensor import Tensor
 from msprobe.core.common.utils import Const, DumpException, ThreadSafe
-from msprobe.core.data_dump.data_processor.base import (
+from msprobe.core.dump.data_dump.data_processor.base import (
     ModuleBackwardInputs,
     ModuleBackwardOutputs,
     ModuleForwardInputsOutputs
 )
-from msprobe.core.hook_manager import BaseHookManager
+from msprobe.core.dump.hook_manager import BaseHookManager
 from msprobe.mindspore.common.log import logger
 
 

msprobe/mindspore/dump/{hook_cell → dump_processor/hook_cell}/support_wrap_ops.yaml

@@ -1,17 +1,18 @@
-# Copyright 2024 Huawei Technologies Co., Ltd
+# -------------------------------------------------------------------------
+#  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.
+# -------------------------------------------------------------------------
 
 # List of ops that register hooks
 
@@ -516,6 +517,7 @@ ops:
   - vander
   - is_nonzero
   - rotary_position_embedding
+  - auto_generate.quant_matmul
 
 tensor:
   - __abs__

msprobe/mindspore/dump/{jit_dump.py → dump_processor/jit_dump.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 os
 import types
@@ -30,9 +31,9 @@ from msprobe.core.common.log import logger
 from msprobe.core.common.const import Const
 from msprobe.core.common.utils import ThreadSafe
 from msprobe.core.common.runtime import Runtime
-from msprobe.core.data_dump.data_processor.base import ModuleForwardInputsOutputs, ModuleBackwardInputsOutputs
+from msprobe.core.dump.data_dump.data_processor.base import ModuleForwardInputsOutputs, ModuleBackwardInputsOutputs
 from msprobe.mindspore.common.const import Const as MsConst
-from msprobe.mindspore.dump.hook_cell.api_register import get_api_register
+from msprobe.mindspore.dump.dump_processor.hook_cell.api_register import get_api_register
 
 _api_register = get_api_register()
 

msprobe/mindspore/dump/{kernel_graph_dump.py → dump_processor/kernel_graph_dump.py}

@@ -1,23 +1,24 @@
-# 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 os
 
 from msprobe.core.common.file_utils import create_directory, save_json
 from msprobe.mindspore.common.log import logger
-from msprobe.mindspore.debugger.debugger_config import DebuggerConfig
+from msprobe.mindspore.dump.debugger.debugger_config import DebuggerConfig
 
 
 class KernelGraphDump: