| { |
| "op": "ai.onnx.STFT", |
| "tunableSpace": { "WORKGROUP_SIZE": [64, 128, 256] }, |
| "cases": [ |
| { |
| "name": "batch4_signal1024_frame256_step128", |
| "inputs": { |
| "signal": { "dtype": "float32", "shape": [4, 1024, 1] }, |
| "window": { "dtype": "float32", "shape": [256] } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [4, 7, 129, 2] } }, |
| "args": { "frameStep": 128 } |
| }, |
| { |
| "name": "fft_frame256_signal2048_step128_batch4", |
| "preset": "smoke", |
| "vars": { "frames": 60, "n": 256, "log2n": 8 }, |
| "inputs": { |
| "signal": { "dtype": "float32", "shape": [4, 2048, 1], "dist": "normal", "seed": 911 }, |
| "window": { "dtype": "float32", "shape": [256], "dist": "uniform", "seed": 912 } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [4, 15, 129, 2] } }, |
| "bench": { "metrics": [{ "type": "gflops", "value": "5 * args.frames * args.n * args.log2n" }] }, |
| "args": { "frameStep": 128 } |
| }, |
| { |
| "name": "fft_frame512_signal4096_step256_batch2", |
| "preset": "smoke", |
| "vars": { "frames": 30, "n": 512, "log2n": 9 }, |
| "inputs": { |
| "signal": { "dtype": "float32", "shape": [2, 4096, 1], "dist": "normal", "seed": 913 }, |
| "window": { "dtype": "float32", "shape": [512], "dist": "uniform", "seed": 914 } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [2, 15, 257, 2] } }, |
| "bench": { "metrics": [{ "type": "gflops", "value": "5 * args.frames * args.n * args.log2n" }] }, |
| "args": { "frameStep": 256 } |
| }, |
| { |
| "name": "fft_frame1024_signal131072_step256_audio", |
| "preset": "smoke", |
| "vars": { "frames": 509, "n": 1024, "log2n": 10 }, |
| "inputs": { |
| "signal": { "dtype": "float32", "shape": [1, 131072, 1], "dist": "normal", "seed": 915 }, |
| "window": { "dtype": "float32", "shape": [1024], "dist": "uniform", "seed": 916 } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [1, 509, 513, 2] } }, |
| "bench": { "primary": true, "metrics": [{ "type": "gflops", "value": "5 * args.frames * args.n * args.log2n" }] }, |
| "args": { "frameStep": 256 } |
| }, |
| { |
| "name": "pow2_gate_FFT_win512_signal8192_step160", |
| "preset": "smoke", |
| "vars": { "batch": 2, "frames": 49, "bins": 257, "n": 512, "log2n": 9 }, |
| "inputs": { |
| "signal": { "dtype": "float32", "shape": [2, 8192, 1], "dist": "normal", "seed": 2001 }, |
| "window": { "dtype": "float32", "shape": [512], "dist": "uniform", "seed": 2002 } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [2, 49, 257, 2] } }, |
| "bench": { "metrics": [{ "type": "gflops", "value": "5 * args.batch * args.frames * args.n * args.log2n" }] }, |
| "args": { "frameStep": 160 } |
| }, |
| { |
| "name": "mixed_radix_win480_signal8192_step160_tail15", |
| "preset": "smoke", |
| "vars": { "batch": 2, "frames": 49, "bins": 241, "n": 480, "log2radix2": 5, "tail": 15 }, |
| "inputs": { |
| "signal": { "dtype": "float32", "shape": [2, 8192, 1], "dist": "normal", "seed": 2001 }, |
| "window": { "dtype": "float32", "shape": [480], "dist": "uniform", "seed": 2003 } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [2, 49, 241, 2] } }, |
| "bench": { |
| "metrics": [ |
| { |
| "type": "gflops", |
| "value": "args.batch * args.frames * (5 * args.n * args.log2radix2 + 8 * args.bins * args.tail)" |
| } |
| ] |
| }, |
| "args": { "frameStep": 160 } |
| }, |
| { |
| "name": "mixed_radix_win400_signal16000_step160_batch4_tail25", |
| "preset": "smoke", |
| "vars": { "batch": 4, "frames": 98, "bins": 201, "n": 400, "log2radix2": 4, "tail": 25 }, |
| "inputs": { |
| "signal": { "dtype": "float32", "shape": [4, 16000, 1], "dist": "normal", "seed": 2010 }, |
| "window": { "dtype": "float32", "shape": [400], "dist": "uniform", "seed": 2011 } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [4, 98, 201, 2] } }, |
| "bench": { |
| "metrics": [ |
| { |
| "type": "gflops", |
| "value": "args.batch * args.frames * (5 * args.n * args.log2radix2 + 8 * args.bins * args.tail)" |
| } |
| ] |
| }, |
| "args": { "frameStep": 160 } |
| }, |
| { |
| "name": "radix2_win512_signal16000_step160_batch4", |
| "preset": "smoke", |
| "vars": { "batch": 4, "frames": 97, "bins": 257, "n": 512, "log2n": 9 }, |
| "inputs": { |
| "signal": { "dtype": "float32", "shape": [4, 16000, 1], "dist": "normal", "seed": 2010 }, |
| "window": { "dtype": "float32", "shape": [512], "dist": "uniform", "seed": 2012 } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [4, 97, 257, 2] } }, |
| "bench": { "metrics": [{ "type": "gflops", "value": "5 * args.batch * args.frames * args.n * args.log2n" }] }, |
| "args": { "frameStep": 160 } |
| }, |
| { |
| "name": "dispatch_cliff_NAIVE_batch48_frames1397_win512_step160", |
| "preset": "stress", |
| "vars": { "batch": 48, "frames": 1397, "bins": 257, "n": 512, "log2n": 9, "signalLength": 224000 }, |
| "inputs": { |
| "signal": { "dtype": "float32", "shape": [48, 224000, 1], "dist": "normal", "seed": 2020 }, |
| "window": { "dtype": "float32", "shape": [512], "dist": "uniform", "seed": 2021 } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [48, 1397, 257, 2] } }, |
| "bench": { |
| "metrics": [ |
| { "type": "gflops", "value": "5 * args.batch * args.frames * args.n * args.log2n" }, |
| { |
| "type": "bandwidth", |
| "value": "4 * (args.batch * args.signalLength + args.n + args.batch * args.frames * args.bins * 2)" |
| } |
| ] |
| }, |
| "args": { "frameStep": 160 } |
| }, |
| { |
| "name": "dispatch_cliff_sibling_FFT_batch32_frames1397_win512_step160", |
| "preset": "stress", |
| "vars": { "batch": 32, "frames": 1397, "bins": 257, "n": 512, "log2n": 9, "signalLength": 224000 }, |
| "inputs": { |
| "signal": { "dtype": "float32", "shape": [32, 224000, 1], "dist": "normal", "seed": 2020 }, |
| "window": { "dtype": "float32", "shape": [512], "dist": "uniform", "seed": 2022 } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [32, 1397, 257, 2] } }, |
| "bench": { |
| "metrics": [ |
| { "type": "gflops", "value": "5 * args.batch * args.frames * args.n * args.log2n" }, |
| { |
| "type": "bandwidth", |
| "value": "4 * (args.batch * args.signalLength + args.n + args.batch * args.frames * args.bins * 2)" |
| } |
| ] |
| }, |
| "args": { "frameStep": 160 } |
| }, |
| { |
| "name": "large_window_FFT_win4096_signal131072_step1024", |
| "preset": "smoke", |
| "vars": { "batch": 1, "frames": 125, "bins": 2049, "n": 4096, "log2n": 12 }, |
| "inputs": { |
| "signal": { "dtype": "float32", "shape": [1, 131072, 1], "dist": "normal", "seed": 2030 }, |
| "window": { "dtype": "float32", "shape": [4096], "dist": "uniform", "seed": 2031 } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [1, 125, 2049, 2] } }, |
| "bench": { "metrics": [{ "type": "gflops", "value": "5 * args.batch * args.frames * args.n * args.log2n" }] }, |
| "args": { "frameStep": 1024 } |
| }, |
| { |
| "name": "large_window_FFT_win8192_signal131072_step2048", |
| "preset": "stress", |
| "vars": { "batch": 1, "frames": 61, "bins": 4097, "n": 8192, "log2n": 13 }, |
| "inputs": { |
| "signal": { "dtype": "float32", "shape": [1, 131072, 1], "dist": "normal", "seed": 2030 }, |
| "window": { "dtype": "float32", "shape": [8192], "dist": "uniform", "seed": 2033 } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [1, 61, 4097, 2] } }, |
| "bench": { "metrics": [{ "type": "gflops", "value": "5 * args.batch * args.frames * args.n * args.log2n" }] }, |
| "args": { "frameStep": 2048 } |
| }, |
| { |
| "name": "large_window_sibling_FFT_win2048_signal131072_step1024", |
| "preset": "smoke", |
| "vars": { "batch": 1, "frames": 127, "bins": 1025, "n": 2048, "log2n": 11 }, |
| "inputs": { |
| "signal": { "dtype": "float32", "shape": [1, 131072, 1], "dist": "normal", "seed": 2030 }, |
| "window": { "dtype": "float32", "shape": [2048], "dist": "uniform", "seed": 2032 } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [1, 127, 1025, 2] } }, |
| "bench": { "metrics": [{ "type": "gflops", "value": "5 * args.batch * args.frames * args.n * args.log2n" }] }, |
| "args": { "frameStep": 1024 } |
| }, |
| { |
| "name": "mixed_radix_win1000_tail125_batch2_step250_twosided", |
| "preset": "stress", |
| "vars": { "batch": 2, "frames": 13, "bins": 1000, "n": 1000 }, |
| "attrs": { "onesided": 0 }, |
| "inputs": { |
| "signal": { "dtype": "float32", "shape": [2, 4000, 1], "dist": "normal", "seed": 3120 }, |
| "window": { "dtype": "float32", "shape": [1000], "dist": "uniform", "seed": 3121 } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [2, 13, 1000, 2], "dist": "empty" } }, |
| "bench": { "metrics": [{ "type": "gflops", "value": "5 * args.batch * args.frames * args.bins * args.n" }] }, |
| "args": { "frameStep": 250 } |
| }, |
| { |
| "name": "tiled_win2047_nonpow2_boundary_batch2_step256", |
| "preset": "stress", |
| "vars": { "batch": 2, "frames": 30, "bins": 1024, "n": 2047 }, |
| "inputs": { |
| "signal": { "dtype": "float32", "shape": [2, 9471, 1], "dist": "normal", "seed": 3101 }, |
| "window": { "dtype": "float32", "shape": [2047], "dist": "uniform", "seed": 3102 } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [2, 30, 1024, 2], "dist": "empty" } }, |
| "bench": { "metrics": [{ "type": "gflops", "value": "5 * args.batch * args.frames * args.bins * args.n" }] }, |
| "args": { "frameStep": 256 } |
| }, |
| { |
| "name": "mixed_radix_win1536_batch4_step384", |
| "preset": "stress", |
| "vars": { "batch": 4, "frames": 13, "bins": 769, "n": 1536 }, |
| "inputs": { |
| "signal": { "dtype": "float32", "shape": [4, 6144, 1], "dist": "normal", "seed": 3110 }, |
| "window": { "dtype": "float32", "shape": [1536], "dist": "uniform", "seed": 3111 } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [4, 13, 769, 2], "dist": "empty" } }, |
| "bench": { "metrics": [{ "type": "gflops", "value": "5 * args.batch * args.frames * args.bins * args.n" }] }, |
| "args": { "frameStep": 384 } |
| }, |
| { |
| "name": "tiled_win513_nonpow2_boundary_batch2_signal8193_step160", |
| "preset": "stress", |
| "vars": { "batch": 2, "frames": 49, "bins": 257, "n": 513 }, |
| "attrs": { "onesided": 1 }, |
| "inputs": { |
| "signal": { "dtype": "float32", "shape": [2, 8193, 1], "dist": "normal", "seed": 3120 }, |
| "window": { "dtype": "float32", "shape": [513], "dist": "uniform", "seed": 3121 } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [2, 49, 257, 2], "dist": "empty" } }, |
| "bench": { "metrics": [{ "type": "gflops", "value": "5 * args.batch * args.frames * args.bins * args.n" }] }, |
| "args": { "frameStep": 160 } |
| }, |
| { |
| "name": "audio-no-window-frame512-batch8-signal160000-pathology", |
| "preset": "stress", |
| "provenance": { |
| "notes": "Large no-window transform that exercises rank3_no_window; the windowed FFT control isolates the optional-window route gap." |
| }, |
| "vars": { "batch": 8, "frames": 997, "bins": 257, "n": 512, "log2n": 9 }, |
| "attrs": { "onesided": 1 }, |
| "inputs": { "signal": { "dtype": "float32", "shape": [8, 160000, 1], "dist": "normal", "seed": 3130 } }, |
| "outputs": { "output": { "dtype": "float32", "shape": [8, 997, 257, 2], "dist": "empty" } }, |
| "bench": { "metrics": [{ "type": "gflops", "value": "5 * args.batch * args.frames * args.n * args.log2n" }] }, |
| "args": { "frameStep": 160, "frameLength": 512 } |
| } |
| ] |
| } |
|
|
