| { |
| "cases": [ |
| { |
| "name": "1x256x256_f32", |
| "attrs": { "align_corners": 0 }, |
| "inputs": { |
| "theta": { |
| "dtype": "float32", |
| "shape": [1, 2, 3], |
| "data": { "kind": "values", "values": [1.0, 0.0, 0.0, 0.0, 1.0, 0.0] } |
| } |
| }, |
| "outputs": { "grid": { "dtype": "float32", "shape": [1, 256, 256, 2] } }, |
| "preset": "smoke" |
| }, |
| { |
| "name": "affine-grid3d-f32-b1-64x128x128-medical-volume", |
| "preset": "stress", |
| "vars": { "dtype": "float32", "points": 1048576 }, |
| "attrs": { "align_corners": 0 }, |
| "inputs": { |
| "theta": { |
| "dtype": "float32", |
| "shape": [1, 3, 4], |
| "data": { "kind": "values", "values": [1.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0] } |
| } |
| }, |
| "outputs": { "grid": { "dtype": "float32", "shape": [1, 64, 128, 128, 3] } }, |
| "bench": { |
| "primary": true, |
| "metrics": [{ "type": "bandwidth", "value": "args.points * 3 * dtypeBytes(args.dtype)" }] |
| } |
| } |
| ] |
| } |
|
|
