diff options
| -rw-r--r-- | ethosu/vela/cascade_builder.py | 1 | ||||
| -rw-r--r-- | ethosu/vela/scheduler.py | 3 |
2 files changed, 3 insertions, 1 deletions
diff --git a/ethosu/vela/cascade_builder.py b/ethosu/vela/cascade_builder.py index 95872cfe..01bbf3bf 100644 --- a/ethosu/vela/cascade_builder.py +++ b/ethosu/vela/cascade_builder.py @@ -101,7 +101,6 @@ class CascadeBuilder: and sched_op.parent_op.read_offsets[0] is None and sched_op.parent_op.read_offsets[1] is None and self.elementwise_cascadable(sched_op) - and not sched_op.parent_op.type.is_resize_op() and not sched_op.parent_op.type == Op.Conv2DBackpropInputSwitchedBias and sched_op.parent_op.attrs.get("padding", None) != Padding.TILE ) 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) |