[MLIR] Fix lowering of affine operations with return values
authorPrateek Gupta <prateek@polymagelabs.com>
Tue, 22 Dec 2020 16:10:17 +0000 (21:40 +0530)
committerUday Bondhugula <uday@polymagelabs.com>
Tue, 22 Dec 2020 16:14:31 +0000 (21:44 +0530)
commit3e07b0b9d3363fb767cbbaa2593fa91ac393fb7e
tree2a1f844aa429d7d0786626e2d66434fb4b326827
parent0586f048d794673867f7307898114a111f6e80ba
[MLIR] Fix lowering of affine operations with return values

This commit addresses the issue of lowering affine.for and
affine.parallel having return values. Relevant test cases are also
added.

Signed-off-by: Prateek Gupta <prateek@polymagelabs.com>
Differential Revision: https://reviews.llvm.org/D93090
mlir/lib/Conversion/AffineToStandard/AffineToStandard.cpp
mlir/test/Conversion/AffineToStandard/lower-affine.mlir