diff --git a/code_size_comparison.py b/code_size_comparison.py index fa31a1a..1e733e5 100644 --- 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}%" ], } )