[mlir] Return new scf.forall handle in fuse_into_containing_op
authorHarsh Menon <harsh@nod-labs.com>
Thu, 25 May 2023 10:32:59 +0000 (03:32 -0700)
committerHarsh Menon <harsh@nod-labs.com>
Thu, 25 May 2023 16:34:46 +0000 (09:34 -0700)
commit19b9c74b42824c5f7d73481d9c1fe8e385a4426c
tree573a3e84d454b828e8644dcca009997840c25a66
parent13e3d4aa5a4e9062bb377207e522cd4f8ed688bd
[mlir] Return new scf.forall handle in fuse_into_containing_op

Since the scf.forall is now consumed by the fuse into
containing op, we need to return a handle to the new scf.forall.
This patch does that and also ensures that the new bbArg
added to the scf.forall is used in its body.

Differential Revision: https://reviews.llvm.org/D151418
mlir/include/mlir/Dialect/Linalg/TransformOps/LinalgTransformOps.td
mlir/lib/Dialect/Linalg/TransformOps/LinalgTransformOps.cpp
mlir/test/Dialect/Linalg/transform-op-fuse-into-containing.mlir
mlir/test/Dialect/Linalg/transform-tile-and-fuse.mlir