trunk/a118dccc3732e8f8135396b6b3cee91ae6e3fa11: [Native DSL] Add CuTeDSL radix-select topk override (#184926)
🔒
https://github.com
«Registers a fp32, last-dim, contiguous topk override at aten::topk /
topk.values for K in {64, 128, 256, 512, 1024}. Fused radix-select +
bitonic sort, one CTA per row, single launch.
Two specialisations selected at comp...»
Automatische Weiterleitung...
1.5s