Updated the settings for the batched benchmarks
parent
d7314d4f8e
commit
4d34083039
|
@ -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))],
|
||||
}
|
||||
]
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue