Minor displaying improvements to the graph plotting scripts

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]