diff options
author | Louis Verhaard <louis.verhaard@arm.com> | 2020-10-09 10:47:04 +0200 |
---|---|---|
committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2020-10-12 07:03:23 +0000 |
commit | 2e186c794b4bdc68d89428a58848e376591f76a0 (patch) | |
tree | 49fa72c5cf3943664ef2b18222c57d67d9fea42b /ethosu/vela/high_level_command_stream_generator.py | |
parent | 8359a474e4f125382fd7b7d5431a612f6013f107 (diff) | |
download | ethos-u-vela-2e186c794b4bdc68d89428a58848e376591f76a0.tar.gz |
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>
Diffstat (limited to 'ethosu/vela/high_level_command_stream_generator.py')
-rw-r--r-- | ethosu/vela/high_level_command_stream_generator.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/high_level_command_stream_generator.py b/ethosu/vela/high_level_command_stream_generator.py index dc52ae52..3e3cda19 100644 --- a/ethosu/vela/high_level_command_stream_generator.py +++ b/ethosu/vela/high_level_command_stream_generator.py @@ -51,7 +51,7 @@ def generate_high_level_command_stream_for_pass(strat, passes, block_configs, id 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 |