Merge pull request #311 from opencompl/sasha/regalloc/more-kernels

include snitch_stream variants in regalloc csv
opencompl · Oct 3, 2024 · 94058bc · 94058bc
2 parents 3988235 + 64f00fb
commit 94058bc
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/Snakefile b/Snakefile
@@ -450,11 +450,9 @@ rule regalloc_stats_to_csv:
         import pandas as pd
 
         df = pd.read_json(input[0], lines=True)
-        df = df[df.variant == "linalg_xdsl"]
-        del df["preallocated_int"]
-        del df["preallocated_float"]
-        del df["variant"]
-        df.set_index(["impl", "params"], inplace=True)
+        df = df[df['variant'].isin(["linalg_xdsl", "snitch_stream"])]
+        df = df.drop(columns=["preallocated_int", "preallocated_float", "variant"])
+        df = df.set_index(["impl", "params"])
         df.to_csv(output[0], index=True)
 
 

diff --git a/results/regalloc.fast.csv b/results/regalloc.fast.csv
@@ -1,10 +1,14 @@
 impl,params,allocated_float,allocated_int
 conv2d_d1_s1_3x3,4x4xf64,8,9
+ddot,128xf64,6,8
+dense,8x8xf64,5,12
 fill,4x4xf64,3,4
 matmul,4x16x8xf64,8,9
+matmul_transb,4x16x16xf32,19,13
 pooling_nchw_max_d1_s2_3x3,4x4xf64,7,7
 pooling_nchw_sum_d1_s2_3x3,4x4xf64,7,7
 relu,4x4xf64,3,6
+relu,4x8xf32,3,6
 sum,4x4xf64,3,8
 sum,4x8xf32,3,8
 sum,8x8xf16,3,8