diff options
| author | Henrik G Olsson <henrik.olsson@arm.com> | 2021-04-15 17:57:26 +0200 |
|---|---|---|
| committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2021-04-30 13:00:31 +0000 |
| commit | 5fabfcaa2b636b02899b4d6e0ccf95d853986475 (patch) | |
| tree | c2dff0cc2e67bbb90b192a209855cab9e8c0fbad /ethosu/vela/high_level_command_stream.py | |
| parent | dec6fbcb16fa2f3d7254c4beb3235ab50f72a923 (diff) | |
| download | ethos-u-vela-5fabfcaa2b636b02899b4d6e0ccf95d853986475.tar.gz | |
MLBEDSW-4350 Use padding instead of skirt for merged SplitSlice
When the operations are merged some later passes are confused by start
and end coordinates for the convolution not being along the edges of
the IFM, and omitting padding. But we need the zero padding to keep
the output the same as before the transformation.
Also fixes bug where Vela could crash if convolution had explicit
start coordinate.
Signed-off-by: Henrik G Olsson <henrik.olsson@arm.com>
Change-Id: I8449d237350d528f83738b2f09124f1ed79c07ca
Diffstat (limited to 'ethosu/vela/high_level_command_stream.py')
| -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 075574e4..53c5e01f 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 == NpuBlockType.ElementWise and len(new_end_coord) >= 1: + if npu_block_type in (NpuBlockType.ElementWise, NpuBlockType.ConvolutionMxN) and 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) |