diff options
Diffstat (limited to 'scripts/benchmark/benchmark.py')
-rw-r--r-- | scripts/benchmark/benchmark.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/scripts/benchmark/benchmark.py b/scripts/benchmark/benchmark.py index d84d5d98..0bb37c10 100644 --- a/scripts/benchmark/benchmark.py +++ b/scripts/benchmark/benchmark.py @@ -148,7 +148,9 @@ def benchmark_single(benchmark, comparisons, platform, device, num_runs, precisi # Retrieves the data from the benchmark settings file_name_suffix = "_tight" if tight_plot else "" pdf_file_name = os.path.join(output_folder, benchmark_name.lower() + "_plot" + file_name_suffix + ".pdf") - titles = [utils.precision_to_letter(precision) + b["name"].upper() + " " + b["title"] for b in benchmarks] + titles = [b["title"] if "BATCHED" in b["name"].upper() else + utils.precision_to_letter(precision) + b["name"].upper() + " " + b["title"] + for b in benchmarks] x_keys = [b["x_keys"] for b in benchmarks] y_keys = [["%s_%d" % (b["y_key"], i) for i in library_ids] for b in benchmarks] x_labels = [b["x_label"] for b in benchmarks] |