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

msprobe/core/compare/compare_cli.py

@@ -1,157 +1,80 @@
-# 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
 
-from msprobe.core.common.file_utils import check_file_type, load_json, check_file_or_directory_path
-from msprobe.core.common.const import FileCheckConst, Const
 from msprobe.core.common.utils import CompareException
 from msprobe.core.common.log import logger
-from msprobe.core.compare.utils import get_paired_dirs
-
-
-def compare_cli(args, depth=1):
-    if depth > 2:
-        logger.error("Recursive compare error, depth exceeds 2.")
-        raise CompareException(CompareException.RECURSION_LIMIT_ERROR)
-
-    if isinstance(args.input_path, dict):  # special for dyn-graph mix compare
-        input_param = args.input_path
-    else:
-        input_param = load_json(args.input_path)
-    if not isinstance(input_param, dict):
-        logger.error("input_param should be dict, please check!")
-        raise CompareException(CompareException.INVALID_OBJECT_TYPE_ERROR)
-
-    npu_path = input_param.get("npu_path", None)
-    bench_path = input_param.get("bench_path", None)
-    if not npu_path:
-        logger.error(f"Missing npu_path in input configuration file, please check!")
-        raise CompareException(CompareException.INVALID_PATH_ERROR)
-    if not bench_path:
-        logger.error(f"Missing bench_path in input configuration file, please check!")
-        raise CompareException(CompareException.INVALID_PATH_ERROR)
-
-    frame_name = args.framework
-    auto_analyze = not args.compare_only
-
-    if frame_name == Const.PT_FRAMEWORK:
-        from msprobe.pytorch.compare.pt_compare import compare
-        from msprobe.pytorch.compare.distributed_compare import compare_distributed
-    else:
-        from msprobe.mindspore.compare.ms_compare import ms_compare
-        from msprobe.mindspore.compare.distributed_compare import ms_compare_distributed, ms_graph_compare
-        from msprobe.mindspore.compare.common_dir_compare import common_dir_compare
-
-    common_kwargs = {
-        "auto_analyze": auto_analyze,
-        "fuzzy_match": args.fuzzy_match,
-        "highlight": args.highlight,
-        "data_mapping": args.data_mapping,
-        "diff_analyze": args.diff_analyze
-    }
-
-    if check_file_type(npu_path) == FileCheckConst.FILE and check_file_type(bench_path) == FileCheckConst.FILE:
-        check_file_or_directory_path(npu_path)
-        check_file_or_directory_path(bench_path)
-        input_param["npu_json_path"] = input_param.pop("npu_path")
-        input_param["bench_json_path"] = input_param.pop("bench_path")
-        if "stack_path" not in input_param:
-            logger.warning(f"Missing stack_path in the configuration file. "
-                           f"Automatically detecting stack.json to determine whether to display NPU_Stack_Info.")
-        else:
-            input_param["stack_json_path"] = input_param.pop("stack_path")
-
-        if frame_name == Const.PT_FRAMEWORK:
-            kwargs = {**common_kwargs, "stack_mode": args.stack_mode}
-            compare(input_param, args.output_path, **kwargs)
-        else:
-            kwargs = {
-                **common_kwargs,
-                "stack_mode": args.stack_mode,
-                "cell_mapping": args.cell_mapping,
-                "api_mapping": args.api_mapping,
-                "layer_mapping": args.layer_mapping
-            }
-            ms_compare(input_param, args.output_path, **kwargs)
-    elif check_file_type(npu_path) == FileCheckConst.DIR and check_file_type(bench_path) == FileCheckConst.DIR:
-        check_file_or_directory_path(npu_path, isdir=True)
-        check_file_or_directory_path(bench_path, isdir=True)
-
-        if depth == 1:
-            mix_compare_success = mix_compare(args, input_param, depth)
-            if mix_compare_success:
-                return
-
-        kwargs = {
-            **common_kwargs,
-            "stack_mode": args.stack_mode,
-            "is_print_compare_log": input_param.get("is_print_compare_log", True),
-            "cell_mapping": args.cell_mapping,
-            "api_mapping": args.api_mapping,
-            "layer_mapping": args.layer_mapping
-        }
-        if input_param.get("rank_id") is not None:
-            ms_graph_compare(input_param, args.output_path)
-            return
-        common = input_param.get("common", False)
-        if isinstance(common, bool) and common:
-            common_dir_compare(input_param, args.output_path)
-            return
-
-        if common_kwargs.get('diff_analyze', False):
-            logger.info("Start finding first diff node......")
-            from msprobe.core.compare.find_first.analyzer import DiffAnalyzer
-            DiffAnalyzer(npu_path, bench_path, args.output_path, frame_name).analyze()
-            return
-
-        if frame_name == Const.PT_FRAMEWORK:
-            compare_distributed(npu_path, bench_path, args.output_path, **kwargs)
-        else:
-            ms_compare_distributed(npu_path, bench_path, args.output_path, **kwargs)
-    else:
-        logger.error("The npu_path and bench_path need to be of the same type.")
+from msprobe.core.compare.atb_data_compare import compare_atb_mode
+from msprobe.core.compare.auto_compare import compare_auto_mode
+from msprobe.core.compare.offline_data_compare import compare_offline_data_mode
+from msprobe.core.compare.torchair_acc_cmp import compare_torchair_mode
+from msprobe.infer.offline.compare.msquickcmp.main import compare_offline_model_mode
+
+
+MODE_DISPATCHER = {
+    'auto': compare_auto_mode,
+    'offline_data': compare_offline_data_mode,
+    'torchair': compare_torchair_mode,
+    'offline_model': compare_offline_model_mode,
+    'atb': compare_atb_mode
+}
+
+VALID_ARGS_MAP = {
+    'auto': [
+        '--mode', '-m', '--target_path', '-tp', '--golden_path', '-gp', '--output_path', '-o', 
+        '--fuzzy_match', '-fm', '--cell_mapping', '-cm', '--api_mapping', '-am', '--data_mapping',
+        '-dm', '--layer_mapping', '-lm', '--diff_analyze', '-da', '--rank', '--step',
+        '--is_print_compare_log', '-tensor_log'
+    ],
+    'offline_data': [
+        "-m", "-tp", "-gp", "-fr", "-qfr", "-cfr", "-o", "--mode", "--target_path", "--golden_path", 
+        "--fusion_rule_file", "--quant_fusion_rule_file", "--close_fusion_rule_file", "--output_path"
+    ],
+    'torchair': [
+        '--mode', '-m', '--target_path', '-tp', '--golden_path', '-gp', '--output_path', '-o'
+    ],
+    'offline_model': [
+        '--mode', '-m', '--target_path', '-tp', '--golden_path', '-gp', '--output_path', '-o',
+        '--rank', '--input_data', '--input_shape', '--output_size', '--dym_shape_range',
+        '-ofs', '--onnx_fusion_switch'
+    ],
+    'atb': [
+        '--mode', '-m', '--target_path', '-tp', '--golden_path', '-gp', '--output_path', '-o'
+    ],
+}
+
+
+def check_valid_args(sys_argv, mode):
+    valid_args = VALID_ARGS_MAP.get(mode)
+    for arg in sys_argv:
+        if arg.startswith("-") and arg not in valid_args:
+            logger.error(f"Invalid argument '{arg}' for mode '{mode}'")
+            raise CompareException(CompareException.INVALID_PARAM_ERROR)
+
+
+def compare_cli(args, sys_argv):
+    """
+    Dispatch comparison based on mode parameter
+    """
+    mode = getattr(args, 'mode', 'auto')
+    # Get the appropriate function based on mode
+    compare_func = MODE_DISPATCHER.get(mode)
+    if compare_func is None:
+        logger.error(f"Invalid mode '{mode}'. Available modes: {list(MODE_DISPATCHER.keys())}")
         raise CompareException(CompareException.INVALID_COMPARE_MODE)
