File size: 1,384 Bytes
9d0cf92 | 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 | {
"op": "ai.onnx.BitCast",
"cases": [
{
"name": "bitcast-f32-to-u32-33m",
"preset": "stress",
"provenance": { "notes": "Stress-only capacity case: input plus output occupy exactly 256 MiB of GPU storage." },
"attrs": { "to": 12 },
"inputs": { "input": { "dtype": "float32", "shape": [4096, 8192], "dist": "normal", "seed": 915, "scale": 1 } },
"outputs": { "output": { "dtype": "uint32", "shape": [4096, 8192] } },
"bench": { "primary": true, "metrics": [{ "type": "bandwidth", "value": "4096 * 8192 * 4 * 2" }] }
},
{
"name": "bitcast-f32-to-u32-1m",
"preset": "smoke",
"attrs": { "to": 12 },
"inputs": { "input": { "dtype": "float32", "shape": [1048576], "dist": "normal", "seed": 913, "scale": 1 } },
"outputs": { "output": { "dtype": "uint32", "shape": [1048576] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "(numel(shapes.input) + numel(shapes.output)) * 4" }] }
},
{
"name": "bitcast-f32-to-u32-scalar-fallback-4m",
"preset": "edge",
"attrs": { "to": 12 },
"inputs": { "input": { "dtype": "float32", "shape": [4194305], "dist": "normal", "seed": 917, "scale": 1 } },
"outputs": { "output": { "dtype": "uint32", "shape": [4194305] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "4194305 * 4 * 2" }] }
}
]
}
|