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

tb_graph_ascend/server/app/utils/i18n.py

@@ -0,0 +1,153 @@
+# -------------------------------------------------------------------------
+#  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.
+# -------------------------------------------------------------------------
+
+# 语言切换
+ZH_CN = "zh-CN"
+EN = "en"
+language = {
+    ZH_CN: {
+        'typeError': '文件类型错误',
+        'searchTypeError': '搜索类型错误',
+        'nodeInfoError': '查询参数nodeInfo及其子项必须存在且必须为字典',
+        'metaDataError': '查询参数“metaData”及其子项必须存在且必须为字典',
+        'dbInitError': '数据库初始化失败',
+        'loadConfigError': '加载图配置信息失败',
+        'rankStepNullError': 'rank或step或micro_step为空',
+        'loadNodeError': '加载图所有节点列表失败',
+        'expandNodeError': '节点展开或收起发生错误',
+        'searchNodeError': '节点搜索发生错误',
+        'graphTypeError': '图类型错误,仅支持 "NPU" 或 "BENCH"',
+        'getNodeInfoError': '获取节点信息失败',
+        'taskTypeError': '任务类型不支持',
+        'matchNodeError': '匹配节点失败',
+        'loadConfigFileError': '读取配置文件失败',
+        'matchNodeLinksNullError': '匹配配置文件为空',
+        'parameterError': '参数错误',   
+        'nodesTypeError': '节点类型不一致, 无法添加匹配关系',
+        'taskTypeWrong': 'task类型错误',
+        'nodeNotExist': '节点不存在',
+        'ioEmptyError': '输入或输出统计误差值为空',
+        'outputEmptyError': '输出统计误差值为空，计算精度误差失败',
+        'operationFailed': '操作失败：',
+        'or': '或',
+        'nodeUnmatchError': '节点未匹配，请先匹配节点',
+        'deleteMatchNodeFailed': '删除匹配节点失败',
+        'noUpdateNodesFound': '未找到可更新的节点',
+        'updateNodeAccuracyFailed': '更新节点精度失败',
+        'updateDatabaseFailed': '更新数据库失败',
+        'updateColorFailed': '更新颜色失败',
+        'saveMatchRelationshipFailed': '保存匹配关系失败',
+        'selectedNodeCantMatched': '选择的节点不可匹配',
+        'loadErrorBySecurityIssue': '文件存在安全问题，读取文件失败',
+        'loadConfigFailed': '获取配置信息失败, 请检查目录中第一个文件',
+        'debug': '调试侧',
+        'bench': '标杆侧',
+        'obtainNodesAccuracyFailed': '获取符合精度误差节点失败',
+        'notSupportOverflow': '多图模式下不支持溢出检测',
+        'obtainNodesOverflowFailed': '获取符合溢出检测节点失败',
+        'updatePrecisionFailed': '更新精度误差失败',
+        'nodeTypeError': '节点类型错误',
+        'readFileFailed': '读取文件失败',
+        'parameterEmpty': '参数为空',
+        'saveFileFailed': '保存文件失败',
+        'dbConncetionNotInit': '数据库链接未初始化',
+        'parseJsonFailed': '解析JSON失败',
+        'getNodeListFailed': '获取节点列表失败',
+        'initGraphJsonFailed': '初始化json图数据失败',
+        'colorsNotObject': 'colors 必须是一个对象',
+        'colotsEmpty': 'colors 不能为空',
+        'illegalColorKey': '非法颜色键: ',
+        'colorValuesNotObject': '颜色值必须是对象: ',
+        'missingValueField': '缺少 value 字段: ',
+        'notArrayOfLength2': 'value 必须是长度为2的数组: ',
+        'notArrayConsistNumbers': 'value 数组必须是数字: ',
+        'invalidValueRange': 'value 区间无效（左 >= 右）: ',
+        'unsupportedValue': '不支持的 value 字符串值: ',
+        'valueTypeError': 'value 必须是数字数组或字符串: ',
+        'descriptionError': 'description 包含恶意内容或格式错误: ',
+        'circularReferenceError1': '检测到循环引用：节点 ',
+        'circularReferenceError2': ' 已存在于路径中',
+        'invalidTag': '无效标签: ',
+        'pathMayNotInSecureDirectory': '该路径可能不在安全目录内',
+        'serverError': '服务异常，请稍后重试'
+    },
+    EN: {
+        'typeError': 'File type error',
+        'searchTypeError': 'Search type error',
+        'nodeInfoError': 'The query parameter nodeInfo and its sub-items must exist and must be a dictionary.',
+        'metaDataError': 'The query parameter “metaData” and its sub-items must exist and must be a dictionary.',
+        'dbInitError': 'Database initialization failed',
+        'loadConfigError': 'load graph config info failed',
+        'rankStepNullError': 'rank or step or micro_step is null',
+        'loadNodeError': 'load graph all node list failed',
+        'expandNodeError': 'node expand or collapse error',
+        'searchNodeError': 'node search error',
+        'graphTypeError': 'graph type error, only support "NPU" or "BENCH"',
+        'getNodeInfoError': 'get node info failed',
+        'taskTypeError': 'task type not support',
+        'matchNodeError': 'match node failed',
+        'loadConfigFileError': 'load config file failed',
+        'parameterError': 'Parameter Error',
+        'nodesTypeError': 'Node types are inconsistent; matching relationships cannot be added.',
+        'taskTypeWrong': 'Task type error',
+        'nodeNotExist': 'Node does not exist',
+        'ioEmptyError': 'The input or output statistical error value is empty.',
+        'outputEmptyError': 'The output statistical error value is empty, and the calculation precision error failed.',
+        'operationFailed': 'Operation failed: ',
+        'or': 'or',
+        'nodeUnmatchError': 'Node not matched. Please match the node first.',
+        'deleteMatchNodeFailed': 'Failed to delete matching node',
+        'noUpdateNodesFound': 'No nodes found to update.',
+        'updateNodeAccuracyFailed': 'Failed to update node accuracy',
+        'updateDatabaseFailed': 'Failed to update the database',
+        'updateColorFailed': 'Color update failed',
+        'saveMatchRelationshipFailed': 'Failed to save matching relationship',
+        'selectedNodeCantMatched': 'The selected node cannot be matched.',
+        'loadErrorBySecurityIssue': 'The file has security issues, and reading the file failed.',
+        'loadConfigFailed': 'Failed to retrieve configuration information. Please check the first file in directory.',
+        'debug': 'Debug',
+        'bench': 'Bench',
+        'expandOrCollapseError': 'An error occurred while expanding or collapsing the node.',
+        'obtainNodesAccuracyFailed': 'Failed to obtain nodes meeting accuracy error requirements',
+        'notSupportOverflow': 'Overflow detection is not supported in multi-image mode.',
+        'obtainNodesOverflowFailed': 'Failed to obtain overflow detection node',
+        'updatePrecisionFailed': 'Update precision error failure',
+        'nodeTypeError': 'Node type error',
+        'readFileFailed': 'Failed to read file',
+        'parameterEmpty': 'The parameter is empty.',
+        'saveFileFailed': 'Failed to save file',
+        'dbConncetionNotInit': 'database connection not init',
+        'parseJsonFailed': 'Failed to parse JSON',
+        'getNodeListFailed': 'Failed to get node list',
+        'initGraphJsonFailed': 'initlize graph json failed',
+        'colorsNotObject': 'Colors must be an object',
+        'colotsEmpty': 'Colors cannot be empty',
+        'illegalColorKey': 'Illegal color key: ',
+        'colorValuesNotObject': 'Color values must be objects: ',
+        'missingValueField': 'Missing value field: ',
+        'notArrayOfLength2': 'The value must be an array of length 2: ',
+        'notArrayConsistNumbers': 'Value array must consist of numbers: ',
+        'invalidValueRange': 'Invalid value range (left >= right): ',
+        'unsupportedValue': 'Unsupported value string values: ',
+        'valueTypeError': 'Value must be a numeric array or string: ',
+        'descriptionError': 'Description contains malicious content or formatting errors: ',
+        'circularReferenceError1': 'Detected circular reference: node ',
+        'circularReferenceError2': ' already exists in the path',
+        'invalidTag': 'Invalid tag: ',
+        'pathMayNotInSecureDirectory': 'The path may not be within a secure directory',
+        'serverError': 'server error, please try again later'
+    }
+}

