[mlir][sparse] accept affine subscripts in outer dimensions of dense memrefs
authorAart Bik <ajcbik@google.com>
Fri, 8 Oct 2021 23:13:29 +0000 (16:13 -0700)
committerAart Bik <ajcbik@google.com>
Mon, 11 Oct 2021 18:45:14 +0000 (11:45 -0700)
commit849f016ce8322d492e7d85eaf0c2d322e8836a64
tree7adc5c6daaa8604ff395565c5f6b9a66b9be84d0
parent011d8633eb11d31c801ae86bd1aecbe72953082a
[mlir][sparse] accept affine subscripts in outer dimensions of dense memrefs

This relaxes vectorization of dense memrefs a bit so that affine expressions
are allowed in more outer dimensions. Vectorization of non unit stride
references is disabled though, since this seems ineffective anyway.

Reviewed By: bixia

Differential Revision: https://reviews.llvm.org/D111469
mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp
mlir/test/Dialect/SparseTensor/sparse_vector.mlir