| { |
| "op": "ai.onnx.ConstantOfShape", |
| "cases": [ |
| { |
| "name": "mask_16m_f32", |
| "preset": "smoke", |
| "attrs": { "value": -10000 }, |
| "inputs": { "input": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [4096, 4096] } } }, |
| "outputs": { "output": { "dtype": "float32", "shape": [4096, 4096] } }, |
| "bench": { "primary": true, "metrics": [{ "type": "bandwidth", "value": "4096 * 4096 * 4" }] } |
| }, |
| { |
| "name": "1m_f32", |
| "attrs": { "value": -10000 }, |
| "inputs": { "input": { "dtype": "uint32", "shape": [1], "data": { "kind": "values", "values": [1048576] } } }, |
| "outputs": { "output": { "dtype": "float32", "shape": [1048576] } } |
| }, |
| { |
| "name": "fill_16m_f32_scalar_fallback", |
| "attrs": { "value": -10000 }, |
| "inputs": { "input": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [4097, 4097] } } }, |
| "outputs": { "output": { "dtype": "float32", "shape": [4097, 4097] } }, |
| "vars": { "dtype": "float32", "count": 16785409 }, |
| "bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype)" }] } |
| }, |
| { |
| "name": "mask_16m_f16", |
| "attrs": { "value": -10000 }, |
| "inputs": { "input": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [4096, 4096] } } }, |
| "outputs": { "output": { "dtype": "float16", "shape": [4096, 4096] } }, |
| "vars": { "dtype": "float16", "count": 16777216 }, |
| "bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype)" }] } |
| }, |
| { |
| "name": "fill_134m_f32_2dfold", |
| "attrs": { "value": -10000 }, |
| "inputs": { "input": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [16384, 8192] } } }, |
| "outputs": { "output": { "dtype": "float32", "shape": [16384, 8192] } }, |
| "vars": { "dtype": "float32", "count": 134217728 }, |
| "bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype)" }] } |
| }, |
| { |
| "name": "fill_16m_f16_scalar_fallback", |
| "preset": "stress", |
| "attrs": { "value": -10000 }, |
| "inputs": { "input": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [4097, 4097] } } }, |
| "outputs": { "output": { "dtype": "float16", "shape": [4097, 4097] } }, |
| "vars": { "dtype": "float16", "count": 16785409 }, |
| "bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype)" }] } |
| } |
| ] |
| } |
|
|
