diff options
| author | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-05-05 08:30:41 +0200 |
|---|---|---|
| committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2021-05-07 09:19:20 +0000 |
| commit | e6b94bb0a6d6021c5e575aef80fbcaaa332d7e33 (patch) | |
| tree | 11e20d616fb1b3a37ec83c598f398541b1813505 | |
| parent | 03d40a24de88452a877f356ffe20e93705e76c43 (diff) | |
| download | ethos-u-vela-e6b94bb0a6d6021c5e575aef80fbcaaa332d7e33.tar.gz | |
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
| -rw-r--r-- | ethosu/vela/high_level_command_stream.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/high_level_command_stream.py b/ethosu/vela/high_level_command_stream.py index 53c5e01f..19a363c3 100644 --- a/ethosu/vela/high_level_command_stream.py +++ b/ethosu/vela/high_level_command_stream.py @@ -62,7 +62,7 @@ class Box: 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) |