Files
Jeff Bolz 879d673759 vulkan: Implement top-k (#17418)
* vulkan: Implement top-k

Each pass launches workgroups that each sort 2^N elements (where N is usually 7-10)
and discards all but the top K. Repeat until only K are left. And there's a fast
path when K==1 to just find the max value rather than sorting.

* fix pipeline selection

* vulkan: Add N-ary search algorithm for topk

* microoptimizations
2025-11-26 16:45:43 +01:00
..
2025-09-29 17:43:58 +03:00
2025-11-25 15:31:43 +02:00
2025-08-05 22:10:36 +03:00
2025-08-05 22:10:36 +03:00
2025-11-25 15:31:43 +02:00
2025-09-05 11:34:28 +02:00