diff --git a/flang/lib/Lower/OpenACC.cpp b/flang/lib/Lower/OpenACC.cpp --- a/flang/lib/Lower/OpenACC.cpp +++ b/flang/lib/Lower/OpenACC.cpp @@ -1355,12 +1355,13 @@ const Fortran::parser::AccClauseList &accClauseList) { mlir::Value ifCond, async; llvm::SmallVector attachEntryOperands, createEntryOperands, - copyEntryOperands, copyoutEntryOperands, dataClauseOperands; + copyEntryOperands, copyoutEntryOperands, dataClauseOperands, waitOperands; - // Async has an optional value but can be present with + // Async and wait have an optional value but can be present with // no value as well. When there is no value, the op has an attribute to // represent the clause. bool addAsyncAttr = false; + bool addWaitAttr = false; fir::FirOpBuilder &builder = converter.getFirOpBuilder(); @@ -1450,12 +1451,14 @@ llvm::SmallVector operandSegments; addOperand(operands, operandSegments, ifCond); addOperand(operands, operandSegments, async); + addOperands(operands, operandSegments, waitOperands); addOperands(operands, operandSegments, dataClauseOperands); auto dataOp = createRegionOp( builder, currentLocation, operands, operandSegments); dataOp.setAsyncAttr(addAsyncAttr); + dataOp.setAsyncAttr(addWaitAttr); auto insPt = builder.saveInsertionPoint(); builder.setInsertionPointAfter(dataOp); diff --git a/mlir/include/mlir/Dialect/OpenACC/OpenACCOps.td b/mlir/include/mlir/Dialect/OpenACC/OpenACCOps.td --- a/mlir/include/mlir/Dialect/OpenACC/OpenACCOps.td +++ b/mlir/include/mlir/Dialect/OpenACC/OpenACCOps.td @@ -867,6 +867,8 @@ let arguments = (ins Optional:$ifCond, Optional:$async, UnitAttr:$asyncAttr, + Variadic:$waitOperands, + UnitAttr:$waitAttr, Variadic:$dataClauseOperands, OptionalAttr:$defaultAttr); @@ -885,6 +887,7 @@ `if` `(` $ifCond `)` | `async` `(` $async `:` type($async) `)` | `dataOperands` `(` $dataClauseOperands `:` type($dataClauseOperands) `)` + | `wait` `(` $waitOperands `:` type($waitOperands) `)` ) $region attr-dict-with-keyword }]; diff --git a/mlir/lib/Dialect/OpenACC/IR/OpenACC.cpp b/mlir/lib/Dialect/OpenACC/IR/OpenACC.cpp --- a/mlir/lib/Dialect/OpenACC/IR/OpenACC.cpp +++ b/mlir/lib/Dialect/OpenACC/IR/OpenACC.cpp @@ -868,6 +868,7 @@ Value DataOp::getDataOperand(unsigned i) { unsigned numOptional = getIfCond() ? 1 : 0; numOptional += getAsync() ? 1 : 0; + numOptional += getWaitOperands().size(); return getOperand(numOptional + i); } diff --git a/mlir/test/Dialect/OpenACC/ops.mlir b/mlir/test/Dialect/OpenACC/ops.mlir --- a/mlir/test/Dialect/OpenACC/ops.mlir +++ b/mlir/test/Dialect/OpenACC/ops.mlir @@ -822,6 +822,13 @@ acc.data async(%a1 : i64) { } attributes { defaultAttr = #acc, async } + acc.data { + } attributes { defaultAttr = #acc, wait } + + %w1 = arith.constant 1 : i64 + acc.data wait(%w1 : i64) { + } attributes { defaultAttr = #acc, wait } + return } @@ -927,6 +934,12 @@ // CHECK: acc.data async(%{{.*}} : i64) { // CHECK-NEXT: } attributes {async, defaultAttr = #acc} +// CHECK: acc.data { +// CHECK-NEXT: } attributes {defaultAttr = #acc, wait} + +// CHECK: acc.data wait(%{{.*}}) { +// CHECK-NEXT: } attributes {defaultAttr = #acc, wait} + // ----- func.func @testupdateop(%a: memref, %b: memref, %c: memref) -> () {