[mlir][sparse] fix bugs when computing the memory size when lowering pack op.
authorPeiming Liu <peiming@google.com>
Thu, 25 May 2023 18:35:28 +0000 (18:35 +0000)
committerPeiming Liu <peiming@google.com>
Thu, 25 May 2023 19:19:52 +0000 (19:19 +0000)
commitf7b8b005ff12b2f4245aa42684d129358396d5df
tree34090e22129655a4a7fe238a5bba2afad75b54ba
parent5c082e7e15e38a2eea1f506725efe636a5b1bf8a
[mlir][sparse] fix bugs when computing the memory size when lowering pack op.

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D151481
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorCodegen.cpp
mlir/test/Dialect/SparseTensor/sparse_pack.mlir
mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_pack.mlir