[mlir][sparse] extend foreach operation to accept reduction arguments; fix sparse...
authorPeiming Liu <peiming@google.com>
Fri, 4 Nov 2022 17:52:21 +0000 (17:52 +0000)
committerPeiming Liu <peiming@google.com>
Mon, 7 Nov 2022 21:40:30 +0000 (21:40 +0000)
commit4fa00ce15c842aa8be495759723e2e2450591380
tree28d13d009ab3852dcd23d501b258f37037e5decf
parent43e52ad553cbafb78386e9e6e1204c8de8a506d4
[mlir][sparse] extend foreach operation to accept reduction arguments; fix sparse tensor rewriting patterns that do not propagate sparse tensor SSA properly.

This patch re-commit D137468 and D137463, which were reverted by mistakes.

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D137579
mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorOps.td
mlir/lib/Dialect/SparseTensor/IR/SparseTensorDialect.cpp
mlir/lib/Dialect/SparseTensor/Transforms/CodegenUtils.cpp
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorRewriting.cpp
mlir/test/Dialect/SparseTensor/convert_dense2sparse.mlir
mlir/test/Dialect/SparseTensor/convert_sparse2sparse.mlir
mlir/test/Dialect/SparseTensor/invalid.mlir
mlir/test/Dialect/SparseTensor/rewriting_for_codegen.mlir
mlir/test/Dialect/SparseTensor/roundtrip.mlir
mlir/test/Dialect/SparseTensor/sparse_concat_codegen.mlir
mlir/test/Dialect/SparseTensor/sparse_reshape.mlir