[Mlir-commits] [mlir] [mlir] Extend SCF loopUnrollByFactor to return the result loops (PR #114573)

Thomas Raoux llvmlistbot at llvm.org
Sun Nov 3 10:47:44 PST 2024


================
@@ -490,7 +494,8 @@ LogicalResult mlir::loopUnrollByFactor(
       annotateFn, iterArgs, yieldedValues);
   // Promote the loop body up if this has turned into a single iteration loop.
   (void)forOp.promoteIfSingleIteration(rewriter);
-  return success();
+  resultLoops.mainLoopOp = forOp;
----------------
ThomasRaoux wrote:

best would be to use `FailureOr<UnrolledLoopInfo>`

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


More information about the Mlir-commits mailing list