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 | |
| 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
| -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 450fa7b..34530ae 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 |