[mlir][sparse] make UnpackOp return the actual filled length of unpacked memory
authorPeiming Liu <peiming@google.com>
Fri, 30 Jun 2023 18:07:21 +0000 (18:07 +0000)
committerPeiming Liu <peiming@google.com>
Fri, 30 Jun 2023 21:35:15 +0000 (21:35 +0000)
commita63d6a00140d25289e0cba294c9228dbb74b06fa
tree3b2a5202e8d6aa4cb35e163aafbe37b6a2963058
parentab345bde81f23112e2bff2fb18975ca9d9ff640f
[mlir][sparse] make UnpackOp return the actual filled length of unpacked memory

This might simplify frontend implementation by avoiding recomputation for the same value.

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D154244
mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorOps.td
mlir/lib/Dialect/SparseTensor/Transforms/BufferizableOpInterfaceImpl.cpp
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorCodegen.cpp
mlir/test/Dialect/SparseTensor/invalid.mlir
mlir/test/Dialect/SparseTensor/roundtrip.mlir
mlir/test/Dialect/SparseTensor/sparse_pack.mlir
mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_pack.mlir