-
-
-def mix_compare(args, input_param, depth):
-    npu_path = input_param.get("npu_path", None)
-    bench_path = input_param.get("bench_path", None)
-
-    npu_bench_same_dirs_set = set(get_paired_dirs(npu_path, bench_path))
-    compare_cross_set = npu_bench_same_dirs_set & Const.MIX_DUMP_NAMES
-
-    if compare_cross_set:
-        logger.info("Start mix compare.")
-        origin_output = args.output_path
-
-        for folder_name in list(compare_cross_set):
-            new_npu_path = os.path.join(npu_path, folder_name)
-            new_bench_path = os.path.join(bench_path, folder_name)
-            paired_steps = get_paired_dirs(new_npu_path, new_bench_path)
-
-            for step_name in paired_steps:
-                logger.info(f"[mix compare] Start comparing {folder_name}/{step_name}")
-                npu_dir = os.path.join(new_npu_path, step_name)
-                bench_dir = os.path.join(new_bench_path, step_name)
-                args.input_path = {
-                    "npu_path": npu_dir,
-                    "bench_path": bench_dir,
-                    "is_print_compare_log": input_param.get("is_print_compare_log", True)
-                }
-                args.output_path = os.path.join(origin_output, folder_name, step_name)
-                compare_cli(args, depth + 1)
-        return True
-    return False
+    check_valid_args(sys_argv, mode)
+    # Execute the comparison function
+    return compare_func(args)

