Files
SmartEdgeAI/scripts/plot_epi.py
2025-10-05 02:54:38 +00:00

35 lines
1.1 KiB
Python
Executable File

#!/usr/bin/env python3
import csv
import os
from collections import defaultdict
import matplotlib.pyplot as plt
ROOT = "/home/carlos/projects/gem5"
OUT_DATA = os.path.join(ROOT, "gem5-data", "SmartEdgeAI", "results")
OUT_IOT = os.path.join(ROOT, "iot", "results")
src = os.path.join(OUT_DATA, "summary_energy.csv")
out_data = os.path.join(OUT_DATA, "fig_epi_across_workloads.png")
out_iot = os.path.join(OUT_IOT, "fig_epi_across_workloads.png")
epi_by_core = defaultdict(list)
with open(src) as f:
r = csv.DictReader(f)
for row in r:
insts = float(row["insts"])
energy = float(row["energy_J"])
epi = 1e12 * energy / insts if insts > 0 else 0.0
epi_by_core[row["core"]].append(epi)
cores = ["big", "little", "hybrid"]
vals = [sum(epi_by_core[c]) / max(1, len(epi_by_core[c])) for c in cores]
plt.figure()
plt.bar(cores, vals)
plt.ylabel("EPI (pJ/inst)")
plt.title("Energy per Instruction across workloads (avg by core mode)")
plt.tight_layout()
plt.savefig(out_data)
plt.savefig(out_iot)
print(f"[plot] wrote {out_data} and mirrored to {out_iot}")