diff options
author | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2021-09-29 10:08:04 +0200 |
---|---|---|
committer | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2021-10-01 07:56:18 +0000 |
commit | 3ff7a4aed60fefababdced93a5aba91409a4050c (patch) | |
tree | 1dfd999beb51fec093d6ce4eb6f68f7bca41b32d /ethosu/vela/scheduler.py | |
parent | 008cd10f6f5b8838de1f37c8c42899ace1c1cf0d (diff) | |
download | ethos-u-vela-3ff7a4aed60fefababdced93a5aba91409a4050c.tar.gz |
MLBEDSW-5013 Output diff for u55-bring-up tests, int16
Fixed output diff for some architectures due to incorrect IFM buffer size
calculation when using NearestNeighbour upscaling.
Signed-off-by: Fredrik Svedberg <fredrik.svedberg@arm.com>
Change-Id: I0d6d1efc606603cdd6188ae282e7f6babfd7e24e
Diffstat (limited to 'ethosu/vela/scheduler.py')
-rw-r--r-- | ethosu/vela/scheduler.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/ethosu/vela/scheduler.py b/ethosu/vela/scheduler.py index e4543e3a..044b246c 100644 --- a/ethosu/vela/scheduler.py +++ b/ethosu/vela/scheduler.py @@ -32,7 +32,6 @@ 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 @@ -269,7 +268,7 @@ class SchedulerOperation: """Returns the amount of IFM required to produce the stripe with shape:'stripe_shape'""" ofm_shape_to_produce = Block.from_shape(stripe_shape.as_list()) - return get_ifm_area_required(ofm_shape_to_produce, self.kernel, to_upscale(self.resampling_mode)) + return get_ifm_area_required(ofm_shape_to_produce, self.kernel, self.resampling_mode) def _calculate_min_stripe_input(self) -> Shape4D: # Calculate the input volume required height and width for the smallest possible stripe (h,w = 1,1) |