msprobe/core/compare/config.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
 
@@ -22,9 +23,7 @@ from msprobe.core.common.file_utils import load_yaml
 class ModeConfig:
     def __init__(self, **kwargs):
         self.stack_mode = kwargs.get('stack_mode', False)
-        self.auto_analyze = kwargs.get('auto_analyze', True)
         self.fuzzy_match = kwargs.get('fuzzy_match', False)
-        self.highlight = kwargs.get('highlight', False)
         self.dump_mode = kwargs.get('dump_mode', Const.SUMMARY)
         self.first_diff_analyze = kwargs.get('first_diff_analyze', False)
         self.diff_analyze = kwargs.get('diff_analyze', False)

msprobe/core/compare/diff_analyze/first_diff_analyze.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,6 +27,8 @@ from msprobe.core.compare.utils import gen_api_batches
 
 cur_dir = os.path.dirname(os.path.realpath(__file__))
 diff_threshold_yaml_path = os.path.join(cur_dir, 'diff_analyze_threshold.yaml')
+ignore_op_list_yaml_path = os.path.join(cur_dir, 'ignore_op_list.yaml')
+ignore_list = load_yaml(ignore_op_list_yaml_path)
 thresholds = load_yaml(diff_threshold_yaml_path)
 cmp_metrics = thresholds.get('compare_metrics')
 
@@ -51,7 +54,7 @@ class FirstDiffAnalyze:
                 return True
         return False
 
