commit bf61268fae3d05ae7687067ae2ab2964067634c9
author Jacob Bohlin <jacob.bohlin@arm.com> Thu Aug 13 09:37:02 2020 +0200
committer Fredrik Knutsson <fredrik.knutsson.hunnebo@gmail.com> Thu Aug 13 13:33:53 2020 +0000
tree a175d80b74eff005dddd5554b3f91c3a0a6c50fd
parent 8b24f2b2a8823720f399018923d2d4cf717fe63b

MLBEDSW-2755: Added check that ifm2_tensor is set

Signed-off-by: Jacob Bohlin <jacob.bohlin@arm.com>
Change-Id: I5b8db6430e79ec7a5836d8dd00a03413647de8ba

ethosu/vela/register_command_stream_generator.py

diff --git a/ethosu/vela/register_command_stream_generator.py b/ethosu/vela/register_command_stream_generator.py
index 471953d..28b6765 100644
--- a/ethosu/vela/register_command_stream_generator.py
+++ b/ethosu/vela/register_command_stream_generator.py
@@ -487,7 +487,7 @@
                     IFM2Broadcast.ReverseOperandOrder if primary_op.attrs.get("reverse_op_order", False) else 0
                 )
 
-                if not ifm_ifm2_correct_order(cmd.ifm_tensor.shape, cmd.ifm2_tensor.shape):
+                if cmd.ifm2_tensor and not ifm_ifm2_correct_order(cmd.ifm_tensor.shape, cmd.ifm2_tensor.shape):
                     # The scalar has to be the ifm2 tensor so switch the ifms
                     cmd.ifm_tensor, cmd.ifm2_tensor = cmd.ifm2_tensor, cmd.ifm_tensor
                     cmd.ifm_box, cmd.ifm2_box = cmd.ifm2_box, cmd.ifm_box