Add simulation script

simulation_script.py

@@ -0,0 +1,76 @@
+import subprocess
+import csv
+import copy
+import dataclasses
+import json
+
+from threading import Thread
+from pathlib import Path
+from typing import Dict
+from configs.pim_config import Configuration, Statistics
+
+gem5 = Path("build/ARM/gem5.opt")
+out_dir_base = Path("pim_out")
+pim_simulation = Path("configs/pim_simulation.py")
+
+
+class Gem5Thread(Thread):
+    def __init__(self, configuration: Configuration) -> None:
+        super().__init__()
+        self.configuration = configuration
+
+    def run(self):
+        serialized_configuration = json.dumps(
+            dataclasses.asdict(self.configuration)
+        )
+
+        out_dir = out_dir_base / configuration.name
+
+        out = subprocess.run(
+            [
+                gem5,
+                "-d" + out_dir.as_posix(),
+                pim_simulation,
+                serialized_configuration,
+            ],
+            capture_output=True,
+        )
+        # print(out.stderr)
+        # print(out.stdout)
+        output = out.stdout.splitlines()[-1]
+        self.statistics = Statistics(**json.loads(output))
+
+
+workload_directory = Path("kernels")
+
+workloads = [
+    "classic_vadd",
+    "classic_vmul",
+    "classic_haxpy",
+    "classic_gemv",
+    # "classic_gemv_layers",
+    "vadd",
+    "vmul",
+    "haxpy",
+    "gemv",
+    # "gemv_layers",
+]
+
+configurations: list[Configuration] = []
+
+for frequency in ["3GHz", "100GHz"]:
+    for workload in workloads:
+        configurations.append(
+            Configuration(workload + "_" + frequency, (workload_directory / workload).as_posix(), frequency)
+        )
+
+threads: list[Gem5Thread] = []
+
+for configuration in configurations:
+    thread = Gem5Thread(configuration)
+    thread.start()
+    threads.append(thread)
+
+for thread in threads:
+    thread.join()
+    print(thread.configuration, thread.statistics)