diff options
author | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2020-12-23 08:50:44 +0100 |
---|---|---|
committer | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2020-12-23 08:50:44 +0100 |
commit | 0a261cd279aeec29a8910960d6c763f69d74c41a (patch) | |
tree | 59e13774b1e357e2a84601edca2031a6a84ec80c /ethosu | |
parent | 70b71a5639484b234515c9130f334d285d76d190 (diff) | |
download | ethos-u-vela-0a261cd279aeec29a8910960d6c763f69d74c41a.tar.gz |
MLBEDSW-3804 Add missing check for npu_op
Add missing check for npu_op
Op ifm ofm shapes only valid for npu_ops.
Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: I73624c8e122fee510ab8320172b8b3a648a6f070
Diffstat (limited to 'ethosu')
-rw-r--r-- | ethosu/vela/pass_packing.py | 18 |
1 files changed, 10 insertions, 8 deletions
diff --git a/ethosu/vela/pass_packing.py b/ethosu/vela/pass_packing.py index 35f53300..7abf3b24 100644 --- a/ethosu/vela/pass_packing.py +++ b/ethosu/vela/pass_packing.py @@ -402,15 +402,17 @@ def pack_into_passes(nng, arch, verbose_packing=False): # Get the corresponding ifm_shapes for op in input_ops_list + [primary_op]: - if ps.ifm_tensor == op.ifm: - ps.ifm_shapes.append(op.ifm_shapes[0]) - elif ps.ifm_tensor == op.ifm2: - ps.ifm_shapes.append(op.ifm_shapes[1]) + if op.run_on_npu: + if ps.ifm_tensor == op.ifm: + ps.ifm_shapes.append(op.ifm_shapes[0]) + elif ps.ifm_tensor == op.ifm2: + ps.ifm_shapes.append(op.ifm_shapes[1]) for op in input_ops_list + [primary_op]: - if ps.ifm2_tensor == op.ifm: - ps.ifm_shapes.append(op.ifm_shapes[0]) - elif ps.ifm2_tensor == op.ifm2: - ps.ifm_shapes.append(op.ifm_shapes[1]) + if op.run_on_npu: + if ps.ifm2_tensor == op.ifm: + ps.ifm_shapes.append(op.ifm_shapes[0]) + elif ps.ifm2_tensor == op.ifm2: + ps.ifm_shapes.append(op.ifm_shapes[1]) else: ps.ifm_tensor = ifm_tensor ps.ifm2_tensor = None |