From 3a9154d7c598a9aadbd6d120bdb50cde75ea19d9 Mon Sep 17 00:00:00 2001 From: l30044004 Date: Tue, 1 Jul 2025 15:31:11 +0800 Subject: [PATCH] =?UTF-8?q?=E5=88=86=E7=BA=A7=E5=8F=AF=E8=A7=86=E5=8C=96?= =?UTF-8?q?=E6=8B=A6=E6=88=AAconstruct.json=E4=B8=BA=E7=A9=BA=E7=9A=84?= =?UTF-8?q?=E5=9C=BA=E6=99=AF?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../test/visualization_ut/input/step0/rank0/construct.json | 5 ++++- .../input_format_correct/step0/rank0/construct.json | 5 ++++- .../input_format_correct/step0/rank1/construct.json | 5 ++++- .../input_format_correct/step1/rank0/construct.json | 5 ++++- .../input_format_correct/step2/rank0/construct.json | 5 ++++- .../msprobe/visualization/builder/graph_builder.py | 6 ++++++ 6 files changed, 26 insertions(+), 5 deletions(-) diff --git a/debug/accuracy_tools/msprobe/test/visualization_ut/input/step0/rank0/construct.json b/debug/accuracy_tools/msprobe/test/visualization_ut/input/step0/rank0/construct.json index 0967ef424b..f38780de74 100644 --- a/debug/accuracy_tools/msprobe/test/visualization_ut/input/step0/rank0/construct.json +++ b/debug/accuracy_tools/msprobe/test/visualization_ut/input/step0/rank0/construct.json @@ -1 +1,4 @@ -{} +{ + "Tensor1": "Module1", + "Module1": null +} diff --git a/debug/accuracy_tools/msprobe/test/visualization_ut/input_format_correct/step0/rank0/construct.json b/debug/accuracy_tools/msprobe/test/visualization_ut/input_format_correct/step0/rank0/construct.json index 0967ef424b..f38780de74 100644 --- a/debug/accuracy_tools/msprobe/test/visualization_ut/input_format_correct/step0/rank0/construct.json +++ b/debug/accuracy_tools/msprobe/test/visualization_ut/input_format_correct/step0/rank0/construct.json @@ -1 +1,4 @@ -{} +{ + "Tensor1": "Module1", + "Module1": null +} diff --git a/debug/accuracy_tools/msprobe/test/visualization_ut/input_format_correct/step0/rank1/construct.json b/debug/accuracy_tools/msprobe/test/visualization_ut/input_format_correct/step0/rank1/construct.json index 0967ef424b..f38780de74 100644 --- a/debug/accuracy_tools/msprobe/test/visualization_ut/input_format_correct/step0/rank1/construct.json +++ b/debug/accuracy_tools/msprobe/test/visualization_ut/input_format_correct/step0/rank1/construct.json @@ -1 +1,4 @@ -{} +{ + "Tensor1": "Module1", + "Module1": null +} diff --git a/debug/accuracy_tools/msprobe/test/visualization_ut/input_format_correct/step1/rank0/construct.json b/debug/accuracy_tools/msprobe/test/visualization_ut/input_format_correct/step1/rank0/construct.json index 0967ef424b..f38780de74 100644 --- a/debug/accuracy_tools/msprobe/test/visualization_ut/input_format_correct/step1/rank0/construct.json +++ b/debug/accuracy_tools/msprobe/test/visualization_ut/input_format_correct/step1/rank0/construct.json @@ -1 +1,4 @@ -{} +{ + "Tensor1": "Module1", + "Module1": null +} diff --git a/debug/accuracy_tools/msprobe/test/visualization_ut/input_format_correct/step2/rank0/construct.json b/debug/accuracy_tools/msprobe/test/visualization_ut/input_format_correct/step2/rank0/construct.json index 0967ef424b..f38780de74 100644 --- a/debug/accuracy_tools/msprobe/test/visualization_ut/input_format_correct/step2/rank0/construct.json +++ b/debug/accuracy_tools/msprobe/test/visualization_ut/input_format_correct/step2/rank0/construct.json @@ -1 +1,4 @@ -{} +{ + "Tensor1": "Module1", + "Module1": null +} diff --git a/debug/accuracy_tools/msprobe/visualization/builder/graph_builder.py b/debug/accuracy_tools/msprobe/visualization/builder/graph_builder.py index f56c6f9a10..538683862c 100644 --- a/debug/accuracy_tools/msprobe/visualization/builder/graph_builder.py +++ b/debug/accuracy_tools/msprobe/visualization/builder/graph_builder.py @@ -19,6 +19,7 @@ from dataclasses import dataclass from msprobe.core.common.const import Const from msprobe.core.common.file_utils import load_json, save_json from msprobe.core.common.utils import load_stack_json +from msprobe.core.common.log import logger from msprobe.visualization.builder.msprobe_adapter import get_input_output from msprobe.visualization.builder.msprobe_adapter import op_patterns from msprobe.visualization.graph.graph import Graph @@ -45,6 +46,11 @@ class GraphBuilder: Returns: Graph，代表图的数据结构 """ construct_dict = load_json(construct_path) + if not construct_dict: + logger.error("The content of 'construct.json' is empty, failed to build graph. " + "When dumping data, it is necessary to select level L0 or mix in order to " + "collect model structure data, that is, the content of 'construct.json' is not empty.") + raise RuntimeError dump_dict = load_json(data_path) stack_dict = load_stack_json(stack_path) if not complete_stack: -- Gitee