From 77b448fa6a63e472928276cfe4e092da1c1395b4 Mon Sep 17 00:00:00 2001 From: Jonas Ohlsson Date: Fri, 11 Mar 2022 16:08:30 +0100 Subject: 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 Change-Id: I92127b0020f12352d923c0c9aa2b6f47e6110764 --- ethosu/vela/npu_performance.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ethosu/vela/npu_performance.py b/ethosu/vela/npu_performance.py index 450fa7b6..34530ae8 100644 --- a/ethosu/vela/npu_performance.py +++ b/ethosu/vela/npu_performance.py @@ -753,7 +753,7 @@ def calc_new_performance_for_network(nng: Graph, arch): # 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 -- cgit v1.2.1