From 91c5a14059f24951f5b46cb448c60269723e0b16 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Johan=20Alfv=C3=A9n?= Date: Sat, 18 Dec 2021 16:45:44 +0100 Subject: MLBEDSW-5740: Fix assert when setting address on identical LUT tensors When an LUT tensor address is updated with another existing LUT tensor address, also make sure to update the equivalence id. Signed-off-by: Johan Alfven Change-Id: I5ce8c608d9ff6d31e16212b1a725b4147dd3f6f1 --- ethosu/vela/lut.py | 1 + 1 file changed, 1 insertion(+) diff --git a/ethosu/vela/lut.py b/ethosu/vela/lut.py index 8a23b51d..3b2e3077 100644 --- a/ethosu/vela/lut.py +++ b/ethosu/vela/lut.py @@ -115,6 +115,7 @@ def optimize_high_level_cmd_stream(sg, arch): existing_tens = lut_state.get_equivalent(lut_tens) if existing_tens is not None: # LUT is already in SHRAM, no need to perform DMA + lut_tens.equivalence_id = existing_tens.equivalence_id lut_tens.address = existing_tens.address cmd.ps.primary_op.activation.lut_index = get_lut_index(arch, existing_tens) continue -- cgit v1.2.1