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

msprobe/core/common/const.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.
+# -------------------------------------------------------------------------
+
 
 import os
 import stat
@@ -112,14 +114,15 @@ class Const:
     STATISTICS = "statistics"
     TENSOR = "tensor"
     OVERFLOW_CHECK = "overflow_check"
+    DIFF_CHECK = "diff_check"
     FREE_BENCHMARK = "free_benchmark"
-    RUN_UT = "run_ut"
-    GRAD_PROBE = "grad_probe"
+    ACC_CHECK = "acc_check"
     STRUCTURE = "structure"
     EXCEPTION_DUMP = "exception_dump"
     DUMP_PRECISION_HIGH = "high"
     DUMP_PRECISION_LOW = "low"
-    TASK_LIST = [TENSOR, STATISTICS, OVERFLOW_CHECK, FREE_BENCHMARK, RUN_UT, GRAD_PROBE, STRUCTURE, EXCEPTION_DUMP]
+    TASK_LIST = [TENSOR, STATISTICS, ACC_CHECK, STRUCTURE, OVERFLOW_CHECK, DIFF_CHECK, EXCEPTION_DUMP]
+    TORCH_TASK_LIST = [TENSOR, STATISTICS, ACC_CHECK, STRUCTURE, DIFF_CHECK]
     DUMP_DATA_COLLECTION_LIST = [STATISTICS, TENSOR, STRUCTURE]
     DUMP_DATA_MODE_LIST = [ALL, INPUT, OUTPUT, FORWARD, BACKWARD]
     DUMP_PRECISION_LIST = [DUMP_PRECISION_LOW, DUMP_PRECISION_HIGH]
@@ -216,7 +219,7 @@ class Const:
 
     WITHOUT_CALL_STACK = "The call stack retrieval failed."
     STACK_FILTER_KEYWORDS = ["msprobe/core", "msprobe/pytorch", "msprobe/mindspore"]
-    CALL_STACK_FLAG = "data_dump/api_registry"
+    CALL_STACK_FLAG = "api_dump/api_registry"
     NEW_STACK_FLAG = "0"
 
     STEP = "step"
@@ -238,6 +241,8 @@ class Const:
     TYPE = 'type'
     DTYPE = 'dtype'
     SHAPE = 'shape'
+    LAYOUT = 'layout'
+    SHARD_SIZE = 'hsdp_shard_size'
     STACK_INFO = 'stack_info'
     MAX = 'Max'
     MIN = 'Min'
@@ -275,6 +280,7 @@ class Const:
     DTENSOR_TYPE = "torch.distributed.tensor.DTensor"
     FAKE_TENSOR_TYPE = "torch._subclasses.fake_tensor.FakeTensor"
     AC_TENSOR_TYPE = "torch.distributed._functional_collectives.AsyncCollectiveTensor"
+    NESTED_TENSOR_TYPE = "torch.nested._internal.nested_tensor.NestedTensor"
 
     SUPPORT_API_FILE_NAME = "support_wrap_ops.yaml"
 
@@ -287,6 +293,7 @@ class Const:
     PT_API_TYPE_NPU = "torch_npu"
     PT_API_TYPE_ATEN = "aten"
     PT_API_TYPE_DIST = "distributed"
+    PT_API_TYPE_DIST_C10D = "distributed_c10d"
     PT_API_TYPE_NPU_DIST = "npu_distributed"
     PT_API_TYPE_MINDSPEED = "mindspeed"
 
@@ -321,6 +328,7 @@ class Const:
             PT_API_TYPE_NPU: PT_API_TYPE_NPU,
             PT_API_TYPE_ATEN: PT_API_TYPE_ATEN,
             PT_API_TYPE_DIST: PT_API_TYPE_DIST,
+            PT_API_TYPE_DIST_C10D: PT_API_TYPE_DIST,
             PT_API_TYPE_NPU_DIST: PT_API_TYPE_NPU_DIST,
             PT_API_TYPE_MINDSPEED: PT_API_TYPE_MINDSPEED
         },
@@ -351,6 +359,7 @@ class Const:
             PT_API_TYPE_NPU: NPU_API_TYPE_PREFIX,
             PT_API_TYPE_ATEN: ATEN_API_TYPE_PREFIX,
             PT_API_TYPE_DIST: DIST_API_TYPE_PREFIX,
