ai.onnx.Pow / build /webgpu /bench.json
Xenova's picture
Xenova HF Staff
sync 91d990483a17
2f3cae7 verified
Raw
History Blame
8.15 kB
{
"tunableSpace": { "WORKGROUP_SIZE": [64, 128, 256] },
"cases": [
{
"name": "pow-f32-1m",
"preset": "smoke",
"vars": { "dtype": "float32", "count": 1048576 },
"inputs": {
"x": { "shape": [1048576], "dtype": "float32", "dist": "uniform", "seed": 502, "scale": 0.5, "offset": 1 },
"y": { "shape": [1048576], "dtype": "float32", "dist": "uniform", "seed": 503, "scale": 0.25, "offset": 1 }
},
"outputs": { "z": { "shape": [1048576], "dtype": "float32" } },
"bench": {
"primary": true,
"metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 3" }]
}
},
{
"name": "pow-f32-negative-square-1m",
"preset": "smoke",
"vars": { "dtype": "float32", "count": 1048576 },
"inputs": {
"x": { "shape": [1048576], "dtype": "float32", "dist": "normal", "seed": 504, "scale": 4 },
"y": { "shape": [1], "dtype": "float32", "dist": "constant", "value": 2 }
},
"outputs": { "z": { "shape": [1048576], "dtype": "float32" } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 3" }] }
},
{
"name": "pow-f32-scalar-exp2-1m",
"preset": "smoke",
"args": { "constantY": 2 },
"vars": { "dtype": "float32", "count": 1048576 },
"inputs": {
"x": { "shape": [1048576], "dtype": "float32", "dist": "normal", "seed": 505, "scale": 4 },
"y": { "shape": [1], "dtype": "float32", "dist": "constant", "value": 2 }
},
"outputs": { "z": { "shape": [1048576], "dtype": "float32" } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 3" }] }
},
{
"name": "pow-f32-scalar-exp2-8m",
"args": { "constantY": 2 },
"vars": { "dtype": "float32", "count": 8388608 },
"inputs": {
"x": { "shape": [8388608], "dtype": "float32", "dist": "normal", "seed": 505, "scale": 4 },
"y": { "shape": [1], "dtype": "float32", "dist": "constant", "value": 2 }
},
"outputs": { "z": { "shape": [8388608], "dtype": "float32" } },
"bench": {
"primary": true,
"metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }]
}
},
{
"name": "pow-f32-scalar-exp3-1m",
"preset": "smoke",
"args": { "constantY": 3 },
"vars": { "dtype": "float32", "count": 1048576 },
"inputs": {
"x": { "shape": [1048576], "dtype": "float32", "dist": "normal", "seed": 506, "scale": 2 },
"y": { "shape": [1], "dtype": "float32", "dist": "constant", "value": 3 }
},
"outputs": { "z": { "shape": [1048576], "dtype": "float32" } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 3" }] }
},
{
"name": "pow-f32-scalar-exp9-1m",
"preset": "smoke",
"args": { "constantY": 9 },
"vars": { "dtype": "float32", "count": 1048576 },
"inputs": {
"x": { "shape": [1048576], "dtype": "float32", "dist": "uniform", "seed": 507, "scale": 1.5, "offset": -0.75 },
"y": { "shape": [1], "dtype": "float32", "dist": "constant", "value": 9 }
},
"outputs": { "z": { "shape": [1048576], "dtype": "float32" } },
"bench": {
"primary": true,
"metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 3" }]
}
},
{
"name": "pow-f32-scalar-exp16-1m",
"preset": "smoke",
"args": { "constantY": 16 },
"vars": { "dtype": "float32", "count": 1048576 },
"inputs": {
"x": { "shape": [1048576], "dtype": "float32", "dist": "uniform", "seed": 508, "scale": 2, "offset": -1 },
"y": { "shape": [1], "dtype": "float32", "dist": "constant", "value": 16 }
},
"outputs": { "z": { "shape": [1048576], "dtype": "float32" } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 3" }] }
},
{
"name": "pow-f32-vector-exp9-1m",
"preset": "smoke",
"vars": { "dtype": "float32", "count": 1048576 },
"inputs": {
"x": { "shape": [1048576], "dtype": "float32", "dist": "uniform", "seed": 509, "scale": 1.5, "offset": -0.75 },
"y": { "shape": [1048576], "dtype": "float32", "dist": "constant", "value": 9 }
},
"outputs": { "z": { "shape": [1048576], "dtype": "float32" } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 3" }] }
},
{
"name": "pow-f16-1m",
"preset": "smoke",
"vars": { "dtype": "float16", "count": 1048576 },
"inputs": {
"x": { "shape": [1048576], "dtype": "float16", "dist": "uniform", "seed": 510, "scale": 0.5, "offset": 1 },
"y": { "shape": [1048576], "dtype": "float16", "dist": "uniform", "seed": 511, "scale": 0.25, "offset": 1 }
},
"outputs": { "z": { "shape": [1048576], "dtype": "float16" } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 3" }] }
},
{
"name": "pow-f16-scalar-exp2-1m",
"preset": "smoke",
"args": { "constantY": 2 },
"vars": { "dtype": "float16", "count": 1048576 },
"inputs": {
"x": { "shape": [1048576], "dtype": "float16", "dist": "uniform", "seed": 512, "scale": 1.5, "offset": 0.5 },
"y": { "shape": [1], "dtype": "float16", "dist": "constant", "value": 2 }
},
"outputs": { "z": { "shape": [1048576], "dtype": "float16" } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 3" }] }
},
{
"name": "pow-f32-broadcast-1m",
"preset": "smoke",
"vars": { "dtype": "float32", "count": 1048576 },
"inputs": {
"x": { "shape": [1024, 1024], "dtype": "float32", "dist": "uniform", "seed": 513, "scale": 0.5, "offset": 1 },
"y": { "shape": [1024], "dtype": "float32", "dist": "uniform", "seed": 514, "scale": 0.25, "offset": 1 }
},
"outputs": { "z": { "shape": [1024, 1024], "dtype": "float32" } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 3" }] }
},
{
"name": "pow-f32-unaligned-1m",
"preset": "smoke",
"vars": { "dtype": "float32", "count": 1048577 },
"inputs": {
"x": { "shape": [1048577], "dtype": "float32", "dist": "uniform", "seed": 515, "scale": 0.5, "offset": 1 },
"y": { "shape": [1048577], "dtype": "float32", "dist": "uniform", "seed": 516, "scale": 0.25, "offset": 1 }
},
"outputs": { "z": { "shape": [1048577], "dtype": "float32" } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 3" }] }
},
{
"name": "pow-f32-broadcast-transcendental-scalar-4m",
"preset": "stress",
"vars": { "dtype": "float32", "count": 4194304 },
"inputs": {
"x": { "shape": [4096, 1024], "dtype": "float32", "dist": "uniform", "seed": 5201, "scale": 3, "offset": 0.5 },
"y": { "shape": [4096, 1], "dtype": "float32", "dist": "uniform", "seed": 5202, "scale": 2, "offset": 0.25 }
},
"outputs": { "z": { "shape": [4096, 1024], "dtype": "float32", "dist": "empty" } },
"bench": {
"primary": true,
"metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }]
}
},
{
"name": "pow-f32-unaligned-transcendental-scalar-4m",
"preset": "stress",
"vars": { "dtype": "float32", "count": 4194305 },
"inputs": {
"x": { "shape": [4194305], "dtype": "float32", "dist": "uniform", "seed": 5203, "scale": 3, "offset": 0.5 },
"y": { "shape": [4194305], "dtype": "float32", "dist": "uniform", "seed": 5204, "scale": 2, "offset": 0.25 }
},
"outputs": { "z": { "shape": [4194305], "dtype": "float32", "dist": "empty" } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 3" }] }
}
]
}