tb_graph_ascend/server/app/utils/request_method.py

@@ -0,0 +1,46 @@
+# -------------------------------------------------------------------------
+#  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 json
+from werkzeug.wrappers.request import Request
+from werkzeug import Response
+from ..utils.graph_utils import GraphUtils
+from ..utils.constant import security_headers
+
+
+def request_method(allowed_method):
+    def decorator(func):
+        def wrapper(*args, **kwargs):
+            try:
+                if len(args) <= 0:
+                    raise RuntimeError('Illegal function call, at least 1 parameter is required but got 0')
+                request = args[0]
+                if not isinstance(request, Request):
+                    raise RuntimeError('The request "parameter" is not in a format supported by werkzeug')
+                actual_method = request.method.upper()
+                expected_method = allowed_method.upper()
+                if actual_method != expected_method:
+                    raise RuntimeError(f"Method Not Allowed: expected '{expected_method}', got '{actual_method}'")
+            except Exception as e:
+                result = {'success': False, 'error': str(e)}
+                return Response(json.dumps(result), content_type="application/json", headers=security_headers)
+            try:
+                return func(*args, **kwargs)
+            except Exception as e:
+                result = {'success': False, 'error': GraphUtils.t('serverError')}
+                return Response(json.dumps(result), content_type="application/json", headers=security_headers)
+        return wrapper
+    return decorator

