Commit 975912c
fused/DSA: add TLE bucket_sort_topk with internal dispatch (#2272)
* fused/DSA: add TLE bucket_sort_topk with internal dispatch
* style: apply black formatting for bin_topk
* style: format use_final expression for black
* fused: export bucket_sort_topk from package init
* Fix lint
* Address bucket_sort_topk review comments
* Fix lint
---------
Co-authored-by: Qiming Teng <tengqm@outlook.com>1 parent b952161 commit 975912c
6 files changed
Lines changed: 938 additions & 50 deletions
File tree
- src/flag_gems
- fused
- DSA
- utils
- tests/test_DSA
0 commit comments