diff --git a/mindinsight/profiler/parser/framework_parser.py b/mindinsight/profiler/parser/framework_parser.py index d8331faa58ec422071e1a9925d44add6a35e99f0..a8e80d72e6cbc5286ec72df0a420ede531282f6e 100644 --- a/mindinsight/profiler/parser/framework_parser.py +++ b/mindinsight/profiler/parser/framework_parser.py @@ -435,9 +435,14 @@ class FrameworkParser: for graph_info in graph_file: result = self._parse_one_row_graph_info(graph_info) task_info = task_cache.get(result[0]) - task_info.extend(result) - csv_writer.writerow(task_info) - del task_cache[result[0]] + if task_info: + task_info.extend(result) + csv_writer.writerow(task_info) + del task_cache[result[0]] + else: + save_info = [None, None, None] + save_info.extend(result) + csv_writer.writerow(save_info) none_list = [None, None, None, None] for key, value in task_cache.items():