[mlir][sparse] Add rewriting rules for sparse_tensor.sort_coo.
authorbixia1 <bixia@google.com>
Wed, 9 Nov 2022 17:07:06 +0000 (09:07 -0800)
committerbixia1 <bixia@google.com>
Mon, 14 Nov 2022 16:48:53 +0000 (08:48 -0800)
commit4f729d5a7056bbb59621c1332598db924c2f1fd6
treefab8e6e6397f1a868b1a7efc2a7d7d02763a1211
parentdf2f9e57f8d1aa858b12aefb89909a7c88c6e0ea
[mlir][sparse] Add rewriting rules for sparse_tensor.sort_coo.

Refactor the rewriting of sparse_tensor.sort to support the implementation of
sparse_tensor.sort_coo.

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D137522
mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorOps.td
mlir/lib/Dialect/SparseTensor/Transforms/SparseBufferRewriting.cpp
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorPasses.cpp
mlir/test/Dialect/SparseTensor/buffer_rewriting.mlir
mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_rewrite_sort_coo.mlir [new file with mode: 0644]