fix: reduce RANK window top-K memory from O(input) to O(K + ties) per partition - #24591

Merged

jayzhan211 merged 5 commits into

apache:mainfrom

jayzhan211:fix/partitioned-topk-rank-memory

Aug 26, 2026

Commits on Aug 23, 2026

fix: stop PartitionedTopKRank from pinning whole input batches per partition
jayzhan211
committed
9ffa610
fix: clarify comments in PartitionedTopKDenseRank regarding admission boundaries
jayzhan211
committed
a01b08c
fix: improve tests for PartitionedTopKRank to address memory usage and boundary conditions
jayzhan211
committed
2d15aeb

Commits on Aug 26, 2026