Skip to content

Commit 13f5fa2

Browse files
committed
Add custom omp loop wrapper
1 parent e453738 commit 13f5fa2

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

mlir/include/mlir/Dialect/OpenMP/OpenMPOps.td

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -308,6 +308,17 @@ def WorkshareOp : OpenMP_Op<"workshare", clauses = [
308308
let hasVerifier = 1;
309309
}
310310

311+
def WorkshareLoopWrapperOp : OpenMP_Op<"workshare_loop_wrapper", traits = [
312+
DeclareOpInterfaceMethods<LoopWrapperInterface>,
313+
RecursiveMemoryEffects, SingleBlock
314+
], singleRegion = true> {
315+
let summary = "contains loop nests to be parallelized by workshare";
316+
317+
let builders = [
318+
OpBuilder<(ins), [{ build($_builder, $_state, {}); }]>
319+
];
320+
}
321+
311322
//===----------------------------------------------------------------------===//
312323
// Loop Nest
313324
//===----------------------------------------------------------------------===//

0 commit comments

Comments
 (0)