File size: 5,139 Bytes
386747d | 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 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 | {
"op": "ai.onnx.MeanVarianceNormalization",
"cases": [
{
"name": "1x32x32x32_default_axes",
"preset": "smoke",
"attrs": { "axes": [0, 2, 3] },
"inputs": { "x": { "dtype": "float32", "shape": [1, 32, 32, 32] } },
"outputs": { "y": { "dtype": "float32", "shape": [1, 32, 32, 32] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "3 * 4 * numel(shapes.x)" }] }
},
{
"name": "8x64x128x128_default_axes",
"preset": "smoke",
"attrs": { "axes": [0, 2, 3] },
"inputs": { "x": { "dtype": "float32", "shape": [8, 64, 128, 128] } },
"outputs": { "y": { "dtype": "float32", "shape": [8, 64, 128, 128] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "3 * 4 * numel(shapes.x)" }] }
},
{
"name": "16x64x64x64_spatial_axes",
"preset": "smoke",
"attrs": { "axes": [2, 3] },
"inputs": { "x": { "dtype": "float32", "shape": [16, 64, 64, 64] } },
"outputs": { "y": { "dtype": "float32", "shape": [16, 64, 64, 64] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "3 * 4 * numel(shapes.x)" }] }
},
{
"name": "f16_vec4_8x64x128x128_default_axes",
"preset": "smoke",
"vars": { "dtype": "float16" },
"attrs": { "axes": [0, 2, 3] },
"inputs": { "x": { "dtype": "float16", "shape": [8, 64, 128, 128], "dist": "normal", "seed": 900, "scale": 0.5 } },
"outputs": { "y": { "dtype": "float16", "shape": [8, 64, 128, 128] } },
"bench": {
"primary": true,
"metrics": [{ "type": "bandwidth", "value": "3 * dtypeBytes(args.dtype) * numel(shapes.x)" }]
}
},
{
"name": "f16_all_axes_flat_split_1x1x256x256",
"preset": "smoke",
"attrs": { "axes": [0, 1, 2, 3] },
"inputs": { "x": { "dtype": "float16", "shape": [1, 1, 256, 256], "dist": "normal", "seed": 906, "scale": 0.5 } },
"outputs": { "y": { "dtype": "float16", "shape": [1, 1, 256, 256] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "3 * 2 * numel(shapes.x)" }] }
},
{
"name": "vec4_align_cliff_8x64x128x130_unaligned_W",
"preset": "smoke",
"attrs": { "axes": [0, 2, 3] },
"inputs": { "x": { "dtype": "float32", "shape": [8, 64, 128, 130], "dist": "normal", "seed": 901, "scale": 0.5 } },
"outputs": { "y": { "dtype": "float32", "shape": [8, 64, 128, 130] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "3 * 4 * numel(shapes.x)" }] }
},
{
"name": "channel_axis_scalar_strided_8x64x128x128",
"preset": "smoke",
"attrs": { "axes": [1] },
"inputs": { "x": { "dtype": "float32", "shape": [8, 64, 128, 128], "dist": "normal", "seed": 902, "scale": 0.5 } },
"outputs": { "y": { "dtype": "float32", "shape": [8, 64, 128, 128] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "3 * 4 * numel(shapes.x)" }] }
},
{
"name": "channel_axis_wg_pow2_cliff_8x65x128x128",
"preset": "stress",
"attrs": { "axes": [1] },
"inputs": { "x": { "dtype": "float32", "shape": [8, 65, 128, 128], "dist": "normal", "seed": 905, "scale": 0.5 } },
"outputs": { "y": { "dtype": "float32", "shape": [8, 65, 128, 128] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "3 * 4 * numel(shapes.x)" }] }
},
{
"name": "rank3_scalar_8x4096x1024_reduce_hidden",
"preset": "stress",
"provenance": { "notes": "Stress-only capacity case: input plus output occupy exactly 256 MiB of GPU storage." },
"attrs": { "axes": [2] },
"inputs": { "x": { "dtype": "float32", "shape": [8, 4096, 1024], "dist": "normal", "seed": 903, "scale": 0.5 } },
"outputs": { "y": { "dtype": "float32", "shape": [8, 4096, 1024] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "3 * 4 * numel(shapes.x)" }] }
},
{
"name": "rank3_rows1_decode_hidden768",
"preset": "smoke",
"attrs": { "axes": [2] },
"inputs": { "x": { "dtype": "float32", "shape": [1, 1, 768], "dist": "normal", "seed": 904, "scale": 0.5 } },
"outputs": { "y": { "dtype": "float32", "shape": [1, 1, 768] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "3 * 4 * numel(shapes.x)" }] }
},
{
"name": "stress_all_axes_single_workgroup_rank4_3p9m",
"preset": "stress",
"attrs": { "axes": [0, 1, 2, 3] },
"inputs": { "x": { "dtype": "float32", "shape": [64, 64, 64, 15], "dist": "normal", "seed": 950, "scale": 0.5 } },
"outputs": { "y": { "dtype": "float32", "shape": [64, 64, 64, 15] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "3 * 4 * numel(shapes.x)" }] }
},
{
"name": "stress_tiny_reduce_launchbound_wg2_4m_rows",
"preset": "stress",
"attrs": { "axes": [2] },
"inputs": { "x": { "dtype": "float32", "shape": [4194304, 1, 2], "dist": "normal", "seed": 951, "scale": 0.5 } },
"outputs": { "y": { "dtype": "float32", "shape": [4194304, 1, 2] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "3 * 4 * numel(shapes.x)" }] }
}
]
}
|