diff options
author | Cedric Nugteren <web@cedricnugteren.nl> | 2022-10-14 08:18:00 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-10-14 08:18:00 +0200 |
commit | 03cffa83c5f7742f8ec0c5e762bb7048e38952f3 (patch) | |
tree | c072c2329337b3b163ceb78971f30ec4dd09134f | |
parent | d55840e16cda7cb8bb8b8c81e5294833c81913e8 (diff) | |
parent | c7d677e4a9b7090fb8abad4af83428f03432025c (diff) |
Merge pull request #447 from CNugteren/small_plotting_fixes
Fix two small issues in the plotting script
-rw-r--r-- | CHANGELOG | 3 | ||||
-rw-r--r-- | scripts/benchmark/plot.py | 10 |
2 files changed, 11 insertions, 2 deletions
@@ -1,3 +1,6 @@ +Development version (next version) +- Fixes two small issues in the plotting script + Version 1.5.3 - Fix a correctness issue with DGEMM on SM 7.5 Turing GPUs - Various minor fixes and enhancements diff --git a/scripts/benchmark/plot.py b/scripts/benchmark/plot.py index b0b63df3..2cf5d04d 100644 --- a/scripts/benchmark/plot.py +++ b/scripts/benchmark/plot.py @@ -76,7 +76,12 @@ def plot_graphs(results, file_name, num_rows, num_cols, for col in range(num_cols): index = row * num_cols + col result = results[index] - ax = axes[row, col] + if num_rows == 1: + ax = axes[col] + elif num_cols == 1: + ax = axes[row] + else: + ax = axes[row, col] plt.sca(ax) print("[plot] Plotting subplot %d" % index) @@ -90,7 +95,8 @@ def plot_graphs(results, file_name, num_rows, num_cols, x_ticks = [v if not (i % 2) else "" for i, v in enumerate(x_ticks)] # Sets the y-data - y_list = [[r[y_key] if y_key in r.keys() else 0 for r in result] for y_key in y_keys[index]] + y_list = [[r[y_key] if y_key in r.keys() and not isinstance(r[y_key], str) else 0 for r in result] + for y_key in y_keys[index]] y_max = [max(y) if len(y) else 1 for y in y_list] y_max = max(y_max) if len(y_list) > 0 else 1 |