+            PT_API_TYPE_DIST_C10D: DIST_API_TYPE_PREFIX,
             PT_API_TYPE_NPU_DIST: DIST_API_TYPE_PREFIX,
             PT_API_TYPE_MINDSPEED: MINDSPEED_API_TYPE_PREFIX
         },
@@ -408,6 +417,43 @@ class Const:
 
     MEGATRON_MICRO_STEP_NUMBER = 'megatron_micro_step_number'
 
+    TORCHAIR_FX_DATA_DIRECTORY = 'msprobe_fx_dump'
+    TORCHAIR_GE_DATA_DIRECTORY = 'msprobe_ge_dump'
+
+    FULL_OP_NAME = "full_op_name"
+    DIRECTION = "direction"
+    CALL_DIRECTION = "call_direction"
+    OP_NO_NUMBER = "op_no_number"
+    BACKWARD_CALL_ORDER = "backward_call_order"
+    SUFFIX = "suffix"
+
+
+class Data2DBConst:
+    BATCH_SIZE = 5000
+    FORWARD = "forward"
+    BACKWARD = "backward"
+    RECOMPUTE = "recompute"
+    PARAMETERS_GRAD = "parameters_grad"
+    METRICS = [FORWARD, BACKWARD, RECOMPUTE, PARAMETERS_GRAD]
+    DEFAULT_TAGS = ["Module", "Cell"]
+    ORDERED_STAT = ["norm", "max", "min", "mean"]
+    TAG_DEFAULT = "default"
+    TAG_INDEX = "index"
+    TAG_LAYER = "layer"
+    TAG_MODULE = "module"
+    TAG_FUNCTION = "function"
+    MAX_PARTITION = 10_000_000
+    MIN_PARTITION = 10
+    MAX_MICRO_STEP = 10000
+    MIN_MICRO_STEP = 1
+    SUPPORT_TYPE = ["torch.Tensor", "mindspore.Tensor", "mindtorch.Tensor"]
+    SUPPORT_DTYPE = [Const.FLOAT16,
+                     Const.FLOAT32,
+                     Const.BFLOAT16,
+                     Const.TORCH_FLOAT16,
+                     Const.TORCH_FLOAT32,
+                     Const.TORCH_BFLOAT16]
+
 
 class CompareConst:
     """
@@ -459,6 +505,7 @@ class CompareConst:
     RESULT = "Result"
     MAGNITUDE = 0.5
     OP_NAME = "op_name"
+    OP_NAME_UPDATE = "op_name_update"
     STRUCT = "struct"
     INPUT_STRUCT = "input_struct"
     KWARGS_STRUCT = "kwargs_struct"
@@ -526,12 +573,12 @@ class CompareConst:
     BASIC_INFO = [NPU_NAME, BENCH_NAME, NPU_DTYPE, BENCH_DTYPE, NPU_SHAPE, BENCH_SHAPE, NPU_REQ_GRAD, BENCH_REQ_GRAD]
     SUMMARY_INFO = [NPU_MAX, NPU_MIN, NPU_MEAN, NPU_NORM, BENCH_MAX, BENCH_MIN, BENCH_MEAN, BENCH_NORM]
 
-    COMPARE_RESULT_HEADER = BASIC_INFO + ALL_COMPARE_INDEX + SUMMARY_INFO + [REQ_GRAD_CONSIST, ACCURACY, ERROR_MESSAGE]
+    COMPARE_RESULT_HEADER = BASIC_INFO + ALL_COMPARE_INDEX + SUMMARY_INFO + [REQ_GRAD_CONSIST, RESULT, ERROR_MESSAGE]
 
     SUMMARY_COMPARE_RESULT_HEADER = BASIC_INFO + SUMMARY_COMPARE_INDEX + SUMMARY_INFO + [REQ_GRAD_CONSIST, RESULT,
                                                                                          ERROR_MESSAGE]
 
-    MD5_COMPARE_RESULT_HEADER = BASIC_INFO + [NPU_MD5, BENCH_MD5, REQ_GRAD_CONSIST] + MD5_COMPARE_INDEX
+    MD5_COMPARE_RESULT_HEADER = BASIC_INFO + [NPU_MD5, BENCH_MD5, REQ_GRAD_CONSIST] + [RESULT, ERROR_MESSAGE]
 
     COMPARE_RESULT_HEADER_STACK = COMPARE_RESULT_HEADER + [STACK]
 
@@ -587,10 +634,11 @@ class CompareConst:
     NONE = 'None'
     SHAPE_UNMATCH = 'shape unmatched'
     DIFF = 'Different'
+    DIFF_FLAG = 'Diff'
     UNSUPPORTED = 'unsupported'
     NAN = 'Nan'
     PASS = 'pass'
-    WARNING = 'Warning'
+    WARNING = 'warning'
     ERROR = 'error'
     TRUE = 'TRUE'
     FALSE = 'FALSE'
@@ -618,26 +666,12 @@ class CompareConst:
     # compare const
     FLOAT_TYPE = [np.half, np.single, float, np.double, np.float64, np.longdouble]
 
-    # highlight xlsx color const
-    RED = "FFFF0000"
-    YELLOW = "FFFF00"
-    BLUE = "0000FF"
-
-    # run_ut const
+    # acc_check const
     MAX_TOKENS = 65536
     SPECIAL_SPARSE_MOED = 4
 
-    # highlight rules const
     OVERFLOW_LIST = ['nan\t', 'inf\t', '-inf\t', 'nan', 'inf', '-inf']
-    MAX_DIFF_RED = 1e+10
-    ORDER_MAGNITUDE_DIFF_YELLOW = 1
-    ONE_THOUSAND_ERROR_IN_RED = 0.9
-    ONE_THOUSAND_ERROR_OUT_RED = 0.6
-    ONE_THOUSAND_ERROR_DIFF_YELLOW = 0.1
-    COSINE_DIFF_YELLOW = 0.1
-    MAX_RELATIVE_OUT_RED = 0.5
-    MAX_RELATIVE_OUT_YELLOW = 0.1
-    MAX_RELATIVE_IN_YELLOW = 0.01
+
     MS_GRAPH_BASE = {
         NPU_NAME: None, BENCH_NAME: None, NPU_DTYPE: None, BENCH_DTYPE: None, NPU_SHAPE: None, BENCH_SHAPE: None,
         NPU_MAX: None, NPU_MIN: None, NPU_MEAN: None, NPU_NORM: None, BENCH_MAX: None, BENCH_MIN: None,
@@ -686,6 +720,17 @@ class CompareConst:
     BENCH_DUMP_DATA_DIR = 'bench_dump_data_dir'
     NO_REAL_DATA_FLAG = '-1'
 
+    # ATB data comparison
+    STATS_COMPARISION_MODE = 'stats'
+    TENSOR_COMPARISION_MODE = 'tensor'
+    UNKNOWN_COMPARISION_MODE = "unknown"
+    ATB_DATA_NAME = 'Data Name'
+    DEVICE_AND_PID = 'Device and PID'
+    EXECUTION_COUNT = 'Execution Count'
+    DATA_TYPE = 'Data Type'
+    DATA_SHAPE = 'Data Shape'
+    TENSOR_PATH = "tensor_path"
+
 
 class FileCheckConst:
     """
