From b81e1bb92be8ea5b29625cf2f361e9160286b16a Mon Sep 17 00:00:00 2001 From: Fredrik Svedberg Date: Tue, 11 Oct 2022 21:50:51 +0200 Subject: MLBEDSW-6971 Fix output diff when cascading elementwise operators Fixed output diff when cascading elementwise operators with reversed operand order. Signed-off-by: Fredrik Svedberg Change-Id: Iac2e28cfb53037b929459af213f4fa7715b3e6de --- ethosu/vela/high_level_command_stream_generator.py | 3 +++ 1 file changed, 3 insertions(+) (limited to '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 7e13b62f..e71fb6ea 100644 --- a/ethosu/vela/high_level_command_stream_generator.py +++ b/ethosu/vela/high_level_command_stream_generator.py @@ -74,6 +74,8 @@ def generate_high_level_commands_for_sched_op(sched_op, schedule): _, _, ) = parent_op.get_ifm_ifm2_weights_biases_ofm() + if sched_op.reversed_operands: + ifm2_tensor, ifm_tensor = ifm_tensor, ifm2_tensor ifm = sched_op.ifm ifm2 = sched_op.ifm2 ofm_shape = sched_op.ofm.shape @@ -236,4 +238,5 @@ def generate_high_level_commands_for_sched_op(sched_op, schedule): ifm2_box=ifm2_box, pad_top=pad_top, pad_bottom=pad_bottom, + reversed_operands=sched_op.reversed_operands, ) -- cgit v1.2.1