commit 1c772e43434bd39aad1cb2e0036dcdf67b038633
author Henrik G Olsson <henrik.olsson@arm.com> Fri Apr 09 10:51:33 2021 +0200
committer patrik.gustavsson <patrik.gustavsson@arm.com> Fri Apr 09 12:40:45 2021 +0000
tree ba1707c43acf7305b3b13159235da8c3eb32bde3
parent b8f400bbe963e75356ab619461bdceb8cc9568a9

MLBEDSW-4073 Handle elementwise ops with same tensor for both inputs

Signed-off-by: Henrik G Olsson <henrik.olsson@arm.com>
Change-Id: I0e6bb46b7b91ed10f5bda34fba66d8b714560f47

ethosu/vela/high_level_command_stream_generator.py

diff --git a/ethosu/vela/high_level_command_stream_generator.py b/ethosu/vela/high_level_command_stream_generator.py
index aad3783..c01790a 100644
--- a/ethosu/vela/high_level_command_stream_generator.py
+++ b/ethosu/vela/high_level_command_stream_generator.py
@@ -46,7 +46,12 @@
     npu_block_type = ps.npu_block_type
     split_offsets = list(ps.primary_op.read_offsets)  # offset for [ifm, ifm2]
 
-    if ps.ifm_tensor is not None and ps.ifm2_tensor is not None and npu_block_type == NpuBlockType.ElementWise:
+    if (
+        len(ps.inputs) == 2
+        and ps.ifm_tensor is not None
+        and ps.ifm2_tensor is not None
+        and npu_block_type == NpuBlockType.ElementWise
+    ):
         # Ensure correct ifm and ifm2 order
         if ps.inputs[0] == ps.primary_op.inputs[1] and ps.inputs[1] == ps.primary_op.inputs[0]:
             ps.ifm_tensor, ps.ifm2_tensor = ps.ifm2_tensor, ps.ifm_tensor