1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987 |
- //===--- ParseOpenMP.cpp - OpenMP directives parsing ----------------------===//
- //
- // The LLVM Compiler Infrastructure
- //
- // This file is distributed under the University of Illinois Open Source
- // License. See LICENSE.TXT for details.
- //
- //===----------------------------------------------------------------------===//
- /// \file
- /// \brief This file implements parsing of all OpenMP directives and clauses.
- ///
- //===----------------------------------------------------------------------===//
- #include "clang/AST/ASTContext.h"
- #include "clang/AST/StmtOpenMP.h"
- #include "clang/Parse/ParseDiagnostic.h"
- #include "clang/Parse/Parser.h"
- #include "clang/Parse/RAIIObjectsForParser.h"
- #include "clang/Sema/Scope.h"
- #include "llvm/ADT/PointerIntPair.h"
- using namespace clang;
- //===----------------------------------------------------------------------===//
- // OpenMP declarative directives.
- //===----------------------------------------------------------------------===//
- namespace {
- enum OpenMPDirectiveKindEx {
- OMPD_cancellation = OMPD_unknown + 1,
- OMPD_data,
- OMPD_declare,
- OMPD_end,
- OMPD_end_declare,
- OMPD_enter,
- OMPD_exit,
- OMPD_point,
- OMPD_reduction,
- OMPD_target_enter,
- OMPD_target_exit,
- OMPD_update,
- OMPD_distribute_parallel,
- OMPD_teams_distribute_parallel,
- OMPD_target_teams_distribute_parallel
- };
- class ThreadprivateListParserHelper final {
- SmallVector<Expr *, 4> Identifiers;
- Parser *P;
- public:
- ThreadprivateListParserHelper(Parser *P) : P(P) {}
- void operator()(CXXScopeSpec &SS, DeclarationNameInfo NameInfo) {
- ExprResult Res =
- P->getActions().ActOnOpenMPIdExpression(P->getCurScope(), SS, NameInfo);
- if (Res.isUsable())
- Identifiers.push_back(Res.get());
- }
- llvm::ArrayRef<Expr *> getIdentifiers() const { return Identifiers; }
- };
- } // namespace
- // Map token string to extended OMP token kind that are
- // OpenMPDirectiveKind + OpenMPDirectiveKindEx.
- static unsigned getOpenMPDirectiveKindEx(StringRef S) {
- auto DKind = getOpenMPDirectiveKind(S);
- if (DKind != OMPD_unknown)
- return DKind;
- return llvm::StringSwitch<unsigned>(S)
- .Case("cancellation", OMPD_cancellation)
- .Case("data", OMPD_data)
- .Case("declare", OMPD_declare)
- .Case("end", OMPD_end)
- .Case("enter", OMPD_enter)
- .Case("exit", OMPD_exit)
- .Case("point", OMPD_point)
- .Case("reduction", OMPD_reduction)
- .Case("update", OMPD_update)
- .Default(OMPD_unknown);
- }
- static OpenMPDirectiveKind parseOpenMPDirectiveKind(Parser &P) {
- // Array of foldings: F[i][0] F[i][1] ===> F[i][2].
- // E.g.: OMPD_for OMPD_simd ===> OMPD_for_simd
- // TODO: add other combined directives in topological order.
- static const unsigned F[][3] = {
- {OMPD_cancellation, OMPD_point, OMPD_cancellation_point},
- {OMPD_declare, OMPD_reduction, OMPD_declare_reduction},
- {OMPD_declare, OMPD_simd, OMPD_declare_simd},
- {OMPD_declare, OMPD_target, OMPD_declare_target},
- {OMPD_distribute, OMPD_parallel, OMPD_distribute_parallel},
- {OMPD_distribute_parallel, OMPD_for, OMPD_distribute_parallel_for},
- {OMPD_distribute_parallel_for, OMPD_simd,
- OMPD_distribute_parallel_for_simd},
- {OMPD_distribute, OMPD_simd, OMPD_distribute_simd},
- {OMPD_end, OMPD_declare, OMPD_end_declare},
- {OMPD_end_declare, OMPD_target, OMPD_end_declare_target},
- {OMPD_target, OMPD_data, OMPD_target_data},
- {OMPD_target, OMPD_enter, OMPD_target_enter},
- {OMPD_target, OMPD_exit, OMPD_target_exit},
- {OMPD_target, OMPD_update, OMPD_target_update},
- {OMPD_target_enter, OMPD_data, OMPD_target_enter_data},
- {OMPD_target_exit, OMPD_data, OMPD_target_exit_data},
- {OMPD_for, OMPD_simd, OMPD_for_simd},
- {OMPD_parallel, OMPD_for, OMPD_parallel_for},
- {OMPD_parallel_for, OMPD_simd, OMPD_parallel_for_simd},
- {OMPD_parallel, OMPD_sections, OMPD_parallel_sections},
- {OMPD_taskloop, OMPD_simd, OMPD_taskloop_simd},
- {OMPD_target, OMPD_parallel, OMPD_target_parallel},
- {OMPD_target, OMPD_simd, OMPD_target_simd},
- {OMPD_target_parallel, OMPD_for, OMPD_target_parallel_for},
- {OMPD_target_parallel_for, OMPD_simd, OMPD_target_parallel_for_simd},
- {OMPD_teams, OMPD_distribute, OMPD_teams_distribute},
- {OMPD_teams_distribute, OMPD_simd, OMPD_teams_distribute_simd},
- {OMPD_teams_distribute, OMPD_parallel, OMPD_teams_distribute_parallel},
- {OMPD_teams_distribute_parallel, OMPD_for,
- OMPD_teams_distribute_parallel_for},
- {OMPD_teams_distribute_parallel_for, OMPD_simd,
- OMPD_teams_distribute_parallel_for_simd},
- {OMPD_target, OMPD_teams, OMPD_target_teams},
- {OMPD_target_teams, OMPD_distribute, OMPD_target_teams_distribute},
- {OMPD_target_teams_distribute, OMPD_parallel,
- OMPD_target_teams_distribute_parallel},
- {OMPD_target_teams_distribute, OMPD_simd,
- OMPD_target_teams_distribute_simd},
- {OMPD_target_teams_distribute_parallel, OMPD_for,
- OMPD_target_teams_distribute_parallel_for},
- {OMPD_target_teams_distribute_parallel_for, OMPD_simd,
- OMPD_target_teams_distribute_parallel_for_simd}};
- enum { CancellationPoint = 0, DeclareReduction = 1, TargetData = 2 };
- Token Tok = P.getCurToken();
- unsigned DKind =
- Tok.isAnnotation()
- ? static_cast<unsigned>(OMPD_unknown)
- : getOpenMPDirectiveKindEx(P.getPreprocessor().getSpelling(Tok));
- if (DKind == OMPD_unknown)
- return OMPD_unknown;
- for (unsigned I = 0; I < llvm::array_lengthof(F); ++I) {
- if (DKind != F[I][0])
- continue;
- Tok = P.getPreprocessor().LookAhead(0);
- unsigned SDKind =
- Tok.isAnnotation()
- ? static_cast<unsigned>(OMPD_unknown)
- : getOpenMPDirectiveKindEx(P.getPreprocessor().getSpelling(Tok));
- if (SDKind == OMPD_unknown)
- continue;
- if (SDKind == F[I][1]) {
- P.ConsumeToken();
- DKind = F[I][2];
- }
- }
- return DKind < OMPD_unknown ? static_cast<OpenMPDirectiveKind>(DKind)
- : OMPD_unknown;
- }
- static DeclarationName parseOpenMPReductionId(Parser &P) {
- Token Tok = P.getCurToken();
- Sema &Actions = P.getActions();
- OverloadedOperatorKind OOK = OO_None;
- // Allow to use 'operator' keyword for C++ operators
- bool WithOperator = false;
- if (Tok.is(tok::kw_operator)) {
- P.ConsumeToken();
- Tok = P.getCurToken();
- WithOperator = true;
- }
- switch (Tok.getKind()) {
- case tok::plus: // '+'
- OOK = OO_Plus;
- break;
- case tok::minus: // '-'
- OOK = OO_Minus;
- break;
- case tok::star: // '*'
- OOK = OO_Star;
- break;
- case tok::amp: // '&'
- OOK = OO_Amp;
- break;
- case tok::pipe: // '|'
- OOK = OO_Pipe;
- break;
- case tok::caret: // '^'
- OOK = OO_Caret;
- break;
- case tok::ampamp: // '&&'
- OOK = OO_AmpAmp;
- break;
- case tok::pipepipe: // '||'
- OOK = OO_PipePipe;
- break;
- case tok::identifier: // identifier
- if (!WithOperator)
- break;
- LLVM_FALLTHROUGH;
- default:
- P.Diag(Tok.getLocation(), diag::err_omp_expected_reduction_identifier);
- P.SkipUntil(tok::colon, tok::r_paren, tok::annot_pragma_openmp_end,
- Parser::StopBeforeMatch);
- return DeclarationName();
- }
- P.ConsumeToken();
- auto &DeclNames = Actions.getASTContext().DeclarationNames;
- return OOK == OO_None ? DeclNames.getIdentifier(Tok.getIdentifierInfo())
- : DeclNames.getCXXOperatorName(OOK);
- }
- /// \brief Parse 'omp declare reduction' construct.
- ///
- /// declare-reduction-directive:
- /// annot_pragma_openmp 'declare' 'reduction'
- /// '(' <reduction_id> ':' <type> {',' <type>} ':' <expression> ')'
- /// ['initializer' '(' ('omp_priv' '=' <expression>)|<function_call> ')']
- /// annot_pragma_openmp_end
- /// <reduction_id> is either a base language identifier or one of the following
- /// operators: '+', '-', '*', '&', '|', '^', '&&' and '||'.
- ///
- Parser::DeclGroupPtrTy
- Parser::ParseOpenMPDeclareReductionDirective(AccessSpecifier AS) {
- // Parse '('.
- BalancedDelimiterTracker T(*this, tok::l_paren, tok::annot_pragma_openmp_end);
- if (T.expectAndConsume(diag::err_expected_lparen_after,
- getOpenMPDirectiveName(OMPD_declare_reduction))) {
- SkipUntil(tok::annot_pragma_openmp_end, StopBeforeMatch);
- return DeclGroupPtrTy();
- }
- DeclarationName Name = parseOpenMPReductionId(*this);
- if (Name.isEmpty() && Tok.is(tok::annot_pragma_openmp_end))
- return DeclGroupPtrTy();
- // Consume ':'.
- bool IsCorrect = !ExpectAndConsume(tok::colon);
- if (!IsCorrect && Tok.is(tok::annot_pragma_openmp_end))
- return DeclGroupPtrTy();
- IsCorrect = IsCorrect && !Name.isEmpty();
- if (Tok.is(tok::colon) || Tok.is(tok::annot_pragma_openmp_end)) {
- Diag(Tok.getLocation(), diag::err_expected_type);
- IsCorrect = false;
- }
- if (!IsCorrect && Tok.is(tok::annot_pragma_openmp_end))
- return DeclGroupPtrTy();
- SmallVector<std::pair<QualType, SourceLocation>, 8> ReductionTypes;
- // Parse list of types until ':' token.
- do {
- ColonProtectionRAIIObject ColonRAII(*this);
- SourceRange Range;
- TypeResult TR =
- ParseTypeName(&Range, DeclaratorContext::PrototypeContext, AS);
- if (TR.isUsable()) {
- QualType ReductionType =
- Actions.ActOnOpenMPDeclareReductionType(Range.getBegin(), TR);
- if (!ReductionType.isNull()) {
- ReductionTypes.push_back(
- std::make_pair(ReductionType, Range.getBegin()));
- }
- } else {
- SkipUntil(tok::comma, tok::colon, tok::annot_pragma_openmp_end,
- StopBeforeMatch);
- }
- if (Tok.is(tok::colon) || Tok.is(tok::annot_pragma_openmp_end))
- break;
- // Consume ','.
- if (ExpectAndConsume(tok::comma)) {
- IsCorrect = false;
- if (Tok.is(tok::annot_pragma_openmp_end)) {
- Diag(Tok.getLocation(), diag::err_expected_type);
- return DeclGroupPtrTy();
- }
- }
- } while (Tok.isNot(tok::annot_pragma_openmp_end));
- if (ReductionTypes.empty()) {
- SkipUntil(tok::annot_pragma_openmp_end, StopBeforeMatch);
- return DeclGroupPtrTy();
- }
- if (!IsCorrect && Tok.is(tok::annot_pragma_openmp_end))
- return DeclGroupPtrTy();
- // Consume ':'.
- if (ExpectAndConsume(tok::colon))
- IsCorrect = false;
- if (Tok.is(tok::annot_pragma_openmp_end)) {
- Diag(Tok.getLocation(), diag::err_expected_expression);
- return DeclGroupPtrTy();
- }
- DeclGroupPtrTy DRD = Actions.ActOnOpenMPDeclareReductionDirectiveStart(
- getCurScope(), Actions.getCurLexicalContext(), Name, ReductionTypes, AS);
- // Parse <combiner> expression and then parse initializer if any for each
- // correct type.
- unsigned I = 0, E = ReductionTypes.size();
- for (Decl *D : DRD.get()) {
- TentativeParsingAction TPA(*this);
- ParseScope OMPDRScope(this, Scope::FnScope | Scope::DeclScope |
- Scope::CompoundStmtScope |
- Scope::OpenMPDirectiveScope);
- // Parse <combiner> expression.
- Actions.ActOnOpenMPDeclareReductionCombinerStart(getCurScope(), D);
- ExprResult CombinerResult =
- Actions.ActOnFinishFullExpr(ParseAssignmentExpression().get(),
- D->getLocation(), /*DiscardedValue=*/true);
- Actions.ActOnOpenMPDeclareReductionCombinerEnd(D, CombinerResult.get());
- if (CombinerResult.isInvalid() && Tok.isNot(tok::r_paren) &&
- Tok.isNot(tok::annot_pragma_openmp_end)) {
- TPA.Commit();
- IsCorrect = false;
- break;
- }
- IsCorrect = !T.consumeClose() && IsCorrect && CombinerResult.isUsable();
- ExprResult InitializerResult;
- if (Tok.isNot(tok::annot_pragma_openmp_end)) {
- // Parse <initializer> expression.
- if (Tok.is(tok::identifier) &&
- Tok.getIdentifierInfo()->isStr("initializer")) {
- ConsumeToken();
- } else {
- Diag(Tok.getLocation(), diag::err_expected) << "'initializer'";
- TPA.Commit();
- IsCorrect = false;
- break;
- }
- // Parse '('.
- BalancedDelimiterTracker T(*this, tok::l_paren,
- tok::annot_pragma_openmp_end);
- IsCorrect =
- !T.expectAndConsume(diag::err_expected_lparen_after, "initializer") &&
- IsCorrect;
- if (Tok.isNot(tok::annot_pragma_openmp_end)) {
- ParseScope OMPDRScope(this, Scope::FnScope | Scope::DeclScope |
- Scope::CompoundStmtScope |
- Scope::OpenMPDirectiveScope);
- // Parse expression.
- VarDecl *OmpPrivParm =
- Actions.ActOnOpenMPDeclareReductionInitializerStart(getCurScope(),
- D);
- // Check if initializer is omp_priv <init_expr> or something else.
- if (Tok.is(tok::identifier) &&
- Tok.getIdentifierInfo()->isStr("omp_priv")) {
- ConsumeToken();
- ParseOpenMPReductionInitializerForDecl(OmpPrivParm);
- } else {
- InitializerResult = Actions.ActOnFinishFullExpr(
- ParseAssignmentExpression().get(), D->getLocation(),
- /*DiscardedValue=*/true);
- }
- Actions.ActOnOpenMPDeclareReductionInitializerEnd(
- D, InitializerResult.get(), OmpPrivParm);
- if (InitializerResult.isInvalid() && Tok.isNot(tok::r_paren) &&
- Tok.isNot(tok::annot_pragma_openmp_end)) {
- TPA.Commit();
- IsCorrect = false;
- break;
- }
- IsCorrect =
- !T.consumeClose() && IsCorrect && !InitializerResult.isInvalid();
- }
- }
- ++I;
- // Revert parsing if not the last type, otherwise accept it, we're done with
- // parsing.
- if (I != E)
- TPA.Revert();
- else
- TPA.Commit();
- }
- return Actions.ActOnOpenMPDeclareReductionDirectiveEnd(getCurScope(), DRD,
- IsCorrect);
- }
- void Parser::ParseOpenMPReductionInitializerForDecl(VarDecl *OmpPrivParm) {
- // Parse declarator '=' initializer.
- // If a '==' or '+=' is found, suggest a fixit to '='.
- if (isTokenEqualOrEqualTypo()) {
- ConsumeToken();
- if (Tok.is(tok::code_completion)) {
- Actions.CodeCompleteInitializer(getCurScope(), OmpPrivParm);
- Actions.FinalizeDeclaration(OmpPrivParm);
- cutOffParsing();
- return;
- }
- ExprResult Init(ParseInitializer());
- if (Init.isInvalid()) {
- SkipUntil(tok::r_paren, tok::annot_pragma_openmp_end, StopBeforeMatch);
- Actions.ActOnInitializerError(OmpPrivParm);
- } else {
- Actions.AddInitializerToDecl(OmpPrivParm, Init.get(),
- /*DirectInit=*/false);
- }
- } else if (Tok.is(tok::l_paren)) {
- // Parse C++ direct initializer: '(' expression-list ')'
- BalancedDelimiterTracker T(*this, tok::l_paren);
- T.consumeOpen();
- ExprVector Exprs;
- CommaLocsTy CommaLocs;
- if (ParseExpressionList(Exprs, CommaLocs, [this, OmpPrivParm, &Exprs] {
- Actions.CodeCompleteConstructor(
- getCurScope(), OmpPrivParm->getType()->getCanonicalTypeInternal(),
- OmpPrivParm->getLocation(), Exprs);
- })) {
- Actions.ActOnInitializerError(OmpPrivParm);
- SkipUntil(tok::r_paren, tok::annot_pragma_openmp_end, StopBeforeMatch);
- } else {
- // Match the ')'.
- T.consumeClose();
- assert(!Exprs.empty() && Exprs.size() - 1 == CommaLocs.size() &&
- "Unexpected number of commas!");
- ExprResult Initializer = Actions.ActOnParenListExpr(
- T.getOpenLocation(), T.getCloseLocation(), Exprs);
- Actions.AddInitializerToDecl(OmpPrivParm, Initializer.get(),
- /*DirectInit=*/true);
- }
- } else if (getLangOpts().CPlusPlus11 && Tok.is(tok::l_brace)) {
- // Parse C++0x braced-init-list.
- Diag(Tok, diag::warn_cxx98_compat_generalized_initializer_lists);
- ExprResult Init(ParseBraceInitializer());
- if (Init.isInvalid()) {
- Actions.ActOnInitializerError(OmpPrivParm);
- } else {
- Actions.AddInitializerToDecl(OmpPrivParm, Init.get(),
- /*DirectInit=*/true);
- }
- } else {
- Actions.ActOnUninitializedDecl(OmpPrivParm);
- }
- }
- namespace {
- /// RAII that recreates function context for correct parsing of clauses of
- /// 'declare simd' construct.
- /// OpenMP, 2.8.2 declare simd Construct
- /// The expressions appearing in the clauses of this directive are evaluated in
- /// the scope of the arguments of the function declaration or definition.
- class FNContextRAII final {
- Parser &P;
- Sema::CXXThisScopeRAII *ThisScope;
- Parser::ParseScope *TempScope;
- Parser::ParseScope *FnScope;
- bool HasTemplateScope = false;
- bool HasFunScope = false;
- FNContextRAII() = delete;
- FNContextRAII(const FNContextRAII &) = delete;
- FNContextRAII &operator=(const FNContextRAII &) = delete;
- public:
- FNContextRAII(Parser &P, Parser::DeclGroupPtrTy Ptr) : P(P) {
- Decl *D = *Ptr.get().begin();
- NamedDecl *ND = dyn_cast<NamedDecl>(D);
- RecordDecl *RD = dyn_cast_or_null<RecordDecl>(D->getDeclContext());
- Sema &Actions = P.getActions();
- // Allow 'this' within late-parsed attributes.
- ThisScope = new Sema::CXXThisScopeRAII(Actions, RD, /*TypeQuals=*/0,
- ND && ND->isCXXInstanceMember());
- // If the Decl is templatized, add template parameters to scope.
- HasTemplateScope = D->isTemplateDecl();
- TempScope =
- new Parser::ParseScope(&P, Scope::TemplateParamScope, HasTemplateScope);
- if (HasTemplateScope)
- Actions.ActOnReenterTemplateScope(Actions.getCurScope(), D);
- // If the Decl is on a function, add function parameters to the scope.
- HasFunScope = D->isFunctionOrFunctionTemplate();
- FnScope = new Parser::ParseScope(
- &P, Scope::FnScope | Scope::DeclScope | Scope::CompoundStmtScope,
- HasFunScope);
- if (HasFunScope)
- Actions.ActOnReenterFunctionContext(Actions.getCurScope(), D);
- }
- ~FNContextRAII() {
- if (HasFunScope) {
- P.getActions().ActOnExitFunctionContext();
- FnScope->Exit(); // Pop scope, and remove Decls from IdResolver
- }
- if (HasTemplateScope)
- TempScope->Exit();
- delete FnScope;
- delete TempScope;
- delete ThisScope;
- }
- };
- } // namespace
- /// Parses clauses for 'declare simd' directive.
- /// clause:
- /// 'inbranch' | 'notinbranch'
- /// 'simdlen' '(' <expr> ')'
- /// { 'uniform' '(' <argument_list> ')' }
- /// { 'aligned '(' <argument_list> [ ':' <alignment> ] ')' }
- /// { 'linear '(' <argument_list> [ ':' <step> ] ')' }
- static bool parseDeclareSimdClauses(
- Parser &P, OMPDeclareSimdDeclAttr::BranchStateTy &BS, ExprResult &SimdLen,
- SmallVectorImpl<Expr *> &Uniforms, SmallVectorImpl<Expr *> &Aligneds,
- SmallVectorImpl<Expr *> &Alignments, SmallVectorImpl<Expr *> &Linears,
- SmallVectorImpl<unsigned> &LinModifiers, SmallVectorImpl<Expr *> &Steps) {
- SourceRange BSRange;
- const Token &Tok = P.getCurToken();
- bool IsError = false;
- while (Tok.isNot(tok::annot_pragma_openmp_end)) {
- if (Tok.isNot(tok::identifier))
- break;
- OMPDeclareSimdDeclAttr::BranchStateTy Out;
- IdentifierInfo *II = Tok.getIdentifierInfo();
- StringRef ClauseName = II->getName();
- // Parse 'inranch|notinbranch' clauses.
- if (OMPDeclareSimdDeclAttr::ConvertStrToBranchStateTy(ClauseName, Out)) {
- if (BS != OMPDeclareSimdDeclAttr::BS_Undefined && BS != Out) {
- P.Diag(Tok, diag::err_omp_declare_simd_inbranch_notinbranch)
- << ClauseName
- << OMPDeclareSimdDeclAttr::ConvertBranchStateTyToStr(BS) << BSRange;
- IsError = true;
- }
- BS = Out;
- BSRange = SourceRange(Tok.getLocation(), Tok.getEndLoc());
- P.ConsumeToken();
- } else if (ClauseName.equals("simdlen")) {
- if (SimdLen.isUsable()) {
- P.Diag(Tok, diag::err_omp_more_one_clause)
- << getOpenMPDirectiveName(OMPD_declare_simd) << ClauseName << 0;
- IsError = true;
- }
- P.ConsumeToken();
- SourceLocation RLoc;
- SimdLen = P.ParseOpenMPParensExpr(ClauseName, RLoc);
- if (SimdLen.isInvalid())
- IsError = true;
- } else {
- OpenMPClauseKind CKind = getOpenMPClauseKind(ClauseName);
- if (CKind == OMPC_uniform || CKind == OMPC_aligned ||
- CKind == OMPC_linear) {
- Parser::OpenMPVarListDataTy Data;
- SmallVectorImpl<Expr *> *Vars = &Uniforms;
- if (CKind == OMPC_aligned)
- Vars = &Aligneds;
- else if (CKind == OMPC_linear)
- Vars = &Linears;
- P.ConsumeToken();
- if (P.ParseOpenMPVarList(OMPD_declare_simd,
- getOpenMPClauseKind(ClauseName), *Vars, Data))
- IsError = true;
- if (CKind == OMPC_aligned) {
- Alignments.append(Aligneds.size() - Alignments.size(), Data.TailExpr);
- } else if (CKind == OMPC_linear) {
- if (P.getActions().CheckOpenMPLinearModifier(Data.LinKind,
- Data.DepLinMapLoc))
- Data.LinKind = OMPC_LINEAR_val;
- LinModifiers.append(Linears.size() - LinModifiers.size(),
- Data.LinKind);
- Steps.append(Linears.size() - Steps.size(), Data.TailExpr);
- }
- } else
- // TODO: add parsing of other clauses.
- break;
- }
- // Skip ',' if any.
- if (Tok.is(tok::comma))
- P.ConsumeToken();
- }
- return IsError;
- }
- /// Parse clauses for '#pragma omp declare simd'.
- Parser::DeclGroupPtrTy
- Parser::ParseOMPDeclareSimdClauses(Parser::DeclGroupPtrTy Ptr,
- CachedTokens &Toks, SourceLocation Loc) {
- PP.EnterToken(Tok);
- PP.EnterTokenStream(Toks, /*DisableMacroExpansion=*/true);
- // Consume the previously pushed token.
- ConsumeAnyToken(/*ConsumeCodeCompletionTok=*/true);
- FNContextRAII FnContext(*this, Ptr);
- OMPDeclareSimdDeclAttr::BranchStateTy BS =
- OMPDeclareSimdDeclAttr::BS_Undefined;
- ExprResult Simdlen;
- SmallVector<Expr *, 4> Uniforms;
- SmallVector<Expr *, 4> Aligneds;
- SmallVector<Expr *, 4> Alignments;
- SmallVector<Expr *, 4> Linears;
- SmallVector<unsigned, 4> LinModifiers;
- SmallVector<Expr *, 4> Steps;
- bool IsError =
- parseDeclareSimdClauses(*this, BS, Simdlen, Uniforms, Aligneds,
- Alignments, Linears, LinModifiers, Steps);
- // Need to check for extra tokens.
- if (Tok.isNot(tok::annot_pragma_openmp_end)) {
- Diag(Tok, diag::warn_omp_extra_tokens_at_eol)
- << getOpenMPDirectiveName(OMPD_declare_simd);
- while (Tok.isNot(tok::annot_pragma_openmp_end))
- ConsumeAnyToken();
- }
- // Skip the last annot_pragma_openmp_end.
- SourceLocation EndLoc = ConsumeAnnotationToken();
- if (IsError)
- return Ptr;
- return Actions.ActOnOpenMPDeclareSimdDirective(
- Ptr, BS, Simdlen.get(), Uniforms, Aligneds, Alignments, Linears,
- LinModifiers, Steps, SourceRange(Loc, EndLoc));
- }
- /// \brief Parsing of declarative OpenMP directives.
- ///
- /// threadprivate-directive:
- /// annot_pragma_openmp 'threadprivate' simple-variable-list
- /// annot_pragma_openmp_end
- ///
- /// declare-reduction-directive:
- /// annot_pragma_openmp 'declare' 'reduction' [...]
- /// annot_pragma_openmp_end
- ///
- /// declare-simd-directive:
- /// annot_pragma_openmp 'declare simd' {<clause> [,]}
- /// annot_pragma_openmp_end
- /// <function declaration/definition>
- ///
- Parser::DeclGroupPtrTy Parser::ParseOpenMPDeclarativeDirectiveWithExtDecl(
- AccessSpecifier &AS, ParsedAttributesWithRange &Attrs,
- DeclSpec::TST TagType, Decl *Tag) {
- assert(Tok.is(tok::annot_pragma_openmp) && "Not an OpenMP directive!");
- ParenBraceBracketBalancer BalancerRAIIObj(*this);
- SourceLocation Loc = ConsumeAnnotationToken();
- OpenMPDirectiveKind DKind = parseOpenMPDirectiveKind(*this);
- switch (DKind) {
- case OMPD_threadprivate: {
- ConsumeToken();
- ThreadprivateListParserHelper Helper(this);
- if (!ParseOpenMPSimpleVarList(OMPD_threadprivate, Helper, true)) {
- // The last seen token is annot_pragma_openmp_end - need to check for
- // extra tokens.
- if (Tok.isNot(tok::annot_pragma_openmp_end)) {
- Diag(Tok, diag::warn_omp_extra_tokens_at_eol)
- << getOpenMPDirectiveName(OMPD_threadprivate);
- SkipUntil(tok::annot_pragma_openmp_end, StopBeforeMatch);
- }
- // Skip the last annot_pragma_openmp_end.
- ConsumeAnnotationToken();
- return Actions.ActOnOpenMPThreadprivateDirective(Loc,
- Helper.getIdentifiers());
- }
- break;
- }
- case OMPD_declare_reduction:
- ConsumeToken();
- if (DeclGroupPtrTy Res = ParseOpenMPDeclareReductionDirective(AS)) {
- // The last seen token is annot_pragma_openmp_end - need to check for
- // extra tokens.
- if (Tok.isNot(tok::annot_pragma_openmp_end)) {
- Diag(Tok, diag::warn_omp_extra_tokens_at_eol)
- << getOpenMPDirectiveName(OMPD_declare_reduction);
- while (Tok.isNot(tok::annot_pragma_openmp_end))
- ConsumeAnyToken();
- }
- // Skip the last annot_pragma_openmp_end.
- ConsumeAnnotationToken();
- return Res;
- }
- break;
- case OMPD_declare_simd: {
- // The syntax is:
- // { #pragma omp declare simd }
- // <function-declaration-or-definition>
- //
- ConsumeToken();
- CachedTokens Toks;
- while(Tok.isNot(tok::annot_pragma_openmp_end)) {
- Toks.push_back(Tok);
- ConsumeAnyToken();
- }
- Toks.push_back(Tok);
- ConsumeAnyToken();
- DeclGroupPtrTy Ptr;
- if (Tok.is(tok::annot_pragma_openmp)) {
- Ptr = ParseOpenMPDeclarativeDirectiveWithExtDecl(AS, Attrs, TagType, Tag);
- } else if (Tok.isNot(tok::r_brace) && !isEofOrEom()) {
- // Here we expect to see some function declaration.
- if (AS == AS_none) {
- assert(TagType == DeclSpec::TST_unspecified);
- MaybeParseCXX11Attributes(Attrs);
- ParsingDeclSpec PDS(*this);
- Ptr = ParseExternalDeclaration(Attrs, &PDS);
- } else {
- Ptr =
- ParseCXXClassMemberDeclarationWithPragmas(AS, Attrs, TagType, Tag);
- }
- }
- if (!Ptr) {
- Diag(Loc, diag::err_omp_decl_in_declare_simd);
- return DeclGroupPtrTy();
- }
- return ParseOMPDeclareSimdClauses(Ptr, Toks, Loc);
- }
- case OMPD_declare_target: {
- SourceLocation DTLoc = ConsumeAnyToken();
- if (Tok.isNot(tok::annot_pragma_openmp_end)) {
- // OpenMP 4.5 syntax with list of entities.
- Sema::NamedDeclSetType SameDirectiveDecls;
- while (Tok.isNot(tok::annot_pragma_openmp_end)) {
- OMPDeclareTargetDeclAttr::MapTypeTy MT =
- OMPDeclareTargetDeclAttr::MT_To;
- if (Tok.is(tok::identifier)) {
- IdentifierInfo *II = Tok.getIdentifierInfo();
- StringRef ClauseName = II->getName();
- // Parse 'to|link' clauses.
- if (!OMPDeclareTargetDeclAttr::ConvertStrToMapTypeTy(ClauseName,
- MT)) {
- Diag(Tok, diag::err_omp_declare_target_unexpected_clause)
- << ClauseName;
- break;
- }
- ConsumeToken();
- }
- auto &&Callback = [this, MT, &SameDirectiveDecls](
- CXXScopeSpec &SS, DeclarationNameInfo NameInfo) {
- Actions.ActOnOpenMPDeclareTargetName(getCurScope(), SS, NameInfo, MT,
- SameDirectiveDecls);
- };
- if (ParseOpenMPSimpleVarList(OMPD_declare_target, Callback,
- /*AllowScopeSpecifier=*/true))
- break;
- // Consume optional ','.
- if (Tok.is(tok::comma))
- ConsumeToken();
- }
- SkipUntil(tok::annot_pragma_openmp_end, StopBeforeMatch);
- ConsumeAnyToken();
- SmallVector<Decl *, 4> Decls(SameDirectiveDecls.begin(),
- SameDirectiveDecls.end());
- if (Decls.empty())
- return DeclGroupPtrTy();
- return Actions.BuildDeclaratorGroup(Decls);
- }
- // Skip the last annot_pragma_openmp_end.
- ConsumeAnyToken();
- if (!Actions.ActOnStartOpenMPDeclareTargetDirective(DTLoc))
- return DeclGroupPtrTy();
- llvm::SmallVector<Decl *, 4> Decls;
- DKind = parseOpenMPDirectiveKind(*this);
- while (DKind != OMPD_end_declare_target && DKind != OMPD_declare_target &&
- Tok.isNot(tok::eof) && Tok.isNot(tok::r_brace)) {
- DeclGroupPtrTy Ptr;
- // Here we expect to see some function declaration.
- if (AS == AS_none) {
- assert(TagType == DeclSpec::TST_unspecified);
- MaybeParseCXX11Attributes(Attrs);
- ParsingDeclSpec PDS(*this);
- Ptr = ParseExternalDeclaration(Attrs, &PDS);
- } else {
- Ptr =
- ParseCXXClassMemberDeclarationWithPragmas(AS, Attrs, TagType, Tag);
- }
- if (Ptr) {
- DeclGroupRef Ref = Ptr.get();
- Decls.append(Ref.begin(), Ref.end());
- }
- if (Tok.isAnnotation() && Tok.is(tok::annot_pragma_openmp)) {
- TentativeParsingAction TPA(*this);
- ConsumeAnnotationToken();
- DKind = parseOpenMPDirectiveKind(*this);
- if (DKind != OMPD_end_declare_target)
- TPA.Revert();
- else
- TPA.Commit();
- }
- }
- if (DKind == OMPD_end_declare_target) {
- ConsumeAnyToken();
- if (Tok.isNot(tok::annot_pragma_openmp_end)) {
- Diag(Tok, diag::warn_omp_extra_tokens_at_eol)
- << getOpenMPDirectiveName(OMPD_end_declare_target);
- SkipUntil(tok::annot_pragma_openmp_end, StopBeforeMatch);
- }
- // Skip the last annot_pragma_openmp_end.
- ConsumeAnyToken();
- } else {
- Diag(Tok, diag::err_expected_end_declare_target);
- Diag(DTLoc, diag::note_matching) << "'#pragma omp declare target'";
- }
- Actions.ActOnFinishOpenMPDeclareTargetDirective();
- return Actions.BuildDeclaratorGroup(Decls);
- }
- case OMPD_unknown:
- Diag(Tok, diag::err_omp_unknown_directive);
- break;
- case OMPD_parallel:
- case OMPD_simd:
- case OMPD_task:
- case OMPD_taskyield:
- case OMPD_barrier:
- case OMPD_taskwait:
- case OMPD_taskgroup:
- case OMPD_flush:
- case OMPD_for:
- case OMPD_for_simd:
- case OMPD_sections:
- case OMPD_section:
- case OMPD_single:
- case OMPD_master:
- case OMPD_ordered:
- case OMPD_critical:
- case OMPD_parallel_for:
- case OMPD_parallel_for_simd:
- case OMPD_parallel_sections:
- case OMPD_atomic:
- case OMPD_target:
- case OMPD_teams:
- case OMPD_cancellation_point:
- case OMPD_cancel:
- case OMPD_target_data:
- case OMPD_target_enter_data:
- case OMPD_target_exit_data:
- case OMPD_target_parallel:
- case OMPD_target_parallel_for:
- case OMPD_taskloop:
- case OMPD_taskloop_simd:
- case OMPD_distribute:
- case OMPD_end_declare_target:
- case OMPD_target_update:
- case OMPD_distribute_parallel_for:
- case OMPD_distribute_parallel_for_simd:
- case OMPD_distribute_simd:
- case OMPD_target_parallel_for_simd:
- case OMPD_target_simd:
- case OMPD_teams_distribute:
- case OMPD_teams_distribute_simd:
- case OMPD_teams_distribute_parallel_for_simd:
- case OMPD_teams_distribute_parallel_for:
- case OMPD_target_teams:
- case OMPD_target_teams_distribute:
- case OMPD_target_teams_distribute_parallel_for:
- case OMPD_target_teams_distribute_parallel_for_simd:
- case OMPD_target_teams_distribute_simd:
- Diag(Tok, diag::err_omp_unexpected_directive)
- << 1 << getOpenMPDirectiveName(DKind);
- break;
- }
- while (Tok.isNot(tok::annot_pragma_openmp_end))
- ConsumeAnyToken();
- ConsumeAnyToken();
- return nullptr;
- }
- /// \brief Parsing of declarative or executable OpenMP directives.
- ///
- /// threadprivate-directive:
- /// annot_pragma_openmp 'threadprivate' simple-variable-list
- /// annot_pragma_openmp_end
- ///
- /// declare-reduction-directive:
- /// annot_pragma_openmp 'declare' 'reduction' '(' <reduction_id> ':'
- /// <type> {',' <type>} ':' <expression> ')' ['initializer' '('
- /// ('omp_priv' '=' <expression>|<function_call>) ')']
- /// annot_pragma_openmp_end
- ///
- /// executable-directive:
- /// annot_pragma_openmp 'parallel' | 'simd' | 'for' | 'sections' |
- /// 'section' | 'single' | 'master' | 'critical' [ '(' <name> ')' ] |
- /// 'parallel for' | 'parallel sections' | 'task' | 'taskyield' |
- /// 'barrier' | 'taskwait' | 'flush' | 'ordered' | 'atomic' |
- /// 'for simd' | 'parallel for simd' | 'target' | 'target data' |
- /// 'taskgroup' | 'teams' | 'taskloop' | 'taskloop simd' |
- /// 'distribute' | 'target enter data' | 'target exit data' |
- /// 'target parallel' | 'target parallel for' |
- /// 'target update' | 'distribute parallel for' |
- /// 'distribute paralle for simd' | 'distribute simd' |
- /// 'target parallel for simd' | 'target simd' |
- /// 'teams distribute' | 'teams distribute simd' |
- /// 'teams distribute parallel for simd' |
- /// 'teams distribute parallel for' | 'target teams' |
- /// 'target teams distribute' |
- /// 'target teams distribute parallel for' |
- /// 'target teams distribute parallel for simd' |
- /// 'target teams distribute simd' {clause}
- /// annot_pragma_openmp_end
- ///
- StmtResult Parser::ParseOpenMPDeclarativeOrExecutableDirective(
- AllowedConstructsKind Allowed) {
- assert(Tok.is(tok::annot_pragma_openmp) && "Not an OpenMP directive!");
- ParenBraceBracketBalancer BalancerRAIIObj(*this);
- SmallVector<OMPClause *, 5> Clauses;
- SmallVector<llvm::PointerIntPair<OMPClause *, 1, bool>, OMPC_unknown + 1>
- FirstClauses(OMPC_unknown + 1);
- unsigned ScopeFlags = Scope::FnScope | Scope::DeclScope |
- Scope::CompoundStmtScope | Scope::OpenMPDirectiveScope;
- SourceLocation Loc = ConsumeAnnotationToken(), EndLoc;
- OpenMPDirectiveKind DKind = parseOpenMPDirectiveKind(*this);
- OpenMPDirectiveKind CancelRegion = OMPD_unknown;
- // Name of critical directive.
- DeclarationNameInfo DirName;
- StmtResult Directive = StmtError();
- bool HasAssociatedStatement = true;
- bool FlushHasClause = false;
- switch (DKind) {
- case OMPD_threadprivate: {
- if (Allowed != ACK_Any) {
- Diag(Tok, diag::err_omp_immediate_directive)
- << getOpenMPDirectiveName(DKind) << 0;
- }
- ConsumeToken();
- ThreadprivateListParserHelper Helper(this);
- if (!ParseOpenMPSimpleVarList(OMPD_threadprivate, Helper, false)) {
- // The last seen token is annot_pragma_openmp_end - need to check for
- // extra tokens.
- if (Tok.isNot(tok::annot_pragma_openmp_end)) {
- Diag(Tok, diag::warn_omp_extra_tokens_at_eol)
- << getOpenMPDirectiveName(OMPD_threadprivate);
- SkipUntil(tok::annot_pragma_openmp_end, StopBeforeMatch);
- }
- DeclGroupPtrTy Res = Actions.ActOnOpenMPThreadprivateDirective(
- Loc, Helper.getIdentifiers());
- Directive = Actions.ActOnDeclStmt(Res, Loc, Tok.getLocation());
- }
- SkipUntil(tok::annot_pragma_openmp_end);
- break;
- }
- case OMPD_declare_reduction:
- ConsumeToken();
- if (DeclGroupPtrTy Res =
- ParseOpenMPDeclareReductionDirective(/*AS=*/AS_none)) {
- // The last seen token is annot_pragma_openmp_end - need to check for
- // extra tokens.
- if (Tok.isNot(tok::annot_pragma_openmp_end)) {
- Diag(Tok, diag::warn_omp_extra_tokens_at_eol)
- << getOpenMPDirectiveName(OMPD_declare_reduction);
- while (Tok.isNot(tok::annot_pragma_openmp_end))
- ConsumeAnyToken();
- }
- ConsumeAnyToken();
- Directive = Actions.ActOnDeclStmt(Res, Loc, Tok.getLocation());
- } else {
- SkipUntil(tok::annot_pragma_openmp_end);
- }
- break;
- case OMPD_flush:
- if (PP.LookAhead(0).is(tok::l_paren)) {
- FlushHasClause = true;
- // Push copy of the current token back to stream to properly parse
- // pseudo-clause OMPFlushClause.
- PP.EnterToken(Tok);
- }
- LLVM_FALLTHROUGH;
- case OMPD_taskyield:
- case OMPD_barrier:
- case OMPD_taskwait:
- case OMPD_cancellation_point:
- case OMPD_cancel:
- case OMPD_target_enter_data:
- case OMPD_target_exit_data:
- case OMPD_target_update:
- if (Allowed == ACK_StatementsOpenMPNonStandalone) {
- Diag(Tok, diag::err_omp_immediate_directive)
- << getOpenMPDirectiveName(DKind) << 0;
- }
- HasAssociatedStatement = false;
- // Fall through for further analysis.
- LLVM_FALLTHROUGH;
- case OMPD_parallel:
- case OMPD_simd:
- case OMPD_for:
- case OMPD_for_simd:
- case OMPD_sections:
- case OMPD_single:
- case OMPD_section:
- case OMPD_master:
- case OMPD_critical:
- case OMPD_parallel_for:
- case OMPD_parallel_for_simd:
- case OMPD_parallel_sections:
- case OMPD_task:
- case OMPD_ordered:
- case OMPD_atomic:
- case OMPD_target:
- case OMPD_teams:
- case OMPD_taskgroup:
- case OMPD_target_data:
- case OMPD_target_parallel:
- case OMPD_target_parallel_for:
- case OMPD_taskloop:
- case OMPD_taskloop_simd:
- case OMPD_distribute:
- case OMPD_distribute_parallel_for:
- case OMPD_distribute_parallel_for_simd:
- case OMPD_distribute_simd:
- case OMPD_target_parallel_for_simd:
- case OMPD_target_simd:
- case OMPD_teams_distribute:
- case OMPD_teams_distribute_simd:
- case OMPD_teams_distribute_parallel_for_simd:
- case OMPD_teams_distribute_parallel_for:
- case OMPD_target_teams:
- case OMPD_target_teams_distribute:
- case OMPD_target_teams_distribute_parallel_for:
- case OMPD_target_teams_distribute_parallel_for_simd:
- case OMPD_target_teams_distribute_simd: {
- ConsumeToken();
- // Parse directive name of the 'critical' directive if any.
- if (DKind == OMPD_critical) {
- BalancedDelimiterTracker T(*this, tok::l_paren,
- tok::annot_pragma_openmp_end);
- if (!T.consumeOpen()) {
- if (Tok.isAnyIdentifier()) {
- DirName =
- DeclarationNameInfo(Tok.getIdentifierInfo(), Tok.getLocation());
- ConsumeAnyToken();
- } else {
- Diag(Tok, diag::err_omp_expected_identifier_for_critical);
- }
- T.consumeClose();
- }
- } else if (DKind == OMPD_cancellation_point || DKind == OMPD_cancel) {
- CancelRegion = parseOpenMPDirectiveKind(*this);
- if (Tok.isNot(tok::annot_pragma_openmp_end))
- ConsumeToken();
- }
- if (isOpenMPLoopDirective(DKind))
- ScopeFlags |= Scope::OpenMPLoopDirectiveScope;
- if (isOpenMPSimdDirective(DKind))
- ScopeFlags |= Scope::OpenMPSimdDirectiveScope;
- ParseScope OMPDirectiveScope(this, ScopeFlags);
- Actions.StartOpenMPDSABlock(DKind, DirName, Actions.getCurScope(), Loc);
- while (Tok.isNot(tok::annot_pragma_openmp_end)) {
- OpenMPClauseKind CKind =
- Tok.isAnnotation()
- ? OMPC_unknown
- : FlushHasClause ? OMPC_flush
- : getOpenMPClauseKind(PP.getSpelling(Tok));
- Actions.StartOpenMPClause(CKind);
- FlushHasClause = false;
- OMPClause *Clause =
- ParseOpenMPClause(DKind, CKind, !FirstClauses[CKind].getInt());
- FirstClauses[CKind].setInt(true);
- if (Clause) {
- FirstClauses[CKind].setPointer(Clause);
- Clauses.push_back(Clause);
- }
- // Skip ',' if any.
- if (Tok.is(tok::comma))
- ConsumeToken();
- Actions.EndOpenMPClause();
- }
- // End location of the directive.
- EndLoc = Tok.getLocation();
- // Consume final annot_pragma_openmp_end.
- ConsumeAnnotationToken();
- // OpenMP [2.13.8, ordered Construct, Syntax]
- // If the depend clause is specified, the ordered construct is a stand-alone
- // directive.
- if (DKind == OMPD_ordered && FirstClauses[OMPC_depend].getInt()) {
- if (Allowed == ACK_StatementsOpenMPNonStandalone) {
- Diag(Loc, diag::err_omp_immediate_directive)
- << getOpenMPDirectiveName(DKind) << 1
- << getOpenMPClauseName(OMPC_depend);
- }
- HasAssociatedStatement = false;
- }
- StmtResult AssociatedStmt;
- if (HasAssociatedStatement) {
- // The body is a block scope like in Lambdas and Blocks.
- Actions.ActOnOpenMPRegionStart(DKind, getCurScope());
- // FIXME: We create a bogus CompoundStmt scope to hold the contents of
- // the captured region. Code elsewhere assumes that any FunctionScopeInfo
- // should have at least one compound statement scope within it.
- AssociatedStmt = (Sema::CompoundScopeRAII(Actions), ParseStatement());
- AssociatedStmt = Actions.ActOnOpenMPRegionEnd(AssociatedStmt, Clauses);
- } else if (DKind == OMPD_target_update || DKind == OMPD_target_enter_data ||
- DKind == OMPD_target_exit_data) {
- Actions.ActOnOpenMPRegionStart(DKind, getCurScope());
- AssociatedStmt = (Sema::CompoundScopeRAII(Actions),
- Actions.ActOnCompoundStmt(Loc, Loc, llvm::None,
- /*isStmtExpr=*/false));
- AssociatedStmt = Actions.ActOnOpenMPRegionEnd(AssociatedStmt, Clauses);
- }
- Directive = Actions.ActOnOpenMPExecutableDirective(
- DKind, DirName, CancelRegion, Clauses, AssociatedStmt.get(), Loc,
- EndLoc);
- // Exit scope.
- Actions.EndOpenMPDSABlock(Directive.get());
- OMPDirectiveScope.Exit();
- break;
- }
- case OMPD_declare_simd:
- case OMPD_declare_target:
- case OMPD_end_declare_target:
- Diag(Tok, diag::err_omp_unexpected_directive)
- << 1 << getOpenMPDirectiveName(DKind);
- SkipUntil(tok::annot_pragma_openmp_end);
- break;
- case OMPD_unknown:
- Diag(Tok, diag::err_omp_unknown_directive);
- SkipUntil(tok::annot_pragma_openmp_end);
- break;
- }
- return Directive;
- }
- // Parses simple list:
- // simple-variable-list:
- // '(' id-expression {, id-expression} ')'
- //
- bool Parser::ParseOpenMPSimpleVarList(
- OpenMPDirectiveKind Kind,
- const llvm::function_ref<void(CXXScopeSpec &, DeclarationNameInfo)> &
- Callback,
- bool AllowScopeSpecifier) {
- // Parse '('.
- BalancedDelimiterTracker T(*this, tok::l_paren, tok::annot_pragma_openmp_end);
- if (T.expectAndConsume(diag::err_expected_lparen_after,
- getOpenMPDirectiveName(Kind)))
- return true;
- bool IsCorrect = true;
- bool NoIdentIsFound = true;
- // Read tokens while ')' or annot_pragma_openmp_end is not found.
- while (Tok.isNot(tok::r_paren) && Tok.isNot(tok::annot_pragma_openmp_end)) {
- CXXScopeSpec SS;
- UnqualifiedId Name;
- // Read var name.
- Token PrevTok = Tok;
- NoIdentIsFound = false;
- if (AllowScopeSpecifier && getLangOpts().CPlusPlus &&
- ParseOptionalCXXScopeSpecifier(SS, nullptr, false)) {
- IsCorrect = false;
- SkipUntil(tok::comma, tok::r_paren, tok::annot_pragma_openmp_end,
- StopBeforeMatch);
- } else if (ParseUnqualifiedId(SS, false, false, false, false, nullptr,
- nullptr, Name)) {
- IsCorrect = false;
- SkipUntil(tok::comma, tok::r_paren, tok::annot_pragma_openmp_end,
- StopBeforeMatch);
- } else if (Tok.isNot(tok::comma) && Tok.isNot(tok::r_paren) &&
- Tok.isNot(tok::annot_pragma_openmp_end)) {
- IsCorrect = false;
- SkipUntil(tok::comma, tok::r_paren, tok::annot_pragma_openmp_end,
- StopBeforeMatch);
- Diag(PrevTok.getLocation(), diag::err_expected)
- << tok::identifier
- << SourceRange(PrevTok.getLocation(), PrevTokLocation);
- } else {
- Callback(SS, Actions.GetNameFromUnqualifiedId(Name));
- }
- // Consume ','.
- if (Tok.is(tok::comma)) {
- ConsumeToken();
- }
- }
- if (NoIdentIsFound) {
- Diag(Tok, diag::err_expected) << tok::identifier;
- IsCorrect = false;
- }
- // Parse ')'.
- IsCorrect = !T.consumeClose() && IsCorrect;
- return !IsCorrect;
- }
- /// \brief Parsing of OpenMP clauses.
- ///
- /// clause:
- /// if-clause | final-clause | num_threads-clause | safelen-clause |
- /// default-clause | private-clause | firstprivate-clause | shared-clause
- /// | linear-clause | aligned-clause | collapse-clause |
- /// lastprivate-clause | reduction-clause | proc_bind-clause |
- /// schedule-clause | copyin-clause | copyprivate-clause | untied-clause |
- /// mergeable-clause | flush-clause | read-clause | write-clause |
- /// update-clause | capture-clause | seq_cst-clause | device-clause |
- /// simdlen-clause | threads-clause | simd-clause | num_teams-clause |
- /// thread_limit-clause | priority-clause | grainsize-clause |
- /// nogroup-clause | num_tasks-clause | hint-clause | to-clause |
- /// from-clause | is_device_ptr-clause | task_reduction-clause |
- /// in_reduction-clause
- ///
- OMPClause *Parser::ParseOpenMPClause(OpenMPDirectiveKind DKind,
- OpenMPClauseKind CKind, bool FirstClause) {
- OMPClause *Clause = nullptr;
- bool ErrorFound = false;
- bool WrongDirective = false;
- // Check if clause is allowed for the given directive.
- if (CKind != OMPC_unknown && !isAllowedClauseForDirective(DKind, CKind)) {
- Diag(Tok, diag::err_omp_unexpected_clause) << getOpenMPClauseName(CKind)
- << getOpenMPDirectiveName(DKind);
- ErrorFound = true;
- WrongDirective = true;
- }
- switch (CKind) {
- case OMPC_final:
- case OMPC_num_threads:
- case OMPC_safelen:
- case OMPC_simdlen:
- case OMPC_collapse:
- case OMPC_ordered:
- case OMPC_device:
- case OMPC_num_teams:
- case OMPC_thread_limit:
- case OMPC_priority:
- case OMPC_grainsize:
- case OMPC_num_tasks:
- case OMPC_hint:
- // OpenMP [2.5, Restrictions]
- // At most one num_threads clause can appear on the directive.
- // OpenMP [2.8.1, simd construct, Restrictions]
- // Only one safelen clause can appear on a simd directive.
- // Only one simdlen clause can appear on a simd directive.
- // Only one collapse clause can appear on a simd directive.
- // OpenMP [2.9.1, target data construct, Restrictions]
- // At most one device clause can appear on the directive.
- // OpenMP [2.11.1, task Construct, Restrictions]
- // At most one if clause can appear on the directive.
- // At most one final clause can appear on the directive.
- // OpenMP [teams Construct, Restrictions]
- // At most one num_teams clause can appear on the directive.
- // At most one thread_limit clause can appear on the directive.
- // OpenMP [2.9.1, task Construct, Restrictions]
- // At most one priority clause can appear on the directive.
- // OpenMP [2.9.2, taskloop Construct, Restrictions]
- // At most one grainsize clause can appear on the directive.
- // OpenMP [2.9.2, taskloop Construct, Restrictions]
- // At most one num_tasks clause can appear on the directive.
- if (!FirstClause) {
- Diag(Tok, diag::err_omp_more_one_clause)
- << getOpenMPDirectiveName(DKind) << getOpenMPClauseName(CKind) << 0;
- ErrorFound = true;
- }
- if (CKind == OMPC_ordered && PP.LookAhead(/*N=*/0).isNot(tok::l_paren))
- Clause = ParseOpenMPClause(CKind, WrongDirective);
- else
- Clause = ParseOpenMPSingleExprClause(CKind, WrongDirective);
- break;
- case OMPC_default:
- case OMPC_proc_bind:
- // OpenMP [2.14.3.1, Restrictions]
- // Only a single default clause may be specified on a parallel, task or
- // teams directive.
- // OpenMP [2.5, parallel Construct, Restrictions]
- // At most one proc_bind clause can appear on the directive.
- if (!FirstClause) {
- Diag(Tok, diag::err_omp_more_one_clause)
- << getOpenMPDirectiveName(DKind) << getOpenMPClauseName(CKind) << 0;
- ErrorFound = true;
- }
- Clause = ParseOpenMPSimpleClause(CKind, WrongDirective);
- break;
- case OMPC_schedule:
- case OMPC_dist_schedule:
- case OMPC_defaultmap:
- // OpenMP [2.7.1, Restrictions, p. 3]
- // Only one schedule clause can appear on a loop directive.
- // OpenMP [2.10.4, Restrictions, p. 106]
- // At most one defaultmap clause can appear on the directive.
- if (!FirstClause) {
- Diag(Tok, diag::err_omp_more_one_clause)
- << getOpenMPDirectiveName(DKind) << getOpenMPClauseName(CKind) << 0;
- ErrorFound = true;
- }
- LLVM_FALLTHROUGH;
- case OMPC_if:
- Clause = ParseOpenMPSingleExprWithArgClause(CKind, WrongDirective);
- break;
- case OMPC_nowait:
- case OMPC_untied:
- case OMPC_mergeable:
- case OMPC_read:
- case OMPC_write:
- case OMPC_update:
- case OMPC_capture:
- case OMPC_seq_cst:
- case OMPC_threads:
- case OMPC_simd:
- case OMPC_nogroup:
- // OpenMP [2.7.1, Restrictions, p. 9]
- // Only one ordered clause can appear on a loop directive.
- // OpenMP [2.7.1, Restrictions, C/C++, p. 4]
- // Only one nowait clause can appear on a for directive.
- if (!FirstClause) {
- Diag(Tok, diag::err_omp_more_one_clause)
- << getOpenMPDirectiveName(DKind) << getOpenMPClauseName(CKind) << 0;
- ErrorFound = true;
- }
- Clause = ParseOpenMPClause(CKind, WrongDirective);
- break;
- case OMPC_private:
- case OMPC_firstprivate:
- case OMPC_lastprivate:
- case OMPC_shared:
- case OMPC_reduction:
- case OMPC_task_reduction:
- case OMPC_in_reduction:
- case OMPC_linear:
- case OMPC_aligned:
- case OMPC_copyin:
- case OMPC_copyprivate:
- case OMPC_flush:
- case OMPC_depend:
- case OMPC_map:
- case OMPC_to:
- case OMPC_from:
- case OMPC_use_device_ptr:
- case OMPC_is_device_ptr:
- Clause = ParseOpenMPVarListClause(DKind, CKind, WrongDirective);
- break;
- case OMPC_unknown:
- Diag(Tok, diag::warn_omp_extra_tokens_at_eol)
- << getOpenMPDirectiveName(DKind);
- SkipUntil(tok::annot_pragma_openmp_end, StopBeforeMatch);
- break;
- case OMPC_threadprivate:
- case OMPC_uniform:
- if (!WrongDirective)
- Diag(Tok, diag::err_omp_unexpected_clause)
- << getOpenMPClauseName(CKind) << getOpenMPDirectiveName(DKind);
- SkipUntil(tok::comma, tok::annot_pragma_openmp_end, StopBeforeMatch);
- break;
- }
- return ErrorFound ? nullptr : Clause;
- }
- /// Parses simple expression in parens for single-expression clauses of OpenMP
- /// constructs.
- /// \param RLoc Returned location of right paren.
- ExprResult Parser::ParseOpenMPParensExpr(StringRef ClauseName,
- SourceLocation &RLoc) {
- BalancedDelimiterTracker T(*this, tok::l_paren, tok::annot_pragma_openmp_end);
- if (T.expectAndConsume(diag::err_expected_lparen_after, ClauseName.data()))
- return ExprError();
- SourceLocation ELoc = Tok.getLocation();
- ExprResult LHS(ParseCastExpression(
- /*isUnaryExpression=*/false, /*isAddressOfOperand=*/false, NotTypeCast));
- ExprResult Val(ParseRHSOfBinaryExpression(LHS, prec::Conditional));
- Val = Actions.ActOnFinishFullExpr(Val.get(), ELoc);
- // Parse ')'.
- T.consumeClose();
- RLoc = T.getCloseLocation();
- return Val;
- }
- /// \brief Parsing of OpenMP clauses with single expressions like 'final',
- /// 'collapse', 'safelen', 'num_threads', 'simdlen', 'num_teams',
- /// 'thread_limit', 'simdlen', 'priority', 'grainsize', 'num_tasks' or 'hint'.
- ///
- /// final-clause:
- /// 'final' '(' expression ')'
- ///
- /// num_threads-clause:
- /// 'num_threads' '(' expression ')'
- ///
- /// safelen-clause:
- /// 'safelen' '(' expression ')'
- ///
- /// simdlen-clause:
- /// 'simdlen' '(' expression ')'
- ///
- /// collapse-clause:
- /// 'collapse' '(' expression ')'
- ///
- /// priority-clause:
- /// 'priority' '(' expression ')'
- ///
- /// grainsize-clause:
- /// 'grainsize' '(' expression ')'
- ///
- /// num_tasks-clause:
- /// 'num_tasks' '(' expression ')'
- ///
- /// hint-clause:
- /// 'hint' '(' expression ')'
- ///
- OMPClause *Parser::ParseOpenMPSingleExprClause(OpenMPClauseKind Kind,
- bool ParseOnly) {
- SourceLocation Loc = ConsumeToken();
- SourceLocation LLoc = Tok.getLocation();
- SourceLocation RLoc;
- ExprResult Val = ParseOpenMPParensExpr(getOpenMPClauseName(Kind), RLoc);
- if (Val.isInvalid())
- return nullptr;
- if (ParseOnly)
- return nullptr;
- return Actions.ActOnOpenMPSingleExprClause(Kind, Val.get(), Loc, LLoc, RLoc);
- }
- /// \brief Parsing of simple OpenMP clauses like 'default' or 'proc_bind'.
- ///
- /// default-clause:
- /// 'default' '(' 'none' | 'shared' ')
- ///
- /// proc_bind-clause:
- /// 'proc_bind' '(' 'master' | 'close' | 'spread' ')
- ///
- OMPClause *Parser::ParseOpenMPSimpleClause(OpenMPClauseKind Kind,
- bool ParseOnly) {
- SourceLocation Loc = Tok.getLocation();
- SourceLocation LOpen = ConsumeToken();
- // Parse '('.
- BalancedDelimiterTracker T(*this, tok::l_paren, tok::annot_pragma_openmp_end);
- if (T.expectAndConsume(diag::err_expected_lparen_after,
- getOpenMPClauseName(Kind)))
- return nullptr;
- unsigned Type = getOpenMPSimpleClauseType(
- Kind, Tok.isAnnotation() ? "" : PP.getSpelling(Tok));
- SourceLocation TypeLoc = Tok.getLocation();
- if (Tok.isNot(tok::r_paren) && Tok.isNot(tok::comma) &&
- Tok.isNot(tok::annot_pragma_openmp_end))
- ConsumeAnyToken();
- // Parse ')'.
- T.consumeClose();
- if (ParseOnly)
- return nullptr;
- return Actions.ActOnOpenMPSimpleClause(Kind, Type, TypeLoc, LOpen, Loc,
- Tok.getLocation());
- }
- /// \brief Parsing of OpenMP clauses like 'ordered'.
- ///
- /// ordered-clause:
- /// 'ordered'
- ///
- /// nowait-clause:
- /// 'nowait'
- ///
- /// untied-clause:
- /// 'untied'
- ///
- /// mergeable-clause:
- /// 'mergeable'
- ///
- /// read-clause:
- /// 'read'
- ///
- /// threads-clause:
- /// 'threads'
- ///
- /// simd-clause:
- /// 'simd'
- ///
- /// nogroup-clause:
- /// 'nogroup'
- ///
- OMPClause *Parser::ParseOpenMPClause(OpenMPClauseKind Kind, bool ParseOnly) {
- SourceLocation Loc = Tok.getLocation();
- ConsumeAnyToken();
- if (ParseOnly)
- return nullptr;
- return Actions.ActOnOpenMPClause(Kind, Loc, Tok.getLocation());
- }
- /// \brief Parsing of OpenMP clauses with single expressions and some additional
- /// argument like 'schedule' or 'dist_schedule'.
- ///
- /// schedule-clause:
- /// 'schedule' '(' [ modifier [ ',' modifier ] ':' ] kind [',' expression ]
- /// ')'
- ///
- /// if-clause:
- /// 'if' '(' [ directive-name-modifier ':' ] expression ')'
- ///
- /// defaultmap:
- /// 'defaultmap' '(' modifier ':' kind ')'
- ///
- OMPClause *Parser::ParseOpenMPSingleExprWithArgClause(OpenMPClauseKind Kind,
- bool ParseOnly) {
- SourceLocation Loc = ConsumeToken();
- SourceLocation DelimLoc;
- // Parse '('.
- BalancedDelimiterTracker T(*this, tok::l_paren, tok::annot_pragma_openmp_end);
- if (T.expectAndConsume(diag::err_expected_lparen_after,
- getOpenMPClauseName(Kind)))
- return nullptr;
- ExprResult Val;
- SmallVector<unsigned, 4> Arg;
- SmallVector<SourceLocation, 4> KLoc;
- if (Kind == OMPC_schedule) {
- enum { Modifier1, Modifier2, ScheduleKind, NumberOfElements };
- Arg.resize(NumberOfElements);
- KLoc.resize(NumberOfElements);
- Arg[Modifier1] = OMPC_SCHEDULE_MODIFIER_unknown;
- Arg[Modifier2] = OMPC_SCHEDULE_MODIFIER_unknown;
- Arg[ScheduleKind] = OMPC_SCHEDULE_unknown;
- unsigned KindModifier = getOpenMPSimpleClauseType(
- Kind, Tok.isAnnotation() ? "" : PP.getSpelling(Tok));
- if (KindModifier > OMPC_SCHEDULE_unknown) {
- // Parse 'modifier'
- Arg[Modifier1] = KindModifier;
- KLoc[Modifier1] = Tok.getLocation();
- if (Tok.isNot(tok::r_paren) && Tok.isNot(tok::comma) &&
- Tok.isNot(tok::annot_pragma_openmp_end))
- ConsumeAnyToken();
- if (Tok.is(tok::comma)) {
- // Parse ',' 'modifier'
- ConsumeAnyToken();
- KindModifier = getOpenMPSimpleClauseType(
- Kind, Tok.isAnnotation() ? "" : PP.getSpelling(Tok));
- Arg[Modifier2] = KindModifier > OMPC_SCHEDULE_unknown
- ? KindModifier
- : (unsigned)OMPC_SCHEDULE_unknown;
- KLoc[Modifier2] = Tok.getLocation();
- if (Tok.isNot(tok::r_paren) && Tok.isNot(tok::comma) &&
- Tok.isNot(tok::annot_pragma_openmp_end))
- ConsumeAnyToken();
- }
- // Parse ':'
- if (Tok.is(tok::colon))
- ConsumeAnyToken();
- else
- Diag(Tok, diag::warn_pragma_expected_colon) << "schedule modifier";
- KindModifier = getOpenMPSimpleClauseType(
- Kind, Tok.isAnnotation() ? "" : PP.getSpelling(Tok));
- }
- Arg[ScheduleKind] = KindModifier;
- KLoc[ScheduleKind] = Tok.getLocation();
- if (Tok.isNot(tok::r_paren) && Tok.isNot(tok::comma) &&
- Tok.isNot(tok::annot_pragma_openmp_end))
- ConsumeAnyToken();
- if ((Arg[ScheduleKind] == OMPC_SCHEDULE_static ||
- Arg[ScheduleKind] == OMPC_SCHEDULE_dynamic ||
- Arg[ScheduleKind] == OMPC_SCHEDULE_guided) &&
- Tok.is(tok::comma))
- DelimLoc = ConsumeAnyToken();
- } else if (Kind == OMPC_dist_schedule) {
- Arg.push_back(getOpenMPSimpleClauseType(
- Kind, Tok.isAnnotation() ? "" : PP.getSpelling(Tok)));
- KLoc.push_back(Tok.getLocation());
- if (Tok.isNot(tok::r_paren) && Tok.isNot(tok::comma) &&
- Tok.isNot(tok::annot_pragma_openmp_end))
- ConsumeAnyToken();
- if (Arg.back() == OMPC_DIST_SCHEDULE_static && Tok.is(tok::comma))
- DelimLoc = ConsumeAnyToken();
- } else if (Kind == OMPC_defaultmap) {
- // Get a defaultmap modifier
- Arg.push_back(getOpenMPSimpleClauseType(
- Kind, Tok.isAnnotation() ? "" : PP.getSpelling(Tok)));
- KLoc.push_back(Tok.getLocation());
- if (Tok.isNot(tok::r_paren) && Tok.isNot(tok::comma) &&
- Tok.isNot(tok::annot_pragma_openmp_end))
- ConsumeAnyToken();
- // Parse ':'
- if (Tok.is(tok::colon))
- ConsumeAnyToken();
- else if (Arg.back() != OMPC_DEFAULTMAP_MODIFIER_unknown)
- Diag(Tok, diag::warn_pragma_expected_colon) << "defaultmap modifier";
- // Get a defaultmap kind
- Arg.push_back(getOpenMPSimpleClauseType(
- Kind, Tok.isAnnotation() ? "" : PP.getSpelling(Tok)));
- KLoc.push_back(Tok.getLocation());
- if (Tok.isNot(tok::r_paren) && Tok.isNot(tok::comma) &&
- Tok.isNot(tok::annot_pragma_openmp_end))
- ConsumeAnyToken();
- } else {
- assert(Kind == OMPC_if);
- KLoc.push_back(Tok.getLocation());
- TentativeParsingAction TPA(*this);
- Arg.push_back(parseOpenMPDirectiveKind(*this));
- if (Arg.back() != OMPD_unknown) {
- ConsumeToken();
- if (Tok.is(tok::colon) && getLangOpts().OpenMP > 40) {
- TPA.Commit();
- DelimLoc = ConsumeToken();
- } else {
- TPA.Revert();
- Arg.back() = OMPD_unknown;
- }
- } else {
- TPA.Revert();
- }
- }
- bool NeedAnExpression = (Kind == OMPC_schedule && DelimLoc.isValid()) ||
- (Kind == OMPC_dist_schedule && DelimLoc.isValid()) ||
- Kind == OMPC_if;
- if (NeedAnExpression) {
- SourceLocation ELoc = Tok.getLocation();
- ExprResult LHS(ParseCastExpression(false, false, NotTypeCast));
- Val = ParseRHSOfBinaryExpression(LHS, prec::Conditional);
- Val = Actions.ActOnFinishFullExpr(Val.get(), ELoc);
- }
- // Parse ')'.
- T.consumeClose();
- if (NeedAnExpression && Val.isInvalid())
- return nullptr;
- if (ParseOnly)
- return nullptr;
- return Actions.ActOnOpenMPSingleExprWithArgClause(
- Kind, Arg, Val.get(), Loc, T.getOpenLocation(), KLoc, DelimLoc,
- T.getCloseLocation());
- }
- static bool ParseReductionId(Parser &P, CXXScopeSpec &ReductionIdScopeSpec,
- UnqualifiedId &ReductionId) {
- if (ReductionIdScopeSpec.isEmpty()) {
- auto OOK = OO_None;
- switch (P.getCurToken().getKind()) {
- case tok::plus:
- OOK = OO_Plus;
- break;
- case tok::minus:
- OOK = OO_Minus;
- break;
- case tok::star:
- OOK = OO_Star;
- break;
- case tok::amp:
- OOK = OO_Amp;
- break;
- case tok::pipe:
- OOK = OO_Pipe;
- break;
- case tok::caret:
- OOK = OO_Caret;
- break;
- case tok::ampamp:
- OOK = OO_AmpAmp;
- break;
- case tok::pipepipe:
- OOK = OO_PipePipe;
- break;
- default:
- break;
- }
- if (OOK != OO_None) {
- SourceLocation OpLoc = P.ConsumeToken();
- SourceLocation SymbolLocations[] = {OpLoc, OpLoc, SourceLocation()};
- ReductionId.setOperatorFunctionId(OpLoc, OOK, SymbolLocations);
- return false;
- }
- }
- return P.ParseUnqualifiedId(ReductionIdScopeSpec, /*EnteringContext*/ false,
- /*AllowDestructorName*/ false,
- /*AllowConstructorName*/ false,
- /*AllowDeductionGuide*/ false,
- nullptr, nullptr, ReductionId);
- }
- /// Parses clauses with list.
- bool Parser::ParseOpenMPVarList(OpenMPDirectiveKind DKind,
- OpenMPClauseKind Kind,
- SmallVectorImpl<Expr *> &Vars,
- OpenMPVarListDataTy &Data) {
- UnqualifiedId UnqualifiedReductionId;
- bool InvalidReductionId = false;
- bool MapTypeModifierSpecified = false;
- // Parse '('.
- BalancedDelimiterTracker T(*this, tok::l_paren, tok::annot_pragma_openmp_end);
- if (T.expectAndConsume(diag::err_expected_lparen_after,
- getOpenMPClauseName(Kind)))
- return true;
- bool NeedRParenForLinear = false;
- BalancedDelimiterTracker LinearT(*this, tok::l_paren,
- tok::annot_pragma_openmp_end);
- // Handle reduction-identifier for reduction clause.
- if (Kind == OMPC_reduction || Kind == OMPC_task_reduction ||
- Kind == OMPC_in_reduction) {
- ColonProtectionRAIIObject ColonRAII(*this);
- if (getLangOpts().CPlusPlus)
- ParseOptionalCXXScopeSpecifier(Data.ReductionIdScopeSpec,
- /*ObjectType=*/nullptr,
- /*EnteringContext=*/false);
- InvalidReductionId = ParseReductionId(*this, Data.ReductionIdScopeSpec,
- UnqualifiedReductionId);
- if (InvalidReductionId) {
- SkipUntil(tok::colon, tok::r_paren, tok::annot_pragma_openmp_end,
- StopBeforeMatch);
- }
- if (Tok.is(tok::colon))
- Data.ColonLoc = ConsumeToken();
- else
- Diag(Tok, diag::warn_pragma_expected_colon) << "reduction identifier";
- if (!InvalidReductionId)
- Data.ReductionId =
- Actions.GetNameFromUnqualifiedId(UnqualifiedReductionId);
- } else if (Kind == OMPC_depend) {
- // Handle dependency type for depend clause.
- ColonProtectionRAIIObject ColonRAII(*this);
- Data.DepKind =
- static_cast<OpenMPDependClauseKind>(getOpenMPSimpleClauseType(
- Kind, Tok.is(tok::identifier) ? PP.getSpelling(Tok) : ""));
- Data.DepLinMapLoc = Tok.getLocation();
- if (Data.DepKind == OMPC_DEPEND_unknown) {
- SkipUntil(tok::colon, tok::r_paren, tok::annot_pragma_openmp_end,
- StopBeforeMatch);
- } else {
- ConsumeToken();
- // Special processing for depend(source) clause.
- if (DKind == OMPD_ordered && Data.DepKind == OMPC_DEPEND_source) {
- // Parse ')'.
- T.consumeClose();
- return false;
- }
- }
- if (Tok.is(tok::colon)) {
- Data.ColonLoc = ConsumeToken();
- } else {
- Diag(Tok, DKind == OMPD_ordered ? diag::warn_pragma_expected_colon_r_paren
- : diag::warn_pragma_expected_colon)
- << "dependency type";
- }
- } else if (Kind == OMPC_linear) {
- // Try to parse modifier if any.
- if (Tok.is(tok::identifier) && PP.LookAhead(0).is(tok::l_paren)) {
- Data.LinKind = static_cast<OpenMPLinearClauseKind>(
- getOpenMPSimpleClauseType(Kind, PP.getSpelling(Tok)));
- Data.DepLinMapLoc = ConsumeToken();
- LinearT.consumeOpen();
- NeedRParenForLinear = true;
- }
- } else if (Kind == OMPC_map) {
- // Handle map type for map clause.
- ColonProtectionRAIIObject ColonRAII(*this);
- /// The map clause modifier token can be either a identifier or the C++
- /// delete keyword.
- auto &&IsMapClauseModifierToken = [](const Token &Tok) -> bool {
- return Tok.isOneOf(tok::identifier, tok::kw_delete);
- };
- // The first identifier may be a list item, a map-type or a
- // map-type-modifier. The map modifier can also be delete which has the same
- // spelling of the C++ delete keyword.
- Data.MapType =
- IsMapClauseModifierToken(Tok)
- ? static_cast<OpenMPMapClauseKind>(
- getOpenMPSimpleClauseType(Kind, PP.getSpelling(Tok)))
- : OMPC_MAP_unknown;
- Data.DepLinMapLoc = Tok.getLocation();
- bool ColonExpected = false;
- if (IsMapClauseModifierToken(Tok)) {
- if (PP.LookAhead(0).is(tok::colon)) {
- if (Data.MapType == OMPC_MAP_unknown)
- Diag(Tok, diag::err_omp_unknown_map_type);
- else if (Data.MapType == OMPC_MAP_always)
- Diag(Tok, diag::err_omp_map_type_missing);
- ConsumeToken();
- } else if (PP.LookAhead(0).is(tok::comma)) {
- if (IsMapClauseModifierToken(PP.LookAhead(1)) &&
- PP.LookAhead(2).is(tok::colon)) {
- Data.MapTypeModifier = Data.MapType;
- if (Data.MapTypeModifier != OMPC_MAP_always) {
- Diag(Tok, diag::err_omp_unknown_map_type_modifier);
- Data.MapTypeModifier = OMPC_MAP_unknown;
- } else {
- MapTypeModifierSpecified = true;
- }
- ConsumeToken();
- ConsumeToken();
- Data.MapType =
- IsMapClauseModifierToken(Tok)
- ? static_cast<OpenMPMapClauseKind>(
- getOpenMPSimpleClauseType(Kind, PP.getSpelling(Tok)))
- : OMPC_MAP_unknown;
- if (Data.MapType == OMPC_MAP_unknown ||
- Data.MapType == OMPC_MAP_always)
- Diag(Tok, diag::err_omp_unknown_map_type);
- ConsumeToken();
- } else {
- Data.MapType = OMPC_MAP_tofrom;
- Data.IsMapTypeImplicit = true;
- }
- } else if (IsMapClauseModifierToken(PP.LookAhead(0))) {
- if (PP.LookAhead(1).is(tok::colon)) {
- Data.MapTypeModifier = Data.MapType;
- if (Data.MapTypeModifier != OMPC_MAP_always) {
- Diag(Tok, diag::err_omp_unknown_map_type_modifier);
- Data.MapTypeModifier = OMPC_MAP_unknown;
- } else {
- MapTypeModifierSpecified = true;
- }
- ConsumeToken();
- Data.MapType =
- IsMapClauseModifierToken(Tok)
- ? static_cast<OpenMPMapClauseKind>(
- getOpenMPSimpleClauseType(Kind, PP.getSpelling(Tok)))
- : OMPC_MAP_unknown;
- if (Data.MapType == OMPC_MAP_unknown ||
- Data.MapType == OMPC_MAP_always)
- Diag(Tok, diag::err_omp_unknown_map_type);
- ConsumeToken();
- } else {
- Data.MapType = OMPC_MAP_tofrom;
- Data.IsMapTypeImplicit = true;
- }
- } else {
- Data.MapType = OMPC_MAP_tofrom;
- Data.IsMapTypeImplicit = true;
- }
- } else {
- Data.MapType = OMPC_MAP_tofrom;
- Data.IsMapTypeImplicit = true;
- }
- if (Tok.is(tok::colon))
- Data.ColonLoc = ConsumeToken();
- else if (ColonExpected)
- Diag(Tok, diag::warn_pragma_expected_colon) << "map type";
- }
- bool IsComma =
- (Kind != OMPC_reduction && Kind != OMPC_task_reduction &&
- Kind != OMPC_in_reduction && Kind != OMPC_depend && Kind != OMPC_map) ||
- (Kind == OMPC_reduction && !InvalidReductionId) ||
- (Kind == OMPC_map && Data.MapType != OMPC_MAP_unknown &&
- (!MapTypeModifierSpecified ||
- Data.MapTypeModifier == OMPC_MAP_always)) ||
- (Kind == OMPC_depend && Data.DepKind != OMPC_DEPEND_unknown);
- const bool MayHaveTail = (Kind == OMPC_linear || Kind == OMPC_aligned);
- while (IsComma || (Tok.isNot(tok::r_paren) && Tok.isNot(tok::colon) &&
- Tok.isNot(tok::annot_pragma_openmp_end))) {
- ColonProtectionRAIIObject ColonRAII(*this, MayHaveTail);
- // Parse variable
- ExprResult VarExpr =
- Actions.CorrectDelayedTyposInExpr(ParseAssignmentExpression());
- if (VarExpr.isUsable()) {
- Vars.push_back(VarExpr.get());
- } else {
- SkipUntil(tok::comma, tok::r_paren, tok::annot_pragma_openmp_end,
- StopBeforeMatch);
- }
- // Skip ',' if any
- IsComma = Tok.is(tok::comma);
- if (IsComma)
- ConsumeToken();
- else if (Tok.isNot(tok::r_paren) &&
- Tok.isNot(tok::annot_pragma_openmp_end) &&
- (!MayHaveTail || Tok.isNot(tok::colon)))
- Diag(Tok, diag::err_omp_expected_punc)
- << ((Kind == OMPC_flush) ? getOpenMPDirectiveName(OMPD_flush)
- : getOpenMPClauseName(Kind))
- << (Kind == OMPC_flush);
- }
- // Parse ')' for linear clause with modifier.
- if (NeedRParenForLinear)
- LinearT.consumeClose();
- // Parse ':' linear-step (or ':' alignment).
- const bool MustHaveTail = MayHaveTail && Tok.is(tok::colon);
- if (MustHaveTail) {
- Data.ColonLoc = Tok.getLocation();
- SourceLocation ELoc = ConsumeToken();
- ExprResult Tail = ParseAssignmentExpression();
- Tail = Actions.ActOnFinishFullExpr(Tail.get(), ELoc);
- if (Tail.isUsable())
- Data.TailExpr = Tail.get();
- else
- SkipUntil(tok::comma, tok::r_paren, tok::annot_pragma_openmp_end,
- StopBeforeMatch);
- }
- // Parse ')'.
- T.consumeClose();
- return (Kind == OMPC_depend && Data.DepKind != OMPC_DEPEND_unknown &&
- Vars.empty()) ||
- (Kind != OMPC_depend && Kind != OMPC_map && Vars.empty()) ||
- (MustHaveTail && !Data.TailExpr) || InvalidReductionId;
- }
- /// \brief Parsing of OpenMP clause 'private', 'firstprivate', 'lastprivate',
- /// 'shared', 'copyin', 'copyprivate', 'flush', 'reduction', 'task_reduction' or
- /// 'in_reduction'.
- ///
- /// private-clause:
- /// 'private' '(' list ')'
- /// firstprivate-clause:
- /// 'firstprivate' '(' list ')'
- /// lastprivate-clause:
- /// 'lastprivate' '(' list ')'
- /// shared-clause:
- /// 'shared' '(' list ')'
- /// linear-clause:
- /// 'linear' '(' linear-list [ ':' linear-step ] ')'
- /// aligned-clause:
- /// 'aligned' '(' list [ ':' alignment ] ')'
- /// reduction-clause:
- /// 'reduction' '(' reduction-identifier ':' list ')'
- /// task_reduction-clause:
- /// 'task_reduction' '(' reduction-identifier ':' list ')'
- /// in_reduction-clause:
- /// 'in_reduction' '(' reduction-identifier ':' list ')'
- /// copyprivate-clause:
- /// 'copyprivate' '(' list ')'
- /// flush-clause:
- /// 'flush' '(' list ')'
- /// depend-clause:
- /// 'depend' '(' in | out | inout : list | source ')'
- /// map-clause:
- /// 'map' '(' [ [ always , ]
- /// to | from | tofrom | alloc | release | delete ':' ] list ')';
- /// to-clause:
- /// 'to' '(' list ')'
- /// from-clause:
- /// 'from' '(' list ')'
- /// use_device_ptr-clause:
- /// 'use_device_ptr' '(' list ')'
- /// is_device_ptr-clause:
- /// 'is_device_ptr' '(' list ')'
- ///
- /// For 'linear' clause linear-list may have the following forms:
- /// list
- /// modifier(list)
- /// where modifier is 'val' (C) or 'ref', 'val' or 'uval'(C++).
- OMPClause *Parser::ParseOpenMPVarListClause(OpenMPDirectiveKind DKind,
- OpenMPClauseKind Kind,
- bool ParseOnly) {
- SourceLocation Loc = Tok.getLocation();
- SourceLocation LOpen = ConsumeToken();
- SmallVector<Expr *, 4> Vars;
- OpenMPVarListDataTy Data;
- if (ParseOpenMPVarList(DKind, Kind, Vars, Data))
- return nullptr;
- if (ParseOnly)
- return nullptr;
- return Actions.ActOnOpenMPVarListClause(
- Kind, Vars, Data.TailExpr, Loc, LOpen, Data.ColonLoc, Tok.getLocation(),
- Data.ReductionIdScopeSpec, Data.ReductionId, Data.DepKind, Data.LinKind,
- Data.MapTypeModifier, Data.MapType, Data.IsMapTypeImplicit,
- Data.DepLinMapLoc);
- }
|