[mlir][sparse] Extend sorting function generator to support operand beyond (lo, hi...
authorbixia1 <bixia@google.com>
Tue, 31 Jan 2023 16:36:29 +0000 (08:36 -0800)
committerbixia1 <bixia@google.com>
Tue, 31 Jan 2023 18:22:28 +0000 (10:22 -0800)
commit8550aebd57d08d34c11ba438d07e1e0942b97f31
tree23747da270b88d95622b589e6f6a01b980947df4
parenta7f8ddc4e2e3dac4c123ed8db1704f2753db5722
[mlir][sparse] Extend sorting function generator to support operand beyond (lo, hi, xs, ys).

This is to prepare for implementing a hybrid quick sort, which switches to heap
sort when the recursive depth exceeds certain limits.

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D142731
mlir/lib/Dialect/SparseTensor/Transforms/SparseBufferRewriting.cpp