diff options
| author | Tim Hall <tim.hall@arm.com> | 2021-12-17 14:09:19 +0000 |
|---|---|---|
| committer | tim.hall <tim.hall@arm.com> | 2021-12-17 16:57:50 +0000 |
| commit | 3df5b9624cdf6c2668ea35edc84076f5612bfc15 (patch) | |
| tree | a73ab69561555890fdf5ee4681d48be90a6a1867 | |
| parent | 86dcee2667bef8f44912faaf6de47d90ee7dc434 (diff) | |
| download | ethos-u-vela-3df5b9624cdf6c2668ea35edc84076f5612bfc15.tar.gz | |
MLBEDSW-5834: split shape is None when split offset is not
- This bug causes an exception to occur when trying to index split
shape in Box.transform_with_strides_and_skirt()
- The bug was due to the read shapes not being initialised when creating
a primary op in pass packing
Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: I3ebd7fc4c7ef5c06488a36d8340a17ae6afd4609
| -rw-r--r-- | ethosu/vela/pass_packing.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/ethosu/vela/pass_packing.py b/ethosu/vela/pass_packing.py index 530ad78..1fefdf4 100644 --- a/ethosu/vela/pass_packing.py +++ b/ethosu/vela/pass_packing.py @@ -399,6 +399,7 @@ def pack_into_passes(nng, arch, verbose_packing=False): avgpool_op.ifm_shapes = op.ifm_shapes.copy() avgpool_op.ofm_shapes = op.ofm_shapes.copy() avgpool_op.read_offsets = op.read_offsets.copy() + avgpool_op.read_shapes = op.read_shapes.copy() op.inputs[0] = avgpool_out op_list.insert(0, avgpool_op) |