From 3d9533c10cc666c0684ddf625e847d4e10413baa Mon Sep 17 00:00:00 2001
From: Derek Christ <christ.derek@gmail.com>
Date: Tue, 27 Feb 2024 22:17:22 +0100
Subject: [PATCH] First plotting scripts

---
 configs/pim_config.py |  2 +-
 pim_plots.py          | 38 +++++++++++++++++++-------------------
 simulation_script.py  | 11 ++++++++---
 3 files changed, 28 insertions(+), 23 deletions(-)

diff --git a/configs/pim_config.py b/configs/pim_config.py
index ffda503e03..f801ea928c 100644
--- a/configs/pim_config.py
+++ b/configs/pim_config.py
@@ -6,8 +6,8 @@ class Configuration:
     name: str
     workload: str
     executable: Path
-    pim: bool
     level: str
+    pim: bool
     frequency: str = "3GHz"
 
 @dataclass(frozen=True)
diff --git a/pim_plots.py b/pim_plots.py
index 40c8b6f70f..15059050a0 100644
--- a/pim_plots.py
+++ b/pim_plots.py
@@ -1,35 +1,35 @@
 import matplotlib.pyplot as plt
+import seaborn as sns
 import pandas as pd
 import numpy as np
 
 df = pd.read_csv("pim_results.csv")
 
-frequency_filter = df["frequency"] == "3GHz"
-
 workloads = df["workload"].unique()
 
-fig, ax = plt.subplots()
+sns.set_theme()
 
-width = 0.25
-index = 0
+def calc_speedup(x):
+    return x.iat[0] / x.iat[1]
 
-# for workload in workloads:
-for level in df["level"].unique():
-    level_filter = df["level"] == level
+for workload in df["workload"].unique():
+    workload_filter = df["workload"] == workload
 
-    for pim in [False, True]:
-        workload_filter = df["workload"] == "vadd"
+    filtered_df = df[workload_filter]
+    preprocessed_df = filtered_df.groupby(["workload", "level", "frequency"], as_index=False).agg({"ticks": calc_speedup}).rename(columns={"ticks":"speedup"})
 
-        filtered_df = df[level_filter & workload_filter & frequency_filter]
-        print(filtered_df)
+    # print(preprocessed_df)
+    # preprocessed_df.to_csv("plot.csv", index=False)
 
-        x = np.arange(len(filtered_df))
+    g = sns.catplot(
+        data=preprocessed_df, kind="bar",
+        x="level", y="speedup", hue="frequency",
+        palette="dark", alpha=.6, height=6
+    )
 
-        offset = 6*width * index
-        print(x+offset)
-        bars = ax.bar(x + offset, "ticks", width, data=filtered_df, label=level)
-        # ax.bar_label(bars, padding=2)
-        index += 1
+    g.despine(left=True)
+    g.set_axis_labels("", "Speedup")
+    g.set(title=workload)
+    g.legend.set_title("")
 
-ax.legend(loc="upper left")
 plt.show()
diff --git a/simulation_script.py b/simulation_script.py
index dec2d27899..f10a5f9c53 100644
--- a/simulation_script.py
+++ b/simulation_script.py
@@ -52,18 +52,23 @@ workloads = [
     # "gemv_layers",
 ]
 
+systems = [
+    "HBM",
+    "PIM-HBM",
+]
+
 configurations: list[Configuration] = []
 
 for frequency in ["3GHz", "100GHz"]:
     for level in ["X1", "X2", "X3", "X4"]:        
-        for pim in [False, True]:
+        for system in systems:
             for workload in workloads:
                 if workload == "gemv_layers" and level != "X4":
                     continue
                 
                 executable = workload
 
-                if pim:
+                if system == "HBM":
                     executable = f"classic_{workload}"                
 
                 executable = (
@@ -78,8 +83,8 @@ for frequency in ["3GHz", "100GHz"]:
                         f"{workload}_{level}_{frequency}",
                         workload,
                         executable.as_posix(),
-                        pim,
                         level,
+                        system == "PIM-HBM",
                         frequency,
                     )
                 )