Update dist_inference.py

microsoft · Dec 14, 2023 · 06e65ad · 06e65ad
1 parent c50be8e
commit 06e65ad
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/superbench/benchmarks/micro_benchmarks/dist_inference.py b/superbench/benchmarks/micro_benchmarks/dist_inference.py
@@ -497,6 +497,7 @@ def _process_raw_result(self, cmd_idx, raw_output):
             for output_line in output_lines:
                 if output_line.startswith('Latency of step'):
                     step_times.append(float(output_line.split(' ms')[0].split()[-1]))
+            assert (self._args.num_steps == len(step_times))
             return self._process_numeric_result(
                 'step_times', step_times, reduce_type=ReduceType.MAX, cal_percentile=True
             )