| { |
| "name": "ai.onnx.TopK", |
| "id": "_ai_onnx_topk_webgpu_52be754", |
| "version": 1, |
| "license": "Apache-2.0", |
| "backend": { "type": "webgpu" }, |
| "digest": { |
| "algorithm": "sha256", |
| "files": { |
| "bench.json": "PurEWvvW6gbYNkFFlfRDPuV5w56IJxPpwT0HV0cXd0A=", |
| "manifest.json": "YqipHsTNrefVMyk2lSH1Vp9eXwjmXHibQ+FXiPhg3iw=", |
| "test.json": "Fr5n36m9d6NTCaetFemmXK+rj/6dXPaw6INsNyX9jqs=", |
| "topk-axis.wgsl.jinja": "Xl+bV55WupVkNPIof1xc0YfVFd0YlmTvWBoVKXuKH0s=", |
| "topk-large-block.wgsl.jinja": "r0f8d02UPzDntSxjLxFsP7ix3osvWFRmzsyt6YQS5K8=", |
| "topk-noop.wgsl.jinja": "rzwkvcZd8vJ3qLSosuI96CfUW+WUQyFxjc2Hgvq1bdo=", |
| "topk-portable-rows-smallk.wgsl.jinja": "WLo2yQv5mq1yMAW6vNAFbCSzRwuQ8fvM3MtuYH/d5PM=", |
| "topk-small-rows-batched.wgsl.jinja": "zaHFjRfth6UmWbFEYmjRbE2hpIdjuqyafituQUBTGvg=", |
| "topk-strided-smallk.wgsl.jinja": "pMdoMVSn+bYSKpP/FHMFLL2hnGeMQyakRPPDTITae4I=", |
| "topk-subgroup-rows.wgsl.jinja": "NPnHo98cYexGK/mRtrzxZvXOjmVF/1soDP/EaBRaApw=", |
| "topk-top1-last-axis.wgsl.jinja": "AYmJPcushgFs3ixDK6sJtDQDNEGbBk/CAAPWRMF3g+Y=" |
| } |
| }, |
| "provenance": { "kernel": { "sha": "91d990483a174128daf7673f3f37a7c890493ae1", "dirty": false } }, |
| "webgpu": { |
| "manifestSpec": "2.0", |
| "variants": { |
| "axis_smallk_tournament": ["topk-strided-smallk.wgsl.jinja"], |
| "last_axis_large_top1": ["topk-top1-last-axis.wgsl.jinja"], |
| "last_axis_top1": ["topk-top1-last-axis.wgsl.jinja"], |
| "axis_bitonic": ["topk-axis.wgsl.jinja"], |
| "top0_noop": ["topk-noop.wgsl.jinja"], |
| "subgroup_rows_smallk": ["topk-subgroup-rows.wgsl.jinja"], |
| "subgroup_min_rows_smallk": ["topk-subgroup-rows.wgsl.jinja"], |
| "portable_rows_smallk": ["topk-portable-rows-smallk.wgsl.jinja"], |
| "small_rows_batched": ["topk-small-rows-batched.wgsl.jinja"], |
| "last_axis_large_one_merge": ["topk-large-block.wgsl.jinja"], |
| "last_axis_large_two_merge": ["topk-large-block.wgsl.jinja"], |
| "last_axis_large_one_merge_int": ["topk-large-block.wgsl.jinja"], |
| "last_axis_large_two_merge_int": ["topk-large-block.wgsl.jinja"], |
| "axis_large_one_merge": ["topk-large-block.wgsl.jinja"], |
| "axis_large_two_merge": ["topk-large-block.wgsl.jinja"], |
| "last_axis_large_three_merge": ["topk-large-block.wgsl.jinja"], |
| "last_axis_large_three_merge_int": ["topk-large-block.wgsl.jinja"] |
| } |
| } |
| } |
|
|