Skip to content

Commit 772cae8

Browse files
committed
Rename the output CSV file from performance-metrics.csv to microbenchmark_data.csv
1 parent bd6a730 commit 772cae8

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

compare_performance_metrics.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
plt.rcParams["figure.dpi"] = 600
1111
plt.rcParams["font.family"] = "Linux Biolinum"
1212

13-
df = pd.read_csv("performance-metrics.csv")
13+
df = pd.read_csv("microbenchmark_data.csv")
1414

1515
for task in df["Task"]:
1616
latex_item = f"\item {_BACKSLASH_CHAR}texttt{{{task.replace('scaled_dot_product_attention', 'sdpa').replace('rotary_position_embedding', 'rope').replace('_', f'{_BACKSLASH_CHAR}_')}}}"

run_experiments.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -155,7 +155,7 @@ def _benchmark_ops(ops, *args, **kwargs):
155155
for i, provider in enumerate(("NineToothed", "Triton", "PyTorch")):
156156
data[provider].append(results[i])
157157

158-
pd.DataFrame(data).set_index("Task").to_csv("performance-metrics.csv")
158+
pd.DataFrame(data).set_index("Task").to_csv("microbenchmark_data.csv")
159159

160160
for max_new_tokens in ALL_MAX_NEW_TOKENS:
161161
for backend in BACKENDS:

0 commit comments

Comments
 (0)