[mlir][sparse] Fix a bug in rewriting for the convert op.
authorbixia1 <bixia@google.com>
Thu, 10 Nov 2022 21:07:57 +0000 (13:07 -0800)
committerbixia1 <bixia@google.com>
Thu, 10 Nov 2022 22:30:51 +0000 (14:30 -0800)
commitccd1a5b9cbf90c41f1e924da42243d92e88de797
tree5654e9f92f1a399d3e10c9abfa73514023df95c7
parent6a8d894ce05353f17367c52c2985a015bcbacf60
[mlir][sparse] Fix a bug in rewriting for the convert op.

The code to retrieve the number of entries isn't correct.

Reviewed By: Peiming

Differential Revision: https://reviews.llvm.org/D137795
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorRewriting.cpp
mlir/test/Dialect/SparseTensor/convert_dense2sparse.mlir
mlir/test/Dialect/SparseTensor/convert_sparse2sparse.mlir