fix a mulitple TP/PP size comparison issue in a table

Signed-off-by: Tsai, Louie <louie.tsai@intel.com>
This commit is contained in:
Tsai, Louie 2025-12-22 00:02:42 -08:00
parent 76862427f1
commit 898e868d28

View File

@ -20,8 +20,8 @@ DEFAULT_INFO_COLS = [
"Dataset Name", "Dataset Name",
"Input Len", "Input Len",
"Output Len", "Output Len",
"TP Size", # "TP Size",
"PP Size", # "PP Size",
"# of max concurrency.", "# of max concurrency.",
"qps", "qps",
] ]
@ -272,7 +272,7 @@ def _apply_two_decimals(
num_cols = df.select_dtypes("number").columns num_cols = df.select_dtypes("number").columns
if len(num_cols) == 0: if len(num_cols) == 0:
return styler return styler
return styler.format({c: "{:.2f}" for c in num_cols}, na_rep="") return styler.format({c: "{:.2f}" for c in num_cols}, na_rep="")
# ----------------------------- # -----------------------------
@ -442,7 +442,7 @@ def build_valid_max_concurrency_summary_html(
if c == "Configuration": if c == "Configuration":
continue continue
# default argument binds per-column formatter correctly # default argument binds per-column formatter correctly
formatters[c] = lambda v: "" if pd.isna(v) else f"{float(v):.2f}" formatters[c] = lambda v: "" if pd.isna(v) else f"{float(v):.2f}"
styler = summary_df.style.format(formatters) styler = summary_df.style.format(formatters)
@ -730,7 +730,8 @@ def write_report_group_first(
for metric_label, (df, _) in metric_cache.items() for metric_label, (df, _) in metric_cache.items()
} }
with open("perf_comparison.html", "w") as main_fh: with open("perf_comparison.html", "w", encoding="utf-8") as main_fh:
main_fh.write('<meta charset="utf-8">\n')
for gkey in group_keys: for gkey in group_keys:
gkey_tuple = normalize_group_key(gkey) gkey_tuple = normalize_group_key(gkey)
suffix = build_group_suffix(group_cols_canonical, gkey_tuple) suffix = build_group_suffix(group_cols_canonical, gkey_tuple)
@ -743,9 +744,9 @@ def write_report_group_first(
) )
main_fh.write(group_header) main_fh.write(group_header)
with open(sub_path, "w") as sub_fh: with open(sub_path, "w", encoding="utf-8") as sub_fh:
sub_fh.write('<meta charset="utf-8">\n')
sub_fh.write(group_header) sub_fh.write(group_header)
tput_group_df = None tput_group_df = None
ttft_group_df = None ttft_group_df = None
tpot_group_df = None tpot_group_df = None