From 3651b5166406527b2085d5b6ed100ddfd4ea2c5b Mon Sep 17 00:00:00 2001 From: Cedric Nugteren Date: Thu, 25 Jan 2018 21:24:18 +0100 Subject: Improved the benchmark scripts; added gemmstridedbatched benchmark --- scripts/benchmark/plot.py | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) (limited to 'scripts/benchmark/plot.py') diff --git a/scripts/benchmark/plot.py b/scripts/benchmark/plot.py index 06fb278f..66d71595 100644 --- a/scripts/benchmark/plot.py +++ b/scripts/benchmark/plot.py @@ -109,9 +109,20 @@ def plot_graphs(results, file_name, num_rows, num_cols, assert len(label_names) == len(y_keys[index]) for i in range(len(y_keys[index])): color = COLORS[i] - if label_names[i] == "cuBLAS": + marker = MARKERS[i] + if label_names[i] in ["CLBlast", "CLBlast FP32"]: + color = BLUEISH + marker = "o-" + elif label_names[i] in ["CLBlast FP16"]: + color = PURPLISH + marker = ".-" + elif label_names[i] in ["clBLAS", "clBLAS FP32"]: + color = REDISH + marker = "x-" + elif label_names[i] == "cuBLAS": color = GREEN - ax.plot(x_location, y_list[i], MARKERS[i], label=label_names[i], color=color) + marker = ".-" + ax.plot(x_location, y_list[i], marker, label=label_names[i], color=color) # Sets the legend leg = ax.legend(loc=(0.02, 1.0 - legend_from_top - legend_from_top_per_item * len(y_keys[index])), -- cgit v1.2.3