[Mlir-commits] [mlir] [mlir][bufferization] Fix crash when returning an allocation twice (PR #220232)

[llvmlistbot at llvm.org]

================
@@ -168,7 +168,20 @@ updateReturnOps(func::FuncOp func, ArrayRef<BlockArgument> appendedEntryArgs,
     }
     OpBuilder builder(op);
     SmallVector<SmallVector<Value>> dynamicSizes;
+    SetVector<Operation *> toErase;
+    DenseMap<Value, BlockArgument> primaryArgs;
+    DenseMap<Value, SmallVector<Value>> dynSizes;
----------------
AkunCube wrote:

Renamed `dynSizes` to `dynamicSizesCache`. The cache stores dynamic sizes by returned memref, while `dynamicSizes` contains one entry for each result occurrence, including duplicates.

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