Simplify even further

Xilinx · Nov 21, 2024 · f5a954f · f5a954f
1 parent bb66650
commit f5a954f
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/mlir/lib/Conversion/SCFToEmitC/SCFToEmitC.cpp b/mlir/lib/Conversion/SCFToEmitC/SCFToEmitC.cpp
@@ -94,12 +94,9 @@ static LogicalResult lowerYield(Operation *op, ValueRange resultVariables,
   rewriter.setInsertionPoint(yield);
 
   SmallVector<Value> yieldOperands;
-  for (auto originalOperand : yield.getOperands()) {
-    Value remappedValue = rewriter.getRemappedValue(originalOperand);
-    if (!remappedValue) {
-      return rewriter.notifyMatchFailure(op, "failed to lower yield operands");
-    }
-    yieldOperands.push_back(remappedValue);
+  auto result = rewriter.getRemappedValues(yield.getOperands(), yieldOperands);
+  if (failed(result)) {
+    return rewriter.notifyMatchFailure(op, "failed to lower yield operands");
   }
 
   assignValues(yieldOperands, resultVariables, rewriter, loc);