Skip to content

Commit 813815d

Browse files
committed
fix: Address review comment
1 parent 2cd9670 commit 813815d

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

tools/perf/benchmark.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,7 @@ do
134134
--batch_size ${bs} \
135135
--truncate \
136136
--backends ${backend} \
137-
--report "bert_base_perf_bs${bs}_backend_${backend}.csv.csv"
137+
--report "bert_base_perf_bs${bs}_backend_${backend}.csv"
138138
done
139139
done
140140

tools/perf/perf_run.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -593,8 +593,10 @@ def recordStats(backend, timings, precision, batch_size=1, compile_time_s=None):
593593
# Generate report
594594
print("Model Summary: ", model_name)
595595
summary = pd.DataFrame(results)
596-
summary["model_name"] = (
597-
model_name_torch if model_name_torch is not None else model_name
596+
summary.insert(
597+
loc=0,
598+
column="model_name",
599+
value=(model_name_torch if model_name_torch is not None else model_name),
598600
)
599601
print(summary)
600602
if args.report:

0 commit comments

Comments
 (0)