From 95a3f82a7243f5fc867fa61845fc937410232a8f Mon Sep 17 00:00:00 2001
From: Sasha Lopoukhine <superlopuh@gmail.com>
Date: Tue, 13 Aug 2024 16:39:35 +0100
Subject: [PATCH] bigger matrix size for pipeline

---
 Snakefile            | 12 ++++++++++++
 results/pipeline.csv | 14 +++++++-------
 scripts/pipeline.py  |  1 -
 3 files changed, 19 insertions(+), 8 deletions(-)

diff --git a/Snakefile b/Snakefile
index 6d9e6f5c..4624400a 100644
--- a/Snakefile
+++ b/Snakefile
@@ -458,6 +458,18 @@ rule regalloc_stats_to_csv:
         df.to_csv(output[0], index=True)
 
 
+rule pipeline:
+    input:
+        kernels="results/kernels.pipeline.csv",
+        regalloc="kernels/regalloc.pipeline.jsonl",
+        frep_count="results/frep_count.csv",
+        pipeline_py="scripts/pipeline.py",
+    output:
+        "results/pipeline.csv",
+    shell:
+        "python {input.pipeline_py} {input.kernels} {input.regalloc} {input.frep_count} -o {output}"
+
+
 rule optimization_pipelines:
     input:
         passes = "kernels/optimization_passes.txt",
diff --git a/results/pipeline.csv b/results/pipeline.csv
index 6da70105..9c26b0da 100644
--- a/results/pipeline.csv
+++ b/results/pipeline.csv
@@ -1,7 +1,7 @@
-variant,F Registers,X Registers,Cycles,FPU Occupancy [%],F Loads,F Stores,FMAdd Issues,FRep Count
-Baseline,3,14,40161,2.49,3000,1005,1000,0
-+ Streams,3,12,19165,5.25,1000,1000,1000,0
-+ Scalar Replacement,3,11,4147,24.28,5,5,1000,0
-+ FRep,3,10,4124,24.42,5,5,1000,2
-+ Fuse Fill,5,9,4130,24.50,0,0,1000,1
-+ Unroll and Jam,8,8,1115,90.67,0,0,1000,1
+variant,F Registers,X Registers,params,Cycles,FPU Occupancy [%],F Loads,F Stores,FMAdd Issues,FRep Count
+Baseline,3,14,1x200x5xf64,40161,2.49,3000,1005,1000,0
++ Streams,3,12,1x200x5xf64,19165,5.25,1000,1000,1000,0
++ Scalar Replacement,3,11,1x200x5xf64,4147,24.28,5,5,1000,0
++ FRep,3,10,1x200x5xf64,4124,24.42,5,5,1000,2
++ Fuse Fill,5,9,1x200x5xf64,4130,24.50,0,0,1000,1
++ Unroll and Jam,8,8,1x200x5xf64,1115,90.67,0,0,1000,1
diff --git a/scripts/pipeline.py b/scripts/pipeline.py
index 93e127a9..ce25ff6a 100644
--- a/scripts/pipeline.py
+++ b/scripts/pipeline.py
@@ -58,7 +58,6 @@ def main():
     regalloc_df = pd.read_json(regalloc_stats, lines=True)
     regalloc_df = regalloc_df[regalloc_df.impl == "matmul"]
     del regalloc_df["impl"]
-    del regalloc_df["params"]
 
     regalloc_df = regalloc_df.set_index("variant")