-    def single_api_check(self, result_slice, header):
+    def single_api_check(self, result_slice, header, api_name=None):
         """
         单个api差异检查
 
@@ -65,14 +68,18 @@ class FirstDiffAnalyze:
         }
 
         column_indices = {name: idx for idx, name in enumerate(header)}
-
+        output_idx = -1
         for line in result_slice:
             op_item = {
                 column_name: line[column_indices[column_name]]
                 for column_name in header
             }
             single_check_result['op_items'].append(op_item)
-
+            if op_item['state'] != 'output':
+                continue
+            output_idx += 1
+            if output_idx in ignore_list.get(api_name, []):
+                continue
             # set is_same
             if self.mode_config.dump_mode == Const.MD5:
                 if line[column_indices[CompareConst.RESULT]] == CompareConst.DIFF:
@@ -117,7 +124,13 @@ class FirstDiffAnalyze:
         with tqdm(total=len(api_batches), desc=bar_desc_add_rank, unit="api/module", ncols=100) as progress_bar:
             for api_batch in api_batches:
                 result_slice = result[api_batch.start: api_batch.params_grad_end_index]
-                check_result[api_batch.api_name] = self.single_api_check(result_slice, header)
+                api_compo = api_batch.api_name.split('.')
+                # suppose name is Tensor.MatMul.0.forward
+                if len(api_compo) < 4:
+                    continue
+                # get MatMul as api_name
+                api_name = api_compo[-3]
+                check_result[api_batch.api_name] = self.single_api_check(result_slice, header, api_name)
                 progress_bar.update(1)
 
         return check_result

msprobe/core/compare/diff_analyze/ignore_op_list.yaml

@@ -0,0 +1,3 @@
+npu_fusion_attention:
+  - 4
+  - 5

msprobe/core/compare/find_first/analyzer.py

@@ -1,39 +1,38 @@
-# 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 time
 from collections import defaultdict
 import os
 from itertools import dropwhile, chain
 
-from msprobe.core.common import const
-from msprobe.core.common.file_utils import check_file_or_directory_path, save_json, make_dir
+from msprobe.core.common.file_utils import save_json, create_directory
 from msprobe.core.common.log import logger
 from msprobe.core.common.const import Const
-from msprobe.core.compare.find_first.data_processor import DataProcessor
+from msprobe.core.compare.utils import compare_distributed_inner
 from msprobe.core.compare.find_first.utils import (RankPath, FileCache, is_communication_op, is_ignore_op,
                                                    DiffAnalyseConst, analyze_diff_in_group)
 from msprobe.core.compare.find_first.graph import DataNode, CommunicationNode
 
 
 class DiffAnalyzer:
-    def __init__(self, npu_path, bench_path, output_path, data_frame=Const.PT_FRAMEWORK):
+    def __init__(self, npu_path, bench_path, output_path):
         self._bench_path = bench_path
         self._npu_path = npu_path
         self._output_path = output_path
-        self.pre_processor = DataProcessor(data_frame)
         self._paths = {}
         self._diff_nodes = []  # 记录所有异常节点
         self._cache = FileCache()
@@ -51,7 +50,8 @@ class DiffAnalyzer:
         logger.info('Cannot find any diff node, no need to generate analyze file.')
 
     def _pre_process(self):
-        self.pre_processor.process(self._npu_path, self._bench_path, self._output_path)
+        logger.info("Start comparing data ......")
+        compare_distributed_inner(self._npu_path, self._bench_path, self._output_path, first_diff_analyze=True)
         self._resolve_input_path(self._output_path)
         logger.info("Pre Process completed.")
 
@@ -272,7 +272,7 @@ class DiffAnalyzer:
 
     def _gen_analyze_info(self):
         if not os.path.exists(self._output_path):
-            make_dir(self._output_path)
+            create_directory(self._output_path)
         file_name = f'diff_analyze_{time.time_ns()}.json'
         result_file = os.path.join(self._output_path, file_name)
         result_content = defaultdict(list)

msprobe/core/compare/find_first/graph.py

@@ -1,17 +1,18 @@
-# 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.
+# -------------------------------------------------------------------------
 
 from dataclasses import dataclass
 from msprobe.core.common.const import Const

msprobe/core/compare/find_first/utils.py

@@ -1,17 +1,18 @@
-# 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.
+# -------------------------------------------------------------------------
 
 from collections import OrderedDict
 from dataclasses import dataclass
@@ -182,7 +183,7 @@ def analyze_diff_in_group(nodes_group):
     input_diff_nodes = list(filter(lambda node: node.is_diff, src_list))
     # 如果有异常回溯计算节点找到异常来源
     # 使用cpu模拟节点进行计算，查看结果是否有问题。需要对所有计算节点录入/映射，暂不实现。
-    get_compute_ops_from_comm_nodes(input_diff_nodes)
+    get_compute_ops_from_comm_nodes(nodes_group)
     # 筛选入参没问题但出参有问题的通信节点
     output_diff_nodes = list(filter(lambda node: node.data.is_diff, nodes_group))
     get_comm_ops(output_diff_nodes)

msprobe/core/compare/indicator_analysis/__init__.py

@@ -0,0 +1,15 @@
+# -------------------------------------------------------------------------
+#  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.
+# -------------------------------------------------------------------------