Skip to content

Commit bd6a730

Browse files
committed
Refactor CSV export to occur within task-processing loop
1 parent 0517108 commit bd6a730

1 file changed

Lines changed: 1 addition & 3 deletions

File tree

run_experiments.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -155,9 +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-
df = pd.DataFrame(data)
159-
160-
df.set_index("Task").to_csv("performance-metrics.csv")
158+
pd.DataFrame(data).set_index("Task").to_csv("performance-metrics.csv")
161159

162160
for max_new_tokens in ALL_MAX_NEW_TOKENS:
163161
for backend in BACKENDS:

0 commit comments

Comments
 (0)