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 0967ef424bce6791893e9a57bb952f80fd536e93..f38780de744675a62cee03c58fb4682448c210a6 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 0967ef424bce6791893e9a57bb952f80fd536e93..f38780de744675a62cee03c58fb4682448c210a6 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 0967ef424bce6791893e9a57bb952f80fd536e93..f38780de744675a62cee03c58fb4682448c210a6 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 0967ef424bce6791893e9a57bb952f80fd536e93..f38780de744675a62cee03c58fb4682448c210a6 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 0967ef424bce6791893e9a57bb952f80fd536e93..f38780de744675a62cee03c58fb4682448c210a6 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 f56c6f9a10e79f834e4f4e96aa5c242cc69d7ac4..538683862c5fc5d0ee2c9826faa2ada112075952 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: