commit b8f400bbe963e75356ab619461bdceb8cc9568a9
author Fredrik Svedberg <Fredrik.Svedberg@arm.com> Thu Apr 08 15:41:52 2021 +0200
committer patrik.gustavsson <patrik.gustavsson@arm.com> Thu Apr 08 14:46:05 2021 +0000
tree a02af02eb2058549ad946dbf2de99cab3c206ae2
parent ee99bb124b088430b97d205df9fc90a1e9412e0c

Fix stats_writer.py exception

Fixed exception in stats_writer.py.

Change-Id: I625390aec185345cadd0d8fa5edb66907b9be242
Signed-off-by: Fredrik Svedberg <Fredrik.Svedberg@arm.com>

ethosu/vela/stats_writer.py

diff --git a/ethosu/vela/stats_writer.py b/ethosu/vela/stats_writer.py
index 9ccee72..597fd15 100644
--- a/ethosu/vela/stats_writer.py
+++ b/ethosu/vela/stats_writer.py
@@ -267,7 +267,7 @@
 
         print(f"Total {aug_label:25}          {memory_used[mem_area] / 1024.0:12.2f} KiB", file=f)
         if mem_area == MemArea.Sram and min_mem_usage:
-            mem_used = memory_used[[mem_area for mem_area, _ in mem_area_labels if "SRAM" in mem_area][0]] / 1024.0
+            mem_used = memory_used[[mem_area for mem_area, label in mem_area_labels if "SRAM" in label][0]] / 1024.0
             fraction = (mem_used - min_mem_usage / 1024.0) / (min_mem_usage / 1024.0)
             print(f"Theoretical minimum SRAM usage{min_mem_usage/1024.0:23.2F} KiB", file=f)
             print(f"Allocator overhead{100*fraction:35.2F} %", file=f)