Fixup idle calculation

1 files changed, 6 insertions, 2 deletions

diff --git a/tpu.c b/tpu.c
index caa6730..3a62e11 100644
--- a/tpu.c
+++ b/tpu.c
@@ -770,7 +770,6 @@ tis_gen_stats(struct tis *tis)
 			stats.steps = MAX(stats.steps, tis->out_ports[i]->io_step + 1);
 		}
 	}
-	if (!stats.steps) stats.steps = tis->steps;
 
 	all_steps = 0;
 	idle_steps = 0;
@@ -784,7 +783,12 @@ tis_gen_stats(struct tis *tis)
 		}
 	}
 
-	stats.idle = (float) idle_steps / (float) all_steps;
+	if (!stats.steps) stats.steps = all_steps;
+
+	if (all_steps == 0)
+		stats.idle = 0;
+	else
+		stats.idle = (float) idle_steps / (float) all_steps;
 
 	return stats;
 }