| { |
| "op": "ai.onnx.OneHot", |
| "cases": [ |
| { |
| "name": "tokens_vocab", |
| "inputs": { |
| "indices": { "dtype": "uint32", "shape": [128], "min": 0, "max": 4096 }, |
| "depth": { "dtype": "uint32", "shape": [1], "data": { "kind": "values", "values": [4096] } }, |
| "values": { "dtype": "float32", "shape": [2], "data": { "kind": "values", "values": [0.0, 1.0] } } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [128, 4096] } } |
| }, |
| { |
| "name": "tokens_vocab_2047x8192", |
| "preset": "smoke", |
| "inputs": { |
| "indices": { "dtype": "uint32", "shape": [2047], "dist": "uniform", "seed": 901, "min": 0, "max": 8000 }, |
| "depth": { "dtype": "uint32", "shape": [1], "data": { "kind": "values", "values": [8192] } }, |
| "values": { "dtype": "float32", "shape": [2], "data": { "kind": "values", "values": [0.0, 1.0] } } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [2047, 8192] } }, |
| "bench": { "primary": true, "metrics": [{ "type": "bandwidth", "value": "numel(shapes.output) * 4" }] } |
| }, |
| { |
| "name": "unaligned_depth_8191_generic_scalar", |
| "preset": "smoke", |
| "inputs": { |
| "indices": { "dtype": "uint32", "shape": [2047], "dist": "uniform", "seed": 902, "min": 0, "max": 8000 }, |
| "depth": { "dtype": "uint32", "shape": [1], "data": { "kind": "values", "values": [8191] } }, |
| "values": { "dtype": "float32", "shape": [2], "data": { "kind": "values", "values": [0.0, 1.0] } } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [2047, 8191] } }, |
| "bench": { "metrics": [{ "type": "bandwidth", "value": "numel(shapes.output) * 4" }] } |
| }, |
| { |
| "name": "axis0_generic_8192x2047", |
| "preset": "smoke", |
| "attrs": { "axis": 0 }, |
| "inputs": { |
| "indices": { "dtype": "uint32", "shape": [2047], "dist": "uniform", "seed": 903, "min": 0, "max": 8000 }, |
| "depth": { "dtype": "uint32", "shape": [1], "data": { "kind": "values", "values": [8192] } }, |
| "values": { "dtype": "float32", "shape": [2], "data": { "kind": "values", "values": [0.0, 1.0] } } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [8192, 2047] } }, |
| "bench": { "metrics": [{ "type": "bandwidth", "value": "numel(shapes.output) * 4" }] } |
| }, |
| { |
| "name": "axis1_middle_generic_2047x1024x8", |
| "preset": "smoke", |
| "attrs": { "axis": 1 }, |
| "inputs": { |
| "indices": { "dtype": "uint32", "shape": [2047, 8], "dist": "uniform", "seed": 904, "min": 0, "max": 1000 }, |
| "depth": { "dtype": "uint32", "shape": [1], "data": { "kind": "values", "values": [1024] } }, |
| "values": { "dtype": "float32", "shape": [2], "data": { "kind": "values", "values": [0.0, 1.0] } } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [2047, 1024, 8] } }, |
| "bench": { "metrics": [{ "type": "bandwidth", "value": "numel(shapes.output) * 4" }] } |
| }, |
| { |
| "name": "tokens_vocab_f16_2047x8192_vec4", |
| "preset": "smoke", |
| "inputs": { |
| "indices": { "dtype": "uint32", "shape": [2047], "dist": "uniform", "seed": 905, "min": 0, "max": 8000 }, |
| "depth": { "dtype": "uint32", "shape": [1], "data": { "kind": "values", "values": [8192] } }, |
| "values": { "dtype": "float16", "shape": [2], "data": { "kind": "values", "values": [0.0, 1.0] } } |
| }, |
| "outputs": { "output": { "dtype": "float16", "shape": [2047, 8192] } }, |
| "bench": { "metrics": [{ "type": "bandwidth", "value": "numel(shapes.output) * 2" }] } |
| }, |
| { |
| "name": "generic_middle_axis_rank5_3p9m", |
| "preset": "stress", |
| "attrs": { "axis": 1 }, |
| "inputs": { |
| "indices": { "dtype": "uint32", "shape": [256, 8, 8, 8], "dist": "uniform", "seed": 911, "min": 0, "max": 29 }, |
| "depth": { "dtype": "uint32", "shape": [1], "data": { "kind": "values", "values": [30] } }, |
| "values": { "dtype": "float32", "shape": [2], "data": { "kind": "values", "values": [0.0, 1.0] } } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [256, 30, 8, 8, 8], "dist": "empty" } }, |
| "bench": { "metrics": [{ "type": "bandwidth", "value": "numel(shapes.output) * 4" }] } |
| } |
| ] |
| } |
|
|
