diff options
| author | Alexander Belyaev <pifon@google.com> | 2021-04-28 14:57:50 +0200 |
|---|---|---|
| committer | Alexander Belyaev <pifon@google.com> | 2021-04-28 14:58:02 +0200 |
| commit | 9a66d334522d6b69d7d1ea0dbc1b04ae0d43b959 (patch) | |
| tree | 1b11ad30e520733cfcd9223adc43b4e75850e26f | |
| parent | cea97fc0fcd86d52f9efa215116356b72faeb17d (diff) | |
| download | llvm-9a66d334522d6b69d7d1ea0dbc1b04ae0d43b959.tar.gz | |
[mlir] Fix the postsubmit comments in https://reviews.llvm.org/D101445
| -rw-r--r-- | mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp b/mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp index 8a80036688cd..17ecab19c2a8 100644 --- a/mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp +++ b/mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp @@ -174,9 +174,9 @@ static LogicalResult foldMemRefCastInTiledLoopOp(TiledLoopOp op) { auto castOp = operand.get().getDefiningOp<memref::CastOp>(); if (castOp && memref::CastOp::canFoldIntoConsumerOp(castOp)) { operand.set(castOp.getOperand()); - auto newBbArg = + BlockArgument newBbArg = body->insertArgument(bbArgIndex, castOp.getOperand().getType()); - auto oldBbArg = body->getArgument(newBbArg.getArgNumber() + 1); + BlockArgument oldBbArg = body->getArgument(newBbArg.getArgNumber() + 1); // Insert memref.cast back to the original type. oldBbArg.replaceAllUsesWith( @@ -2120,7 +2120,6 @@ struct TiledLoopInputsFolder : public OpRewritePattern<linalg::TiledLoopOp> { if (!in.getType().isa<RankedTensorType>() || !bbArg.use_empty()) { oldInputIdToNew[index] = newInputs.size(); newInputs.push_back(in); - continue; } } if (newInputs.size() == tiledLoop.inputs().size()) |