diff options
| author | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2022-06-30 10:44:12 +0200 |
|---|---|---|
| committer | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2022-07-13 15:29:38 +0000 |
| commit | d03dc504452dbb32db383121f3dad81f4280bb3c (patch) | |
| tree | a491e706d5d18a099116b30b8d924667ff0f7245 /ethosu/vela/scheduler.py | |
| parent | a04f2f7322e7b83d93e875313d2e5b4d0dca94fb (diff) | |
| download | ethos-u-vela-d03dc504452dbb32db383121f3dad81f4280bb3c.tar.gz | |
MLBEDSW-6496 mlperf_deeplabv3_mnv2_ade20k_int8 fails at verify_output for u65
Added check to see if additional stripe data is needed from producer op
when cascading to make sure the stripes are not overwriting data still
being used. Also changed scheduler to make sure ResizeBilinear always
runs with even stripe height.
Signed-off-by: Fredrik Svedberg <fredrik.svedberg@arm.com>
Change-Id: If7d723e6be29575c2b55c400eebbe8275a1aa328
Diffstat (limited to 'ethosu/vela/scheduler.py')
| -rw-r--r-- | ethosu/vela/scheduler.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/ethosu/vela/scheduler.py b/ethosu/vela/scheduler.py index 7e989a7d..d01942bb 100644 --- a/ethosu/vela/scheduler.py +++ b/ethosu/vela/scheduler.py @@ -44,6 +44,7 @@ from . import weight_compressor from .architecture_allocator import ArchitectureBlockConfig from .architecture_allocator import find_block_config from .architecture_allocator import get_ifm_area_required +from .architecture_allocator import to_upscale from .architecture_features import ArchitectureFeatures from .architecture_features import Block from .cascade_builder import CascadeBuilder @@ -906,7 +907,8 @@ class Scheduler: striped_schedule.cost_map[sched_op] = cost # Calculate the preceeding Op's stripe - stripe = sched_op.ifm.shape.with_height(stripe.height * sched_op.kernel.stride.y) + height = stripe.height + stripe.height % to_upscale(sched_op.resampling_mode) + stripe = sched_op.ifm.shape.with_height(height * sched_op.kernel.stride.y) return striped_schedule |