From 180532ea398891c2366b8ca6cfcc215208528f2c Mon Sep 17 00:00:00 2001 From: Cedric Nugteren Date: Sat, 27 Jan 2018 20:06:13 +0100 Subject: Some fixes to the benchmark scripts --- scripts/benchmark/plot.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'scripts/benchmark/plot.py') diff --git a/scripts/benchmark/plot.py b/scripts/benchmark/plot.py index 66d71595..6337b78f 100644 --- a/scripts/benchmark/plot.py +++ b/scripts/benchmark/plot.py @@ -62,10 +62,10 @@ def plot_graphs(results, file_name, num_rows, num_cols, # Initializes the plot size_x = plot_size * num_cols size_y = plot_size * num_rows + rcParams.update({'font.size': font_size}) fig, axes = plt.subplots(nrows=num_rows, ncols=num_cols, figsize=(size_x, size_y), facecolor='w', edgecolor='k') fig.text(.5, 0.92, title, horizontalalignment="center", fontsize=font_size_title) plt.subplots_adjust(wspace=w_space, hspace=h_space) - rcParams.update({'font.size': font_size}) # Loops over each subplot for row in range(num_rows): @@ -116,10 +116,10 @@ def plot_graphs(results, file_name, num_rows, num_cols, elif label_names[i] in ["CLBlast FP16"]: color = PURPLISH marker = ".-" - elif label_names[i] in ["clBLAS", "clBLAS FP32"]: + elif label_names[i] in ["clBLAS", "clBLAS FP32", "clBLAS (non-batched)"]: color = REDISH marker = "x-" - elif label_names[i] == "cuBLAS": + elif label_names[i] in ["cuBLAS", "cuBLAS (non-batched)"]: color = GREEN marker = ".-" ax.plot(x_location, y_list[i], marker, label=label_names[i], color=color) -- cgit v1.2.3