fix bug of step tag analyser

mindinsight/profiler/parser/step_trace_parser.py

@@ -65,7 +65,8 @@ class StepTraceParser:
             summary_info['total_steps'] = len(self._result) - 1
         print('\nStep trace summary info (unit: syscnt):')
         print(summary_info)
-        print('\nThe step trace parse result saves under ${summary_dir}/profiler/%s' % self.output_file)
+        print('\nThe step trace parse result saves under ${summary_dir}/profiler/%s'
+              % self.output_file)
 
     def parse_and_save(self):
         """Parse step trace files and save the result."""
@@ -131,10 +132,6 @@ class StepTraceParser:
             self._construct_event_info(next_event, event_info)
             if event_info.get('end'):
                 yield event_info
-                event_info = {
-                    'start': event_info.get('end'),
-                    'reduce': {}
-                }
 
     def _get_trace_struct(self, bin_info):
         """Translate event info to StepTraceStruct."""
@@ -146,7 +143,7 @@ class StepTraceParser:
     def _construct_event_info(self, next_event, event_info):
         """Construct event info according to next_event."""
         min_job_id = 255
-        step_flag: bool = lambda tag: tag > min_job_id
+        step_flag: bool = lambda tag: tag > min_job_id or tag == 0
         end_flag: bool = lambda tag: tag == min_job_id
         fp_flag: bool = lambda tag: tag == 1
         bp_flag: bool = lambda tag: tag == 2
@@ -154,10 +151,9 @@ class StepTraceParser:
         def _on_step_event():
             """Handle step event."""
             self._validate_tag_id(tag_id)
-            if event_info.get('start'):
-                event_info['end'] = sys_count
-            else:
-                event_info['start'] = sys_count
+            start_time = event_info.get('end', '-')
+            event_info.clear()
+            event_info['start'] = start_time
             event_info['reduce'] = {}
 
         def _on_reduce_event():
@@ -198,7 +194,8 @@ class StepTraceParser:
         if not (start_time and end_time and fp_time and bp_time):
             log.warning("The step %d is missing basic time.", self._step_num)
             return
-
+        if start_time == '-':
+            start_time = fp_time
         row_data = {
             'step_num': self._step_num,
             'start_point': start_time,
@@ -215,7 +212,7 @@ class StepTraceParser:
         # save the row data
         if not self._header:
             self._header = list(row_data.keys())
-        row_data_list = [row_data[header_name] for header_name in self._header]
+        row_data_list = [row_data.get(header_name, 0) for header_name in self._header]
         self._result.append(row_data_list)
 
     @staticmethod
@@ -236,10 +233,9 @@ class StepTraceParser:
     def _record_average_info(self):
         """Calculate average info."""
         result_size = len(self._result)
-        if result_size < 2:
-            return
         # calculate average data for each column in result data
         average_data = [0] * len(self._header)
+        if result_size >= 2:
             for row_info in self._result[1:]:
                 average_data = [
                     Decimal(i) + Decimal(j) for i, j in zip(row_info, average_data)