| { |
| "op": "ai.onnx.Col2Im", |
| "tunableSpace": { "WORKGROUP_SIZE": [64, 128, 256] }, |
| "cases": [ |
| { |
| "name": "nchw_1x16_32x32_k3", |
| "attrs": {}, |
| "inputs": { |
| "input": { "dtype": "float32", "shape": [1, 144, 900] }, |
| "image_shape": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [32, 32] } }, |
| "block_shape": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [3, 3] } } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [1, 16, 32, 32] } } |
| }, |
| { |
| "name": "nchw_4m_k3_overlap_healthy", |
| "preset": "smoke", |
| "attrs": {}, |
| "inputs": { |
| "input": { "dtype": "float32", "shape": [1, 144, 260100], "dist": "normal", "seed": 411, "scale": 0.2 }, |
| "image_shape": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [512, 512] } }, |
| "block_shape": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [3, 3] } } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [1, 16, 512, 512] } }, |
| "bench": { |
| "primary": true, |
| "metrics": [{ "type": "bandwidth", "value": "(numel(shapes.input) + numel(shapes.output)) * 4" }] |
| } |
| }, |
| { |
| "name": "nchw_dispatch_cliff_8m_gidy_fold", |
| "preset": "smoke", |
| "attrs": {}, |
| "inputs": { |
| "input": { "dtype": "float32", "shape": [1, 32, 262144], "dist": "normal", "seed": 412, "scale": 0.2 }, |
| "image_shape": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [512, 512] } }, |
| "block_shape": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [1, 1] } } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [1, 32, 512, 512] } }, |
| "bench": { "metrics": [{ "type": "bandwidth", "value": "(numel(shapes.input) + numel(shapes.output)) * 4" }] } |
| }, |
| { |
| "name": "nchw_k1x1_copy_baseline", |
| "preset": "smoke", |
| "attrs": {}, |
| "inputs": { |
| "input": { "dtype": "float32", "shape": [1, 16, 262144], "dist": "normal", "seed": 413, "scale": 0.2 }, |
| "image_shape": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [512, 512] } }, |
| "block_shape": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [1, 1] } } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [1, 16, 512, 512] } }, |
| "bench": { "metrics": [{ "type": "bandwidth", "value": "(numel(shapes.input) + numel(shapes.output)) * 4" }] } |
| }, |
| { |
| "name": "nchw-k1-fold-f32-b1c320-128x128-diffusion-feature", |
| "preset": "stress", |
| "attrs": {}, |
| "inputs": { |
| "input": { "dtype": "float32", "shape": [1, 320, 16384], "dist": "normal", "seed": 416, "scale": 0.2 }, |
| "image_shape": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [128, 128] } }, |
| "block_shape": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [1, 1] } } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [1, 320, 128, 128] } }, |
| "bench": { |
| "primary": true, |
| "metrics": [{ "type": "bandwidth", "value": "(numel(shapes.input) + numel(shapes.output)) * 4" }] |
| } |
| }, |
| { |
| "name": "nchw_k5x5_overlap_heavy_gather", |
| "preset": "stress", |
| "provenance": { |
| "notes": "Stress-only capacity case: the overlap-heavy gather declares 409 MiB of GPU-resident tensors." |
| }, |
| "attrs": {}, |
| "inputs": { |
| "input": { "dtype": "float32", "shape": [1, 400, 258064], "dist": "normal", "seed": 414, "scale": 0.1 }, |
| "image_shape": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [512, 512] } }, |
| "block_shape": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [5, 5] } } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [1, 16, 512, 512] } }, |
| "bench": { "metrics": [{ "type": "bandwidth", "value": "(numel(shapes.input) + numel(shapes.output)) * 4" }] } |
| }, |
| { |
| "name": "ncdhw_3d_2m_copy_variant", |
| "preset": "smoke", |
| "attrs": {}, |
| "inputs": { |
| "input": { "dtype": "float32", "shape": [1, 16, 131072], "dist": "normal", "seed": 415, "scale": 0.2 }, |
| "image_shape": { "dtype": "uint32", "shape": [3], "data": { "kind": "values", "values": [32, 64, 64] } }, |
| "block_shape": { "dtype": "uint32", "shape": [3], "data": { "kind": "values", "values": [1, 1, 1] } } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [1, 16, 32, 64, 64] } }, |
| "bench": { "metrics": [{ "type": "bandwidth", "value": "(numel(shapes.input) + numel(shapes.output)) * 4" }] } |
| } |
| ] |
| } |
|
|
