[Mlir-commits] [mlir] [mlir][func]-Add deduplicate funcOp arguments transform (PR #158266)

[Amir Bishara]

================
@@ -76,8 +128,10 @@ func::replaceFuncWithNewOrder(RewriterBase &rewriter, func::FuncOp funcOp,
   auto returnOp = cast<func::ReturnOp>(
       newFuncOp.getFunctionBody().begin()->getTerminator());
   SmallVector<Value> newReturnValues;
-  for (unsigned int idx : newResultsOrder)
-    newReturnValues.push_back(returnOp.getOperand(idx));
+  for (auto [newResIdx, oldResIdx] : llvm::enumerate(newResToOldResIdxs)) {
+    std::ignore = newResIdx;
+    newReturnValues.push_back(returnOp.getOperand(oldResIdx.front()));
+  }
----------------
amirBish wrote:

Sure fixed, missed it after the changes.

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