| { |
| "op": "ai.onnx.EyeLike", |
| "cases": [ |
| { |
| "name": "eyelike-f32-2048-square", |
| "inputs": { "input": { "dtype": "float32", "shape": [2048, 2048] } }, |
| "outputs": { "output": { "dtype": "float32", "shape": [2048, 2048] } } |
| }, |
| { |
| "name": "eyelike-f32-4097-square-tail-healthy", |
| "preset": "smoke", |
| "vars": { "dtype": "float32", "count": 16785409 }, |
| "inputs": { "input": { "dtype": "float32", "shape": [4097, 4097] } }, |
| "outputs": { "output": { "dtype": "float32", "shape": [4097, 4097] } }, |
| "bench": { "primary": true, "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype)" }] } |
| }, |
| { |
| "name": "eyelike-f16-4096-square-narrow-store", |
| "preset": "smoke", |
| "vars": { "dtype": "float16", "count": 16777216 }, |
| "inputs": { "input": { "dtype": "float16", "shape": [4096, 4096] } }, |
| "outputs": { "output": { "dtype": "float16", "shape": [4096, 4096] } }, |
| "bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype)" }] } |
| }, |
| { |
| "name": "eyelike-int8-4096-square-word-store-pathology", |
| "preset": "smoke", |
| "vars": { "dtype": "int8", "count": 16777216 }, |
| "inputs": { "input": { "dtype": "int8", "shape": [4096, 4096] } }, |
| "outputs": { "output": { "dtype": "int8", "shape": [4096, 4096] } }, |
| "bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * 4" }] } |
| }, |
| { |
| "name": "eyelike-f32-134m-2dfold-dispatch-cliff", |
| "preset": "stress", |
| "provenance": { "notes": "Stress-only capacity case: the input and output together occupy 1 GiB of GPU storage." }, |
| "vars": { "dtype": "float32", "count": 134217728 }, |
| "inputs": { "input": { "dtype": "float32", "shape": [16384, 8192] } }, |
| "outputs": { "output": { "dtype": "float32", "shape": [16384, 8192] } }, |
| "bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype)" }] } |
| }, |
| { |
| "name": "eyelike-f32-8192-square-long-context-mask", |
| "preset": "stress", |
| "vars": { "dtype": "float32", "count": 67108864 }, |
| "inputs": { "input": { "dtype": "float32", "shape": [8192, 8192] } }, |
| "outputs": { "output": { "dtype": "float32", "shape": [8192, 8192] } }, |
| "bench": { "primary": true, "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype)" }] } |
| } |
| ] |
| } |
|
|
