diff options
author | Jonas Ohlsson <jonas.ohlsson@arm.com> | 2022-03-11 16:08:30 +0100 |
---|---|---|
committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2022-03-14 08:51:31 +0000 |
commit | 77b448fa6a63e472928276cfe4e092da1c1395b4 (patch) | |
tree | 559dad5c34b0328800abfb382ecd7869b15101d7 /ethosu/vela/npu_performance.py | |
parent | 1ddb2ed5fdc0a5e9944c5aeafcfc5ed4c07ea5cf (diff) | |
download | ethos-u-vela-77b448fa6a63e472928276cfe4e092da1c1395b4.tar.gz |
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
Diffstat (limited to 'ethosu/vela/npu_performance.py')
-rw-r--r-- | ethosu/vela/npu_performance.py | 2 |
1 files changed, 1 insertions, 1 deletions
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 |