maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

[mlir][linalg] Assert tile loop nest invariants in fusion. 

Assert the tile loop nest invariants are satisfied instead of failing silently.

Reviewed By: nicolasvasilache

Differential Revision: https://reviews.llvm.org/D110137
This commit is contained in:
Tobias Gysi 2021-09-21 13:51:07 +00:00
parent 744ec74b30
commit c8eed8f9a7
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
mlir/lib/Dialect/Linalg/Transforms/FusionOnTensors.cpp
View File

@ -301,8 +301,13 @@ LogicalResult TileLoopNest::tileRootOp(OpBuilder &b,
FailureOr<LinalgOp> TileLoopNest::fuseProducer(OpBuilder &b,
OpOperand *rootOpOperand) {
// Check the tile loop nest is non-empty and satisfies all invariants.
if (isEmpty() || !isValid())
assert(rootOpOperand->getOwner() == rootOp &&
"expect the root op to be the owner of the operand to fuse");
assert(this->isValid() &&
"expect the tile loop nest to satisfy all invariants");
// Check the tile loop nest is non-empty.
if (isEmpty())
return failure();
// Check `rootOpOperand` is defined by an ExtractSliceOp.
@ -310,9 +315,8 @@ FailureOr<LinalgOp> TileLoopNest::fuseProducer(OpBuilder &b,
if (!sliceOp)
return failure();
// Check `tileLoopNest` tiles `sliceOp` and `rootOpOperand`.
if (sliceOp->getParentOp() != rootOp->getParentOp() ||
rootOpOperand->getOwner() != rootOp)
// Check `sliceOp` is tiled by the tile loop nest.
if (sliceOp->getParentOp() != rootOp->getParentOp())
return failure();
// Check if the producer is a LinalgOp possibly passed by iteration argument.