@@ -724,9 +769,14 @@ class FileCheckConst:
     SHELL_SUFFIX = ".sh"
     LOG_SUFFIX = ".log"
     DB_SUFFIX = '.db'
+    ONNX_SUFFIX = '.onnx'
+    OM_SUFFIX = '.om'
+    BIN_SUFFIX = '.bin'
+    MAX_BIN_SIZE = 10737418240          # 10G
+    MAX_COMMON_FILE_SIZE = 10737418240  # 10G
     MAX_PKL_SIZE = 1073741824  # 1 * 1024 * 1024 * 1024
     MAX_NUMPY_SIZE = 10737418240  # 10 * 1024 * 1024 * 1024
-    MAX_JSON_SIZE = 1073741824  # 1 * 1024 * 1024 * 1024
+    MAX_JSON_SIZE = 10737418240  # 10 * 1024 * 1024 * 1024
     MAX_PT_SIZE = 10737418240  # 10 * 1024 * 1024 * 1024
     MAX_CSV_SIZE = 1073741824  # 1 * 1024 * 1024 * 1024
     MAX_XLSX_SIZE = 1073741824  # 1 * 1024 * 1024 * 1024
@@ -735,9 +785,11 @@ class FileCheckConst:
     MAX_ZIP_SIZE = 10737418240  # 10 * 1024 * 1024 * 1024
     MAX_FILE_IN_ZIP_SIZE = 1073741824  # 1 * 1024 * 1024 * 1024
     MAX_FILE_SIZE = 1073741824  # 1 * 1024 * 1024 * 1024
