File size: 2,104 Bytes
6324155 6c46cd9 6324155 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 | {
"cases": [
{
"name": "f32_16m",
"preset": "smoke",
"inputs": { "x": { "dtype": "float32", "shape": [16777216], "dist": "normal", "seed": 771, "scale": 1 } },
"outputs": { "y": { "dtype": "float32", "shape": [16777216] } },
"bench": { "primary": true, "metrics": [{ "type": "bandwidth", "value": "16777216 * 4 * 2" }] }
},
{
"name": "f32_1m",
"preset": "smoke",
"inputs": { "x": { "dtype": "float32", "shape": [1048576] } },
"outputs": { "y": { "dtype": "float32", "shape": [1048576] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "(numel(shapes.x) + numel(shapes.y)) * 4" }] }
},
{
"name": "f32_scalar_path_non_aligned",
"preset": "edge",
"inputs": { "x": { "dtype": "float32", "shape": [16777213], "dist": "normal", "seed": 883, "scale": 1 } },
"outputs": { "y": { "dtype": "float32", "shape": [16777213] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "16777213 * 4 * 2" }] }
},
{
"name": "float32_16m_tail1_mask",
"preset": "edge",
"inputs": { "x": { "dtype": "float32", "shape": [16777213], "dist": "normal", "seed": 884 } },
"outputs": { "y": { "dtype": "float32", "shape": [16777213] }, "mask": { "dtype": "bool", "shape": [16777213] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "201326556" }] }
},
{
"name": "float16_16m_tail1",
"preset": "edge",
"inputs": { "x": { "dtype": "float16", "shape": [16777213], "dist": "normal", "seed": 884 } },
"outputs": { "y": { "dtype": "float16", "shape": [16777213] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "67108852" }] }
},
{
"name": "float16_16m_tail1_mask",
"preset": "edge",
"inputs": { "x": { "dtype": "float16", "shape": [16777213], "dist": "normal", "seed": 884 } },
"outputs": { "y": { "dtype": "float16", "shape": [16777213] }, "mask": { "dtype": "bool", "shape": [16777213] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "134217704" }] }
}
]
}
|