commit 2e186c794b4bdc68d89428a58848e376591f76a0
author Louis Verhaard <louis.verhaard@arm.com> Fri Oct 09 10:47:04 2020 +0200
committer patrik.gustavsson <patrik.gustavsson@arm.com> Mon Oct 12 07:03:23 2020 +0000
tree 49fa72c5cf3943664ef2b18222c57d67d9fea42b
parent 8359a474e4f125382fd7b7d5431a612f6013f107

MLBEDSW-3154 Bug fix for LUT ops with IFM from SplitSliceRead

- Incorrect length check in high level command stream generator
- Improved tensor names related to LUT based operations

Change-Id: Ib8844a35a986e2dbef095df23f143f4633b255f9
Signed-off-by: Louis Verhaard <louis.verhaard@arm.com>

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 dc52ae5..3e3cda1 100644
--- a/ethosu/vela/high_level_command_stream_generator.py
+++ b/ethosu/vela/high_level_command_stream_generator.py
@@ -51,7 +51,7 @@
     npu_block_type = ps.npu_block_type
     split_offsets = [None, None]  # offset for [ifm, ifm2]
 
-    if len(ps.inputs) == 2 and npu_block_type == NpuBlockType.ElementWise:
+    if 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 match_tensor(ps.inputs[0], ps.primary_op.inputs[1]) and match_tensor(ps.inputs[1], ps.primary_op.inputs[0]):
             ps.ifm_tensor, ps.ifm2_tensor = ps.ifm2_tensor, ps.ifm_tensor