Skip to content

Commit f608cf4

Browse files
committed
Changed yield op replacment code
1 parent 4185a87 commit f608cf4

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

mlir/lib/Dialect/SCF/IR/SCF.cpp

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -355,9 +355,8 @@ struct ExecuteRegionForwardingEliminator
355355
// results. scf.execute_region must have at least one yield operation.
356356
for (auto *yieldOp : yieldOps) {
357357
rewriter.setInsertionPoint(yieldOp);
358-
rewriter.eraseOp(yieldOp);
359-
rewriter.create<scf::YieldOp>(yieldOp->getLoc(),
360-
ValueRange(internalValues));
358+
rewriter.replaceOpWithNewOp<scf::YieldOp>(yieldOp,
359+
ValueRange(internalValues));
361360
}
362361

363362
// Replace the old operation with the external values directly.

0 commit comments

Comments
 (0)