diff options
| author | Johan Alfvén <johan.alfven@arm.com> | 2021-12-18 16:45:44 +0100 |
|---|---|---|
| committer | tim.hall <tim.hall@arm.com> | 2021-12-20 18:34:26 +0000 |
| commit | 91c5a14059f24951f5b46cb448c60269723e0b16 (patch) | |
| tree | 5cff5c4503b554579be1900818a402f54aecd92b | |
| parent | 51a8dce2c709b6d0f5c1775d8beea27028ed819a (diff) | |
| download | ethos-u-vela-91c5a14059f24951f5b46cb448c60269723e0b16.tar.gz | |
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 <johan.alfven@arm.com>
Change-Id: I5ce8c608d9ff6d31e16212b1a725b4147dd3f6f1
| -rw-r--r-- | ethosu/vela/lut.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/ethosu/vela/lut.py b/ethosu/vela/lut.py index 8a23b51..3b2e307 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 |