ai.onnx.Atan / build /webgpu /bench.json
Xenova's picture
Xenova HF Staff
sync 6fdf6301e2bb
6cf6f1d verified
Raw History Blame
13.4 kB
{
"tunableSpace": { "WORKGROUP_SIZE": [64, 128, 256] },
"cases": [
{
"name": "atan-f32-1m-vec4",
"preset": "smoke",
"vars": { "dtype": "float32", "count": 1048576 },
"inputs": { "x": { "shape": [1048576], "dtype": "float32", "dist": "normal", "seed": 820, "scale": 4 } },
"outputs": { "y": { "shape": [1048576], "dtype": "float32" } },
"bench": {
"primary": true,
"metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }]
}
},
{
"name": "atan-f32-scalar-non-mult4",
"preset": "smoke",
"vars": { "dtype": "float32", "count": 1048575 },
"inputs": { "x": { "shape": [1048575], "dtype": "float32", "dist": "normal", "seed": 821, "scale": 4 } },
"outputs": { "y": { "shape": [1048575], "dtype": "float32" } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
},
{
"name": "atan-f16-8m-vec4",
"preset": "smoke",
"vars": { "dtype": "float16", "count": 8388608 },
"inputs": { "x": { "shape": [8388608], "dtype": "float16", "dist": "normal", "seed": 822, "scale": 4 } },
"outputs": { "y": { "shape": [8388608], "dtype": "float16" } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
},
{
"name": "atan-f16-scalar-non-mult4",
"preset": "smoke",
"vars": { "dtype": "float16", "count": 8388607 },
"inputs": { "x": { "shape": [8388607], "dtype": "float16", "dist": "normal", "seed": 823, "scale": 4 } },
"outputs": { "y": { "shape": [8388607], "dtype": "float16" } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
},
{
"name": "atan-f32-8m-dispatch",
"preset": "smoke",
"vars": { "dtype": "float32", "count": 8388608 },
"inputs": { "x": { "shape": [8388608], "dtype": "float32", "dist": "normal", "seed": 824, "scale": 4 } },
"outputs": { "y": { "shape": [8388608], "dtype": "float32" } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
},
{
"name": "vector_tail_float32_n524287",
"preset": "smoke",
"vars": { "dtype": "float32", "count": 524287 },
"inputs": { "x": { "shape": [524287], "dtype": "float32", "dist": "normal", "seed": 823, "scale": 4 } },
"outputs": { "y": { "dtype": "float32", "shape": [524287] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
},
{
"name": "vector_tail_float32_n786431",
"preset": "smoke",
"vars": { "dtype": "float32", "count": 786431 },
"inputs": { "x": { "shape": [786431], "dtype": "float32", "dist": "normal", "seed": 823, "scale": 4 } },
"outputs": { "y": { "dtype": "float32", "shape": [786431] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
},
{
"name": "vector_tail_float32_n1048571",
"preset": "smoke",
"vars": { "dtype": "float32", "count": 1048571 },
"inputs": { "x": { "shape": [1048571], "dtype": "float32", "dist": "normal", "seed": 823, "scale": 4 } },
"outputs": { "y": { "dtype": "float32", "shape": [1048571] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
},
{
"name": "vector_tail_float32_n1048573",
"preset": "smoke",
"vars": { "dtype": "float32", "count": 1048573 },
"inputs": { "x": { "shape": [1048573], "dtype": "float32", "dist": "normal", "seed": 823, "scale": 4 } },
"outputs": { "y": { "dtype": "float32", "shape": [1048573] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
},
{
"name": "vector_tail_float32_n1048574",
"preset": "smoke",
"vars": { "dtype": "float32", "count": 1048574 },
"inputs": { "x": { "shape": [1048574], "dtype": "float32", "dist": "normal", "seed": 823, "scale": 4 } },
"outputs": { "y": { "dtype": "float32", "shape": [1048574] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
},
{
"name": "vector_tail_float32_n1048577",
"preset": "smoke",
"vars": { "dtype": "float32", "count": 1048577 },
"inputs": { "x": { "shape": [1048577], "dtype": "float32", "dist": "normal", "seed": 823, "scale": 4 } },
"outputs": { "y": { "dtype": "float32", "shape": [1048577] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
},
{
"name": "vector_tail_float32_n1048578",
"preset": "smoke",
"vars": { "dtype": "float32", "count": 1048578 },
"inputs": { "x": { "shape": [1048578], "dtype": "float32", "dist": "normal", "seed": 823, "scale": 4 } },
"outputs": { "y": { "dtype": "float32", "shape": [1048578] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
},
{
"name": "vector_tail_float32_n1048579",
"preset": "smoke",
"vars": { "dtype": "float32", "count": 1048579 },
"inputs": { "x": { "shape": [1048579], "dtype": "float32", "dist": "normal", "seed": 823, "scale": 4 } },
"outputs": { "y": { "dtype": "float32", "shape": [1048579] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
},
{
"name": "vector_tail_float32_n4194303",
"preset": "smoke",
"vars": { "dtype": "float32", "count": 4194303 },
"inputs": { "x": { "shape": [4194303], "dtype": "float32", "dist": "normal", "seed": 823, "scale": 4 } },
"outputs": { "y": { "dtype": "float32", "shape": [4194303] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
},
{
"name": "vector_tail_float32_n4194305",
"preset": "smoke",
"vars": { "dtype": "float32", "count": 4194305 },
"inputs": { "x": { "shape": [4194305], "dtype": "float32", "dist": "normal", "seed": 823, "scale": 4 } },
"outputs": { "y": { "dtype": "float32", "shape": [4194305] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
},
{
"name": "vector_tail_float32_n8388609",
"preset": "smoke",
"vars": { "dtype": "float32", "count": 8388609 },
"inputs": { "x": { "shape": [8388609], "dtype": "float32", "dist": "normal", "seed": 823, "scale": 4 } },
"outputs": { "y": { "dtype": "float32", "shape": [8388609] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
},
{
"name": "vector_tail_float32_n16777215",
"preset": "smoke",
"vars": { "dtype": "float32", "count": 16777215 },
"inputs": { "x": { "shape": [16777215], "dtype": "float32", "dist": "normal", "seed": 823, "scale": 4 } },
"outputs": { "y": { "dtype": "float32", "shape": [16777215] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
},
{
"name": "vector_tail_float32_n16777217",
"preset": "smoke",
"vars": { "dtype": "float32", "count": 16777217 },
"inputs": { "x": { "shape": [16777217], "dtype": "float32", "dist": "normal", "seed": 823, "scale": 4 } },
"outputs": { "y": { "dtype": "float32", "shape": [16777217] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
},
{
"name": "vector_tail_float16_n524287",
"preset": "smoke",
"vars": { "dtype": "float16", "count": 524287 },
"inputs": { "x": { "shape": [524287], "dtype": "float16", "dist": "normal", "seed": 823, "scale": 4 } },
"outputs": { "y": { "dtype": "float16", "shape": [524287] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
},
{
"name": "vector_tail_float16_n786431",
"preset": "smoke",
"vars": { "dtype": "float16", "count": 786431 },
"inputs": { "x": { "shape": [786431], "dtype": "float16", "dist": "normal", "seed": 823, "scale": 4 } },
"outputs": { "y": { "dtype": "float16", "shape": [786431] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
},
{
"name": "vector_tail_float16_n1048571",
"preset": "smoke",
"vars": { "dtype": "float16", "count": 1048571 },
"inputs": { "x": { "shape": [1048571], "dtype": "float16", "dist": "normal", "seed": 823, "scale": 4 } },
"outputs": { "y": { "dtype": "float16", "shape": [1048571] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
},
{
"name": "vector_tail_float16_n1048573",
"preset": "smoke",
"vars": { "dtype": "float16", "count": 1048573 },
"inputs": { "x": { "shape": [1048573], "dtype": "float16", "dist": "normal", "seed": 823, "scale": 4 } },
"outputs": { "y": { "dtype": "float16", "shape": [1048573] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
},
{
"name": "vector_tail_float16_n1048574",
"preset": "smoke",
"vars": { "dtype": "float16", "count": 1048574 },
"inputs": { "x": { "shape": [1048574], "dtype": "float16", "dist": "normal", "seed": 823, "scale": 4 } },
"outputs": { "y": { "dtype": "float16", "shape": [1048574] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
},
{
"name": "vector_tail_float16_n1048577",
"preset": "smoke",
"vars": { "dtype": "float16", "count": 1048577 },
"inputs": { "x": { "shape": [1048577], "dtype": "float16", "dist": "normal", "seed": 823, "scale": 4 } },
"outputs": { "y": { "dtype": "float16", "shape": [1048577] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
},
{
"name": "vector_tail_float16_n1048578",
"preset": "smoke",
"vars": { "dtype": "float16", "count": 1048578 },
"inputs": { "x": { "shape": [1048578], "dtype": "float16", "dist": "normal", "seed": 823, "scale": 4 } },
"outputs": { "y": { "dtype": "float16", "shape": [1048578] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
},
{
"name": "vector_tail_float16_n1048579",
"preset": "smoke",
"vars": { "dtype": "float16", "count": 1048579 },
"inputs": { "x": { "shape": [1048579], "dtype": "float16", "dist": "normal", "seed": 823, "scale": 4 } },
"outputs": { "y": { "dtype": "float16", "shape": [1048579] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
},
{
"name": "vector_tail_float16_n4194303",
"preset": "smoke",
"vars": { "dtype": "float16", "count": 4194303 },
"inputs": { "x": { "shape": [4194303], "dtype": "float16", "dist": "normal", "seed": 823, "scale": 4 } },
"outputs": { "y": { "dtype": "float16", "shape": [4194303] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
},
{
"name": "vector_tail_float16_n4194305",
"preset": "smoke",
"vars": { "dtype": "float16", "count": 4194305 },
"inputs": { "x": { "shape": [4194305], "dtype": "float16", "dist": "normal", "seed": 823, "scale": 4 } },
"outputs": { "y": { "dtype": "float16", "shape": [4194305] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
},
{
"name": "vector_tail_float16_n8388609",
"preset": "smoke",
"vars": { "dtype": "float16", "count": 8388609 },
"inputs": { "x": { "shape": [8388609], "dtype": "float16", "dist": "normal", "seed": 823, "scale": 4 } },
"outputs": { "y": { "dtype": "float16", "shape": [8388609] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
},
{
"name": "vector_tail_float16_n16777215",
"preset": "smoke",
"vars": { "dtype": "float16", "count": 16777215 },
"inputs": { "x": { "shape": [16777215], "dtype": "float16", "dist": "normal", "seed": 823, "scale": 4 } },
"outputs": { "y": { "dtype": "float16", "shape": [16777215] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
},
{
"name": "vector_tail_float16_n16777217",
"preset": "smoke",
"vars": { "dtype": "float16", "count": 16777217 },
"inputs": { "x": { "shape": [16777217], "dtype": "float16", "dist": "normal", "seed": 823, "scale": 4 } },
"outputs": { "y": { "dtype": "float16", "shape": [16777217] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
}
]
}