diff options
author | Ayaan Masood <Ayaan.Masood@arm.com> | 2022-02-22 11:28:55 +0000 |
---|---|---|
committer | Ayaan Masood <Ayaan.Masood@arm.com> | 2022-02-22 11:28:55 +0000 |
commit | b801dda26bbcff8ec4f7967d60f38239fd16912b (patch) | |
tree | bb8aa6bc3141f1dd48ffb22630a1eac1cc209825 /ethosu/vela/stats_writer.py | |
parent | 5e0ae5598ab1d7debd603bdd32c7e8f9cad9d581 (diff) | |
download | ethos-u-vela-b801dda26bbcff8ec4f7967d60f38239fd16912b.tar.gz |
MLBEDSW-5880 Fixed Vela verbose weight flag
*Original weights and encoded NPU weight now report correct size instead
of zero when running vela with --verbose-weights flag
(Code to update the aforementioned attributes was missing)
*Removed print references to unencoded NPU weight size
Change-Id: I6d3e41c04cc46d24eeb54cab89818a35e5df27be
Signed-off-by: Ayaan Masood <Ayaan.Masood@arm.com>
Diffstat (limited to 'ethosu/vela/stats_writer.py')
-rw-r--r-- | ethosu/vela/stats_writer.py | 8 |
1 files changed, 1 insertions, 7 deletions
diff --git a/ethosu/vela/stats_writer.py b/ethosu/vela/stats_writer.py index 86f531a8..d8a274b0 100644 --- a/ethosu/vela/stats_writer.py +++ b/ethosu/vela/stats_writer.py @@ -110,7 +110,6 @@ def write_summary_metrics_csv(nng, summary_filename, arch): data_items += [midpoint_fps, nng.batch_size, midpoint_inference_time, n_passes, n_cascaded_passes] data_items += [nng.memory_used.get(mem_area, 0) / 1024.0 for mem_area in mem_areas] data_items += [nng.total_original_weights] - data_items += [nng.total_npu_weights] data_items += [nng.total_npu_encoded_weights] for mem_area in mem_areas: @@ -325,7 +324,6 @@ def print_performance_metrics_for_strat( if weights_data: print(f"Original Weights Size {weights_data['original'] / 1024.0:12.2f} KiB", file=f) - print(f"NPU Weights Size {weights_data['npu'] / 1024.0:12.2f} KiB", file=f) print(f"NPU Encoded Weights Size {weights_data['npu_encoded'] / 1024.0:12.2f} KiB", file=f) print(file=f) @@ -372,11 +370,7 @@ def print_performance_metrics(nng, arch, show_cpu_operations=False, verbose_weig npu_operations.append(op) weights_data = ( - { - "original": nng.total_original_weights, - "npu": nng.total_npu_weights, - "npu_encoded": nng.total_npu_encoded_weights, - } + {"original": nng.total_original_weights, "npu_encoded": nng.total_npu_encoded_weights} if verbose_weights else None ) |