diff options
| author | Cedric Nugteren <web@cedricnugteren.nl> | 2018-01-25 21:24:18 +0100 |
|---|---|---|
| committer | Cedric Nugteren <web@cedricnugteren.nl> | 2018-01-25 21:24:18 +0100 |
| commit | 3651b5166406527b2085d5b6ed100ddfd4ea2c5b (patch) | |
| tree | 40f531b8e6e7937e808d6fea452188a42fca307f /scripts/benchmark/plot.py | |
| parent | 19fd263fb26f292a12f5d0fc047174934c6daf04 (diff) | |
Improved the benchmark scripts; added gemmstridedbatched benchmark
Diffstat (limited to 'scripts/benchmark/plot.py')
| -rw-r--r-- | scripts/benchmark/plot.py | 15 |
1 files changed, 13 insertions, 2 deletions
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])), |