tb_graph_ascend/server/app/views/__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.
+# -------------------------------------------------------------------------

tb_graph_ascend/server/app/views/graph_views.py

@@ -0,0 +1,304 @@
+# -------------------------------------------------------------------------
+#  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
+import json
+from pathlib import Path
+from werkzeug import wrappers, Response, exceptions
+from ..utils.request_method import request_method
+from ..service import ServiceFactory, GraphServiceStrategy
+from ..utils.file_check_wrapper import check_file_type
+from ..utils.graph_utils import GraphUtils
+from ..utils.constant import DataType
+from ..utils.global_state import GraphState
+from ..utils.constant import security_headers
+
+
+class GraphView:
+    service_factory = ServiceFactory()
+
+    # 静态文件路由
+    @staticmethod
+    @wrappers.Request.application
+    def static_file_route(request):
+        filename = os.path.basename(request.path)
+
+        extension = os.path.splitext(filename)[1]
+        if extension == '.html':
+            content_type = 'text/html'
+        elif extension == '.js':
+            content_type = 'application/javascript'
+        else:
+            content_type = 'application/octet-stream'
+
+        try:
+            # 添加白名单校验
+            if filename != 'index.html' and filename != 'index.js':
+                raise exceptions.NotFound('404 Not Found') from e
+            current_dir = Path(__file__).resolve().parent
+            server_dir = current_dir.parent.parent
+            dir_path = (server_dir / "static").resolve()
+            filepath = (server_dir / "static" / filename).resolve()  # resolve() 规范化路径
+            if not GraphUtils.is_relative_to(filepath, dir_path):
+                raise exceptions.NotFound('404 Not Found')
+            # 前端打包后产生的内部输入文件，不需要要安全校验
+            with open(filepath, 'rb') as infile:
+                contents = infile.read()
+        except IOError as e:
+            raise exceptions.NotFound('404 Not Found') from e
+        return Response(contents, content_type=content_type, headers=security_headers)
+
+    @staticmethod
+    @wrappers.Request.application
+    @request_method('GET')
+    def load_meta_dir(request):
+        """Scan logdir for directories containing .vis files, modified to return a tuple of (run, tag)."""
+        result = GraphServiceStrategy.load_meta_dir()
+        return Response(json.dumps(result), content_type="application/json", headers=security_headers)
+
+    # 读取当前图数据
+    @staticmethod
+    @wrappers.Request.application
+    @request_method('GET')
+    def load_graph_data(request):
+        meta_data = {
+            'run': request.args.get('run'),
+            'tag': request.args.get('tag'),
+            'type': request.args.get('type'),
+        }
+        lang = request.args.get('lang')
+        GraphState.set_global_value('lang', lang)
+        strategy = GraphView._get_strategy(meta_data)
+        if meta_data.get('type') == DataType.DB.value:
+            result = strategy.load_graph_data()
+            return Response(result, content_type="application/json", headers=security_headers)
+        elif meta_data.get('type') == DataType.JSON.value:
+            return Response(
+                strategy.load_graph_data(),
+                mimetype='text/event-stream',
+                headers={
+                    'Cache-Control': 'no-cache',
+                    'Connection': 'close',  # TCP链接不复用，请求结束释放资源
+                } | security_headers
+            )
+        else: 
+            result = {'success': False, 'error': GraphUtils.t('typeError')}
+            return Response(json.dumps(result), content_type="application/json", headers=security_headers)
+
+    # 获取当前图数据配置信息
+    @staticmethod
+    @wrappers.Request.application
+    @check_file_type
+    @request_method('POST')
+    def load_graph_config_info(request):
+        data = GraphUtils.safe_json_loads(request.get_data().decode('utf-8'), {})
+        meta_data = data.get('metaData')
+        strategy = GraphView._get_strategy(meta_data)
+        result = strategy.load_graph_config_info()
+        # 创建响应对象
+        response = Response(json.dumps(result), content_type="application/json", headers=security_headers)
+        return response
+
+    # 获取当前图所有节点列表
+    @staticmethod
+    @wrappers.Request.application
+    @check_file_type
+    @request_method('POST')
+    def load_graph_all_node_list(request):
+        data = GraphUtils.safe_json_loads(request.get_data().decode('utf-8'), {})
+        meta_data = data.get('metaData')
+        strategy = GraphView._get_strategy(meta_data)
+        result = strategy.load_graph_all_node_list(meta_data)
+        response = Response(json.dumps(result), content_type="application/json", headers=security_headers)
+        return response          
+
+    # 根据精度误差搜索节点
+    @staticmethod
+    @wrappers.Request.application
+    @check_file_type
+    @request_method('POST')
+    def search_node(request):
+        data = GraphUtils.safe_json_loads(request.get_data().decode('utf-8'))
+        meta_data = data.get("metaData")
+        search_type = data.get("type")
+        values = data.get("values")
+        strategy = GraphView._get_strategy(meta_data)
+        if search_type == 'precision':
+            result = strategy.search_node_by_precision(meta_data, values)
+        elif search_type == 'overflow':
+            result = strategy.search_node_by_overflow(meta_data, values)
+        else:
+            result = {
+                'success': False,
+                'error': GraphUtils.t('searchTypeError')
+            }
+        return Response(json.dumps(result), content_type="application/json", headers=security_headers) 
+
+    # 更新误差节点
+    @staticmethod
+    @wrappers.Request.application
+    @check_file_type
+    @request_method('POST')
+    def update_precision_error(request):
+        data = GraphUtils.safe_json_loads(request.get_data().decode('utf-8'))
+        meta_data = data.get('metaData')
+        filter_value = data.get("filterValue")
+        strategy = GraphView._get_strategy(meta_data)
+        result = strategy.update_precision_error(meta_data, filter_value)
+        return Response(json.dumps(result), content_type="application/json", headers=security_headers)
+
+    # 展开关闭节点
+    @staticmethod
+    @wrappers.Request.application
+    @check_file_type
+    @request_method('POST')
+    def change_node_expand_state(request):
+        data = GraphUtils.safe_json_loads(request.get_data().decode('utf-8'), {})
+        node_info = GraphUtils.safe_get_node_info(data)
+        result = {'success': False, 'error': ''}
+        meta_data = data.get('metaData')
+        if node_info is None or not isinstance(node_info, dict):
+            result['error'] = GraphUtils.t('nodeInfoError') 
+        strategy = GraphView._get_strategy(meta_data)
+        hierarchy = strategy.change_node_expand_state(node_info, meta_data)
+        return Response(json.dumps(hierarchy), content_type="application/json", headers=security_headers)
+
+    # 更新当前图节点信息
+    @staticmethod
+    @wrappers.Request.application
+    @check_file_type
+    @request_method('POST')
+    def update_hierarchy_data(request):
+        data = GraphUtils.safe_json_loads(request.get_data().decode('utf-8'), {})
+        graph_type = data.get("graphType")
+        meta_data = data.get('metaData')
+        strategy = GraphView._get_strategy(meta_data)
+        hierarchy = strategy.update_hierarchy_data(graph_type)
+        return Response(json.dumps(hierarchy), content_type="application/json", headers=security_headers)
+
+    # 获取当前节点对应节点的信息看板数据
+    @staticmethod
+    @wrappers.Request.application
+    @check_file_type
+    @request_method('POST')
+    def get_node_info(request):
+        data = GraphUtils.safe_json_loads(request.get_data().decode('utf-8'), {})
+        node_info = GraphUtils.safe_get_node_info(data)
+        result = {'success': False, 'error': ''}
+        meta_data = data.get('metaData')
+        if node_info is None or not isinstance(node_info, dict):
+            result['error'] = GraphUtils.t('nodeInfoError') 
+            return Response(result, content_type="application/json", headers=security_headers)
+        strategy = GraphView._get_strategy(meta_data)
+        node_detail = strategy.get_node_info(node_info, meta_data)
+        return Response(json.dumps(node_detail), content_type="application/json", headers=security_headers)
+
+    # 根据配置文件添加匹配节点
+    @staticmethod
+    @wrappers.Request.application
+    @check_file_type
+    @request_method('POST')
+    def add_match_nodes_by_config(request):
+        data = GraphUtils.safe_json_loads(request.get_data().decode('utf-8'), {})
+        config_file = data.get('configFile')
+        meta_data = data.get('metaData')
+        strategy = GraphView._get_strategy(meta_data)
+        match_result = strategy.add_match_nodes_by_config(config_file, meta_data)
+        return Response(json.dumps(match_result), content_type="application/json", headers=security_headers)
+
+    # 添加匹配节点
+    @staticmethod
+    @wrappers.Request.application
+    @check_file_type
+    @request_method('POST')
+    def add_match_nodes(request):
+        data = GraphUtils.safe_json_loads(request.get_data().decode('utf-8'), {})
+        npu_node_name = data.get("npuNodeName")
+        bench_node_name = data.get("benchNodeName")
+        meta_data = data.get('metaData')
+        is_match_children = data.get("isMatchChildren")
+        strategy = GraphView._get_strategy(meta_data)
+        match_result = strategy.add_match_nodes(npu_node_name, bench_node_name, meta_data, is_match_children)
+        return Response(json.dumps(match_result), content_type="application/json", headers=security_headers)
+
+    # 取消节点匹配
+    @staticmethod
+    @wrappers.Request.application
+    @check_file_type
+    @request_method('POST')
+    def delete_match_nodes(request):
+        data = GraphUtils.safe_json_loads(request.get_data().decode('utf-8'), {})
+        npu_node_name = data.get("npuNodeName")
+        bench_node_name = data.get("benchNodeName")
+        meta_data = data.get('metaData')
+        is_unmatch_children = data.get("isUnMatchChildren")
+        strategy = GraphView._get_strategy(meta_data)
+        match_result = strategy.delete_match_nodes(npu_node_name, bench_node_name, meta_data, is_unmatch_children)
+        return Response(json.dumps(match_result), content_type="application/json", headers=security_headers)
+
+    # 保存匹配节点列表
+    @staticmethod
+    @wrappers.Request.application
+    @check_file_type
+    @request_method('POST')
+    def save_data(request):
+        data = GraphUtils.safe_json_loads(request.get_data().decode('utf-8'), {})
+        meta_data = data.get('metaData')
+        strategy = GraphView._get_strategy(meta_data)
+        save_result = strategy.save_data(meta_data)
+        return Response(json.dumps(save_result), content_type="application/json", headers=security_headers)
+
+    # 更新颜色信息
+    @staticmethod
+    @wrappers.Request.application
+    @check_file_type
+    @request_method('POST')
+    def update_colors(request):
+        data = GraphUtils.safe_json_loads(request.get_data().decode('utf-8'), {})
+        meta_data = data.get('metaData')
+        colors = GraphUtils.safe_json_loads(data.get('colors'))
+        success, error_msg, colors = GraphUtils.validate_colors_param(colors)
+        if not success:
+            result = {
+                'success': False,
+                'error': error_msg
+            }
+            return Response(json.dumps(result), content_type="application/json", headers=security_headers)
+        strategy = GraphView._get_strategy(meta_data, no_tag=True)
+        update_result = strategy.update_colors(colors)
+        return Response(json.dumps(update_result), content_type="application/json", headers=security_headers)
+
+    # 保存匹配关系
+    @staticmethod
+    @wrappers.Request.application
+    @check_file_type
+    @request_method('POST')
+    def save_matched_relations(request):
+        data = GraphUtils.safe_json_loads(request.get_data().decode('utf-8'), {})
+        meta_data = data.get('metaData')
+        strategy = GraphView._get_strategy(meta_data)
+        save_result = strategy.save_matched_relations(meta_data)
+        return Response(json.dumps(save_result), content_type="application/json", headers=security_headers)
+
+    @staticmethod
+    def _get_strategy(meta_data, no_tag=False):
+        run = meta_data.get('run')
+        data_type = meta_data.get('type')
+        if no_tag:
+            return GraphView.service_factory.create_strategy_without_tag(data_type, run)
+        
+        tag = meta_data.get('tag')
+        return GraphView.service_factory.create_strategy(data_type, run, tag)

