diff options
Diffstat (limited to 'ethosu/vela/stats_writer.py')
| -rw-r--r-- | ethosu/vela/stats_writer.py | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/ethosu/vela/stats_writer.py b/ethosu/vela/stats_writer.py index c90d9876..af7b6997 100644 --- a/ethosu/vela/stats_writer.py +++ b/ethosu/vela/stats_writer.py @@ -85,7 +85,10 @@ def write_summary_metrics_csv(nng, summary_filename, arch): ) midpoint_inference_time = nng.cycles[PassCycles.Total] / arch.npu_clock - midpoint_fps = 1 / midpoint_inference_time + if midpoint_inference_time > 0: + midpoint_fps = 1 / midpoint_inference_time + else: + midpoint_fps = np.nan n_passes = sum(len(sg.passes) for sg in nng.subgraphs) n_cascaded_passes = sum(len(sg.cascaded_passes) for sg in nng.subgraphs) @@ -231,7 +234,10 @@ def print_performance_metrics_for_strat( orig_mem_areas_labels = [(v, v.display_name()) for v in MemArea.all()] midpoint_inference_time = cycles[PassCycles.Total] / arch.npu_clock - midpoint_fps = 1 / midpoint_inference_time + if midpoint_inference_time > 0: + midpoint_fps = 1 / midpoint_inference_time + else: + midpoint_fps = np.nan mem_area_labels = [ (mem_area, label) for mem_area, label in orig_mem_areas_labels if np.sum(bandwidths[mem_area]) > 0 |