457 lines · plain
1// Tablegen tests for the automatic population of OpenMP dialect operation2// fields from clauses. Arguments and extra class declarations are checked from3// the output of tablegen declarations. Summary, description, assembly format4// and traits are checked from the output of tablegen docs.5 6// Run tablegen to generate OmpCommon.td in temp directory first.7// RUN: mkdir -p %t/mlir/Dialect/OpenMP8// RUN: mlir-tblgen --gen-directive-decl --directives-dialect=OpenMP \9// RUN: %S/../../../llvm/include/llvm/Frontend/OpenMP/OMP.td \10// RUN: -I %S/../../../llvm/include > %t/mlir/Dialect/OpenMP/OmpCommon.td11 12// RUN: mlir-tblgen -gen-op-decls -I %S/../../include -I %t %s | FileCheck %s --check-prefix=DECL13// RUN: mlir-tblgen -gen-op-doc -I %S/../../include -I %t %s | FileCheck %s --check-prefix=DOC14 15include "mlir/Dialect/OpenMP/OpenMPOpBase.td"16 17// Dummy traits.18 19def TraitOne : NativeOpTrait<"TraitOne">;20def TraitTwo : NativeOpTrait<"TraitTwo">;21def TraitThree : NativeOpTrait<"TraitThree">;22 23// Test clauses.24 25class OptClauseSkip<26 bit traits = false, bit arguments = false, bit assemblyFormat = false,27 bit description = false, bit extraClassDeclaration = false28 > : OpenMP_Clause<traits, arguments, assemblyFormat, description,29 extraClassDeclaration> {30 let traits = [31 TraitOne32 ];33 34 let arguments = (ins35 Optional<AnyInteger>:$opt_simple_val36 );37 38 let optAssemblyFormat = [{39 `optsimple` `(` $opt_simple_val `:` type($opt_simple_val) `)`40 }];41 42 let description = [{43 Optional clause description.44 }];45 46 let extraClassDeclaration = [{47 void optClauseExtraClassDecl();48 }];49}50 51def OptClause : OptClauseSkip<>;52 53def OptClauseTwo : OpenMP_Clause<54 /*skipTraits=*/false, /*skipArguments=*/false, /*skipAssemblyFormat=*/false,55 /*skipDescription=*/false, /*skipExtraClassDeclaration=*/false> {56 let traits = [57 TraitOne, TraitTwo58 ];59 60 let arguments = (ins61 Optional<AnyInteger>:$opt_two_val62 );63 64 let optAssemblyFormat = [{65 `opt_two` `(` $opt_two_val `:` type($opt_two_val) `)`66 }];67 68 let description = [{69 Optional clause two description.70 }];71 72 let extraClassDeclaration = [{73 void optClauseTwoExtraClassDecl();74 }];75}76 77class ReqClauseSkip<78 bit traits = false, bit arguments = false, bit assemblyFormat = false,79 bit description = false, bit extraClassDeclaration = false80 > : OpenMP_Clause<traits, arguments, assemblyFormat, description,81 extraClassDeclaration> {82 let traits = [83 TraitTwo84 ];85 86 let arguments = (ins87 AnyInteger:$req_simple_val88 );89 90 let reqAssemblyFormat = [{91 `reqsimple` `(` $req_simple_val `:` type($req_simple_val) `)`92 }];93 94 let description = [{95 Required clause description.96 }];97 98 let extraClassDeclaration = [{99 void reqClauseExtraClassDecl();100 }];101}102 103def ReqClause : ReqClauseSkip<>;104 105def ReqClauseTwo : OpenMP_Clause<106 /*skipTraits=*/false, /*skipArguments=*/false, /*skipAssemblyFormat=*/false,107 /*skipDescription=*/false, /*skipExtraClassDeclaration=*/false> {108 let traits = [109 TraitTwo, TraitThree110 ];111 112 let arguments = (ins113 AnyInteger:$req_two_val114 );115 116 let reqAssemblyFormat = [{117 `req_two` `(` $req_two_val `:` type($req_two_val) `)`118 }];119 120 let description = [{121 Required clause two description.122 }];123 124 let extraClassDeclaration = [{125 void reqClauseTwoExtraClassDecl();126 }];127}128 129// Clause-based operation definitions.130 131def OpAddArguments : OpenMP_Op<"op_add_arguments",132 traits=[AttrSizedOperandSegments], clauses=[ReqClause, OptClause]> {133 let summary = "operation with clauses - arguments added";134 let description = [{135 Description of operation with clauses - arguments added.136 }] # clausesDescription;137 let arguments = !con(clausesArgs, (ins Optional<AnyInteger>:$opt_added,138 AnyInteger:$req_added));139 let assemblyFormat = clausesReqAssemblyFormat #140 "`req_added` `(` $req_added `:` type($req_added) `)` oilist(" #141 clausesOptAssemblyFormat #142 "|`opt_added` `(` $opt_added `:` type($opt_added) `)`) attr-dict";143}144 145// DECL-LABEL: class OpAddArguments : public ::mlir::Op<146// DECL: getReqSimpleVal() {147// DECL: getOptSimpleVal() {148// DECL: void reqClauseExtraClassDecl();149// DECL: void optClauseExtraClassDecl();150 151// DOC-LABEL: omp::OpAddArguments152// DOC: _Operation with clauses - arguments added_153// DOC: operation ::= `omp.op_add_arguments`154// DOC: `reqsimple` `(` $req_simple_val `:` type($req_simple_val) `)`155// DOC: `req_added` `(` $req_added `:` type($req_added) `)`156// DOC: oilist(157// DOC: `optsimple` `(` $opt_simple_val `:` type($opt_simple_val) `)`158// DOC: |`opt_added` `(` $opt_added `:` type($opt_added) `)`159// DOC-NOT: $region160// DOC: attr-dict161// DOC: Description of operation with clauses - arguments added.162// DOC: Required clause description.163// DOC: Optional clause description.164// DOC: Traits: `AttrSizedOperandSegments`, `TraitOne`, `TraitTwo`165 166def OpOptClause : OpenMP_Op<"op_with_opt",167 traits=[TraitThree], clauses=[OptClause]> {168 let summary = "operation with optional clause";169 let description = [{170 Description of operation with optional clause.171 }] # clausesDescription;172}173 174// DECL-LABEL: class OpOptClause : public ::mlir::Op<175// DECL: getOptSimpleVal() {176// DECL: void optClauseExtraClassDecl();177 178// DOC-LABEL: omp::OpOptClause179// DOC: _Operation with optional clause_180// DOC: operation ::= `omp.op_with_opt`181// DOC: oilist(182// DOC: `optsimple` `(` $opt_simple_val `:` type($opt_simple_val) `)`183// DOC-NOT: $region184// DOC: attr-dict185// DOC: Description of operation with optional clause.186// DOC: Optional clause description.187// DOC: Traits: `TraitOne`, `TraitThree`188 189def OpReqClause : OpenMP_Op<"op_with_req",190 traits=[TraitThree], clauses=[ReqClause]> {191 let summary = "operation with required clause";192 let description = [{193 Description of operation with required clause.194 }] # clausesDescription;195}196 197// DECL-LABEL: class OpReqClause : public ::mlir::Op<198// DECL: getReqSimpleVal() {199// DECL: void reqClauseExtraClassDecl();200 201// DOC-LABEL: omp::OpReqClause202// DOC: _Operation with required clause_203// DOC: operation ::= `omp.op_with_req`204// DOC-NOT: oilist(205// DOC: `reqsimple` `(` $req_simple_val `:` type($req_simple_val) `)`206// DOC-NOT: $region207// DOC: attr-dict208// DOC: Description of operation with required clause.209// DOC: Required clause description.210// DOC: Traits: `TraitThree`, `TraitTwo`211 212def OpReqOptClause : OpenMP_Op<"op_with_req_and_opt",213 traits=[TraitThree], clauses=[ReqClause, OptClause]> {214 let summary = "operation with required and optional clauses";215 let description = [{216 Description of operation with required and optional clauses.217 }] # clausesDescription;218}219 220// DECL-LABEL: class OpReqOptClause : public ::mlir::Op<221// DECL: getReqSimpleVal() {222// DECL: getOptSimpleVal() {223// DECL: void reqClauseExtraClassDecl();224// DECL: void optClauseExtraClassDecl();225 226// DOC-LABEL: omp::OpReqOptClause227// DOC: _Operation with required and optional clauses_228// DOC: operation ::= `omp.op_with_req_and_opt`229// DOC: `reqsimple` `(` $req_simple_val `:` type($req_simple_val) `)`230// DOC: oilist(231// DOC: `optsimple` `(` $opt_simple_val `:` type($opt_simple_val) `)`232// DOC-NOT: $region233// DOC: attr-dict234// DOC: Description of operation with required and optional clauses.235// DOC: Required clause description.236// DOC: Optional clause description.237// DOC: Traits: `TraitOne`, `TraitThree`, `TraitTwo`238 239def OpSingleRegion : OpenMP_Op<"op_single_region",240 clauses=[OptClause], singleRegion=true> {241 let summary = "operation with a single region";242 let description = [{243 Description of operation with a single region.244 }] # clausesDescription;245}246 247// DECL-LABEL: class OpSingleRegion : public ::mlir::Op<248// DECL: getOptSimpleVal() {249// DECL: getRegion() {250// DECL: void optClauseExtraClassDecl();251 252// DOC-LABEL: omp::OpSingleRegion253// DOC: _Operation with a single region_254// DOC: operation ::= `omp.op_single_region`255// DOC: oilist(256// DOC: `optsimple` `(` $opt_simple_val `:` type($opt_simple_val) `)`257// DOC: ) $region attr-dict258// DOC: Description of operation with a single region.259// DOC: Optional clause description.260// DOC: Traits: `TraitOne`261 262def OpSkipArguments : OpenMP_Op<"op_skip_arguments",263 traits=[TraitThree], clauses=[OptClauseSkip<arguments=true>]> {264 let summary = "operation with clause - arguments skipped";265 let description = [{266 Description of operation with clause - arguments skipped.267 }] # clausesDescription;268 let arguments = !con(clausesArgs, (ins Optional<AnyInteger>:$opt_simple_val));269}270 271// DECL-LABEL: class OpSkipArguments : public ::mlir::Op<272// DECL: getOptSimpleVal() {273// DECL: void optClauseExtraClassDecl();274 275// DOC-LABEL: omp::OpSkipArguments276// DOC: _Operation with clause - arguments skipped_277// DOC: operation ::= `omp.op_skip_arguments`278// DOC: oilist(279// DOC: `optsimple` `(` $opt_simple_val `:` type($opt_simple_val) `)`280// DOC-NOT: $region281// DOC: attr-dict282// DOC: Description of operation with clause - arguments skipped.283// DOC: Optional clause description.284// DOC: Traits: `TraitOne`, `TraitThree`285 286def OpSkipAssemblyFormat : OpenMP_Op<"op_skip_assembly_format",287 traits=[TraitThree], clauses=[ReqClauseSkip<assemblyFormat=true>,288 OptClause]> {289 let summary = "operation with clauses - assemblyFormat skipped";290 let description = [{291 Description of operation with clauses - assemblyFormat skipped.292 }] # clausesDescription;293 let assemblyFormat = [{294 `alt_assembly_format` `(` $req_simple_val `:` type($req_simple_val) `)`295 }] # clausesAssemblyFormat # "attr-dict";296}297 298// DECL-LABEL: class OpSkipAssemblyFormat : public ::mlir::Op<299// DECL: getReqSimpleVal() {300// DECL: getOptSimpleVal() {301// DECL: void reqClauseExtraClassDecl();302// DECL: void optClauseExtraClassDecl();303 304// DOC-LABEL: omp::OpSkipAssemblyFormat305// DOC: _Operation with clauses - assemblyFormat skipped_306// DOC: operation ::= `omp.op_skip_assembly_format`307// DOC: `alt_assembly_format` `(` $req_simple_val `:` type($req_simple_val) `)`308// DOC: oilist(309// DOC: `optsimple` `(` $opt_simple_val `:` type($opt_simple_val) `)`310// DOC-NOT: $region311// DOC: attr-dict312// DOC: Description of operation with clauses - assemblyFormat skipped.313// DOC: Required clause description.314// DOC: Optional clause description.315// DOC: Traits: `TraitOne`, `TraitThree`316 317def OpSkipDescription : OpenMP_Op<"op_skip_description",318 traits=[TraitThree], clauses=[OptClauseSkip<description=true>]> {319 let summary = "operation with clause - description skipped";320 let description = [{321 Description of operation with clause - description skipped.322 }] # clausesDescription;323}324 325// DECL-LABEL: class OpSkipDescription : public ::mlir::Op<326// DECL: getOptSimpleVal() {327// DECL: void optClauseExtraClassDecl();328 329// DOC-LABEL: omp::OpSkipDescription330// DOC: _Operation with clause - description skipped_331// DOC: operation ::= `omp.op_skip_description`332// DOC: oilist(333// DOC: `optsimple` `(` $opt_simple_val `:` type($opt_simple_val) `)`334// DOC-NOT: $region335// DOC: attr-dict336// DOC: Description of operation with clause - description skipped.337// DOC-NOT: Optional clause description.338// DOC: Traits: `TraitOne`, `TraitThree`339 340def OpSkipExtraClassDeclaration : OpenMP_Op<"op_skip_extra_class_declaration",341 traits=[TraitThree], clauses=[OptClauseSkip<extraClassDeclaration=true>]> {342 let summary = "operation with clause - extraClassDeclaration skipped";343 let description = [{344 Description of operation with clause - extraClassDeclaration skipped.345 }] # clausesDescription;346}347 348// DECL-LABEL: class OpSkipExtraClassDeclaration : public ::mlir::Op<349// DECL: getOptSimpleVal() {350// DECL-NOT: void optClauseExtraClassDecl();351 352// DOC-LABEL: omp::OpSkipExtraClassDeclaration353// DOC: _Operation with clause - extraClassDeclaration skipped_354// DOC: operation ::= `omp.op_skip_extra_class_declaration`355// DOC: oilist(356// DOC: `optsimple` `(` $opt_simple_val `:` type($opt_simple_val) `)`357// DOC-NOT: $region358// DOC: attr-dict359// DOC: Description of operation with clause - extraClassDeclaration skipped.360// DOC: Optional clause description.361// DOC: Traits: `TraitOne`, `TraitThree`362 363def OpSkipTraits : OpenMP_Op<"op_skip_traits",364 traits=[TraitThree], clauses=[OptClauseSkip<traits=true>]> {365 let summary = "operation with clause - traits skipped";366 let description = [{367 Description of operation with clause - traits skipped.368 }] # clausesDescription;369}370 371// DECL-LABEL: class OpSkipTraits : public ::mlir::Op<372// DECL: getOptSimpleVal() {373// DECL: void optClauseExtraClassDecl();374 375// DOC-LABEL: omp::OpSkipTraits376// DOC: _Operation with clause - traits skipped_377// DOC: operation ::= `omp.op_skip_traits`378// DOC: oilist(379// DOC: `optsimple` `(` $opt_simple_val `:` type($opt_simple_val) `)`380// DOC-NOT: $region381// DOC: attr-dict382// DOC: Description of operation with clause - traits skipped.383// DOC: Optional clause description.384// DOC: Traits: `TraitThree`385 386def OpTwoOptionalClauses : OpenMP_Op<"op_two_opt_clauses",387 traits=[AttrSizedOperandSegments], clauses=[OptClause, OptClauseTwo]> {388 let summary = "operation with two optional clauses";389 let description = [{390 Description of operation with two optional clauses.391 }] # clausesDescription;392}393 394// DECL-LABEL: class OpTwoOptionalClauses : public ::mlir::Op<395// DECL: getOptSimpleVal() {396// DECL: getOptTwoVal() {397// DECL: void optClauseExtraClassDecl();398// DECL: void optClauseTwoExtraClassDecl();399 400// DOC-LABEL: omp::OpTwoOptionalClauses401// DOC: _Operation with two optional clauses_402// DOC: operation ::= `omp.op_two_opt_clauses`403// DOC: oilist(404// DOC: `optsimple` `(` $opt_simple_val `:` type($opt_simple_val) `)`405// DOC: |406// DOC: `opt_two` `(` $opt_two_val `:` type($opt_two_val) `)`407// DOC-NOT: $region408// DOC: attr-dict409// DOC: Description of operation with two optional clauses.410// DOC: Optional clause description.411// DOC: Optional clause two description.412// DOC: Traits: `AttrSizedOperandSegments`, `TraitOne`, `TraitTwo`413 414def OpTwoRequiredClauses : OpenMP_Op<"op_two_req_clauses",415 clauses=[ReqClause, ReqClauseTwo]> {416 let summary = "operation with two required clauses";417 let description = [{418 Description of operation with two required clauses.419 }] # clausesDescription;420}421 422// DECL-LABEL: class OpTwoRequiredClauses : public ::mlir::Op<423// DECL: getReqSimpleVal() {424// DECL: getReqTwoVal() {425// DECL: void reqClauseExtraClassDecl();426// DECL: void reqClauseTwoExtraClassDecl();427 428// DOC-LABEL: omp::OpTwoRequiredClauses429// DOC: _Operation with two required clauses_430// DOC: operation ::= `omp.op_two_req_clauses`431// DOC-NOT: oilist(432// DOC: `reqsimple` `(` $req_simple_val `:` type($req_simple_val) `)`433// DOC-NOT: |434// DOC: `req_two` `(` $req_two_val `:` type($req_two_val) `)`435// DOC-NOT: $region436// DOC: attr-dict437// DOC: Description of operation with two required clauses.438// DOC: Required clause description.439// DOC: Required clause two description.440// DOC: Traits: `TraitThree`, `TraitTwo`441 442def OpZeroClauses : OpenMP_Op<"op_zero_clauses"> {443 let summary = "operation with no clauses";444 let description = [{445 Description of operation with no clauses.446 }] # clausesDescription;447}448 449// DECL-LABEL: class OpZeroClauses : public ::mlir::Op<450// DECL: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState);451 452// DOC-LABEL: omp::OpZeroClauses453// DOC: _Operation with no clauses_454// DOC-NOT: operation ::=455// DOC: Description of operation with no clauses.456// DOC-NOT: Traits:457