Skip to content

Commit

Permalink
Metrics - llama-2 - Add package name and version to engine of onnxrun…
Browse files Browse the repository at this point in the history
…time (#19325)

### Description
<!-- Describe your changes. -->



### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->
  • Loading branch information
ironmangithub authored Feb 1, 2024
1 parent 1bdd7d9 commit 8a2646c
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -253,7 +253,7 @@ def save_results(results, filename):
# Save results to csv with standard format
records = []
for _, row in df.iterrows():
if row["Engine"] == "optimum-ort":
if row["Engine"] in ["optimum-ort", "onnxruntime"]:
record = BenchmarkRecord(
row["Model Name"], row["Precision"], "onnxruntime", row["Device"], ort_pkg_name, ort_pkg_version
)
Expand Down

0 comments on commit 8a2646c

Please sign in to comment.