From 6bb8f6700e0bc1a254fab745b8d3fac767270304 Mon Sep 17 00:00:00 2001 From: Patrik Gustavsson Date: Mon, 21 Dec 2020 14:49:13 +0100 Subject: MLBEDSW-3786 Fix index out of range on ofm_shapes Ofm_shapes only set on operator for npu_ops. Signed-off-by: Patrik Gustavsson Change-Id: Iab98e24132f3a4004debce9013355e2ef16b0b6f --- ethosu/vela/pass_packing.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/ethosu/vela/pass_packing.py b/ethosu/vela/pass_packing.py index 8f6660c2..0820034d 100644 --- a/ethosu/vela/pass_packing.py +++ b/ethosu/vela/pass_packing.py @@ -231,9 +231,11 @@ def pack_into_passes(nng, arch, verbose_packing=False): ofm_tensor = op.ofm if ofm_tensor is None: ofm_tensor = op.outputs[0] - build_pass((op,), ofm_tensor, op.ofm_shapes[0].clone()) + ofm_shape = op.ofm_shapes[0].clone() if op.run_on_npu else None - def build_pass(start_ops_to_process, ofm_tensor=None, ofm_shapes=None): + build_pass((op,), ofm_tensor, ofm_shape) + + def build_pass(start_ops_to_process, ofm_tensor=None, ofm_shape=None): reverse_ops_list = [] curr_flags = PassFlags.Empty npu_block_type = NpuBlockType.Default @@ -416,7 +418,7 @@ def pack_into_passes(nng, arch, verbose_packing=False): ps.ifm_shapes.append(ps.primary_op.ifm_shapes[0]) ps.ofm_tensor = ofm_tensor - ps.ofm_shapes.append(ofm_shapes) + ps.ofm_shapes.append(ofm_shape) assert ps.placement != PassPlacement.Npu or ps.ofm_tensor is not None ps.weight_tensor = ps.get_primary_op_ifm_weights()[1] -- cgit v1.2.1