diff options
author | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2020-08-13 10:02:53 +0200 |
---|---|---|
committer | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2020-08-19 07:27:56 +0200 |
commit | 597fd3f88397501d61855a327c9632fc1dab3f57 (patch) | |
tree | 2c42401ed06b82a0252663568154bf8472fd8024 /ethosu/vela/shared_buffer_allocation.py | |
parent | 30cb47abd68b125d5c3fc315948187a9ea8dbd43 (diff) | |
download | ethos-u-vela-597fd3f88397501d61855a327c9632fc1dab3f57.tar.gz |
[MLBEDSW-2657] Softmax uint8/int8
Added graph rewrite of Softmax for uint8/int8.
Signed-off-by: Fredrik Svedberg <fredrik.svedberg@arm.com>
Change-Id: Iecdd5d2cd3156a601b3313debba4a3562e6be5d7
Diffstat (limited to 'ethosu/vela/shared_buffer_allocation.py')
-rw-r--r-- | ethosu/vela/shared_buffer_allocation.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ethosu/vela/shared_buffer_allocation.py b/ethosu/vela/shared_buffer_allocation.py index c9686194..053377c4 100644 --- a/ethosu/vela/shared_buffer_allocation.py +++ b/ethosu/vela/shared_buffer_allocation.py @@ -85,8 +85,8 @@ class SharedBufferAllocation: 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 + elif is_elementwise or ps.npu_block_type == NpuBlockType.ReduceSum and self.ifm_bits == 32: + self.use_ifm_element = SHRAMElements.IFM32 else: assert self.ifm_bits == 8, "Unexpected IFM bitdepth" |