| { |
| "op": "ai.onnx.Sub", |
| "tunableSpace": { "WORKGROUP_SIZE": [64, 128, 256] }, |
| "cases": [ |
| { |
| "name": "sub-f32-4m", |
| "preset": "smoke", |
| "vars": { "dtype": "float32", "count": 4194304 }, |
| "inputs": { |
| "a": { "shape": [4194304], "dtype": "float32", "dist": "normal", "seed": 302, "scale": 2 }, |
| "b": { "shape": [4194304], "dtype": "float32", "dist": "normal", "seed": 303, "scale": 2 } |
| }, |
| "outputs": { "c": { "shape": [4194304], "dtype": "float32" } }, |
| "bench": { |
| "primary": true, |
| "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 3" }] |
| } |
| }, |
| { |
| "name": "sub-f16-4m-same-shape", |
| "preset": "smoke", |
| "vars": { "dtype": "float16", "count": 4194304 }, |
| "inputs": { |
| "a": { "shape": [4194304], "dtype": "float16", "dist": "normal", "seed": 312, "scale": 2 }, |
| "b": { "shape": [4194304], "dtype": "float16", "dist": "normal", "seed": 313, "scale": 2 } |
| }, |
| "outputs": { "c": { "shape": [4194304], "dtype": "float16" } }, |
| "bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 3" }] } |
| }, |
| { |
| "name": "sub-f32-rank4-broadcast", |
| "preset": "smoke", |
| "vars": { "dtype": "float32", "count": 2097152 }, |
| "inputs": { |
| "a": { "shape": [8, 128, 32, 64], "dtype": "float32", "dist": "normal", "seed": 322, "scale": 2 }, |
| "b": { "shape": [1, 128, 1, 64], "dtype": "float32", "dist": "normal", "seed": 323, "scale": 2 } |
| }, |
| "outputs": { "c": { "shape": [8, 128, 32, 64], "dtype": "float32" } }, |
| "bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] } |
| }, |
| { |
| "name": "sub-f32-scalar-rhs-odd-lastdim", |
| "preset": "smoke", |
| "vars": { "dtype": "float32", "count": 3000009 }, |
| "inputs": { |
| "a": { "shape": [3, 1000003], "dtype": "float32", "dist": "normal", "seed": 332, "scale": 2 }, |
| "b": { "shape": [], "dtype": "float32", "data": { "kind": "values", "values": [0.5] } } |
| }, |
| "outputs": { "c": { "shape": [3, 1000003], "dtype": "float32" } }, |
| "bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] } |
| }, |
| { |
| "name": "sub-f32-broadcast-scalar-fallback-rank3-odd", |
| "preset": "stress", |
| "vars": { "dtype": "float32", "count": 3407872 }, |
| "inputs": { |
| "a": { "shape": [512, 512, 13], "dtype": "float32", "dist": "normal", "seed": 371, "scale": 2 }, |
| "b": { "shape": [1, 1, 13], "dtype": "float32", "dist": "normal", "seed": 372, "scale": 2 } |
| }, |
| "outputs": { "c": { "shape": [512, 512, 13], "dtype": "float32", "dist": "empty" } }, |
| "bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] } |
| } |
| ] |
| } |
|
|
