diff options
author | Tim Hall <tim.hall@arm.com> | 2021-06-24 18:29:53 +0100 |
---|---|---|
committer | Tim Hall <tim.hall@arm.com> | 2021-07-09 11:51:22 +0100 |
commit | ffe8e288c8321dfc55a3b75f1aedc08769ecb23a (patch) | |
tree | 7445c018851422a5c1b08f1ae5d7f5529fa461aa /ethosu/vela/pass_packing.py | |
parent | 5e26eda0e0f359b6e22b1f1eeb9344cd15e0f093 (diff) | |
download | ethos-u-vela-ffe8e288c8321dfc55a3b75f1aedc08769ecb23a.tar.gz |
MLBEDSW-4839: Fix issues with Elementwise IFM/OFM overlap
- Fixed typo with not using ifm.mem_type
- Fixed bug with using ifm1 properties when only ifm2 is a potential match
- Removed restriction on not considering SHL and SHR for overlap
- Removed some dead reshape code
Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: Id9bcc3c2b3ee9ac7b6276187d3e2f513b4acd4b5
Diffstat (limited to 'ethosu/vela/pass_packing.py')
-rw-r--r-- | ethosu/vela/pass_packing.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/ethosu/vela/pass_packing.py b/ethosu/vela/pass_packing.py index 518b2436..b28f4eb4 100644 --- a/ethosu/vela/pass_packing.py +++ b/ethosu/vela/pass_packing.py @@ -348,8 +348,7 @@ def pack_into_passes(nng, arch, verbose_packing=False): 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 op.run_on_npu: + if ps.ifm2_tensor == op.ifm: ps.ifm_shapes.append(op.ifm_shapes[0]) elif ps.ifm2_tensor == op.ifm2: |