[Mlir-commits] [mlir] Enable custom alloc-like ops in `promoteBufferResultsToOutParams` (PR #120288)

Matthias Springer llvmlistbot at llvm.org
Fri Dec 20 02:23:35 PST 2024


================
@@ -167,6 +174,10 @@ struct BufferResultsToOutParamsOpts {
     return true;
   };
 
+  /// Allocation function; used to allocate a memref.
+  /// If this is empty, memref.alloc is used
+  std::optional<AllocationFn> allocationFn;
----------------
matthias-springer wrote:

Could this be avoided by adding a `buildAlloc` interface method to `AllocationOpInterface`? Maybe the transformation could also supported mixed alloc ops then>

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


More information about the Mlir-commits mailing list