diff options
author | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2022-10-11 21:50:51 +0200 |
---|---|---|
committer | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2022-10-12 11:22:17 +0000 |
commit | b81e1bb92be8ea5b29625cf2f361e9160286b16a (patch) | |
tree | 0963851b6d449224f67cce10ba510830db7b4172 /ethosu/vela/high_level_command_stream.py | |
parent | 3e3faa90def7c2c4aaaf71543726c881f05f53b5 (diff) | |
download | ethos-u-vela-b81e1bb92be8ea5b29625cf2f361e9160286b16a.tar.gz |
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 <fredrik.svedberg@arm.com>
Change-Id: Iac2e28cfb53037b929459af213f4fa7715b3e6de
Diffstat (limited to 'ethosu/vela/high_level_command_stream.py')
-rw-r--r-- | ethosu/vela/high_level_command_stream.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/ethosu/vela/high_level_command_stream.py b/ethosu/vela/high_level_command_stream.py index 4a41edd0..bfe5bce1 100644 --- a/ethosu/vela/high_level_command_stream.py +++ b/ethosu/vela/high_level_command_stream.py @@ -210,6 +210,7 @@ class NpuStripe(Command): ifm2_box=None, pad_top=0, pad_bottom=0, + reversed_operands=False, ): self.ps = ps self.block_config = block_config @@ -226,6 +227,7 @@ class NpuStripe(Command): self.weight_box = weight_box self.pad_top = pad_top self.pad_bottom = pad_bottom + self.reversed_operands = reversed_operands for i in range(len(self.ofm_box.end_coord)): assert self.ofm_box.end_coord[i] <= ps.ofm_shapes[0][i] |