-    COMMOM_FILE_SIZE = 1048576  # 1 * 1024 * 1024
+    COMMON_FILE_SIZE = 1073741824  # 1 * 1024 * 1024 * 1024
     MAX_LOG_SIZE = 10737418240  # 1 * 1024 * 1024 * 1024
-    MAX_DB_SIZE = 10737418240  # 10 * 1024 * 1024 * 1024
+    MAX_DB_SIZE = 107374182400  # 100 * 1024 * 1024 * 1024
+    MAX_ONNX_SIZE = 10737418240  # 1 * 1024 * 1024 * 1024
+    MAX_OM_SIZE = 10737418240  # 10 * 1024 * 1024 * 1024
     DIR = "dir"
     FILE = "file"
     DATA_DIR_AUTHORITY = 0o750
@@ -753,7 +805,10 @@ class FileCheckConst:
         IR_SUFFIX: MAX_IR_SIZE,
         ZIP_SUFFIX: MAX_ZIP_SIZE,
         LOG_SUFFIX: MAX_LOG_SIZE,
-        DB_SUFFIX: MAX_DB_SIZE
+        DB_SUFFIX: MAX_DB_SIZE,
+        ONNX_SUFFIX: MAX_ONNX_SIZE,
+        OM_SUFFIX: MAX_OM_SIZE,
+        BIN_SUFFIX: MAX_BIN_SIZE
     }
     CSV_BLACK_LIST = r'^[＋－＝％＠\+\-=%@]|;[＋－＝％＠\+\-=%@]'
 
@@ -817,6 +872,10 @@ class MonitorConst:
         "entropy", "softmax_max", "sr", "kernel_norm", "std_x", "jacobian",
         "proxy", "token_similarity"
     ]
+    METRICS_TRENDVIS_SUPPORTED = [
+        "actv", "actv_grad", "exp_avg", "exp_avg_sq",
+        "grad_unreduced", "grad_reduced", "param_origin", "param_updated", "other"
+    ]
     MONITOR_OUTPUT_DIR = "MONITOR_OUTPUT_DIR"
     DEFAULT_MONITOR_OUTPUT_DIR = "./monitor_output"
     DATABASE = "database"

msprobe/core/common/db_manager.py

@@ -1,23 +1,25 @@
-# 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 re
 import sqlite3
 from typing import List, Tuple, Dict, Any
 from functools import wraps
 
-from msprobe.pytorch.common.log import logger
+from msprobe.core.common.log import logger
 from msprobe.core.common.file_utils import check_path_before_create, change_mode
 from msprobe.core.common.const import FileCheckConst
 
@@ -254,3 +256,122 @@ class DBManager:
         except sqlite3.Error as err:
             logger.error(f"Failed to release database connection: {err}")
         change_mode(self.db_path, FileCheckConst.DATA_FILE_AUTHORITY)
