[mlir][sparse] support iteration over compressed-hi dimension level in loop emitter
authorPeiming Liu <peiming@google.com>
Tue, 18 Apr 2023 22:05:31 +0000 (22:05 +0000)
committerPeiming Liu <peiming@google.com>
Thu, 20 Apr 2023 00:57:08 +0000 (00:57 +0000)
commitabd66d918ae95fbbdad076c38fe2ea6a459a9646
tree5bd7f9786c0a1b21c77b8536f704aa227ecdf75f
parent6f1a9ed072de612584a90adf3469744781da17fc
[mlir][sparse] support iteration over compressed-hi dimension level in loop emitter

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D148668
mlir/lib/Dialect/SparseTensor/Transforms/LoopEmitter.cpp
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorCodegen.cpp
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorStorageLayout.cpp
mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp
mlir/lib/Dialect/SparseTensor/Utils/Merger.cpp
mlir/test/Dialect/SparseTensor/sparse_foreach.mlir