[mlir][sparse] implement loop index value vectorization
authorAart Bik <ajcbik@google.com>
Fri, 18 Mar 2022 20:28:59 +0000 (13:28 -0700)
committerAart Bik <ajcbik@google.com>
Mon, 21 Mar 2022 17:40:38 +0000 (10:40 -0700)
commit69a7759b402fd42c7ac3fc27ef35682001f305e5
tree068b55e597ec8c7cfd79dd6d83f7ef31ea88ac80
parent059a953d8819b361b33205c109b5e903801613a2
[mlir][sparse] implement loop index value vectorization

with CHECK and integration test

Reviewed By: bixia

Differential Revision: https://reviews.llvm.org/D122040
mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp
mlir/test/Dialect/SparseTensor/sparse_vector_index.mlir [new file with mode: 0644]
mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_index_dense.mlir [new file with mode: 0644]