[mlir][async] Add AutomaticAllocationScope to async::ExecuteOp

Differential Revision: https://reviews.llvm.org/D118761
2022-02-02 05:01:42 -05:00 · 2022-02-02 05:01:42 -05:00 · aea4929dbb
parent 281d29b8fe
commit aea4929dbb
1 changed files with 2 additions and 1 deletions
--- a/mlir/include/mlir/Dialect/Async/IR/AsyncOps.td
+++ b/mlir/include/mlir/Dialect/Async/IR/AsyncOps.td
@ -31,7 +31,8 @@ def Async_ExecuteOp :
  Async_Op<"execute", [SingleBlockImplicitTerminator<"YieldOp">,
                       DeclareOpInterfaceMethods<RegionBranchOpInterface,
                                                 ["getSuccessorEntryOperands"]>,
-                       AttrSizedOperandSegments]> {
+                       AttrSizedOperandSegments,
+                       AutomaticAllocationScope]> {
  let summary = "Asynchronous execute operation";
  let description = [{
    The `body` region attached to the `async.execute` operation semantically