diff options
Diffstat (limited to 'ethosu/vela/stats_writer.py')
-rw-r--r-- | ethosu/vela/stats_writer.py | 11 |
1 files changed, 7 insertions, 4 deletions
diff --git a/ethosu/vela/stats_writer.py b/ethosu/vela/stats_writer.py index c4b4cd9e..3fd29d12 100644 --- a/ethosu/vela/stats_writer.py +++ b/ethosu/vela/stats_writer.py @@ -18,12 +18,15 @@ # Description: # Writes out per-pass and summary performance statistics to CSV files. +import csv +import sys + import numpy as np -from .nn_graph import MemArea, TensorPurpose, PassPlacement + +from .tensor import MemArea, TensorPurpose +from .nn_graph import PassPlacement from .npu_performance import PassCycles, MacCount, BandwidthDirection -import csv from .numeric_util import round_up_to_int -import sys def write_summary_metrics_csv(nng, summary_filename, arch): @@ -246,7 +249,7 @@ def print_performance_metrics_for_strat( print(file=f) for mem_area, label in mem_area_labels: - if not mem_area in memory_used: + if mem_area not in memory_used: continue aug_label = label + " used" |