commit e6b94bb0a6d6021c5e575aef80fbcaaa332d7e33
author Patrik Gustavsson <patrik.gustavsson@arm.com> Wed May 05 08:30:41 2021 +0200
committer patrik.gustavsson <patrik.gustavsson@arm.com> Fri May 07 09:19:20 2021 +0000
tree 11e20d616fb1b3a37ec83c598f398541b1813505
parent 03d40a24de88452a877f356ffe20e93705e76c43

MLBEDSW-4534 Limit ifm box depth

Limit the ifm box depth to ifm shape depth

Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: I889aed9ef7e338faa1fca074fb2843fa2cedecc8

ethosu/vela/high_level_command_stream.py

diff --git a/ethosu/vela/high_level_command_stream.py b/ethosu/vela/high_level_command_stream.py
index 53c5e01..19a363c 100644
--- a/ethosu/vela/high_level_command_stream.py
+++ b/ethosu/vela/high_level_command_stream.py
@@ -62,7 +62,7 @@
             new_start_coord[-1] = 0
             new_end_coord[-1] = ifm_shape.depth
 
-        if npu_block_type in (NpuBlockType.ElementWise, NpuBlockType.ConvolutionMxN) and len(new_end_coord) >= 1:
+        if len(new_end_coord) >= 1:
             new_end_coord[-1] = min(new_end_coord[-1], ifm_shape.depth)
         if len(new_end_coord) >= 2:
             new_end_coord[-2] = min(new_end_coord[-2], ifm_shape.width * upscaling_factor)