commit a0c3624899edc601525a589643c802469003f89d
author Fredrik Svedberg <fredrik.svedberg@arm.com> Wed Jun 03 15:43:31 2020 +0200
committer Louis Verhaard <louis.verhaard@arm.com> Wed Aug 05 16:26:04 2020 +0200
tree 4fc52db04cd29901b3e5d4a7425a7a641e9647fb
parent 9a03fdff316662be69a1adc4e391e43bc6519b08

[MLBEDSW-2335] SoftMax int16

Added graph rewrite of Softmax for int16.

Change-Id: Id7885af6056a23e8b8362fb61ae94283251eb398
Signed-off-by: Fredrik Svedberg <fredrik.svedberg@arm.com>

ethosu/vela/shared_buffer_allocation.py

diff --git a/ethosu/vela/shared_buffer_allocation.py b/ethosu/vela/shared_buffer_allocation.py
index 07637f3..7268d9f 100644
--- a/ethosu/vela/shared_buffer_allocation.py
+++ b/ethosu/vela/shared_buffer_allocation.py
@@ -82,6 +82,8 @@
                 assert (self.use_ifm_element == SHRAMElements.IFM16) or (
                     self.use_ifm_element == SHRAMElements.IFM16_Elementwise
                 )
+            elif is_elementwise and self.ifm_bits == 32:
+                self.use_ifm_element = SHRAMElements.IFM32_Elementwise
             else:
                 assert self.ifm_bits == 8, "Unexpected IFM bitdepth"
 
@@ -168,7 +170,7 @@
     if arch.override_block_config:
         config = alloc.try_block(arch.override_block_config)
         if config is None:
-            raise VelaError("Block config override '{0}' cannot be allocated".format(arch.override_block_config) )
+            raise VelaError("Block config override '{0}' cannot be allocated".format(arch.override_block_config))
         return [config]
 
     # Constrain the search space if the OFM is smaller than the max block size