commit 2b7a1628ab26b453a0932ce845f41fef8e5ade34
author Tim Hall <tim.hall@arm.com> Tue Sep 08 17:00:33 2020 +0100
committer tim.hall <tim.hall@arm.com> Wed Sep 09 16:04:10 2020 +0000
tree 9a9cc630a5a948a8a72dc6a082bd0d596616ea71
parent 11b0bdbf029edc1e166f55a6da1bbaf40da068c4

vela: SHRAM alloc too big for some elementwise ops

 - Fixed SHRAM allocation for 8 and 16-bit elementwise ops

Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: I909a86d76e4ee6eab612aae827038b82703f28dc

ethosu/vela/shared_buffer_allocation.py

diff --git a/ethosu/vela/shared_buffer_allocation.py b/ethosu/vela/shared_buffer_allocation.py
index 0f5c1c8..55cc34c 100644
--- a/ethosu/vela/shared_buffer_allocation.py
+++ b/ethosu/vela/shared_buffer_allocation.py
@@ -90,7 +90,8 @@
                 assert (self.use_ifm_element == SHRAMElements.IFM16) or (
                     self.use_ifm_element == SHRAMElements.IFM16_Elementwise
                 )
-            elif is_elementwise or ps.npu_block_type == NpuBlockType.ReduceSum and self.ifm_bits == 32:
+            elif self.ifm_bits == 32:
+                assert is_elementwise or ps.npu_block_type == NpuBlockType.ReduceSum, "Unsupported 32-bit IFM operation"
                 self.use_ifm_element = SHRAMElements.IFM32
             else:
                 assert self.ifm_bits == 8, "Unexpected IFM bitdepth"