[mlir][sparse] extend foreach operation to accept reduction arguments.
authorPeiming Liu <peiming@google.com>
Fri, 4 Nov 2022 17:52:21 +0000 (17:52 +0000)
committerPeiming Liu <peiming@google.com>
Fri, 4 Nov 2022 23:34:16 +0000 (23:34 +0000)
commit53d5d3401120f2aa741a73a5a9ba0ce012ca532c
tree3fa82c69b320bcfb3414ea2b75b4b1d50247a4a9
parent026ddced176e66657e51ffb73b26019b45485db0
[mlir][sparse] extend foreach operation to accept reduction arguments.

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D137463
mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorOps.td
mlir/lib/Dialect/SparseTensor/IR/SparseTensorDialect.cpp
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorRewriting.cpp
mlir/test/Dialect/SparseTensor/invalid.mlir
mlir/test/Dialect/SparseTensor/roundtrip.mlir