1 //===--- ParseOpenMP.cpp - OpenMP directives parsing ----------------------===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 /// \file 10 /// \brief This file implements parsing of all OpenMP directives and clauses. 11 /// 12 //===----------------------------------------------------------------------===// 13 14 #include "clang/AST/ASTContext.h" 15 #include "clang/AST/StmtOpenMP.h" 16 #include "clang/Parse/ParseDiagnostic.h" 17 #include "clang/Parse/Parser.h" 18 #include "clang/Parse/RAIIObjectsForParser.h" 19 #include "clang/Sema/Scope.h" 20 #include "llvm/ADT/PointerIntPair.h" 21 22 using namespace clang; 23 24 //===----------------------------------------------------------------------===// 25 // OpenMP declarative directives. 26 //===----------------------------------------------------------------------===// 27 28 namespace { 29 enum OpenMPDirectiveKindEx { 30 OMPD_cancellation = OMPD_unknown + 1, 31 OMPD_data, 32 OMPD_declare, 33 OMPD_end, 34 OMPD_end_declare, 35 OMPD_enter, 36 OMPD_exit, 37 OMPD_point, 38 OMPD_reduction, 39 OMPD_target_enter, 40 OMPD_target_exit, 41 OMPD_update, 42 OMPD_distribute_parallel, 43 OMPD_teams_distribute_parallel, 44 OMPD_target_teams_distribute_parallel 45 }; 46 47 class ThreadprivateListParserHelper final { 48 SmallVector<Expr *, 4> Identifiers; 49 Parser *P; 50 51 public: 52 ThreadprivateListParserHelper(Parser *P) : P(P) {} 53 void operator()(CXXScopeSpec &SS, DeclarationNameInfo NameInfo) { 54 ExprResult Res = 55 P->getActions().ActOnOpenMPIdExpression(P->getCurScope(), SS, NameInfo); 56 if (Res.isUsable()) 57 Identifiers.push_back(Res.get()); 58 } 59 llvm::ArrayRef<Expr *> getIdentifiers() const { return Identifiers; } 60 }; 61 } // namespace 62 63 // Map token string to extended OMP token kind that are 64 // OpenMPDirectiveKind + OpenMPDirectiveKindEx. 65 static unsigned getOpenMPDirectiveKindEx(StringRef S) { 66 auto DKind = getOpenMPDirectiveKind(S); 67 if (DKind != OMPD_unknown) 68 return DKind; 69 70 return llvm::StringSwitch<unsigned>(S) 71 .Case("cancellation", OMPD_cancellation) 72 .Case("data", OMPD_data) 73 .Case("declare", OMPD_declare) 74 .Case("end", OMPD_end) 75 .Case("enter", OMPD_enter) 76 .Case("exit", OMPD_exit) 77 .Case("point", OMPD_point) 78 .Case("reduction", OMPD_reduction) 79 .Case("update", OMPD_update) 80 .Default(OMPD_unknown); 81 } 82 83 static OpenMPDirectiveKind ParseOpenMPDirectiveKind(Parser &P) { 84 // Array of foldings: F[i][0] F[i][1] ===> F[i][2]. 85 // E.g.: OMPD_for OMPD_simd ===> OMPD_for_simd 86 // TODO: add other combined directives in topological order. 87 static const unsigned F[][3] = { 88 { OMPD_cancellation, OMPD_point, OMPD_cancellation_point }, 89 { OMPD_declare, OMPD_reduction, OMPD_declare_reduction }, 90 { OMPD_declare, OMPD_simd, OMPD_declare_simd }, 91 { OMPD_declare, OMPD_target, OMPD_declare_target }, 92 { OMPD_distribute, OMPD_parallel, OMPD_distribute_parallel }, 93 { OMPD_distribute_parallel, OMPD_for, OMPD_distribute_parallel_for }, 94 { OMPD_distribute_parallel_for, OMPD_simd, 95 OMPD_distribute_parallel_for_simd }, 96 { OMPD_distribute, OMPD_simd, OMPD_distribute_simd }, 97 { OMPD_end, OMPD_declare, OMPD_end_declare }, 98 { OMPD_end_declare, OMPD_target, OMPD_end_declare_target }, 99 { OMPD_target, OMPD_data, OMPD_target_data }, 100 { OMPD_target, OMPD_enter, OMPD_target_enter }, 101 { OMPD_target, OMPD_exit, OMPD_target_exit }, 102 { OMPD_target, OMPD_update, OMPD_target_update }, 103 { OMPD_target_enter, OMPD_data, OMPD_target_enter_data }, 104 { OMPD_target_exit, OMPD_data, OMPD_target_exit_data }, 105 { OMPD_for, OMPD_simd, OMPD_for_simd }, 106 { OMPD_parallel, OMPD_for, OMPD_parallel_for }, 107 { OMPD_parallel_for, OMPD_simd, OMPD_parallel_for_simd }, 108 { OMPD_parallel, OMPD_sections, OMPD_parallel_sections }, 109 { OMPD_taskloop, OMPD_simd, OMPD_taskloop_simd }, 110 { OMPD_target, OMPD_parallel, OMPD_target_parallel }, 111 { OMPD_target, OMPD_simd, OMPD_target_simd }, 112 { OMPD_target_parallel, OMPD_for, OMPD_target_parallel_for }, 113 { OMPD_target_parallel_for, OMPD_simd, OMPD_target_parallel_for_simd }, 114 { OMPD_teams, OMPD_distribute, OMPD_teams_distribute }, 115 { OMPD_teams_distribute, OMPD_simd, OMPD_teams_distribute_simd }, 116 { OMPD_teams_distribute, OMPD_parallel, OMPD_teams_distribute_parallel }, 117 { OMPD_teams_distribute_parallel, OMPD_for, OMPD_teams_distribute_parallel_for }, 118 { OMPD_teams_distribute_parallel_for, OMPD_simd, OMPD_teams_distribute_parallel_for_simd }, 119 { OMPD_target, OMPD_teams, OMPD_target_teams }, 120 { OMPD_target_teams, OMPD_distribute, OMPD_target_teams_distribute }, 121 { OMPD_target_teams_distribute, OMPD_parallel, OMPD_target_teams_distribute_parallel }, 122 { OMPD_target_teams_distribute, OMPD_simd, OMPD_target_teams_distribute_simd }, 123 { OMPD_target_teams_distribute_parallel, OMPD_for, OMPD_target_teams_distribute_parallel_for }, 124 { OMPD_target_teams_distribute_parallel_for, OMPD_simd, OMPD_target_teams_distribute_parallel_for_simd } 125 }; 126 enum { CancellationPoint = 0, DeclareReduction = 1, TargetData = 2 }; 127 auto Tok = P.getCurToken(); 128 unsigned DKind = 129 Tok.isAnnotation() 130 ? static_cast<unsigned>(OMPD_unknown) 131 : getOpenMPDirectiveKindEx(P.getPreprocessor().getSpelling(Tok)); 132 if (DKind == OMPD_unknown) 133 return OMPD_unknown; 134 135 for (unsigned i = 0; i < llvm::array_lengthof(F); ++i) { 136 if (DKind != F[i][0]) 137 continue; 138 139 Tok = P.getPreprocessor().LookAhead(0); 140 unsigned SDKind = 141 Tok.isAnnotation() 142 ? static_cast<unsigned>(OMPD_unknown) 143 : getOpenMPDirectiveKindEx(P.getPreprocessor().getSpelling(Tok)); 144 if (SDKind == OMPD_unknown) 145 continue; 146 147 if (SDKind == F[i][1]) { 148 P.ConsumeToken(); 149 DKind = F[i][2]; 150 } 151 } 152 return DKind < OMPD_unknown ? static_cast<OpenMPDirectiveKind>(DKind) 153 : OMPD_unknown; 154 } 155 156 static DeclarationName parseOpenMPReductionId(Parser &P) { 157 Token Tok = P.getCurToken(); 158 Sema &Actions = P.getActions(); 159 OverloadedOperatorKind OOK = OO_None; 160 // Allow to use 'operator' keyword for C++ operators 161 bool WithOperator = false; 162 if (Tok.is(tok::kw_operator)) { 163 P.ConsumeToken(); 164 Tok = P.getCurToken(); 165 WithOperator = true; 166 } 167 switch (Tok.getKind()) { 168 case tok::plus: // '+' 169 OOK = OO_Plus; 170 break; 171 case tok::minus: // '-' 172 OOK = OO_Minus; 173 break; 174 case tok::star: // '*' 175 OOK = OO_Star; 176 break; 177 case tok::amp: // '&' 178 OOK = OO_Amp; 179 break; 180 case tok::pipe: // '|' 181 OOK = OO_Pipe; 182 break; 183 case tok::caret: // '^' 184 OOK = OO_Caret; 185 break; 186 case tok::ampamp: // '&&' 187 OOK = OO_AmpAmp; 188 break; 189 case tok::pipepipe: // '||' 190 OOK = OO_PipePipe; 191 break; 192 case tok::identifier: // identifier 193 if (!WithOperator) 194 break; 195 LLVM_FALLTHROUGH; 196 default: 197 P.Diag(Tok.getLocation(), diag::err_omp_expected_reduction_identifier); 198 P.SkipUntil(tok::colon, tok::r_paren, tok::annot_pragma_openmp_end, 199 Parser::StopBeforeMatch); 200 return DeclarationName(); 201 } 202 P.ConsumeToken(); 203 auto &DeclNames = Actions.getASTContext().DeclarationNames; 204 return OOK == OO_None ? DeclNames.getIdentifier(Tok.getIdentifierInfo()) 205 : DeclNames.getCXXOperatorName(OOK); 206 } 207 208 /// \brief Parse 'omp declare reduction' construct. 209 /// 210 /// declare-reduction-directive: 211 /// annot_pragma_openmp 'declare' 'reduction' 212 /// '(' <reduction_id> ':' <type> {',' <type>} ':' <expression> ')' 213 /// ['initializer' '(' ('omp_priv' '=' <expression>)|<function_call> ')'] 214 /// annot_pragma_openmp_end 215 /// <reduction_id> is either a base language identifier or one of the following 216 /// operators: '+', '-', '*', '&', '|', '^', '&&' and '||'. 217 /// 218 Parser::DeclGroupPtrTy 219 Parser::ParseOpenMPDeclareReductionDirective(AccessSpecifier AS) { 220 // Parse '('. 221 BalancedDelimiterTracker T(*this, tok::l_paren, tok::annot_pragma_openmp_end); 222 if (T.expectAndConsume(diag::err_expected_lparen_after, 223 getOpenMPDirectiveName(OMPD_declare_reduction))) { 224 SkipUntil(tok::annot_pragma_openmp_end, StopBeforeMatch); 225 return DeclGroupPtrTy(); 226 } 227 228 DeclarationName Name = parseOpenMPReductionId(*this); 229 if (Name.isEmpty() && Tok.is(tok::annot_pragma_openmp_end)) 230 return DeclGroupPtrTy(); 231 232 // Consume ':'. 233 bool IsCorrect = !ExpectAndConsume(tok::colon); 234 235 if (!IsCorrect && Tok.is(tok::annot_pragma_openmp_end)) 236 return DeclGroupPtrTy(); 237 238 IsCorrect = IsCorrect && !Name.isEmpty(); 239 240 if (Tok.is(tok::colon) || Tok.is(tok::annot_pragma_openmp_end)) { 241 Diag(Tok.getLocation(), diag::err_expected_type); 242 IsCorrect = false; 243 } 244 245 if (!IsCorrect && Tok.is(tok::annot_pragma_openmp_end)) 246 return DeclGroupPtrTy(); 247 248 SmallVector<std::pair<QualType, SourceLocation>, 8> ReductionTypes; 249 // Parse list of types until ':' token. 250 do { 251 ColonProtectionRAIIObject ColonRAII(*this); 252 SourceRange Range; 253 TypeResult TR = 254 ParseTypeName(&Range, DeclaratorContext::PrototypeContext, AS); 255 if (TR.isUsable()) { 256 auto ReductionType = 257 Actions.ActOnOpenMPDeclareReductionType(Range.getBegin(), TR); 258 if (!ReductionType.isNull()) { 259 ReductionTypes.push_back( 260 std::make_pair(ReductionType, Range.getBegin())); 261 } 262 } else { 263 SkipUntil(tok::comma, tok::colon, tok::annot_pragma_openmp_end, 264 StopBeforeMatch); 265 } 266 267 if (Tok.is(tok::colon) || Tok.is(tok::annot_pragma_openmp_end)) 268 break; 269 270 // Consume ','. 271 if (ExpectAndConsume(tok::comma)) { 272 IsCorrect = false; 273 if (Tok.is(tok::annot_pragma_openmp_end)) { 274 Diag(Tok.getLocation(), diag::err_expected_type); 275 return DeclGroupPtrTy(); 276 } 277 } 278 } while (Tok.isNot(tok::annot_pragma_openmp_end)); 279 280 if (ReductionTypes.empty()) { 281 SkipUntil(tok::annot_pragma_openmp_end, StopBeforeMatch); 282 return DeclGroupPtrTy(); 283 } 284 285 if (!IsCorrect && Tok.is(tok::annot_pragma_openmp_end)) 286 return DeclGroupPtrTy(); 287 288 // Consume ':'. 289 if (ExpectAndConsume(tok::colon)) 290 IsCorrect = false; 291 292 if (Tok.is(tok::annot_pragma_openmp_end)) { 293 Diag(Tok.getLocation(), diag::err_expected_expression); 294 return DeclGroupPtrTy(); 295 } 296 297 DeclGroupPtrTy DRD = Actions.ActOnOpenMPDeclareReductionDirectiveStart( 298 getCurScope(), Actions.getCurLexicalContext(), Name, ReductionTypes, AS); 299 300 // Parse <combiner> expression and then parse initializer if any for each 301 // correct type. 302 unsigned I = 0, E = ReductionTypes.size(); 303 for (auto *D : DRD.get()) { 304 TentativeParsingAction TPA(*this); 305 ParseScope OMPDRScope(this, Scope::FnScope | Scope::DeclScope | 306 Scope::CompoundStmtScope | 307 Scope::OpenMPDirectiveScope); 308 // Parse <combiner> expression. 309 Actions.ActOnOpenMPDeclareReductionCombinerStart(getCurScope(), D); 310 ExprResult CombinerResult = 311 Actions.ActOnFinishFullExpr(ParseAssignmentExpression().get(), 312 D->getLocation(), /*DiscardedValue=*/true); 313 Actions.ActOnOpenMPDeclareReductionCombinerEnd(D, CombinerResult.get()); 314 315 if (CombinerResult.isInvalid() && Tok.isNot(tok::r_paren) && 316 Tok.isNot(tok::annot_pragma_openmp_end)) { 317 TPA.Commit(); 318 IsCorrect = false; 319 break; 320 } 321 IsCorrect = !T.consumeClose() && IsCorrect && CombinerResult.isUsable(); 322 ExprResult InitializerResult; 323 if (Tok.isNot(tok::annot_pragma_openmp_end)) { 324 // Parse <initializer> expression. 325 if (Tok.is(tok::identifier) && 326 Tok.getIdentifierInfo()->isStr("initializer")) 327 ConsumeToken(); 328 else { 329 Diag(Tok.getLocation(), diag::err_expected) << "'initializer'"; 330 TPA.Commit(); 331 IsCorrect = false; 332 break; 333 } 334 // Parse '('. 335 BalancedDelimiterTracker T(*this, tok::l_paren, 336 tok::annot_pragma_openmp_end); 337 IsCorrect = 338 !T.expectAndConsume(diag::err_expected_lparen_after, "initializer") && 339 IsCorrect; 340 if (Tok.isNot(tok::annot_pragma_openmp_end)) { 341 ParseScope OMPDRScope(this, Scope::FnScope | Scope::DeclScope | 342 Scope::CompoundStmtScope | 343 Scope::OpenMPDirectiveScope); 344 // Parse expression. 345 VarDecl *OmpPrivParm = 346 Actions.ActOnOpenMPDeclareReductionInitializerStart(getCurScope(), 347 D); 348 // Check if initializer is omp_priv <init_expr> or something else. 349 if (Tok.is(tok::identifier) && 350 Tok.getIdentifierInfo()->isStr("omp_priv")) { 351 ConsumeToken(); 352 ParseOpenMPReductionInitializerForDecl(OmpPrivParm); 353 } else { 354 InitializerResult = Actions.ActOnFinishFullExpr( 355 ParseAssignmentExpression().get(), D->getLocation(), 356 /*DiscardedValue=*/true); 357 } 358 Actions.ActOnOpenMPDeclareReductionInitializerEnd( 359 D, InitializerResult.get(), OmpPrivParm); 360 if (InitializerResult.isInvalid() && Tok.isNot(tok::r_paren) && 361 Tok.isNot(tok::annot_pragma_openmp_end)) { 362 TPA.Commit(); 363 IsCorrect = false; 364 break; 365 } 366 IsCorrect = 367 !T.consumeClose() && IsCorrect && !InitializerResult.isInvalid(); 368 } 369 } 370 371 ++I; 372 // Revert parsing if not the last type, otherwise accept it, we're done with 373 // parsing. 374 if (I != E) 375 TPA.Revert(); 376 else 377 TPA.Commit(); 378 } 379 return Actions.ActOnOpenMPDeclareReductionDirectiveEnd(getCurScope(), DRD, 380 IsCorrect); 381 } 382 383 void Parser::ParseOpenMPReductionInitializerForDecl(VarDecl *OmpPrivParm) { 384 // Parse declarator '=' initializer. 385 // If a '==' or '+=' is found, suggest a fixit to '='. 386 if (isTokenEqualOrEqualTypo()) { 387 ConsumeToken(); 388 389 if (Tok.is(tok::code_completion)) { 390 Actions.CodeCompleteInitializer(getCurScope(), OmpPrivParm); 391 Actions.FinalizeDeclaration(OmpPrivParm); 392 cutOffParsing(); 393 return; 394 } 395 396 ExprResult Init(ParseInitializer()); 397 398 if (Init.isInvalid()) { 399 SkipUntil(tok::r_paren, tok::annot_pragma_openmp_end, StopBeforeMatch); 400 Actions.ActOnInitializerError(OmpPrivParm); 401 } else { 402 Actions.AddInitializerToDecl(OmpPrivParm, Init.get(), 403 /*DirectInit=*/false); 404 } 405 } else if (Tok.is(tok::l_paren)) { 406 // Parse C++ direct initializer: '(' expression-list ')' 407 BalancedDelimiterTracker T(*this, tok::l_paren); 408 T.consumeOpen(); 409 410 ExprVector Exprs; 411 CommaLocsTy CommaLocs; 412 413 if (ParseExpressionList(Exprs, CommaLocs, [this, OmpPrivParm, &Exprs] { 414 Actions.CodeCompleteConstructor( 415 getCurScope(), OmpPrivParm->getType()->getCanonicalTypeInternal(), 416 OmpPrivParm->getLocation(), Exprs); 417 })) { 418 Actions.ActOnInitializerError(OmpPrivParm); 419 SkipUntil(tok::r_paren, tok::annot_pragma_openmp_end, StopBeforeMatch); 420 } else { 421 // Match the ')'. 422 T.consumeClose(); 423 424 assert(!Exprs.empty() && Exprs.size() - 1 == CommaLocs.size() && 425 "Unexpected number of commas!"); 426 427 ExprResult Initializer = Actions.ActOnParenListExpr( 428 T.getOpenLocation(), T.getCloseLocation(), Exprs); 429 Actions.AddInitializerToDecl(OmpPrivParm, Initializer.get(), 430 /*DirectInit=*/true); 431 } 432 } else if (getLangOpts().CPlusPlus11 && Tok.is(tok::l_brace)) { 433 // Parse C++0x braced-init-list. 434 Diag(Tok, diag::warn_cxx98_compat_generalized_initializer_lists); 435 436 ExprResult Init(ParseBraceInitializer()); 437 438 if (Init.isInvalid()) { 439 Actions.ActOnInitializerError(OmpPrivParm); 440 } else { 441 Actions.AddInitializerToDecl(OmpPrivParm, Init.get(), 442 /*DirectInit=*/true); 443 } 444 } else { 445 Actions.ActOnUninitializedDecl(OmpPrivParm); 446 } 447 } 448 449 namespace { 450 /// RAII that recreates function context for correct parsing of clauses of 451 /// 'declare simd' construct. 452 /// OpenMP, 2.8.2 declare simd Construct 453 /// The expressions appearing in the clauses of this directive are evaluated in 454 /// the scope of the arguments of the function declaration or definition. 455 class FNContextRAII final { 456 Parser &P; 457 Sema::CXXThisScopeRAII *ThisScope; 458 Parser::ParseScope *TempScope; 459 Parser::ParseScope *FnScope; 460 bool HasTemplateScope = false; 461 bool HasFunScope = false; 462 FNContextRAII() = delete; 463 FNContextRAII(const FNContextRAII &) = delete; 464 FNContextRAII &operator=(const FNContextRAII &) = delete; 465 466 public: 467 FNContextRAII(Parser &P, Parser::DeclGroupPtrTy Ptr) : P(P) { 468 Decl *D = *Ptr.get().begin(); 469 NamedDecl *ND = dyn_cast<NamedDecl>(D); 470 RecordDecl *RD = dyn_cast_or_null<RecordDecl>(D->getDeclContext()); 471 Sema &Actions = P.getActions(); 472 473 // Allow 'this' within late-parsed attributes. 474 ThisScope = new Sema::CXXThisScopeRAII(Actions, RD, /*TypeQuals=*/0, 475 ND && ND->isCXXInstanceMember()); 476 477 // If the Decl is templatized, add template parameters to scope. 478 HasTemplateScope = D->isTemplateDecl(); 479 TempScope = 480 new Parser::ParseScope(&P, Scope::TemplateParamScope, HasTemplateScope); 481 if (HasTemplateScope) 482 Actions.ActOnReenterTemplateScope(Actions.getCurScope(), D); 483 484 // If the Decl is on a function, add function parameters to the scope. 485 HasFunScope = D->isFunctionOrFunctionTemplate(); 486 FnScope = new Parser::ParseScope( 487 &P, Scope::FnScope | Scope::DeclScope | Scope::CompoundStmtScope, 488 HasFunScope); 489 if (HasFunScope) 490 Actions.ActOnReenterFunctionContext(Actions.getCurScope(), D); 491 } 492 ~FNContextRAII() { 493 if (HasFunScope) { 494 P.getActions().ActOnExitFunctionContext(); 495 FnScope->Exit(); // Pop scope, and remove Decls from IdResolver 496 } 497 if (HasTemplateScope) 498 TempScope->Exit(); 499 delete FnScope; 500 delete TempScope; 501 delete ThisScope; 502 } 503 }; 504 } // namespace 505 506 /// Parses clauses for 'declare simd' directive. 507 /// clause: 508 /// 'inbranch' | 'notinbranch' 509 /// 'simdlen' '(' <expr> ')' 510 /// { 'uniform' '(' <argument_list> ')' } 511 /// { 'aligned '(' <argument_list> [ ':' <alignment> ] ')' } 512 /// { 'linear '(' <argument_list> [ ':' <step> ] ')' } 513 static bool parseDeclareSimdClauses( 514 Parser &P, OMPDeclareSimdDeclAttr::BranchStateTy &BS, ExprResult &SimdLen, 515 SmallVectorImpl<Expr *> &Uniforms, SmallVectorImpl<Expr *> &Aligneds, 516 SmallVectorImpl<Expr *> &Alignments, SmallVectorImpl<Expr *> &Linears, 517 SmallVectorImpl<unsigned> &LinModifiers, SmallVectorImpl<Expr *> &Steps) { 518 SourceRange BSRange; 519 const Token &Tok = P.getCurToken(); 520 bool IsError = false; 521 while (Tok.isNot(tok::annot_pragma_openmp_end)) { 522 if (Tok.isNot(tok::identifier)) 523 break; 524 OMPDeclareSimdDeclAttr::BranchStateTy Out; 525 IdentifierInfo *II = Tok.getIdentifierInfo(); 526 StringRef ClauseName = II->getName(); 527 // Parse 'inranch|notinbranch' clauses. 528 if (OMPDeclareSimdDeclAttr::ConvertStrToBranchStateTy(ClauseName, Out)) { 529 if (BS != OMPDeclareSimdDeclAttr::BS_Undefined && BS != Out) { 530 P.Diag(Tok, diag::err_omp_declare_simd_inbranch_notinbranch) 531 << ClauseName 532 << OMPDeclareSimdDeclAttr::ConvertBranchStateTyToStr(BS) << BSRange; 533 IsError = true; 534 } 535 BS = Out; 536 BSRange = SourceRange(Tok.getLocation(), Tok.getEndLoc()); 537 P.ConsumeToken(); 538 } else if (ClauseName.equals("simdlen")) { 539 if (SimdLen.isUsable()) { 540 P.Diag(Tok, diag::err_omp_more_one_clause) 541 << getOpenMPDirectiveName(OMPD_declare_simd) << ClauseName << 0; 542 IsError = true; 543 } 544 P.ConsumeToken(); 545 SourceLocation RLoc; 546 SimdLen = P.ParseOpenMPParensExpr(ClauseName, RLoc); 547 if (SimdLen.isInvalid()) 548 IsError = true; 549 } else { 550 OpenMPClauseKind CKind = getOpenMPClauseKind(ClauseName); 551 if (CKind == OMPC_uniform || CKind == OMPC_aligned || 552 CKind == OMPC_linear) { 553 Parser::OpenMPVarListDataTy Data; 554 auto *Vars = &Uniforms; 555 if (CKind == OMPC_aligned) 556 Vars = &Aligneds; 557 else if (CKind == OMPC_linear) 558 Vars = &Linears; 559 560 P.ConsumeToken(); 561 if (P.ParseOpenMPVarList(OMPD_declare_simd, 562 getOpenMPClauseKind(ClauseName), *Vars, Data)) 563 IsError = true; 564 if (CKind == OMPC_aligned) 565 Alignments.append(Aligneds.size() - Alignments.size(), Data.TailExpr); 566 else if (CKind == OMPC_linear) { 567 if (P.getActions().CheckOpenMPLinearModifier(Data.LinKind, 568 Data.DepLinMapLoc)) 569 Data.LinKind = OMPC_LINEAR_val; 570 LinModifiers.append(Linears.size() - LinModifiers.size(), 571 Data.LinKind); 572 Steps.append(Linears.size() - Steps.size(), Data.TailExpr); 573 } 574 } else 575 // TODO: add parsing of other clauses. 576 break; 577 } 578 // Skip ',' if any. 579 if (Tok.is(tok::comma)) 580 P.ConsumeToken(); 581 } 582 return IsError; 583 } 584 585 /// Parse clauses for '#pragma omp declare simd'. 586 Parser::DeclGroupPtrTy 587 Parser::ParseOMPDeclareSimdClauses(Parser::DeclGroupPtrTy Ptr, 588 CachedTokens &Toks, SourceLocation Loc) { 589 PP.EnterToken(Tok); 590 PP.EnterTokenStream(Toks, /*DisableMacroExpansion=*/true); 591 // Consume the previously pushed token. 592 ConsumeAnyToken(/*ConsumeCodeCompletionTok=*/true); 593 594 FNContextRAII FnContext(*this, Ptr); 595 OMPDeclareSimdDeclAttr::BranchStateTy BS = 596 OMPDeclareSimdDeclAttr::BS_Undefined; 597 ExprResult Simdlen; 598 SmallVector<Expr *, 4> Uniforms; 599 SmallVector<Expr *, 4> Aligneds; 600 SmallVector<Expr *, 4> Alignments; 601 SmallVector<Expr *, 4> Linears; 602 SmallVector<unsigned, 4> LinModifiers; 603 SmallVector<Expr *, 4> Steps; 604 bool IsError = 605 parseDeclareSimdClauses(*this, BS, Simdlen, Uniforms, Aligneds, 606 Alignments, Linears, LinModifiers, Steps); 607 // Need to check for extra tokens. 608 if (Tok.isNot(tok::annot_pragma_openmp_end)) { 609 Diag(Tok, diag::warn_omp_extra_tokens_at_eol) 610 << getOpenMPDirectiveName(OMPD_declare_simd); 611 while (Tok.isNot(tok::annot_pragma_openmp_end)) 612 ConsumeAnyToken(); 613 } 614 // Skip the last annot_pragma_openmp_end. 615 SourceLocation EndLoc = ConsumeAnnotationToken(); 616 if (!IsError) { 617 return Actions.ActOnOpenMPDeclareSimdDirective( 618 Ptr, BS, Simdlen.get(), Uniforms, Aligneds, Alignments, Linears, 619 LinModifiers, Steps, SourceRange(Loc, EndLoc)); 620 } 621 return Ptr; 622 } 623 624 /// \brief Parsing of declarative OpenMP directives. 625 /// 626 /// threadprivate-directive: 627 /// annot_pragma_openmp 'threadprivate' simple-variable-list 628 /// annot_pragma_openmp_end 629 /// 630 /// declare-reduction-directive: 631 /// annot_pragma_openmp 'declare' 'reduction' [...] 632 /// annot_pragma_openmp_end 633 /// 634 /// declare-simd-directive: 635 /// annot_pragma_openmp 'declare simd' {<clause> [,]} 636 /// annot_pragma_openmp_end 637 /// <function declaration/definition> 638 /// 639 Parser::DeclGroupPtrTy Parser::ParseOpenMPDeclarativeDirectiveWithExtDecl( 640 AccessSpecifier &AS, ParsedAttributesWithRange &Attrs, 641 DeclSpec::TST TagType, Decl *Tag) { 642 assert(Tok.is(tok::annot_pragma_openmp) && "Not an OpenMP directive!"); 643 ParenBraceBracketBalancer BalancerRAIIObj(*this); 644 645 SourceLocation Loc = ConsumeAnnotationToken(); 646 auto DKind = ParseOpenMPDirectiveKind(*this); 647 648 switch (DKind) { 649 case OMPD_threadprivate: { 650 ConsumeToken(); 651 ThreadprivateListParserHelper Helper(this); 652 if (!ParseOpenMPSimpleVarList(OMPD_threadprivate, Helper, true)) { 653 // The last seen token is annot_pragma_openmp_end - need to check for 654 // extra tokens. 655 if (Tok.isNot(tok::annot_pragma_openmp_end)) { 656 Diag(Tok, diag::warn_omp_extra_tokens_at_eol) 657 << getOpenMPDirectiveName(OMPD_threadprivate); 658 SkipUntil(tok::annot_pragma_openmp_end, StopBeforeMatch); 659 } 660 // Skip the last annot_pragma_openmp_end. 661 ConsumeAnnotationToken(); 662 return Actions.ActOnOpenMPThreadprivateDirective(Loc, 663 Helper.getIdentifiers()); 664 } 665 break; 666 } 667 case OMPD_declare_reduction: 668 ConsumeToken(); 669 if (auto Res = ParseOpenMPDeclareReductionDirective(AS)) { 670 // The last seen token is annot_pragma_openmp_end - need to check for 671 // extra tokens. 672 if (Tok.isNot(tok::annot_pragma_openmp_end)) { 673 Diag(Tok, diag::warn_omp_extra_tokens_at_eol) 674 << getOpenMPDirectiveName(OMPD_declare_reduction); 675 while (Tok.isNot(tok::annot_pragma_openmp_end)) 676 ConsumeAnyToken(); 677 } 678 // Skip the last annot_pragma_openmp_end. 679 ConsumeAnnotationToken(); 680 return Res; 681 } 682 break; 683 case OMPD_declare_simd: { 684 // The syntax is: 685 // { #pragma omp declare simd } 686 // <function-declaration-or-definition> 687 // 688 ConsumeToken(); 689 CachedTokens Toks; 690 while(Tok.isNot(tok::annot_pragma_openmp_end)) { 691 Toks.push_back(Tok); 692 ConsumeAnyToken(); 693 } 694 Toks.push_back(Tok); 695 ConsumeAnyToken(); 696 697 DeclGroupPtrTy Ptr; 698 if (Tok.is(tok::annot_pragma_openmp)) 699 Ptr = ParseOpenMPDeclarativeDirectiveWithExtDecl(AS, Attrs, TagType, Tag); 700 else if (Tok.isNot(tok::r_brace) && !isEofOrEom()) { 701 // Here we expect to see some function declaration. 702 if (AS == AS_none) { 703 assert(TagType == DeclSpec::TST_unspecified); 704 MaybeParseCXX11Attributes(Attrs); 705 ParsingDeclSpec PDS(*this); 706 Ptr = ParseExternalDeclaration(Attrs, &PDS); 707 } else { 708 Ptr = 709 ParseCXXClassMemberDeclarationWithPragmas(AS, Attrs, TagType, Tag); 710 } 711 } 712 if (!Ptr) { 713 Diag(Loc, diag::err_omp_decl_in_declare_simd); 714 return DeclGroupPtrTy(); 715 } 716 return ParseOMPDeclareSimdClauses(Ptr, Toks, Loc); 717 } 718 case OMPD_declare_target: { 719 SourceLocation DTLoc = ConsumeAnyToken(); 720 if (Tok.isNot(tok::annot_pragma_openmp_end)) { 721 // OpenMP 4.5 syntax with list of entities. 722 llvm::SmallSetVector<const NamedDecl*, 16> SameDirectiveDecls; 723 while (Tok.isNot(tok::annot_pragma_openmp_end)) { 724 OMPDeclareTargetDeclAttr::MapTypeTy MT = 725 OMPDeclareTargetDeclAttr::MT_To; 726 if (Tok.is(tok::identifier)) { 727 IdentifierInfo *II = Tok.getIdentifierInfo(); 728 StringRef ClauseName = II->getName(); 729 // Parse 'to|link' clauses. 730 if (!OMPDeclareTargetDeclAttr::ConvertStrToMapTypeTy(ClauseName, 731 MT)) { 732 Diag(Tok, diag::err_omp_declare_target_unexpected_clause) 733 << ClauseName; 734 break; 735 } 736 ConsumeToken(); 737 } 738 auto Callback = [this, MT, &SameDirectiveDecls]( 739 CXXScopeSpec &SS, DeclarationNameInfo NameInfo) { 740 Actions.ActOnOpenMPDeclareTargetName(getCurScope(), SS, NameInfo, MT, 741 SameDirectiveDecls); 742 }; 743 if (ParseOpenMPSimpleVarList(OMPD_declare_target, Callback, true)) 744 break; 745 746 // Consume optional ','. 747 if (Tok.is(tok::comma)) 748 ConsumeToken(); 749 } 750 SkipUntil(tok::annot_pragma_openmp_end, StopBeforeMatch); 751 ConsumeAnyToken(); 752 return DeclGroupPtrTy(); 753 } 754 755 // Skip the last annot_pragma_openmp_end. 756 ConsumeAnyToken(); 757 758 if (!Actions.ActOnStartOpenMPDeclareTargetDirective(DTLoc)) 759 return DeclGroupPtrTy(); 760 761 DKind = ParseOpenMPDirectiveKind(*this); 762 while (DKind != OMPD_end_declare_target && DKind != OMPD_declare_target && 763 Tok.isNot(tok::eof) && Tok.isNot(tok::r_brace)) { 764 DeclGroupPtrTy Ptr; 765 // Here we expect to see some function declaration. 766 if (AS == AS_none) { 767 assert(TagType == DeclSpec::TST_unspecified); 768 MaybeParseCXX11Attributes(Attrs); 769 ParsingDeclSpec PDS(*this); 770 Ptr = ParseExternalDeclaration(Attrs, &PDS); 771 } else { 772 Ptr = 773 ParseCXXClassMemberDeclarationWithPragmas(AS, Attrs, TagType, Tag); 774 } 775 if (Tok.isAnnotation() && Tok.is(tok::annot_pragma_openmp)) { 776 TentativeParsingAction TPA(*this); 777 ConsumeAnnotationToken(); 778 DKind = ParseOpenMPDirectiveKind(*this); 779 if (DKind != OMPD_end_declare_target) 780 TPA.Revert(); 781 else 782 TPA.Commit(); 783 } 784 } 785 786 if (DKind == OMPD_end_declare_target) { 787 ConsumeAnyToken(); 788 if (Tok.isNot(tok::annot_pragma_openmp_end)) { 789 Diag(Tok, diag::warn_omp_extra_tokens_at_eol) 790 << getOpenMPDirectiveName(OMPD_end_declare_target); 791 SkipUntil(tok::annot_pragma_openmp_end, StopBeforeMatch); 792 } 793 // Skip the last annot_pragma_openmp_end. 794 ConsumeAnyToken(); 795 } else { 796 Diag(Tok, diag::err_expected_end_declare_target); 797 Diag(DTLoc, diag::note_matching) << "'#pragma omp declare target'"; 798 } 799 Actions.ActOnFinishOpenMPDeclareTargetDirective(); 800 return DeclGroupPtrTy(); 801 } 802 case OMPD_unknown: 803 Diag(Tok, diag::err_omp_unknown_directive); 804 break; 805 case OMPD_parallel: 806 case OMPD_simd: 807 case OMPD_task: 808 case OMPD_taskyield: 809 case OMPD_barrier: 810 case OMPD_taskwait: 811 case OMPD_taskgroup: 812 case OMPD_flush: 813 case OMPD_for: 814 case OMPD_for_simd: 815 case OMPD_sections: 816 case OMPD_section: 817 case OMPD_single: 818 case OMPD_master: 819 case OMPD_ordered: 820 case OMPD_critical: 821 case OMPD_parallel_for: 822 case OMPD_parallel_for_simd: 823 case OMPD_parallel_sections: 824 case OMPD_atomic: 825 case OMPD_target: 826 case OMPD_teams: 827 case OMPD_cancellation_point: 828 case OMPD_cancel: 829 case OMPD_target_data: 830 case OMPD_target_enter_data: 831 case OMPD_target_exit_data: 832 case OMPD_target_parallel: 833 case OMPD_target_parallel_for: 834 case OMPD_taskloop: 835 case OMPD_taskloop_simd: 836 case OMPD_distribute: 837 case OMPD_end_declare_target: 838 case OMPD_target_update: 839 case OMPD_distribute_parallel_for: 840 case OMPD_distribute_parallel_for_simd: 841 case OMPD_distribute_simd: 842 case OMPD_target_parallel_for_simd: 843 case OMPD_target_simd: 844 case OMPD_teams_distribute: 845 case OMPD_teams_distribute_simd: 846 case OMPD_teams_distribute_parallel_for_simd: 847 case OMPD_teams_distribute_parallel_for: 848 case OMPD_target_teams: 849 case OMPD_target_teams_distribute: 850 case OMPD_target_teams_distribute_parallel_for: 851 case OMPD_target_teams_distribute_parallel_for_simd: 852 case OMPD_target_teams_distribute_simd: 853 Diag(Tok, diag::err_omp_unexpected_directive) 854 << getOpenMPDirectiveName(DKind); 855 break; 856 } 857 while (Tok.isNot(tok::annot_pragma_openmp_end)) 858 ConsumeAnyToken(); 859 ConsumeAnyToken(); 860 return nullptr; 861 } 862 863 /// \brief Parsing of declarative or executable OpenMP directives. 864 /// 865 /// threadprivate-directive: 866 /// annot_pragma_openmp 'threadprivate' simple-variable-list 867 /// annot_pragma_openmp_end 868 /// 869 /// declare-reduction-directive: 870 /// annot_pragma_openmp 'declare' 'reduction' '(' <reduction_id> ':' 871 /// <type> {',' <type>} ':' <expression> ')' ['initializer' '(' 872 /// ('omp_priv' '=' <expression>|<function_call>) ')'] 873 /// annot_pragma_openmp_end 874 /// 875 /// executable-directive: 876 /// annot_pragma_openmp 'parallel' | 'simd' | 'for' | 'sections' | 877 /// 'section' | 'single' | 'master' | 'critical' [ '(' <name> ')' ] | 878 /// 'parallel for' | 'parallel sections' | 'task' | 'taskyield' | 879 /// 'barrier' | 'taskwait' | 'flush' | 'ordered' | 'atomic' | 880 /// 'for simd' | 'parallel for simd' | 'target' | 'target data' | 881 /// 'taskgroup' | 'teams' | 'taskloop' | 'taskloop simd' | 882 /// 'distribute' | 'target enter data' | 'target exit data' | 883 /// 'target parallel' | 'target parallel for' | 884 /// 'target update' | 'distribute parallel for' | 885 /// 'distribute paralle for simd' | 'distribute simd' | 886 /// 'target parallel for simd' | 'target simd' | 887 /// 'teams distribute' | 'teams distribute simd' | 888 /// 'teams distribute parallel for simd' | 889 /// 'teams distribute parallel for' | 'target teams' | 890 /// 'target teams distribute' | 891 /// 'target teams distribute parallel for' | 892 /// 'target teams distribute parallel for simd' | 893 /// 'target teams distribute simd' {clause} 894 /// annot_pragma_openmp_end 895 /// 896 StmtResult Parser::ParseOpenMPDeclarativeOrExecutableDirective( 897 AllowedConstructsKind Allowed) { 898 assert(Tok.is(tok::annot_pragma_openmp) && "Not an OpenMP directive!"); 899 ParenBraceBracketBalancer BalancerRAIIObj(*this); 900 SmallVector<OMPClause *, 5> Clauses; 901 SmallVector<llvm::PointerIntPair<OMPClause *, 1, bool>, OMPC_unknown + 1> 902 FirstClauses(OMPC_unknown + 1); 903 unsigned ScopeFlags = Scope::FnScope | Scope::DeclScope | 904 Scope::CompoundStmtScope | Scope::OpenMPDirectiveScope; 905 SourceLocation Loc = ConsumeAnnotationToken(), EndLoc; 906 auto DKind = ParseOpenMPDirectiveKind(*this); 907 OpenMPDirectiveKind CancelRegion = OMPD_unknown; 908 // Name of critical directive. 909 DeclarationNameInfo DirName; 910 StmtResult Directive = StmtError(); 911 bool HasAssociatedStatement = true; 912 bool FlushHasClause = false; 913 914 switch (DKind) { 915 case OMPD_threadprivate: { 916 if (Allowed != ACK_Any) { 917 Diag(Tok, diag::err_omp_immediate_directive) 918 << getOpenMPDirectiveName(DKind) << 0; 919 } 920 ConsumeToken(); 921 ThreadprivateListParserHelper Helper(this); 922 if (!ParseOpenMPSimpleVarList(OMPD_threadprivate, Helper, false)) { 923 // The last seen token is annot_pragma_openmp_end - need to check for 924 // extra tokens. 925 if (Tok.isNot(tok::annot_pragma_openmp_end)) { 926 Diag(Tok, diag::warn_omp_extra_tokens_at_eol) 927 << getOpenMPDirectiveName(OMPD_threadprivate); 928 SkipUntil(tok::annot_pragma_openmp_end, StopBeforeMatch); 929 } 930 DeclGroupPtrTy Res = Actions.ActOnOpenMPThreadprivateDirective( 931 Loc, Helper.getIdentifiers()); 932 Directive = Actions.ActOnDeclStmt(Res, Loc, Tok.getLocation()); 933 } 934 SkipUntil(tok::annot_pragma_openmp_end); 935 break; 936 } 937 case OMPD_declare_reduction: 938 ConsumeToken(); 939 if (auto Res = ParseOpenMPDeclareReductionDirective(/*AS=*/AS_none)) { 940 // The last seen token is annot_pragma_openmp_end - need to check for 941 // extra tokens. 942 if (Tok.isNot(tok::annot_pragma_openmp_end)) { 943 Diag(Tok, diag::warn_omp_extra_tokens_at_eol) 944 << getOpenMPDirectiveName(OMPD_declare_reduction); 945 while (Tok.isNot(tok::annot_pragma_openmp_end)) 946 ConsumeAnyToken(); 947 } 948 ConsumeAnyToken(); 949 Directive = Actions.ActOnDeclStmt(Res, Loc, Tok.getLocation()); 950 } else 951 SkipUntil(tok::annot_pragma_openmp_end); 952 break; 953 case OMPD_flush: 954 if (PP.LookAhead(0).is(tok::l_paren)) { 955 FlushHasClause = true; 956 // Push copy of the current token back to stream to properly parse 957 // pseudo-clause OMPFlushClause. 958 PP.EnterToken(Tok); 959 } 960 LLVM_FALLTHROUGH; 961 case OMPD_taskyield: 962 case OMPD_barrier: 963 case OMPD_taskwait: 964 case OMPD_cancellation_point: 965 case OMPD_cancel: 966 case OMPD_target_enter_data: 967 case OMPD_target_exit_data: 968 case OMPD_target_update: 969 if (Allowed == ACK_StatementsOpenMPNonStandalone) { 970 Diag(Tok, diag::err_omp_immediate_directive) 971 << getOpenMPDirectiveName(DKind) << 0; 972 } 973 HasAssociatedStatement = false; 974 // Fall through for further analysis. 975 LLVM_FALLTHROUGH; 976 case OMPD_parallel: 977 case OMPD_simd: 978 case OMPD_for: 979 case OMPD_for_simd: 980 case OMPD_sections: 981 case OMPD_single: 982 case OMPD_section: 983 case OMPD_master: 984 case OMPD_critical: 985 case OMPD_parallel_for: 986 case OMPD_parallel_for_simd: 987 case OMPD_parallel_sections: 988 case OMPD_task: 989 case OMPD_ordered: 990 case OMPD_atomic: 991 case OMPD_target: 992 case OMPD_teams: 993 case OMPD_taskgroup: 994 case OMPD_target_data: 995 case OMPD_target_parallel: 996 case OMPD_target_parallel_for: 997 case OMPD_taskloop: 998 case OMPD_taskloop_simd: 999 case OMPD_distribute: 1000 case OMPD_distribute_parallel_for: 1001 case OMPD_distribute_parallel_for_simd: 1002 case OMPD_distribute_simd: 1003 case OMPD_target_parallel_for_simd: 1004 case OMPD_target_simd: 1005 case OMPD_teams_distribute: 1006 case OMPD_teams_distribute_simd: 1007 case OMPD_teams_distribute_parallel_for_simd: 1008 case OMPD_teams_distribute_parallel_for: 1009 case OMPD_target_teams: 1010 case OMPD_target_teams_distribute: 1011 case OMPD_target_teams_distribute_parallel_for: 1012 case OMPD_target_teams_distribute_parallel_for_simd: 1013 case OMPD_target_teams_distribute_simd: { 1014 ConsumeToken(); 1015 // Parse directive name of the 'critical' directive if any. 1016 if (DKind == OMPD_critical) { 1017 BalancedDelimiterTracker T(*this, tok::l_paren, 1018 tok::annot_pragma_openmp_end); 1019 if (!T.consumeOpen()) { 1020 if (Tok.isAnyIdentifier()) { 1021 DirName = 1022 DeclarationNameInfo(Tok.getIdentifierInfo(), Tok.getLocation()); 1023 ConsumeAnyToken(); 1024 } else { 1025 Diag(Tok, diag::err_omp_expected_identifier_for_critical); 1026 } 1027 T.consumeClose(); 1028 } 1029 } else if (DKind == OMPD_cancellation_point || DKind == OMPD_cancel) { 1030 CancelRegion = ParseOpenMPDirectiveKind(*this); 1031 if (Tok.isNot(tok::annot_pragma_openmp_end)) 1032 ConsumeToken(); 1033 } 1034 1035 if (isOpenMPLoopDirective(DKind)) 1036 ScopeFlags |= Scope::OpenMPLoopDirectiveScope; 1037 if (isOpenMPSimdDirective(DKind)) 1038 ScopeFlags |= Scope::OpenMPSimdDirectiveScope; 1039 ParseScope OMPDirectiveScope(this, ScopeFlags); 1040 Actions.StartOpenMPDSABlock(DKind, DirName, Actions.getCurScope(), Loc); 1041 1042 while (Tok.isNot(tok::annot_pragma_openmp_end)) { 1043 OpenMPClauseKind CKind = 1044 Tok.isAnnotation() 1045 ? OMPC_unknown 1046 : FlushHasClause ? OMPC_flush 1047 : getOpenMPClauseKind(PP.getSpelling(Tok)); 1048 Actions.StartOpenMPClause(CKind); 1049 FlushHasClause = false; 1050 OMPClause *Clause = 1051 ParseOpenMPClause(DKind, CKind, !FirstClauses[CKind].getInt()); 1052 FirstClauses[CKind].setInt(true); 1053 if (Clause) { 1054 FirstClauses[CKind].setPointer(Clause); 1055 Clauses.push_back(Clause); 1056 } 1057 1058 // Skip ',' if any. 1059 if (Tok.is(tok::comma)) 1060 ConsumeToken(); 1061 Actions.EndOpenMPClause(); 1062 } 1063 // End location of the directive. 1064 EndLoc = Tok.getLocation(); 1065 // Consume final annot_pragma_openmp_end. 1066 ConsumeAnnotationToken(); 1067 1068 // OpenMP [2.13.8, ordered Construct, Syntax] 1069 // If the depend clause is specified, the ordered construct is a stand-alone 1070 // directive. 1071 if (DKind == OMPD_ordered && FirstClauses[OMPC_depend].getInt()) { 1072 if (Allowed == ACK_StatementsOpenMPNonStandalone) { 1073 Diag(Loc, diag::err_omp_immediate_directive) 1074 << getOpenMPDirectiveName(DKind) << 1 1075 << getOpenMPClauseName(OMPC_depend); 1076 } 1077 HasAssociatedStatement = false; 1078 } 1079 1080 StmtResult AssociatedStmt; 1081 if (HasAssociatedStatement) { 1082 // The body is a block scope like in Lambdas and Blocks. 1083 Sema::CompoundScopeRAII CompoundScope(Actions); 1084 Actions.ActOnOpenMPRegionStart(DKind, getCurScope()); 1085 Actions.ActOnStartOfCompoundStmt(); 1086 // Parse statement 1087 AssociatedStmt = ParseStatement(); 1088 Actions.ActOnFinishOfCompoundStmt(); 1089 AssociatedStmt = Actions.ActOnOpenMPRegionEnd(AssociatedStmt, Clauses); 1090 } else if (DKind == OMPD_target_update || DKind == OMPD_target_enter_data || 1091 DKind == OMPD_target_exit_data) { 1092 Sema::CompoundScopeRAII CompoundScope(Actions); 1093 Actions.ActOnOpenMPRegionStart(DKind, getCurScope()); 1094 Actions.ActOnStartOfCompoundStmt(); 1095 AssociatedStmt = 1096 Actions.ActOnCompoundStmt(Loc, Loc, llvm::None, /*isStmtExpr=*/false); 1097 Actions.ActOnFinishOfCompoundStmt(); 1098 AssociatedStmt = Actions.ActOnOpenMPRegionEnd(AssociatedStmt, Clauses); 1099 } 1100 Directive = Actions.ActOnOpenMPExecutableDirective( 1101 DKind, DirName, CancelRegion, Clauses, AssociatedStmt.get(), Loc, 1102 EndLoc); 1103 1104 // Exit scope. 1105 Actions.EndOpenMPDSABlock(Directive.get()); 1106 OMPDirectiveScope.Exit(); 1107 break; 1108 } 1109 case OMPD_declare_simd: 1110 case OMPD_declare_target: 1111 case OMPD_end_declare_target: 1112 Diag(Tok, diag::err_omp_unexpected_directive) 1113 << getOpenMPDirectiveName(DKind); 1114 SkipUntil(tok::annot_pragma_openmp_end); 1115 break; 1116 case OMPD_unknown: 1117 Diag(Tok, diag::err_omp_unknown_directive); 1118 SkipUntil(tok::annot_pragma_openmp_end); 1119 break; 1120 } 1121 return Directive; 1122 } 1123 1124 // Parses simple list: 1125 // simple-variable-list: 1126 // '(' id-expression {, id-expression} ')' 1127 // 1128 bool Parser::ParseOpenMPSimpleVarList( 1129 OpenMPDirectiveKind Kind, 1130 const llvm::function_ref<void(CXXScopeSpec &, DeclarationNameInfo)> & 1131 Callback, 1132 bool AllowScopeSpecifier) { 1133 // Parse '('. 1134 BalancedDelimiterTracker T(*this, tok::l_paren, tok::annot_pragma_openmp_end); 1135 if (T.expectAndConsume(diag::err_expected_lparen_after, 1136 getOpenMPDirectiveName(Kind))) 1137 return true; 1138 bool IsCorrect = true; 1139 bool NoIdentIsFound = true; 1140 1141 // Read tokens while ')' or annot_pragma_openmp_end is not found. 1142 while (Tok.isNot(tok::r_paren) && Tok.isNot(tok::annot_pragma_openmp_end)) { 1143 CXXScopeSpec SS; 1144 SourceLocation TemplateKWLoc; 1145 UnqualifiedId Name; 1146 // Read var name. 1147 Token PrevTok = Tok; 1148 NoIdentIsFound = false; 1149 1150 if (AllowScopeSpecifier && getLangOpts().CPlusPlus && 1151 ParseOptionalCXXScopeSpecifier(SS, nullptr, false)) { 1152 IsCorrect = false; 1153 SkipUntil(tok::comma, tok::r_paren, tok::annot_pragma_openmp_end, 1154 StopBeforeMatch); 1155 } else if (ParseUnqualifiedId(SS, false, false, false, false, nullptr, 1156 TemplateKWLoc, Name)) { 1157 IsCorrect = false; 1158 SkipUntil(tok::comma, tok::r_paren, tok::annot_pragma_openmp_end, 1159 StopBeforeMatch); 1160 } else if (Tok.isNot(tok::comma) && Tok.isNot(tok::r_paren) && 1161 Tok.isNot(tok::annot_pragma_openmp_end)) { 1162 IsCorrect = false; 1163 SkipUntil(tok::comma, tok::r_paren, tok::annot_pragma_openmp_end, 1164 StopBeforeMatch); 1165 Diag(PrevTok.getLocation(), diag::err_expected) 1166 << tok::identifier 1167 << SourceRange(PrevTok.getLocation(), PrevTokLocation); 1168 } else { 1169 Callback(SS, Actions.GetNameFromUnqualifiedId(Name)); 1170 } 1171 // Consume ','. 1172 if (Tok.is(tok::comma)) { 1173 ConsumeToken(); 1174 } 1175 } 1176 1177 if (NoIdentIsFound) { 1178 Diag(Tok, diag::err_expected) << tok::identifier; 1179 IsCorrect = false; 1180 } 1181 1182 // Parse ')'. 1183 IsCorrect = !T.consumeClose() && IsCorrect; 1184 1185 return !IsCorrect; 1186 } 1187 1188 /// \brief Parsing of OpenMP clauses. 1189 /// 1190 /// clause: 1191 /// if-clause | final-clause | num_threads-clause | safelen-clause | 1192 /// default-clause | private-clause | firstprivate-clause | shared-clause 1193 /// | linear-clause | aligned-clause | collapse-clause | 1194 /// lastprivate-clause | reduction-clause | proc_bind-clause | 1195 /// schedule-clause | copyin-clause | copyprivate-clause | untied-clause | 1196 /// mergeable-clause | flush-clause | read-clause | write-clause | 1197 /// update-clause | capture-clause | seq_cst-clause | device-clause | 1198 /// simdlen-clause | threads-clause | simd-clause | num_teams-clause | 1199 /// thread_limit-clause | priority-clause | grainsize-clause | 1200 /// nogroup-clause | num_tasks-clause | hint-clause | to-clause | 1201 /// from-clause | is_device_ptr-clause | task_reduction-clause | 1202 /// in_reduction-clause 1203 /// 1204 OMPClause *Parser::ParseOpenMPClause(OpenMPDirectiveKind DKind, 1205 OpenMPClauseKind CKind, bool FirstClause) { 1206 OMPClause *Clause = nullptr; 1207 bool ErrorFound = false; 1208 // Check if clause is allowed for the given directive. 1209 if (CKind != OMPC_unknown && !isAllowedClauseForDirective(DKind, CKind)) { 1210 Diag(Tok, diag::err_omp_unexpected_clause) << getOpenMPClauseName(CKind) 1211 << getOpenMPDirectiveName(DKind); 1212 ErrorFound = true; 1213 } 1214 1215 switch (CKind) { 1216 case OMPC_final: 1217 case OMPC_num_threads: 1218 case OMPC_safelen: 1219 case OMPC_simdlen: 1220 case OMPC_collapse: 1221 case OMPC_ordered: 1222 case OMPC_device: 1223 case OMPC_num_teams: 1224 case OMPC_thread_limit: 1225 case OMPC_priority: 1226 case OMPC_grainsize: 1227 case OMPC_num_tasks: 1228 case OMPC_hint: 1229 // OpenMP [2.5, Restrictions] 1230 // At most one num_threads clause can appear on the directive. 1231 // OpenMP [2.8.1, simd construct, Restrictions] 1232 // Only one safelen clause can appear on a simd directive. 1233 // Only one simdlen clause can appear on a simd directive. 1234 // Only one collapse clause can appear on a simd directive. 1235 // OpenMP [2.9.1, target data construct, Restrictions] 1236 // At most one device clause can appear on the directive. 1237 // OpenMP [2.11.1, task Construct, Restrictions] 1238 // At most one if clause can appear on the directive. 1239 // At most one final clause can appear on the directive. 1240 // OpenMP [teams Construct, Restrictions] 1241 // At most one num_teams clause can appear on the directive. 1242 // At most one thread_limit clause can appear on the directive. 1243 // OpenMP [2.9.1, task Construct, Restrictions] 1244 // At most one priority clause can appear on the directive. 1245 // OpenMP [2.9.2, taskloop Construct, Restrictions] 1246 // At most one grainsize clause can appear on the directive. 1247 // OpenMP [2.9.2, taskloop Construct, Restrictions] 1248 // At most one num_tasks clause can appear on the directive. 1249 if (!FirstClause) { 1250 Diag(Tok, diag::err_omp_more_one_clause) 1251 << getOpenMPDirectiveName(DKind) << getOpenMPClauseName(CKind) << 0; 1252 ErrorFound = true; 1253 } 1254 1255 if (CKind == OMPC_ordered && PP.LookAhead(/*N=*/0).isNot(tok::l_paren)) 1256 Clause = ParseOpenMPClause(CKind); 1257 else 1258 Clause = ParseOpenMPSingleExprClause(CKind); 1259 break; 1260 case OMPC_default: 1261 case OMPC_proc_bind: 1262 // OpenMP [2.14.3.1, Restrictions] 1263 // Only a single default clause may be specified on a parallel, task or 1264 // teams directive. 1265 // OpenMP [2.5, parallel Construct, Restrictions] 1266 // At most one proc_bind clause can appear on the directive. 1267 if (!FirstClause) { 1268 Diag(Tok, diag::err_omp_more_one_clause) 1269 << getOpenMPDirectiveName(DKind) << getOpenMPClauseName(CKind) << 0; 1270 ErrorFound = true; 1271 } 1272 1273 Clause = ParseOpenMPSimpleClause(CKind); 1274 break; 1275 case OMPC_schedule: 1276 case OMPC_dist_schedule: 1277 case OMPC_defaultmap: 1278 // OpenMP [2.7.1, Restrictions, p. 3] 1279 // Only one schedule clause can appear on a loop directive. 1280 // OpenMP [2.10.4, Restrictions, p. 106] 1281 // At most one defaultmap clause can appear on the directive. 1282 if (!FirstClause) { 1283 Diag(Tok, diag::err_omp_more_one_clause) 1284 << getOpenMPDirectiveName(DKind) << getOpenMPClauseName(CKind) << 0; 1285 ErrorFound = true; 1286 } 1287 LLVM_FALLTHROUGH; 1288 1289 case OMPC_if: 1290 Clause = ParseOpenMPSingleExprWithArgClause(CKind); 1291 break; 1292 case OMPC_nowait: 1293 case OMPC_untied: 1294 case OMPC_mergeable: 1295 case OMPC_read: 1296 case OMPC_write: 1297 case OMPC_update: 1298 case OMPC_capture: 1299 case OMPC_seq_cst: 1300 case OMPC_threads: 1301 case OMPC_simd: 1302 case OMPC_nogroup: 1303 // OpenMP [2.7.1, Restrictions, p. 9] 1304 // Only one ordered clause can appear on a loop directive. 1305 // OpenMP [2.7.1, Restrictions, C/C++, p. 4] 1306 // Only one nowait clause can appear on a for directive. 1307 if (!FirstClause) { 1308 Diag(Tok, diag::err_omp_more_one_clause) 1309 << getOpenMPDirectiveName(DKind) << getOpenMPClauseName(CKind) << 0; 1310 ErrorFound = true; 1311 } 1312 1313 Clause = ParseOpenMPClause(CKind); 1314 break; 1315 case OMPC_private: 1316 case OMPC_firstprivate: 1317 case OMPC_lastprivate: 1318 case OMPC_shared: 1319 case OMPC_reduction: 1320 case OMPC_task_reduction: 1321 case OMPC_in_reduction: 1322 case OMPC_linear: 1323 case OMPC_aligned: 1324 case OMPC_copyin: 1325 case OMPC_copyprivate: 1326 case OMPC_flush: 1327 case OMPC_depend: 1328 case OMPC_map: 1329 case OMPC_to: 1330 case OMPC_from: 1331 case OMPC_use_device_ptr: 1332 case OMPC_is_device_ptr: 1333 Clause = ParseOpenMPVarListClause(DKind, CKind); 1334 break; 1335 case OMPC_unknown: 1336 Diag(Tok, diag::warn_omp_extra_tokens_at_eol) 1337 << getOpenMPDirectiveName(DKind); 1338 SkipUntil(tok::annot_pragma_openmp_end, StopBeforeMatch); 1339 break; 1340 case OMPC_threadprivate: 1341 case OMPC_uniform: 1342 Diag(Tok, diag::err_omp_unexpected_clause) << getOpenMPClauseName(CKind) 1343 << getOpenMPDirectiveName(DKind); 1344 SkipUntil(tok::comma, tok::annot_pragma_openmp_end, StopBeforeMatch); 1345 break; 1346 } 1347 return ErrorFound ? nullptr : Clause; 1348 } 1349 1350 /// Parses simple expression in parens for single-expression clauses of OpenMP 1351 /// constructs. 1352 /// \param RLoc Returned location of right paren. 1353 ExprResult Parser::ParseOpenMPParensExpr(StringRef ClauseName, 1354 SourceLocation &RLoc) { 1355 BalancedDelimiterTracker T(*this, tok::l_paren, tok::annot_pragma_openmp_end); 1356 if (T.expectAndConsume(diag::err_expected_lparen_after, ClauseName.data())) 1357 return ExprError(); 1358 1359 SourceLocation ELoc = Tok.getLocation(); 1360 ExprResult LHS(ParseCastExpression( 1361 /*isUnaryExpression=*/false, /*isAddressOfOperand=*/false, NotTypeCast)); 1362 ExprResult Val(ParseRHSOfBinaryExpression(LHS, prec::Conditional)); 1363 Val = Actions.ActOnFinishFullExpr(Val.get(), ELoc); 1364 1365 // Parse ')'. 1366 T.consumeClose(); 1367 1368 RLoc = T.getCloseLocation(); 1369 return Val; 1370 } 1371 1372 /// \brief Parsing of OpenMP clauses with single expressions like 'final', 1373 /// 'collapse', 'safelen', 'num_threads', 'simdlen', 'num_teams', 1374 /// 'thread_limit', 'simdlen', 'priority', 'grainsize', 'num_tasks' or 'hint'. 1375 /// 1376 /// final-clause: 1377 /// 'final' '(' expression ')' 1378 /// 1379 /// num_threads-clause: 1380 /// 'num_threads' '(' expression ')' 1381 /// 1382 /// safelen-clause: 1383 /// 'safelen' '(' expression ')' 1384 /// 1385 /// simdlen-clause: 1386 /// 'simdlen' '(' expression ')' 1387 /// 1388 /// collapse-clause: 1389 /// 'collapse' '(' expression ')' 1390 /// 1391 /// priority-clause: 1392 /// 'priority' '(' expression ')' 1393 /// 1394 /// grainsize-clause: 1395 /// 'grainsize' '(' expression ')' 1396 /// 1397 /// num_tasks-clause: 1398 /// 'num_tasks' '(' expression ')' 1399 /// 1400 /// hint-clause: 1401 /// 'hint' '(' expression ')' 1402 /// 1403 OMPClause *Parser::ParseOpenMPSingleExprClause(OpenMPClauseKind Kind) { 1404 SourceLocation Loc = ConsumeToken(); 1405 SourceLocation LLoc = Tok.getLocation(); 1406 SourceLocation RLoc; 1407 1408 ExprResult Val = ParseOpenMPParensExpr(getOpenMPClauseName(Kind), RLoc); 1409 1410 if (Val.isInvalid()) 1411 return nullptr; 1412 1413 return Actions.ActOnOpenMPSingleExprClause(Kind, Val.get(), Loc, LLoc, RLoc); 1414 } 1415 1416 /// \brief Parsing of simple OpenMP clauses like 'default' or 'proc_bind'. 1417 /// 1418 /// default-clause: 1419 /// 'default' '(' 'none' | 'shared' ') 1420 /// 1421 /// proc_bind-clause: 1422 /// 'proc_bind' '(' 'master' | 'close' | 'spread' ') 1423 /// 1424 OMPClause *Parser::ParseOpenMPSimpleClause(OpenMPClauseKind Kind) { 1425 SourceLocation Loc = Tok.getLocation(); 1426 SourceLocation LOpen = ConsumeToken(); 1427 // Parse '('. 1428 BalancedDelimiterTracker T(*this, tok::l_paren, tok::annot_pragma_openmp_end); 1429 if (T.expectAndConsume(diag::err_expected_lparen_after, 1430 getOpenMPClauseName(Kind))) 1431 return nullptr; 1432 1433 unsigned Type = getOpenMPSimpleClauseType( 1434 Kind, Tok.isAnnotation() ? "" : PP.getSpelling(Tok)); 1435 SourceLocation TypeLoc = Tok.getLocation(); 1436 if (Tok.isNot(tok::r_paren) && Tok.isNot(tok::comma) && 1437 Tok.isNot(tok::annot_pragma_openmp_end)) 1438 ConsumeAnyToken(); 1439 1440 // Parse ')'. 1441 T.consumeClose(); 1442 1443 return Actions.ActOnOpenMPSimpleClause(Kind, Type, TypeLoc, LOpen, Loc, 1444 Tok.getLocation()); 1445 } 1446 1447 /// \brief Parsing of OpenMP clauses like 'ordered'. 1448 /// 1449 /// ordered-clause: 1450 /// 'ordered' 1451 /// 1452 /// nowait-clause: 1453 /// 'nowait' 1454 /// 1455 /// untied-clause: 1456 /// 'untied' 1457 /// 1458 /// mergeable-clause: 1459 /// 'mergeable' 1460 /// 1461 /// read-clause: 1462 /// 'read' 1463 /// 1464 /// threads-clause: 1465 /// 'threads' 1466 /// 1467 /// simd-clause: 1468 /// 'simd' 1469 /// 1470 /// nogroup-clause: 1471 /// 'nogroup' 1472 /// 1473 OMPClause *Parser::ParseOpenMPClause(OpenMPClauseKind Kind) { 1474 SourceLocation Loc = Tok.getLocation(); 1475 ConsumeAnyToken(); 1476 1477 return Actions.ActOnOpenMPClause(Kind, Loc, Tok.getLocation()); 1478 } 1479 1480 1481 /// \brief Parsing of OpenMP clauses with single expressions and some additional 1482 /// argument like 'schedule' or 'dist_schedule'. 1483 /// 1484 /// schedule-clause: 1485 /// 'schedule' '(' [ modifier [ ',' modifier ] ':' ] kind [',' expression ] 1486 /// ')' 1487 /// 1488 /// if-clause: 1489 /// 'if' '(' [ directive-name-modifier ':' ] expression ')' 1490 /// 1491 /// defaultmap: 1492 /// 'defaultmap' '(' modifier ':' kind ')' 1493 /// 1494 OMPClause *Parser::ParseOpenMPSingleExprWithArgClause(OpenMPClauseKind Kind) { 1495 SourceLocation Loc = ConsumeToken(); 1496 SourceLocation DelimLoc; 1497 // Parse '('. 1498 BalancedDelimiterTracker T(*this, tok::l_paren, tok::annot_pragma_openmp_end); 1499 if (T.expectAndConsume(diag::err_expected_lparen_after, 1500 getOpenMPClauseName(Kind))) 1501 return nullptr; 1502 1503 ExprResult Val; 1504 SmallVector<unsigned, 4> Arg; 1505 SmallVector<SourceLocation, 4> KLoc; 1506 if (Kind == OMPC_schedule) { 1507 enum { Modifier1, Modifier2, ScheduleKind, NumberOfElements }; 1508 Arg.resize(NumberOfElements); 1509 KLoc.resize(NumberOfElements); 1510 Arg[Modifier1] = OMPC_SCHEDULE_MODIFIER_unknown; 1511 Arg[Modifier2] = OMPC_SCHEDULE_MODIFIER_unknown; 1512 Arg[ScheduleKind] = OMPC_SCHEDULE_unknown; 1513 auto KindModifier = getOpenMPSimpleClauseType( 1514 Kind, Tok.isAnnotation() ? "" : PP.getSpelling(Tok)); 1515 if (KindModifier > OMPC_SCHEDULE_unknown) { 1516 // Parse 'modifier' 1517 Arg[Modifier1] = KindModifier; 1518 KLoc[Modifier1] = Tok.getLocation(); 1519 if (Tok.isNot(tok::r_paren) && Tok.isNot(tok::comma) && 1520 Tok.isNot(tok::annot_pragma_openmp_end)) 1521 ConsumeAnyToken(); 1522 if (Tok.is(tok::comma)) { 1523 // Parse ',' 'modifier' 1524 ConsumeAnyToken(); 1525 KindModifier = getOpenMPSimpleClauseType( 1526 Kind, Tok.isAnnotation() ? "" : PP.getSpelling(Tok)); 1527 Arg[Modifier2] = KindModifier > OMPC_SCHEDULE_unknown 1528 ? KindModifier 1529 : (unsigned)OMPC_SCHEDULE_unknown; 1530 KLoc[Modifier2] = Tok.getLocation(); 1531 if (Tok.isNot(tok::r_paren) && Tok.isNot(tok::comma) && 1532 Tok.isNot(tok::annot_pragma_openmp_end)) 1533 ConsumeAnyToken(); 1534 } 1535 // Parse ':' 1536 if (Tok.is(tok::colon)) 1537 ConsumeAnyToken(); 1538 else 1539 Diag(Tok, diag::warn_pragma_expected_colon) << "schedule modifier"; 1540 KindModifier = getOpenMPSimpleClauseType( 1541 Kind, Tok.isAnnotation() ? "" : PP.getSpelling(Tok)); 1542 } 1543 Arg[ScheduleKind] = KindModifier; 1544 KLoc[ScheduleKind] = Tok.getLocation(); 1545 if (Tok.isNot(tok::r_paren) && Tok.isNot(tok::comma) && 1546 Tok.isNot(tok::annot_pragma_openmp_end)) 1547 ConsumeAnyToken(); 1548 if ((Arg[ScheduleKind] == OMPC_SCHEDULE_static || 1549 Arg[ScheduleKind] == OMPC_SCHEDULE_dynamic || 1550 Arg[ScheduleKind] == OMPC_SCHEDULE_guided) && 1551 Tok.is(tok::comma)) 1552 DelimLoc = ConsumeAnyToken(); 1553 } else if (Kind == OMPC_dist_schedule) { 1554 Arg.push_back(getOpenMPSimpleClauseType( 1555 Kind, Tok.isAnnotation() ? "" : PP.getSpelling(Tok))); 1556 KLoc.push_back(Tok.getLocation()); 1557 if (Tok.isNot(tok::r_paren) && Tok.isNot(tok::comma) && 1558 Tok.isNot(tok::annot_pragma_openmp_end)) 1559 ConsumeAnyToken(); 1560 if (Arg.back() == OMPC_DIST_SCHEDULE_static && Tok.is(tok::comma)) 1561 DelimLoc = ConsumeAnyToken(); 1562 } else if (Kind == OMPC_defaultmap) { 1563 // Get a defaultmap modifier 1564 Arg.push_back(getOpenMPSimpleClauseType( 1565 Kind, Tok.isAnnotation() ? "" : PP.getSpelling(Tok))); 1566 KLoc.push_back(Tok.getLocation()); 1567 if (Tok.isNot(tok::r_paren) && Tok.isNot(tok::comma) && 1568 Tok.isNot(tok::annot_pragma_openmp_end)) 1569 ConsumeAnyToken(); 1570 // Parse ':' 1571 if (Tok.is(tok::colon)) 1572 ConsumeAnyToken(); 1573 else if (Arg.back() != OMPC_DEFAULTMAP_MODIFIER_unknown) 1574 Diag(Tok, diag::warn_pragma_expected_colon) << "defaultmap modifier"; 1575 // Get a defaultmap kind 1576 Arg.push_back(getOpenMPSimpleClauseType( 1577 Kind, Tok.isAnnotation() ? "" : PP.getSpelling(Tok))); 1578 KLoc.push_back(Tok.getLocation()); 1579 if (Tok.isNot(tok::r_paren) && Tok.isNot(tok::comma) && 1580 Tok.isNot(tok::annot_pragma_openmp_end)) 1581 ConsumeAnyToken(); 1582 } else { 1583 assert(Kind == OMPC_if); 1584 KLoc.push_back(Tok.getLocation()); 1585 TentativeParsingAction TPA(*this); 1586 Arg.push_back(ParseOpenMPDirectiveKind(*this)); 1587 if (Arg.back() != OMPD_unknown) { 1588 ConsumeToken(); 1589 if (Tok.is(tok::colon) && getLangOpts().OpenMP > 40) { 1590 TPA.Commit(); 1591 DelimLoc = ConsumeToken(); 1592 } else { 1593 TPA.Revert(); 1594 Arg.back() = OMPD_unknown; 1595 } 1596 } else 1597 TPA.Revert(); 1598 } 1599 1600 bool NeedAnExpression = (Kind == OMPC_schedule && DelimLoc.isValid()) || 1601 (Kind == OMPC_dist_schedule && DelimLoc.isValid()) || 1602 Kind == OMPC_if; 1603 if (NeedAnExpression) { 1604 SourceLocation ELoc = Tok.getLocation(); 1605 ExprResult LHS(ParseCastExpression(false, false, NotTypeCast)); 1606 Val = ParseRHSOfBinaryExpression(LHS, prec::Conditional); 1607 Val = Actions.ActOnFinishFullExpr(Val.get(), ELoc); 1608 } 1609 1610 // Parse ')'. 1611 T.consumeClose(); 1612 1613 if (NeedAnExpression && Val.isInvalid()) 1614 return nullptr; 1615 1616 return Actions.ActOnOpenMPSingleExprWithArgClause( 1617 Kind, Arg, Val.get(), Loc, T.getOpenLocation(), KLoc, DelimLoc, 1618 T.getCloseLocation()); 1619 } 1620 1621 static bool ParseReductionId(Parser &P, CXXScopeSpec &ReductionIdScopeSpec, 1622 UnqualifiedId &ReductionId) { 1623 SourceLocation TemplateKWLoc; 1624 if (ReductionIdScopeSpec.isEmpty()) { 1625 auto OOK = OO_None; 1626 switch (P.getCurToken().getKind()) { 1627 case tok::plus: 1628 OOK = OO_Plus; 1629 break; 1630 case tok::minus: 1631 OOK = OO_Minus; 1632 break; 1633 case tok::star: 1634 OOK = OO_Star; 1635 break; 1636 case tok::amp: 1637 OOK = OO_Amp; 1638 break; 1639 case tok::pipe: 1640 OOK = OO_Pipe; 1641 break; 1642 case tok::caret: 1643 OOK = OO_Caret; 1644 break; 1645 case tok::ampamp: 1646 OOK = OO_AmpAmp; 1647 break; 1648 case tok::pipepipe: 1649 OOK = OO_PipePipe; 1650 break; 1651 default: 1652 break; 1653 } 1654 if (OOK != OO_None) { 1655 SourceLocation OpLoc = P.ConsumeToken(); 1656 SourceLocation SymbolLocations[] = {OpLoc, OpLoc, SourceLocation()}; 1657 ReductionId.setOperatorFunctionId(OpLoc, OOK, SymbolLocations); 1658 return false; 1659 } 1660 } 1661 return P.ParseUnqualifiedId(ReductionIdScopeSpec, /*EnteringContext*/ false, 1662 /*AllowDestructorName*/ false, 1663 /*AllowConstructorName*/ false, 1664 /*AllowDeductionGuide*/ false, 1665 nullptr, TemplateKWLoc, ReductionId); 1666 } 1667 1668 /// Parses clauses with list. 1669 bool Parser::ParseOpenMPVarList(OpenMPDirectiveKind DKind, 1670 OpenMPClauseKind Kind, 1671 SmallVectorImpl<Expr *> &Vars, 1672 OpenMPVarListDataTy &Data) { 1673 UnqualifiedId UnqualifiedReductionId; 1674 bool InvalidReductionId = false; 1675 bool MapTypeModifierSpecified = false; 1676 1677 // Parse '('. 1678 BalancedDelimiterTracker T(*this, tok::l_paren, tok::annot_pragma_openmp_end); 1679 if (T.expectAndConsume(diag::err_expected_lparen_after, 1680 getOpenMPClauseName(Kind))) 1681 return true; 1682 1683 bool NeedRParenForLinear = false; 1684 BalancedDelimiterTracker LinearT(*this, tok::l_paren, 1685 tok::annot_pragma_openmp_end); 1686 // Handle reduction-identifier for reduction clause. 1687 if (Kind == OMPC_reduction || Kind == OMPC_task_reduction || 1688 Kind == OMPC_in_reduction) { 1689 ColonProtectionRAIIObject ColonRAII(*this); 1690 if (getLangOpts().CPlusPlus) 1691 ParseOptionalCXXScopeSpecifier(Data.ReductionIdScopeSpec, 1692 /*ObjectType=*/nullptr, 1693 /*EnteringContext=*/false); 1694 InvalidReductionId = ParseReductionId(*this, Data.ReductionIdScopeSpec, 1695 UnqualifiedReductionId); 1696 if (InvalidReductionId) { 1697 SkipUntil(tok::colon, tok::r_paren, tok::annot_pragma_openmp_end, 1698 StopBeforeMatch); 1699 } 1700 if (Tok.is(tok::colon)) 1701 Data.ColonLoc = ConsumeToken(); 1702 else 1703 Diag(Tok, diag::warn_pragma_expected_colon) << "reduction identifier"; 1704 if (!InvalidReductionId) 1705 Data.ReductionId = 1706 Actions.GetNameFromUnqualifiedId(UnqualifiedReductionId); 1707 } else if (Kind == OMPC_depend) { 1708 // Handle dependency type for depend clause. 1709 ColonProtectionRAIIObject ColonRAII(*this); 1710 Data.DepKind = 1711 static_cast<OpenMPDependClauseKind>(getOpenMPSimpleClauseType( 1712 Kind, Tok.is(tok::identifier) ? PP.getSpelling(Tok) : "")); 1713 Data.DepLinMapLoc = Tok.getLocation(); 1714 1715 if (Data.DepKind == OMPC_DEPEND_unknown) { 1716 SkipUntil(tok::colon, tok::r_paren, tok::annot_pragma_openmp_end, 1717 StopBeforeMatch); 1718 } else { 1719 ConsumeToken(); 1720 // Special processing for depend(source) clause. 1721 if (DKind == OMPD_ordered && Data.DepKind == OMPC_DEPEND_source) { 1722 // Parse ')'. 1723 T.consumeClose(); 1724 return false; 1725 } 1726 } 1727 if (Tok.is(tok::colon)) 1728 Data.ColonLoc = ConsumeToken(); 1729 else { 1730 Diag(Tok, DKind == OMPD_ordered ? diag::warn_pragma_expected_colon_r_paren 1731 : diag::warn_pragma_expected_colon) 1732 << "dependency type"; 1733 } 1734 } else if (Kind == OMPC_linear) { 1735 // Try to parse modifier if any. 1736 if (Tok.is(tok::identifier) && PP.LookAhead(0).is(tok::l_paren)) { 1737 Data.LinKind = static_cast<OpenMPLinearClauseKind>( 1738 getOpenMPSimpleClauseType(Kind, PP.getSpelling(Tok))); 1739 Data.DepLinMapLoc = ConsumeToken(); 1740 LinearT.consumeOpen(); 1741 NeedRParenForLinear = true; 1742 } 1743 } else if (Kind == OMPC_map) { 1744 // Handle map type for map clause. 1745 ColonProtectionRAIIObject ColonRAII(*this); 1746 1747 /// The map clause modifier token can be either a identifier or the C++ 1748 /// delete keyword. 1749 auto &&IsMapClauseModifierToken = [](const Token &Tok) -> bool { 1750 return Tok.isOneOf(tok::identifier, tok::kw_delete); 1751 }; 1752 1753 // The first identifier may be a list item, a map-type or a 1754 // map-type-modifier. The map modifier can also be delete which has the same 1755 // spelling of the C++ delete keyword. 1756 Data.MapType = 1757 IsMapClauseModifierToken(Tok) 1758 ? static_cast<OpenMPMapClauseKind>( 1759 getOpenMPSimpleClauseType(Kind, PP.getSpelling(Tok))) 1760 : OMPC_MAP_unknown; 1761 Data.DepLinMapLoc = Tok.getLocation(); 1762 bool ColonExpected = false; 1763 1764 if (IsMapClauseModifierToken(Tok)) { 1765 if (PP.LookAhead(0).is(tok::colon)) { 1766 if (Data.MapType == OMPC_MAP_unknown) 1767 Diag(Tok, diag::err_omp_unknown_map_type); 1768 else if (Data.MapType == OMPC_MAP_always) 1769 Diag(Tok, diag::err_omp_map_type_missing); 1770 ConsumeToken(); 1771 } else if (PP.LookAhead(0).is(tok::comma)) { 1772 if (IsMapClauseModifierToken(PP.LookAhead(1)) && 1773 PP.LookAhead(2).is(tok::colon)) { 1774 Data.MapTypeModifier = Data.MapType; 1775 if (Data.MapTypeModifier != OMPC_MAP_always) { 1776 Diag(Tok, diag::err_omp_unknown_map_type_modifier); 1777 Data.MapTypeModifier = OMPC_MAP_unknown; 1778 } else 1779 MapTypeModifierSpecified = true; 1780 1781 ConsumeToken(); 1782 ConsumeToken(); 1783 1784 Data.MapType = 1785 IsMapClauseModifierToken(Tok) 1786 ? static_cast<OpenMPMapClauseKind>( 1787 getOpenMPSimpleClauseType(Kind, PP.getSpelling(Tok))) 1788 : OMPC_MAP_unknown; 1789 if (Data.MapType == OMPC_MAP_unknown || 1790 Data.MapType == OMPC_MAP_always) 1791 Diag(Tok, diag::err_omp_unknown_map_type); 1792 ConsumeToken(); 1793 } else { 1794 Data.MapType = OMPC_MAP_tofrom; 1795 Data.IsMapTypeImplicit = true; 1796 } 1797 } else if (IsMapClauseModifierToken(PP.LookAhead(0))) { 1798 if (PP.LookAhead(1).is(tok::colon)) { 1799 Data.MapTypeModifier = Data.MapType; 1800 if (Data.MapTypeModifier != OMPC_MAP_always) { 1801 Diag(Tok, diag::err_omp_unknown_map_type_modifier); 1802 Data.MapTypeModifier = OMPC_MAP_unknown; 1803 } else 1804 MapTypeModifierSpecified = true; 1805 1806 ConsumeToken(); 1807 1808 Data.MapType = 1809 IsMapClauseModifierToken(Tok) 1810 ? static_cast<OpenMPMapClauseKind>( 1811 getOpenMPSimpleClauseType(Kind, PP.getSpelling(Tok))) 1812 : OMPC_MAP_unknown; 1813 if (Data.MapType == OMPC_MAP_unknown || 1814 Data.MapType == OMPC_MAP_always) 1815 Diag(Tok, diag::err_omp_unknown_map_type); 1816 ConsumeToken(); 1817 } else { 1818 Data.MapType = OMPC_MAP_tofrom; 1819 Data.IsMapTypeImplicit = true; 1820 } 1821 } else { 1822 Data.MapType = OMPC_MAP_tofrom; 1823 Data.IsMapTypeImplicit = true; 1824 } 1825 } else { 1826 Data.MapType = OMPC_MAP_tofrom; 1827 Data.IsMapTypeImplicit = true; 1828 } 1829 1830 if (Tok.is(tok::colon)) 1831 Data.ColonLoc = ConsumeToken(); 1832 else if (ColonExpected) 1833 Diag(Tok, diag::warn_pragma_expected_colon) << "map type"; 1834 } 1835 1836 bool IsComma = 1837 (Kind != OMPC_reduction && Kind != OMPC_task_reduction && 1838 Kind != OMPC_in_reduction && Kind != OMPC_depend && Kind != OMPC_map) || 1839 (Kind == OMPC_reduction && !InvalidReductionId) || 1840 (Kind == OMPC_map && Data.MapType != OMPC_MAP_unknown && 1841 (!MapTypeModifierSpecified || 1842 Data.MapTypeModifier == OMPC_MAP_always)) || 1843 (Kind == OMPC_depend && Data.DepKind != OMPC_DEPEND_unknown); 1844 const bool MayHaveTail = (Kind == OMPC_linear || Kind == OMPC_aligned); 1845 while (IsComma || (Tok.isNot(tok::r_paren) && Tok.isNot(tok::colon) && 1846 Tok.isNot(tok::annot_pragma_openmp_end))) { 1847 ColonProtectionRAIIObject ColonRAII(*this, MayHaveTail); 1848 // Parse variable 1849 ExprResult VarExpr = 1850 Actions.CorrectDelayedTyposInExpr(ParseAssignmentExpression()); 1851 if (VarExpr.isUsable()) 1852 Vars.push_back(VarExpr.get()); 1853 else { 1854 SkipUntil(tok::comma, tok::r_paren, tok::annot_pragma_openmp_end, 1855 StopBeforeMatch); 1856 } 1857 // Skip ',' if any 1858 IsComma = Tok.is(tok::comma); 1859 if (IsComma) 1860 ConsumeToken(); 1861 else if (Tok.isNot(tok::r_paren) && 1862 Tok.isNot(tok::annot_pragma_openmp_end) && 1863 (!MayHaveTail || Tok.isNot(tok::colon))) 1864 Diag(Tok, diag::err_omp_expected_punc) 1865 << ((Kind == OMPC_flush) ? getOpenMPDirectiveName(OMPD_flush) 1866 : getOpenMPClauseName(Kind)) 1867 << (Kind == OMPC_flush); 1868 } 1869 1870 // Parse ')' for linear clause with modifier. 1871 if (NeedRParenForLinear) 1872 LinearT.consumeClose(); 1873 1874 // Parse ':' linear-step (or ':' alignment). 1875 const bool MustHaveTail = MayHaveTail && Tok.is(tok::colon); 1876 if (MustHaveTail) { 1877 Data.ColonLoc = Tok.getLocation(); 1878 SourceLocation ELoc = ConsumeToken(); 1879 ExprResult Tail = ParseAssignmentExpression(); 1880 Tail = Actions.ActOnFinishFullExpr(Tail.get(), ELoc); 1881 if (Tail.isUsable()) 1882 Data.TailExpr = Tail.get(); 1883 else 1884 SkipUntil(tok::comma, tok::r_paren, tok::annot_pragma_openmp_end, 1885 StopBeforeMatch); 1886 } 1887 1888 // Parse ')'. 1889 T.consumeClose(); 1890 if ((Kind == OMPC_depend && Data.DepKind != OMPC_DEPEND_unknown && 1891 Vars.empty()) || 1892 (Kind != OMPC_depend && Kind != OMPC_map && Vars.empty()) || 1893 (MustHaveTail && !Data.TailExpr) || InvalidReductionId) 1894 return true; 1895 return false; 1896 } 1897 1898 /// \brief Parsing of OpenMP clause 'private', 'firstprivate', 'lastprivate', 1899 /// 'shared', 'copyin', 'copyprivate', 'flush', 'reduction', 'task_reduction' or 1900 /// 'in_reduction'. 1901 /// 1902 /// private-clause: 1903 /// 'private' '(' list ')' 1904 /// firstprivate-clause: 1905 /// 'firstprivate' '(' list ')' 1906 /// lastprivate-clause: 1907 /// 'lastprivate' '(' list ')' 1908 /// shared-clause: 1909 /// 'shared' '(' list ')' 1910 /// linear-clause: 1911 /// 'linear' '(' linear-list [ ':' linear-step ] ')' 1912 /// aligned-clause: 1913 /// 'aligned' '(' list [ ':' alignment ] ')' 1914 /// reduction-clause: 1915 /// 'reduction' '(' reduction-identifier ':' list ')' 1916 /// task_reduction-clause: 1917 /// 'task_reduction' '(' reduction-identifier ':' list ')' 1918 /// in_reduction-clause: 1919 /// 'in_reduction' '(' reduction-identifier ':' list ')' 1920 /// copyprivate-clause: 1921 /// 'copyprivate' '(' list ')' 1922 /// flush-clause: 1923 /// 'flush' '(' list ')' 1924 /// depend-clause: 1925 /// 'depend' '(' in | out | inout : list | source ')' 1926 /// map-clause: 1927 /// 'map' '(' [ [ always , ] 1928 /// to | from | tofrom | alloc | release | delete ':' ] list ')'; 1929 /// to-clause: 1930 /// 'to' '(' list ')' 1931 /// from-clause: 1932 /// 'from' '(' list ')' 1933 /// use_device_ptr-clause: 1934 /// 'use_device_ptr' '(' list ')' 1935 /// is_device_ptr-clause: 1936 /// 'is_device_ptr' '(' list ')' 1937 /// 1938 /// For 'linear' clause linear-list may have the following forms: 1939 /// list 1940 /// modifier(list) 1941 /// where modifier is 'val' (C) or 'ref', 'val' or 'uval'(C++). 1942 OMPClause *Parser::ParseOpenMPVarListClause(OpenMPDirectiveKind DKind, 1943 OpenMPClauseKind Kind) { 1944 SourceLocation Loc = Tok.getLocation(); 1945 SourceLocation LOpen = ConsumeToken(); 1946 SmallVector<Expr *, 4> Vars; 1947 OpenMPVarListDataTy Data; 1948 1949 if (ParseOpenMPVarList(DKind, Kind, Vars, Data)) 1950 return nullptr; 1951 1952 return Actions.ActOnOpenMPVarListClause( 1953 Kind, Vars, Data.TailExpr, Loc, LOpen, Data.ColonLoc, Tok.getLocation(), 1954 Data.ReductionIdScopeSpec, Data.ReductionId, Data.DepKind, Data.LinKind, 1955 Data.MapTypeModifier, Data.MapType, Data.IsMapTypeImplicit, 1956 Data.DepLinMapLoc); 1957 } 1958 1959