add back benchmarks

binpash · Jan 11, 2025 · 5a73db8 · 5a73db8
1 parent 44b8897
commit 5a73db8
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/infrastructure/viz_dynamic.py b/infrastructure/viz_dynamic.py
@@ -217,5 +217,18 @@ def main():
     plot_io(df)
     plot_memory(df)
 
+    plot_benchmark_times(df, ylabel='Time per input byte',
+                         ticks=([0, 0.00000001, 0.0000001, 0.000001, 0.00001, 0.0001], 
+                                ['0', '10ns', '100ns',    '1us',    '10us', '100us']),
+                                linthresh=0.00000001)
+    plot_io(df, ylabel='IO per input byte',
+            ticks=([0,    1,   10,     100,    1000], 
+                   ['0', '1B', '10B', '100B', '1KB']),
+                   linthresh=1)
+    plot_memory(df, ylabel='Memory per input byte',
+                ticks=([0,   0.001,   0.01,     0.1,  1,    10,    100,   1000,  10000], 
+                       ['0', '0.001B', '0.01B', '0.1B', '1B', '10B', '100B', '1KB', '10KB']),
+                linthresh=0.001)
+
 if __name__ == '__main__':
     main()