diff options
-rw-r--r-- | scripts/benchmark/benchmark.py | 4 | ||||
-rw-r--r-- | scripts/benchmark/settings.py | 12 |
2 files changed, 9 insertions, 7 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] diff --git a/scripts/benchmark/settings.py b/scripts/benchmark/settings.py index 98b97530..7a2df2d6 100644 --- a/scripts/benchmark/settings.py +++ b/scripts/benchmark/settings.py @@ -64,7 +64,7 @@ AXPYBATCHED = { "benchmarks": [ { "name": "axpybatched", "num_runs": 10, - "title": "b=8", + "title": "num batches = 8", "x_label": "sizes (n)", "x_keys": ["n"], "y_label": "GB/s (higher is better)", "y_key": "GBs", "arguments": [{"batch_num": 8, "n": n, "incx": 1, "incy": 1, "step": 0, "num_steps": 1} @@ -72,7 +72,7 @@ AXPYBATCHED = { }, { "name": "axpybatched", "num_runs": 5, - "title": "b=64", + "title": "num batches = 64", "x_label": "sizes (n)", "x_keys": ["n"], "y_label": "GB/s (higher is better)", "y_key": "GBs", "arguments": [{"batch_num": 64, "n": n, "incx": 1, "incy": 1, "step": 0, "num_steps": 1} @@ -222,7 +222,7 @@ GEMMBATCHED = { "benchmarks": [ { "name": "gemmbatched", "num_runs": 20, - "title": "b=8", + "title": "num batches = 8", "x_label": "sizes (m=n=k)", "x_keys": ["m"], "y_label": "GFLOPS (higher is better)", "y_key": "GFLOPS", "arguments": [{"batch_num": 8, "m": 32, "n": 32, "k": 32, "layout": 102, @@ -230,7 +230,7 @@ GEMMBATCHED = { }, { "name": "gemmbatched", "num_runs": 10, - "title": "b=64", + "title": "num batches = 64", "x_label": "sizes (m=n=k)", "x_keys": ["m"], "y_label": "GFLOPS (higher is better)", "y_key": "GFLOPS", "arguments": [{"batch_num": 64, "m": 32, "n": 32, "k": 32, "layout": 102, @@ -252,7 +252,7 @@ GEMMSTRIDEDBATCHED = { "benchmarks": [ { "name": "gemmstridedbatched", "num_runs": 20, - "title": "b=8", + "title": "num batches = 8", "x_label": "sizes (m=n=k)", "x_keys": ["m"], "y_label": "GFLOPS (higher is better)", "y_key": "GFLOPS", "arguments": [{"batch_num": 8, "m": 32, "n": 32, "k": 32, "layout": 102, @@ -260,7 +260,7 @@ GEMMSTRIDEDBATCHED = { }, { "name": "gemmstridedbatched", "num_runs": 10, - "title": "b=64", + "title": "num batches = 64", "x_label": "sizes (m=n=k)", "x_keys": ["m"], "y_label": "GFLOPS (higher is better)", "y_key": "GFLOPS", "arguments": [{"batch_num": 64, "m": 32, "n": 32, "k": 32, "layout": 102, |