Add overall comparison into code_size_comparison.py

InfiniTensor · Jan 13, 2025 · 13dc6be · 13dc6be
1 parent 69b626d
commit 13dc6be
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/code_size_comparison.py b/code_size_comparison.py
@@ -42,12 +42,15 @@
 print(
     pd.DataFrame(
         {
-            "Kernel": kernels,
+            "Kernel": kernels + ("Overall",),
             "Relative Code Size Change (%)": [
                 f"{ninetoothed_lines / triton_lines * 100:.2f}%"
                 for ninetoothed_lines, triton_lines in zip(
                     lines_of_code["NineToothed"], lines_of_code["Triton"]
                 )
+            ]
+            + [
+                f"{sum(lines_of_code['NineToothed']) / sum(lines_of_code['Triton']) * 100:.2f}%"
             ],
         }
     )