Bug fix divide by zero (#10)

- Return load_balance as 0 in case computation_time is 0
neuronsimulator · Jun 5, 2019 · d3b40f9 · d3b40f9
1 parent b2c8384
commit d3b40f9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/ringtest.py b/ringtest.py
@@ -91,7 +91,10 @@ def prun(tstop):
 
     avg_comp_time = pc.allreduce(computation_time, 1) / settings.nhost
 
-    load_balance = avg_comp_time / max_cw_time
+    if max_cw_time == 0:
+        load_balance = 0
+    else:
+        load_balance = avg_comp_time / max_cw_time
 
     # spike communication time
     spk_time = (pc.allreduce(wait, 2), pc.allreduce(wait, 3))