diff --git a/mlir/include/mlir/Dialect/OpenMP/OpenMPOps.td b/mlir/include/mlir/Dialect/OpenMP/OpenMPOps.td --- a/mlir/include/mlir/Dialect/OpenMP/OpenMPOps.td +++ b/mlir/include/mlir/Dialect/OpenMP/OpenMPOps.td @@ -80,7 +80,8 @@ def ClauseDefaultAttr : EnumAttr; -def ParallelOp : OpenMP_Op<"parallel", [AttrSizedOperandSegments, +def ParallelOp : OpenMP_Op<"parallel", [AutomaticAllocationScope, + AttrSizedOperandSegments, DeclareOpInterfaceMethods]> { let summary = "parallel construct"; let description = [{ diff --git a/mlir/include/mlir/Dialect/SCF/SCFOps.td b/mlir/include/mlir/Dialect/SCF/SCFOps.td --- a/mlir/include/mlir/Dialect/SCF/SCFOps.td +++ b/mlir/include/mlir/Dialect/SCF/SCFOps.td @@ -404,7 +404,8 @@ } def ParallelOp : SCF_Op<"parallel", - [AttrSizedOperandSegments, + [AutomaticAllocationScope, + AttrSizedOperandSegments, DeclareOpInterfaceMethods, RecursiveSideEffects, SingleBlockImplicitTerminator<"scf::YieldOp">]> {