[mlir][sparse] Replace sparse_tensor.sort with sparse_tensor.sort_coo for sorting...
authorbixia1 <bixia@google.com>
Thu, 5 Jan 2023 17:39:23 +0000 (09:39 -0800)
committerbixia1 <bixia@google.com>
Thu, 5 Jan 2023 23:42:57 +0000 (15:42 -0800)
commit81e3079d0f0cc407f4e295417c3d9c8b6203e736
tree247cc544d6af99e14a0b14835680a6f1d3b6de21
parent47232bea9e5ecc5ad999f53ec2895d2c07427572
[mlir][sparse] Replace sparse_tensor.sort with sparse_tensor.sort_coo for sorting COO tensors.

Add codegen pattern for sparse_tensor.indices_buffer.

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D140871
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorCodegen.cpp
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorRewriting.cpp
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorStorageLayout.h
mlir/test/Dialect/SparseTensor/codegen.mlir
mlir/test/Dialect/SparseTensor/convert_dense2sparse.mlir