Function plot_speedups

benchmarking/vllm/plot_tpot.py:223–253

(
    results_dir: Path, imgs_dir: Path, max_concurrency: int, fmms_name: str, fmt: str = "png"
)

            color="gray",
            style="italic",
        )

        fig.tight_layout()
        out = tpots_dir / f"{model}.{fmt}"
        fig.savefig(out, dpi=150, bbox_inches="tight")
        plt.close(fig)
        print(f"Saved to {out}")


# ---------------------------------------------------------------------------
# Per-model speedup plots (imgs/speedups/)
# ---------------------------------------------------------------------------


def plot_speedups(
    results_dir: Path, imgs_dir: Path, max_concurrency: int, fmms_name: str, fmt: str = "png"
):
    speedups_dir = imgs_dir / "speedups"
    tp = tp_from_dir(results_dir)

    for model in MODELS:
        model_dirs = resolve_model_dirs(results_dir, model)
        if not model_dirs:
            continue
        candidate_series = []
        for candidate_dir, candidate_name in [(FI2_DIR, FI2_NAME), (FMMS_DIR, fmms_name)]:
            sdf = paired_speedups(model_dirs, candidate_dir, max_concurrency)
            if not sdf.empty:
                candidate_series.append((candidate_name, sdf))
        if not candidate_series:
            continue

        speedups_dir.mkdir(parents=True, exist_ok=True)
        fig, ax = plt.subplots(figsize=FIGSIZE)