{ "op": "ai.onnx.LpNormalization", "tunableSpace": { "WORKGROUP_SIZE": [64, 128, 256], "SPLIT_TARGET_DIM": [128, 256, 512, 1024], "MAX_SPLITS": [8, 16, 32, 64, 128, 256] }, "cases": [ { "name": "4096x4096_last_axis_p2", "preset": "smoke", "vars": { "dtype": "float32", "rows": 4096, "dim": 4096 }, "attrs": { "axis": -1, "p": 2 }, "inputs": { "input": { "dtype": "float32", "shape": [4096, 4096], "dist": "normal", "seed": 760, "scale": 0.5 } }, "outputs": { "output": { "dtype": "float32", "shape": [4096, 4096] } }, "bench": { "primary": true, "metrics": [{ "type": "bandwidth", "value": "args.rows * args.dim * 2 * dtypeBytes(args.dtype)" }] } }, { "name": "4096x4096_axis0_p2_nonlast", "preset": "smoke", "vars": { "dtype": "float32", "rows": 4096, "dim": 4096 }, "attrs": { "axis": 0, "p": 2 }, "inputs": { "input": { "dtype": "float32", "shape": [4096, 4096], "dist": "normal", "seed": 761, "scale": 0.5 } }, "outputs": { "output": { "dtype": "float32", "shape": [4096, 4096] } }, "bench": { "metrics": [ { "type": "bandwidth", "name": "reduce-read plus divide read-write traffic", "value": "args.rows * args.dim * 3 * dtypeBytes(args.dtype)" } ] } }, { "name": "1024x1024_axis1", "preset": "smoke", "attrs": { "axis": 1, "p": 2 }, "inputs": { "input": { "dtype": "float32", "shape": [1024, 1024] } }, "outputs": { "output": { "dtype": "float32", "shape": [1024, 1024] } } }, { "name": "axis0_fewrows_hugedim_p2_f32_starved_reduce", "preset": "stress", "vars": { "dtype": "float32", "rows": 128, "dim": 16384 }, "attrs": { "axis": 0, "p": 2 }, "inputs": { "input": { "dtype": "float32", "shape": [16384, 128], "dist": "normal", "seed": 911, "scale": 0.5 } }, "outputs": { "output": { "dtype": "float32", "shape": [16384, 128], "dist": "empty" } }, "bench": { "metrics": [ { "type": "bandwidth", "name": "reduce-read plus divide read-write traffic", "value": "args.rows * args.dim * 3 * dtypeBytes(args.dtype)" } ] } }, { "name": "axis0_64rows_dim32k_p1_f32_sharper_starvation", "preset": "stress", "vars": { "dtype": "float32", "rows": 64, "dim": 32768 }, "attrs": { "axis": 0, "p": 1 }, "inputs": { "input": { "dtype": "float32", "shape": [32768, 64], "dist": "normal", "seed": 912, "scale": 0.5 } }, "outputs": { "output": { "dtype": "float32", "shape": [32768, 64], "dist": "empty" } }, "bench": { "metrics": [ { "type": "bandwidth", "name": "reduce-read plus divide read-write traffic", "value": "args.rows * args.dim * 3 * dtypeBytes(args.dtype)" } ] } } ] }