[mlir][sparse] fix bug in workspace dimension computation
authorAart Bik <ajcbik@google.com>
Thu, 8 Sep 2022 05:46:31 +0000 (22:46 -0700)
committerAart Bik <ajcbik@google.com>
Thu, 8 Sep 2022 15:25:02 +0000 (08:25 -0700)
commitec8f2905a33ba970543c8edb4141c47f30d325f7
tree53b327d2010460b41cde584e9a1775a3a528fdb3
parentac3b8df8f2f433193827b8c0396ba8d41d1af565
[mlir][sparse] fix bug in workspace dimension computation

Access pattern expansion is always done along the innermost stored
dimension, but this was incorrectly reordered due to using a
general utility typically used by original dimensions only.

Reviewed By: bixia

Differential Revision: https://reviews.llvm.org/D133472
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorConversion.cpp
mlir/test/Dialect/SparseTensor/conversion.mlir
mlir/test/Dialect/SparseTensor/sparse_expand.mlir
mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_expand.mlir [new file with mode: 0644]