68 lines
2.0 KiB
Python
Executable File
68 lines
2.0 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import csv
|
|
import os
|
|
|
|
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.csv")
|
|
dst_data = os.path.join(OUT_DATA, "summary_energy.csv")
|
|
dst_iot = os.path.join(OUT_IOT, "summary_energy.csv")
|
|
|
|
# Your model constants (document in Methods)
|
|
EPI_PJ = {"big": 200.0, "little": 80.0, "hybrid": 104.0}
|
|
E_MEM_PJ = 600.0
|
|
DROWSY_SCALE = 0.85
|
|
|
|
rows = []
|
|
if not os.path.exists(src):
|
|
print(f"[energy] ERROR: Source file {src} does not exist")
|
|
exit(1)
|
|
|
|
with open(src) as f:
|
|
r = csv.DictReader(f)
|
|
for row in r:
|
|
# Handle empty or invalid values
|
|
try:
|
|
insts = float(row["insts"]) if row["insts"] else 0.0
|
|
secs = float(row["sim_seconds"]) if row["sim_seconds"] else 0.0
|
|
mr = float(row["l2_miss_rate"]) if row["l2_miss_rate"] else 0.0
|
|
except (ValueError, TypeError):
|
|
print(f"[energy] WARNING: Invalid data in row: {row}")
|
|
insts = 0.0
|
|
secs = 0.0
|
|
mr = 0.0
|
|
|
|
core = row["core"]
|
|
drowsy = int(row["drowsy"]) if row["drowsy"] else 0
|
|
epi = EPI_PJ.get(core, EPI_PJ["little"])
|
|
|
|
l2_misses = mr * insts
|
|
energy = (epi * 1e-12) * insts + (E_MEM_PJ * 1e-12) * l2_misses
|
|
if drowsy == 1:
|
|
energy *= DROWSY_SCALE
|
|
power = energy / secs if secs > 0 else 0.0
|
|
edp = energy * secs
|
|
|
|
row.update(
|
|
{
|
|
"energy_J": f"{energy:.6f}",
|
|
"power_W": f"{power:.6f}",
|
|
"edp": f"{edp:.6e}",
|
|
"epi_model_pJ": f"{epi:.1f}",
|
|
}
|
|
)
|
|
rows.append(row)
|
|
|
|
if not rows:
|
|
print(f"[energy] ERROR: No data found in {src}")
|
|
exit(1)
|
|
|
|
for path in (dst_data, dst_iot):
|
|
with open(path, "w", newline="") as f:
|
|
w = csv.DictWriter(f, fieldnames=list(rows[0].keys()))
|
|
w.writeheader()
|
|
w.writerows(rows)
|
|
print(f"[energy] wrote {dst_data} and mirrored to {dst_iot}")
|