From bf61268fae3d05ae7687067ae2ab2964067634c9 Mon Sep 17 00:00:00 2001 From: Jacob Bohlin Date: Thu, 13 Aug 2020 09:37:02 +0200 Subject: MLBEDSW-2755: Added check that ifm2_tensor is set Signed-off-by: Jacob Bohlin Change-Id: I5b8db6430e79ec7a5836d8dd00a03413647de8ba --- ethosu/vela/register_command_stream_generator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'ethosu/vela') diff --git a/ethosu/vela/register_command_stream_generator.py b/ethosu/vela/register_command_stream_generator.py index 471953d9..28b67656 100644 --- a/ethosu/vela/register_command_stream_generator.py +++ b/ethosu/vela/register_command_stream_generator.py @@ -487,7 +487,7 @@ def generate_register_command_stream(nng, sg, arch, verbose=False): 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 -- cgit v1.2.1