commit 77b448fa6a63e472928276cfe4e092da1c1395b4
author Jonas Ohlsson <jonas.ohlsson@arm.com> Fri Mar 11 16:08:30 2022 +0100
committer patrik.gustavsson <patrik.gustavsson@arm.com> Mon Mar 14 08:51:31 2022 +0000
tree 559dad5c34b0328800abfb382ecd7869b15101d7
parent 1ddb2ed5fdc0a5e9944c5aeafcfc5ed4c07ea5cf

Fix bug storing encoded NPU weight UUIDs

Fix bug when storing the encoded NPU weight UUID in the
NPU performance estimation.

Signed-off-by: Jonas Ohlsson <jonas.ohlsson@arm.com>
Change-Id: I92127b0020f12352d923c0c9aa2b6f47e6110764

ethosu/vela/npu_performance.py

diff --git a/ethosu/vela/npu_performance.py b/ethosu/vela/npu_performance.py
index 450fa7b..34530ae 100644
--- a/ethosu/vela/npu_performance.py
+++ b/ethosu/vela/npu_performance.py
@@ -753,7 +753,7 @@
             # Save UUIDs of encoded_npu_weight so only unique instances of tensors are used to calculate weights
             if encoded_npu_weight and (encoded_npu_weight.equivalence_id not in encoded_npu_weight_uuids):
 
-                encoded_npu_weight_uuids.add(encoded_npu_weight)
+                encoded_npu_weight_uuids.add(encoded_npu_weight.equivalence_id)
                 total_encoded_weight_size += len(encoded_npu_weight.buffer)
 
             total_bws += bws