[Mlir-commits] [mlir] [mlir] Fix loop pipelining when the operand of `yield` is not defined in the loop body (PR #75423)

Thomas Raoux llvmlistbot at llvm.org
Wed Dec 13 18:10:35 PST 2023


================
@@ -685,14 +696,15 @@ FailureOr<ForOp> mlir::scf::pipelineForLoop(RewriterBase &rewriter, ForOp forOp,
   if (options.peelEpilogue) {
     // 4. Emit the epilogue after the new forOp.
     rewriter.setInsertionPointAfter(newForOp);
-    returnValues = pipeliner.emitEpilogue(rewriter);
+    pipeliner.emitEpilogue(rewriter, returnValues);
   }
   // 5. Erase the original loop and replace the uses with the epilogue output.
   if (forOp->getNumResults() > 0)
     rewriter.replaceOp(forOp, returnValues);
   else
     rewriter.eraseOp(forOp);
 
+  llvm::errs() << *newForOp->getParentOp() << "\n";
----------------
ThomasRaoux wrote:

left over debug?

https://github.com/llvm/llvm-project/pull/75423


More information about the Mlir-commits mailing list