aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/scheduler.py
diff options
context:
space:
mode:
Diffstat (limited to 'ethosu/vela/scheduler.py')
-rw-r--r--ethosu/vela/scheduler.py3
1 files changed, 3 insertions, 0 deletions
diff --git a/ethosu/vela/scheduler.py b/ethosu/vela/scheduler.py
index 6e2cd4a6..2174a6e3 100644
--- a/ethosu/vela/scheduler.py
+++ b/ethosu/vela/scheduler.py
@@ -880,6 +880,9 @@ class Scheduler:
for index, sched_op in enumerate(reversed(self.sched_ops)):
progress_print(verbose_progress, "Processing SchedulerOp", index, self.sched_ops)
min_stripe_height = prev_op.kernel.stride.y if prev_op else 1
+ if is_nearest(sched_op.resampling_mode):
+ # is nearest requires even stripes
+ min_stripe_height += min_stripe_height % 2
min_stripe = sched_op.ofm.shape.with_height(min_stripe_height)
cost = sched_op.create_scheduler_info(self.nng, min_stripe)