commit 3ff7a4aed60fefababdced93a5aba91409a4050c
author Fredrik Svedberg <fredrik.svedberg@arm.com> Wed Sep 29 10:08:04 2021 +0200
committer Fredrik Svedberg <fredrik.svedberg@arm.com> Fri Oct 01 07:56:18 2021 +0000
tree 1dfd999beb51fec093d6ce4eb6f68f7bca41b32d
parent 008cd10f6f5b8838de1f37c8c42899ace1c1cf0d

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

ethosu/vela/scheduler.py

diff --git a/ethosu/vela/scheduler.py b/ethosu/vela/scheduler.py
index e4543e3..044b246 100644
--- a/ethosu/vela/scheduler.py
+++ b/ethosu/vela/scheduler.py
@@ -32,7 +32,6 @@
 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 @@
         """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)