Fix small m for zbv

app.py

@@ -71,7 +71,7 @@ def calculate(p, m, f, b, w, c, mem):
   zb_bubble=percentage(zb_time/(f+b+w)/m - 1)
   zb_acceleration=percentage(baseline_time/zb_time - 1) if baseline_time is not None else None
 
-  if mem < p or m < 2 * p:
+  if mem < p:
     zbv_time=None
     zbv_bubble=None
     zbv_acceleration=None

v_schedule.py

@@ -145,7 +145,7 @@ class PipelineGraph(object):
                 end_tmp = cur_time[0] + self.fbw_cost[1]
                 continue
             tmp = end_tmp + self.c_cost
-            while count[i][0] + count[i][1] < count[i - 1][0] + count[i - 1][1]:
+            while count[i][0] + count[i][1] < count[i - 1][0] + count[i - 1][1] or count[i][1] <= count[i - 1][1] < self.n_micro:
                 for j in range(self.n_stage - 1, i - 1, -1):
                     if count[j][iter_chunk_] < self.n_micro:
                         put(0, iter_chunk_, j)
@@ -471,4 +471,4 @@ if __name__ == '__main__':
                 max_mem=mem_f * (p * 2 + m_offset),
             )
             graph.get_v_schedule()
-            break
+            break