xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-05-17 03:34:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

improve table readability

Browse Source 
Signed-off-by: Tsai, Louie <louie.tsai@intel.com>
This commit is contained in:
Tsai, Louie 2025-12-19 19:16:59 -08:00
parent b735255f17
commit ba0bf189c8
1 changed files with 36 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
.buildkite/performance-benchmarks/scripts/compare-json-results.py
View File

@ -364,7 +364,7 @@ if __name__ == "__main__":
# For Plot feature, insert y axis from one of info_cols # For Plot feature, insert y axis from one of info_cols
raw_data_cols.insert(0, info_cols[y_axis_index]) raw_data_cols.insert(0, info_cols[y_axis_index])
filtered_info_cols = info_cols[:-2] filtered_info_cols = info_cols[:-4]
existing_group_cols = [ existing_group_cols = [
c for c in filtered_info_cols if c in output_df.columns c for c in filtered_info_cols if c in output_df.columns
] ]
@ -382,31 +382,57 @@ if __name__ == "__main__":
",".join(map(str, name)).replace(",", "_").replace("/", "-") ",".join(map(str, name)).replace(",", "_").replace("/", "-")
) )
group_html_name = "perf_comparison_" + group_name + ".html" group_html_name = "perf_comparison_" + group_name + ".html"
import html as _html
name_vals = name if isinstance(name, tuple) else (name,)
group_title_suffix = ", ".join(
f"{col}={val}" for col, val in zip(existing_group_cols, name_vals)
)
# ---------------------------------------------
# DROP group columns from DISPLAY ONLY
# ---------------------------------------------
display_group = group.drop(columns=existing_group_cols, errors="ignore")
metric_name = str(data_cols_to_compare[i]).lower() metric_name = str(data_cols_to_compare[i]).lower()
if "tok/s" in metric_name: if "tok/s" in metric_name:
html = group.to_html() html = (
f'<div style="font-size: 1.25em; font-weight: 600; margin: 12px 0;">'
f'{_html.escape(data_cols_to_compare[i])}'
f' — {_html.escape(group_title_suffix)}'
f'</div>\n'
+ display_group.to_html(index=False)
)
elif "ttft" in metric_name: elif "ttft" in metric_name:
styler = _highlight_threshold(group, args.ttft_max_ms).format( styler = _highlight_threshold(display_group, args.ttft_max_ms).format(
{c: "{:.2f}" for c in group.select_dtypes("number").columns}, {c: "{:.2f}" for c in display_group.select_dtypes("number").columns},
na_rep="—", na_rep="—",
) )
html = styler.to_html( html = (
table_attributes='border="1" class="dataframe"' f'<div style="font-size: 1.25em; font-weight: 600; margin: 12px 0;">'
f'{_html.escape(data_cols_to_compare[i])}'
f' — {_html.escape(group_title_suffix)}'
f'</div>\n'
+ styler.to_html(table_attributes='border="1" class="dataframe"')
) )
elif ( elif (
"tpot" in metric_name "tpot" in metric_name
or "median" in metric_name or "median" in metric_name
or "p99" in metric_name or "p99" in metric_name
): ):
styler = _highlight_threshold(group, args.tpot_max_ms).format( styler = _highlight_threshold(display_group, args.tpot_max_ms).format(
{c: "{:.2f}" for c in group.select_dtypes("number").columns}, {c: "{:.2f}" for c in display_group.select_dtypes("number").columns},
na_rep="—", na_rep="—",
) )
html = styler.to_html( html = (
table_attributes='border="1" class="dataframe"' f'<div style="font-size: 1.25em; font-weight: 600; margin: 12px 0;">'
f'{_html.escape(data_cols_to_compare[i])}'
f' — {_html.escape(group_title_suffix)}'
f'</div>\n'
+ styler.to_html(table_attributes='border="1" class="dataframe"')
) )
text_file.write(html_msgs_for_data_cols[i]) text_file.write(html_msgs_for_data_cols[i])
text_file.write(html) text_file.write(html)
with open(group_html_name, "a+") as sub_text_file: with open(group_html_name, "a+") as sub_text_file: