aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--ethosu/vela/stats_writer.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/stats_writer.py b/ethosu/vela/stats_writer.py
index 22605a6b..f3e2dbac 100644
--- a/ethosu/vela/stats_writer.py
+++ b/ethosu/vela/stats_writer.py
@@ -277,7 +277,7 @@ def print_performance_metrics_for_strat(
n_cpu_operations = len(cpu_operations)
n_npu_operations = len(npu_operations)
- n_total_operations = n_cpu_operations + n_npu_operations
+ n_total_operations = max(n_cpu_operations + n_npu_operations, 1) # avoid potential divide by zero
def format_tens_list(lst):
return " ".join(str(list(tens.shape)) for tens in lst)