tb_graph_ascend/server/plugin.py

@@ -0,0 +1,108 @@
+# Copyright 2017 The TensorFlow Authors. 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.
+#
+# Copyright (c) 2025, Huawei Technologies.
+# Adapt to the model hierarchical visualization data collected by the msprobe tool
+# ==============================================================================
+
+"""The TensorBoard Graphs plugin."""
+
+import os
+from tensorboard.plugins import base_plugin
+from tensorboard.util import tb_logging
+
+from .app.views.graph_views import GraphView
+from .app.utils.graph_utils import GraphUtils
+from .app.utils.global_state import GraphState
+from .app.utils.constant import Extension
+
+logger = tb_logging.get_logger()
+
+PLUGIN_NAME = 'graph_ascend'
+PLUGIN_NAME_RUN_METADATA_WITH_GRAPH = 'graph_ascend_run_metadata_graph'
+DB_EXT = Extension.DB.value
+JSON_EXT = Extension.JSON.value
+
+
+class GraphsPlugin(base_plugin.TBPlugin):
+    """Graphs Plugin for TensorBoard."""
+
+    plugin_name = PLUGIN_NAME
+
+    def __init__(self, context):
+        """Instantiates GraphsPlugin via TensorBoard core.
+
+        Args:
+          context: A base_plugin.TBContext instance.
+        """
+        super().__init__(context)
+        GraphState.reset_global_state()
+        self.logdir = os.path.abspath(os.path.expanduser(context.logdir.rstrip('/')))
+        # 将logdir赋值给global_state中的logdir属性,方便其他模块使用
+        GraphState.set_global_value('logdir', self.logdir)
+
+    def get_plugin_apps(self):
+        return {     
+            '/index.js': GraphView.static_file_route,
+            '/index.html': GraphView.static_file_route,
+            "/load_meta_dir": GraphView.load_meta_dir,
+            "/screen": GraphView.search_node,
+            '/loadGraphData': GraphView.load_graph_data,
+            '/loadGraphConfigInfo': GraphView.load_graph_config_info,
+            '/loadGraphAllNodeList': GraphView.load_graph_all_node_list,
+            '/changeNodeExpandState': GraphView.change_node_expand_state,
+            '/updateHierarchyData': GraphView.update_hierarchy_data,
+            '/getNodeInfo': GraphView.get_node_info,
+            '/addMatchNodes': GraphView.add_match_nodes,
+            '/addMatchNodesByConfig': GraphView.add_match_nodes_by_config,
+            '/deleteMatchNodes': GraphView.delete_match_nodes,
+            '/saveData': GraphView.save_data,
+            '/updateColors': GraphView.update_colors,
+            '/saveMatchedRelations': GraphView.save_matched_relations,
+            '/updatePrecisionError': GraphView.update_precision_error,
+        }
+
+    def is_active(self):
+        """The graphs plugin is active if any run has a graph."""
+
+        def _is_vis(path, file_name):
+            return os.path.isfile(path) and (file_name.endswith(DB_EXT) or file_name.endswith(JSON_EXT))
+        
+        _, error = GraphUtils.safe_check_load_file_path(self.logdir, True)
+        if error:
+            return False
+        for content in os.listdir(self.logdir):
+            content_path = os.path.join(self.logdir, content)
+            if _is_vis(content_path, content):
+                return True
+            if os.path.isdir(content_path):
+                _, error = GraphUtils.safe_check_load_file_path(content_path, True)
+                if error:
+                    continue
+                for file in os.listdir(content_path):
+                    if _is_vis(os.path.join(content_path, file), file):
+                        return True
+        return False
+
+    def data_plugin_names(self):
+        return (
+            PLUGIN_NAME,
+            PLUGIN_NAME_RUN_METADATA_WITH_GRAPH,
+        )
+
+    def frontend_metadata(self):
+        return base_plugin.FrontendMetadata(
+            es_module_path='/index.js',
+            disable_reload=True,
+        )