MLBEDSW-6296: Updated condition for the opt size weight buffering schedule
Allow schedule do be used when calculations says zero total improvement
but calculations on the other hand shows there are dram improvement.
When testing on real target, total performance is improvement.
Signed-off-by: Johan Alfven <johan.alfven@arm.com>
Change-Id: Ib4f2a37710dc7954b72b48c38fce4817ccd7187b
diff --git a/ethosu/vela/scheduler.py b/ethosu/vela/scheduler.py
index 67b890e..bc05876 100644
--- a/ethosu/vela/scheduler.py
+++ b/ethosu/vela/scheduler.py
@@ -1077,7 +1077,7 @@
improvement_dram = round((default_dram_cycles - new_dram_cycles) / default_dram_cycles, 2)
# Compare both total and dram improvement
- if not (improvement_tot > 0 and improvement_dram > 0):
+ if not (improvement_tot >= 0 and improvement_dram > 0):
# No improvement, restore the default schedule
for sched_op in self.sched_ops:
sched_op.evicted_fms_size = 0