Small fixes and data update

bar_plot.py

@@ -72,7 +72,7 @@ def create_matplotlib_bar_plot() -> None:
 
     # Create figure with dark theme - maximum size for full screen
     plt.style.use('dark_background')
-    fig, axs = plt.subplots(2, 1, figsize=(30, 16), sharex=True) 
+    fig, axs = plt.subplots(2, 1, figsize=(20, 11), sharex=True)  # used to be 30, 16
     fig.patch.set_facecolor('#000000')
 
     # Load data and ensure coherence
@@ -103,7 +103,7 @@ def create_matplotlib_bar_plot() -> None:
         "Time to first token and inter-token latency (lower is better)",
         f"Batch size: {batch_size},  sequence length: {sequence_length},  new tokens: {num_tokens_to_generate}",
     ])
-    fig.suptitle(title, color='white', fontsize=20, y=1.005)
+    fig.suptitle(title, color='white', fontsize=20, y=1.005, linespacing=1.5)
     plt.tight_layout()
 
     # Add common legend with full text
@@ -135,9 +135,16 @@ def create_matplotlib_bar_plot() -> None:
     return html
 
 
-def draw_bar_plot(ax: plt.Axes, bar_kwargs: dict, errors: list, ylabel: str, xticks: list[tuple[float, str]]):
+def draw_bar_plot(
+    ax: plt.Axes, 
+    bar_kwargs: dict,
+    errors: list,
+    ylabel: str,
+    xticks: list[tuple[float, str]],
+    adapt_ylim: bool = False,
+) -> None:
     ax.set_facecolor('#000000')
-    ax.grid(True, alpha=0.2, color='white', zorder=0)
+    ax.grid(True, alpha=0.3, color='white', axis='y', zorder=0)
     # Draw bars
     _ = ax.bar(**bar_kwargs, width=1.0, edgecolor='white', linewidth=1, zorder=3)
     # Add error bars
@@ -151,9 +158,10 @@ def draw_bar_plot(ax: plt.Axes, bar_kwargs: dict, errors: list, ylabel: str, xti
     ax.tick_params(colors='white', labelsize=13)
     ax.set_xticks([xt[0] for xt in xticks], [xt[1] for xt in xticks], fontsize=16)
     # Truncate axis to better fit the bars
-    new_ymin, new_ymax = 1e9, -1e9
-    for h, e in zip(bar_kwargs["height"], errors):
-        new_ymin = min(new_ymin, 0.98 * (h - e))
-        new_ymax = max(new_ymax, 1.02 * (h + e))
-    ymin, ymax = ax.get_ylim() 
-    ax.set_ylim(max(ymin, new_ymin), min(ymax, new_ymax))
+    if adapt_ylim:
+        new_ymin, new_ymax = 1e9, -1e9
+        for h, e in zip(bar_kwargs["height"], errors):
+            new_ymin = min(new_ymin, 0.98 * (h - e))
+            new_ymax = max(new_ymax, 1.02 * (h + e))
+        ymin, ymax = ax.get_ylim() 
+        ax.set_ylim(max(ymin, new_ymin), min(ymax, new_ymax))

h100_data.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d8558afaf128de6a17a9d87fdde0ca5450943eaeacff4a008615aa5d882b63ab
-size 3372040
+oid sha256:2f843d3f436d7919f67c071824fde3bc247b7e3a096a92d8abb191988d86a9d2
+size 2627476

mi325_data.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6dd568ec6695735008ba6934d8f52203c110cd7287cbb748af03d1fb8f9b1bae
-size 3539636
+oid sha256:628d50b1cbd1eca36f26d3e2c3e31d3062996ffbeb203f3c7ee04b604fec039e
+size 2674620