1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463346434653466346734683469347034713472347334743475347634773478347934803481348234833484348534863487348834893490349134923493349434953496349734983499350035013502350335043505350635073508350935103511351235133514351535163517351835193520352135223523352435253526352735283529353035313532353335343535353635373538353935403541354235433544354535463547354835493550355135523553355435553556355735583559356035613562356335643565356635673568356935703571357235733574357535763577357835793580358135823583358435853586358735883589359035913592359335943595359635973598359936003601360236033604360536063607360836093610361136123613361436153616361736183619362036213622362336243625362636273628362936303631363236333634363536363637363836393640364136423643364436453646364736483649365036513652365336543655365636573658365936603661366236633664366536663667366836693670367136723673367436753676367736783679368036813682368336843685368636873688368936903691369236933694369536963697369836993700370137023703370437053706370737083709371037113712371337143715371637173718371937203721372237233724372537263727372837293730373137323733373437353736373737383739374037413742374337443745374637473748374937503751375237533754375537563757375837593760376137623763376437653766376737683769377037713772377337743775377637773778377937803781378237833784378537863787378837893790379137923793379437953796379737983799380038013802380338043805380638073808380938103811381238133814381538163817381838193820382138223823382438253826382738283829383038313832383338343835383638373838383938403841384238433844384538463847384838493850385138523853385438553856385738583859386038613862386338643865386638673868386938703871387238733874387538763877387838793880388138823883388438853886388738883889389038913892389338943895389638973898389939003901390239033904390539063907390839093910391139123913391439153916391739183919392039213922392339243925392639273928392939303931393239333934393539363937393839393940394139423943394439453946394739483949395039513952395339543955395639573958395939603961396239633964396539663967396839693970397139723973397439753976397739783979398039813982398339843985398639873988398939903991399239933994399539963997399839994000400140024003400440054006400740084009401040114012401340144015401640174018401940204021402240234024402540264027402840294030403140324033403440354036403740384039404040414042404340444045404640474048404940504051405240534054405540564057405840594060406140624063406440654066406740684069407040714072407340744075407640774078407940804081408240834084408540864087408840894090409140924093409440954096409740984099410041014102410341044105410641074108410941104111411241134114411541164117411841194120412141224123412441254126412741284129413041314132413341344135413641374138413941404141414241434144414541464147414841494150415141524153415441554156415741584159416041614162416341644165416641674168416941704171417241734174417541764177417841794180418141824183418441854186418741884189419041914192419341944195419641974198419942004201420242034204420542064207420842094210421142124213421442154216421742184219422042214222422342244225422642274228422942304231423242334234423542364237423842394240424142424243424442454246424742484249425042514252425342544255425642574258425942604261426242634264426542664267426842694270427142724273427442754276427742784279428042814282428342844285428642874288428942904291429242934294429542964297429842994300430143024303430443054306430743084309431043114312431343144315431643174318431943204321432243234324432543264327432843294330433143324333433443354336433743384339434043414342434343444345434643474348434943504351435243534354435543564357435843594360436143624363436443654366436743684369437043714372437343744375437643774378437943804381438243834384438543864387438843894390439143924393439443954396439743984399440044014402440344044405440644074408440944104411441244134414441544164417441844194420442144224423442444254426442744284429443044314432443344344435443644374438443944404441444244434444444544464447444844494450445144524453445444554456445744584459446044614462446344644465446644674468446944704471447244734474447544764477447844794480448144824483448444854486448744884489449044914492449344944495449644974498449945004501450245034504450545064507450845094510451145124513451445154516451745184519452045214522452345244525452645274528452945304531453245334534453545364537453845394540454145424543454445454546454745484549455045514552455345544555455645574558455945604561456245634564456545664567456845694570457145724573457445754576457745784579458045814582458345844585458645874588458945904591459245934594459545964597459845994600460146024603460446054606460746084609461046114612461346144615461646174618461946204621462246234624462546264627462846294630463146324633463446354636463746384639464046414642464346444645464646474648464946504651465246534654465546564657465846594660466146624663466446654666466746684669467046714672467346744675467646774678467946804681468246834684468546864687468846894690469146924693469446954696469746984699470047014702470347044705470647074708470947104711471247134714471547164717471847194720 |
- //===--- ParseOpenMP.cpp - OpenMP directives parsing ----------------------===//
- //
- // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
- // See https://llvm.org/LICENSE.txt for license information.
- // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
- //
- //===----------------------------------------------------------------------===//
- /// \file
- /// This file implements parsing of all OpenMP directives and clauses.
- ///
- //===----------------------------------------------------------------------===//
- #include "clang/AST/ASTContext.h"
- #include "clang/AST/OpenMPClause.h"
- #include "clang/AST/StmtOpenMP.h"
- #include "clang/Basic/OpenMPKinds.h"
- #include "clang/Basic/TargetInfo.h"
- #include "clang/Basic/TokenKinds.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"
- #include "llvm/ADT/StringSwitch.h"
- #include "llvm/ADT/UniqueVector.h"
- #include "llvm/Frontend/OpenMP/OMPAssume.h"
- #include "llvm/Frontend/OpenMP/OMPContext.h"
- #include <optional>
- using namespace clang;
- using namespace llvm::omp;
- //===----------------------------------------------------------------------===//
- // OpenMP declarative directives.
- //===----------------------------------------------------------------------===//
- namespace {
- enum OpenMPDirectiveKindEx {
- OMPD_cancellation = llvm::omp::Directive_enumSize + 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,
- OMPD_mapper,
- OMPD_variant,
- OMPD_begin,
- OMPD_begin_declare,
- };
- // Helper to unify the enum class OpenMPDirectiveKind with its extension
- // the OpenMPDirectiveKindEx enum which allows to use them together as if they
- // are unsigned values.
- struct OpenMPDirectiveKindExWrapper {
- OpenMPDirectiveKindExWrapper(unsigned Value) : Value(Value) {}
- OpenMPDirectiveKindExWrapper(OpenMPDirectiveKind DK) : Value(unsigned(DK)) {}
- bool operator==(OpenMPDirectiveKindExWrapper V) const {
- return Value == V.Value;
- }
- bool operator!=(OpenMPDirectiveKindExWrapper V) const {
- return Value != V.Value;
- }
- bool operator==(OpenMPDirectiveKind V) const { return Value == unsigned(V); }
- bool operator!=(OpenMPDirectiveKind V) const { return Value != unsigned(V); }
- bool operator<(OpenMPDirectiveKind V) const { return Value < unsigned(V); }
- operator unsigned() const { return Value; }
- operator OpenMPDirectiveKind() const { return OpenMPDirectiveKind(Value); }
- unsigned Value;
- };
- class DeclDirectiveListParserHelper final {
- SmallVector<Expr *, 4> Identifiers;
- Parser *P;
- OpenMPDirectiveKind Kind;
- public:
- DeclDirectiveListParserHelper(Parser *P, OpenMPDirectiveKind Kind)
- : P(P), Kind(Kind) {}
- void operator()(CXXScopeSpec &SS, DeclarationNameInfo NameInfo) {
- ExprResult Res = P->getActions().ActOnOpenMPIdExpression(
- P->getCurScope(), SS, NameInfo, Kind);
- 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) {
- OpenMPDirectiveKindExWrapper DKind = getOpenMPDirectiveKind(S);
- if (DKind != OMPD_unknown)
- return DKind;
- return llvm::StringSwitch<OpenMPDirectiveKindExWrapper>(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)
- .Case("mapper", OMPD_mapper)
- .Case("variant", OMPD_variant)
- .Case("begin", OMPD_begin)
- .Default(OMPD_unknown);
- }
- static OpenMPDirectiveKindExWrapper 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 OpenMPDirectiveKindExWrapper F[][3] = {
- {OMPD_begin, OMPD_declare, OMPD_begin_declare},
- {OMPD_begin, OMPD_assumes, OMPD_begin_assumes},
- {OMPD_end, OMPD_declare, OMPD_end_declare},
- {OMPD_end, OMPD_assumes, OMPD_end_assumes},
- {OMPD_cancellation, OMPD_point, OMPD_cancellation_point},
- {OMPD_declare, OMPD_reduction, OMPD_declare_reduction},
- {OMPD_declare, OMPD_mapper, OMPD_declare_mapper},
- {OMPD_declare, OMPD_simd, OMPD_declare_simd},
- {OMPD_declare, OMPD_target, OMPD_declare_target},
- {OMPD_declare, OMPD_variant, OMPD_declare_variant},
- {OMPD_begin_declare, OMPD_target, OMPD_begin_declare_target},
- {OMPD_begin_declare, OMPD_variant, OMPD_begin_declare_variant},
- {OMPD_end_declare, OMPD_variant, OMPD_end_declare_variant},
- {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_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_loop, OMPD_parallel_loop},
- {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_loop, OMPD_target_parallel_loop},
- {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_teams, OMPD_loop, OMPD_teams_loop},
- {OMPD_target, OMPD_teams, OMPD_target_teams},
- {OMPD_target_teams, OMPD_distribute, OMPD_target_teams_distribute},
- {OMPD_target_teams, OMPD_loop, OMPD_target_teams_loop},
- {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},
- {OMPD_master, OMPD_taskloop, OMPD_master_taskloop},
- {OMPD_masked, OMPD_taskloop, OMPD_masked_taskloop},
- {OMPD_master_taskloop, OMPD_simd, OMPD_master_taskloop_simd},
- {OMPD_masked_taskloop, OMPD_simd, OMPD_masked_taskloop_simd},
- {OMPD_parallel, OMPD_master, OMPD_parallel_master},
- {OMPD_parallel, OMPD_masked, OMPD_parallel_masked},
- {OMPD_parallel_master, OMPD_taskloop, OMPD_parallel_master_taskloop},
- {OMPD_parallel_masked, OMPD_taskloop, OMPD_parallel_masked_taskloop},
- {OMPD_parallel_master_taskloop, OMPD_simd,
- OMPD_parallel_master_taskloop_simd},
- {OMPD_parallel_masked_taskloop, OMPD_simd,
- OMPD_parallel_masked_taskloop_simd}};
- enum { CancellationPoint = 0, DeclareReduction = 1, TargetData = 2 };
- Token Tok = P.getCurToken();
- OpenMPDirectiveKindExWrapper DKind =
- Tok.isAnnotation()
- ? static_cast<unsigned>(OMPD_unknown)
- : getOpenMPDirectiveKindEx(P.getPreprocessor().getSpelling(Tok));
- if (DKind == OMPD_unknown)
- return OMPD_unknown;
- for (const auto &I : F) {
- if (DKind != I[0])
- continue;
- Tok = P.getPreprocessor().LookAhead(0);
- OpenMPDirectiveKindExWrapper SDKind =
- Tok.isAnnotation()
- ? static_cast<unsigned>(OMPD_unknown)
- : getOpenMPDirectiveKindEx(P.getPreprocessor().getSpelling(Tok));
- if (SDKind == OMPD_unknown)
- continue;
- if (SDKind == I[1]) {
- P.ConsumeToken();
- DKind = I[2];
- }
- }
- return unsigned(DKind) < llvm::omp::Directive_enumSize
- ? 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;
- [[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);
- }
- /// 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).data())) {
- 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::Prototype, 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(
- ParseExpression().get(), D->getLocation(), /*DiscardedValue*/ false);
- 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*/ false);
- }
- 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)) {
- cutOffParsing();
- Actions.CodeCompleteInitializer(getCurScope(), OmpPrivParm);
- Actions.FinalizeDeclaration(OmpPrivParm);
- return;
- }
- PreferredType.enterVariableInit(Tok.getLocation(), OmpPrivParm);
- 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;
- SourceLocation LParLoc = T.getOpenLocation();
- auto RunSignatureHelp = [this, OmpPrivParm, LParLoc, &Exprs]() {
- QualType PreferredType = Actions.ProduceConstructorSignatureHelp(
- OmpPrivParm->getType()->getCanonicalTypeInternal(),
- OmpPrivParm->getLocation(), Exprs, LParLoc, /*Braced=*/false);
- CalledSignatureHelp = true;
- return PreferredType;
- };
- if (ParseExpressionList(Exprs, [&] {
- PreferredType.enterFunctionArgument(Tok.getLocation(),
- RunSignatureHelp);
- })) {
- if (PP.isCodeCompletionReached() && !CalledSignatureHelp)
- RunSignatureHelp();
- Actions.ActOnInitializerError(OmpPrivParm);
- SkipUntil(tok::r_paren, tok::annot_pragma_openmp_end, StopBeforeMatch);
- } else {
- // Match the ')'.
- SourceLocation RLoc = Tok.getLocation();
- if (!T.consumeClose())
- RLoc = T.getCloseLocation();
- ExprResult Initializer =
- Actions.ActOnParenListExpr(T.getOpenLocation(), RLoc, 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);
- }
- }
- /// Parses 'omp declare mapper' directive.
- ///
- /// declare-mapper-directive:
- /// annot_pragma_openmp 'declare' 'mapper' '(' [<mapper-identifier> ':']
- /// <type> <var> ')' [<clause>[[,] <clause>] ... ]
- /// annot_pragma_openmp_end
- /// <mapper-identifier> and <var> are base language identifiers.
- ///
- Parser::DeclGroupPtrTy
- Parser::ParseOpenMPDeclareMapperDirective(AccessSpecifier AS) {
- bool IsCorrect = true;
- // Parse '('
- BalancedDelimiterTracker T(*this, tok::l_paren, tok::annot_pragma_openmp_end);
- if (T.expectAndConsume(diag::err_expected_lparen_after,
- getOpenMPDirectiveName(OMPD_declare_mapper).data())) {
- SkipUntil(tok::annot_pragma_openmp_end, StopBeforeMatch);
- return DeclGroupPtrTy();
- }
- // Parse <mapper-identifier>
- auto &DeclNames = Actions.getASTContext().DeclarationNames;
- DeclarationName MapperId;
- if (PP.LookAhead(0).is(tok::colon)) {
- if (Tok.isNot(tok::identifier) && Tok.isNot(tok::kw_default)) {
- Diag(Tok.getLocation(), diag::err_omp_mapper_illegal_identifier);
- IsCorrect = false;
- } else {
- MapperId = DeclNames.getIdentifier(Tok.getIdentifierInfo());
- }
- ConsumeToken();
- // Consume ':'.
- ExpectAndConsume(tok::colon);
- } else {
- // If no mapper identifier is provided, its name is "default" by default
- MapperId =
- DeclNames.getIdentifier(&Actions.getASTContext().Idents.get("default"));
- }
- if (!IsCorrect && Tok.is(tok::annot_pragma_openmp_end))
- return DeclGroupPtrTy();
- // Parse <type> <var>
- DeclarationName VName;
- QualType MapperType;
- SourceRange Range;
- TypeResult ParsedType = parseOpenMPDeclareMapperVarDecl(Range, VName, AS);
- if (ParsedType.isUsable())
- MapperType =
- Actions.ActOnOpenMPDeclareMapperType(Range.getBegin(), ParsedType);
- if (MapperType.isNull())
- IsCorrect = false;
- if (!IsCorrect) {
- SkipUntil(tok::annot_pragma_openmp_end, Parser::StopBeforeMatch);
- return DeclGroupPtrTy();
- }
- // Consume ')'.
- IsCorrect &= !T.consumeClose();
- if (!IsCorrect) {
- SkipUntil(tok::annot_pragma_openmp_end, Parser::StopBeforeMatch);
- return DeclGroupPtrTy();
- }
- // Enter scope.
- DeclarationNameInfo DirName;
- SourceLocation Loc = Tok.getLocation();
- unsigned ScopeFlags = Scope::FnScope | Scope::DeclScope |
- Scope::CompoundStmtScope | Scope::OpenMPDirectiveScope;
- ParseScope OMPDirectiveScope(this, ScopeFlags);
- Actions.StartOpenMPDSABlock(OMPD_declare_mapper, DirName, getCurScope(), Loc);
- // Add the mapper variable declaration.
- ExprResult MapperVarRef = Actions.ActOnOpenMPDeclareMapperDirectiveVarDecl(
- getCurScope(), MapperType, Range.getBegin(), VName);
- // Parse map clauses.
- SmallVector<OMPClause *, 6> Clauses;
- while (Tok.isNot(tok::annot_pragma_openmp_end)) {
- OpenMPClauseKind CKind = Tok.isAnnotation()
- ? OMPC_unknown
- : getOpenMPClauseKind(PP.getSpelling(Tok));
- Actions.StartOpenMPClause(CKind);
- OMPClause *Clause =
- ParseOpenMPClause(OMPD_declare_mapper, CKind, Clauses.empty());
- if (Clause)
- Clauses.push_back(Clause);
- else
- IsCorrect = false;
- // Skip ',' if any.
- if (Tok.is(tok::comma))
- ConsumeToken();
- Actions.EndOpenMPClause();
- }
- if (Clauses.empty()) {
- Diag(Tok, diag::err_omp_expected_clause)
- << getOpenMPDirectiveName(OMPD_declare_mapper);
- IsCorrect = false;
- }
- // Exit scope.
- Actions.EndOpenMPDSABlock(nullptr);
- OMPDirectiveScope.Exit();
- DeclGroupPtrTy DG = Actions.ActOnOpenMPDeclareMapperDirective(
- getCurScope(), Actions.getCurLexicalContext(), MapperId, MapperType,
- Range.getBegin(), VName, AS, MapperVarRef.get(), Clauses);
- if (!IsCorrect)
- return DeclGroupPtrTy();
- return DG;
- }
- TypeResult Parser::parseOpenMPDeclareMapperVarDecl(SourceRange &Range,
- DeclarationName &Name,
- AccessSpecifier AS) {
- // Parse the common declaration-specifiers piece.
- Parser::DeclSpecContext DSC = Parser::DeclSpecContext::DSC_type_specifier;
- DeclSpec DS(AttrFactory);
- ParseSpecifierQualifierList(DS, AS, DSC);
- // Parse the declarator.
- DeclaratorContext Context = DeclaratorContext::Prototype;
- Declarator DeclaratorInfo(DS, ParsedAttributesView::none(), Context);
- ParseDeclarator(DeclaratorInfo);
- Range = DeclaratorInfo.getSourceRange();
- if (DeclaratorInfo.getIdentifier() == nullptr) {
- Diag(Tok.getLocation(), diag::err_omp_mapper_expected_declarator);
- return true;
- }
- Name = Actions.GetNameForDeclarator(DeclaratorInfo).getName();
- return Actions.ActOnOpenMPDeclareMapperVarDecl(getCurScope(), DeclaratorInfo);
- }
- 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::MultiParseScope Scopes;
- bool HasFunScope = false;
- FNContextRAII() = delete;
- FNContextRAII(const FNContextRAII &) = delete;
- FNContextRAII &operator=(const FNContextRAII &) = delete;
- public:
- FNContextRAII(Parser &P, Parser::DeclGroupPtrTy Ptr) : P(P), Scopes(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, Qualifiers(),
- ND && ND->isCXXInstanceMember());
- // If the Decl is templatized, add template parameters to scope.
- // FIXME: Track CurTemplateDepth?
- P.ReenterTemplateScopes(Scopes, D);
- // If the Decl is on a function, add function parameters to the scope.
- if (D->isFunctionOrFunctionTemplate()) {
- HasFunScope = true;
- Scopes.Enter(Scope::FnScope | Scope::DeclScope |
- Scope::CompoundStmtScope);
- Actions.ActOnReenterFunctionContext(Actions.getCurScope(), D);
- }
- }
- ~FNContextRAII() {
- if (HasFunScope)
- P.getActions().ActOnExitFunctionContext();
- 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) {
- Sema::OpenMPVarListDataTy Data;
- SmallVectorImpl<Expr *> *Vars = &Uniforms;
- if (CKind == OMPC_aligned) {
- Vars = &Aligneds;
- } else if (CKind == OMPC_linear) {
- Data.ExtraModifier = OMPC_LINEAR_val;
- 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.DepModOrTailExpr);
- } else if (CKind == OMPC_linear) {
- assert(0 <= Data.ExtraModifier &&
- Data.ExtraModifier <= OMPC_LINEAR_unknown &&
- "Unexpected linear modifier.");
- if (P.getActions().CheckOpenMPLinearModifier(
- static_cast<OpenMPLinearClauseKind>(Data.ExtraModifier),
- Data.ExtraModifierLoc))
- Data.ExtraModifier = OMPC_LINEAR_val;
- LinModifiers.append(Linears.size() - LinModifiers.size(),
- Data.ExtraModifier);
- Steps.append(Linears.size() - Steps.size(), Data.DepModOrTailExpr);
- }
- } 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, /*IsReinject*/ true);
- PP.EnterTokenStream(Toks, /*DisableMacroExpansion=*/true,
- /*IsReinject*/ true);
- // Consume the previously pushed token.
- ConsumeAnyToken(/*ConsumeCodeCompletionTok=*/true);
- 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);
- skipUntilPragmaOpenMPEnd(OMPD_declare_simd);
- // 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));
- }
- namespace {
- /// Constant used in the diagnostics to distinguish the levels in an OpenMP
- /// contexts: selector-set={selector(trait, ...), ...}, ....
- enum OMPContextLvl {
- CONTEXT_SELECTOR_SET_LVL = 0,
- CONTEXT_SELECTOR_LVL = 1,
- CONTEXT_TRAIT_LVL = 2,
- };
- static StringRef stringLiteralParser(Parser &P) {
- ExprResult Res = P.ParseStringLiteralExpression(true);
- return Res.isUsable() ? Res.getAs<StringLiteral>()->getString() : "";
- }
- static StringRef getNameFromIdOrString(Parser &P, Token &Tok,
- OMPContextLvl Lvl) {
- if (Tok.is(tok::identifier) || Tok.is(tok::kw_for)) {
- llvm::SmallString<16> Buffer;
- StringRef Name = P.getPreprocessor().getSpelling(Tok, Buffer);
- (void)P.ConsumeToken();
- return Name;
- }
- if (tok::isStringLiteral(Tok.getKind()))
- return stringLiteralParser(P);
- P.Diag(Tok.getLocation(),
- diag::warn_omp_declare_variant_string_literal_or_identifier)
- << Lvl;
- return "";
- }
- static bool checkForDuplicates(Parser &P, StringRef Name,
- SourceLocation NameLoc,
- llvm::StringMap<SourceLocation> &Seen,
- OMPContextLvl Lvl) {
- auto Res = Seen.try_emplace(Name, NameLoc);
- if (Res.second)
- return false;
- // Each trait-set-selector-name, trait-selector-name and trait-name can
- // only be specified once.
- P.Diag(NameLoc, diag::warn_omp_declare_variant_ctx_mutiple_use)
- << Lvl << Name;
- P.Diag(Res.first->getValue(), diag::note_omp_declare_variant_ctx_used_here)
- << Lvl << Name;
- return true;
- }
- } // namespace
- void Parser::parseOMPTraitPropertyKind(OMPTraitProperty &TIProperty,
- llvm::omp::TraitSet Set,
- llvm::omp::TraitSelector Selector,
- llvm::StringMap<SourceLocation> &Seen) {
- TIProperty.Kind = TraitProperty::invalid;
- SourceLocation NameLoc = Tok.getLocation();
- StringRef Name = getNameFromIdOrString(*this, Tok, CONTEXT_TRAIT_LVL);
- if (Name.empty()) {
- Diag(Tok.getLocation(), diag::note_omp_declare_variant_ctx_options)
- << CONTEXT_TRAIT_LVL << listOpenMPContextTraitProperties(Set, Selector);
- return;
- }
- TIProperty.RawString = Name;
- TIProperty.Kind = getOpenMPContextTraitPropertyKind(Set, Selector, Name);
- if (TIProperty.Kind != TraitProperty::invalid) {
- if (checkForDuplicates(*this, Name, NameLoc, Seen, CONTEXT_TRAIT_LVL))
- TIProperty.Kind = TraitProperty::invalid;
- return;
- }
- // It follows diagnosis and helping notes.
- // FIXME: We should move the diagnosis string generation into libFrontend.
- Diag(NameLoc, diag::warn_omp_declare_variant_ctx_not_a_property)
- << Name << getOpenMPContextTraitSelectorName(Selector)
- << getOpenMPContextTraitSetName(Set);
- TraitSet SetForName = getOpenMPContextTraitSetKind(Name);
- if (SetForName != TraitSet::invalid) {
- Diag(NameLoc, diag::note_omp_declare_variant_ctx_is_a)
- << Name << CONTEXT_SELECTOR_SET_LVL << CONTEXT_TRAIT_LVL;
- Diag(NameLoc, diag::note_omp_declare_variant_ctx_try)
- << Name << "<selector-name>"
- << "(<property-name>)";
- return;
- }
- TraitSelector SelectorForName = getOpenMPContextTraitSelectorKind(Name);
- if (SelectorForName != TraitSelector::invalid) {
- Diag(NameLoc, diag::note_omp_declare_variant_ctx_is_a)
- << Name << CONTEXT_SELECTOR_LVL << CONTEXT_TRAIT_LVL;
- bool AllowsTraitScore = false;
- bool RequiresProperty = false;
- isValidTraitSelectorForTraitSet(
- SelectorForName, getOpenMPContextTraitSetForSelector(SelectorForName),
- AllowsTraitScore, RequiresProperty);
- Diag(NameLoc, diag::note_omp_declare_variant_ctx_try)
- << getOpenMPContextTraitSetName(
- getOpenMPContextTraitSetForSelector(SelectorForName))
- << Name << (RequiresProperty ? "(<property-name>)" : "");
- return;
- }
- for (const auto &PotentialSet :
- {TraitSet::construct, TraitSet::user, TraitSet::implementation,
- TraitSet::device}) {
- TraitProperty PropertyForName =
- getOpenMPContextTraitPropertyKind(PotentialSet, Selector, Name);
- if (PropertyForName == TraitProperty::invalid)
- continue;
- Diag(NameLoc, diag::note_omp_declare_variant_ctx_try)
- << getOpenMPContextTraitSetName(
- getOpenMPContextTraitSetForProperty(PropertyForName))
- << getOpenMPContextTraitSelectorName(
- getOpenMPContextTraitSelectorForProperty(PropertyForName))
- << ("(" + Name + ")").str();
- return;
- }
- Diag(NameLoc, diag::note_omp_declare_variant_ctx_options)
- << CONTEXT_TRAIT_LVL << listOpenMPContextTraitProperties(Set, Selector);
- }
- static bool checkExtensionProperty(Parser &P, SourceLocation Loc,
- OMPTraitProperty &TIProperty,
- OMPTraitSelector &TISelector,
- llvm::StringMap<SourceLocation> &Seen) {
- assert(TISelector.Kind ==
- llvm::omp::TraitSelector::implementation_extension &&
- "Only for extension properties, e.g., "
- "`implementation={extension(PROPERTY)}`");
- if (TIProperty.Kind == TraitProperty::invalid)
- return false;
- if (TIProperty.Kind ==
- TraitProperty::implementation_extension_disable_implicit_base)
- return true;
- if (TIProperty.Kind ==
- TraitProperty::implementation_extension_allow_templates)
- return true;
- if (TIProperty.Kind ==
- TraitProperty::implementation_extension_bind_to_declaration)
- return true;
- auto IsMatchExtension = [](OMPTraitProperty &TP) {
- return (TP.Kind ==
- llvm::omp::TraitProperty::implementation_extension_match_all ||
- TP.Kind ==
- llvm::omp::TraitProperty::implementation_extension_match_any ||
- TP.Kind ==
- llvm::omp::TraitProperty::implementation_extension_match_none);
- };
- if (IsMatchExtension(TIProperty)) {
- for (OMPTraitProperty &SeenProp : TISelector.Properties)
- if (IsMatchExtension(SeenProp)) {
- P.Diag(Loc, diag::err_omp_variant_ctx_second_match_extension);
- StringRef SeenName = llvm::omp::getOpenMPContextTraitPropertyName(
- SeenProp.Kind, SeenProp.RawString);
- SourceLocation SeenLoc = Seen[SeenName];
- P.Diag(SeenLoc, diag::note_omp_declare_variant_ctx_used_here)
- << CONTEXT_TRAIT_LVL << SeenName;
- return false;
- }
- return true;
- }
- llvm_unreachable("Unknown extension property!");
- }
- void Parser::parseOMPContextProperty(OMPTraitSelector &TISelector,
- llvm::omp::TraitSet Set,
- llvm::StringMap<SourceLocation> &Seen) {
- assert(TISelector.Kind != TraitSelector::user_condition &&
- "User conditions are special properties not handled here!");
- SourceLocation PropertyLoc = Tok.getLocation();
- OMPTraitProperty TIProperty;
- parseOMPTraitPropertyKind(TIProperty, Set, TISelector.Kind, Seen);
- if (TISelector.Kind == llvm::omp::TraitSelector::implementation_extension)
- if (!checkExtensionProperty(*this, Tok.getLocation(), TIProperty,
- TISelector, Seen))
- TIProperty.Kind = TraitProperty::invalid;
- // If we have an invalid property here we already issued a warning.
- if (TIProperty.Kind == TraitProperty::invalid) {
- if (PropertyLoc != Tok.getLocation())
- Diag(Tok.getLocation(), diag::note_omp_declare_variant_ctx_continue_here)
- << CONTEXT_TRAIT_LVL;
- return;
- }
- if (isValidTraitPropertyForTraitSetAndSelector(TIProperty.Kind,
- TISelector.Kind, Set)) {
- // If we make it here the property, selector, set, score, condition, ... are
- // all valid (or have been corrected). Thus we can record the property.
- TISelector.Properties.push_back(TIProperty);
- return;
- }
- Diag(PropertyLoc, diag::warn_omp_ctx_incompatible_property_for_selector)
- << getOpenMPContextTraitPropertyName(TIProperty.Kind,
- TIProperty.RawString)
- << getOpenMPContextTraitSelectorName(TISelector.Kind)
- << getOpenMPContextTraitSetName(Set);
- Diag(PropertyLoc, diag::note_omp_ctx_compatible_set_and_selector_for_property)
- << getOpenMPContextTraitPropertyName(TIProperty.Kind,
- TIProperty.RawString)
- << getOpenMPContextTraitSelectorName(
- getOpenMPContextTraitSelectorForProperty(TIProperty.Kind))
- << getOpenMPContextTraitSetName(
- getOpenMPContextTraitSetForProperty(TIProperty.Kind));
- Diag(Tok.getLocation(), diag::note_omp_declare_variant_ctx_continue_here)
- << CONTEXT_TRAIT_LVL;
- }
- void Parser::parseOMPTraitSelectorKind(OMPTraitSelector &TISelector,
- llvm::omp::TraitSet Set,
- llvm::StringMap<SourceLocation> &Seen) {
- TISelector.Kind = TraitSelector::invalid;
- SourceLocation NameLoc = Tok.getLocation();
- StringRef Name = getNameFromIdOrString(*this, Tok, CONTEXT_SELECTOR_LVL);
- if (Name.empty()) {
- Diag(Tok.getLocation(), diag::note_omp_declare_variant_ctx_options)
- << CONTEXT_SELECTOR_LVL << listOpenMPContextTraitSelectors(Set);
- return;
- }
- TISelector.Kind = getOpenMPContextTraitSelectorKind(Name);
- if (TISelector.Kind != TraitSelector::invalid) {
- if (checkForDuplicates(*this, Name, NameLoc, Seen, CONTEXT_SELECTOR_LVL))
- TISelector.Kind = TraitSelector::invalid;
- return;
- }
- // It follows diagnosis and helping notes.
- Diag(NameLoc, diag::warn_omp_declare_variant_ctx_not_a_selector)
- << Name << getOpenMPContextTraitSetName(Set);
- TraitSet SetForName = getOpenMPContextTraitSetKind(Name);
- if (SetForName != TraitSet::invalid) {
- Diag(NameLoc, diag::note_omp_declare_variant_ctx_is_a)
- << Name << CONTEXT_SELECTOR_SET_LVL << CONTEXT_SELECTOR_LVL;
- Diag(NameLoc, diag::note_omp_declare_variant_ctx_try)
- << Name << "<selector-name>"
- << "<property-name>";
- return;
- }
- for (const auto &PotentialSet :
- {TraitSet::construct, TraitSet::user, TraitSet::implementation,
- TraitSet::device}) {
- TraitProperty PropertyForName = getOpenMPContextTraitPropertyKind(
- PotentialSet, TraitSelector::invalid, Name);
- if (PropertyForName == TraitProperty::invalid)
- continue;
- Diag(NameLoc, diag::note_omp_declare_variant_ctx_is_a)
- << Name << CONTEXT_TRAIT_LVL << CONTEXT_SELECTOR_LVL;
- Diag(NameLoc, diag::note_omp_declare_variant_ctx_try)
- << getOpenMPContextTraitSetName(
- getOpenMPContextTraitSetForProperty(PropertyForName))
- << getOpenMPContextTraitSelectorName(
- getOpenMPContextTraitSelectorForProperty(PropertyForName))
- << ("(" + Name + ")").str();
- return;
- }
- Diag(NameLoc, diag::note_omp_declare_variant_ctx_options)
- << CONTEXT_SELECTOR_LVL << listOpenMPContextTraitSelectors(Set);
- }
- /// Parse optional 'score' '(' <expr> ')' ':'.
- static ExprResult parseContextScore(Parser &P) {
- ExprResult ScoreExpr;
- llvm::SmallString<16> Buffer;
- StringRef SelectorName =
- P.getPreprocessor().getSpelling(P.getCurToken(), Buffer);
- if (!SelectorName.equals("score"))
- return ScoreExpr;
- (void)P.ConsumeToken();
- SourceLocation RLoc;
- ScoreExpr = P.ParseOpenMPParensExpr(SelectorName, RLoc);
- // Parse ':'
- if (P.getCurToken().is(tok::colon))
- (void)P.ConsumeAnyToken();
- else
- P.Diag(P.getCurToken(), diag::warn_omp_declare_variant_expected)
- << "':'"
- << "score expression";
- return ScoreExpr;
- }
- /// Parses an OpenMP context selector.
- ///
- /// <trait-selector-name> ['('[<trait-score>] <trait-property> [, <t-p>]* ')']
- void Parser::parseOMPContextSelector(
- OMPTraitSelector &TISelector, llvm::omp::TraitSet Set,
- llvm::StringMap<SourceLocation> &SeenSelectors) {
- unsigned short OuterPC = ParenCount;
- // If anything went wrong we issue an error or warning and then skip the rest
- // of the selector. However, commas are ambiguous so we look for the nesting
- // of parentheses here as well.
- auto FinishSelector = [OuterPC, this]() -> void {
- bool Done = false;
- while (!Done) {
- while (!SkipUntil({tok::r_brace, tok::r_paren, tok::comma,
- tok::annot_pragma_openmp_end},
- StopBeforeMatch))
- ;
- if (Tok.is(tok::r_paren) && OuterPC > ParenCount)
- (void)ConsumeParen();
- if (OuterPC <= ParenCount) {
- Done = true;
- break;
- }
- if (!Tok.is(tok::comma) && !Tok.is(tok::r_paren)) {
- Done = true;
- break;
- }
- (void)ConsumeAnyToken();
- }
- Diag(Tok.getLocation(), diag::note_omp_declare_variant_ctx_continue_here)
- << CONTEXT_SELECTOR_LVL;
- };
- SourceLocation SelectorLoc = Tok.getLocation();
- parseOMPTraitSelectorKind(TISelector, Set, SeenSelectors);
- if (TISelector.Kind == TraitSelector::invalid)
- return FinishSelector();
- bool AllowsTraitScore = false;
- bool RequiresProperty = false;
- if (!isValidTraitSelectorForTraitSet(TISelector.Kind, Set, AllowsTraitScore,
- RequiresProperty)) {
- Diag(SelectorLoc, diag::warn_omp_ctx_incompatible_selector_for_set)
- << getOpenMPContextTraitSelectorName(TISelector.Kind)
- << getOpenMPContextTraitSetName(Set);
- Diag(SelectorLoc, diag::note_omp_ctx_compatible_set_for_selector)
- << getOpenMPContextTraitSelectorName(TISelector.Kind)
- << getOpenMPContextTraitSetName(
- getOpenMPContextTraitSetForSelector(TISelector.Kind))
- << RequiresProperty;
- return FinishSelector();
- }
- if (!RequiresProperty) {
- TISelector.Properties.push_back(
- {getOpenMPContextTraitPropertyForSelector(TISelector.Kind),
- getOpenMPContextTraitSelectorName(TISelector.Kind)});
- return;
- }
- if (!Tok.is(tok::l_paren)) {
- Diag(SelectorLoc, diag::warn_omp_ctx_selector_without_properties)
- << getOpenMPContextTraitSelectorName(TISelector.Kind)
- << getOpenMPContextTraitSetName(Set);
- return FinishSelector();
- }
- if (TISelector.Kind == TraitSelector::user_condition) {
- SourceLocation RLoc;
- ExprResult Condition = ParseOpenMPParensExpr("user condition", RLoc);
- if (!Condition.isUsable())
- return FinishSelector();
- TISelector.ScoreOrCondition = Condition.get();
- TISelector.Properties.push_back(
- {TraitProperty::user_condition_unknown, "<condition>"});
- return;
- }
- BalancedDelimiterTracker BDT(*this, tok::l_paren,
- tok::annot_pragma_openmp_end);
- // Parse '('.
- (void)BDT.consumeOpen();
- SourceLocation ScoreLoc = Tok.getLocation();
- ExprResult Score = parseContextScore(*this);
- if (!AllowsTraitScore && !Score.isUnset()) {
- if (Score.isUsable()) {
- Diag(ScoreLoc, diag::warn_omp_ctx_incompatible_score_for_property)
- << getOpenMPContextTraitSelectorName(TISelector.Kind)
- << getOpenMPContextTraitSetName(Set) << Score.get();
- } else {
- Diag(ScoreLoc, diag::warn_omp_ctx_incompatible_score_for_property)
- << getOpenMPContextTraitSelectorName(TISelector.Kind)
- << getOpenMPContextTraitSetName(Set) << "<invalid>";
- }
- Score = ExprResult();
- }
- if (Score.isUsable())
- TISelector.ScoreOrCondition = Score.get();
- llvm::StringMap<SourceLocation> SeenProperties;
- do {
- parseOMPContextProperty(TISelector, Set, SeenProperties);
- } while (TryConsumeToken(tok::comma));
- // Parse ')'.
- BDT.consumeClose();
- }
- void Parser::parseOMPTraitSetKind(OMPTraitSet &TISet,
- llvm::StringMap<SourceLocation> &Seen) {
- TISet.Kind = TraitSet::invalid;
- SourceLocation NameLoc = Tok.getLocation();
- StringRef Name = getNameFromIdOrString(*this, Tok, CONTEXT_SELECTOR_SET_LVL);
- if (Name.empty()) {
- Diag(Tok.getLocation(), diag::note_omp_declare_variant_ctx_options)
- << CONTEXT_SELECTOR_SET_LVL << listOpenMPContextTraitSets();
- return;
- }
- TISet.Kind = getOpenMPContextTraitSetKind(Name);
- if (TISet.Kind != TraitSet::invalid) {
- if (checkForDuplicates(*this, Name, NameLoc, Seen,
- CONTEXT_SELECTOR_SET_LVL))
- TISet.Kind = TraitSet::invalid;
- return;
- }
- // It follows diagnosis and helping notes.
- Diag(NameLoc, diag::warn_omp_declare_variant_ctx_not_a_set) << Name;
- TraitSelector SelectorForName = getOpenMPContextTraitSelectorKind(Name);
- if (SelectorForName != TraitSelector::invalid) {
- Diag(NameLoc, diag::note_omp_declare_variant_ctx_is_a)
- << Name << CONTEXT_SELECTOR_LVL << CONTEXT_SELECTOR_SET_LVL;
- bool AllowsTraitScore = false;
- bool RequiresProperty = false;
- isValidTraitSelectorForTraitSet(
- SelectorForName, getOpenMPContextTraitSetForSelector(SelectorForName),
- AllowsTraitScore, RequiresProperty);
- Diag(NameLoc, diag::note_omp_declare_variant_ctx_try)
- << getOpenMPContextTraitSetName(
- getOpenMPContextTraitSetForSelector(SelectorForName))
- << Name << (RequiresProperty ? "(<property-name>)" : "");
- return;
- }
- for (const auto &PotentialSet :
- {TraitSet::construct, TraitSet::user, TraitSet::implementation,
- TraitSet::device}) {
- TraitProperty PropertyForName = getOpenMPContextTraitPropertyKind(
- PotentialSet, TraitSelector::invalid, Name);
- if (PropertyForName == TraitProperty::invalid)
- continue;
- Diag(NameLoc, diag::note_omp_declare_variant_ctx_is_a)
- << Name << CONTEXT_TRAIT_LVL << CONTEXT_SELECTOR_SET_LVL;
- Diag(NameLoc, diag::note_omp_declare_variant_ctx_try)
- << getOpenMPContextTraitSetName(
- getOpenMPContextTraitSetForProperty(PropertyForName))
- << getOpenMPContextTraitSelectorName(
- getOpenMPContextTraitSelectorForProperty(PropertyForName))
- << ("(" + Name + ")").str();
- return;
- }
- Diag(NameLoc, diag::note_omp_declare_variant_ctx_options)
- << CONTEXT_SELECTOR_SET_LVL << listOpenMPContextTraitSets();
- }
- /// Parses an OpenMP context selector set.
- ///
- /// <trait-set-selector-name> '=' '{' <trait-selector> [, <trait-selector>]* '}'
- void Parser::parseOMPContextSelectorSet(
- OMPTraitSet &TISet, llvm::StringMap<SourceLocation> &SeenSets) {
- auto OuterBC = BraceCount;
- // If anything went wrong we issue an error or warning and then skip the rest
- // of the set. However, commas are ambiguous so we look for the nesting
- // of braces here as well.
- auto FinishSelectorSet = [this, OuterBC]() -> void {
- bool Done = false;
- while (!Done) {
- while (!SkipUntil({tok::comma, tok::r_brace, tok::r_paren,
- tok::annot_pragma_openmp_end},
- StopBeforeMatch))
- ;
- if (Tok.is(tok::r_brace) && OuterBC > BraceCount)
- (void)ConsumeBrace();
- if (OuterBC <= BraceCount) {
- Done = true;
- break;
- }
- if (!Tok.is(tok::comma) && !Tok.is(tok::r_brace)) {
- Done = true;
- break;
- }
- (void)ConsumeAnyToken();
- }
- Diag(Tok.getLocation(), diag::note_omp_declare_variant_ctx_continue_here)
- << CONTEXT_SELECTOR_SET_LVL;
- };
- parseOMPTraitSetKind(TISet, SeenSets);
- if (TISet.Kind == TraitSet::invalid)
- return FinishSelectorSet();
- // Parse '='.
- if (!TryConsumeToken(tok::equal))
- Diag(Tok.getLocation(), diag::warn_omp_declare_variant_expected)
- << "="
- << ("context set name \"" + getOpenMPContextTraitSetName(TISet.Kind) +
- "\"")
- .str();
- // Parse '{'.
- if (Tok.is(tok::l_brace)) {
- (void)ConsumeBrace();
- } else {
- Diag(Tok.getLocation(), diag::warn_omp_declare_variant_expected)
- << "{"
- << ("'=' that follows the context set name \"" +
- getOpenMPContextTraitSetName(TISet.Kind) + "\"")
- .str();
- }
- llvm::StringMap<SourceLocation> SeenSelectors;
- do {
- OMPTraitSelector TISelector;
- parseOMPContextSelector(TISelector, TISet.Kind, SeenSelectors);
- if (TISelector.Kind != TraitSelector::invalid &&
- !TISelector.Properties.empty())
- TISet.Selectors.push_back(TISelector);
- } while (TryConsumeToken(tok::comma));
- // Parse '}'.
- if (Tok.is(tok::r_brace)) {
- (void)ConsumeBrace();
- } else {
- Diag(Tok.getLocation(), diag::warn_omp_declare_variant_expected)
- << "}"
- << ("context selectors for the context set \"" +
- getOpenMPContextTraitSetName(TISet.Kind) + "\"")
- .str();
- }
- }
- /// Parse OpenMP context selectors:
- ///
- /// <trait-set-selector> [, <trait-set-selector>]*
- bool Parser::parseOMPContextSelectors(SourceLocation Loc, OMPTraitInfo &TI) {
- llvm::StringMap<SourceLocation> SeenSets;
- do {
- OMPTraitSet TISet;
- parseOMPContextSelectorSet(TISet, SeenSets);
- if (TISet.Kind != TraitSet::invalid && !TISet.Selectors.empty())
- TI.Sets.push_back(TISet);
- } while (TryConsumeToken(tok::comma));
- return false;
- }
- /// Parse clauses for '#pragma omp declare variant ( variant-func-id ) clause'.
- void Parser::ParseOMPDeclareVariantClauses(Parser::DeclGroupPtrTy Ptr,
- CachedTokens &Toks,
- SourceLocation Loc) {
- PP.EnterToken(Tok, /*IsReinject*/ true);
- PP.EnterTokenStream(Toks, /*DisableMacroExpansion=*/true,
- /*IsReinject*/ true);
- // Consume the previously pushed token.
- ConsumeAnyToken(/*ConsumeCodeCompletionTok=*/true);
- ConsumeAnyToken(/*ConsumeCodeCompletionTok=*/true);
- FNContextRAII FnContext(*this, Ptr);
- // Parse function declaration id.
- SourceLocation RLoc;
- // Parse with IsAddressOfOperand set to true to parse methods as DeclRefExprs
- // instead of MemberExprs.
- ExprResult AssociatedFunction;
- {
- // Do not mark function as is used to prevent its emission if this is the
- // only place where it is used.
- EnterExpressionEvaluationContext Unevaluated(
- Actions, Sema::ExpressionEvaluationContext::Unevaluated);
- AssociatedFunction = ParseOpenMPParensExpr(
- getOpenMPDirectiveName(OMPD_declare_variant), RLoc,
- /*IsAddressOfOperand=*/true);
- }
- if (!AssociatedFunction.isUsable()) {
- if (!Tok.is(tok::annot_pragma_openmp_end))
- while (!SkipUntil(tok::annot_pragma_openmp_end, StopBeforeMatch))
- ;
- // Skip the last annot_pragma_openmp_end.
- (void)ConsumeAnnotationToken();
- return;
- }
- OMPTraitInfo *ParentTI = Actions.getOMPTraitInfoForSurroundingScope();
- ASTContext &ASTCtx = Actions.getASTContext();
- OMPTraitInfo &TI = ASTCtx.getNewOMPTraitInfo();
- SmallVector<Expr *, 6> AdjustNothing;
- SmallVector<Expr *, 6> AdjustNeedDevicePtr;
- SmallVector<OMPInteropInfo, 3> AppendArgs;
- SourceLocation AdjustArgsLoc, AppendArgsLoc;
- // At least one clause is required.
- if (Tok.is(tok::annot_pragma_openmp_end)) {
- Diag(Tok.getLocation(), diag::err_omp_declare_variant_wrong_clause)
- << (getLangOpts().OpenMP < 51 ? 0 : 1);
- }
- bool IsError = false;
- while (Tok.isNot(tok::annot_pragma_openmp_end)) {
- OpenMPClauseKind CKind = Tok.isAnnotation()
- ? OMPC_unknown
- : getOpenMPClauseKind(PP.getSpelling(Tok));
- if (!isAllowedClauseForDirective(OMPD_declare_variant, CKind,
- getLangOpts().OpenMP)) {
- Diag(Tok.getLocation(), diag::err_omp_declare_variant_wrong_clause)
- << (getLangOpts().OpenMP < 51 ? 0 : 1);
- IsError = true;
- }
- if (!IsError) {
- switch (CKind) {
- case OMPC_match:
- IsError = parseOMPDeclareVariantMatchClause(Loc, TI, ParentTI);
- break;
- case OMPC_adjust_args: {
- AdjustArgsLoc = Tok.getLocation();
- ConsumeToken();
- Sema::OpenMPVarListDataTy Data;
- SmallVector<Expr *> Vars;
- IsError = ParseOpenMPVarList(OMPD_declare_variant, OMPC_adjust_args,
- Vars, Data);
- if (!IsError)
- llvm::append_range(Data.ExtraModifier == OMPC_ADJUST_ARGS_nothing
- ? AdjustNothing
- : AdjustNeedDevicePtr,
- Vars);
- break;
- }
- case OMPC_append_args:
- if (!AppendArgs.empty()) {
- Diag(AppendArgsLoc, diag::err_omp_more_one_clause)
- << getOpenMPDirectiveName(OMPD_declare_variant)
- << getOpenMPClauseName(CKind) << 0;
- IsError = true;
- }
- if (!IsError) {
- AppendArgsLoc = Tok.getLocation();
- ConsumeToken();
- IsError = parseOpenMPAppendArgs(AppendArgs);
- }
- break;
- default:
- llvm_unreachable("Unexpected clause for declare variant.");
- }
- }
- if (IsError) {
- while (!SkipUntil(tok::annot_pragma_openmp_end, StopBeforeMatch))
- ;
- // Skip the last annot_pragma_openmp_end.
- (void)ConsumeAnnotationToken();
- return;
- }
- // Skip ',' if any.
- if (Tok.is(tok::comma))
- ConsumeToken();
- }
- std::optional<std::pair<FunctionDecl *, Expr *>> DeclVarData =
- Actions.checkOpenMPDeclareVariantFunction(
- Ptr, AssociatedFunction.get(), TI, AppendArgs.size(),
- SourceRange(Loc, Tok.getLocation()));
- if (DeclVarData && !TI.Sets.empty())
- Actions.ActOnOpenMPDeclareVariantDirective(
- DeclVarData->first, DeclVarData->second, TI, AdjustNothing,
- AdjustNeedDevicePtr, AppendArgs, AdjustArgsLoc, AppendArgsLoc,
- SourceRange(Loc, Tok.getLocation()));
- // Skip the last annot_pragma_openmp_end.
- (void)ConsumeAnnotationToken();
- }
- bool Parser::parseOpenMPAppendArgs(
- SmallVectorImpl<OMPInteropInfo> &InteropInfos) {
- bool HasError = false;
- // Parse '('.
- BalancedDelimiterTracker T(*this, tok::l_paren, tok::annot_pragma_openmp_end);
- if (T.expectAndConsume(diag::err_expected_lparen_after,
- getOpenMPClauseName(OMPC_append_args).data()))
- return true;
- // Parse the list of append-ops, each is;
- // interop(interop-type[,interop-type]...)
- while (Tok.is(tok::identifier) && Tok.getIdentifierInfo()->isStr("interop")) {
- ConsumeToken();
- BalancedDelimiterTracker IT(*this, tok::l_paren,
- tok::annot_pragma_openmp_end);
- if (IT.expectAndConsume(diag::err_expected_lparen_after, "interop"))
- return true;
- OMPInteropInfo InteropInfo;
- if (ParseOMPInteropInfo(InteropInfo, OMPC_append_args))
- HasError = true;
- else
- InteropInfos.push_back(InteropInfo);
- IT.consumeClose();
- if (Tok.is(tok::comma))
- ConsumeToken();
- }
- if (!HasError && InteropInfos.empty()) {
- HasError = true;
- Diag(Tok.getLocation(), diag::err_omp_unexpected_append_op);
- SkipUntil(tok::comma, tok::r_paren, tok::annot_pragma_openmp_end,
- StopBeforeMatch);
- }
- HasError = T.consumeClose() || HasError;
- return HasError;
- }
- bool Parser::parseOMPDeclareVariantMatchClause(SourceLocation Loc,
- OMPTraitInfo &TI,
- OMPTraitInfo *ParentTI) {
- // Parse 'match'.
- OpenMPClauseKind CKind = Tok.isAnnotation()
- ? OMPC_unknown
- : getOpenMPClauseKind(PP.getSpelling(Tok));
- if (CKind != OMPC_match) {
- Diag(Tok.getLocation(), diag::err_omp_declare_variant_wrong_clause)
- << (getLangOpts().OpenMP < 51 ? 0 : 1);
- return true;
- }
- (void)ConsumeToken();
- // Parse '('.
- BalancedDelimiterTracker T(*this, tok::l_paren, tok::annot_pragma_openmp_end);
- if (T.expectAndConsume(diag::err_expected_lparen_after,
- getOpenMPClauseName(OMPC_match).data()))
- return true;
- // Parse inner context selectors.
- parseOMPContextSelectors(Loc, TI);
- // Parse ')'
- (void)T.consumeClose();
- if (!ParentTI)
- return false;
- // Merge the parent/outer trait info into the one we just parsed and diagnose
- // problems.
- // TODO: Keep some source location in the TI to provide better diagnostics.
- // TODO: Perform some kind of equivalence check on the condition and score
- // expressions.
- for (const OMPTraitSet &ParentSet : ParentTI->Sets) {
- bool MergedSet = false;
- for (OMPTraitSet &Set : TI.Sets) {
- if (Set.Kind != ParentSet.Kind)
- continue;
- MergedSet = true;
- for (const OMPTraitSelector &ParentSelector : ParentSet.Selectors) {
- bool MergedSelector = false;
- for (OMPTraitSelector &Selector : Set.Selectors) {
- if (Selector.Kind != ParentSelector.Kind)
- continue;
- MergedSelector = true;
- for (const OMPTraitProperty &ParentProperty :
- ParentSelector.Properties) {
- bool MergedProperty = false;
- for (OMPTraitProperty &Property : Selector.Properties) {
- // Ignore "equivalent" properties.
- if (Property.Kind != ParentProperty.Kind)
- continue;
- // If the kind is the same but the raw string not, we don't want
- // to skip out on the property.
- MergedProperty |= Property.RawString == ParentProperty.RawString;
- if (Property.RawString == ParentProperty.RawString &&
- Selector.ScoreOrCondition == ParentSelector.ScoreOrCondition)
- continue;
- if (Selector.Kind == llvm::omp::TraitSelector::user_condition) {
- Diag(Loc, diag::err_omp_declare_variant_nested_user_condition);
- } else if (Selector.ScoreOrCondition !=
- ParentSelector.ScoreOrCondition) {
- Diag(Loc, diag::err_omp_declare_variant_duplicate_nested_trait)
- << getOpenMPContextTraitPropertyName(
- ParentProperty.Kind, ParentProperty.RawString)
- << getOpenMPContextTraitSelectorName(ParentSelector.Kind)
- << getOpenMPContextTraitSetName(ParentSet.Kind);
- }
- }
- if (!MergedProperty)
- Selector.Properties.push_back(ParentProperty);
- }
- }
- if (!MergedSelector)
- Set.Selectors.push_back(ParentSelector);
- }
- }
- if (!MergedSet)
- TI.Sets.push_back(ParentSet);
- }
- return false;
- }
- /// <clause> [clause[ [,] clause] ... ]
- ///
- /// clauses: for error directive
- /// 'at' '(' compilation | execution ')'
- /// 'severity' '(' fatal | warning ')'
- /// 'message' '(' msg-string ')'
- /// ....
- void Parser::ParseOpenMPClauses(OpenMPDirectiveKind DKind,
- SmallVectorImpl<OMPClause *> &Clauses,
- SourceLocation Loc) {
- SmallVector<llvm::PointerIntPair<OMPClause *, 1, bool>,
- llvm::omp::Clause_enumSize + 1>
- FirstClauses(llvm::omp::Clause_enumSize + 1);
- while (Tok.isNot(tok::annot_pragma_openmp_end)) {
- OpenMPClauseKind CKind = Tok.isAnnotation()
- ? OMPC_unknown
- : getOpenMPClauseKind(PP.getSpelling(Tok));
- Actions.StartOpenMPClause(CKind);
- OMPClause *Clause = ParseOpenMPClause(
- DKind, CKind, !FirstClauses[unsigned(CKind)].getInt());
- SkipUntil(tok::comma, tok::identifier, tok::annot_pragma_openmp_end,
- StopBeforeMatch);
- FirstClauses[unsigned(CKind)].setInt(true);
- if (Clause != nullptr)
- Clauses.push_back(Clause);
- if (Tok.is(tok::annot_pragma_openmp_end)) {
- Actions.EndOpenMPClause();
- break;
- }
- // Skip ',' if any.
- if (Tok.is(tok::comma))
- ConsumeToken();
- Actions.EndOpenMPClause();
- }
- }
- /// `omp assumes` or `omp begin/end assumes` <clause> [[,]<clause>]...
- /// where
- ///
- /// clause:
- /// 'ext_IMPL_DEFINED'
- /// 'absent' '(' directive-name [, directive-name]* ')'
- /// 'contains' '(' directive-name [, directive-name]* ')'
- /// 'holds' '(' scalar-expression ')'
- /// 'no_openmp'
- /// 'no_openmp_routines'
- /// 'no_parallelism'
- ///
- void Parser::ParseOpenMPAssumesDirective(OpenMPDirectiveKind DKind,
- SourceLocation Loc) {
- SmallVector<std::string, 4> Assumptions;
- bool SkippedClauses = false;
- auto SkipBraces = [&](llvm::StringRef Spelling, bool IssueNote) {
- BalancedDelimiterTracker T(*this, tok::l_paren,
- tok::annot_pragma_openmp_end);
- if (T.expectAndConsume(diag::err_expected_lparen_after, Spelling.data()))
- return;
- T.skipToEnd();
- if (IssueNote && T.getCloseLocation().isValid())
- Diag(T.getCloseLocation(),
- diag::note_omp_assumption_clause_continue_here);
- };
- /// Helper to determine which AssumptionClauseMapping (ACM) in the
- /// AssumptionClauseMappings table matches \p RawString. The return value is
- /// the index of the matching ACM into the table or -1 if there was no match.
- auto MatchACMClause = [&](StringRef RawString) {
- llvm::StringSwitch<int> SS(RawString);
- unsigned ACMIdx = 0;
- for (const AssumptionClauseMappingInfo &ACMI : AssumptionClauseMappings) {
- if (ACMI.StartsWith)
- SS.StartsWith(ACMI.Identifier, ACMIdx++);
- else
- SS.Case(ACMI.Identifier, ACMIdx++);
- }
- return SS.Default(-1);
- };
- while (Tok.isNot(tok::annot_pragma_openmp_end)) {
- IdentifierInfo *II = nullptr;
- SourceLocation StartLoc = Tok.getLocation();
- int Idx = -1;
- if (Tok.isAnyIdentifier()) {
- II = Tok.getIdentifierInfo();
- Idx = MatchACMClause(II->getName());
- }
- ConsumeAnyToken();
- bool NextIsLPar = Tok.is(tok::l_paren);
- // Handle unknown clauses by skipping them.
- if (Idx == -1) {
- Diag(StartLoc, diag::warn_omp_unknown_assumption_clause_missing_id)
- << llvm::omp::getOpenMPDirectiveName(DKind)
- << llvm::omp::getAllAssumeClauseOptions() << NextIsLPar;
- if (NextIsLPar)
- SkipBraces(II ? II->getName() : "", /* IssueNote */ true);
- SkippedClauses = true;
- continue;
- }
- const AssumptionClauseMappingInfo &ACMI = AssumptionClauseMappings[Idx];
- if (ACMI.HasDirectiveList || ACMI.HasExpression) {
- // TODO: We ignore absent, contains, and holds assumptions for now. We
- // also do not verify the content in the parenthesis at all.
- SkippedClauses = true;
- SkipBraces(II->getName(), /* IssueNote */ false);
- continue;
- }
- if (NextIsLPar) {
- Diag(Tok.getLocation(),
- diag::warn_omp_unknown_assumption_clause_without_args)
- << II;
- SkipBraces(II->getName(), /* IssueNote */ true);
- }
- assert(II && "Expected an identifier clause!");
- std::string Assumption = II->getName().str();
- if (ACMI.StartsWith)
- Assumption = "ompx_" + Assumption.substr(ACMI.Identifier.size());
- else
- Assumption = "omp_" + Assumption;
- Assumptions.push_back(Assumption);
- }
- Actions.ActOnOpenMPAssumesDirective(Loc, DKind, Assumptions, SkippedClauses);
- }
- void Parser::ParseOpenMPEndAssumesDirective(SourceLocation Loc) {
- if (Actions.isInOpenMPAssumeScope())
- Actions.ActOnOpenMPEndAssumesDirective();
- else
- Diag(Loc, diag::err_expected_begin_assumes);
- }
- /// Parsing of simple OpenMP clauses like 'default' or 'proc_bind'.
- ///
- /// default-clause:
- /// 'default' '(' 'none' | 'shared' | 'private' | 'firstprivate' ')
- ///
- /// proc_bind-clause:
- /// 'proc_bind' '(' 'master' | 'close' | 'spread' ')
- ///
- /// device_type-clause:
- /// 'device_type' '(' 'host' | 'nohost' | 'any' )'
- namespace {
- struct SimpleClauseData {
- unsigned Type;
- SourceLocation Loc;
- SourceLocation LOpen;
- SourceLocation TypeLoc;
- SourceLocation RLoc;
- SimpleClauseData(unsigned Type, SourceLocation Loc, SourceLocation LOpen,
- SourceLocation TypeLoc, SourceLocation RLoc)
- : Type(Type), Loc(Loc), LOpen(LOpen), TypeLoc(TypeLoc), RLoc(RLoc) {}
- };
- } // anonymous namespace
- static std::optional<SimpleClauseData>
- parseOpenMPSimpleClause(Parser &P, OpenMPClauseKind Kind) {
- const Token &Tok = P.getCurToken();
- SourceLocation Loc = Tok.getLocation();
- SourceLocation LOpen = P.ConsumeToken();
- // Parse '('.
- BalancedDelimiterTracker T(P, tok::l_paren, tok::annot_pragma_openmp_end);
- if (T.expectAndConsume(diag::err_expected_lparen_after,
- getOpenMPClauseName(Kind).data()))
- return std::nullopt;
- unsigned Type = getOpenMPSimpleClauseType(
- Kind, Tok.isAnnotation() ? "" : P.getPreprocessor().getSpelling(Tok),
- P.getLangOpts());
- SourceLocation TypeLoc = Tok.getLocation();
- if (Tok.isNot(tok::r_paren) && Tok.isNot(tok::comma) &&
- Tok.isNot(tok::annot_pragma_openmp_end))
- P.ConsumeAnyToken();
- // Parse ')'.
- SourceLocation RLoc = Tok.getLocation();
- if (!T.consumeClose())
- RLoc = T.getCloseLocation();
- return SimpleClauseData(Type, Loc, LOpen, TypeLoc, RLoc);
- }
- void Parser::ParseOMPDeclareTargetClauses(
- Sema::DeclareTargetContextInfo &DTCI) {
- SourceLocation DeviceTypeLoc;
- bool RequiresToOrLinkOrIndirectClause = false;
- bool HasToOrLinkOrIndirectClause = false;
- while (Tok.isNot(tok::annot_pragma_openmp_end)) {
- OMPDeclareTargetDeclAttr::MapTypeTy MT = OMPDeclareTargetDeclAttr::MT_To;
- bool HasIdentifier = Tok.is(tok::identifier);
- if (HasIdentifier) {
- // If we see any clause we need a to or link clause.
- RequiresToOrLinkOrIndirectClause = true;
- IdentifierInfo *II = Tok.getIdentifierInfo();
- StringRef ClauseName = II->getName();
- bool IsDeviceTypeClause =
- getLangOpts().OpenMP >= 50 &&
- getOpenMPClauseKind(ClauseName) == OMPC_device_type;
- bool IsIndirectClause = getLangOpts().OpenMP >= 51 &&
- getOpenMPClauseKind(ClauseName) == OMPC_indirect;
- if (DTCI.Indirect && IsIndirectClause) {
- Diag(Tok, diag::err_omp_more_one_clause)
- << getOpenMPDirectiveName(OMPD_declare_target)
- << getOpenMPClauseName(OMPC_indirect) << 0;
- break;
- }
- bool IsToEnterOrLinkClause =
- OMPDeclareTargetDeclAttr::ConvertStrToMapTypeTy(ClauseName, MT);
- assert((!IsDeviceTypeClause || !IsToEnterOrLinkClause) &&
- "Cannot be both!");
- // Starting with OpenMP 5.2 the `to` clause has been replaced by the
- // `enter` clause.
- if (getLangOpts().OpenMP >= 52 && ClauseName == "to") {
- Diag(Tok, diag::err_omp_declare_target_unexpected_to_clause);
- break;
- }
- if (getLangOpts().OpenMP <= 51 && ClauseName == "enter") {
- Diag(Tok, diag::err_omp_declare_target_unexpected_enter_clause);
- break;
- }
- if (!IsDeviceTypeClause && !IsIndirectClause &&
- DTCI.Kind == OMPD_begin_declare_target) {
- Diag(Tok, diag::err_omp_declare_target_unexpected_clause)
- << ClauseName << (getLangOpts().OpenMP >= 51 ? 3 : 0);
- break;
- }
- if (!IsDeviceTypeClause && !IsToEnterOrLinkClause && !IsIndirectClause) {
- Diag(Tok, getLangOpts().OpenMP >= 52
- ? diag::err_omp_declare_target_unexpected_clause_52
- : diag::err_omp_declare_target_unexpected_clause)
- << ClauseName
- << (getLangOpts().OpenMP >= 51
- ? 4
- : getLangOpts().OpenMP >= 50 ? 2 : 1);
- break;
- }
- if (IsToEnterOrLinkClause || IsIndirectClause)
- HasToOrLinkOrIndirectClause = true;
- if (IsIndirectClause) {
- if (!ParseOpenMPIndirectClause(DTCI, /*ParseOnly*/ false))
- break;
- continue;
- }
- // Parse 'device_type' clause and go to next clause if any.
- if (IsDeviceTypeClause) {
- std::optional<SimpleClauseData> DevTypeData =
- parseOpenMPSimpleClause(*this, OMPC_device_type);
- if (DevTypeData) {
- if (DeviceTypeLoc.isValid()) {
- // We already saw another device_type clause, diagnose it.
- Diag(DevTypeData->Loc,
- diag::warn_omp_more_one_device_type_clause);
- break;
- }
- switch (static_cast<OpenMPDeviceType>(DevTypeData->Type)) {
- case OMPC_DEVICE_TYPE_any:
- DTCI.DT = OMPDeclareTargetDeclAttr::DT_Any;
- break;
- case OMPC_DEVICE_TYPE_host:
- DTCI.DT = OMPDeclareTargetDeclAttr::DT_Host;
- break;
- case OMPC_DEVICE_TYPE_nohost:
- DTCI.DT = OMPDeclareTargetDeclAttr::DT_NoHost;
- break;
- case OMPC_DEVICE_TYPE_unknown:
- llvm_unreachable("Unexpected device_type");
- }
- DeviceTypeLoc = DevTypeData->Loc;
- }
- continue;
- }
- ConsumeToken();
- }
- if (DTCI.Kind == OMPD_declare_target || HasIdentifier) {
- auto &&Callback = [this, MT, &DTCI](CXXScopeSpec &SS,
- DeclarationNameInfo NameInfo) {
- NamedDecl *ND =
- Actions.lookupOpenMPDeclareTargetName(getCurScope(), SS, NameInfo);
- if (!ND)
- return;
- Sema::DeclareTargetContextInfo::MapInfo MI{MT, NameInfo.getLoc()};
- bool FirstMapping = DTCI.ExplicitlyMapped.try_emplace(ND, MI).second;
- if (!FirstMapping)
- Diag(NameInfo.getLoc(), diag::err_omp_declare_target_multiple)
- << NameInfo.getName();
- };
- if (ParseOpenMPSimpleVarList(OMPD_declare_target, Callback,
- /*AllowScopeSpecifier=*/true))
- break;
- }
- if (Tok.is(tok::l_paren)) {
- Diag(Tok,
- diag::err_omp_begin_declare_target_unexpected_implicit_to_clause);
- break;
- }
- if (!HasIdentifier && Tok.isNot(tok::annot_pragma_openmp_end)) {
- Diag(Tok,
- getLangOpts().OpenMP >= 52
- ? diag::err_omp_declare_target_wrong_clause_after_implicit_enter
- : diag::err_omp_declare_target_wrong_clause_after_implicit_to);
- break;
- }
- // Consume optional ','.
- if (Tok.is(tok::comma))
- ConsumeToken();
- }
- if (DTCI.Indirect && DTCI.DT != OMPDeclareTargetDeclAttr::DT_Any)
- Diag(DeviceTypeLoc, diag::err_omp_declare_target_indirect_device_type);
- // For declare target require at least 'to' or 'link' to be present.
- if (DTCI.Kind == OMPD_declare_target && RequiresToOrLinkOrIndirectClause &&
- !HasToOrLinkOrIndirectClause)
- Diag(DTCI.Loc,
- getLangOpts().OpenMP >= 52
- ? diag::err_omp_declare_target_missing_enter_or_link_clause
- : diag::err_omp_declare_target_missing_to_or_link_clause)
- << (getLangOpts().OpenMP >= 51 ? 1 : 0);
- SkipUntil(tok::annot_pragma_openmp_end, StopBeforeMatch);
- }
- void Parser::skipUntilPragmaOpenMPEnd(OpenMPDirectiveKind DKind) {
- // The last seen token is annot_pragma_openmp_end - need to check for
- // extra tokens.
- if (Tok.is(tok::annot_pragma_openmp_end))
- return;
- Diag(Tok, diag::warn_omp_extra_tokens_at_eol)
- << getOpenMPDirectiveName(DKind);
- while (Tok.isNot(tok::annot_pragma_openmp_end))
- ConsumeAnyToken();
- }
- void Parser::parseOMPEndDirective(OpenMPDirectiveKind BeginKind,
- OpenMPDirectiveKind ExpectedKind,
- OpenMPDirectiveKind FoundKind,
- SourceLocation BeginLoc,
- SourceLocation FoundLoc,
- bool SkipUntilOpenMPEnd) {
- int DiagSelection = ExpectedKind == OMPD_end_declare_target ? 0 : 1;
- if (FoundKind == ExpectedKind) {
- ConsumeAnyToken();
- skipUntilPragmaOpenMPEnd(ExpectedKind);
- return;
- }
- Diag(FoundLoc, diag::err_expected_end_declare_target_or_variant)
- << DiagSelection;
- Diag(BeginLoc, diag::note_matching)
- << ("'#pragma omp " + getOpenMPDirectiveName(BeginKind) + "'").str();
- if (SkipUntilOpenMPEnd)
- SkipUntil(tok::annot_pragma_openmp_end, StopBeforeMatch);
- }
- void Parser::ParseOMPEndDeclareTargetDirective(OpenMPDirectiveKind BeginDKind,
- OpenMPDirectiveKind EndDKind,
- SourceLocation DKLoc) {
- parseOMPEndDirective(BeginDKind, OMPD_end_declare_target, EndDKind, DKLoc,
- Tok.getLocation(),
- /* SkipUntilOpenMPEnd */ false);
- // Skip the last annot_pragma_openmp_end.
- if (Tok.is(tok::annot_pragma_openmp_end))
- ConsumeAnnotationToken();
- }
- /// Parsing of declarative OpenMP directives.
- ///
- /// threadprivate-directive:
- /// annot_pragma_openmp 'threadprivate' simple-variable-list
- /// annot_pragma_openmp_end
- ///
- /// allocate-directive:
- /// annot_pragma_openmp 'allocate' simple-variable-list [<clause>]
- /// annot_pragma_openmp_end
- ///
- /// declare-reduction-directive:
- /// annot_pragma_openmp 'declare' 'reduction' [...]
- /// annot_pragma_openmp_end
- ///
- /// declare-mapper-directive:
- /// annot_pragma_openmp 'declare' 'mapper' '(' [<mapper-identifer> ':']
- /// <type> <var> ')' [<clause>[[,] <clause>] ... ]
- /// annot_pragma_openmp_end
- ///
- /// declare-simd-directive:
- /// annot_pragma_openmp 'declare simd' {<clause> [,]}
- /// annot_pragma_openmp_end
- /// <function declaration/definition>
- ///
- /// requires directive:
- /// annot_pragma_openmp 'requires' <clause> [[[,] <clause>] ... ]
- /// annot_pragma_openmp_end
- ///
- /// assumes directive:
- /// annot_pragma_openmp 'assumes' <clause> [[[,] <clause>] ... ]
- /// annot_pragma_openmp_end
- /// or
- /// annot_pragma_openmp 'begin assumes' <clause> [[[,] <clause>] ... ]
- /// annot_pragma_openmp 'end assumes'
- /// annot_pragma_openmp_end
- ///
- Parser::DeclGroupPtrTy Parser::ParseOpenMPDeclarativeDirectiveWithExtDecl(
- AccessSpecifier &AS, ParsedAttributes &Attrs, bool Delayed,
- DeclSpec::TST TagType, Decl *Tag) {
- assert(Tok.isOneOf(tok::annot_pragma_openmp, tok::annot_attr_openmp) &&
- "Not an OpenMP directive!");
- ParsingOpenMPDirectiveRAII DirScope(*this);
- ParenBraceBracketBalancer BalancerRAIIObj(*this);
- SourceLocation Loc;
- OpenMPDirectiveKind DKind;
- if (Delayed) {
- TentativeParsingAction TPA(*this);
- Loc = ConsumeAnnotationToken();
- DKind = parseOpenMPDirectiveKind(*this);
- if (DKind == OMPD_declare_reduction || DKind == OMPD_declare_mapper) {
- // Need to delay parsing until completion of the parent class.
- TPA.Revert();
- CachedTokens Toks;
- unsigned Cnt = 1;
- Toks.push_back(Tok);
- while (Cnt && Tok.isNot(tok::eof)) {
- (void)ConsumeAnyToken();
- if (Tok.isOneOf(tok::annot_pragma_openmp, tok::annot_attr_openmp))
- ++Cnt;
- else if (Tok.is(tok::annot_pragma_openmp_end))
- --Cnt;
- Toks.push_back(Tok);
- }
- // Skip last annot_pragma_openmp_end.
- if (Cnt == 0)
- (void)ConsumeAnyToken();
- auto *LP = new LateParsedPragma(this, AS);
- LP->takeToks(Toks);
- getCurrentClass().LateParsedDeclarations.push_back(LP);
- return nullptr;
- }
- TPA.Commit();
- } else {
- Loc = ConsumeAnnotationToken();
- DKind = parseOpenMPDirectiveKind(*this);
- }
- switch (DKind) {
- case OMPD_threadprivate: {
- ConsumeToken();
- DeclDirectiveListParserHelper Helper(this, DKind);
- if (!ParseOpenMPSimpleVarList(DKind, Helper,
- /*AllowScopeSpecifier=*/true)) {
- skipUntilPragmaOpenMPEnd(DKind);
- // Skip the last annot_pragma_openmp_end.
- ConsumeAnnotationToken();
- return Actions.ActOnOpenMPThreadprivateDirective(Loc,
- Helper.getIdentifiers());
- }
- break;
- }
- case OMPD_allocate: {
- ConsumeToken();
- DeclDirectiveListParserHelper Helper(this, DKind);
- if (!ParseOpenMPSimpleVarList(DKind, Helper,
- /*AllowScopeSpecifier=*/true)) {
- SmallVector<OMPClause *, 1> Clauses;
- if (Tok.isNot(tok::annot_pragma_openmp_end)) {
- SmallVector<llvm::PointerIntPair<OMPClause *, 1, bool>,
- llvm::omp::Clause_enumSize + 1>
- FirstClauses(llvm::omp::Clause_enumSize + 1);
- while (Tok.isNot(tok::annot_pragma_openmp_end)) {
- OpenMPClauseKind CKind =
- Tok.isAnnotation() ? OMPC_unknown
- : getOpenMPClauseKind(PP.getSpelling(Tok));
- Actions.StartOpenMPClause(CKind);
- OMPClause *Clause = ParseOpenMPClause(
- OMPD_allocate, CKind, !FirstClauses[unsigned(CKind)].getInt());
- SkipUntil(tok::comma, tok::identifier, tok::annot_pragma_openmp_end,
- StopBeforeMatch);
- FirstClauses[unsigned(CKind)].setInt(true);
- if (Clause != nullptr)
- Clauses.push_back(Clause);
- if (Tok.is(tok::annot_pragma_openmp_end)) {
- Actions.EndOpenMPClause();
- break;
- }
- // Skip ',' if any.
- if (Tok.is(tok::comma))
- ConsumeToken();
- Actions.EndOpenMPClause();
- }
- skipUntilPragmaOpenMPEnd(DKind);
- }
- // Skip the last annot_pragma_openmp_end.
- ConsumeAnnotationToken();
- return Actions.ActOnOpenMPAllocateDirective(Loc, Helper.getIdentifiers(),
- Clauses);
- }
- break;
- }
- case OMPD_requires: {
- SourceLocation StartLoc = ConsumeToken();
- SmallVector<OMPClause *, 5> Clauses;
- SmallVector<llvm::PointerIntPair<OMPClause *, 1, bool>,
- llvm::omp::Clause_enumSize + 1>
- FirstClauses(llvm::omp::Clause_enumSize + 1);
- if (Tok.is(tok::annot_pragma_openmp_end)) {
- Diag(Tok, diag::err_omp_expected_clause)
- << getOpenMPDirectiveName(OMPD_requires);
- break;
- }
- while (Tok.isNot(tok::annot_pragma_openmp_end)) {
- OpenMPClauseKind CKind = Tok.isAnnotation()
- ? OMPC_unknown
- : getOpenMPClauseKind(PP.getSpelling(Tok));
- Actions.StartOpenMPClause(CKind);
- OMPClause *Clause = ParseOpenMPClause(
- OMPD_requires, CKind, !FirstClauses[unsigned(CKind)].getInt());
- SkipUntil(tok::comma, tok::identifier, tok::annot_pragma_openmp_end,
- StopBeforeMatch);
- FirstClauses[unsigned(CKind)].setInt(true);
- if (Clause != nullptr)
- Clauses.push_back(Clause);
- if (Tok.is(tok::annot_pragma_openmp_end)) {
- Actions.EndOpenMPClause();
- break;
- }
- // Skip ',' if any.
- if (Tok.is(tok::comma))
- ConsumeToken();
- Actions.EndOpenMPClause();
- }
- // Consume final annot_pragma_openmp_end
- if (Clauses.empty()) {
- Diag(Tok, diag::err_omp_expected_clause)
- << getOpenMPDirectiveName(OMPD_requires);
- ConsumeAnnotationToken();
- return nullptr;
- }
- ConsumeAnnotationToken();
- return Actions.ActOnOpenMPRequiresDirective(StartLoc, Clauses);
- }
- case OMPD_error: {
- SmallVector<OMPClause *, 1> Clauses;
- SourceLocation StartLoc = ConsumeToken();
- ParseOpenMPClauses(DKind, Clauses, StartLoc);
- Actions.ActOnOpenMPErrorDirective(Clauses, StartLoc, SourceLocation(),
- /*InExContext = */ false);
- break;
- }
- case OMPD_assumes:
- case OMPD_begin_assumes:
- ParseOpenMPAssumesDirective(DKind, ConsumeToken());
- break;
- case OMPD_end_assumes:
- ParseOpenMPEndAssumesDirective(ConsumeToken());
- break;
- case OMPD_declare_reduction:
- ConsumeToken();
- if (DeclGroupPtrTy Res = ParseOpenMPDeclareReductionDirective(AS)) {
- skipUntilPragmaOpenMPEnd(OMPD_declare_reduction);
- // Skip the last annot_pragma_openmp_end.
- ConsumeAnnotationToken();
- return Res;
- }
- break;
- case OMPD_declare_mapper: {
- ConsumeToken();
- if (DeclGroupPtrTy Res = ParseOpenMPDeclareMapperDirective(AS)) {
- // Skip the last annot_pragma_openmp_end.
- ConsumeAnnotationToken();
- return Res;
- }
- break;
- }
- case OMPD_begin_declare_variant: {
- // The syntax is:
- // { #pragma omp begin declare variant clause }
- // <function-declaration-or-definition-sequence>
- // { #pragma omp end declare variant }
- //
- ConsumeToken();
- OMPTraitInfo *ParentTI = Actions.getOMPTraitInfoForSurroundingScope();
- ASTContext &ASTCtx = Actions.getASTContext();
- OMPTraitInfo &TI = ASTCtx.getNewOMPTraitInfo();
- if (parseOMPDeclareVariantMatchClause(Loc, TI, ParentTI)) {
- while (!SkipUntil(tok::annot_pragma_openmp_end, Parser::StopBeforeMatch))
- ;
- // Skip the last annot_pragma_openmp_end.
- (void)ConsumeAnnotationToken();
- break;
- }
- // Skip last tokens.
- skipUntilPragmaOpenMPEnd(OMPD_begin_declare_variant);
- ParsingOpenMPDirectiveRAII NormalScope(*this, /*Value=*/false);
- VariantMatchInfo VMI;
- TI.getAsVariantMatchInfo(ASTCtx, VMI);
- std::function<void(StringRef)> DiagUnknownTrait =
- [this, Loc](StringRef ISATrait) {
- // TODO Track the selector locations in a way that is accessible here
- // to improve the diagnostic location.
- Diag(Loc, diag::warn_unknown_declare_variant_isa_trait) << ISATrait;
- };
- TargetOMPContext OMPCtx(
- ASTCtx, std::move(DiagUnknownTrait),
- /* CurrentFunctionDecl */ nullptr,
- /* ConstructTraits */ ArrayRef<llvm::omp::TraitProperty>());
- if (isVariantApplicableInContext(VMI, OMPCtx, /* DeviceSetOnly */ true)) {
- Actions.ActOnOpenMPBeginDeclareVariant(Loc, TI);
- break;
- }
- // Elide all the code till the matching end declare variant was found.
- unsigned Nesting = 1;
- SourceLocation DKLoc;
- OpenMPDirectiveKind DK = OMPD_unknown;
- do {
- DKLoc = Tok.getLocation();
- DK = parseOpenMPDirectiveKind(*this);
- if (DK == OMPD_end_declare_variant)
- --Nesting;
- else if (DK == OMPD_begin_declare_variant)
- ++Nesting;
- if (!Nesting || isEofOrEom())
- break;
- ConsumeAnyToken();
- } while (true);
- parseOMPEndDirective(OMPD_begin_declare_variant, OMPD_end_declare_variant,
- DK, Loc, DKLoc, /* SkipUntilOpenMPEnd */ true);
- if (isEofOrEom())
- return nullptr;
- break;
- }
- case OMPD_end_declare_variant: {
- if (Actions.isInOpenMPDeclareVariantScope())
- Actions.ActOnOpenMPEndDeclareVariant();
- else
- Diag(Loc, diag::err_expected_begin_declare_variant);
- ConsumeToken();
- break;
- }
- case OMPD_declare_variant:
- case OMPD_declare_simd: {
- // The syntax is:
- // { #pragma omp declare {simd|variant} }
- // <function-declaration-or-definition>
- //
- CachedTokens Toks;
- Toks.push_back(Tok);
- ConsumeToken();
- while (Tok.isNot(tok::annot_pragma_openmp_end)) {
- Toks.push_back(Tok);
- ConsumeAnyToken();
- }
- Toks.push_back(Tok);
- ConsumeAnyToken();
- DeclGroupPtrTy Ptr;
- if (Tok.isOneOf(tok::annot_pragma_openmp, tok::annot_attr_openmp)) {
- Ptr = ParseOpenMPDeclarativeDirectiveWithExtDecl(AS, Attrs, Delayed,
- 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);
- ParsedAttributes EmptyDeclSpecAttrs(AttrFactory);
- MaybeParseCXX11Attributes(Attrs);
- ParsingDeclSpec PDS(*this);
- Ptr = ParseExternalDeclaration(Attrs, EmptyDeclSpecAttrs, &PDS);
- } else {
- Ptr =
- ParseCXXClassMemberDeclarationWithPragmas(AS, Attrs, TagType, Tag);
- }
- }
- if (!Ptr) {
- Diag(Loc, diag::err_omp_decl_in_declare_simd_variant)
- << (DKind == OMPD_declare_simd ? 0 : 1);
- return DeclGroupPtrTy();
- }
- if (DKind == OMPD_declare_simd)
- return ParseOMPDeclareSimdClauses(Ptr, Toks, Loc);
- assert(DKind == OMPD_declare_variant &&
- "Expected declare variant directive only");
- ParseOMPDeclareVariantClauses(Ptr, Toks, Loc);
- return Ptr;
- }
- case OMPD_begin_declare_target:
- case OMPD_declare_target: {
- SourceLocation DTLoc = ConsumeAnyToken();
- bool HasClauses = Tok.isNot(tok::annot_pragma_openmp_end);
- Sema::DeclareTargetContextInfo DTCI(DKind, DTLoc);
- if (HasClauses)
- ParseOMPDeclareTargetClauses(DTCI);
- bool HasImplicitMappings = DKind == OMPD_begin_declare_target ||
- !HasClauses ||
- (DTCI.ExplicitlyMapped.empty() && DTCI.Indirect);
- // Skip the last annot_pragma_openmp_end.
- ConsumeAnyToken();
- if (HasImplicitMappings) {
- Actions.ActOnStartOpenMPDeclareTargetContext(DTCI);
- return nullptr;
- }
- Actions.ActOnFinishedOpenMPDeclareTargetContext(DTCI);
- llvm::SmallVector<Decl *, 4> Decls;
- for (auto &It : DTCI.ExplicitlyMapped)
- Decls.push_back(It.first);
- return Actions.BuildDeclaratorGroup(Decls);
- }
- case OMPD_end_declare_target: {
- if (!Actions.isInOpenMPDeclareTargetContext()) {
- Diag(Tok, diag::err_omp_unexpected_directive)
- << 1 << getOpenMPDirectiveName(DKind);
- break;
- }
- const Sema::DeclareTargetContextInfo &DTCI =
- Actions.ActOnOpenMPEndDeclareTargetDirective();
- ParseOMPEndDeclareTargetDirective(DTCI.Kind, DKind, DTCI.Loc);
- return nullptr;
- }
- case OMPD_unknown:
- Diag(Tok, diag::err_omp_unknown_directive);
- break;
- case OMPD_parallel:
- case OMPD_simd:
- case OMPD_tile:
- case OMPD_unroll:
- case OMPD_task:
- case OMPD_taskyield:
- case OMPD_barrier:
- case OMPD_taskwait:
- case OMPD_taskgroup:
- case OMPD_flush:
- case OMPD_depobj:
- case OMPD_scan:
- 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_parallel_master:
- case OMPD_parallel_masked:
- 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_master_taskloop:
- case OMPD_master_taskloop_simd:
- case OMPD_parallel_master_taskloop:
- case OMPD_parallel_master_taskloop_simd:
- case OMPD_masked_taskloop:
- case OMPD_masked_taskloop_simd:
- case OMPD_parallel_masked_taskloop:
- case OMPD_parallel_masked_taskloop_simd:
- case OMPD_distribute:
- 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:
- case OMPD_dispatch:
- case OMPD_masked:
- case OMPD_metadirective:
- case OMPD_loop:
- case OMPD_teams_loop:
- case OMPD_target_teams_loop:
- case OMPD_parallel_loop:
- case OMPD_target_parallel_loop:
- Diag(Tok, diag::err_omp_unexpected_directive)
- << 1 << getOpenMPDirectiveName(DKind);
- break;
- default:
- break;
- }
- while (Tok.isNot(tok::annot_pragma_openmp_end))
- ConsumeAnyToken();
- ConsumeAnyToken();
- return nullptr;
- }
- /// Parsing of declarative or executable OpenMP directives.
- ///
- /// threadprivate-directive:
- /// annot_pragma_openmp 'threadprivate' simple-variable-list
- /// annot_pragma_openmp_end
- ///
- /// allocate-directive:
- /// annot_pragma_openmp 'allocate' 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
- ///
- /// declare-mapper-directive:
- /// annot_pragma_openmp 'declare' 'mapper' '(' [<mapper-identifer> ':']
- /// <type> <var> ')' [<clause>[[,] <clause>] ... ]
- /// annot_pragma_openmp_end
- ///
- /// executable-directive:
- /// annot_pragma_openmp 'parallel' | 'simd' | 'for' | 'sections' |
- /// 'section' | 'single' | 'master' | 'critical' [ '(' <name> ')' ] |
- /// 'parallel for' | 'parallel sections' | 'parallel master' | 'task' |
- /// 'taskyield' | 'barrier' | 'taskwait' | 'flush' | 'ordered' | 'error'
- /// | 'atomic' | 'for simd' | 'parallel for simd' | 'target' | 'target
- /// data' | 'taskgroup' | 'teams' | 'taskloop' | 'taskloop simd' |
- /// 'master taskloop' | 'master taskloop simd' | 'parallel master
- /// taskloop' | 'parallel master 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' | 'masked' {clause}
- /// annot_pragma_openmp_end
- ///
- StmtResult Parser::ParseOpenMPDeclarativeOrExecutableDirective(
- ParsedStmtContext StmtCtx, bool ReadDirectiveWithinMetadirective) {
- if (!ReadDirectiveWithinMetadirective)
- assert(Tok.isOneOf(tok::annot_pragma_openmp, tok::annot_attr_openmp) &&
- "Not an OpenMP directive!");
- ParsingOpenMPDirectiveRAII DirScope(*this);
- ParenBraceBracketBalancer BalancerRAIIObj(*this);
- SmallVector<OMPClause *, 5> Clauses;
- SmallVector<llvm::PointerIntPair<OMPClause *, 1, bool>,
- llvm::omp::Clause_enumSize + 1>
- FirstClauses(llvm::omp::Clause_enumSize + 1);
- unsigned ScopeFlags = Scope::FnScope | Scope::DeclScope |
- Scope::CompoundStmtScope | Scope::OpenMPDirectiveScope;
- SourceLocation Loc = ReadDirectiveWithinMetadirective
- ? Tok.getLocation()
- : ConsumeAnnotationToken(),
- EndLoc;
- OpenMPDirectiveKind DKind = parseOpenMPDirectiveKind(*this);
- if (ReadDirectiveWithinMetadirective && DKind == OMPD_unknown) {
- Diag(Tok, diag::err_omp_unknown_directive);
- return StmtError();
- }
- OpenMPDirectiveKind CancelRegion = OMPD_unknown;
- // Name of critical directive.
- DeclarationNameInfo DirName;
- StmtResult Directive = StmtError();
- bool HasAssociatedStatement = true;
- switch (DKind) {
- case OMPD_nothing:
- if ((StmtCtx & ParsedStmtContext::AllowStandaloneOpenMPDirectives) ==
- ParsedStmtContext())
- Diag(Tok, diag::err_omp_immediate_directive)
- << getOpenMPDirectiveName(DKind) << 0;
- ConsumeToken();
- skipUntilPragmaOpenMPEnd(DKind);
- if (Tok.is(tok::annot_pragma_openmp_end))
- ConsumeAnnotationToken();
- break;
- case OMPD_metadirective: {
- ConsumeToken();
- SmallVector<VariantMatchInfo, 4> VMIs;
- // First iteration of parsing all clauses of metadirective.
- // This iteration only parses and collects all context selector ignoring the
- // associated directives.
- TentativeParsingAction TPA(*this);
- ASTContext &ASTContext = Actions.getASTContext();
- BalancedDelimiterTracker T(*this, tok::l_paren,
- tok::annot_pragma_openmp_end);
- while (Tok.isNot(tok::annot_pragma_openmp_end)) {
- OpenMPClauseKind CKind = Tok.isAnnotation()
- ? OMPC_unknown
- : getOpenMPClauseKind(PP.getSpelling(Tok));
- SourceLocation Loc = ConsumeToken();
- // Parse '('.
- if (T.expectAndConsume(diag::err_expected_lparen_after,
- getOpenMPClauseName(CKind).data()))
- return Directive;
- OMPTraitInfo &TI = Actions.getASTContext().getNewOMPTraitInfo();
- if (CKind == OMPC_when) {
- // parse and get OMPTraitInfo to pass to the When clause
- parseOMPContextSelectors(Loc, TI);
- if (TI.Sets.size() == 0) {
- Diag(Tok, diag::err_omp_expected_context_selector) << "when clause";
- TPA.Commit();
- return Directive;
- }
- // Parse ':'
- if (Tok.is(tok::colon))
- ConsumeAnyToken();
- else {
- Diag(Tok, diag::err_omp_expected_colon) << "when clause";
- TPA.Commit();
- return Directive;
- }
- }
- // Skip Directive for now. We will parse directive in the second iteration
- int paren = 0;
- while (Tok.isNot(tok::r_paren) || paren != 0) {
- if (Tok.is(tok::l_paren))
- paren++;
- if (Tok.is(tok::r_paren))
- paren--;
- if (Tok.is(tok::annot_pragma_openmp_end)) {
- Diag(Tok, diag::err_omp_expected_punc)
- << getOpenMPClauseName(CKind) << 0;
- TPA.Commit();
- return Directive;
- }
- ConsumeAnyToken();
- }
- // Parse ')'
- if (Tok.is(tok::r_paren))
- T.consumeClose();
- VariantMatchInfo VMI;
- TI.getAsVariantMatchInfo(ASTContext, VMI);
- VMIs.push_back(VMI);
- }
- TPA.Revert();
- // End of the first iteration. Parser is reset to the start of metadirective
- std::function<void(StringRef)> DiagUnknownTrait =
- [this, Loc](StringRef ISATrait) {
- // TODO Track the selector locations in a way that is accessible here
- // to improve the diagnostic location.
- Diag(Loc, diag::warn_unknown_declare_variant_isa_trait) << ISATrait;
- };
- TargetOMPContext OMPCtx(ASTContext, std::move(DiagUnknownTrait),
- /* CurrentFunctionDecl */ nullptr,
- ArrayRef<llvm::omp::TraitProperty>());
- // A single match is returned for OpenMP 5.0
- int BestIdx = getBestVariantMatchForContext(VMIs, OMPCtx);
- int Idx = 0;
- // In OpenMP 5.0 metadirective is either replaced by another directive or
- // ignored.
- // TODO: In OpenMP 5.1 generate multiple directives based upon the matches
- // found by getBestWhenMatchForContext.
- while (Tok.isNot(tok::annot_pragma_openmp_end)) {
- // OpenMP 5.0 implementation - Skip to the best index found.
- if (Idx++ != BestIdx) {
- ConsumeToken(); // Consume clause name
- T.consumeOpen(); // Consume '('
- int paren = 0;
- // Skip everything inside the clause
- while (Tok.isNot(tok::r_paren) || paren != 0) {
- if (Tok.is(tok::l_paren))
- paren++;
- if (Tok.is(tok::r_paren))
- paren--;
- ConsumeAnyToken();
- }
- // Parse ')'
- if (Tok.is(tok::r_paren))
- T.consumeClose();
- continue;
- }
- OpenMPClauseKind CKind = Tok.isAnnotation()
- ? OMPC_unknown
- : getOpenMPClauseKind(PP.getSpelling(Tok));
- SourceLocation Loc = ConsumeToken();
- // Parse '('.
- T.consumeOpen();
- // Skip ContextSelectors for when clause
- if (CKind == OMPC_when) {
- OMPTraitInfo &TI = Actions.getASTContext().getNewOMPTraitInfo();
- // parse and skip the ContextSelectors
- parseOMPContextSelectors(Loc, TI);
- // Parse ':'
- ConsumeAnyToken();
- }
- // If no directive is passed, skip in OpenMP 5.0.
- // TODO: Generate nothing directive from OpenMP 5.1.
- if (Tok.is(tok::r_paren)) {
- SkipUntil(tok::annot_pragma_openmp_end);
- break;
- }
- // Parse Directive
- Directive = ParseOpenMPDeclarativeOrExecutableDirective(
- StmtCtx,
- /*ReadDirectiveWithinMetadirective=*/true);
- break;
- }
- break;
- }
- case OMPD_threadprivate: {
- // FIXME: Should this be permitted in C++?
- if ((StmtCtx & ParsedStmtContext::AllowDeclarationsInC) ==
- ParsedStmtContext()) {
- Diag(Tok, diag::err_omp_immediate_directive)
- << getOpenMPDirectiveName(DKind) << 0;
- }
- ConsumeToken();
- DeclDirectiveListParserHelper Helper(this, DKind);
- if (!ParseOpenMPSimpleVarList(DKind, Helper,
- /*AllowScopeSpecifier=*/false)) {
- skipUntilPragmaOpenMPEnd(DKind);
- DeclGroupPtrTy Res = Actions.ActOnOpenMPThreadprivateDirective(
- Loc, Helper.getIdentifiers());
- Directive = Actions.ActOnDeclStmt(Res, Loc, Tok.getLocation());
- }
- SkipUntil(tok::annot_pragma_openmp_end);
- break;
- }
- case OMPD_allocate: {
- // FIXME: Should this be permitted in C++?
- if ((StmtCtx & ParsedStmtContext::AllowDeclarationsInC) ==
- ParsedStmtContext()) {
- Diag(Tok, diag::err_omp_immediate_directive)
- << getOpenMPDirectiveName(DKind) << 0;
- }
- ConsumeToken();
- DeclDirectiveListParserHelper Helper(this, DKind);
- if (!ParseOpenMPSimpleVarList(DKind, Helper,
- /*AllowScopeSpecifier=*/false)) {
- SmallVector<OMPClause *, 1> Clauses;
- if (Tok.isNot(tok::annot_pragma_openmp_end)) {
- SmallVector<llvm::PointerIntPair<OMPClause *, 1, bool>,
- llvm::omp::Clause_enumSize + 1>
- FirstClauses(llvm::omp::Clause_enumSize + 1);
- while (Tok.isNot(tok::annot_pragma_openmp_end)) {
- OpenMPClauseKind CKind =
- Tok.isAnnotation() ? OMPC_unknown
- : getOpenMPClauseKind(PP.getSpelling(Tok));
- Actions.StartOpenMPClause(CKind);
- OMPClause *Clause = ParseOpenMPClause(
- OMPD_allocate, CKind, !FirstClauses[unsigned(CKind)].getInt());
- SkipUntil(tok::comma, tok::identifier, tok::annot_pragma_openmp_end,
- StopBeforeMatch);
- FirstClauses[unsigned(CKind)].setInt(true);
- if (Clause != nullptr)
- Clauses.push_back(Clause);
- if (Tok.is(tok::annot_pragma_openmp_end)) {
- Actions.EndOpenMPClause();
- break;
- }
- // Skip ',' if any.
- if (Tok.is(tok::comma))
- ConsumeToken();
- Actions.EndOpenMPClause();
- }
- skipUntilPragmaOpenMPEnd(DKind);
- }
- DeclGroupPtrTy Res = Actions.ActOnOpenMPAllocateDirective(
- Loc, Helper.getIdentifiers(), Clauses);
- 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)) {
- skipUntilPragmaOpenMPEnd(OMPD_declare_reduction);
- ConsumeAnyToken();
- Directive = Actions.ActOnDeclStmt(Res, Loc, Tok.getLocation());
- } else {
- SkipUntil(tok::annot_pragma_openmp_end);
- }
- break;
- case OMPD_declare_mapper: {
- ConsumeToken();
- if (DeclGroupPtrTy Res =
- ParseOpenMPDeclareMapperDirective(/*AS=*/AS_none)) {
- // Skip the last annot_pragma_openmp_end.
- ConsumeAnnotationToken();
- Directive = Actions.ActOnDeclStmt(Res, Loc, Tok.getLocation());
- } else {
- SkipUntil(tok::annot_pragma_openmp_end);
- }
- break;
- }
- case OMPD_flush:
- case OMPD_depobj:
- case OMPD_scan:
- case OMPD_taskyield:
- case OMPD_error:
- 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:
- case OMPD_interop:
- if ((StmtCtx & ParsedStmtContext::AllowStandaloneOpenMPDirectives) ==
- ParsedStmtContext()) {
- Diag(Tok, diag::err_omp_immediate_directive)
- << getOpenMPDirectiveName(DKind) << 0;
- if (DKind == OMPD_error) {
- SkipUntil(tok::annot_pragma_openmp_end);
- break;
- }
- }
- HasAssociatedStatement = false;
- // Fall through for further analysis.
- [[fallthrough]];
- case OMPD_parallel:
- case OMPD_simd:
- case OMPD_tile:
- case OMPD_unroll:
- 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_parallel_master:
- case OMPD_parallel_masked:
- 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_loop:
- case OMPD_teams_loop:
- case OMPD_target_teams_loop:
- case OMPD_parallel_loop:
- case OMPD_target_parallel_loop:
- case OMPD_taskloop:
- case OMPD_taskloop_simd:
- case OMPD_master_taskloop:
- case OMPD_masked_taskloop:
- case OMPD_master_taskloop_simd:
- case OMPD_masked_taskloop_simd:
- case OMPD_parallel_master_taskloop:
- case OMPD_parallel_masked_taskloop:
- case OMPD_parallel_master_taskloop_simd:
- case OMPD_parallel_masked_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:
- case OMPD_dispatch:
- case OMPD_masked: {
- // Special processing for flush and depobj clauses.
- Token ImplicitTok;
- bool ImplicitClauseAllowed = false;
- if (DKind == OMPD_flush || DKind == OMPD_depobj) {
- ImplicitTok = Tok;
- ImplicitClauseAllowed = true;
- }
- 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)) {
- // If we are parsing for a directive within a metadirective, the directive
- // ends with a ')'.
- if (ReadDirectiveWithinMetadirective && Tok.is(tok::r_paren)) {
- while (Tok.isNot(tok::annot_pragma_openmp_end))
- ConsumeAnyToken();
- break;
- }
- bool HasImplicitClause = false;
- if (ImplicitClauseAllowed && Tok.is(tok::l_paren)) {
- HasImplicitClause = true;
- // Push copy of the current token back to stream to properly parse
- // pseudo-clause OMPFlushClause or OMPDepobjClause.
- PP.EnterToken(Tok, /*IsReinject*/ true);
- PP.EnterToken(ImplicitTok, /*IsReinject*/ true);
- ConsumeAnyToken();
- }
- OpenMPClauseKind CKind = Tok.isAnnotation()
- ? OMPC_unknown
- : getOpenMPClauseKind(PP.getSpelling(Tok));
- if (HasImplicitClause) {
- assert(CKind == OMPC_unknown && "Must be unknown implicit clause.");
- if (DKind == OMPD_flush) {
- CKind = OMPC_flush;
- } else {
- assert(DKind == OMPD_depobj &&
- "Expected flush or depobj directives.");
- CKind = OMPC_depobj;
- }
- }
- // No more implicit clauses allowed.
- ImplicitClauseAllowed = false;
- Actions.StartOpenMPClause(CKind);
- HasImplicitClause = false;
- OMPClause *Clause = ParseOpenMPClause(
- DKind, CKind, !FirstClauses[unsigned(CKind)].getInt());
- FirstClauses[unsigned(CKind)].setInt(true);
- if (Clause) {
- FirstClauses[unsigned(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[unsigned(OMPC_depend)].getInt()) {
- if ((StmtCtx & ParsedStmtContext::AllowStandaloneOpenMPDirectives) ==
- ParsedStmtContext()) {
- Diag(Loc, diag::err_omp_immediate_directive)
- << getOpenMPDirectiveName(DKind) << 1
- << getOpenMPClauseName(OMPC_depend);
- }
- HasAssociatedStatement = false;
- }
- if (DKind == OMPD_tile && !FirstClauses[unsigned(OMPC_sizes)].getInt()) {
- Diag(Loc, diag::err_omp_required_clause)
- << getOpenMPDirectiveName(OMPD_tile) << "sizes";
- }
- 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.
- ParsingOpenMPDirectiveRAII NormalScope(*this, /*Value=*/false);
- {
- Sema::CompoundScopeRAII Scope(Actions);
- AssociatedStmt = ParseStatement();
- if (AssociatedStmt.isUsable() && isOpenMPLoopDirective(DKind) &&
- getLangOpts().OpenMPIRBuilder)
- AssociatedStmt = Actions.ActOnOpenMPLoopnest(AssociatedStmt.get());
- }
- 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, std::nullopt,
- /*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_begin_declare_target:
- case OMPD_end_declare_target:
- case OMPD_requires:
- case OMPD_begin_declare_variant:
- case OMPD_end_declare_variant:
- case OMPD_declare_variant:
- Diag(Tok, diag::err_omp_unexpected_directive)
- << 1 << getOpenMPDirectiveName(DKind);
- SkipUntil(tok::annot_pragma_openmp_end);
- break;
- case OMPD_unknown:
- default:
- 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).data()))
- 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, /*ObjectType=*/nullptr,
- /*ObjectHasErrors=*/false, false)) {
- IsCorrect = false;
- SkipUntil(tok::comma, tok::r_paren, tok::annot_pragma_openmp_end,
- StopBeforeMatch);
- } else if (ParseUnqualifiedId(SS, /*ObjectType=*/nullptr,
- /*ObjectHadErrors=*/false, false, false,
- false, false, 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;
- }
- OMPClause *Parser::ParseOpenMPSizesClause() {
- SourceLocation ClauseNameLoc = ConsumeToken();
- SmallVector<Expr *, 4> ValExprs;
- BalancedDelimiterTracker T(*this, tok::l_paren, tok::annot_pragma_openmp_end);
- if (T.consumeOpen()) {
- Diag(Tok, diag::err_expected) << tok::l_paren;
- return nullptr;
- }
- while (true) {
- ExprResult Val = ParseConstantExpression();
- if (!Val.isUsable()) {
- T.skipToEnd();
- return nullptr;
- }
- ValExprs.push_back(Val.get());
- if (Tok.is(tok::r_paren) || Tok.is(tok::annot_pragma_openmp_end))
- break;
- ExpectAndConsume(tok::comma);
- }
- T.consumeClose();
- return Actions.ActOnOpenMPSizesClause(
- ValExprs, ClauseNameLoc, T.getOpenLocation(), T.getCloseLocation());
- }
- OMPClause *Parser::ParseOpenMPUsesAllocatorClause(OpenMPDirectiveKind DKind) {
- SourceLocation Loc = Tok.getLocation();
- ConsumeAnyToken();
- // Parse '('.
- BalancedDelimiterTracker T(*this, tok::l_paren, tok::annot_pragma_openmp_end);
- if (T.expectAndConsume(diag::err_expected_lparen_after, "uses_allocator"))
- return nullptr;
- SmallVector<Sema::UsesAllocatorsData, 4> Data;
- do {
- ExprResult Allocator =
- getLangOpts().CPlusPlus ? ParseCXXIdExpression() : ParseExpression();
- if (Allocator.isInvalid()) {
- SkipUntil(tok::comma, tok::r_paren, tok::annot_pragma_openmp_end,
- StopBeforeMatch);
- break;
- }
- Sema::UsesAllocatorsData &D = Data.emplace_back();
- D.Allocator = Allocator.get();
- if (Tok.is(tok::l_paren)) {
- BalancedDelimiterTracker T(*this, tok::l_paren,
- tok::annot_pragma_openmp_end);
- T.consumeOpen();
- ExprResult AllocatorTraits =
- getLangOpts().CPlusPlus ? ParseCXXIdExpression() : ParseExpression();
- T.consumeClose();
- if (AllocatorTraits.isInvalid()) {
- SkipUntil(tok::comma, tok::r_paren, tok::annot_pragma_openmp_end,
- StopBeforeMatch);
- break;
- }
- D.AllocatorTraits = AllocatorTraits.get();
- D.LParenLoc = T.getOpenLocation();
- D.RParenLoc = T.getCloseLocation();
- }
- if (Tok.isNot(tok::comma) && Tok.isNot(tok::r_paren))
- Diag(Tok, diag::err_omp_expected_punc) << "uses_allocators" << 0;
- // Parse ','
- if (Tok.is(tok::comma))
- ConsumeAnyToken();
- } while (Tok.isNot(tok::r_paren) && Tok.isNot(tok::annot_pragma_openmp_end));
- T.consumeClose();
- return Actions.ActOnOpenMPUsesAllocatorClause(Loc, T.getOpenLocation(),
- T.getCloseLocation(), Data);
- }
- /// 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 | bind-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 | allocator-clause | allocate-clause |
- /// acq_rel-clause | acquire-clause | release-clause | relaxed-clause |
- /// depobj-clause | destroy-clause | detach-clause | inclusive-clause |
- /// exclusive-clause | uses_allocators-clause | use_device_addr-clause |
- /// has_device_addr
- ///
- OMPClause *Parser::ParseOpenMPClause(OpenMPDirectiveKind DKind,
- OpenMPClauseKind CKind, bool FirstClause) {
- OMPClauseKind = CKind;
- 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, getLangOpts().OpenMP)) {
- 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_num_teams:
- case OMPC_thread_limit:
- case OMPC_priority:
- case OMPC_grainsize:
- case OMPC_num_tasks:
- case OMPC_hint:
- case OMPC_allocator:
- case OMPC_depobj:
- case OMPC_detach:
- case OMPC_novariants:
- case OMPC_nocontext:
- case OMPC_filter:
- case OMPC_partial:
- case OMPC_align:
- case OMPC_message:
- case OMPC_ompx_dyn_cgroup_mem:
- // 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.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.
- // OpenMP [2.11.3, allocate Directive, Restrictions]
- // At most one allocator clause can appear on the directive.
- // OpenMP 5.0, 2.10.1 task Construct, Restrictions.
- // At most one detach clause can appear on the directive.
- // OpenMP 5.1, 2.3.6 dispatch Construct, Restrictions.
- // At most one novariants clause can appear on a dispatch directive.
- // At most one nocontext clause can appear on a dispatch directive.
- // OpenMP [5.1, error directive, Restrictions]
- // At most one message 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 || CKind == OMPC_partial) &&
- PP.LookAhead(/*N=*/0).isNot(tok::l_paren))
- Clause = ParseOpenMPClause(CKind, WrongDirective);
- else if (CKind == OMPC_grainsize || CKind == OMPC_num_tasks)
- Clause = ParseOpenMPSingleExprWithArgClause(DKind, CKind, WrongDirective);
- else
- Clause = ParseOpenMPSingleExprClause(CKind, WrongDirective);
- break;
- case OMPC_default:
- case OMPC_proc_bind:
- case OMPC_atomic_default_mem_order:
- case OMPC_at:
- case OMPC_severity:
- case OMPC_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.
- // OpenMP [5.0, Requires directive, Restrictions]
- // At most one atomic_default_mem_order clause can appear
- // on the directive
- // OpenMP [5.1, error directive, Restrictions]
- // At most one at clause can appear on the directive
- // At most one severity clause can appear on the directive
- // OpenMP 5.1, 2.11.7 loop Construct, Restrictions.
- // At most one bind clause can appear on a loop 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_device:
- case OMPC_schedule:
- case OMPC_dist_schedule:
- case OMPC_defaultmap:
- case OMPC_order:
- // OpenMP [2.7.1, Restrictions, p. 3]
- // Only one schedule clause can appear on a loop directive.
- // OpenMP 4.5 [2.10.4, Restrictions, p. 106]
- // At most one defaultmap clause can appear on the directive.
- // OpenMP 5.0 [2.12.5, target construct, Restrictions]
- // At most one device clause can appear on the directive.
- // OpenMP 5.1 [2.11.3, order clause, Restrictions]
- // At most one order clause may appear on a construct.
- if ((getLangOpts().OpenMP < 50 || CKind != OMPC_defaultmap) &&
- (CKind != OMPC_order || getLangOpts().OpenMP >= 51) && !FirstClause) {
- Diag(Tok, diag::err_omp_more_one_clause)
- << getOpenMPDirectiveName(DKind) << getOpenMPClauseName(CKind) << 0;
- ErrorFound = true;
- }
- [[fallthrough]];
- case OMPC_if:
- Clause = ParseOpenMPSingleExprWithArgClause(DKind, CKind, WrongDirective);
- break;
- case OMPC_nowait:
- case OMPC_untied:
- case OMPC_mergeable:
- case OMPC_read:
- case OMPC_write:
- case OMPC_capture:
- case OMPC_compare:
- case OMPC_seq_cst:
- case OMPC_acq_rel:
- case OMPC_acquire:
- case OMPC_release:
- case OMPC_relaxed:
- case OMPC_threads:
- case OMPC_simd:
- case OMPC_nogroup:
- case OMPC_unified_address:
- case OMPC_unified_shared_memory:
- case OMPC_reverse_offload:
- case OMPC_dynamic_allocators:
- case OMPC_full:
- // 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.
- // OpenMP [5.0, Requires directive, Restrictions]
- // Each of the requires clauses can appear at most once on the 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_update:
- if (!FirstClause) {
- Diag(Tok, diag::err_omp_more_one_clause)
- << getOpenMPDirectiveName(DKind) << getOpenMPClauseName(CKind) << 0;
- ErrorFound = true;
- }
- Clause = (DKind == OMPD_depobj)
- ? ParseOpenMPSimpleClause(CKind, WrongDirective)
- : 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_use_device_addr:
- case OMPC_is_device_ptr:
- case OMPC_has_device_addr:
- case OMPC_allocate:
- case OMPC_nontemporal:
- case OMPC_inclusive:
- case OMPC_exclusive:
- case OMPC_affinity:
- Clause = ParseOpenMPVarListClause(DKind, CKind, WrongDirective);
- break;
- case OMPC_sizes:
- if (!FirstClause) {
- Diag(Tok, diag::err_omp_more_one_clause)
- << getOpenMPDirectiveName(DKind) << getOpenMPClauseName(CKind) << 0;
- ErrorFound = true;
- }
- Clause = ParseOpenMPSizesClause();
- break;
- case OMPC_uses_allocators:
- Clause = ParseOpenMPUsesAllocatorClause(DKind);
- break;
- case OMPC_destroy:
- if (DKind != OMPD_interop) {
- if (!FirstClause) {
- Diag(Tok, diag::err_omp_more_one_clause)
- << getOpenMPDirectiveName(DKind) << getOpenMPClauseName(CKind) << 0;
- ErrorFound = true;
- }
- Clause = ParseOpenMPClause(CKind, WrongDirective);
- break;
- }
- [[fallthrough]];
- case OMPC_init:
- case OMPC_use:
- Clause = ParseOpenMPInteropClause(CKind, WrongDirective);
- break;
- case OMPC_device_type:
- case OMPC_unknown:
- skipUntilPragmaOpenMPEnd(DKind);
- break;
- case OMPC_threadprivate:
- case OMPC_uniform:
- case OMPC_match:
- if (!WrongDirective)
- Diag(Tok, diag::err_omp_unexpected_clause)
- << getOpenMPClauseName(CKind) << getOpenMPDirectiveName(DKind);
- SkipUntil(tok::comma, tok::annot_pragma_openmp_end, StopBeforeMatch);
- break;
- default:
- 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,
- bool IsAddressOfOperand) {
- 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(AnyCastExpr, IsAddressOfOperand, NotTypeCast));
- ExprResult Val(ParseRHSOfBinaryExpression(LHS, prec::Conditional));
- Val = Actions.ActOnFinishFullExpr(Val.get(), ELoc, /*DiscardedValue*/ false);
- // Parse ')'.
- RLoc = Tok.getLocation();
- if (!T.consumeClose())
- RLoc = T.getCloseLocation();
- return Val;
- }
- /// Parsing of OpenMP clauses with single expressions like 'final',
- /// 'collapse', 'safelen', 'num_threads', 'simdlen', 'num_teams',
- /// 'thread_limit', 'simdlen', 'priority', 'grainsize', 'num_tasks', 'hint' or
- /// 'detach'.
- ///
- /// 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 ')'
- ///
- /// allocator-clause:
- /// 'allocator' '(' expression ')'
- ///
- /// detach-clause:
- /// 'detach' '(' event-handler-expression ')'
- ///
- /// align-clause
- /// 'align' '(' positive-integer-constant ')'
- ///
- 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);
- }
- /// Parse indirect clause for '#pragma omp declare target' directive.
- /// 'indirect' '[' '(' invoked-by-fptr ')' ']'
- /// where invoked-by-fptr is a constant boolean expression that evaluates to
- /// true or false at compile time.
- bool Parser::ParseOpenMPIndirectClause(Sema::DeclareTargetContextInfo &DTCI,
- bool ParseOnly) {
- SourceLocation Loc = ConsumeToken();
- SourceLocation RLoc;
- if (Tok.isNot(tok::l_paren)) {
- if (ParseOnly)
- return false;
- DTCI.Indirect = nullptr;
- return true;
- }
- ExprResult Val =
- ParseOpenMPParensExpr(getOpenMPClauseName(OMPC_indirect), RLoc);
- if (Val.isInvalid())
- return false;
- if (ParseOnly)
- return false;
- if (!Val.get()->isValueDependent() && !Val.get()->isTypeDependent() &&
- !Val.get()->isInstantiationDependent() &&
- !Val.get()->containsUnexpandedParameterPack()) {
- ExprResult Ret = Actions.CheckBooleanCondition(Loc, Val.get());
- if (Ret.isInvalid())
- return false;
- llvm::APSInt Result;
- Ret = Actions.VerifyIntegerConstantExpression(Val.get(), &Result,
- Sema::AllowFold);
- if (Ret.isInvalid())
- return false;
- DTCI.Indirect = Val.get();
- return true;
- }
- return false;
- }
- /// Parses a comma-separated list of interop-types and a prefer_type list.
- ///
- bool Parser::ParseOMPInteropInfo(OMPInteropInfo &InteropInfo,
- OpenMPClauseKind Kind) {
- const Token &Tok = getCurToken();
- bool HasError = false;
- bool IsTarget = false;
- bool IsTargetSync = false;
- while (Tok.is(tok::identifier)) {
- // Currently prefer_type is only allowed with 'init' and it must be first.
- bool PreferTypeAllowed = Kind == OMPC_init &&
- InteropInfo.PreferTypes.empty() && !IsTarget &&
- !IsTargetSync;
- if (Tok.getIdentifierInfo()->isStr("target")) {
- // OpenMP 5.1 [2.15.1, interop Construct, Restrictions]
- // Each interop-type may be specified on an action-clause at most
- // once.
- if (IsTarget)
- Diag(Tok, diag::warn_omp_more_one_interop_type) << "target";
- IsTarget = true;
- ConsumeToken();
- } else if (Tok.getIdentifierInfo()->isStr("targetsync")) {
- if (IsTargetSync)
- Diag(Tok, diag::warn_omp_more_one_interop_type) << "targetsync";
- IsTargetSync = true;
- ConsumeToken();
- } else if (Tok.getIdentifierInfo()->isStr("prefer_type") &&
- PreferTypeAllowed) {
- ConsumeToken();
- BalancedDelimiterTracker PT(*this, tok::l_paren,
- tok::annot_pragma_openmp_end);
- if (PT.expectAndConsume(diag::err_expected_lparen_after, "prefer_type"))
- HasError = true;
- while (Tok.isNot(tok::r_paren)) {
- SourceLocation Loc = Tok.getLocation();
- ExprResult LHS = ParseCastExpression(AnyCastExpr);
- ExprResult PTExpr = Actions.CorrectDelayedTyposInExpr(
- ParseRHSOfBinaryExpression(LHS, prec::Conditional));
- PTExpr = Actions.ActOnFinishFullExpr(PTExpr.get(), Loc,
- /*DiscardedValue=*/false);
- if (PTExpr.isUsable()) {
- InteropInfo.PreferTypes.push_back(PTExpr.get());
- } else {
- HasError = true;
- SkipUntil(tok::comma, tok::r_paren, tok::annot_pragma_openmp_end,
- StopBeforeMatch);
- }
- if (Tok.is(tok::comma))
- ConsumeToken();
- }
- PT.consumeClose();
- } else {
- HasError = true;
- Diag(Tok, diag::err_omp_expected_interop_type);
- ConsumeToken();
- }
- if (!Tok.is(tok::comma))
- break;
- ConsumeToken();
- }
- if (!HasError && !IsTarget && !IsTargetSync) {
- Diag(Tok, diag::err_omp_expected_interop_type);
- HasError = true;
- }
- if (Kind == OMPC_init) {
- if (Tok.isNot(tok::colon) && (IsTarget || IsTargetSync))
- Diag(Tok, diag::warn_pragma_expected_colon) << "interop types";
- if (Tok.is(tok::colon))
- ConsumeToken();
- }
- // As of OpenMP 5.1,there are two interop-types, "target" and
- // "targetsync". Either or both are allowed for a single interop.
- InteropInfo.IsTarget = IsTarget;
- InteropInfo.IsTargetSync = IsTargetSync;
- return HasError;
- }
- /// Parsing of OpenMP clauses that use an interop-var.
- ///
- /// init-clause:
- /// init([interop-modifier, ]interop-type[[, interop-type] ... ]:interop-var)
- ///
- /// destroy-clause:
- /// destroy(interop-var)
- ///
- /// use-clause:
- /// use(interop-var)
- ///
- /// interop-modifier:
- /// prefer_type(preference-list)
- ///
- /// preference-list:
- /// foreign-runtime-id [, foreign-runtime-id]...
- ///
- /// foreign-runtime-id:
- /// <string-literal> | <constant-integral-expression>
- ///
- /// interop-type:
- /// target | targetsync
- ///
- OMPClause *Parser::ParseOpenMPInteropClause(OpenMPClauseKind Kind,
- bool ParseOnly) {
- SourceLocation Loc = ConsumeToken();
- // Parse '('.
- BalancedDelimiterTracker T(*this, tok::l_paren, tok::annot_pragma_openmp_end);
- if (T.expectAndConsume(diag::err_expected_lparen_after,
- getOpenMPClauseName(Kind).data()))
- return nullptr;
- bool InteropError = false;
- OMPInteropInfo InteropInfo;
- if (Kind == OMPC_init)
- InteropError = ParseOMPInteropInfo(InteropInfo, OMPC_init);
- // Parse the variable.
- SourceLocation VarLoc = Tok.getLocation();
- ExprResult InteropVarExpr =
- Actions.CorrectDelayedTyposInExpr(ParseAssignmentExpression());
- if (!InteropVarExpr.isUsable()) {
- SkipUntil(tok::comma, tok::r_paren, tok::annot_pragma_openmp_end,
- StopBeforeMatch);
- }
- // Parse ')'.
- SourceLocation RLoc = Tok.getLocation();
- if (!T.consumeClose())
- RLoc = T.getCloseLocation();
- if (ParseOnly || !InteropVarExpr.isUsable() || InteropError)
- return nullptr;
- if (Kind == OMPC_init)
- return Actions.ActOnOpenMPInitClause(InteropVarExpr.get(), InteropInfo, Loc,
- T.getOpenLocation(), VarLoc, RLoc);
- if (Kind == OMPC_use)
- return Actions.ActOnOpenMPUseClause(InteropVarExpr.get(), Loc,
- T.getOpenLocation(), VarLoc, RLoc);
- if (Kind == OMPC_destroy)
- return Actions.ActOnOpenMPDestroyClause(InteropVarExpr.get(), Loc,
- T.getOpenLocation(), VarLoc, RLoc);
- llvm_unreachable("Unexpected interop variable clause.");
- }
- /// Parsing of simple OpenMP clauses like 'default' or 'proc_bind'.
- ///
- /// default-clause:
- /// 'default' '(' 'none' | 'shared' | 'private' | 'firstprivate' ')'
- ///
- /// proc_bind-clause:
- /// 'proc_bind' '(' 'master' | 'close' | 'spread' ')'
- ///
- /// bind-clause:
- /// 'bind' '(' 'teams' | 'parallel' | 'thread' ')'
- ///
- /// update-clause:
- /// 'update' '(' 'in' | 'out' | 'inout' | 'mutexinoutset' |
- /// 'inoutset' ')'
- ///
- OMPClause *Parser::ParseOpenMPSimpleClause(OpenMPClauseKind Kind,
- bool ParseOnly) {
- std::optional<SimpleClauseData> Val = parseOpenMPSimpleClause(*this, Kind);
- if (!Val || ParseOnly)
- return nullptr;
- if (getLangOpts().OpenMP < 51 && Kind == OMPC_default &&
- (static_cast<DefaultKind>(Val->Type) == OMP_DEFAULT_private ||
- static_cast<DefaultKind>(Val->Type) ==
- OMP_DEFAULT_firstprivate)) {
- Diag(Val->LOpen, diag::err_omp_invalid_dsa)
- << getOpenMPClauseName(static_cast<DefaultKind>(Val->Type) ==
- OMP_DEFAULT_private
- ? OMPC_private
- : OMPC_firstprivate)
- << getOpenMPClauseName(OMPC_default) << "5.1";
- return nullptr;
- }
- return Actions.ActOnOpenMPSimpleClause(Kind, Val->Type,
- Val->TypeLoc, Val->LOpen,
- Val->Loc, Val->RLoc);
- }
- /// 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());
- }
- /// 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 ] ')'
- ///
- /// device-clause:
- /// 'device' '(' [ device-modifier ':' ] expression ')'
- ///
- OMPClause *Parser::ParseOpenMPSingleExprWithArgClause(OpenMPDirectiveKind DKind,
- 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).data()))
- 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), getLangOpts());
- 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), getLangOpts());
- 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), getLangOpts());
- }
- 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), getLangOpts()));
- 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
- unsigned Modifier = getOpenMPSimpleClauseType(
- Kind, Tok.isAnnotation() ? "" : PP.getSpelling(Tok), getLangOpts());
- // Set defaultmap modifier to unknown if it is either scalar, aggregate, or
- // pointer
- if (Modifier < OMPC_DEFAULTMAP_MODIFIER_unknown)
- Modifier = OMPC_DEFAULTMAP_MODIFIER_unknown;
- Arg.push_back(Modifier);
- 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) || getLangOpts().OpenMP < 50) {
- 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), getLangOpts()));
- KLoc.push_back(Tok.getLocation());
- if (Tok.isNot(tok::r_paren) && Tok.isNot(tok::comma) &&
- Tok.isNot(tok::annot_pragma_openmp_end))
- ConsumeAnyToken();
- } else {
- Arg.push_back(OMPC_DEFAULTMAP_unknown);
- KLoc.push_back(SourceLocation());
- }
- } else if (Kind == OMPC_order) {
- enum { Modifier, OrderKind, NumberOfElements };
- Arg.resize(NumberOfElements);
- KLoc.resize(NumberOfElements);
- Arg[Modifier] = OMPC_ORDER_MODIFIER_unknown;
- Arg[OrderKind] = OMPC_ORDER_unknown;
- unsigned KindModifier = getOpenMPSimpleClauseType(
- Kind, Tok.isAnnotation() ? "" : PP.getSpelling(Tok), getLangOpts());
- if (KindModifier > OMPC_ORDER_unknown) {
- // Parse 'modifier'
- Arg[Modifier] = KindModifier;
- KLoc[Modifier] = 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) << "order modifier";
- KindModifier = getOpenMPSimpleClauseType(
- Kind, Tok.isAnnotation() ? "" : PP.getSpelling(Tok), getLangOpts());
- }
- Arg[OrderKind] = KindModifier;
- KLoc[OrderKind] = Tok.getLocation();
- if (Tok.isNot(tok::r_paren) && Tok.isNot(tok::comma) &&
- Tok.isNot(tok::annot_pragma_openmp_end))
- ConsumeAnyToken();
- } else if (Kind == OMPC_device) {
- // Only target executable directives support extended device construct.
- if (isOpenMPTargetExecutionDirective(DKind) && getLangOpts().OpenMP >= 50 &&
- NextToken().is(tok::colon)) {
- // Parse optional <device modifier> ':'
- Arg.push_back(getOpenMPSimpleClauseType(
- Kind, Tok.isAnnotation() ? "" : PP.getSpelling(Tok), getLangOpts()));
- KLoc.push_back(Tok.getLocation());
- ConsumeAnyToken();
- // Parse ':'
- ConsumeAnyToken();
- } else {
- Arg.push_back(OMPC_DEVICE_unknown);
- KLoc.emplace_back();
- }
- } else if (Kind == OMPC_grainsize) {
- // Parse optional <grainsize modifier> ':'
- OpenMPGrainsizeClauseModifier Modifier =
- static_cast<OpenMPGrainsizeClauseModifier>(getOpenMPSimpleClauseType(
- Kind, Tok.isAnnotation() ? "" : PP.getSpelling(Tok),
- getLangOpts()));
- if (getLangOpts().OpenMP >= 51) {
- if (NextToken().is(tok::colon)) {
- Arg.push_back(Modifier);
- KLoc.push_back(Tok.getLocation());
- // Parse modifier
- ConsumeAnyToken();
- // Parse ':'
- ConsumeAnyToken();
- } else {
- if (Modifier == OMPC_GRAINSIZE_strict) {
- Diag(Tok, diag::err_modifier_expected_colon) << "strict";
- // Parse modifier
- ConsumeAnyToken();
- }
- Arg.push_back(OMPC_GRAINSIZE_unknown);
- KLoc.emplace_back();
- }
- } else {
- Arg.push_back(OMPC_GRAINSIZE_unknown);
- KLoc.emplace_back();
- }
- } else if (Kind == OMPC_num_tasks) {
- // Parse optional <num_tasks modifier> ':'
- OpenMPNumTasksClauseModifier Modifier =
- static_cast<OpenMPNumTasksClauseModifier>(getOpenMPSimpleClauseType(
- Kind, Tok.isAnnotation() ? "" : PP.getSpelling(Tok),
- getLangOpts()));
- if (getLangOpts().OpenMP >= 51) {
- if (NextToken().is(tok::colon)) {
- Arg.push_back(Modifier);
- KLoc.push_back(Tok.getLocation());
- // Parse modifier
- ConsumeAnyToken();
- // Parse ':'
- ConsumeAnyToken();
- } else {
- if (Modifier == OMPC_NUMTASKS_strict) {
- Diag(Tok, diag::err_modifier_expected_colon) << "strict";
- // Parse modifier
- ConsumeAnyToken();
- }
- Arg.push_back(OMPC_NUMTASKS_unknown);
- KLoc.emplace_back();
- }
- } else {
- Arg.push_back(OMPC_NUMTASKS_unknown);
- KLoc.emplace_back();
- }
- } else {
- assert(Kind == OMPC_if);
- KLoc.push_back(Tok.getLocation());
- TentativeParsingAction TPA(*this);
- auto DK = parseOpenMPDirectiveKind(*this);
- Arg.push_back(DK);
- if (DK != OMPD_unknown) {
- ConsumeToken();
- if (Tok.is(tok::colon) && getLangOpts().OpenMP > 40) {
- TPA.Commit();
- DelimLoc = ConsumeToken();
- } else {
- TPA.Revert();
- Arg.back() = unsigned(OMPD_unknown);
- }
- } else {
- TPA.Revert();
- }
- }
- bool NeedAnExpression = (Kind == OMPC_schedule && DelimLoc.isValid()) ||
- (Kind == OMPC_dist_schedule && DelimLoc.isValid()) ||
- Kind == OMPC_if || Kind == OMPC_device ||
- Kind == OMPC_grainsize || Kind == OMPC_num_tasks;
- if (NeedAnExpression) {
- SourceLocation ELoc = Tok.getLocation();
- ExprResult LHS(ParseCastExpression(AnyCastExpr, false, NotTypeCast));
- Val = ParseRHSOfBinaryExpression(LHS, prec::Conditional);
- Val =
- Actions.ActOnFinishFullExpr(Val.get(), ELoc, /*DiscardedValue*/ false);
- }
- // Parse ')'.
- SourceLocation RLoc = Tok.getLocation();
- if (!T.consumeClose())
- RLoc = T.getCloseLocation();
- if (NeedAnExpression && Val.isInvalid())
- return nullptr;
- if (ParseOnly)
- return nullptr;
- return Actions.ActOnOpenMPSingleExprWithArgClause(
- Kind, Arg, Val.get(), Loc, T.getOpenLocation(), KLoc, DelimLoc, RLoc);
- }
- 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, /*ObjectType=*/nullptr,
- /*ObjectHadErrors=*/false, /*EnteringContext*/ false,
- /*AllowDestructorName*/ false,
- /*AllowConstructorName*/ false,
- /*AllowDeductionGuide*/ false, nullptr, ReductionId);
- }
- /// Checks if the token is a valid map-type-modifier.
- /// FIXME: It will return an OpenMPMapClauseKind if that's what it parses.
- static OpenMPMapModifierKind isMapModifier(Parser &P) {
- Token Tok = P.getCurToken();
- if (!Tok.is(tok::identifier))
- return OMPC_MAP_MODIFIER_unknown;
- Preprocessor &PP = P.getPreprocessor();
- OpenMPMapModifierKind TypeModifier =
- static_cast<OpenMPMapModifierKind>(getOpenMPSimpleClauseType(
- OMPC_map, PP.getSpelling(Tok), P.getLangOpts()));
- return TypeModifier;
- }
- /// Parse the mapper modifier in map, to, and from clauses.
- bool Parser::parseMapperModifier(Sema::OpenMPVarListDataTy &Data) {
- // Parse '('.
- BalancedDelimiterTracker T(*this, tok::l_paren, tok::colon);
- if (T.expectAndConsume(diag::err_expected_lparen_after, "mapper")) {
- SkipUntil(tok::colon, tok::r_paren, tok::annot_pragma_openmp_end,
- StopBeforeMatch);
- return true;
- }
- // Parse mapper-identifier
- if (getLangOpts().CPlusPlus)
- ParseOptionalCXXScopeSpecifier(Data.ReductionOrMapperIdScopeSpec,
- /*ObjectType=*/nullptr,
- /*ObjectHasErrors=*/false,
- /*EnteringContext=*/false);
- if (Tok.isNot(tok::identifier) && Tok.isNot(tok::kw_default)) {
- Diag(Tok.getLocation(), diag::err_omp_mapper_illegal_identifier);
- SkipUntil(tok::colon, tok::r_paren, tok::annot_pragma_openmp_end,
- StopBeforeMatch);
- return true;
- }
- auto &DeclNames = Actions.getASTContext().DeclarationNames;
- Data.ReductionOrMapperId = DeclarationNameInfo(
- DeclNames.getIdentifier(Tok.getIdentifierInfo()), Tok.getLocation());
- ConsumeToken();
- // Parse ')'.
- return T.consumeClose();
- }
- /// Parse map-type-modifiers in map clause.
- /// map([ [map-type-modifier[,] [map-type-modifier[,] ...] map-type : ] list)
- /// where, map-type-modifier ::= always | close | mapper(mapper-identifier) |
- /// present
- bool Parser::parseMapTypeModifiers(Sema::OpenMPVarListDataTy &Data) {
- while (getCurToken().isNot(tok::colon)) {
- OpenMPMapModifierKind TypeModifier = isMapModifier(*this);
- if (TypeModifier == OMPC_MAP_MODIFIER_always ||
- TypeModifier == OMPC_MAP_MODIFIER_close ||
- TypeModifier == OMPC_MAP_MODIFIER_present ||
- TypeModifier == OMPC_MAP_MODIFIER_ompx_hold) {
- Data.MapTypeModifiers.push_back(TypeModifier);
- Data.MapTypeModifiersLoc.push_back(Tok.getLocation());
- ConsumeToken();
- } else if (TypeModifier == OMPC_MAP_MODIFIER_mapper) {
- Data.MapTypeModifiers.push_back(TypeModifier);
- Data.MapTypeModifiersLoc.push_back(Tok.getLocation());
- ConsumeToken();
- if (parseMapperModifier(Data))
- return true;
- } else {
- // For the case of unknown map-type-modifier or a map-type.
- // Map-type is followed by a colon; the function returns when it
- // encounters a token followed by a colon.
- if (Tok.is(tok::comma)) {
- Diag(Tok, diag::err_omp_map_type_modifier_missing);
- ConsumeToken();
- continue;
- }
- // Potential map-type token as it is followed by a colon.
- if (PP.LookAhead(0).is(tok::colon))
- return false;
- Diag(Tok, diag::err_omp_unknown_map_type_modifier)
- << (getLangOpts().OpenMP >= 51 ? (getLangOpts().OpenMP >= 52 ? 2 : 1)
- : 0)
- << getLangOpts().OpenMPExtensions;
- ConsumeToken();
- }
- if (getCurToken().is(tok::comma))
- ConsumeToken();
- }
- return false;
- }
- /// Checks if the token is a valid map-type.
- /// FIXME: It will return an OpenMPMapModifierKind if that's what it parses.
- static OpenMPMapClauseKind isMapType(Parser &P) {
- Token Tok = P.getCurToken();
- // The map-type token can be either an identifier or the C++ delete keyword.
- if (!Tok.isOneOf(tok::identifier, tok::kw_delete))
- return OMPC_MAP_unknown;
- Preprocessor &PP = P.getPreprocessor();
- OpenMPMapClauseKind MapType =
- static_cast<OpenMPMapClauseKind>(getOpenMPSimpleClauseType(
- OMPC_map, PP.getSpelling(Tok), P.getLangOpts()));
- return MapType;
- }
- /// Parse map-type in map clause.
- /// map([ [map-type-modifier[,] [map-type-modifier[,] ...] map-type : ] list)
- /// where, map-type ::= to | from | tofrom | alloc | release | delete
- static void parseMapType(Parser &P, Sema::OpenMPVarListDataTy &Data) {
- Token Tok = P.getCurToken();
- if (Tok.is(tok::colon)) {
- P.Diag(Tok, diag::err_omp_map_type_missing);
- return;
- }
- Data.ExtraModifier = isMapType(P);
- if (Data.ExtraModifier == OMPC_MAP_unknown)
- P.Diag(Tok, diag::err_omp_unknown_map_type);
- P.ConsumeToken();
- }
- /// Parses simple expression in parens for single-expression clauses of OpenMP
- /// constructs.
- ExprResult Parser::ParseOpenMPIteratorsExpr() {
- assert(Tok.is(tok::identifier) && PP.getSpelling(Tok) == "iterator" &&
- "Expected 'iterator' token.");
- SourceLocation IteratorKwLoc = ConsumeToken();
- BalancedDelimiterTracker T(*this, tok::l_paren, tok::annot_pragma_openmp_end);
- if (T.expectAndConsume(diag::err_expected_lparen_after, "iterator"))
- return ExprError();
- SourceLocation LLoc = T.getOpenLocation();
- SmallVector<Sema::OMPIteratorData, 4> Data;
- while (Tok.isNot(tok::r_paren) && Tok.isNot(tok::annot_pragma_openmp_end)) {
- // Check if the type parsing is required.
- ParsedType IteratorType;
- if (Tok.isNot(tok::identifier) || NextToken().isNot(tok::equal)) {
- // identifier '=' is not found - parse type.
- TypeResult TR = ParseTypeName();
- if (TR.isInvalid()) {
- T.skipToEnd();
- return ExprError();
- }
- IteratorType = TR.get();
- }
- // Parse identifier.
- IdentifierInfo *II = nullptr;
- SourceLocation IdLoc;
- if (Tok.is(tok::identifier)) {
- II = Tok.getIdentifierInfo();
- IdLoc = ConsumeToken();
- } else {
- Diag(Tok, diag::err_expected_unqualified_id) << 0;
- }
- // Parse '='.
- SourceLocation AssignLoc;
- if (Tok.is(tok::equal))
- AssignLoc = ConsumeToken();
- else
- Diag(Tok, diag::err_omp_expected_equal_in_iterator);
- // Parse range-specification - <begin> ':' <end> [ ':' <step> ]
- ColonProtectionRAIIObject ColonRAII(*this);
- // Parse <begin>
- SourceLocation Loc = Tok.getLocation();
- ExprResult LHS = ParseCastExpression(AnyCastExpr);
- ExprResult Begin = Actions.CorrectDelayedTyposInExpr(
- ParseRHSOfBinaryExpression(LHS, prec::Conditional));
- Begin = Actions.ActOnFinishFullExpr(Begin.get(), Loc,
- /*DiscardedValue=*/false);
- // Parse ':'.
- SourceLocation ColonLoc;
- if (Tok.is(tok::colon))
- ColonLoc = ConsumeToken();
- // Parse <end>
- Loc = Tok.getLocation();
- LHS = ParseCastExpression(AnyCastExpr);
- ExprResult End = Actions.CorrectDelayedTyposInExpr(
- ParseRHSOfBinaryExpression(LHS, prec::Conditional));
- End = Actions.ActOnFinishFullExpr(End.get(), Loc,
- /*DiscardedValue=*/false);
- SourceLocation SecColonLoc;
- ExprResult Step;
- // Parse optional step.
- if (Tok.is(tok::colon)) {
- // Parse ':'
- SecColonLoc = ConsumeToken();
- // Parse <step>
- Loc = Tok.getLocation();
- LHS = ParseCastExpression(AnyCastExpr);
- Step = Actions.CorrectDelayedTyposInExpr(
- ParseRHSOfBinaryExpression(LHS, prec::Conditional));
- Step = Actions.ActOnFinishFullExpr(Step.get(), Loc,
- /*DiscardedValue=*/false);
- }
- // Parse ',' or ')'
- if (Tok.isNot(tok::comma) && Tok.isNot(tok::r_paren))
- Diag(Tok, diag::err_omp_expected_punc_after_iterator);
- if (Tok.is(tok::comma))
- ConsumeToken();
- Sema::OMPIteratorData &D = Data.emplace_back();
- D.DeclIdent = II;
- D.DeclIdentLoc = IdLoc;
- D.Type = IteratorType;
- D.AssignLoc = AssignLoc;
- D.ColonLoc = ColonLoc;
- D.SecColonLoc = SecColonLoc;
- D.Range.Begin = Begin.get();
- D.Range.End = End.get();
- D.Range.Step = Step.get();
- }
- // Parse ')'.
- SourceLocation RLoc = Tok.getLocation();
- if (!T.consumeClose())
- RLoc = T.getCloseLocation();
- return Actions.ActOnOMPIteratorExpr(getCurScope(), IteratorKwLoc, LLoc, RLoc,
- Data);
- }
- bool Parser::ParseOpenMPReservedLocator(OpenMPClauseKind Kind,
- Sema::OpenMPVarListDataTy &Data,
- const LangOptions &LangOpts) {
- // Currently the only reserved locator is 'omp_all_memory' which is only
- // allowed on a depend clause.
- if (Kind != OMPC_depend || LangOpts.OpenMP < 51)
- return false;
- if (Tok.is(tok::identifier) &&
- Tok.getIdentifierInfo()->isStr("omp_all_memory")) {
- if (Data.ExtraModifier == OMPC_DEPEND_outallmemory ||
- Data.ExtraModifier == OMPC_DEPEND_inoutallmemory)
- Diag(Tok, diag::warn_omp_more_one_omp_all_memory);
- else if (Data.ExtraModifier != OMPC_DEPEND_out &&
- Data.ExtraModifier != OMPC_DEPEND_inout)
- Diag(Tok, diag::err_omp_requires_out_inout_depend_type);
- else
- Data.ExtraModifier = Data.ExtraModifier == OMPC_DEPEND_out
- ? OMPC_DEPEND_outallmemory
- : OMPC_DEPEND_inoutallmemory;
- ConsumeToken();
- return true;
- }
- return false;
- }
- /// Parses clauses with list.
- bool Parser::ParseOpenMPVarList(OpenMPDirectiveKind DKind,
- OpenMPClauseKind Kind,
- SmallVectorImpl<Expr *> &Vars,
- Sema::OpenMPVarListDataTy &Data) {
- UnqualifiedId UnqualifiedReductionId;
- bool InvalidReductionId = false;
- bool IsInvalidMapperModifier = false;
- // Parse '('.
- BalancedDelimiterTracker T(*this, tok::l_paren, tok::annot_pragma_openmp_end);
- if (T.expectAndConsume(diag::err_expected_lparen_after,
- getOpenMPClauseName(Kind).data()))
- return true;
- bool HasIterator = false;
- bool InvalidIterator = false;
- 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) {
- Data.ExtraModifier = OMPC_REDUCTION_unknown;
- if (Kind == OMPC_reduction && getLangOpts().OpenMP >= 50 &&
- (Tok.is(tok::identifier) || Tok.is(tok::kw_default)) &&
- NextToken().is(tok::comma)) {
- // Parse optional reduction modifier.
- Data.ExtraModifier =
- getOpenMPSimpleClauseType(Kind, PP.getSpelling(Tok), getLangOpts());
- Data.ExtraModifierLoc = Tok.getLocation();
- ConsumeToken();
- assert(Tok.is(tok::comma) && "Expected comma.");
- (void)ConsumeToken();
- }
- ColonProtectionRAIIObject ColonRAII(*this);
- if (getLangOpts().CPlusPlus)
- ParseOptionalCXXScopeSpecifier(Data.ReductionOrMapperIdScopeSpec,
- /*ObjectType=*/nullptr,
- /*ObjectHasErrors=*/false,
- /*EnteringContext=*/false);
- InvalidReductionId = ParseReductionId(
- *this, Data.ReductionOrMapperIdScopeSpec, 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.ReductionOrMapperId =
- Actions.GetNameFromUnqualifiedId(UnqualifiedReductionId);
- } else if (Kind == OMPC_depend) {
- if (getLangOpts().OpenMP >= 50) {
- if (Tok.is(tok::identifier) && PP.getSpelling(Tok) == "iterator") {
- // Handle optional dependence modifier.
- // iterator(iterators-definition)
- // where iterators-definition is iterator-specifier [,
- // iterators-definition ]
- // where iterator-specifier is [ iterator-type ] identifier =
- // range-specification
- HasIterator = true;
- EnterScope(Scope::OpenMPDirectiveScope | Scope::DeclScope);
- ExprResult IteratorRes = ParseOpenMPIteratorsExpr();
- Data.DepModOrTailExpr = IteratorRes.get();
- // Parse ','
- ExpectAndConsume(tok::comma);
- }
- }
- // Handle dependency type for depend clause.
- ColonProtectionRAIIObject ColonRAII(*this);
- Data.ExtraModifier = getOpenMPSimpleClauseType(
- Kind, Tok.is(tok::identifier) ? PP.getSpelling(Tok) : "",
- getLangOpts());
- Data.ExtraModifierLoc = Tok.getLocation();
- if (Data.ExtraModifier == 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.ExtraModifier == 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.
- Data.ExtraModifier = OMPC_LINEAR_val;
- if (Tok.is(tok::identifier) && PP.LookAhead(0).is(tok::l_paren)) {
- Data.ExtraModifier =
- getOpenMPSimpleClauseType(Kind, PP.getSpelling(Tok), getLangOpts());
- Data.ExtraModifierLoc = ConsumeToken();
- LinearT.consumeOpen();
- NeedRParenForLinear = true;
- }
- } else if (Kind == OMPC_lastprivate) {
- // Try to parse modifier if any.
- Data.ExtraModifier = OMPC_LASTPRIVATE_unknown;
- // Conditional modifier allowed only in OpenMP 5.0 and not supported in
- // distribute and taskloop based directives.
- if ((getLangOpts().OpenMP >= 50 && !isOpenMPDistributeDirective(DKind) &&
- !isOpenMPTaskLoopDirective(DKind)) &&
- Tok.is(tok::identifier) && PP.LookAhead(0).is(tok::colon)) {
- Data.ExtraModifier =
- getOpenMPSimpleClauseType(Kind, PP.getSpelling(Tok), getLangOpts());
- Data.ExtraModifierLoc = Tok.getLocation();
- ConsumeToken();
- assert(Tok.is(tok::colon) && "Expected colon.");
- Data.ColonLoc = ConsumeToken();
- }
- } else if (Kind == OMPC_map) {
- // Handle optional iterator map modifier.
- if (Tok.is(tok::identifier) && PP.getSpelling(Tok) == "iterator") {
- HasIterator = true;
- EnterScope(Scope::OpenMPDirectiveScope | Scope::DeclScope);
- Data.MapTypeModifiers.push_back(OMPC_MAP_MODIFIER_iterator);
- Data.MapTypeModifiersLoc.push_back(Tok.getLocation());
- ExprResult IteratorRes = ParseOpenMPIteratorsExpr();
- Data.IteratorExpr = IteratorRes.get();
- // Parse ','
- ExpectAndConsume(tok::comma);
- if (getLangOpts().OpenMP < 52) {
- Diag(Tok, diag::err_omp_unknown_map_type_modifier)
- << (getLangOpts().OpenMP >= 51 ? 1 : 0)
- << getLangOpts().OpenMPExtensions;
- InvalidIterator = true;
- }
- }
- // Handle map type for map clause.
- ColonProtectionRAIIObject ColonRAII(*this);
- // The first identifier may be a list item, a map-type or a
- // map-type-modifier. The map-type can also be delete which has the same
- // spelling of the C++ delete keyword.
- Data.ExtraModifier = OMPC_MAP_unknown;
- Data.ExtraModifierLoc = Tok.getLocation();
- // Check for presence of a colon in the map clause.
- TentativeParsingAction TPA(*this);
- bool ColonPresent = false;
- if (SkipUntil(tok::colon, tok::r_paren, tok::annot_pragma_openmp_end,
- StopBeforeMatch)) {
- if (Tok.is(tok::colon))
- ColonPresent = true;
- }
- TPA.Revert();
- // Only parse map-type-modifier[s] and map-type if a colon is present in
- // the map clause.
- if (ColonPresent) {
- IsInvalidMapperModifier = parseMapTypeModifiers(Data);
- if (!IsInvalidMapperModifier)
- parseMapType(*this, Data);
- else
- SkipUntil(tok::colon, tok::annot_pragma_openmp_end, StopBeforeMatch);
- }
- if (Data.ExtraModifier == OMPC_MAP_unknown) {
- Data.ExtraModifier = OMPC_MAP_tofrom;
- if (getLangOpts().OpenMP >= 52) {
- if (DKind == OMPD_target_enter_data)
- Data.ExtraModifier = OMPC_MAP_to;
- else if (DKind == OMPD_target_exit_data)
- Data.ExtraModifier = OMPC_MAP_from;
- }
- Data.IsMapTypeImplicit = true;
- }
- if (Tok.is(tok::colon))
- Data.ColonLoc = ConsumeToken();
- } else if (Kind == OMPC_to || Kind == OMPC_from) {
- while (Tok.is(tok::identifier)) {
- auto Modifier = static_cast<OpenMPMotionModifierKind>(
- getOpenMPSimpleClauseType(Kind, PP.getSpelling(Tok), getLangOpts()));
- if (Modifier == OMPC_MOTION_MODIFIER_unknown)
- break;
- Data.MotionModifiers.push_back(Modifier);
- Data.MotionModifiersLoc.push_back(Tok.getLocation());
- ConsumeToken();
- if (Modifier == OMPC_MOTION_MODIFIER_mapper) {
- IsInvalidMapperModifier = parseMapperModifier(Data);
- if (IsInvalidMapperModifier)
- break;
- }
- // OpenMP < 5.1 doesn't permit a ',' or additional modifiers.
- if (getLangOpts().OpenMP < 51)
- break;
- // OpenMP 5.1 accepts an optional ',' even if the next character is ':'.
- // TODO: Is that intentional?
- if (Tok.is(tok::comma))
- ConsumeToken();
- }
- if (!Data.MotionModifiers.empty() && Tok.isNot(tok::colon)) {
- if (!IsInvalidMapperModifier) {
- if (getLangOpts().OpenMP < 51)
- Diag(Tok, diag::warn_pragma_expected_colon) << ")";
- else
- Diag(Tok, diag::warn_pragma_expected_colon) << "motion modifier";
- }
- SkipUntil(tok::colon, tok::r_paren, tok::annot_pragma_openmp_end,
- StopBeforeMatch);
- }
- // OpenMP 5.1 permits a ':' even without a preceding modifier. TODO: Is
- // that intentional?
- if ((!Data.MotionModifiers.empty() || getLangOpts().OpenMP >= 51) &&
- Tok.is(tok::colon))
- Data.ColonLoc = ConsumeToken();
- } else if (Kind == OMPC_allocate ||
- (Kind == OMPC_affinity && Tok.is(tok::identifier) &&
- PP.getSpelling(Tok) == "iterator")) {
- // Handle optional allocator expression followed by colon delimiter.
- ColonProtectionRAIIObject ColonRAII(*this);
- TentativeParsingAction TPA(*this);
- // OpenMP 5.0, 2.10.1, task Construct.
- // where aff-modifier is one of the following:
- // iterator(iterators-definition)
- ExprResult Tail;
- if (Kind == OMPC_allocate) {
- Tail = ParseAssignmentExpression();
- } else {
- HasIterator = true;
- EnterScope(Scope::OpenMPDirectiveScope | Scope::DeclScope);
- Tail = ParseOpenMPIteratorsExpr();
- }
- Tail = Actions.CorrectDelayedTyposInExpr(Tail);
- Tail = Actions.ActOnFinishFullExpr(Tail.get(), T.getOpenLocation(),
- /*DiscardedValue=*/false);
- if (Tail.isUsable()) {
- if (Tok.is(tok::colon)) {
- Data.DepModOrTailExpr = Tail.get();
- Data.ColonLoc = ConsumeToken();
- TPA.Commit();
- } else {
- // Colon not found, parse only list of variables.
- TPA.Revert();
- }
- } else {
- // Parsing was unsuccessfull, revert and skip to the end of clause or
- // directive.
- TPA.Revert();
- SkipUntil(tok::comma, tok::r_paren, tok::annot_pragma_openmp_end,
- StopBeforeMatch);
- }
- } else if (Kind == OMPC_adjust_args) {
- // Handle adjust-op for adjust_args clause.
- ColonProtectionRAIIObject ColonRAII(*this);
- Data.ExtraModifier = getOpenMPSimpleClauseType(
- Kind, Tok.is(tok::identifier) ? PP.getSpelling(Tok) : "",
- getLangOpts());
- Data.ExtraModifierLoc = Tok.getLocation();
- if (Data.ExtraModifier == OMPC_ADJUST_ARGS_unknown) {
- SkipUntil(tok::colon, tok::r_paren, tok::annot_pragma_openmp_end,
- StopBeforeMatch);
- } else {
- ConsumeToken();
- if (Tok.is(tok::colon))
- Data.ColonLoc = Tok.getLocation();
- ExpectAndConsume(tok::colon, diag::warn_pragma_expected_colon,
- "adjust-op");
- }
- }
- 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.ExtraModifier != OMPC_MAP_unknown) ||
- (Kind == OMPC_depend && Data.ExtraModifier != OMPC_DEPEND_unknown) ||
- (Kind == OMPC_adjust_args &&
- Data.ExtraModifier != OMPC_ADJUST_ARGS_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))) {
- ParseScope OMPListScope(this, Scope::OpenMPDirectiveScope);
- ColonProtectionRAIIObject ColonRAII(*this, MayHaveTail);
- if (!ParseOpenMPReservedLocator(Kind, Data, getLangOpts())) {
- // 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, /*DiscardedValue*/ false);
- if (Tail.isUsable())
- Data.DepModOrTailExpr = Tail.get();
- else
- SkipUntil(tok::comma, tok::r_paren, tok::annot_pragma_openmp_end,
- StopBeforeMatch);
- }
- // Parse ')'.
- Data.RLoc = Tok.getLocation();
- if (!T.consumeClose())
- Data.RLoc = T.getCloseLocation();
- // Exit from scope when the iterator is used in depend clause.
- if (HasIterator)
- ExitScope();
- return (Kind != OMPC_depend && Kind != OMPC_map && Vars.empty()) ||
- (MustHaveTail && !Data.DepModOrTailExpr) || InvalidReductionId ||
- IsInvalidMapperModifier || InvalidIterator;
- }
- /// Parsing of OpenMP clause 'private', 'firstprivate', 'lastprivate',
- /// 'shared', 'copyin', 'copyprivate', 'flush', 'reduction', 'task_reduction',
- /// 'in_reduction', 'nontemporal', 'exclusive' or 'inclusive'.
- ///
- /// 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' '(' [ modifier ',' ] 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 [,] ] [ close [,] ]
- /// [ mapper '(' mapper-identifier ')' [,] ]
- /// to | from | tofrom | alloc | release | delete ':' ] list ')';
- /// to-clause:
- /// 'to' '(' [ mapper '(' mapper-identifier ')' ':' ] list ')'
- /// from-clause:
- /// 'from' '(' [ mapper '(' mapper-identifier ')' ':' ] list ')'
- /// use_device_ptr-clause:
- /// 'use_device_ptr' '(' list ')'
- /// use_device_addr-clause:
- /// 'use_device_addr' '(' list ')'
- /// is_device_ptr-clause:
- /// 'is_device_ptr' '(' list ')'
- /// has_device_addr-clause:
- /// 'has_device_addr' '(' list ')'
- /// allocate-clause:
- /// 'allocate' '(' [ allocator ':' ] list ')'
- /// nontemporal-clause:
- /// 'nontemporal' '(' list ')'
- /// inclusive-clause:
- /// 'inclusive' '(' list ')'
- /// exclusive-clause:
- /// 'exclusive' '(' 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;
- Sema::OpenMPVarListDataTy Data;
- if (ParseOpenMPVarList(DKind, Kind, Vars, Data))
- return nullptr;
- if (ParseOnly)
- return nullptr;
- OMPVarListLocTy Locs(Loc, LOpen, Data.RLoc);
- return Actions.ActOnOpenMPVarListClause(Kind, Vars, Locs, Data);
- }
|