Skip to content

Commit 01457fe

Browse files
authored
[CI][benchmarks] Update reporting schema for third party benchmarks (#4825)
First part of #4824
1 parent 5f430ce commit 01457fe

File tree

2 files changed

+13
-8
lines changed

2 files changed

+13
-8
lines changed

benchmarks/third_party/e2e-flex_attention/transform_results.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -62,12 +62,13 @@ def parse_llm_log(log_file_path, tag, model, max_new_tokens, batch_size):
6262
# Create one row for each metric
6363
for metric_name, metric_value in metrics.items():
6464
row = {
65-
'benchmark': 'e2e-flex-attention',
6665
'run_uuid': run_uuid,
67-
'datetime': current_datetime,
66+
'ts': current_datetime,
67+
'benchmark_group': 'e2e-flex-attention',
68+
'benchmark': 'e2e-flex-attention',
6869
'compiler': 'triton',
69-
'metric_name': metric_name,
70-
'metric_value': metric_value,
70+
'value_name': metric_name + '_s',
71+
'value': metric_value,
7172
'params': params_json,
7273
'tag': tag,
7374
}

benchmarks/third_party/liger_kernels/transform.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -18,17 +18,21 @@ def parse_args():
1818
def transform_df(df, tag):
1919
df_results = pd.DataFrame()
2020

21-
df = df[df["metric_name"].eq("speed") & ~df["gpu_name"].str.contains("NVIDIA")]
21+
df = df[~df["gpu_name"].str.contains("NVIDIA")]
2222

2323
if len(df) == 0:
2424
raise ValueError("No new results found, did all benchmarks just fail?")
2525

26-
df_results["benchmark"] = df["kernel_name"] + "-" + df["kernel_operation_mode"]
26+
# df_results["benchmark"] = df["kernel_name"] + "-" + df["kernel_operation_mode"]
27+
df_results["benchmark"] = df["kernel_name"]
28+
mapping = {"speed": "_ms", "memory": "_memory_mb"}
29+
df_results["value_name"] = df["kernel_operation_mode"] + df["metric_name"].map(mapping)
30+
df_results["value"] = df["y_value_50"]
31+
df_results["benchmark_group"] = "liger"
2732
df_results["run_uuid"] = uuid.uuid4().hex # Generate a unique run ID
28-
df_results["datetime"] = datetime.datetime.now()
33+
df_results["ts"] = datetime.datetime.now()
2934
df_results["compiler"] = df["kernel_provider"]
3035
# Use the 50th percentile value.
31-
df_results["time_ms"] = df["y_value_50"]
3236
df_results["comment"] = "" # Empty comment
3337

3438
# Create the parameters JSON, handling different x_value types correctly.

0 commit comments

Comments
 (0)