+
+
+class TrendSql:
+    """趋势可视化数据库表参数类"""
+
+    @staticmethod
+    def create_monitoring_targets_table():
+        """监测目标表"""
+        return """
+        CREATE TABLE IF NOT EXISTS monitoring_targets (
+            target_id INTEGER PRIMARY KEY AUTOINCREMENT,
+            target_name TEXT NOT NULL,
+            vpp_stage INTEGER NOT NULL,
+            micro_step INTEGER NOT NULL DEFAULT 0,
+            UNIQUE(target_name, vpp_stage, micro_step) 
+        )"""
+
+    @staticmethod
+    def create_monitoring_metrics_table():
+        """监测指标表"""
+        return """
+        CREATE TABLE IF NOT EXISTS monitoring_metrics (
+            metric_id INTEGER PRIMARY KEY AUTOINCREMENT,
+            metric_name TEXT UNIQUE NOT NULL
+        )"""
+
+    @staticmethod
+    def get_metric_mapping_sql():
+        """从monitoring_metrics表获取所有metric名称和ID"""
+        return """
+        SELECT metric_id, metric_name
+        FROM monitoring_metrics
+        """
+
+    @staticmethod
+    def get_global_stats_sql():
+        """从global_stats表获取最新的统计配置"""
+        return """
+        SELECT * FROM global_stats 
+        ORDER BY ROWID DESC 
+        LIMIT 1
+        """
+
+    @staticmethod
+    def create_global_stats_table(columns_config):
+        """根据字典配置创建全局统计表"""
+        # 根据字典的键值类型动态创建列
+        column_definitions = []
+        for column_name, column_value in columns_config.items():
+            if isinstance(column_value, (list, set)):
+                column_definitions.append(f"{column_name} TEXT DEFAULT NULL")
+            elif isinstance(column_value, (int, float)):
+                column_definitions.append(f"{column_name} INTEGER DEFAULT 0")
+        
+        create_sql = f"""
+        CREATE TABLE IF NOT EXISTS global_stats (
+            {', '.join(column_definitions)}
+        )"""
+        return create_sql
+
+    @staticmethod
+    def create_tags_table():
+        # 新增：标签表
+        return """
+        CREATE TABLE IF NOT EXISTS monitoring_tags (
+            tag_id INTEGER PRIMARY KEY AUTOINCREMENT,
+            tag_name TEXT NOT NULL,
+            category TEXT NOT NULL,
+            metric_id INTEGER NOT NULL,
+            UNIQUE(tag_name, category, metric_id),
+            FOREIGN KEY (metric_id) REFERENCES monitoring_metrics(metric_id)
+        )"""
+
+    @staticmethod
+    def create_tag_mapping_table():
+        return """
+        CREATE TABLE IF NOT EXISTS tag_target_mapping (
+            tag_id INTEGER NOT NULL,
+            target_id INTEGER NOT NULL,
+            PRIMARY KEY (tag_id, target_id),
+            FOREIGN KEY (tag_id) REFERENCES monitoring_tags(tag_id),
+            FOREIGN KEY (target_id) REFERENCES monitoring_target(target_id)
+        )"""
+
+    @staticmethod
+    def create_trend_table(stats):
+        stat_columns = [f"{stat} REAL DEFAULT NULL" for stat in stats]
+        create_sql = f"""
+        CREATE TABLE IF NOT EXISTS trend_data (
+            rank INTEGER NOT NULL,
+            step INTEGER NOT NULL,
+            target_id INTEGER NOT NULL,
+            metric_id INTEGER NOT NULL,
+            {', '.join(stat_columns)},
+            PRIMARY KEY (rank, step, target_id, metric_id),
+            FOREIGN KEY (target_id) REFERENCES monitoring_targets(target_id),
+            FOREIGN KEY (metric_id) REFERENCES monitoring_metrics(metric_id)
+        ) WITHOUT ROWID"""
+        return create_sql
+    
+    @classmethod
+    def get_table_definition(cls, table_name=""):
+        """
+        获取表定义SQL
+        :param table_name: 表名
+        :return: 建表SQL语句
+        :raises ValueError: 当表名不存在时
+        """
+        table_creators = {
+            "monitoring_targets": cls.create_monitoring_targets_table,
+            "monitoring_metrics": cls.create_monitoring_metrics_table,
+            "monitoring_tags": cls.create_tags_table,
+            "tag_target_mapping": cls.create_tag_mapping_table,
+        }
+        if not table_name:
+            return [table_creators.get(table, lambda x: "")() for table in table_creators]
+        if table_name not in table_creators:
+            raise ValueError(f"Unsupported table name: {table_name}")
+        return table_creators[table_name]()

msprobe/core/common/decorator.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.
+# -------------------------------------------------------------------------
 
 from collections import defaultdict
 from functools import wraps

msprobe/core/common/exceptions.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.
+# -------------------------------------------------------------------------
 
 class CodedException(Exception):
     def __init__(self, code, error_info=''):