Updated the settings for the batched benchmarks

pull/148/head
Cedric Nugteren 2017-04-20 22:19:29 +02:00
parent d7314d4f8e
commit 4d34083039
1 changed files with 8 additions and 8 deletions

View File

@ -70,7 +70,7 @@ AXPYBATCHED = {
"x_label": "sizes (n)", "x_keys": ["n"],
"y_label": "GB/s (higher is better)", "y_keys": ["GBs_1", "GBs_2"],
"arguments": [{"batch_num": 8, "n": n, "incx": 1, "incy": 1, "step": 0, "num_steps": 1}
for n in utils.powers_of_2(utils.k(8), utils.m(2))],
for n in utils.powers_of_2(utils.k(8), utils.m(4))],
},
{
"name": "axpybatched", "num_runs": 20,
@ -78,7 +78,7 @@ AXPYBATCHED = {
"x_label": "sizes (n)", "x_keys": ["n"],
"y_label": "GB/s (higher is better)", "y_keys": ["GBs_1", "GBs_2"],
"arguments": [{"batch_num": 64, "n": n, "incx": 1, "incy": 1, "step": 0, "num_steps": 1}
for n in utils.powers_of_2(utils.k(8), utils.m(2))],
for n in utils.powers_of_2(utils.k(8), utils.m(4))],
},
{
"name": "axpybatched", "num_runs": 40,
@ -86,7 +86,7 @@ AXPYBATCHED = {
"x_label": "number of batches", "x_keys": ["batch_num"],
"y_label": "GB/s (higher is better)", "y_keys": ["GBs_1", "GBs_2"],
"arguments": [{"batch_num": b, "n": utils.k(512), "incx": 1, "incy": 1, "step": 1, "num_steps": 1}
for b in utils.powers_of_2(1, 256)],
for b in utils.powers_of_2(1, 512)],
}
]
}
@ -232,7 +232,7 @@ GEMMBATCHED = {
"x_label": "sizes (m=n=k)", "x_keys": ["m"],
"y_label": "GFLOPS (higher is better)", "y_keys": ["GFLOPS_1", "GFLOPS_2"],
"arguments": [{"batch_num": 8, "m": 32, "n": 32, "k": 32, "layout": 102,
"transA": 111, "transB": 111, "step": 32, "num_steps": 16}],
"transA": 111, "transB": 111, "step": 32, "num_steps": 20}],
},
{
"name": "gemmbatched", "num_runs": 20,
@ -240,15 +240,15 @@ GEMMBATCHED = {
"x_label": "sizes (m=n=k)", "x_keys": ["m"],
"y_label": "GFLOPS (higher is better)", "y_keys": ["GFLOPS_1", "GFLOPS_2"],
"arguments": [{"batch_num": 64, "m": 32, "n": 32, "k": 32, "layout": 102,
"transA": 111, "transB": 111, "step": 32, "num_steps": 16}],
"transA": 111, "transB": 111, "step": 32, "num_steps": 20}],
},
{
"name": "gemmbatched", "num_runs": 30,
"title": "m=n=k=64",
"title": "m=n=k=128",
"x_label": "number of batches", "x_keys": ["batch_num"],
"y_label": "GFLOPS (higher is better)", "y_keys": ["GFLOPS_1", "GFLOPS_2"],
"arguments": [{"batch_num": b, "m": 64, "n": 64, "k": 64, "layout": 102,
"transA": 111, "transB": 111} for b in utils.powers_of_2(1, utils.k(8))],
"arguments": [{"batch_num": b, "m": 128, "n": 128, "k": 128, "layout": 102,
"transA": 111, "transB": 111} for b in utils.powers_of_2(1, utils.k(16))],
}
]
}