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

Mehdi Amini llvmlistbot at llvm.org
Fri Sep 12 09:36:40 PDT 2025


================
@@ -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()));
+  }
----------------
joker-eph wrote:

```suggestion
  for (int oldResIdx : newResToOldResIdxs)
    newReturnValues.push_back(returnOp.getOperand(oldResIdx.front()));
```

(is it equivalent?)

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


More information about the Mlir-commits mailing list