From 3a4eb5a2e8b7bbc5a5d4450b322bf26367d50217 Mon Sep 17 00:00:00 2001
From: tanner-andrulis <andrulis@mit.edu>
Date: Fri, 31 May 2024 10:00:47 -0400
Subject: [PATCH] Parsed output area&energy dicts now have all components even
 if zero energy

---
 timeloopfe/v4/output_parsing.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/timeloopfe/v4/output_parsing.py b/timeloopfe/v4/output_parsing.py
index efdc814..2b12644 100644
--- a/timeloopfe/v4/output_parsing.py
+++ b/timeloopfe/v4/output_parsing.py
@@ -416,6 +416,10 @@ def parse_timeloop_output(
     cycles, computes, percent_utilization, energy = parse_stats_file(stats_path)
     area = get_area_from_art(art_path)
 
+    for k in list(area.keys()) + list(energy.keys()):
+        area.setdefault(k, 0)
+        energy.setdefault(k, 0)
+
     spec.parse_expressions()
     mapping = None
     if os.path.exists(stats_path.replace(".stats.txt", ".map.txt")):