[mlir][sparse] Implement hybrid quick sort for sparse_tensor.sort.
authorbixia1 <bixia@google.com>
Wed, 8 Feb 2023 17:02:13 +0000 (09:02 -0800)
committerbixia1 <bixia@google.com>
Wed, 8 Feb 2023 22:06:31 +0000 (14:06 -0800)
commita1507668807e6108c12ffecf3740cb339b15018d
tree669b021bef24c26ba5737033c112c2dfd629be3c
parent550cb763dac4234b5d76bcafdda0471e38be05fa
[mlir][sparse] Implement hybrid quick sort for sparse_tensor.sort.

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D143227
mlir/lib/Dialect/SparseTensor/Transforms/CodegenUtils.h
mlir/lib/Dialect/SparseTensor/Transforms/SparseBufferRewriting.cpp
mlir/test/Dialect/SparseTensor/buffer_rewriting.mlir
mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_rewrite_sort.mlir
mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_rewrite_sort_coo.mlir