Function main

benchmarking/vllm/plot_tpot.py:369–403

()

                    fontweight="bold",
                    color=ann_color,
                )

        ax.set_xticks(range(n_conc))
        ax.set_xticklabels([str(int(c)) for c in concurrencies])
        ax.set_xlabel("Batch Size")
        ax.set_ylabel("Median TPOT (ms)")
        ax.set_title(f"E2E Speedups for {model} TP{tp}")
        ax.grid(axis="y", alpha=0.3)

        if legend_handles:
            ax.legend(
                legend_handles.values(),
                legend_handles.keys(),
                loc="upper left",
                framealpha=0.9,
            )

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


def main():
    parser = argparse.ArgumentParser(description="Plot TPOT and speedup from vLLM benchmarks")
    parser.add_argument(
        "--results-dir",
        type=Path,
        default=Path(__file__).parent,
        help="Directory containing model subdirectories (default: benchmarking/vllm/)",
    )
    parser.add_argument(
        "--fmt",
        default="png",
        help="Output image format (default: png)",
    )
    parser.add_argument(