108 lines
3.3 KiB
Python
Executable File
108 lines
3.3 KiB
Python
Executable File
# scripts/hetero_big_little.py
|
|
# Minimal SmartEdgeAI heterogeneous ARM system
|
|
# Supports --cmd, --mem, --l2, --dvfs, and --drowsy
|
|
# Generates valid stats.txt for all workloads
|
|
|
|
import argparse, m5
|
|
from m5.objects import *
|
|
|
|
# -------------------------------
|
|
# Argument parsing
|
|
# -------------------------------
|
|
ap = argparse.ArgumentParser()
|
|
ap.add_argument("--cmd", required=True)
|
|
ap.add_argument("--mem", default="16GB")
|
|
ap.add_argument("--l2", default="1MB")
|
|
ap.add_argument("--dvfs", choices=["high","low"], default="high")
|
|
ap.add_argument("--drowsy", type=int, choices=[0,1], default=0)
|
|
ap.add_argument("--core", choices=["big","little","hybrid"], default="big")
|
|
ap.add_argument("--outdir", default="m5out")
|
|
args = ap.parse_args()
|
|
|
|
# -------------------------------
|
|
# Clock & Voltage (DVFS)
|
|
# -------------------------------
|
|
v = VoltageDomain(voltage="1.0V" if args.dvfs == "high" else "0.8V")
|
|
clk = "2GHz" if args.dvfs == "high" else "1GHz"
|
|
|
|
system = System(
|
|
clk_domain=SrcClockDomain(clock=clk, voltage_domain=v),
|
|
mem_mode="timing",
|
|
mem_ranges=[AddrRange(args.mem)]
|
|
)
|
|
|
|
# -------------------------------
|
|
# CPU cluster: Configure based on core type
|
|
# -------------------------------
|
|
if args.core == "big":
|
|
system.cpu = [O3CPU(cpu_id=0)]
|
|
elif args.core == "little":
|
|
system.cpu = [TimingSimpleCPU(cpu_id=0)]
|
|
elif args.core == "hybrid":
|
|
# 1 big + 1 little
|
|
system.cpu = [O3CPU(cpu_id=0), TimingSimpleCPU(cpu_id=1)]
|
|
|
|
# -------------------------------
|
|
# Cache hierarchy
|
|
# -------------------------------
|
|
system.l2bus = L2XBar()
|
|
for c in system.cpu:
|
|
c.icache = Cache(size="32kB", assoc=2)
|
|
c.dcache = Cache(size="32kB", assoc=2)
|
|
c.icache.cpu_side = c.icache_port
|
|
c.dcache.cpu_side = c.dcache_port
|
|
c.icache.mem_side = system.l2bus.slave
|
|
c.dcache.mem_side = system.l2bus.slave
|
|
|
|
system.l2 = Cache(size=args.l2, assoc=8)
|
|
system.membus = SystemXBar()
|
|
system.l2.cpu_side = system.l2bus.master
|
|
system.l2.mem_side = system.membus.slave
|
|
|
|
# -------------------------------
|
|
# Drowsy cache behavior
|
|
# -------------------------------
|
|
if args.drowsy:
|
|
system.l2.tag_latency = 24
|
|
system.l2.data_latency = 24
|
|
|
|
# -------------------------------
|
|
# Memory controller
|
|
# -------------------------------
|
|
system.mem_ctrl = SimpleMemory()
|
|
system.mem_ctrl.range = system.mem_ranges[0]
|
|
system.mem_ctrl.port = system.membus.master
|
|
system.system_port = system.membus.slave
|
|
|
|
# -------------------------------
|
|
# Workload setup
|
|
# -------------------------------
|
|
proc = Process()
|
|
proc.executable = args.cmd
|
|
proc.cmd = [args.cmd]
|
|
proc.env = {'GLIBC_TUNABLES': 'glibc.pthread.rseq=0'}
|
|
|
|
# Only assign workload to the first CPU (primary execution)
|
|
system.cpu[0].workload = proc
|
|
system.cpu[0].createThreads()
|
|
|
|
# Other CPUs remain idle
|
|
for i in range(1, len(system.cpu)):
|
|
system.cpu[i].workload = SEWorkload.init_compatible("hello")
|
|
system.cpu[i].createThreads()
|
|
|
|
# -------------------------------
|
|
# Stats configuration
|
|
# -------------------------------
|
|
# Note: Stats will be automatically written to stats.txt in the output directory
|
|
|
|
# -------------------------------
|
|
# Instantiate and simulate
|
|
# -------------------------------
|
|
root = Root(full_system=False, system=system)
|
|
m5.instantiate()
|
|
print("[SmartEdgeAI] Starting simulation...")
|
|
exit_event = m5.simulate()
|
|
print(f"[SmartEdgeAI] Exiting @ tick {m5.curTick()} because {exit_event.getCause()}")
|
|
|