[mlir] Add MemoryEffects::Allocate to memref::CloneOp
authorButygin <ivan.butygin@intel.com>
Sat, 26 Jun 2021 10:05:29 +0000 (13:05 +0300)
committerButygin <ivan.butygin@intel.com>
Tue, 29 Jun 2021 10:37:32 +0000 (13:37 +0300)
commit293064222a013055cda9240647110f5bcf8e1f31
treea377f29fdd7d2fa078891262209b64ccf6bbce59
parent6cf6f6f65fde9638a2ca64cd8013d0d0ab1d473c
[mlir] Add MemoryEffects::Allocate to memref::CloneOp

Without it BufferDeallocationPass process only CloneOps created during pass itself and ignore all CloneOps that were already present in IR.

For our specific usecase:

```
func @dealloc_existing_clones(%arg0: memref<?x?xf64>, %arg1: memref<?x?xf64>) -> memref<?x?xf64> {
  return %arg0 : memref<?x?xf64>
}
```

Input arguments will be freed immediately after return from function and we want to prolong lifetime for the returned argument.
To achieve this we explicitly add clones to all input memrefs and expect that BufferDeallocationPass will add correct deallocs to them (unnessesary clone+dealloc pairs will be canonicalized away later).

Differential Revision: https://reviews.llvm.org/D104973
mlir/lib/Dialect/MemRef/IR/MemRefOps.cpp
mlir/test/Transforms/buffer-deallocation.mlir