123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771377237733774377537763777377837793780378137823783378437853786378737883789379037913792379337943795379637973798379938003801380238033804380538063807380838093810381138123813381438153816381738183819382038213822382338243825382638273828382938303831383238333834383538363837383838393840384138423843384438453846384738483849385038513852385338543855385638573858385938603861386238633864386538663867386838693870387138723873387438753876387738783879388038813882388338843885388638873888388938903891389238933894389538963897389838993900390139023903390439053906390739083909391039113912391339143915391639173918391939203921392239233924392539263927392839293930393139323933393439353936393739383939394039413942394339443945394639473948394939503951395239533954395539563957395839593960396139623963396439653966396739683969397039713972397339743975397639773978397939803981398239833984398539863987398839893990399139923993399439953996399739983999400040014002400340044005400640074008400940104011401240134014401540164017401840194020402140224023402440254026402740284029403040314032403340344035403640374038403940404041404240434044404540464047404840494050405140524053405440554056405740584059406040614062406340644065406640674068406940704071407240734074407540764077407840794080408140824083408440854086408740884089409040914092409340944095409640974098409941004101410241034104410541064107410841094110411141124113411441154116411741184119412041214122412341244125412641274128412941304131413241334134413541364137413841394140414141424143414441454146414741484149415041514152415341544155415641574158415941604161416241634164416541664167416841694170417141724173417441754176417741784179418041814182418341844185418641874188418941904191419241934194419541964197419841994200420142024203420442054206420742084209421042114212421342144215421642174218421942204221422242234224422542264227422842294230423142324233423442354236423742384239424042414242424342444245424642474248424942504251425242534254425542564257425842594260426142624263426442654266426742684269427042714272427342744275427642774278427942804281428242834284428542864287428842894290429142924293429442954296429742984299430043014302430343044305430643074308430943104311431243134314431543164317431843194320432143224323432443254326432743284329433043314332433343344335433643374338433943404341434243434344434543464347434843494350435143524353435443554356435743584359436043614362436343644365436643674368436943704371437243734374437543764377437843794380438143824383438443854386438743884389439043914392439343944395439643974398439944004401440244034404440544064407440844094410441144124413441444154416441744184419442044214422442344244425442644274428442944304431443244334434443544364437443844394440444144424443444444454446444744484449445044514452445344544455445644574458445944604461446244634464446544664467446844694470447144724473447444754476447744784479448044814482448344844485448644874488448944904491449244934494449544964497449844994500450145024503450445054506450745084509451045114512451345144515451645174518451945204521452245234524452545264527452845294530453145324533453445354536453745384539454045414542454345444545454645474548454945504551455245534554455545564557455845594560456145624563456445654566456745684569457045714572457345744575457645774578457945804581458245834584458545864587458845894590459145924593459445954596459745984599460046014602460346044605460646074608460946104611461246134614461546164617461846194620462146224623462446254626462746284629463046314632463346344635463646374638463946404641464246434644464546464647464846494650465146524653465446554656465746584659466046614662466346644665466646674668466946704671467246734674467546764677467846794680468146824683468446854686468746884689469046914692469346944695469646974698469947004701470247034704470547064707470847094710471147124713471447154716471747184719472047214722472347244725472647274728472947304731473247334734473547364737473847394740474147424743474447454746474747484749475047514752475347544755475647574758475947604761476247634764476547664767476847694770477147724773477447754776477747784779478047814782478347844785478647874788478947904791479247934794479547964797479847994800480148024803480448054806480748084809481048114812481348144815481648174818481948204821482248234824482548264827482848294830483148324833483448354836483748384839484048414842484348444845484648474848484948504851485248534854485548564857485848594860486148624863486448654866486748684869487048714872487348744875487648774878487948804881488248834884488548864887488848894890489148924893489448954896489748984899490049014902490349044905490649074908490949104911491249134914491549164917491849194920492149224923492449254926492749284929493049314932493349344935493649374938493949404941494249434944494549464947494849494950495149524953495449554956495749584959496049614962496349644965496649674968496949704971497249734974497549764977497849794980498149824983498449854986498749884989499049914992499349944995499649974998499950005001500250035004500550065007500850095010501150125013501450155016501750185019502050215022502350245025502650275028502950305031503250335034503550365037503850395040504150425043504450455046504750485049505050515052505350545055505650575058505950605061506250635064506550665067506850695070507150725073507450755076507750785079508050815082508350845085508650875088508950905091509250935094509550965097509850995100510151025103510451055106510751085109511051115112511351145115511651175118511951205121512251235124512551265127512851295130513151325133513451355136513751385139514051415142514351445145514651475148514951505151515251535154515551565157515851595160516151625163516451655166516751685169517051715172517351745175517651775178517951805181518251835184518551865187518851895190519151925193519451955196519751985199520052015202520352045205520652075208520952105211521252135214521552165217521852195220522152225223522452255226522752285229523052315232523352345235523652375238523952405241524252435244524552465247524852495250525152525253525452555256525752585259526052615262526352645265526652675268526952705271527252735274527552765277527852795280528152825283528452855286528752885289529052915292529352945295529652975298529953005301530253035304530553065307530853095310531153125313531453155316531753185319532053215322532353245325532653275328532953305331533253335334533553365337533853395340534153425343534453455346534753485349535053515352535353545355535653575358535953605361536253635364536553665367536853695370537153725373537453755376537753785379538053815382538353845385538653875388538953905391539253935394539553965397539853995400540154025403540454055406540754085409541054115412541354145415541654175418541954205421542254235424542554265427542854295430543154325433543454355436543754385439544054415442544354445445544654475448544954505451545254535454545554565457545854595460546154625463546454655466546754685469547054715472547354745475547654775478547954805481548254835484548554865487548854895490549154925493549454955496549754985499550055015502550355045505550655075508550955105511551255135514551555165517551855195520552155225523552455255526552755285529553055315532553355345535553655375538553955405541554255435544554555465547554855495550555155525553555455555556555755585559556055615562556355645565556655675568556955705571557255735574557555765577557855795580558155825583558455855586558755885589559055915592559355945595559655975598559956005601560256035604560556065607560856095610561156125613561456155616561756185619562056215622562356245625562656275628562956305631563256335634563556365637563856395640564156425643564456455646564756485649565056515652565356545655565656575658565956605661566256635664566556665667566856695670567156725673567456755676567756785679568056815682568356845685568656875688568956905691569256935694569556965697569856995700570157025703570457055706570757085709571057115712571357145715571657175718571957205721572257235724572557265727572857295730573157325733573457355736573757385739574057415742574357445745574657475748574957505751575257535754575557565757575857595760576157625763576457655766576757685769577057715772577357745775577657775778577957805781578257835784578557865787578857895790579157925793579457955796579757985799580058015802580358045805580658075808580958105811581258135814581558165817581858195820582158225823582458255826582758285829583058315832583358345835583658375838583958405841584258435844584558465847584858495850585158525853585458555856585758585859586058615862586358645865586658675868586958705871587258735874587558765877587858795880588158825883588458855886588758885889589058915892589358945895589658975898589959005901590259035904590559065907590859095910591159125913591459155916591759185919592059215922592359245925592659275928592959305931593259335934593559365937593859395940594159425943594459455946594759485949595059515952595359545955595659575958595959605961596259635964596559665967596859695970597159725973597459755976597759785979598059815982598359845985598659875988598959905991599259935994599559965997599859996000600160026003600460056006600760086009601060116012601360146015601660176018601960206021602260236024602560266027602860296030603160326033603460356036603760386039604060416042604360446045604660476048604960506051605260536054605560566057605860596060606160626063606460656066606760686069607060716072607360746075607660776078607960806081608260836084608560866087608860896090609160926093609460956096609760986099610061016102610361046105610661076108610961106111611261136114611561166117611861196120612161226123612461256126612761286129613061316132613361346135613661376138613961406141614261436144614561466147614861496150615161526153615461556156615761586159616061616162616361646165616661676168616961706171617261736174617561766177617861796180618161826183618461856186618761886189619061916192619361946195619661976198619962006201620262036204620562066207620862096210621162126213621462156216621762186219622062216222622362246225622662276228622962306231623262336234623562366237623862396240624162426243624462456246624762486249625062516252625362546255625662576258625962606261626262636264626562666267626862696270627162726273627462756276627762786279628062816282628362846285628662876288628962906291629262936294629562966297629862996300630163026303630463056306630763086309631063116312631363146315631663176318631963206321632263236324632563266327632863296330633163326333633463356336633763386339634063416342634363446345634663476348634963506351635263536354635563566357635863596360636163626363636463656366636763686369637063716372637363746375637663776378637963806381638263836384638563866387638863896390639163926393639463956396639763986399640064016402640364046405640664076408640964106411641264136414641564166417641864196420642164226423642464256426642764286429643064316432643364346435643664376438643964406441644264436444644564466447644864496450645164526453645464556456645764586459646064616462646364646465646664676468646964706471647264736474647564766477647864796480648164826483648464856486648764886489649064916492649364946495649664976498649965006501650265036504650565066507650865096510651165126513651465156516651765186519652065216522652365246525652665276528652965306531653265336534653565366537653865396540654165426543654465456546654765486549655065516552655365546555655665576558655965606561656265636564656565666567656865696570657165726573657465756576657765786579658065816582658365846585658665876588658965906591659265936594659565966597659865996600660166026603660466056606660766086609661066116612661366146615661666176618661966206621662266236624662566266627662866296630663166326633663466356636663766386639664066416642664366446645664666476648664966506651665266536654665566566657665866596660666166626663666466656666666766686669667066716672667366746675667666776678667966806681668266836684668566866687668866896690669166926693669466956696669766986699670067016702670367046705670667076708670967106711671267136714671567166717671867196720672167226723672467256726672767286729673067316732673367346735673667376738673967406741674267436744674567466747674867496750675167526753675467556756675767586759676067616762676367646765676667676768676967706771677267736774677567766777677867796780678167826783678467856786678767886789679067916792679367946795679667976798679968006801680268036804680568066807680868096810681168126813681468156816681768186819682068216822682368246825682668276828682968306831683268336834683568366837683868396840684168426843684468456846684768486849685068516852685368546855685668576858685968606861686268636864686568666867686868696870687168726873687468756876687768786879688068816882688368846885688668876888688968906891689268936894689568966897689868996900690169026903690469056906690769086909691069116912691369146915691669176918691969206921692269236924692569266927692869296930693169326933693469356936693769386939694069416942694369446945694669476948694969506951695269536954695569566957695869596960696169626963696469656966696769686969697069716972697369746975697669776978697969806981698269836984698569866987698869896990699169926993699469956996699769986999700070017002700370047005700670077008700970107011701270137014701570167017701870197020702170227023702470257026702770287029703070317032703370347035703670377038703970407041704270437044704570467047704870497050705170527053705470557056705770587059706070617062706370647065706670677068706970707071707270737074707570767077707870797080708170827083708470857086708770887089709070917092709370947095709670977098709971007101710271037104710571067107710871097110711171127113711471157116711771187119712071217122712371247125712671277128712971307131713271337134713571367137713871397140714171427143714471457146714771487149715071517152715371547155715671577158715971607161716271637164716571667167716871697170717171727173717471757176717771787179718071817182718371847185718671877188718971907191719271937194719571967197719871997200720172027203720472057206720772087209721072117212721372147215721672177218721972207221722272237224722572267227722872297230723172327233723472357236723772387239724072417242724372447245724672477248724972507251725272537254725572567257725872597260726172627263726472657266726772687269727072717272727372747275727672777278727972807281728272837284728572867287728872897290729172927293729472957296729772987299730073017302730373047305730673077308730973107311731273137314731573167317731873197320732173227323732473257326732773287329733073317332733373347335733673377338733973407341734273437344734573467347734873497350735173527353735473557356735773587359736073617362736373647365736673677368736973707371737273737374737573767377737873797380738173827383738473857386738773887389739073917392739373947395739673977398739974007401740274037404740574067407740874097410741174127413741474157416741774187419742074217422742374247425742674277428742974307431743274337434743574367437743874397440744174427443744474457446744774487449745074517452745374547455745674577458745974607461746274637464746574667467746874697470747174727473747474757476747774787479748074817482748374847485748674877488748974907491749274937494749574967497749874997500750175027503750475057506750775087509751075117512751375147515751675177518751975207521752275237524752575267527752875297530753175327533753475357536753775387539754075417542754375447545754675477548754975507551755275537554755575567557755875597560756175627563756475657566756775687569757075717572757375747575757675777578757975807581758275837584758575867587758875897590759175927593759475957596759775987599760076017602760376047605760676077608760976107611761276137614761576167617761876197620762176227623762476257626762776287629763076317632763376347635763676377638763976407641764276437644764576467647764876497650765176527653765476557656765776587659766076617662766376647665766676677668766976707671767276737674767576767677767876797680768176827683768476857686768776887689769076917692769376947695769676977698769977007701770277037704770577067707770877097710771177127713771477157716771777187719772077217722772377247725772677277728772977307731773277337734773577367737773877397740774177427743774477457746774777487749775077517752775377547755775677577758775977607761776277637764776577667767776877697770777177727773777477757776777777787779778077817782778377847785778677877788778977907791779277937794779577967797779877997800780178027803780478057806780778087809781078117812781378147815781678177818781978207821782278237824782578267827782878297830783178327833783478357836783778387839784078417842784378447845784678477848784978507851785278537854785578567857785878597860786178627863786478657866786778687869787078717872787378747875787678777878787978807881788278837884788578867887788878897890789178927893789478957896789778987899790079017902790379047905790679077908790979107911791279137914791579167917791879197920792179227923792479257926792779287929793079317932793379347935793679377938793979407941794279437944794579467947794879497950795179527953795479557956795779587959796079617962796379647965796679677968796979707971797279737974797579767977797879797980798179827983798479857986798779887989799079917992799379947995799679977998799980008001800280038004800580068007800880098010801180128013801480158016801780188019802080218022802380248025802680278028802980308031803280338034803580368037803880398040804180428043804480458046804780488049805080518052805380548055805680578058805980608061806280638064806580668067806880698070807180728073807480758076807780788079808080818082808380848085808680878088808980908091809280938094809580968097809880998100810181028103810481058106810781088109811081118112811381148115811681178118811981208121812281238124812581268127812881298130813181328133813481358136813781388139814081418142814381448145814681478148814981508151815281538154815581568157815881598160816181628163816481658166816781688169817081718172817381748175817681778178817981808181818281838184818581868187818881898190819181928193819481958196819781988199820082018202820382048205820682078208820982108211821282138214821582168217821882198220822182228223822482258226822782288229823082318232823382348235823682378238823982408241824282438244824582468247824882498250825182528253825482558256825782588259826082618262826382648265826682678268826982708271827282738274827582768277827882798280828182828283828482858286828782888289829082918292829382948295829682978298829983008301830283038304830583068307830883098310831183128313831483158316831783188319832083218322832383248325832683278328832983308331833283338334833583368337833883398340834183428343834483458346834783488349835083518352835383548355835683578358835983608361836283638364836583668367836883698370837183728373837483758376837783788379838083818382838383848385838683878388838983908391839283938394839583968397839883998400840184028403840484058406840784088409841084118412841384148415841684178418841984208421842284238424842584268427842884298430843184328433843484358436843784388439844084418442844384448445844684478448844984508451845284538454845584568457845884598460846184628463846484658466846784688469847084718472847384748475847684778478847984808481848284838484848584868487848884898490849184928493849484958496849784988499850085018502850385048505850685078508850985108511851285138514851585168517851885198520852185228523852485258526852785288529853085318532853385348535853685378538853985408541854285438544854585468547854885498550855185528553855485558556855785588559856085618562856385648565856685678568856985708571857285738574857585768577857885798580858185828583858485858586858785888589859085918592859385948595859685978598859986008601860286038604860586068607860886098610861186128613861486158616861786188619862086218622862386248625862686278628862986308631863286338634863586368637863886398640864186428643864486458646864786488649865086518652865386548655865686578658865986608661866286638664866586668667866886698670867186728673867486758676867786788679868086818682868386848685868686878688868986908691869286938694869586968697869886998700870187028703870487058706870787088709871087118712871387148715871687178718871987208721872287238724872587268727872887298730873187328733873487358736873787388739874087418742874387448745874687478748874987508751875287538754875587568757875887598760876187628763876487658766876787688769877087718772877387748775877687778778877987808781878287838784878587868787878887898790879187928793879487958796879787988799880088018802880388048805880688078808880988108811881288138814881588168817881888198820882188228823882488258826882788288829883088318832883388348835883688378838883988408841884288438844884588468847884888498850885188528853885488558856885788588859886088618862886388648865886688678868886988708871887288738874887588768877887888798880888188828883888488858886888788888889889088918892889388948895889688978898889989008901890289038904890589068907890889098910891189128913891489158916891789188919892089218922892389248925892689278928892989308931893289338934893589368937893889398940894189428943894489458946894789488949895089518952895389548955895689578958895989608961896289638964896589668967896889698970897189728973897489758976897789788979898089818982898389848985898689878988898989908991899289938994899589968997899889999000900190029003900490059006900790089009901090119012901390149015901690179018901990209021902290239024902590269027902890299030903190329033903490359036903790389039904090419042904390449045904690479048904990509051905290539054905590569057905890599060906190629063906490659066906790689069907090719072907390749075907690779078907990809081908290839084908590869087908890899090909190929093909490959096909790989099910091019102910391049105910691079108910991109111911291139114911591169117911891199120912191229123912491259126912791289129913091319132913391349135913691379138913991409141914291439144914591469147914891499150915191529153915491559156915791589159916091619162916391649165916691679168916991709171917291739174917591769177917891799180918191829183918491859186918791889189919091919192919391949195919691979198919992009201920292039204920592069207920892099210921192129213921492159216921792189219922092219222922392249225922692279228922992309231923292339234923592369237923892399240924192429243924492459246924792489249925092519252925392549255925692579258925992609261926292639264926592669267926892699270927192729273927492759276927792789279928092819282928392849285928692879288928992909291929292939294929592969297929892999300930193029303930493059306930793089309931093119312931393149315931693179318931993209321932293239324932593269327932893299330933193329333933493359336933793389339934093419342934393449345934693479348934993509351935293539354935593569357935893599360936193629363936493659366936793689369937093719372937393749375937693779378937993809381938293839384938593869387938893899390939193929393939493959396939793989399940094019402940394049405940694079408940994109411941294139414941594169417941894199420942194229423942494259426942794289429943094319432943394349435943694379438943994409441944294439444944594469447944894499450945194529453945494559456945794589459946094619462946394649465946694679468946994709471947294739474947594769477947894799480948194829483948494859486948794889489949094919492949394949495949694979498949995009501950295039504950595069507950895099510951195129513951495159516951795189519952095219522952395249525952695279528952995309531953295339534953595369537953895399540954195429543954495459546954795489549955095519552955395549555955695579558955995609561956295639564956595669567956895699570957195729573957495759576957795789579958095819582958395849585958695879588958995909591959295939594959595969597959895999600960196029603960496059606960796089609961096119612961396149615961696179618961996209621962296239624962596269627962896299630963196329633963496359636963796389639964096419642964396449645964696479648964996509651965296539654965596569657965896599660966196629663966496659666966796689669967096719672967396749675967696779678967996809681968296839684968596869687968896899690969196929693969496959696969796989699970097019702970397049705970697079708970997109711971297139714971597169717971897199720972197229723972497259726972797289729973097319732973397349735973697379738973997409741974297439744974597469747974897499750975197529753975497559756975797589759976097619762976397649765976697679768976997709771977297739774977597769777977897799780978197829783978497859786978797889789979097919792979397949795979697979798979998009801980298039804980598069807980898099810981198129813981498159816981798189819982098219822982398249825982698279828982998309831983298339834983598369837983898399840984198429843984498459846984798489849985098519852985398549855985698579858985998609861986298639864986598669867986898699870987198729873987498759876987798789879988098819882988398849885988698879888988998909891989298939894989598969897989898999900990199029903990499059906990799089909991099119912991399149915991699179918991999209921992299239924992599269927992899299930993199329933993499359936993799389939994099419942994399449945994699479948994999509951995299539954995599569957995899599960996199629963996499659966996799689969997099719972997399749975997699779978997999809981998299839984998599869987998899899990999199929993999499959996999799989999100001000110002100031000410005100061000710008100091001010011100121001310014100151001610017100181001910020100211002210023100241002510026100271002810029100301003110032100331003410035100361003710038100391004010041100421004310044100451004610047100481004910050100511005210053100541005510056100571005810059100601006110062100631006410065100661006710068100691007010071100721007310074100751007610077100781007910080100811008210083100841008510086100871008810089100901009110092100931009410095100961009710098100991010010101101021010310104101051010610107101081010910110101111011210113101141011510116101171011810119101201012110122101231012410125101261012710128101291013010131101321013310134101351013610137101381013910140101411014210143101441014510146101471014810149101501015110152101531015410155101561015710158101591016010161101621016310164101651016610167101681016910170101711017210173101741017510176101771017810179101801018110182101831018410185101861018710188101891019010191101921019310194101951019610197101981019910200102011020210203102041020510206102071020810209102101021110212102131021410215102161021710218102191022010221102221022310224102251022610227102281022910230102311023210233102341023510236102371023810239102401024110242102431024410245102461024710248102491025010251102521025310254102551025610257102581025910260102611026210263102641026510266102671026810269102701027110272102731027410275102761027710278102791028010281102821028310284102851028610287102881028910290102911029210293102941029510296102971029810299103001030110302103031030410305103061030710308103091031010311103121031310314103151031610317103181031910320103211032210323103241032510326103271032810329103301033110332103331033410335103361033710338103391034010341103421034310344103451034610347103481034910350103511035210353103541035510356103571035810359103601036110362103631036410365103661036710368103691037010371103721037310374103751037610377103781037910380103811038210383103841038510386103871038810389103901039110392103931039410395103961039710398103991040010401104021040310404104051040610407104081040910410104111041210413104141041510416104171041810419104201042110422104231042410425104261042710428104291043010431104321043310434104351043610437104381043910440104411044210443104441044510446104471044810449104501045110452104531045410455104561045710458104591046010461104621046310464104651046610467104681046910470104711047210473104741047510476104771047810479104801048110482104831048410485104861048710488104891049010491104921049310494104951049610497104981049910500105011050210503105041050510506105071050810509105101051110512105131051410515105161051710518105191052010521105221052310524105251052610527105281052910530105311053210533105341053510536105371053810539105401054110542105431054410545105461054710548105491055010551105521055310554105551055610557105581055910560105611056210563105641056510566105671056810569105701057110572105731057410575105761057710578105791058010581105821058310584105851058610587105881058910590105911059210593105941059510596105971059810599106001060110602106031060410605106061060710608106091061010611106121061310614106151061610617106181061910620106211062210623106241062510626106271062810629106301063110632106331063410635106361063710638106391064010641106421064310644106451064610647106481064910650106511065210653106541065510656106571065810659106601066110662106631066410665106661066710668106691067010671106721067310674106751067610677106781067910680106811068210683106841068510686106871068810689106901069110692106931069410695106961069710698106991070010701107021070310704107051070610707107081070910710107111071210713107141071510716107171071810719107201072110722107231072410725107261072710728107291073010731107321073310734107351073610737107381073910740107411074210743107441074510746107471074810749107501075110752107531075410755107561075710758107591076010761107621076310764107651076610767107681076910770107711077210773107741077510776107771077810779107801078110782107831078410785107861078710788107891079010791107921079310794107951079610797107981079910800108011080210803108041080510806108071080810809108101081110812108131081410815108161081710818108191082010821108221082310824108251082610827108281082910830108311083210833108341083510836108371083810839108401084110842108431084410845108461084710848108491085010851108521085310854108551085610857108581085910860108611086210863108641086510866108671086810869108701087110872108731087410875108761087710878108791088010881108821088310884108851088610887108881088910890108911089210893108941089510896108971089810899109001090110902109031090410905109061090710908109091091010911109121091310914109151091610917109181091910920109211092210923109241092510926109271092810929109301093110932109331093410935109361093710938109391094010941109421094310944109451094610947109481094910950109511095210953109541095510956109571095810959109601096110962109631096410965109661096710968109691097010971109721097310974109751097610977109781097910980109811098210983109841098510986109871098810989109901099110992109931099410995109961099710998109991100011001110021100311004110051100611007110081100911010110111101211013110141101511016110171101811019110201102111022110231102411025110261102711028110291103011031110321103311034110351103611037110381103911040110411104211043110441104511046110471104811049110501105111052110531105411055110561105711058110591106011061110621106311064110651106611067110681106911070110711107211073110741107511076110771107811079110801108111082110831108411085110861108711088110891109011091110921109311094110951109611097110981109911100111011110211103111041110511106111071110811109111101111111112111131111411115111161111711118111191112011121111221112311124111251112611127111281112911130111311113211133111341113511136111371113811139111401114111142111431114411145111461114711148111491115011151111521115311154111551115611157111581115911160111611116211163111641116511166111671116811169111701117111172111731117411175111761117711178111791118011181111821118311184111851118611187111881118911190111911119211193111941119511196111971119811199112001120111202112031120411205112061120711208112091121011211112121121311214112151121611217112181121911220112211122211223112241122511226112271122811229112301123111232112331123411235112361123711238112391124011241112421124311244112451124611247112481124911250112511125211253112541125511256112571125811259112601126111262112631126411265112661126711268112691127011271112721127311274112751127611277112781127911280112811128211283112841128511286112871128811289112901129111292112931129411295112961129711298112991130011301113021130311304113051130611307113081130911310113111131211313113141131511316113171131811319113201132111322113231132411325113261132711328113291133011331113321133311334113351133611337113381133911340113411134211343113441134511346113471134811349113501135111352113531135411355113561135711358113591136011361113621136311364113651136611367113681136911370113711137211373113741137511376113771137811379113801138111382113831138411385113861138711388113891139011391113921139311394113951139611397113981139911400114011140211403114041140511406114071140811409114101141111412114131141411415114161141711418114191142011421114221142311424114251142611427114281142911430114311143211433114341143511436114371143811439114401144111442114431144411445114461144711448114491145011451114521145311454114551145611457114581145911460114611146211463114641146511466114671146811469114701147111472114731147411475114761147711478114791148011481114821148311484114851148611487114881148911490114911149211493114941149511496114971149811499115001150111502115031150411505115061150711508115091151011511115121151311514115151151611517115181151911520115211152211523115241152511526115271152811529115301153111532115331153411535115361153711538115391154011541115421154311544115451154611547115481154911550115511155211553115541155511556115571155811559115601156111562115631156411565115661156711568115691157011571115721157311574115751157611577115781157911580115811158211583115841158511586115871158811589115901159111592115931159411595115961159711598115991160011601116021160311604116051160611607116081160911610116111161211613116141161511616116171161811619116201162111622116231162411625116261162711628116291163011631116321163311634116351163611637116381163911640116411164211643116441164511646116471164811649116501165111652116531165411655116561165711658116591166011661116621166311664116651166611667116681166911670116711167211673116741167511676116771167811679116801168111682116831168411685116861168711688116891169011691116921169311694116951169611697116981169911700117011170211703117041170511706117071170811709117101171111712117131171411715117161171711718117191172011721117221172311724117251172611727117281172911730117311173211733117341173511736117371173811739117401174111742117431174411745117461174711748117491175011751117521175311754117551175611757117581175911760117611176211763117641176511766117671176811769117701177111772117731177411775117761177711778117791178011781117821178311784117851178611787117881178911790117911179211793117941179511796117971179811799118001180111802118031180411805118061180711808118091181011811118121181311814118151181611817118181181911820118211182211823118241182511826118271182811829118301183111832118331183411835118361183711838118391184011841118421184311844118451184611847118481184911850118511185211853118541185511856118571185811859118601186111862118631186411865118661186711868118691187011871118721187311874118751187611877118781187911880118811188211883118841188511886118871188811889118901189111892118931189411895118961189711898118991190011901119021190311904119051190611907119081190911910119111191211913119141191511916119171191811919119201192111922119231192411925119261192711928119291193011931119321193311934119351193611937119381193911940119411194211943119441194511946119471194811949119501195111952119531195411955119561195711958119591196011961119621196311964119651196611967119681196911970119711197211973119741197511976119771197811979119801198111982119831198411985119861198711988119891199011991119921199311994119951199611997119981199912000120011200212003120041200512006120071200812009120101201112012120131201412015120161201712018120191202012021120221202312024120251202612027120281202912030120311203212033120341203512036120371203812039120401204112042120431204412045120461204712048120491205012051120521205312054120551205612057120581205912060120611206212063120641206512066120671206812069120701207112072120731207412075120761207712078120791208012081120821208312084120851208612087120881208912090120911209212093120941209512096120971209812099121001210112102121031210412105121061210712108121091211012111121121211312114121151211612117121181211912120121211212212123121241212512126121271212812129121301213112132121331213412135121361213712138121391214012141121421214312144121451214612147121481214912150121511215212153121541215512156121571215812159121601216112162121631216412165121661216712168121691217012171121721217312174121751217612177121781217912180121811218212183121841218512186121871218812189121901219112192121931219412195121961219712198121991220012201122021220312204122051220612207122081220912210122111221212213122141221512216122171221812219122201222112222122231222412225122261222712228122291223012231122321223312234122351223612237122381223912240122411224212243122441224512246122471224812249122501225112252122531225412255122561225712258122591226012261122621226312264122651226612267122681226912270122711227212273122741227512276122771227812279122801228112282122831228412285122861228712288122891229012291122921229312294122951229612297122981229912300123011230212303123041230512306123071230812309123101231112312123131231412315123161231712318123191232012321123221232312324123251232612327123281232912330123311233212333123341233512336123371233812339123401234112342123431234412345123461234712348123491235012351123521235312354123551235612357123581235912360123611236212363123641236512366123671236812369123701237112372123731237412375123761237712378123791238012381123821238312384123851238612387123881238912390123911239212393123941239512396123971239812399124001240112402124031240412405124061240712408124091241012411124121241312414124151241612417124181241912420124211242212423124241242512426124271242812429124301243112432124331243412435124361243712438124391244012441124421244312444124451244612447124481244912450124511245212453124541245512456124571245812459124601246112462124631246412465124661246712468124691247012471124721247312474124751247612477124781247912480124811248212483124841248512486124871248812489124901249112492124931249412495124961249712498124991250012501125021250312504125051250612507125081250912510125111251212513125141251512516125171251812519125201252112522125231252412525125261252712528125291253012531125321253312534125351253612537125381253912540125411254212543125441254512546125471254812549125501255112552125531255412555125561255712558125591256012561125621256312564125651256612567125681256912570125711257212573125741257512576125771257812579125801258112582125831258412585125861258712588125891259012591125921259312594125951259612597125981259912600126011260212603126041260512606126071260812609126101261112612126131261412615126161261712618126191262012621126221262312624126251262612627126281262912630126311263212633126341263512636126371263812639126401264112642126431264412645126461264712648126491265012651126521265312654126551265612657126581265912660126611266212663126641266512666126671266812669126701267112672126731267412675126761267712678126791268012681126821268312684126851268612687126881268912690126911269212693126941269512696126971269812699127001270112702127031270412705127061270712708127091271012711127121271312714127151271612717127181271912720127211272212723127241272512726127271272812729127301273112732127331273412735127361273712738127391274012741127421274312744127451274612747127481274912750127511275212753127541275512756127571275812759127601276112762127631276412765127661276712768127691277012771127721277312774127751277612777127781277912780127811278212783127841278512786127871278812789127901279112792127931279412795127961279712798127991280012801128021280312804128051280612807128081280912810128111281212813128141281512816128171281812819128201282112822128231282412825128261282712828128291283012831128321283312834128351283612837128381283912840128411284212843128441284512846128471284812849128501285112852128531285412855128561285712858128591286012861128621286312864128651286612867128681286912870128711287212873128741287512876128771287812879128801288112882128831288412885128861288712888128891289012891128921289312894128951289612897128981289912900129011290212903129041290512906129071290812909129101291112912129131291412915129161291712918129191292012921129221292312924129251292612927129281292912930129311293212933129341293512936129371293812939129401294112942129431294412945129461294712948129491295012951129521295312954129551295612957129581295912960129611296212963129641296512966129671296812969129701297112972129731297412975129761297712978129791298012981129821298312984129851298612987129881298912990129911299212993129941299512996129971299812999130001300113002130031300413005130061300713008130091301013011130121301313014130151301613017130181301913020130211302213023130241302513026130271302813029130301303113032130331303413035130361303713038130391304013041130421304313044130451304613047130481304913050130511305213053130541305513056130571305813059130601306113062130631306413065130661306713068130691307013071130721307313074130751307613077130781307913080130811308213083130841308513086130871308813089130901309113092130931309413095130961309713098130991310013101131021310313104131051310613107131081310913110131111311213113131141311513116131171311813119131201312113122131231312413125131261312713128131291313013131131321313313134131351313613137131381313913140131411314213143131441314513146131471314813149131501315113152131531315413155131561315713158131591316013161131621316313164131651316613167131681316913170131711317213173131741317513176131771317813179131801318113182131831318413185131861318713188131891319013191131921319313194131951319613197131981319913200132011320213203132041320513206132071320813209132101321113212132131321413215132161321713218132191322013221132221322313224132251322613227132281322913230132311323213233132341323513236132371323813239132401324113242132431324413245132461324713248132491325013251132521325313254132551325613257132581325913260132611326213263132641326513266132671326813269132701327113272132731327413275132761327713278132791328013281132821328313284132851328613287132881328913290132911329213293132941329513296132971329813299133001330113302133031330413305133061330713308133091331013311133121331313314133151331613317133181331913320133211332213323133241332513326133271332813329133301333113332133331333413335133361333713338133391334013341133421334313344133451334613347133481334913350133511335213353133541335513356133571335813359133601336113362133631336413365133661336713368133691337013371133721337313374133751337613377133781337913380133811338213383133841338513386133871338813389133901339113392133931339413395133961339713398133991340013401134021340313404134051340613407134081340913410134111341213413134141341513416134171341813419134201342113422134231342413425134261342713428134291343013431134321343313434134351343613437134381343913440134411344213443134441344513446134471344813449134501345113452134531345413455134561345713458134591346013461134621346313464134651346613467134681346913470134711347213473134741347513476134771347813479134801348113482134831348413485134861348713488134891349013491134921349313494134951349613497134981349913500135011350213503135041350513506135071350813509135101351113512135131351413515135161351713518135191352013521135221352313524135251352613527135281352913530135311353213533135341353513536135371353813539135401354113542135431354413545135461354713548135491355013551135521355313554135551355613557135581355913560135611356213563135641356513566135671356813569135701357113572135731357413575135761357713578135791358013581135821358313584135851358613587135881358913590135911359213593135941359513596135971359813599136001360113602136031360413605136061360713608136091361013611136121361313614136151361613617136181361913620136211362213623136241362513626136271362813629136301363113632136331363413635136361363713638136391364013641136421364313644136451364613647136481364913650136511365213653136541365513656136571365813659136601366113662136631366413665136661366713668136691367013671136721367313674136751367613677136781367913680136811368213683136841368513686136871368813689136901369113692136931369413695136961369713698136991370013701137021370313704137051370613707137081370913710137111371213713137141371513716137171371813719137201372113722137231372413725137261372713728137291373013731137321373313734137351373613737137381373913740137411374213743137441374513746137471374813749137501375113752137531375413755137561375713758137591376013761137621376313764137651376613767137681376913770137711377213773137741377513776137771377813779137801378113782137831378413785137861378713788137891379013791137921379313794137951379613797 |
- //===--- SemaOpenMP.cpp - Semantic Analysis for OpenMP constructs ---------===//
- //
- // The LLVM Compiler Infrastructure
- //
- // This file is distributed under the University of Illinois Open Source
- // License. See LICENSE.TXT for details.
- //
- //===----------------------------------------------------------------------===//
- /// \file
- /// This file implements semantic analysis for OpenMP directives and
- /// clauses.
- ///
- //===----------------------------------------------------------------------===//
- #include "TreeTransform.h"
- #include "clang/AST/ASTContext.h"
- #include "clang/AST/ASTMutationListener.h"
- #include "clang/AST/CXXInheritance.h"
- #include "clang/AST/Decl.h"
- #include "clang/AST/DeclCXX.h"
- #include "clang/AST/DeclOpenMP.h"
- #include "clang/AST/StmtCXX.h"
- #include "clang/AST/StmtOpenMP.h"
- #include "clang/AST/StmtVisitor.h"
- #include "clang/Basic/OpenMPKinds.h"
- #include "clang/Sema/Initialization.h"
- #include "clang/Sema/Lookup.h"
- #include "clang/Sema/Scope.h"
- #include "clang/Sema/ScopeInfo.h"
- #include "clang/Sema/SemaInternal.h"
- #include "llvm/ADT/PointerEmbeddedInt.h"
- using namespace clang;
- //===----------------------------------------------------------------------===//
- // Stack of data-sharing attributes for variables
- //===----------------------------------------------------------------------===//
- static const Expr *checkMapClauseExpressionBase(
- Sema &SemaRef, Expr *E,
- OMPClauseMappableExprCommon::MappableExprComponentList &CurComponents,
- OpenMPClauseKind CKind, bool NoDiagnose);
- namespace {
- /// Default data sharing attributes, which can be applied to directive.
- enum DefaultDataSharingAttributes {
- DSA_unspecified = 0, /// Data sharing attribute not specified.
- DSA_none = 1 << 0, /// Default data sharing attribute 'none'.
- DSA_shared = 1 << 1, /// Default data sharing attribute 'shared'.
- };
- /// Attributes of the defaultmap clause.
- enum DefaultMapAttributes {
- DMA_unspecified, /// Default mapping is not specified.
- DMA_tofrom_scalar, /// Default mapping is 'tofrom:scalar'.
- };
- /// Stack for tracking declarations used in OpenMP directives and
- /// clauses and their data-sharing attributes.
- class DSAStackTy {
- public:
- struct DSAVarData {
- OpenMPDirectiveKind DKind = OMPD_unknown;
- OpenMPClauseKind CKind = OMPC_unknown;
- const Expr *RefExpr = nullptr;
- DeclRefExpr *PrivateCopy = nullptr;
- SourceLocation ImplicitDSALoc;
- DSAVarData() = default;
- DSAVarData(OpenMPDirectiveKind DKind, OpenMPClauseKind CKind,
- const Expr *RefExpr, DeclRefExpr *PrivateCopy,
- SourceLocation ImplicitDSALoc)
- : DKind(DKind), CKind(CKind), RefExpr(RefExpr),
- PrivateCopy(PrivateCopy), ImplicitDSALoc(ImplicitDSALoc) {}
- };
- using OperatorOffsetTy =
- llvm::SmallVector<std::pair<Expr *, OverloadedOperatorKind>, 4>;
- using DoacrossDependMapTy =
- llvm::DenseMap<OMPDependClause *, OperatorOffsetTy>;
- private:
- struct DSAInfo {
- OpenMPClauseKind Attributes = OMPC_unknown;
- /// Pointer to a reference expression and a flag which shows that the
- /// variable is marked as lastprivate(true) or not (false).
- llvm::PointerIntPair<const Expr *, 1, bool> RefExpr;
- DeclRefExpr *PrivateCopy = nullptr;
- };
- using DeclSAMapTy = llvm::SmallDenseMap<const ValueDecl *, DSAInfo, 8>;
- using AlignedMapTy = llvm::SmallDenseMap<const ValueDecl *, const Expr *, 8>;
- using LCDeclInfo = std::pair<unsigned, VarDecl *>;
- using LoopControlVariablesMapTy =
- llvm::SmallDenseMap<const ValueDecl *, LCDeclInfo, 8>;
- /// Struct that associates a component with the clause kind where they are
- /// found.
- struct MappedExprComponentTy {
- OMPClauseMappableExprCommon::MappableExprComponentLists Components;
- OpenMPClauseKind Kind = OMPC_unknown;
- };
- using MappedExprComponentsTy =
- llvm::DenseMap<const ValueDecl *, MappedExprComponentTy>;
- using CriticalsWithHintsTy =
- llvm::StringMap<std::pair<const OMPCriticalDirective *, llvm::APSInt>>;
- struct ReductionData {
- using BOKPtrType = llvm::PointerEmbeddedInt<BinaryOperatorKind, 16>;
- SourceRange ReductionRange;
- llvm::PointerUnion<const Expr *, BOKPtrType> ReductionOp;
- ReductionData() = default;
- void set(BinaryOperatorKind BO, SourceRange RR) {
- ReductionRange = RR;
- ReductionOp = BO;
- }
- void set(const Expr *RefExpr, SourceRange RR) {
- ReductionRange = RR;
- ReductionOp = RefExpr;
- }
- };
- using DeclReductionMapTy =
- llvm::SmallDenseMap<const ValueDecl *, ReductionData, 4>;
- struct SharingMapTy {
- DeclSAMapTy SharingMap;
- DeclReductionMapTy ReductionMap;
- AlignedMapTy AlignedMap;
- MappedExprComponentsTy MappedExprComponents;
- LoopControlVariablesMapTy LCVMap;
- DefaultDataSharingAttributes DefaultAttr = DSA_unspecified;
- SourceLocation DefaultAttrLoc;
- DefaultMapAttributes DefaultMapAttr = DMA_unspecified;
- SourceLocation DefaultMapAttrLoc;
- OpenMPDirectiveKind Directive = OMPD_unknown;
- DeclarationNameInfo DirectiveName;
- Scope *CurScope = nullptr;
- SourceLocation ConstructLoc;
- /// Set of 'depend' clauses with 'sink|source' dependence kind. Required to
- /// get the data (loop counters etc.) about enclosing loop-based construct.
- /// This data is required during codegen.
- DoacrossDependMapTy DoacrossDepends;
- /// first argument (Expr *) contains optional argument of the
- /// 'ordered' clause, the second one is true if the regions has 'ordered'
- /// clause, false otherwise.
- llvm::Optional<std::pair<const Expr *, OMPOrderedClause *>> OrderedRegion;
- unsigned AssociatedLoops = 1;
- const Decl *PossiblyLoopCounter = nullptr;
- bool NowaitRegion = false;
- bool CancelRegion = false;
- bool LoopStart = false;
- SourceLocation InnerTeamsRegionLoc;
- /// Reference to the taskgroup task_reduction reference expression.
- Expr *TaskgroupReductionRef = nullptr;
- SharingMapTy(OpenMPDirectiveKind DKind, DeclarationNameInfo Name,
- Scope *CurScope, SourceLocation Loc)
- : Directive(DKind), DirectiveName(Name), CurScope(CurScope),
- ConstructLoc(Loc) {}
- SharingMapTy() = default;
- };
- using StackTy = SmallVector<SharingMapTy, 4>;
- /// Stack of used declaration and their data-sharing attributes.
- DeclSAMapTy Threadprivates;
- const FunctionScopeInfo *CurrentNonCapturingFunctionScope = nullptr;
- SmallVector<std::pair<StackTy, const FunctionScopeInfo *>, 4> Stack;
- /// true, if check for DSA must be from parent directive, false, if
- /// from current directive.
- OpenMPClauseKind ClauseKindMode = OMPC_unknown;
- Sema &SemaRef;
- bool ForceCapturing = false;
- /// true if all the vaiables in the target executable directives must be
- /// captured by reference.
- bool ForceCaptureByReferenceInTargetExecutable = false;
- CriticalsWithHintsTy Criticals;
- using iterator = StackTy::const_reverse_iterator;
- DSAVarData getDSA(iterator &Iter, ValueDecl *D) const;
- /// Checks if the variable is a local for OpenMP region.
- bool isOpenMPLocal(VarDecl *D, iterator Iter) const;
- bool isStackEmpty() const {
- return Stack.empty() ||
- Stack.back().second != CurrentNonCapturingFunctionScope ||
- Stack.back().first.empty();
- }
- /// Vector of previously declared requires directives
- SmallVector<const OMPRequiresDecl *, 2> RequiresDecls;
- public:
- explicit DSAStackTy(Sema &S) : SemaRef(S) {}
- bool isClauseParsingMode() const { return ClauseKindMode != OMPC_unknown; }
- OpenMPClauseKind getClauseParsingMode() const {
- assert(isClauseParsingMode() && "Must be in clause parsing mode.");
- return ClauseKindMode;
- }
- void setClauseParsingMode(OpenMPClauseKind K) { ClauseKindMode = K; }
- bool isForceVarCapturing() const { return ForceCapturing; }
- void setForceVarCapturing(bool V) { ForceCapturing = V; }
- void setForceCaptureByReferenceInTargetExecutable(bool V) {
- ForceCaptureByReferenceInTargetExecutable = V;
- }
- bool isForceCaptureByReferenceInTargetExecutable() const {
- return ForceCaptureByReferenceInTargetExecutable;
- }
- void push(OpenMPDirectiveKind DKind, const DeclarationNameInfo &DirName,
- Scope *CurScope, SourceLocation Loc) {
- if (Stack.empty() ||
- Stack.back().second != CurrentNonCapturingFunctionScope)
- Stack.emplace_back(StackTy(), CurrentNonCapturingFunctionScope);
- Stack.back().first.emplace_back(DKind, DirName, CurScope, Loc);
- Stack.back().first.back().DefaultAttrLoc = Loc;
- }
- void pop() {
- assert(!Stack.back().first.empty() &&
- "Data-sharing attributes stack is empty!");
- Stack.back().first.pop_back();
- }
- /// Marks that we're started loop parsing.
- void loopInit() {
- assert(isOpenMPLoopDirective(getCurrentDirective()) &&
- "Expected loop-based directive.");
- Stack.back().first.back().LoopStart = true;
- }
- /// Start capturing of the variables in the loop context.
- void loopStart() {
- assert(isOpenMPLoopDirective(getCurrentDirective()) &&
- "Expected loop-based directive.");
- Stack.back().first.back().LoopStart = false;
- }
- /// true, if variables are captured, false otherwise.
- bool isLoopStarted() const {
- assert(isOpenMPLoopDirective(getCurrentDirective()) &&
- "Expected loop-based directive.");
- return !Stack.back().first.back().LoopStart;
- }
- /// Marks (or clears) declaration as possibly loop counter.
- void resetPossibleLoopCounter(const Decl *D = nullptr) {
- Stack.back().first.back().PossiblyLoopCounter =
- D ? D->getCanonicalDecl() : D;
- }
- /// Gets the possible loop counter decl.
- const Decl *getPossiblyLoopCunter() const {
- return Stack.back().first.back().PossiblyLoopCounter;
- }
- /// Start new OpenMP region stack in new non-capturing function.
- void pushFunction() {
- const FunctionScopeInfo *CurFnScope = SemaRef.getCurFunction();
- assert(!isa<CapturingScopeInfo>(CurFnScope));
- CurrentNonCapturingFunctionScope = CurFnScope;
- }
- /// Pop region stack for non-capturing function.
- void popFunction(const FunctionScopeInfo *OldFSI) {
- if (!Stack.empty() && Stack.back().second == OldFSI) {
- assert(Stack.back().first.empty());
- Stack.pop_back();
- }
- CurrentNonCapturingFunctionScope = nullptr;
- for (const FunctionScopeInfo *FSI : llvm::reverse(SemaRef.FunctionScopes)) {
- if (!isa<CapturingScopeInfo>(FSI)) {
- CurrentNonCapturingFunctionScope = FSI;
- break;
- }
- }
- }
- void addCriticalWithHint(const OMPCriticalDirective *D, llvm::APSInt Hint) {
- Criticals.try_emplace(D->getDirectiveName().getAsString(), D, Hint);
- }
- const std::pair<const OMPCriticalDirective *, llvm::APSInt>
- getCriticalWithHint(const DeclarationNameInfo &Name) const {
- auto I = Criticals.find(Name.getAsString());
- if (I != Criticals.end())
- return I->second;
- return std::make_pair(nullptr, llvm::APSInt());
- }
- /// If 'aligned' declaration for given variable \a D was not seen yet,
- /// add it and return NULL; otherwise return previous occurrence's expression
- /// for diagnostics.
- const Expr *addUniqueAligned(const ValueDecl *D, const Expr *NewDE);
- /// Register specified variable as loop control variable.
- void addLoopControlVariable(const ValueDecl *D, VarDecl *Capture);
- /// Check if the specified variable is a loop control variable for
- /// current region.
- /// \return The index of the loop control variable in the list of associated
- /// for-loops (from outer to inner).
- const LCDeclInfo isLoopControlVariable(const ValueDecl *D) const;
- /// Check if the specified variable is a loop control variable for
- /// parent region.
- /// \return The index of the loop control variable in the list of associated
- /// for-loops (from outer to inner).
- const LCDeclInfo isParentLoopControlVariable(const ValueDecl *D) const;
- /// Get the loop control variable for the I-th loop (or nullptr) in
- /// parent directive.
- const ValueDecl *getParentLoopControlVariable(unsigned I) const;
- /// Adds explicit data sharing attribute to the specified declaration.
- void addDSA(const ValueDecl *D, const Expr *E, OpenMPClauseKind A,
- DeclRefExpr *PrivateCopy = nullptr);
- /// Adds additional information for the reduction items with the reduction id
- /// represented as an operator.
- void addTaskgroupReductionData(const ValueDecl *D, SourceRange SR,
- BinaryOperatorKind BOK);
- /// Adds additional information for the reduction items with the reduction id
- /// represented as reduction identifier.
- void addTaskgroupReductionData(const ValueDecl *D, SourceRange SR,
- const Expr *ReductionRef);
- /// Returns the location and reduction operation from the innermost parent
- /// region for the given \p D.
- const DSAVarData
- getTopMostTaskgroupReductionData(const ValueDecl *D, SourceRange &SR,
- BinaryOperatorKind &BOK,
- Expr *&TaskgroupDescriptor) const;
- /// Returns the location and reduction operation from the innermost parent
- /// region for the given \p D.
- const DSAVarData
- getTopMostTaskgroupReductionData(const ValueDecl *D, SourceRange &SR,
- const Expr *&ReductionRef,
- Expr *&TaskgroupDescriptor) const;
- /// Return reduction reference expression for the current taskgroup.
- Expr *getTaskgroupReductionRef() const {
- assert(Stack.back().first.back().Directive == OMPD_taskgroup &&
- "taskgroup reference expression requested for non taskgroup "
- "directive.");
- return Stack.back().first.back().TaskgroupReductionRef;
- }
- /// Checks if the given \p VD declaration is actually a taskgroup reduction
- /// descriptor variable at the \p Level of OpenMP regions.
- bool isTaskgroupReductionRef(const ValueDecl *VD, unsigned Level) const {
- return Stack.back().first[Level].TaskgroupReductionRef &&
- cast<DeclRefExpr>(Stack.back().first[Level].TaskgroupReductionRef)
- ->getDecl() == VD;
- }
- /// Returns data sharing attributes from top of the stack for the
- /// specified declaration.
- const DSAVarData getTopDSA(ValueDecl *D, bool FromParent);
- /// Returns data-sharing attributes for the specified declaration.
- const DSAVarData getImplicitDSA(ValueDecl *D, bool FromParent) const;
- /// Checks if the specified variables has data-sharing attributes which
- /// match specified \a CPred predicate in any directive which matches \a DPred
- /// predicate.
- const DSAVarData
- hasDSA(ValueDecl *D, const llvm::function_ref<bool(OpenMPClauseKind)> CPred,
- const llvm::function_ref<bool(OpenMPDirectiveKind)> DPred,
- bool FromParent) const;
- /// Checks if the specified variables has data-sharing attributes which
- /// match specified \a CPred predicate in any innermost directive which
- /// matches \a DPred predicate.
- const DSAVarData
- hasInnermostDSA(ValueDecl *D,
- const llvm::function_ref<bool(OpenMPClauseKind)> CPred,
- const llvm::function_ref<bool(OpenMPDirectiveKind)> DPred,
- bool FromParent) const;
- /// Checks if the specified variables has explicit data-sharing
- /// attributes which match specified \a CPred predicate at the specified
- /// OpenMP region.
- bool hasExplicitDSA(const ValueDecl *D,
- const llvm::function_ref<bool(OpenMPClauseKind)> CPred,
- unsigned Level, bool NotLastprivate = false) const;
- /// Returns true if the directive at level \Level matches in the
- /// specified \a DPred predicate.
- bool hasExplicitDirective(
- const llvm::function_ref<bool(OpenMPDirectiveKind)> DPred,
- unsigned Level) const;
- /// Finds a directive which matches specified \a DPred predicate.
- bool hasDirective(
- const llvm::function_ref<bool(
- OpenMPDirectiveKind, const DeclarationNameInfo &, SourceLocation)>
- DPred,
- bool FromParent) const;
- /// Returns currently analyzed directive.
- OpenMPDirectiveKind getCurrentDirective() const {
- return isStackEmpty() ? OMPD_unknown : Stack.back().first.back().Directive;
- }
- /// Returns directive kind at specified level.
- OpenMPDirectiveKind getDirective(unsigned Level) const {
- assert(!isStackEmpty() && "No directive at specified level.");
- return Stack.back().first[Level].Directive;
- }
- /// Returns parent directive.
- OpenMPDirectiveKind getParentDirective() const {
- if (isStackEmpty() || Stack.back().first.size() == 1)
- return OMPD_unknown;
- return std::next(Stack.back().first.rbegin())->Directive;
- }
- /// Add requires decl to internal vector
- void addRequiresDecl(OMPRequiresDecl *RD) {
- RequiresDecls.push_back(RD);
- }
- /// Checks for a duplicate clause amongst previously declared requires
- /// directives
- bool hasDuplicateRequiresClause(ArrayRef<OMPClause *> ClauseList) const {
- bool IsDuplicate = false;
- for (OMPClause *CNew : ClauseList) {
- for (const OMPRequiresDecl *D : RequiresDecls) {
- for (const OMPClause *CPrev : D->clauselists()) {
- if (CNew->getClauseKind() == CPrev->getClauseKind()) {
- SemaRef.Diag(CNew->getBeginLoc(),
- diag::err_omp_requires_clause_redeclaration)
- << getOpenMPClauseName(CNew->getClauseKind());
- SemaRef.Diag(CPrev->getBeginLoc(),
- diag::note_omp_requires_previous_clause)
- << getOpenMPClauseName(CPrev->getClauseKind());
- IsDuplicate = true;
- }
- }
- }
- }
- return IsDuplicate;
- }
- /// Set default data sharing attribute to none.
- void setDefaultDSANone(SourceLocation Loc) {
- assert(!isStackEmpty());
- Stack.back().first.back().DefaultAttr = DSA_none;
- Stack.back().first.back().DefaultAttrLoc = Loc;
- }
- /// Set default data sharing attribute to shared.
- void setDefaultDSAShared(SourceLocation Loc) {
- assert(!isStackEmpty());
- Stack.back().first.back().DefaultAttr = DSA_shared;
- Stack.back().first.back().DefaultAttrLoc = Loc;
- }
- /// Set default data mapping attribute to 'tofrom:scalar'.
- void setDefaultDMAToFromScalar(SourceLocation Loc) {
- assert(!isStackEmpty());
- Stack.back().first.back().DefaultMapAttr = DMA_tofrom_scalar;
- Stack.back().first.back().DefaultMapAttrLoc = Loc;
- }
- DefaultDataSharingAttributes getDefaultDSA() const {
- return isStackEmpty() ? DSA_unspecified
- : Stack.back().first.back().DefaultAttr;
- }
- SourceLocation getDefaultDSALocation() const {
- return isStackEmpty() ? SourceLocation()
- : Stack.back().first.back().DefaultAttrLoc;
- }
- DefaultMapAttributes getDefaultDMA() const {
- return isStackEmpty() ? DMA_unspecified
- : Stack.back().first.back().DefaultMapAttr;
- }
- DefaultMapAttributes getDefaultDMAAtLevel(unsigned Level) const {
- return Stack.back().first[Level].DefaultMapAttr;
- }
- SourceLocation getDefaultDMALocation() const {
- return isStackEmpty() ? SourceLocation()
- : Stack.back().first.back().DefaultMapAttrLoc;
- }
- /// Checks if the specified variable is a threadprivate.
- bool isThreadPrivate(VarDecl *D) {
- const DSAVarData DVar = getTopDSA(D, false);
- return isOpenMPThreadPrivate(DVar.CKind);
- }
- /// Marks current region as ordered (it has an 'ordered' clause).
- void setOrderedRegion(bool IsOrdered, const Expr *Param,
- OMPOrderedClause *Clause) {
- assert(!isStackEmpty());
- if (IsOrdered)
- Stack.back().first.back().OrderedRegion.emplace(Param, Clause);
- else
- Stack.back().first.back().OrderedRegion.reset();
- }
- /// Returns true, if region is ordered (has associated 'ordered' clause),
- /// false - otherwise.
- bool isOrderedRegion() const {
- if (isStackEmpty())
- return false;
- return Stack.back().first.rbegin()->OrderedRegion.hasValue();
- }
- /// Returns optional parameter for the ordered region.
- std::pair<const Expr *, OMPOrderedClause *> getOrderedRegionParam() const {
- if (isStackEmpty() ||
- !Stack.back().first.rbegin()->OrderedRegion.hasValue())
- return std::make_pair(nullptr, nullptr);
- return Stack.back().first.rbegin()->OrderedRegion.getValue();
- }
- /// Returns true, if parent region is ordered (has associated
- /// 'ordered' clause), false - otherwise.
- bool isParentOrderedRegion() const {
- if (isStackEmpty() || Stack.back().first.size() == 1)
- return false;
- return std::next(Stack.back().first.rbegin())->OrderedRegion.hasValue();
- }
- /// Returns optional parameter for the ordered region.
- std::pair<const Expr *, OMPOrderedClause *>
- getParentOrderedRegionParam() const {
- if (isStackEmpty() || Stack.back().first.size() == 1 ||
- !std::next(Stack.back().first.rbegin())->OrderedRegion.hasValue())
- return std::make_pair(nullptr, nullptr);
- return std::next(Stack.back().first.rbegin())->OrderedRegion.getValue();
- }
- /// Marks current region as nowait (it has a 'nowait' clause).
- void setNowaitRegion(bool IsNowait = true) {
- assert(!isStackEmpty());
- Stack.back().first.back().NowaitRegion = IsNowait;
- }
- /// Returns true, if parent region is nowait (has associated
- /// 'nowait' clause), false - otherwise.
- bool isParentNowaitRegion() const {
- if (isStackEmpty() || Stack.back().first.size() == 1)
- return false;
- return std::next(Stack.back().first.rbegin())->NowaitRegion;
- }
- /// Marks parent region as cancel region.
- void setParentCancelRegion(bool Cancel = true) {
- if (!isStackEmpty() && Stack.back().first.size() > 1) {
- auto &StackElemRef = *std::next(Stack.back().first.rbegin());
- StackElemRef.CancelRegion |= StackElemRef.CancelRegion || Cancel;
- }
- }
- /// Return true if current region has inner cancel construct.
- bool isCancelRegion() const {
- return isStackEmpty() ? false : Stack.back().first.back().CancelRegion;
- }
- /// Set collapse value for the region.
- void setAssociatedLoops(unsigned Val) {
- assert(!isStackEmpty());
- Stack.back().first.back().AssociatedLoops = Val;
- }
- /// Return collapse value for region.
- unsigned getAssociatedLoops() const {
- return isStackEmpty() ? 0 : Stack.back().first.back().AssociatedLoops;
- }
- /// Marks current target region as one with closely nested teams
- /// region.
- void setParentTeamsRegionLoc(SourceLocation TeamsRegionLoc) {
- if (!isStackEmpty() && Stack.back().first.size() > 1) {
- std::next(Stack.back().first.rbegin())->InnerTeamsRegionLoc =
- TeamsRegionLoc;
- }
- }
- /// Returns true, if current region has closely nested teams region.
- bool hasInnerTeamsRegion() const {
- return getInnerTeamsRegionLoc().isValid();
- }
- /// Returns location of the nested teams region (if any).
- SourceLocation getInnerTeamsRegionLoc() const {
- return isStackEmpty() ? SourceLocation()
- : Stack.back().first.back().InnerTeamsRegionLoc;
- }
- Scope *getCurScope() const {
- return isStackEmpty() ? nullptr : Stack.back().first.back().CurScope;
- }
- SourceLocation getConstructLoc() const {
- return isStackEmpty() ? SourceLocation()
- : Stack.back().first.back().ConstructLoc;
- }
- /// Do the check specified in \a Check to all component lists and return true
- /// if any issue is found.
- bool checkMappableExprComponentListsForDecl(
- const ValueDecl *VD, bool CurrentRegionOnly,
- const llvm::function_ref<
- bool(OMPClauseMappableExprCommon::MappableExprComponentListRef,
- OpenMPClauseKind)>
- Check) const {
- if (isStackEmpty())
- return false;
- auto SI = Stack.back().first.rbegin();
- auto SE = Stack.back().first.rend();
- if (SI == SE)
- return false;
- if (CurrentRegionOnly)
- SE = std::next(SI);
- else
- std::advance(SI, 1);
- for (; SI != SE; ++SI) {
- auto MI = SI->MappedExprComponents.find(VD);
- if (MI != SI->MappedExprComponents.end())
- for (OMPClauseMappableExprCommon::MappableExprComponentListRef L :
- MI->second.Components)
- if (Check(L, MI->second.Kind))
- return true;
- }
- return false;
- }
- /// Do the check specified in \a Check to all component lists at a given level
- /// and return true if any issue is found.
- bool checkMappableExprComponentListsForDeclAtLevel(
- const ValueDecl *VD, unsigned Level,
- const llvm::function_ref<
- bool(OMPClauseMappableExprCommon::MappableExprComponentListRef,
- OpenMPClauseKind)>
- Check) const {
- if (isStackEmpty())
- return false;
- auto StartI = Stack.back().first.begin();
- auto EndI = Stack.back().first.end();
- if (std::distance(StartI, EndI) <= (int)Level)
- return false;
- std::advance(StartI, Level);
- auto MI = StartI->MappedExprComponents.find(VD);
- if (MI != StartI->MappedExprComponents.end())
- for (OMPClauseMappableExprCommon::MappableExprComponentListRef L :
- MI->second.Components)
- if (Check(L, MI->second.Kind))
- return true;
- return false;
- }
- /// Create a new mappable expression component list associated with a given
- /// declaration and initialize it with the provided list of components.
- void addMappableExpressionComponents(
- const ValueDecl *VD,
- OMPClauseMappableExprCommon::MappableExprComponentListRef Components,
- OpenMPClauseKind WhereFoundClauseKind) {
- assert(!isStackEmpty() &&
- "Not expecting to retrieve components from a empty stack!");
- MappedExprComponentTy &MEC =
- Stack.back().first.back().MappedExprComponents[VD];
- // Create new entry and append the new components there.
- MEC.Components.resize(MEC.Components.size() + 1);
- MEC.Components.back().append(Components.begin(), Components.end());
- MEC.Kind = WhereFoundClauseKind;
- }
- unsigned getNestingLevel() const {
- assert(!isStackEmpty());
- return Stack.back().first.size() - 1;
- }
- void addDoacrossDependClause(OMPDependClause *C,
- const OperatorOffsetTy &OpsOffs) {
- assert(!isStackEmpty() && Stack.back().first.size() > 1);
- SharingMapTy &StackElem = *std::next(Stack.back().first.rbegin());
- assert(isOpenMPWorksharingDirective(StackElem.Directive));
- StackElem.DoacrossDepends.try_emplace(C, OpsOffs);
- }
- llvm::iterator_range<DoacrossDependMapTy::const_iterator>
- getDoacrossDependClauses() const {
- assert(!isStackEmpty());
- const SharingMapTy &StackElem = Stack.back().first.back();
- if (isOpenMPWorksharingDirective(StackElem.Directive)) {
- const DoacrossDependMapTy &Ref = StackElem.DoacrossDepends;
- return llvm::make_range(Ref.begin(), Ref.end());
- }
- return llvm::make_range(StackElem.DoacrossDepends.end(),
- StackElem.DoacrossDepends.end());
- }
- };
- bool isParallelOrTaskRegion(OpenMPDirectiveKind DKind) {
- return isOpenMPParallelDirective(DKind) || isOpenMPTaskingDirective(DKind) ||
- isOpenMPTeamsDirective(DKind) || DKind == OMPD_unknown;
- }
- } // namespace
- static const Expr *getExprAsWritten(const Expr *E) {
- if (const auto *FE = dyn_cast<FullExpr>(E))
- E = FE->getSubExpr();
- if (const auto *MTE = dyn_cast<MaterializeTemporaryExpr>(E))
- E = MTE->GetTemporaryExpr();
- while (const auto *Binder = dyn_cast<CXXBindTemporaryExpr>(E))
- E = Binder->getSubExpr();
- if (const auto *ICE = dyn_cast<ImplicitCastExpr>(E))
- E = ICE->getSubExprAsWritten();
- return E->IgnoreParens();
- }
- static Expr *getExprAsWritten(Expr *E) {
- return const_cast<Expr *>(getExprAsWritten(const_cast<const Expr *>(E)));
- }
- static const ValueDecl *getCanonicalDecl(const ValueDecl *D) {
- if (const auto *CED = dyn_cast<OMPCapturedExprDecl>(D))
- if (const auto *ME = dyn_cast<MemberExpr>(getExprAsWritten(CED->getInit())))
- D = ME->getMemberDecl();
- const auto *VD = dyn_cast<VarDecl>(D);
- const auto *FD = dyn_cast<FieldDecl>(D);
- if (VD != nullptr) {
- VD = VD->getCanonicalDecl();
- D = VD;
- } else {
- assert(FD);
- FD = FD->getCanonicalDecl();
- D = FD;
- }
- return D;
- }
- static ValueDecl *getCanonicalDecl(ValueDecl *D) {
- return const_cast<ValueDecl *>(
- getCanonicalDecl(const_cast<const ValueDecl *>(D)));
- }
- DSAStackTy::DSAVarData DSAStackTy::getDSA(iterator &Iter,
- ValueDecl *D) const {
- D = getCanonicalDecl(D);
- auto *VD = dyn_cast<VarDecl>(D);
- const auto *FD = dyn_cast<FieldDecl>(D);
- DSAVarData DVar;
- if (isStackEmpty() || Iter == Stack.back().first.rend()) {
- // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced
- // in a region but not in construct]
- // File-scope or namespace-scope variables referenced in called routines
- // in the region are shared unless they appear in a threadprivate
- // directive.
- if (VD && !VD->isFunctionOrMethodVarDecl() && !isa<ParmVarDecl>(VD))
- DVar.CKind = OMPC_shared;
- // OpenMP [2.9.1.2, Data-sharing Attribute Rules for Variables Referenced
- // in a region but not in construct]
- // Variables with static storage duration that are declared in called
- // routines in the region are shared.
- if (VD && VD->hasGlobalStorage())
- DVar.CKind = OMPC_shared;
- // Non-static data members are shared by default.
- if (FD)
- DVar.CKind = OMPC_shared;
- return DVar;
- }
- // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced
- // in a Construct, C/C++, predetermined, p.1]
- // Variables with automatic storage duration that are declared in a scope
- // inside the construct are private.
- if (VD && isOpenMPLocal(VD, Iter) && VD->isLocalVarDecl() &&
- (VD->getStorageClass() == SC_Auto || VD->getStorageClass() == SC_None)) {
- DVar.CKind = OMPC_private;
- return DVar;
- }
- DVar.DKind = Iter->Directive;
- // Explicitly specified attributes and local variables with predetermined
- // attributes.
- if (Iter->SharingMap.count(D)) {
- const DSAInfo &Data = Iter->SharingMap.lookup(D);
- DVar.RefExpr = Data.RefExpr.getPointer();
- DVar.PrivateCopy = Data.PrivateCopy;
- DVar.CKind = Data.Attributes;
- DVar.ImplicitDSALoc = Iter->DefaultAttrLoc;
- return DVar;
- }
- // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced
- // in a Construct, C/C++, implicitly determined, p.1]
- // In a parallel or task construct, the data-sharing attributes of these
- // variables are determined by the default clause, if present.
- switch (Iter->DefaultAttr) {
- case DSA_shared:
- DVar.CKind = OMPC_shared;
- DVar.ImplicitDSALoc = Iter->DefaultAttrLoc;
- return DVar;
- case DSA_none:
- return DVar;
- case DSA_unspecified:
- // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced
- // in a Construct, implicitly determined, p.2]
- // In a parallel construct, if no default clause is present, these
- // variables are shared.
- DVar.ImplicitDSALoc = Iter->DefaultAttrLoc;
- if (isOpenMPParallelDirective(DVar.DKind) ||
- isOpenMPTeamsDirective(DVar.DKind)) {
- DVar.CKind = OMPC_shared;
- return DVar;
- }
- // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced
- // in a Construct, implicitly determined, p.4]
- // In a task construct, if no default clause is present, a variable that in
- // the enclosing context is determined to be shared by all implicit tasks
- // bound to the current team is shared.
- if (isOpenMPTaskingDirective(DVar.DKind)) {
- DSAVarData DVarTemp;
- iterator I = Iter, E = Stack.back().first.rend();
- do {
- ++I;
- // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables
- // Referenced in a Construct, implicitly determined, p.6]
- // In a task construct, if no default clause is present, a variable
- // whose data-sharing attribute is not determined by the rules above is
- // firstprivate.
- DVarTemp = getDSA(I, D);
- if (DVarTemp.CKind != OMPC_shared) {
- DVar.RefExpr = nullptr;
- DVar.CKind = OMPC_firstprivate;
- return DVar;
- }
- } while (I != E && !isParallelOrTaskRegion(I->Directive));
- DVar.CKind =
- (DVarTemp.CKind == OMPC_unknown) ? OMPC_firstprivate : OMPC_shared;
- return DVar;
- }
- }
- // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced
- // in a Construct, implicitly determined, p.3]
- // For constructs other than task, if no default clause is present, these
- // variables inherit their data-sharing attributes from the enclosing
- // context.
- return getDSA(++Iter, D);
- }
- const Expr *DSAStackTy::addUniqueAligned(const ValueDecl *D,
- const Expr *NewDE) {
- assert(!isStackEmpty() && "Data sharing attributes stack is empty");
- D = getCanonicalDecl(D);
- SharingMapTy &StackElem = Stack.back().first.back();
- auto It = StackElem.AlignedMap.find(D);
- if (It == StackElem.AlignedMap.end()) {
- assert(NewDE && "Unexpected nullptr expr to be added into aligned map");
- StackElem.AlignedMap[D] = NewDE;
- return nullptr;
- }
- assert(It->second && "Unexpected nullptr expr in the aligned map");
- return It->second;
- }
- void DSAStackTy::addLoopControlVariable(const ValueDecl *D, VarDecl *Capture) {
- assert(!isStackEmpty() && "Data-sharing attributes stack is empty");
- D = getCanonicalDecl(D);
- SharingMapTy &StackElem = Stack.back().first.back();
- StackElem.LCVMap.try_emplace(
- D, LCDeclInfo(StackElem.LCVMap.size() + 1, Capture));
- }
- const DSAStackTy::LCDeclInfo
- DSAStackTy::isLoopControlVariable(const ValueDecl *D) const {
- assert(!isStackEmpty() && "Data-sharing attributes stack is empty");
- D = getCanonicalDecl(D);
- const SharingMapTy &StackElem = Stack.back().first.back();
- auto It = StackElem.LCVMap.find(D);
- if (It != StackElem.LCVMap.end())
- return It->second;
- return {0, nullptr};
- }
- const DSAStackTy::LCDeclInfo
- DSAStackTy::isParentLoopControlVariable(const ValueDecl *D) const {
- assert(!isStackEmpty() && Stack.back().first.size() > 1 &&
- "Data-sharing attributes stack is empty");
- D = getCanonicalDecl(D);
- const SharingMapTy &StackElem = *std::next(Stack.back().first.rbegin());
- auto It = StackElem.LCVMap.find(D);
- if (It != StackElem.LCVMap.end())
- return It->second;
- return {0, nullptr};
- }
- const ValueDecl *DSAStackTy::getParentLoopControlVariable(unsigned I) const {
- assert(!isStackEmpty() && Stack.back().first.size() > 1 &&
- "Data-sharing attributes stack is empty");
- const SharingMapTy &StackElem = *std::next(Stack.back().first.rbegin());
- if (StackElem.LCVMap.size() < I)
- return nullptr;
- for (const auto &Pair : StackElem.LCVMap)
- if (Pair.second.first == I)
- return Pair.first;
- return nullptr;
- }
- void DSAStackTy::addDSA(const ValueDecl *D, const Expr *E, OpenMPClauseKind A,
- DeclRefExpr *PrivateCopy) {
- D = getCanonicalDecl(D);
- if (A == OMPC_threadprivate) {
- DSAInfo &Data = Threadprivates[D];
- Data.Attributes = A;
- Data.RefExpr.setPointer(E);
- Data.PrivateCopy = nullptr;
- } else {
- assert(!isStackEmpty() && "Data-sharing attributes stack is empty");
- DSAInfo &Data = Stack.back().first.back().SharingMap[D];
- assert(Data.Attributes == OMPC_unknown || (A == Data.Attributes) ||
- (A == OMPC_firstprivate && Data.Attributes == OMPC_lastprivate) ||
- (A == OMPC_lastprivate && Data.Attributes == OMPC_firstprivate) ||
- (isLoopControlVariable(D).first && A == OMPC_private));
- if (A == OMPC_lastprivate && Data.Attributes == OMPC_firstprivate) {
- Data.RefExpr.setInt(/*IntVal=*/true);
- return;
- }
- const bool IsLastprivate =
- A == OMPC_lastprivate || Data.Attributes == OMPC_lastprivate;
- Data.Attributes = A;
- Data.RefExpr.setPointerAndInt(E, IsLastprivate);
- Data.PrivateCopy = PrivateCopy;
- if (PrivateCopy) {
- DSAInfo &Data =
- Stack.back().first.back().SharingMap[PrivateCopy->getDecl()];
- Data.Attributes = A;
- Data.RefExpr.setPointerAndInt(PrivateCopy, IsLastprivate);
- Data.PrivateCopy = nullptr;
- }
- }
- }
- /// Build a variable declaration for OpenMP loop iteration variable.
- static VarDecl *buildVarDecl(Sema &SemaRef, SourceLocation Loc, QualType Type,
- StringRef Name, const AttrVec *Attrs = nullptr,
- DeclRefExpr *OrigRef = nullptr) {
- DeclContext *DC = SemaRef.CurContext;
- IdentifierInfo *II = &SemaRef.PP.getIdentifierTable().get(Name);
- TypeSourceInfo *TInfo = SemaRef.Context.getTrivialTypeSourceInfo(Type, Loc);
- auto *Decl =
- VarDecl::Create(SemaRef.Context, DC, Loc, Loc, II, Type, TInfo, SC_None);
- if (Attrs) {
- for (specific_attr_iterator<AlignedAttr> I(Attrs->begin()), E(Attrs->end());
- I != E; ++I)
- Decl->addAttr(*I);
- }
- Decl->setImplicit();
- if (OrigRef) {
- Decl->addAttr(
- OMPReferencedVarAttr::CreateImplicit(SemaRef.Context, OrigRef));
- }
- return Decl;
- }
- static DeclRefExpr *buildDeclRefExpr(Sema &S, VarDecl *D, QualType Ty,
- SourceLocation Loc,
- bool RefersToCapture = false) {
- D->setReferenced();
- D->markUsed(S.Context);
- return DeclRefExpr::Create(S.getASTContext(), NestedNameSpecifierLoc(),
- SourceLocation(), D, RefersToCapture, Loc, Ty,
- VK_LValue);
- }
- void DSAStackTy::addTaskgroupReductionData(const ValueDecl *D, SourceRange SR,
- BinaryOperatorKind BOK) {
- D = getCanonicalDecl(D);
- assert(!isStackEmpty() && "Data-sharing attributes stack is empty");
- assert(
- Stack.back().first.back().SharingMap[D].Attributes == OMPC_reduction &&
- "Additional reduction info may be specified only for reduction items.");
- ReductionData &ReductionData = Stack.back().first.back().ReductionMap[D];
- assert(ReductionData.ReductionRange.isInvalid() &&
- Stack.back().first.back().Directive == OMPD_taskgroup &&
- "Additional reduction info may be specified only once for reduction "
- "items.");
- ReductionData.set(BOK, SR);
- Expr *&TaskgroupReductionRef =
- Stack.back().first.back().TaskgroupReductionRef;
- if (!TaskgroupReductionRef) {
- VarDecl *VD = buildVarDecl(SemaRef, SR.getBegin(),
- SemaRef.Context.VoidPtrTy, ".task_red.");
- TaskgroupReductionRef =
- buildDeclRefExpr(SemaRef, VD, SemaRef.Context.VoidPtrTy, SR.getBegin());
- }
- }
- void DSAStackTy::addTaskgroupReductionData(const ValueDecl *D, SourceRange SR,
- const Expr *ReductionRef) {
- D = getCanonicalDecl(D);
- assert(!isStackEmpty() && "Data-sharing attributes stack is empty");
- assert(
- Stack.back().first.back().SharingMap[D].Attributes == OMPC_reduction &&
- "Additional reduction info may be specified only for reduction items.");
- ReductionData &ReductionData = Stack.back().first.back().ReductionMap[D];
- assert(ReductionData.ReductionRange.isInvalid() &&
- Stack.back().first.back().Directive == OMPD_taskgroup &&
- "Additional reduction info may be specified only once for reduction "
- "items.");
- ReductionData.set(ReductionRef, SR);
- Expr *&TaskgroupReductionRef =
- Stack.back().first.back().TaskgroupReductionRef;
- if (!TaskgroupReductionRef) {
- VarDecl *VD = buildVarDecl(SemaRef, SR.getBegin(),
- SemaRef.Context.VoidPtrTy, ".task_red.");
- TaskgroupReductionRef =
- buildDeclRefExpr(SemaRef, VD, SemaRef.Context.VoidPtrTy, SR.getBegin());
- }
- }
- const DSAStackTy::DSAVarData DSAStackTy::getTopMostTaskgroupReductionData(
- const ValueDecl *D, SourceRange &SR, BinaryOperatorKind &BOK,
- Expr *&TaskgroupDescriptor) const {
- D = getCanonicalDecl(D);
- assert(!isStackEmpty() && "Data-sharing attributes stack is empty.");
- if (Stack.back().first.empty())
- return DSAVarData();
- for (iterator I = std::next(Stack.back().first.rbegin(), 1),
- E = Stack.back().first.rend();
- I != E; std::advance(I, 1)) {
- const DSAInfo &Data = I->SharingMap.lookup(D);
- if (Data.Attributes != OMPC_reduction || I->Directive != OMPD_taskgroup)
- continue;
- const ReductionData &ReductionData = I->ReductionMap.lookup(D);
- if (!ReductionData.ReductionOp ||
- ReductionData.ReductionOp.is<const Expr *>())
- return DSAVarData();
- SR = ReductionData.ReductionRange;
- BOK = ReductionData.ReductionOp.get<ReductionData::BOKPtrType>();
- assert(I->TaskgroupReductionRef && "taskgroup reduction reference "
- "expression for the descriptor is not "
- "set.");
- TaskgroupDescriptor = I->TaskgroupReductionRef;
- return DSAVarData(OMPD_taskgroup, OMPC_reduction, Data.RefExpr.getPointer(),
- Data.PrivateCopy, I->DefaultAttrLoc);
- }
- return DSAVarData();
- }
- const DSAStackTy::DSAVarData DSAStackTy::getTopMostTaskgroupReductionData(
- const ValueDecl *D, SourceRange &SR, const Expr *&ReductionRef,
- Expr *&TaskgroupDescriptor) const {
- D = getCanonicalDecl(D);
- assert(!isStackEmpty() && "Data-sharing attributes stack is empty.");
- if (Stack.back().first.empty())
- return DSAVarData();
- for (iterator I = std::next(Stack.back().first.rbegin(), 1),
- E = Stack.back().first.rend();
- I != E; std::advance(I, 1)) {
- const DSAInfo &Data = I->SharingMap.lookup(D);
- if (Data.Attributes != OMPC_reduction || I->Directive != OMPD_taskgroup)
- continue;
- const ReductionData &ReductionData = I->ReductionMap.lookup(D);
- if (!ReductionData.ReductionOp ||
- !ReductionData.ReductionOp.is<const Expr *>())
- return DSAVarData();
- SR = ReductionData.ReductionRange;
- ReductionRef = ReductionData.ReductionOp.get<const Expr *>();
- assert(I->TaskgroupReductionRef && "taskgroup reduction reference "
- "expression for the descriptor is not "
- "set.");
- TaskgroupDescriptor = I->TaskgroupReductionRef;
- return DSAVarData(OMPD_taskgroup, OMPC_reduction, Data.RefExpr.getPointer(),
- Data.PrivateCopy, I->DefaultAttrLoc);
- }
- return DSAVarData();
- }
- bool DSAStackTy::isOpenMPLocal(VarDecl *D, iterator Iter) const {
- D = D->getCanonicalDecl();
- if (!isStackEmpty()) {
- iterator I = Iter, E = Stack.back().first.rend();
- Scope *TopScope = nullptr;
- while (I != E && !isParallelOrTaskRegion(I->Directive) &&
- !isOpenMPTargetExecutionDirective(I->Directive))
- ++I;
- if (I == E)
- return false;
- TopScope = I->CurScope ? I->CurScope->getParent() : nullptr;
- Scope *CurScope = getCurScope();
- while (CurScope != TopScope && !CurScope->isDeclScope(D))
- CurScope = CurScope->getParent();
- return CurScope != TopScope;
- }
- return false;
- }
- const DSAStackTy::DSAVarData DSAStackTy::getTopDSA(ValueDecl *D,
- bool FromParent) {
- D = getCanonicalDecl(D);
- DSAVarData DVar;
- auto *VD = dyn_cast<VarDecl>(D);
- auto TI = Threadprivates.find(D);
- if (TI != Threadprivates.end()) {
- DVar.RefExpr = TI->getSecond().RefExpr.getPointer();
- DVar.CKind = OMPC_threadprivate;
- return DVar;
- }
- if (VD && VD->hasAttr<OMPThreadPrivateDeclAttr>()) {
- DVar.RefExpr = buildDeclRefExpr(
- SemaRef, VD, D->getType().getNonReferenceType(),
- VD->getAttr<OMPThreadPrivateDeclAttr>()->getLocation());
- DVar.CKind = OMPC_threadprivate;
- addDSA(D, DVar.RefExpr, OMPC_threadprivate);
- return DVar;
- }
- // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced
- // in a Construct, C/C++, predetermined, p.1]
- // Variables appearing in threadprivate directives are threadprivate.
- if ((VD && VD->getTLSKind() != VarDecl::TLS_None &&
- !(VD->hasAttr<OMPThreadPrivateDeclAttr>() &&
- SemaRef.getLangOpts().OpenMPUseTLS &&
- SemaRef.getASTContext().getTargetInfo().isTLSSupported())) ||
- (VD && VD->getStorageClass() == SC_Register &&
- VD->hasAttr<AsmLabelAttr>() && !VD->isLocalVarDecl())) {
- DVar.RefExpr = buildDeclRefExpr(
- SemaRef, VD, D->getType().getNonReferenceType(), D->getLocation());
- DVar.CKind = OMPC_threadprivate;
- addDSA(D, DVar.RefExpr, OMPC_threadprivate);
- return DVar;
- }
- if (SemaRef.getLangOpts().OpenMPCUDAMode && VD &&
- VD->isLocalVarDeclOrParm() && !isStackEmpty() &&
- !isLoopControlVariable(D).first) {
- iterator IterTarget =
- std::find_if(Stack.back().first.rbegin(), Stack.back().first.rend(),
- [](const SharingMapTy &Data) {
- return isOpenMPTargetExecutionDirective(Data.Directive);
- });
- if (IterTarget != Stack.back().first.rend()) {
- iterator ParentIterTarget = std::next(IterTarget, 1);
- for (iterator Iter = Stack.back().first.rbegin();
- Iter != ParentIterTarget; std::advance(Iter, 1)) {
- if (isOpenMPLocal(VD, Iter)) {
- DVar.RefExpr =
- buildDeclRefExpr(SemaRef, VD, D->getType().getNonReferenceType(),
- D->getLocation());
- DVar.CKind = OMPC_threadprivate;
- return DVar;
- }
- }
- if (!isClauseParsingMode() || IterTarget != Stack.back().first.rbegin()) {
- auto DSAIter = IterTarget->SharingMap.find(D);
- if (DSAIter != IterTarget->SharingMap.end() &&
- isOpenMPPrivate(DSAIter->getSecond().Attributes)) {
- DVar.RefExpr = DSAIter->getSecond().RefExpr.getPointer();
- DVar.CKind = OMPC_threadprivate;
- return DVar;
- }
- iterator End = Stack.back().first.rend();
- if (!SemaRef.isOpenMPCapturedByRef(
- D, std::distance(ParentIterTarget, End))) {
- DVar.RefExpr =
- buildDeclRefExpr(SemaRef, VD, D->getType().getNonReferenceType(),
- IterTarget->ConstructLoc);
- DVar.CKind = OMPC_threadprivate;
- return DVar;
- }
- }
- }
- }
- if (isStackEmpty())
- // Not in OpenMP execution region and top scope was already checked.
- return DVar;
- // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced
- // in a Construct, C/C++, predetermined, p.4]
- // Static data members are shared.
- // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced
- // in a Construct, C/C++, predetermined, p.7]
- // Variables with static storage duration that are declared in a scope
- // inside the construct are shared.
- auto &&MatchesAlways = [](OpenMPDirectiveKind) { return true; };
- if (VD && VD->isStaticDataMember()) {
- DSAVarData DVarTemp = hasDSA(D, isOpenMPPrivate, MatchesAlways, FromParent);
- if (DVarTemp.CKind != OMPC_unknown && DVarTemp.RefExpr)
- return DVar;
- DVar.CKind = OMPC_shared;
- return DVar;
- }
- QualType Type = D->getType().getNonReferenceType().getCanonicalType();
- bool IsConstant = Type.isConstant(SemaRef.getASTContext());
- Type = SemaRef.getASTContext().getBaseElementType(Type);
- // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced
- // in a Construct, C/C++, predetermined, p.6]
- // Variables with const qualified type having no mutable member are
- // shared.
- const CXXRecordDecl *RD =
- SemaRef.getLangOpts().CPlusPlus ? Type->getAsCXXRecordDecl() : nullptr;
- if (const auto *CTSD = dyn_cast_or_null<ClassTemplateSpecializationDecl>(RD))
- if (const ClassTemplateDecl *CTD = CTSD->getSpecializedTemplate())
- RD = CTD->getTemplatedDecl();
- if (IsConstant &&
- !(SemaRef.getLangOpts().CPlusPlus && RD && RD->hasDefinition() &&
- RD->hasMutableFields())) {
- // Variables with const-qualified type having no mutable member may be
- // listed in a firstprivate clause, even if they are static data members.
- DSAVarData DVarTemp =
- hasDSA(D, [](OpenMPClauseKind C) { return C == OMPC_firstprivate; },
- MatchesAlways, FromParent);
- if (DVarTemp.CKind == OMPC_firstprivate && DVarTemp.RefExpr)
- return DVarTemp;
- DVar.CKind = OMPC_shared;
- return DVar;
- }
- // Explicitly specified attributes and local variables with predetermined
- // attributes.
- iterator I = Stack.back().first.rbegin();
- iterator EndI = Stack.back().first.rend();
- if (FromParent && I != EndI)
- std::advance(I, 1);
- auto It = I->SharingMap.find(D);
- if (It != I->SharingMap.end()) {
- const DSAInfo &Data = It->getSecond();
- DVar.RefExpr = Data.RefExpr.getPointer();
- DVar.PrivateCopy = Data.PrivateCopy;
- DVar.CKind = Data.Attributes;
- DVar.ImplicitDSALoc = I->DefaultAttrLoc;
- DVar.DKind = I->Directive;
- }
- return DVar;
- }
- const DSAStackTy::DSAVarData DSAStackTy::getImplicitDSA(ValueDecl *D,
- bool FromParent) const {
- if (isStackEmpty()) {
- iterator I;
- return getDSA(I, D);
- }
- D = getCanonicalDecl(D);
- iterator StartI = Stack.back().first.rbegin();
- iterator EndI = Stack.back().first.rend();
- if (FromParent && StartI != EndI)
- std::advance(StartI, 1);
- return getDSA(StartI, D);
- }
- const DSAStackTy::DSAVarData
- DSAStackTy::hasDSA(ValueDecl *D,
- const llvm::function_ref<bool(OpenMPClauseKind)> CPred,
- const llvm::function_ref<bool(OpenMPDirectiveKind)> DPred,
- bool FromParent) const {
- if (isStackEmpty())
- return {};
- D = getCanonicalDecl(D);
- iterator I = Stack.back().first.rbegin();
- iterator EndI = Stack.back().first.rend();
- if (FromParent && I != EndI)
- std::advance(I, 1);
- for (; I != EndI; std::advance(I, 1)) {
- if (!DPred(I->Directive) && !isParallelOrTaskRegion(I->Directive))
- continue;
- iterator NewI = I;
- DSAVarData DVar = getDSA(NewI, D);
- if (I == NewI && CPred(DVar.CKind))
- return DVar;
- }
- return {};
- }
- const DSAStackTy::DSAVarData DSAStackTy::hasInnermostDSA(
- ValueDecl *D, const llvm::function_ref<bool(OpenMPClauseKind)> CPred,
- const llvm::function_ref<bool(OpenMPDirectiveKind)> DPred,
- bool FromParent) const {
- if (isStackEmpty())
- return {};
- D = getCanonicalDecl(D);
- iterator StartI = Stack.back().first.rbegin();
- iterator EndI = Stack.back().first.rend();
- if (FromParent && StartI != EndI)
- std::advance(StartI, 1);
- if (StartI == EndI || !DPred(StartI->Directive))
- return {};
- iterator NewI = StartI;
- DSAVarData DVar = getDSA(NewI, D);
- return (NewI == StartI && CPred(DVar.CKind)) ? DVar : DSAVarData();
- }
- bool DSAStackTy::hasExplicitDSA(
- const ValueDecl *D, const llvm::function_ref<bool(OpenMPClauseKind)> CPred,
- unsigned Level, bool NotLastprivate) const {
- if (isStackEmpty())
- return false;
- D = getCanonicalDecl(D);
- auto StartI = Stack.back().first.begin();
- auto EndI = Stack.back().first.end();
- if (std::distance(StartI, EndI) <= (int)Level)
- return false;
- std::advance(StartI, Level);
- auto I = StartI->SharingMap.find(D);
- if ((I != StartI->SharingMap.end()) &&
- I->getSecond().RefExpr.getPointer() &&
- CPred(I->getSecond().Attributes) &&
- (!NotLastprivate || !I->getSecond().RefExpr.getInt()))
- return true;
- // Check predetermined rules for the loop control variables.
- auto LI = StartI->LCVMap.find(D);
- if (LI != StartI->LCVMap.end())
- return CPred(OMPC_private);
- return false;
- }
- bool DSAStackTy::hasExplicitDirective(
- const llvm::function_ref<bool(OpenMPDirectiveKind)> DPred,
- unsigned Level) const {
- if (isStackEmpty())
- return false;
- auto StartI = Stack.back().first.begin();
- auto EndI = Stack.back().first.end();
- if (std::distance(StartI, EndI) <= (int)Level)
- return false;
- std::advance(StartI, Level);
- return DPred(StartI->Directive);
- }
- bool DSAStackTy::hasDirective(
- const llvm::function_ref<bool(OpenMPDirectiveKind,
- const DeclarationNameInfo &, SourceLocation)>
- DPred,
- bool FromParent) const {
- // We look only in the enclosing region.
- if (isStackEmpty())
- return false;
- auto StartI = std::next(Stack.back().first.rbegin());
- auto EndI = Stack.back().first.rend();
- if (FromParent && StartI != EndI)
- StartI = std::next(StartI);
- for (auto I = StartI, EE = EndI; I != EE; ++I) {
- if (DPred(I->Directive, I->DirectiveName, I->ConstructLoc))
- return true;
- }
- return false;
- }
- void Sema::InitDataSharingAttributesStack() {
- VarDataSharingAttributesStack = new DSAStackTy(*this);
- }
- #define DSAStack static_cast<DSAStackTy *>(VarDataSharingAttributesStack)
- void Sema::pushOpenMPFunctionRegion() {
- DSAStack->pushFunction();
- }
- void Sema::popOpenMPFunctionRegion(const FunctionScopeInfo *OldFSI) {
- DSAStack->popFunction(OldFSI);
- }
- bool Sema::isOpenMPCapturedByRef(const ValueDecl *D, unsigned Level) const {
- assert(LangOpts.OpenMP && "OpenMP is not allowed");
- ASTContext &Ctx = getASTContext();
- bool IsByRef = true;
- // Find the directive that is associated with the provided scope.
- D = cast<ValueDecl>(D->getCanonicalDecl());
- QualType Ty = D->getType();
- if (DSAStack->hasExplicitDirective(isOpenMPTargetExecutionDirective, Level)) {
- // This table summarizes how a given variable should be passed to the device
- // given its type and the clauses where it appears. This table is based on
- // the description in OpenMP 4.5 [2.10.4, target Construct] and
- // OpenMP 4.5 [2.15.5, Data-mapping Attribute Rules and Clauses].
- //
- // =========================================================================
- // | type | defaultmap | pvt | first | is_device_ptr | map | res. |
- // | |(tofrom:scalar)| | pvt | | | |
- // =========================================================================
- // | scl | | | | - | | bycopy|
- // | scl | | - | x | - | - | bycopy|
- // | scl | | x | - | - | - | null |
- // | scl | x | | | - | | byref |
- // | scl | x | - | x | - | - | bycopy|
- // | scl | x | x | - | - | - | null |
- // | scl | | - | - | - | x | byref |
- // | scl | x | - | - | - | x | byref |
- //
- // | agg | n.a. | | | - | | byref |
- // | agg | n.a. | - | x | - | - | byref |
- // | agg | n.a. | x | - | - | - | null |
- // | agg | n.a. | - | - | - | x | byref |
- // | agg | n.a. | - | - | - | x[] | byref |
- //
- // | ptr | n.a. | | | - | | bycopy|
- // | ptr | n.a. | - | x | - | - | bycopy|
- // | ptr | n.a. | x | - | - | - | null |
- // | ptr | n.a. | - | - | - | x | byref |
- // | ptr | n.a. | - | - | - | x[] | bycopy|
- // | ptr | n.a. | - | - | x | | bycopy|
- // | ptr | n.a. | - | - | x | x | bycopy|
- // | ptr | n.a. | - | - | x | x[] | bycopy|
- // =========================================================================
- // Legend:
- // scl - scalar
- // ptr - pointer
- // agg - aggregate
- // x - applies
- // - - invalid in this combination
- // [] - mapped with an array section
- // byref - should be mapped by reference
- // byval - should be mapped by value
- // null - initialize a local variable to null on the device
- //
- // Observations:
- // - All scalar declarations that show up in a map clause have to be passed
- // by reference, because they may have been mapped in the enclosing data
- // environment.
- // - If the scalar value does not fit the size of uintptr, it has to be
- // passed by reference, regardless the result in the table above.
- // - For pointers mapped by value that have either an implicit map or an
- // array section, the runtime library may pass the NULL value to the
- // device instead of the value passed to it by the compiler.
- if (Ty->isReferenceType())
- Ty = Ty->castAs<ReferenceType>()->getPointeeType();
- // Locate map clauses and see if the variable being captured is referred to
- // in any of those clauses. Here we only care about variables, not fields,
- // because fields are part of aggregates.
- bool IsVariableUsedInMapClause = false;
- bool IsVariableAssociatedWithSection = false;
- DSAStack->checkMappableExprComponentListsForDeclAtLevel(
- D, Level,
- [&IsVariableUsedInMapClause, &IsVariableAssociatedWithSection, D](
- OMPClauseMappableExprCommon::MappableExprComponentListRef
- MapExprComponents,
- OpenMPClauseKind WhereFoundClauseKind) {
- // Only the map clause information influences how a variable is
- // captured. E.g. is_device_ptr does not require changing the default
- // behavior.
- if (WhereFoundClauseKind != OMPC_map)
- return false;
- auto EI = MapExprComponents.rbegin();
- auto EE = MapExprComponents.rend();
- assert(EI != EE && "Invalid map expression!");
- if (isa<DeclRefExpr>(EI->getAssociatedExpression()))
- IsVariableUsedInMapClause |= EI->getAssociatedDeclaration() == D;
- ++EI;
- if (EI == EE)
- return false;
- if (isa<ArraySubscriptExpr>(EI->getAssociatedExpression()) ||
- isa<OMPArraySectionExpr>(EI->getAssociatedExpression()) ||
- isa<MemberExpr>(EI->getAssociatedExpression())) {
- IsVariableAssociatedWithSection = true;
- // There is nothing more we need to know about this variable.
- return true;
- }
- // Keep looking for more map info.
- return false;
- });
- if (IsVariableUsedInMapClause) {
- // If variable is identified in a map clause it is always captured by
- // reference except if it is a pointer that is dereferenced somehow.
- IsByRef = !(Ty->isPointerType() && IsVariableAssociatedWithSection);
- } else {
- // By default, all the data that has a scalar type is mapped by copy
- // (except for reduction variables).
- IsByRef =
- (DSAStack->isForceCaptureByReferenceInTargetExecutable() &&
- !Ty->isAnyPointerType()) ||
- !Ty->isScalarType() ||
- DSAStack->getDefaultDMAAtLevel(Level) == DMA_tofrom_scalar ||
- DSAStack->hasExplicitDSA(
- D, [](OpenMPClauseKind K) { return K == OMPC_reduction; }, Level);
- }
- }
- if (IsByRef && Ty.getNonReferenceType()->isScalarType()) {
- IsByRef =
- ((DSAStack->isForceCaptureByReferenceInTargetExecutable() &&
- !Ty->isAnyPointerType()) ||
- !DSAStack->hasExplicitDSA(
- D,
- [](OpenMPClauseKind K) -> bool { return K == OMPC_firstprivate; },
- Level, /*NotLastprivate=*/true)) &&
- // If the variable is artificial and must be captured by value - try to
- // capture by value.
- !(isa<OMPCapturedExprDecl>(D) && !D->hasAttr<OMPCaptureNoInitAttr>() &&
- !cast<OMPCapturedExprDecl>(D)->getInit()->isGLValue());
- }
- // When passing data by copy, we need to make sure it fits the uintptr size
- // and alignment, because the runtime library only deals with uintptr types.
- // If it does not fit the uintptr size, we need to pass the data by reference
- // instead.
- if (!IsByRef &&
- (Ctx.getTypeSizeInChars(Ty) >
- Ctx.getTypeSizeInChars(Ctx.getUIntPtrType()) ||
- Ctx.getDeclAlign(D) > Ctx.getTypeAlignInChars(Ctx.getUIntPtrType()))) {
- IsByRef = true;
- }
- return IsByRef;
- }
- unsigned Sema::getOpenMPNestingLevel() const {
- assert(getLangOpts().OpenMP);
- return DSAStack->getNestingLevel();
- }
- bool Sema::isInOpenMPTargetExecutionDirective() const {
- return (isOpenMPTargetExecutionDirective(DSAStack->getCurrentDirective()) &&
- !DSAStack->isClauseParsingMode()) ||
- DSAStack->hasDirective(
- [](OpenMPDirectiveKind K, const DeclarationNameInfo &,
- SourceLocation) -> bool {
- return isOpenMPTargetExecutionDirective(K);
- },
- false);
- }
- VarDecl *Sema::isOpenMPCapturedDecl(ValueDecl *D) {
- assert(LangOpts.OpenMP && "OpenMP is not allowed");
- D = getCanonicalDecl(D);
- // If we are attempting to capture a global variable in a directive with
- // 'target' we return true so that this global is also mapped to the device.
- //
- auto *VD = dyn_cast<VarDecl>(D);
- if (VD && !VD->hasLocalStorage()) {
- if (isInOpenMPDeclareTargetContext() &&
- (getCurCapturedRegion() || getCurBlock() || getCurLambda())) {
- // Try to mark variable as declare target if it is used in capturing
- // regions.
- if (!OMPDeclareTargetDeclAttr::isDeclareTargetDeclaration(VD))
- checkDeclIsAllowedInOpenMPTarget(nullptr, VD);
- return nullptr;
- } else if (isInOpenMPTargetExecutionDirective()) {
- // If the declaration is enclosed in a 'declare target' directive,
- // then it should not be captured.
- //
- if (OMPDeclareTargetDeclAttr::isDeclareTargetDeclaration(VD))
- return nullptr;
- return VD;
- }
- }
- // Capture variables captured by reference in lambdas for target-based
- // directives.
- if (VD && !DSAStack->isClauseParsingMode()) {
- if (const auto *RD = VD->getType()
- .getCanonicalType()
- .getNonReferenceType()
- ->getAsCXXRecordDecl()) {
- bool SavedForceCaptureByReferenceInTargetExecutable =
- DSAStack->isForceCaptureByReferenceInTargetExecutable();
- DSAStack->setForceCaptureByReferenceInTargetExecutable(/*V=*/true);
- if (RD->isLambda()) {
- llvm::DenseMap<const VarDecl *, FieldDecl *> Captures;
- FieldDecl *ThisCapture;
- RD->getCaptureFields(Captures, ThisCapture);
- for (const LambdaCapture &LC : RD->captures()) {
- if (LC.getCaptureKind() == LCK_ByRef) {
- VarDecl *VD = LC.getCapturedVar();
- DeclContext *VDC = VD->getDeclContext();
- if (!VDC->Encloses(CurContext))
- continue;
- DSAStackTy::DSAVarData DVarPrivate =
- DSAStack->getTopDSA(VD, /*FromParent=*/false);
- // Do not capture already captured variables.
- if (!OMPDeclareTargetDeclAttr::isDeclareTargetDeclaration(VD) &&
- DVarPrivate.CKind == OMPC_unknown &&
- !DSAStack->checkMappableExprComponentListsForDecl(
- D, /*CurrentRegionOnly=*/true,
- [](OMPClauseMappableExprCommon::
- MappableExprComponentListRef,
- OpenMPClauseKind) { return true; }))
- MarkVariableReferenced(LC.getLocation(), LC.getCapturedVar());
- } else if (LC.getCaptureKind() == LCK_This) {
- QualType ThisTy = getCurrentThisType();
- if (!ThisTy.isNull() &&
- Context.typesAreCompatible(ThisTy, ThisCapture->getType()))
- CheckCXXThisCapture(LC.getLocation());
- }
- }
- }
- DSAStack->setForceCaptureByReferenceInTargetExecutable(
- SavedForceCaptureByReferenceInTargetExecutable);
- }
- }
- if (DSAStack->getCurrentDirective() != OMPD_unknown &&
- (!DSAStack->isClauseParsingMode() ||
- DSAStack->getParentDirective() != OMPD_unknown)) {
- auto &&Info = DSAStack->isLoopControlVariable(D);
- if (Info.first ||
- (VD && VD->hasLocalStorage() &&
- isParallelOrTaskRegion(DSAStack->getCurrentDirective())) ||
- (VD && DSAStack->isForceVarCapturing()))
- return VD ? VD : Info.second;
- DSAStackTy::DSAVarData DVarPrivate =
- DSAStack->getTopDSA(D, DSAStack->isClauseParsingMode());
- if (DVarPrivate.CKind != OMPC_unknown && isOpenMPPrivate(DVarPrivate.CKind))
- return VD ? VD : cast<VarDecl>(DVarPrivate.PrivateCopy->getDecl());
- DVarPrivate = DSAStack->hasDSA(D, isOpenMPPrivate,
- [](OpenMPDirectiveKind) { return true; },
- DSAStack->isClauseParsingMode());
- if (DVarPrivate.CKind != OMPC_unknown)
- return VD ? VD : cast<VarDecl>(DVarPrivate.PrivateCopy->getDecl());
- }
- return nullptr;
- }
- void Sema::adjustOpenMPTargetScopeIndex(unsigned &FunctionScopesIndex,
- unsigned Level) const {
- SmallVector<OpenMPDirectiveKind, 4> Regions;
- getOpenMPCaptureRegions(Regions, DSAStack->getDirective(Level));
- FunctionScopesIndex -= Regions.size();
- }
- void Sema::startOpenMPLoop() {
- assert(LangOpts.OpenMP && "OpenMP must be enabled.");
- if (isOpenMPLoopDirective(DSAStack->getCurrentDirective()))
- DSAStack->loopInit();
- }
- bool Sema::isOpenMPPrivateDecl(const ValueDecl *D, unsigned Level) const {
- assert(LangOpts.OpenMP && "OpenMP is not allowed");
- if (isOpenMPLoopDirective(DSAStack->getCurrentDirective())) {
- if (DSAStack->getAssociatedLoops() > 0 &&
- !DSAStack->isLoopStarted()) {
- DSAStack->resetPossibleLoopCounter(D);
- DSAStack->loopStart();
- return true;
- }
- if ((DSAStack->getPossiblyLoopCunter() == D->getCanonicalDecl() ||
- DSAStack->isLoopControlVariable(D).first) &&
- !DSAStack->hasExplicitDSA(
- D, [](OpenMPClauseKind K) { return K != OMPC_private; }, Level) &&
- !isOpenMPSimdDirective(DSAStack->getCurrentDirective()))
- return true;
- }
- return DSAStack->hasExplicitDSA(
- D, [](OpenMPClauseKind K) { return K == OMPC_private; }, Level) ||
- (DSAStack->isClauseParsingMode() &&
- DSAStack->getClauseParsingMode() == OMPC_private) ||
- // Consider taskgroup reduction descriptor variable a private to avoid
- // possible capture in the region.
- (DSAStack->hasExplicitDirective(
- [](OpenMPDirectiveKind K) { return K == OMPD_taskgroup; },
- Level) &&
- DSAStack->isTaskgroupReductionRef(D, Level));
- }
- void Sema::setOpenMPCaptureKind(FieldDecl *FD, const ValueDecl *D,
- unsigned Level) {
- assert(LangOpts.OpenMP && "OpenMP is not allowed");
- D = getCanonicalDecl(D);
- OpenMPClauseKind OMPC = OMPC_unknown;
- for (unsigned I = DSAStack->getNestingLevel() + 1; I > Level; --I) {
- const unsigned NewLevel = I - 1;
- if (DSAStack->hasExplicitDSA(D,
- [&OMPC](const OpenMPClauseKind K) {
- if (isOpenMPPrivate(K)) {
- OMPC = K;
- return true;
- }
- return false;
- },
- NewLevel))
- break;
- if (DSAStack->checkMappableExprComponentListsForDeclAtLevel(
- D, NewLevel,
- [](OMPClauseMappableExprCommon::MappableExprComponentListRef,
- OpenMPClauseKind) { return true; })) {
- OMPC = OMPC_map;
- break;
- }
- if (DSAStack->hasExplicitDirective(isOpenMPTargetExecutionDirective,
- NewLevel)) {
- OMPC = OMPC_map;
- if (D->getType()->isScalarType() &&
- DSAStack->getDefaultDMAAtLevel(NewLevel) !=
- DefaultMapAttributes::DMA_tofrom_scalar)
- OMPC = OMPC_firstprivate;
- break;
- }
- }
- if (OMPC != OMPC_unknown)
- FD->addAttr(OMPCaptureKindAttr::CreateImplicit(Context, OMPC));
- }
- bool Sema::isOpenMPTargetCapturedDecl(const ValueDecl *D,
- unsigned Level) const {
- assert(LangOpts.OpenMP && "OpenMP is not allowed");
- // Return true if the current level is no longer enclosed in a target region.
- const auto *VD = dyn_cast<VarDecl>(D);
- return VD && !VD->hasLocalStorage() &&
- DSAStack->hasExplicitDirective(isOpenMPTargetExecutionDirective,
- Level);
- }
- void Sema::DestroyDataSharingAttributesStack() { delete DSAStack; }
- void Sema::StartOpenMPDSABlock(OpenMPDirectiveKind DKind,
- const DeclarationNameInfo &DirName,
- Scope *CurScope, SourceLocation Loc) {
- DSAStack->push(DKind, DirName, CurScope, Loc);
- PushExpressionEvaluationContext(
- ExpressionEvaluationContext::PotentiallyEvaluated);
- }
- void Sema::StartOpenMPClause(OpenMPClauseKind K) {
- DSAStack->setClauseParsingMode(K);
- }
- void Sema::EndOpenMPClause() {
- DSAStack->setClauseParsingMode(/*K=*/OMPC_unknown);
- }
- void Sema::EndOpenMPDSABlock(Stmt *CurDirective) {
- // OpenMP [2.14.3.5, Restrictions, C/C++, p.1]
- // A variable of class type (or array thereof) that appears in a lastprivate
- // clause requires an accessible, unambiguous default constructor for the
- // class type, unless the list item is also specified in a firstprivate
- // clause.
- if (const auto *D = dyn_cast_or_null<OMPExecutableDirective>(CurDirective)) {
- for (OMPClause *C : D->clauses()) {
- if (auto *Clause = dyn_cast<OMPLastprivateClause>(C)) {
- SmallVector<Expr *, 8> PrivateCopies;
- for (Expr *DE : Clause->varlists()) {
- if (DE->isValueDependent() || DE->isTypeDependent()) {
- PrivateCopies.push_back(nullptr);
- continue;
- }
- auto *DRE = cast<DeclRefExpr>(DE->IgnoreParens());
- auto *VD = cast<VarDecl>(DRE->getDecl());
- QualType Type = VD->getType().getNonReferenceType();
- const DSAStackTy::DSAVarData DVar =
- DSAStack->getTopDSA(VD, /*FromParent=*/false);
- if (DVar.CKind == OMPC_lastprivate) {
- // Generate helper private variable and initialize it with the
- // default value. The address of the original variable is replaced
- // by the address of the new private variable in CodeGen. This new
- // variable is not added to IdResolver, so the code in the OpenMP
- // region uses original variable for proper diagnostics.
- VarDecl *VDPrivate = buildVarDecl(
- *this, DE->getExprLoc(), Type.getUnqualifiedType(),
- VD->getName(), VD->hasAttrs() ? &VD->getAttrs() : nullptr, DRE);
- ActOnUninitializedDecl(VDPrivate);
- if (VDPrivate->isInvalidDecl())
- continue;
- PrivateCopies.push_back(buildDeclRefExpr(
- *this, VDPrivate, DE->getType(), DE->getExprLoc()));
- } else {
- // The variable is also a firstprivate, so initialization sequence
- // for private copy is generated already.
- PrivateCopies.push_back(nullptr);
- }
- }
- // Set initializers to private copies if no errors were found.
- if (PrivateCopies.size() == Clause->varlist_size())
- Clause->setPrivateCopies(PrivateCopies);
- }
- }
- }
- DSAStack->pop();
- DiscardCleanupsInEvaluationContext();
- PopExpressionEvaluationContext();
- }
- static bool FinishOpenMPLinearClause(OMPLinearClause &Clause, DeclRefExpr *IV,
- Expr *NumIterations, Sema &SemaRef,
- Scope *S, DSAStackTy *Stack);
- namespace {
- class VarDeclFilterCCC final : public CorrectionCandidateCallback {
- private:
- Sema &SemaRef;
- public:
- explicit VarDeclFilterCCC(Sema &S) : SemaRef(S) {}
- bool ValidateCandidate(const TypoCorrection &Candidate) override {
- NamedDecl *ND = Candidate.getCorrectionDecl();
- if (const auto *VD = dyn_cast_or_null<VarDecl>(ND)) {
- return VD->hasGlobalStorage() &&
- SemaRef.isDeclInScope(ND, SemaRef.getCurLexicalContext(),
- SemaRef.getCurScope());
- }
- return false;
- }
- };
- class VarOrFuncDeclFilterCCC final : public CorrectionCandidateCallback {
- private:
- Sema &SemaRef;
- public:
- explicit VarOrFuncDeclFilterCCC(Sema &S) : SemaRef(S) {}
- bool ValidateCandidate(const TypoCorrection &Candidate) override {
- NamedDecl *ND = Candidate.getCorrectionDecl();
- if (ND && (isa<VarDecl>(ND) || isa<FunctionDecl>(ND))) {
- return SemaRef.isDeclInScope(ND, SemaRef.getCurLexicalContext(),
- SemaRef.getCurScope());
- }
- return false;
- }
- };
- } // namespace
- ExprResult Sema::ActOnOpenMPIdExpression(Scope *CurScope,
- CXXScopeSpec &ScopeSpec,
- const DeclarationNameInfo &Id) {
- LookupResult Lookup(*this, Id, LookupOrdinaryName);
- LookupParsedName(Lookup, CurScope, &ScopeSpec, true);
- if (Lookup.isAmbiguous())
- return ExprError();
- VarDecl *VD;
- if (!Lookup.isSingleResult()) {
- if (TypoCorrection Corrected = CorrectTypo(
- Id, LookupOrdinaryName, CurScope, nullptr,
- llvm::make_unique<VarDeclFilterCCC>(*this), CTK_ErrorRecovery)) {
- diagnoseTypo(Corrected,
- PDiag(Lookup.empty()
- ? diag::err_undeclared_var_use_suggest
- : diag::err_omp_expected_var_arg_suggest)
- << Id.getName());
- VD = Corrected.getCorrectionDeclAs<VarDecl>();
- } else {
- Diag(Id.getLoc(), Lookup.empty() ? diag::err_undeclared_var_use
- : diag::err_omp_expected_var_arg)
- << Id.getName();
- return ExprError();
- }
- } else if (!(VD = Lookup.getAsSingle<VarDecl>())) {
- Diag(Id.getLoc(), diag::err_omp_expected_var_arg) << Id.getName();
- Diag(Lookup.getFoundDecl()->getLocation(), diag::note_declared_at);
- return ExprError();
- }
- Lookup.suppressDiagnostics();
- // OpenMP [2.9.2, Syntax, C/C++]
- // Variables must be file-scope, namespace-scope, or static block-scope.
- if (!VD->hasGlobalStorage()) {
- Diag(Id.getLoc(), diag::err_omp_global_var_arg)
- << getOpenMPDirectiveName(OMPD_threadprivate) << !VD->isStaticLocal();
- bool IsDecl =
- VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly;
- Diag(VD->getLocation(),
- IsDecl ? diag::note_previous_decl : diag::note_defined_here)
- << VD;
- return ExprError();
- }
- VarDecl *CanonicalVD = VD->getCanonicalDecl();
- NamedDecl *ND = CanonicalVD;
- // OpenMP [2.9.2, Restrictions, C/C++, p.2]
- // A threadprivate directive for file-scope variables must appear outside
- // any definition or declaration.
- if (CanonicalVD->getDeclContext()->isTranslationUnit() &&
- !getCurLexicalContext()->isTranslationUnit()) {
- Diag(Id.getLoc(), diag::err_omp_var_scope)
- << getOpenMPDirectiveName(OMPD_threadprivate) << VD;
- bool IsDecl =
- VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly;
- Diag(VD->getLocation(),
- IsDecl ? diag::note_previous_decl : diag::note_defined_here)
- << VD;
- return ExprError();
- }
- // OpenMP [2.9.2, Restrictions, C/C++, p.3]
- // A threadprivate directive for static class member variables must appear
- // in the class definition, in the same scope in which the member
- // variables are declared.
- if (CanonicalVD->isStaticDataMember() &&
- !CanonicalVD->getDeclContext()->Equals(getCurLexicalContext())) {
- Diag(Id.getLoc(), diag::err_omp_var_scope)
- << getOpenMPDirectiveName(OMPD_threadprivate) << VD;
- bool IsDecl =
- VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly;
- Diag(VD->getLocation(),
- IsDecl ? diag::note_previous_decl : diag::note_defined_here)
- << VD;
- return ExprError();
- }
- // OpenMP [2.9.2, Restrictions, C/C++, p.4]
- // A threadprivate directive for namespace-scope variables must appear
- // outside any definition or declaration other than the namespace
- // definition itself.
- if (CanonicalVD->getDeclContext()->isNamespace() &&
- (!getCurLexicalContext()->isFileContext() ||
- !getCurLexicalContext()->Encloses(CanonicalVD->getDeclContext()))) {
- Diag(Id.getLoc(), diag::err_omp_var_scope)
- << getOpenMPDirectiveName(OMPD_threadprivate) << VD;
- bool IsDecl =
- VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly;
- Diag(VD->getLocation(),
- IsDecl ? diag::note_previous_decl : diag::note_defined_here)
- << VD;
- return ExprError();
- }
- // OpenMP [2.9.2, Restrictions, C/C++, p.6]
- // A threadprivate directive for static block-scope variables must appear
- // in the scope of the variable and not in a nested scope.
- if (CanonicalVD->isStaticLocal() && CurScope &&
- !isDeclInScope(ND, getCurLexicalContext(), CurScope)) {
- Diag(Id.getLoc(), diag::err_omp_var_scope)
- << getOpenMPDirectiveName(OMPD_threadprivate) << VD;
- bool IsDecl =
- VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly;
- Diag(VD->getLocation(),
- IsDecl ? diag::note_previous_decl : diag::note_defined_here)
- << VD;
- return ExprError();
- }
- // OpenMP [2.9.2, Restrictions, C/C++, p.2-6]
- // A threadprivate directive must lexically precede all references to any
- // of the variables in its list.
- if (VD->isUsed() && !DSAStack->isThreadPrivate(VD)) {
- Diag(Id.getLoc(), diag::err_omp_var_used)
- << getOpenMPDirectiveName(OMPD_threadprivate) << VD;
- return ExprError();
- }
- QualType ExprType = VD->getType().getNonReferenceType();
- return DeclRefExpr::Create(Context, NestedNameSpecifierLoc(),
- SourceLocation(), VD,
- /*RefersToEnclosingVariableOrCapture=*/false,
- Id.getLoc(), ExprType, VK_LValue);
- }
- Sema::DeclGroupPtrTy
- Sema::ActOnOpenMPThreadprivateDirective(SourceLocation Loc,
- ArrayRef<Expr *> VarList) {
- if (OMPThreadPrivateDecl *D = CheckOMPThreadPrivateDecl(Loc, VarList)) {
- CurContext->addDecl(D);
- return DeclGroupPtrTy::make(DeclGroupRef(D));
- }
- return nullptr;
- }
- namespace {
- class LocalVarRefChecker final
- : public ConstStmtVisitor<LocalVarRefChecker, bool> {
- Sema &SemaRef;
- public:
- bool VisitDeclRefExpr(const DeclRefExpr *E) {
- if (const auto *VD = dyn_cast<VarDecl>(E->getDecl())) {
- if (VD->hasLocalStorage()) {
- SemaRef.Diag(E->getBeginLoc(),
- diag::err_omp_local_var_in_threadprivate_init)
- << E->getSourceRange();
- SemaRef.Diag(VD->getLocation(), diag::note_defined_here)
- << VD << VD->getSourceRange();
- return true;
- }
- }
- return false;
- }
- bool VisitStmt(const Stmt *S) {
- for (const Stmt *Child : S->children()) {
- if (Child && Visit(Child))
- return true;
- }
- return false;
- }
- explicit LocalVarRefChecker(Sema &SemaRef) : SemaRef(SemaRef) {}
- };
- } // namespace
- OMPThreadPrivateDecl *
- Sema::CheckOMPThreadPrivateDecl(SourceLocation Loc, ArrayRef<Expr *> VarList) {
- SmallVector<Expr *, 8> Vars;
- for (Expr *RefExpr : VarList) {
- auto *DE = cast<DeclRefExpr>(RefExpr);
- auto *VD = cast<VarDecl>(DE->getDecl());
- SourceLocation ILoc = DE->getExprLoc();
- // Mark variable as used.
- VD->setReferenced();
- VD->markUsed(Context);
- QualType QType = VD->getType();
- if (QType->isDependentType() || QType->isInstantiationDependentType()) {
- // It will be analyzed later.
- Vars.push_back(DE);
- continue;
- }
- // OpenMP [2.9.2, Restrictions, C/C++, p.10]
- // A threadprivate variable must not have an incomplete type.
- if (RequireCompleteType(ILoc, VD->getType(),
- diag::err_omp_threadprivate_incomplete_type)) {
- continue;
- }
- // OpenMP [2.9.2, Restrictions, C/C++, p.10]
- // A threadprivate variable must not have a reference type.
- if (VD->getType()->isReferenceType()) {
- Diag(ILoc, diag::err_omp_ref_type_arg)
- << getOpenMPDirectiveName(OMPD_threadprivate) << VD->getType();
- bool IsDecl =
- VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly;
- Diag(VD->getLocation(),
- IsDecl ? diag::note_previous_decl : diag::note_defined_here)
- << VD;
- continue;
- }
- // Check if this is a TLS variable. If TLS is not being supported, produce
- // the corresponding diagnostic.
- if ((VD->getTLSKind() != VarDecl::TLS_None &&
- !(VD->hasAttr<OMPThreadPrivateDeclAttr>() &&
- getLangOpts().OpenMPUseTLS &&
- getASTContext().getTargetInfo().isTLSSupported())) ||
- (VD->getStorageClass() == SC_Register && VD->hasAttr<AsmLabelAttr>() &&
- !VD->isLocalVarDecl())) {
- Diag(ILoc, diag::err_omp_var_thread_local)
- << VD << ((VD->getTLSKind() != VarDecl::TLS_None) ? 0 : 1);
- bool IsDecl =
- VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly;
- Diag(VD->getLocation(),
- IsDecl ? diag::note_previous_decl : diag::note_defined_here)
- << VD;
- continue;
- }
- // Check if initial value of threadprivate variable reference variable with
- // local storage (it is not supported by runtime).
- if (const Expr *Init = VD->getAnyInitializer()) {
- LocalVarRefChecker Checker(*this);
- if (Checker.Visit(Init))
- continue;
- }
- Vars.push_back(RefExpr);
- DSAStack->addDSA(VD, DE, OMPC_threadprivate);
- VD->addAttr(OMPThreadPrivateDeclAttr::CreateImplicit(
- Context, SourceRange(Loc, Loc)));
- if (ASTMutationListener *ML = Context.getASTMutationListener())
- ML->DeclarationMarkedOpenMPThreadPrivate(VD);
- }
- OMPThreadPrivateDecl *D = nullptr;
- if (!Vars.empty()) {
- D = OMPThreadPrivateDecl::Create(Context, getCurLexicalContext(), Loc,
- Vars);
- D->setAccess(AS_public);
- }
- return D;
- }
- Sema::DeclGroupPtrTy
- Sema::ActOnOpenMPRequiresDirective(SourceLocation Loc,
- ArrayRef<OMPClause *> ClauseList) {
- OMPRequiresDecl *D = nullptr;
- if (!CurContext->isFileContext()) {
- Diag(Loc, diag::err_omp_invalid_scope) << "requires";
- } else {
- D = CheckOMPRequiresDecl(Loc, ClauseList);
- if (D) {
- CurContext->addDecl(D);
- DSAStack->addRequiresDecl(D);
- }
- }
- return DeclGroupPtrTy::make(DeclGroupRef(D));
- }
- OMPRequiresDecl *Sema::CheckOMPRequiresDecl(SourceLocation Loc,
- ArrayRef<OMPClause *> ClauseList) {
- if (!DSAStack->hasDuplicateRequiresClause(ClauseList))
- return OMPRequiresDecl::Create(Context, getCurLexicalContext(), Loc,
- ClauseList);
- return nullptr;
- }
- static void reportOriginalDsa(Sema &SemaRef, const DSAStackTy *Stack,
- const ValueDecl *D,
- const DSAStackTy::DSAVarData &DVar,
- bool IsLoopIterVar = false) {
- if (DVar.RefExpr) {
- SemaRef.Diag(DVar.RefExpr->getExprLoc(), diag::note_omp_explicit_dsa)
- << getOpenMPClauseName(DVar.CKind);
- return;
- }
- enum {
- PDSA_StaticMemberShared,
- PDSA_StaticLocalVarShared,
- PDSA_LoopIterVarPrivate,
- PDSA_LoopIterVarLinear,
- PDSA_LoopIterVarLastprivate,
- PDSA_ConstVarShared,
- PDSA_GlobalVarShared,
- PDSA_TaskVarFirstprivate,
- PDSA_LocalVarPrivate,
- PDSA_Implicit
- } Reason = PDSA_Implicit;
- bool ReportHint = false;
- auto ReportLoc = D->getLocation();
- auto *VD = dyn_cast<VarDecl>(D);
- if (IsLoopIterVar) {
- if (DVar.CKind == OMPC_private)
- Reason = PDSA_LoopIterVarPrivate;
- else if (DVar.CKind == OMPC_lastprivate)
- Reason = PDSA_LoopIterVarLastprivate;
- else
- Reason = PDSA_LoopIterVarLinear;
- } else if (isOpenMPTaskingDirective(DVar.DKind) &&
- DVar.CKind == OMPC_firstprivate) {
- Reason = PDSA_TaskVarFirstprivate;
- ReportLoc = DVar.ImplicitDSALoc;
- } else if (VD && VD->isStaticLocal())
- Reason = PDSA_StaticLocalVarShared;
- else if (VD && VD->isStaticDataMember())
- Reason = PDSA_StaticMemberShared;
- else if (VD && VD->isFileVarDecl())
- Reason = PDSA_GlobalVarShared;
- else if (D->getType().isConstant(SemaRef.getASTContext()))
- Reason = PDSA_ConstVarShared;
- else if (VD && VD->isLocalVarDecl() && DVar.CKind == OMPC_private) {
- ReportHint = true;
- Reason = PDSA_LocalVarPrivate;
- }
- if (Reason != PDSA_Implicit) {
- SemaRef.Diag(ReportLoc, diag::note_omp_predetermined_dsa)
- << Reason << ReportHint
- << getOpenMPDirectiveName(Stack->getCurrentDirective());
- } else if (DVar.ImplicitDSALoc.isValid()) {
- SemaRef.Diag(DVar.ImplicitDSALoc, diag::note_omp_implicit_dsa)
- << getOpenMPClauseName(DVar.CKind);
- }
- }
- namespace {
- class DSAAttrChecker final : public StmtVisitor<DSAAttrChecker, void> {
- DSAStackTy *Stack;
- Sema &SemaRef;
- bool ErrorFound = false;
- CapturedStmt *CS = nullptr;
- llvm::SmallVector<Expr *, 4> ImplicitFirstprivate;
- llvm::SmallVector<Expr *, 4> ImplicitMap;
- Sema::VarsWithInheritedDSAType VarsWithInheritedDSA;
- llvm::SmallDenseSet<const ValueDecl *, 4> ImplicitDeclarations;
- void VisitSubCaptures(OMPExecutableDirective *S) {
- // Check implicitly captured variables.
- if (!S->hasAssociatedStmt() || !S->getAssociatedStmt())
- return;
- for (const CapturedStmt::Capture &Cap :
- S->getInnermostCapturedStmt()->captures()) {
- if (!Cap.capturesVariable())
- continue;
- VarDecl *VD = Cap.getCapturedVar();
- // Do not try to map the variable if it or its sub-component was mapped
- // already.
- if (isOpenMPTargetExecutionDirective(Stack->getCurrentDirective()) &&
- Stack->checkMappableExprComponentListsForDecl(
- VD, /*CurrentRegionOnly=*/true,
- [](OMPClauseMappableExprCommon::MappableExprComponentListRef,
- OpenMPClauseKind) { return true; }))
- continue;
- DeclRefExpr *DRE = buildDeclRefExpr(
- SemaRef, VD, VD->getType().getNonLValueExprType(SemaRef.Context),
- Cap.getLocation(), /*RefersToCapture=*/true);
- Visit(DRE);
- }
- }
- public:
- void VisitDeclRefExpr(DeclRefExpr *E) {
- if (E->isTypeDependent() || E->isValueDependent() ||
- E->containsUnexpandedParameterPack() || E->isInstantiationDependent())
- return;
- if (auto *VD = dyn_cast<VarDecl>(E->getDecl())) {
- VD = VD->getCanonicalDecl();
- // Skip internally declared variables.
- if (VD->hasLocalStorage() && !CS->capturesVariable(VD))
- return;
- DSAStackTy::DSAVarData DVar = Stack->getTopDSA(VD, /*FromParent=*/false);
- // Check if the variable has explicit DSA set and stop analysis if it so.
- if (DVar.RefExpr || !ImplicitDeclarations.insert(VD).second)
- return;
- // Skip internally declared static variables.
- llvm::Optional<OMPDeclareTargetDeclAttr::MapTypeTy> Res =
- OMPDeclareTargetDeclAttr::isDeclareTargetDeclaration(VD);
- if (VD->hasGlobalStorage() && !CS->capturesVariable(VD) &&
- (!Res || *Res != OMPDeclareTargetDeclAttr::MT_Link))
- return;
- SourceLocation ELoc = E->getExprLoc();
- OpenMPDirectiveKind DKind = Stack->getCurrentDirective();
- // The default(none) clause requires that each variable that is referenced
- // in the construct, and does not have a predetermined data-sharing
- // attribute, must have its data-sharing attribute explicitly determined
- // by being listed in a data-sharing attribute clause.
- if (DVar.CKind == OMPC_unknown && Stack->getDefaultDSA() == DSA_none &&
- isParallelOrTaskRegion(DKind) &&
- VarsWithInheritedDSA.count(VD) == 0) {
- VarsWithInheritedDSA[VD] = E;
- return;
- }
- if (isOpenMPTargetExecutionDirective(DKind) &&
- !Stack->isLoopControlVariable(VD).first) {
- if (!Stack->checkMappableExprComponentListsForDecl(
- VD, /*CurrentRegionOnly=*/true,
- [](OMPClauseMappableExprCommon::MappableExprComponentListRef
- StackComponents,
- OpenMPClauseKind) {
- // Variable is used if it has been marked as an array, array
- // section or the variable iself.
- return StackComponents.size() == 1 ||
- std::all_of(
- std::next(StackComponents.rbegin()),
- StackComponents.rend(),
- [](const OMPClauseMappableExprCommon::
- MappableComponent &MC) {
- return MC.getAssociatedDeclaration() ==
- nullptr &&
- (isa<OMPArraySectionExpr>(
- MC.getAssociatedExpression()) ||
- isa<ArraySubscriptExpr>(
- MC.getAssociatedExpression()));
- });
- })) {
- bool IsFirstprivate = false;
- // By default lambdas are captured as firstprivates.
- if (const auto *RD =
- VD->getType().getNonReferenceType()->getAsCXXRecordDecl())
- IsFirstprivate = RD->isLambda();
- IsFirstprivate =
- IsFirstprivate ||
- (VD->getType().getNonReferenceType()->isScalarType() &&
- Stack->getDefaultDMA() != DMA_tofrom_scalar && !Res);
- if (IsFirstprivate)
- ImplicitFirstprivate.emplace_back(E);
- else
- ImplicitMap.emplace_back(E);
- return;
- }
- }
- // OpenMP [2.9.3.6, Restrictions, p.2]
- // A list item that appears in a reduction clause of the innermost
- // enclosing worksharing or parallel construct may not be accessed in an
- // explicit task.
- DVar = Stack->hasInnermostDSA(
- VD, [](OpenMPClauseKind C) { return C == OMPC_reduction; },
- [](OpenMPDirectiveKind K) {
- return isOpenMPParallelDirective(K) ||
- isOpenMPWorksharingDirective(K) || isOpenMPTeamsDirective(K);
- },
- /*FromParent=*/true);
- if (isOpenMPTaskingDirective(DKind) && DVar.CKind == OMPC_reduction) {
- ErrorFound = true;
- SemaRef.Diag(ELoc, diag::err_omp_reduction_in_task);
- reportOriginalDsa(SemaRef, Stack, VD, DVar);
- return;
- }
- // Define implicit data-sharing attributes for task.
- DVar = Stack->getImplicitDSA(VD, /*FromParent=*/false);
- if (isOpenMPTaskingDirective(DKind) && DVar.CKind != OMPC_shared &&
- !Stack->isLoopControlVariable(VD).first)
- ImplicitFirstprivate.push_back(E);
- }
- }
- void VisitMemberExpr(MemberExpr *E) {
- if (E->isTypeDependent() || E->isValueDependent() ||
- E->containsUnexpandedParameterPack() || E->isInstantiationDependent())
- return;
- auto *FD = dyn_cast<FieldDecl>(E->getMemberDecl());
- OpenMPDirectiveKind DKind = Stack->getCurrentDirective();
- if (isa<CXXThisExpr>(E->getBase()->IgnoreParens())) {
- if (!FD)
- return;
- DSAStackTy::DSAVarData DVar = Stack->getTopDSA(FD, /*FromParent=*/false);
- // Check if the variable has explicit DSA set and stop analysis if it
- // so.
- if (DVar.RefExpr || !ImplicitDeclarations.insert(FD).second)
- return;
- if (isOpenMPTargetExecutionDirective(DKind) &&
- !Stack->isLoopControlVariable(FD).first &&
- !Stack->checkMappableExprComponentListsForDecl(
- FD, /*CurrentRegionOnly=*/true,
- [](OMPClauseMappableExprCommon::MappableExprComponentListRef
- StackComponents,
- OpenMPClauseKind) {
- return isa<CXXThisExpr>(
- cast<MemberExpr>(
- StackComponents.back().getAssociatedExpression())
- ->getBase()
- ->IgnoreParens());
- })) {
- // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, C/C++, p.3]
- // A bit-field cannot appear in a map clause.
- //
- if (FD->isBitField())
- return;
- ImplicitMap.emplace_back(E);
- return;
- }
- SourceLocation ELoc = E->getExprLoc();
- // OpenMP [2.9.3.6, Restrictions, p.2]
- // A list item that appears in a reduction clause of the innermost
- // enclosing worksharing or parallel construct may not be accessed in
- // an explicit task.
- DVar = Stack->hasInnermostDSA(
- FD, [](OpenMPClauseKind C) { return C == OMPC_reduction; },
- [](OpenMPDirectiveKind K) {
- return isOpenMPParallelDirective(K) ||
- isOpenMPWorksharingDirective(K) || isOpenMPTeamsDirective(K);
- },
- /*FromParent=*/true);
- if (isOpenMPTaskingDirective(DKind) && DVar.CKind == OMPC_reduction) {
- ErrorFound = true;
- SemaRef.Diag(ELoc, diag::err_omp_reduction_in_task);
- reportOriginalDsa(SemaRef, Stack, FD, DVar);
- return;
- }
- // Define implicit data-sharing attributes for task.
- DVar = Stack->getImplicitDSA(FD, /*FromParent=*/false);
- if (isOpenMPTaskingDirective(DKind) && DVar.CKind != OMPC_shared &&
- !Stack->isLoopControlVariable(FD).first) {
- // Check if there is a captured expression for the current field in the
- // region. Do not mark it as firstprivate unless there is no captured
- // expression.
- // TODO: try to make it firstprivate.
- if (DVar.CKind != OMPC_unknown)
- ImplicitFirstprivate.push_back(E);
- }
- return;
- }
- if (isOpenMPTargetExecutionDirective(DKind)) {
- OMPClauseMappableExprCommon::MappableExprComponentList CurComponents;
- if (!checkMapClauseExpressionBase(SemaRef, E, CurComponents, OMPC_map,
- /*NoDiagnose=*/true))
- return;
- const auto *VD = cast<ValueDecl>(
- CurComponents.back().getAssociatedDeclaration()->getCanonicalDecl());
- if (!Stack->checkMappableExprComponentListsForDecl(
- VD, /*CurrentRegionOnly=*/true,
- [&CurComponents](
- OMPClauseMappableExprCommon::MappableExprComponentListRef
- StackComponents,
- OpenMPClauseKind) {
- auto CCI = CurComponents.rbegin();
- auto CCE = CurComponents.rend();
- for (const auto &SC : llvm::reverse(StackComponents)) {
- // Do both expressions have the same kind?
- if (CCI->getAssociatedExpression()->getStmtClass() !=
- SC.getAssociatedExpression()->getStmtClass())
- if (!(isa<OMPArraySectionExpr>(
- SC.getAssociatedExpression()) &&
- isa<ArraySubscriptExpr>(
- CCI->getAssociatedExpression())))
- return false;
- const Decl *CCD = CCI->getAssociatedDeclaration();
- const Decl *SCD = SC.getAssociatedDeclaration();
- CCD = CCD ? CCD->getCanonicalDecl() : nullptr;
- SCD = SCD ? SCD->getCanonicalDecl() : nullptr;
- if (SCD != CCD)
- return false;
- std::advance(CCI, 1);
- if (CCI == CCE)
- break;
- }
- return true;
- })) {
- Visit(E->getBase());
- }
- } else {
- Visit(E->getBase());
- }
- }
- void VisitOMPExecutableDirective(OMPExecutableDirective *S) {
- for (OMPClause *C : S->clauses()) {
- // Skip analysis of arguments of implicitly defined firstprivate clause
- // for task|target directives.
- // Skip analysis of arguments of implicitly defined map clause for target
- // directives.
- if (C && !((isa<OMPFirstprivateClause>(C) || isa<OMPMapClause>(C)) &&
- C->isImplicit())) {
- for (Stmt *CC : C->children()) {
- if (CC)
- Visit(CC);
- }
- }
- }
- // Check implicitly captured variables.
- VisitSubCaptures(S);
- }
- void VisitStmt(Stmt *S) {
- for (Stmt *C : S->children()) {
- if (C) {
- if (auto *OED = dyn_cast<OMPExecutableDirective>(C)) {
- // Check implicitly captured variables in the task-based directives to
- // check if they must be firstprivatized.
- VisitSubCaptures(OED);
- } else {
- Visit(C);
- }
- }
- }
- }
- bool isErrorFound() const { return ErrorFound; }
- ArrayRef<Expr *> getImplicitFirstprivate() const {
- return ImplicitFirstprivate;
- }
- ArrayRef<Expr *> getImplicitMap() const { return ImplicitMap; }
- const Sema::VarsWithInheritedDSAType &getVarsWithInheritedDSA() const {
- return VarsWithInheritedDSA;
- }
- DSAAttrChecker(DSAStackTy *S, Sema &SemaRef, CapturedStmt *CS)
- : Stack(S), SemaRef(SemaRef), ErrorFound(false), CS(CS) {}
- };
- } // namespace
- void Sema::ActOnOpenMPRegionStart(OpenMPDirectiveKind DKind, Scope *CurScope) {
- switch (DKind) {
- case OMPD_parallel:
- case OMPD_parallel_for:
- case OMPD_parallel_for_simd:
- case OMPD_parallel_sections:
- case OMPD_teams:
- case OMPD_teams_distribute:
- case OMPD_teams_distribute_simd: {
- QualType KmpInt32Ty = Context.getIntTypeForBitwidth(32, 1).withConst();
- QualType KmpInt32PtrTy =
- Context.getPointerType(KmpInt32Ty).withConst().withRestrict();
- Sema::CapturedParamNameType Params[] = {
- std::make_pair(".global_tid.", KmpInt32PtrTy),
- std::make_pair(".bound_tid.", KmpInt32PtrTy),
- std::make_pair(StringRef(), QualType()) // __context with shared vars
- };
- ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
- Params);
- break;
- }
- case OMPD_target_teams:
- case OMPD_target_parallel:
- case OMPD_target_parallel_for:
- case OMPD_target_parallel_for_simd:
- case OMPD_target_teams_distribute:
- case OMPD_target_teams_distribute_simd: {
- QualType KmpInt32Ty = Context.getIntTypeForBitwidth(32, 1).withConst();
- QualType VoidPtrTy = Context.VoidPtrTy.withConst().withRestrict();
- QualType KmpInt32PtrTy =
- Context.getPointerType(KmpInt32Ty).withConst().withRestrict();
- QualType Args[] = {VoidPtrTy};
- FunctionProtoType::ExtProtoInfo EPI;
- EPI.Variadic = true;
- QualType CopyFnType = Context.getFunctionType(Context.VoidTy, Args, EPI);
- Sema::CapturedParamNameType Params[] = {
- std::make_pair(".global_tid.", KmpInt32Ty),
- std::make_pair(".part_id.", KmpInt32PtrTy),
- std::make_pair(".privates.", VoidPtrTy),
- std::make_pair(
- ".copy_fn.",
- Context.getPointerType(CopyFnType).withConst().withRestrict()),
- std::make_pair(".task_t.", Context.VoidPtrTy.withConst()),
- std::make_pair(StringRef(), QualType()) // __context with shared vars
- };
- ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
- Params);
- // Mark this captured region as inlined, because we don't use outlined
- // function directly.
- getCurCapturedRegion()->TheCapturedDecl->addAttr(
- AlwaysInlineAttr::CreateImplicit(
- Context, AlwaysInlineAttr::Keyword_forceinline));
- Sema::CapturedParamNameType ParamsTarget[] = {
- std::make_pair(StringRef(), QualType()) // __context with shared vars
- };
- // Start a captured region for 'target' with no implicit parameters.
- ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
- ParamsTarget);
- Sema::CapturedParamNameType ParamsTeamsOrParallel[] = {
- std::make_pair(".global_tid.", KmpInt32PtrTy),
- std::make_pair(".bound_tid.", KmpInt32PtrTy),
- std::make_pair(StringRef(), QualType()) // __context with shared vars
- };
- // Start a captured region for 'teams' or 'parallel'. Both regions have
- // the same implicit parameters.
- ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
- ParamsTeamsOrParallel);
- break;
- }
- case OMPD_target:
- case OMPD_target_simd: {
- QualType KmpInt32Ty = Context.getIntTypeForBitwidth(32, 1).withConst();
- QualType VoidPtrTy = Context.VoidPtrTy.withConst().withRestrict();
- QualType KmpInt32PtrTy =
- Context.getPointerType(KmpInt32Ty).withConst().withRestrict();
- QualType Args[] = {VoidPtrTy};
- FunctionProtoType::ExtProtoInfo EPI;
- EPI.Variadic = true;
- QualType CopyFnType = Context.getFunctionType(Context.VoidTy, Args, EPI);
- Sema::CapturedParamNameType Params[] = {
- std::make_pair(".global_tid.", KmpInt32Ty),
- std::make_pair(".part_id.", KmpInt32PtrTy),
- std::make_pair(".privates.", VoidPtrTy),
- std::make_pair(
- ".copy_fn.",
- Context.getPointerType(CopyFnType).withConst().withRestrict()),
- std::make_pair(".task_t.", Context.VoidPtrTy.withConst()),
- std::make_pair(StringRef(), QualType()) // __context with shared vars
- };
- ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
- Params);
- // Mark this captured region as inlined, because we don't use outlined
- // function directly.
- getCurCapturedRegion()->TheCapturedDecl->addAttr(
- AlwaysInlineAttr::CreateImplicit(
- Context, AlwaysInlineAttr::Keyword_forceinline));
- ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
- std::make_pair(StringRef(), QualType()));
- break;
- }
- case OMPD_simd:
- case OMPD_for:
- case OMPD_for_simd:
- case OMPD_sections:
- case OMPD_section:
- case OMPD_single:
- case OMPD_master:
- case OMPD_critical:
- case OMPD_taskgroup:
- case OMPD_distribute:
- case OMPD_distribute_simd:
- case OMPD_ordered:
- case OMPD_atomic:
- case OMPD_target_data: {
- Sema::CapturedParamNameType Params[] = {
- std::make_pair(StringRef(), QualType()) // __context with shared vars
- };
- ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
- Params);
- break;
- }
- case OMPD_task: {
- QualType KmpInt32Ty = Context.getIntTypeForBitwidth(32, 1).withConst();
- QualType VoidPtrTy = Context.VoidPtrTy.withConst().withRestrict();
- QualType KmpInt32PtrTy =
- Context.getPointerType(KmpInt32Ty).withConst().withRestrict();
- QualType Args[] = {VoidPtrTy};
- FunctionProtoType::ExtProtoInfo EPI;
- EPI.Variadic = true;
- QualType CopyFnType = Context.getFunctionType(Context.VoidTy, Args, EPI);
- Sema::CapturedParamNameType Params[] = {
- std::make_pair(".global_tid.", KmpInt32Ty),
- std::make_pair(".part_id.", KmpInt32PtrTy),
- std::make_pair(".privates.", VoidPtrTy),
- std::make_pair(
- ".copy_fn.",
- Context.getPointerType(CopyFnType).withConst().withRestrict()),
- std::make_pair(".task_t.", Context.VoidPtrTy.withConst()),
- std::make_pair(StringRef(), QualType()) // __context with shared vars
- };
- ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
- Params);
- // Mark this captured region as inlined, because we don't use outlined
- // function directly.
- getCurCapturedRegion()->TheCapturedDecl->addAttr(
- AlwaysInlineAttr::CreateImplicit(
- Context, AlwaysInlineAttr::Keyword_forceinline));
- break;
- }
- case OMPD_taskloop:
- case OMPD_taskloop_simd: {
- QualType KmpInt32Ty =
- Context.getIntTypeForBitwidth(/*DestWidth=*/32, /*Signed=*/1)
- .withConst();
- QualType KmpUInt64Ty =
- Context.getIntTypeForBitwidth(/*DestWidth=*/64, /*Signed=*/0)
- .withConst();
- QualType KmpInt64Ty =
- Context.getIntTypeForBitwidth(/*DestWidth=*/64, /*Signed=*/1)
- .withConst();
- QualType VoidPtrTy = Context.VoidPtrTy.withConst().withRestrict();
- QualType KmpInt32PtrTy =
- Context.getPointerType(KmpInt32Ty).withConst().withRestrict();
- QualType Args[] = {VoidPtrTy};
- FunctionProtoType::ExtProtoInfo EPI;
- EPI.Variadic = true;
- QualType CopyFnType = Context.getFunctionType(Context.VoidTy, Args, EPI);
- Sema::CapturedParamNameType Params[] = {
- std::make_pair(".global_tid.", KmpInt32Ty),
- std::make_pair(".part_id.", KmpInt32PtrTy),
- std::make_pair(".privates.", VoidPtrTy),
- std::make_pair(
- ".copy_fn.",
- Context.getPointerType(CopyFnType).withConst().withRestrict()),
- std::make_pair(".task_t.", Context.VoidPtrTy.withConst()),
- std::make_pair(".lb.", KmpUInt64Ty),
- std::make_pair(".ub.", KmpUInt64Ty),
- std::make_pair(".st.", KmpInt64Ty),
- std::make_pair(".liter.", KmpInt32Ty),
- std::make_pair(".reductions.", VoidPtrTy),
- std::make_pair(StringRef(), QualType()) // __context with shared vars
- };
- ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
- Params);
- // Mark this captured region as inlined, because we don't use outlined
- // function directly.
- getCurCapturedRegion()->TheCapturedDecl->addAttr(
- AlwaysInlineAttr::CreateImplicit(
- Context, AlwaysInlineAttr::Keyword_forceinline));
- break;
- }
- case OMPD_distribute_parallel_for_simd:
- case OMPD_distribute_parallel_for: {
- QualType KmpInt32Ty = Context.getIntTypeForBitwidth(32, 1).withConst();
- QualType KmpInt32PtrTy =
- Context.getPointerType(KmpInt32Ty).withConst().withRestrict();
- Sema::CapturedParamNameType Params[] = {
- std::make_pair(".global_tid.", KmpInt32PtrTy),
- std::make_pair(".bound_tid.", KmpInt32PtrTy),
- std::make_pair(".previous.lb.", Context.getSizeType().withConst()),
- std::make_pair(".previous.ub.", Context.getSizeType().withConst()),
- std::make_pair(StringRef(), QualType()) // __context with shared vars
- };
- ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
- Params);
- break;
- }
- case OMPD_target_teams_distribute_parallel_for:
- case OMPD_target_teams_distribute_parallel_for_simd: {
- QualType KmpInt32Ty = Context.getIntTypeForBitwidth(32, 1).withConst();
- QualType KmpInt32PtrTy =
- Context.getPointerType(KmpInt32Ty).withConst().withRestrict();
- QualType VoidPtrTy = Context.VoidPtrTy.withConst().withRestrict();
- QualType Args[] = {VoidPtrTy};
- FunctionProtoType::ExtProtoInfo EPI;
- EPI.Variadic = true;
- QualType CopyFnType = Context.getFunctionType(Context.VoidTy, Args, EPI);
- Sema::CapturedParamNameType Params[] = {
- std::make_pair(".global_tid.", KmpInt32Ty),
- std::make_pair(".part_id.", KmpInt32PtrTy),
- std::make_pair(".privates.", VoidPtrTy),
- std::make_pair(
- ".copy_fn.",
- Context.getPointerType(CopyFnType).withConst().withRestrict()),
- std::make_pair(".task_t.", Context.VoidPtrTy.withConst()),
- std::make_pair(StringRef(), QualType()) // __context with shared vars
- };
- ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
- Params);
- // Mark this captured region as inlined, because we don't use outlined
- // function directly.
- getCurCapturedRegion()->TheCapturedDecl->addAttr(
- AlwaysInlineAttr::CreateImplicit(
- Context, AlwaysInlineAttr::Keyword_forceinline));
- Sema::CapturedParamNameType ParamsTarget[] = {
- std::make_pair(StringRef(), QualType()) // __context with shared vars
- };
- // Start a captured region for 'target' with no implicit parameters.
- ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
- ParamsTarget);
- Sema::CapturedParamNameType ParamsTeams[] = {
- std::make_pair(".global_tid.", KmpInt32PtrTy),
- std::make_pair(".bound_tid.", KmpInt32PtrTy),
- std::make_pair(StringRef(), QualType()) // __context with shared vars
- };
- // Start a captured region for 'target' with no implicit parameters.
- ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
- ParamsTeams);
- Sema::CapturedParamNameType ParamsParallel[] = {
- std::make_pair(".global_tid.", KmpInt32PtrTy),
- std::make_pair(".bound_tid.", KmpInt32PtrTy),
- std::make_pair(".previous.lb.", Context.getSizeType().withConst()),
- std::make_pair(".previous.ub.", Context.getSizeType().withConst()),
- std::make_pair(StringRef(), QualType()) // __context with shared vars
- };
- // Start a captured region for 'teams' or 'parallel'. Both regions have
- // the same implicit parameters.
- ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
- ParamsParallel);
- break;
- }
- case OMPD_teams_distribute_parallel_for:
- case OMPD_teams_distribute_parallel_for_simd: {
- QualType KmpInt32Ty = Context.getIntTypeForBitwidth(32, 1).withConst();
- QualType KmpInt32PtrTy =
- Context.getPointerType(KmpInt32Ty).withConst().withRestrict();
- Sema::CapturedParamNameType ParamsTeams[] = {
- std::make_pair(".global_tid.", KmpInt32PtrTy),
- std::make_pair(".bound_tid.", KmpInt32PtrTy),
- std::make_pair(StringRef(), QualType()) // __context with shared vars
- };
- // Start a captured region for 'target' with no implicit parameters.
- ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
- ParamsTeams);
- Sema::CapturedParamNameType ParamsParallel[] = {
- std::make_pair(".global_tid.", KmpInt32PtrTy),
- std::make_pair(".bound_tid.", KmpInt32PtrTy),
- std::make_pair(".previous.lb.", Context.getSizeType().withConst()),
- std::make_pair(".previous.ub.", Context.getSizeType().withConst()),
- std::make_pair(StringRef(), QualType()) // __context with shared vars
- };
- // Start a captured region for 'teams' or 'parallel'. Both regions have
- // the same implicit parameters.
- ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
- ParamsParallel);
- break;
- }
- case OMPD_target_update:
- case OMPD_target_enter_data:
- case OMPD_target_exit_data: {
- QualType KmpInt32Ty = Context.getIntTypeForBitwidth(32, 1).withConst();
- QualType VoidPtrTy = Context.VoidPtrTy.withConst().withRestrict();
- QualType KmpInt32PtrTy =
- Context.getPointerType(KmpInt32Ty).withConst().withRestrict();
- QualType Args[] = {VoidPtrTy};
- FunctionProtoType::ExtProtoInfo EPI;
- EPI.Variadic = true;
- QualType CopyFnType = Context.getFunctionType(Context.VoidTy, Args, EPI);
- Sema::CapturedParamNameType Params[] = {
- std::make_pair(".global_tid.", KmpInt32Ty),
- std::make_pair(".part_id.", KmpInt32PtrTy),
- std::make_pair(".privates.", VoidPtrTy),
- std::make_pair(
- ".copy_fn.",
- Context.getPointerType(CopyFnType).withConst().withRestrict()),
- std::make_pair(".task_t.", Context.VoidPtrTy.withConst()),
- std::make_pair(StringRef(), QualType()) // __context with shared vars
- };
- ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
- Params);
- // Mark this captured region as inlined, because we don't use outlined
- // function directly.
- getCurCapturedRegion()->TheCapturedDecl->addAttr(
- AlwaysInlineAttr::CreateImplicit(
- Context, AlwaysInlineAttr::Keyword_forceinline));
- break;
- }
- case OMPD_threadprivate:
- case OMPD_taskyield:
- case OMPD_barrier:
- case OMPD_taskwait:
- case OMPD_cancellation_point:
- case OMPD_cancel:
- case OMPD_flush:
- case OMPD_declare_reduction:
- case OMPD_declare_simd:
- case OMPD_declare_target:
- case OMPD_end_declare_target:
- case OMPD_requires:
- llvm_unreachable("OpenMP Directive is not allowed");
- case OMPD_unknown:
- llvm_unreachable("Unknown OpenMP directive");
- }
- }
- int Sema::getOpenMPCaptureLevels(OpenMPDirectiveKind DKind) {
- SmallVector<OpenMPDirectiveKind, 4> CaptureRegions;
- getOpenMPCaptureRegions(CaptureRegions, DKind);
- return CaptureRegions.size();
- }
- static OMPCapturedExprDecl *buildCaptureDecl(Sema &S, IdentifierInfo *Id,
- Expr *CaptureExpr, bool WithInit,
- bool AsExpression) {
- assert(CaptureExpr);
- ASTContext &C = S.getASTContext();
- Expr *Init = AsExpression ? CaptureExpr : CaptureExpr->IgnoreImpCasts();
- QualType Ty = Init->getType();
- if (CaptureExpr->getObjectKind() == OK_Ordinary && CaptureExpr->isGLValue()) {
- if (S.getLangOpts().CPlusPlus) {
- Ty = C.getLValueReferenceType(Ty);
- } else {
- Ty = C.getPointerType(Ty);
- ExprResult Res =
- S.CreateBuiltinUnaryOp(CaptureExpr->getExprLoc(), UO_AddrOf, Init);
- if (!Res.isUsable())
- return nullptr;
- Init = Res.get();
- }
- WithInit = true;
- }
- auto *CED = OMPCapturedExprDecl::Create(C, S.CurContext, Id, Ty,
- CaptureExpr->getBeginLoc());
- if (!WithInit)
- CED->addAttr(OMPCaptureNoInitAttr::CreateImplicit(C));
- S.CurContext->addHiddenDecl(CED);
- S.AddInitializerToDecl(CED, Init, /*DirectInit=*/false);
- return CED;
- }
- static DeclRefExpr *buildCapture(Sema &S, ValueDecl *D, Expr *CaptureExpr,
- bool WithInit) {
- OMPCapturedExprDecl *CD;
- if (VarDecl *VD = S.isOpenMPCapturedDecl(D))
- CD = cast<OMPCapturedExprDecl>(VD);
- else
- CD = buildCaptureDecl(S, D->getIdentifier(), CaptureExpr, WithInit,
- /*AsExpression=*/false);
- return buildDeclRefExpr(S, CD, CD->getType().getNonReferenceType(),
- CaptureExpr->getExprLoc());
- }
- static ExprResult buildCapture(Sema &S, Expr *CaptureExpr, DeclRefExpr *&Ref) {
- CaptureExpr = S.DefaultLvalueConversion(CaptureExpr).get();
- if (!Ref) {
- OMPCapturedExprDecl *CD = buildCaptureDecl(
- S, &S.getASTContext().Idents.get(".capture_expr."), CaptureExpr,
- /*WithInit=*/true, /*AsExpression=*/true);
- Ref = buildDeclRefExpr(S, CD, CD->getType().getNonReferenceType(),
- CaptureExpr->getExprLoc());
- }
- ExprResult Res = Ref;
- if (!S.getLangOpts().CPlusPlus &&
- CaptureExpr->getObjectKind() == OK_Ordinary && CaptureExpr->isGLValue() &&
- Ref->getType()->isPointerType()) {
- Res = S.CreateBuiltinUnaryOp(CaptureExpr->getExprLoc(), UO_Deref, Ref);
- if (!Res.isUsable())
- return ExprError();
- }
- return S.DefaultLvalueConversion(Res.get());
- }
- namespace {
- // OpenMP directives parsed in this section are represented as a
- // CapturedStatement with an associated statement. If a syntax error
- // is detected during the parsing of the associated statement, the
- // compiler must abort processing and close the CapturedStatement.
- //
- // Combined directives such as 'target parallel' have more than one
- // nested CapturedStatements. This RAII ensures that we unwind out
- // of all the nested CapturedStatements when an error is found.
- class CaptureRegionUnwinderRAII {
- private:
- Sema &S;
- bool &ErrorFound;
- OpenMPDirectiveKind DKind = OMPD_unknown;
- public:
- CaptureRegionUnwinderRAII(Sema &S, bool &ErrorFound,
- OpenMPDirectiveKind DKind)
- : S(S), ErrorFound(ErrorFound), DKind(DKind) {}
- ~CaptureRegionUnwinderRAII() {
- if (ErrorFound) {
- int ThisCaptureLevel = S.getOpenMPCaptureLevels(DKind);
- while (--ThisCaptureLevel >= 0)
- S.ActOnCapturedRegionError();
- }
- }
- };
- } // namespace
- StmtResult Sema::ActOnOpenMPRegionEnd(StmtResult S,
- ArrayRef<OMPClause *> Clauses) {
- bool ErrorFound = false;
- CaptureRegionUnwinderRAII CaptureRegionUnwinder(
- *this, ErrorFound, DSAStack->getCurrentDirective());
- if (!S.isUsable()) {
- ErrorFound = true;
- return StmtError();
- }
- SmallVector<OpenMPDirectiveKind, 4> CaptureRegions;
- getOpenMPCaptureRegions(CaptureRegions, DSAStack->getCurrentDirective());
- OMPOrderedClause *OC = nullptr;
- OMPScheduleClause *SC = nullptr;
- SmallVector<const OMPLinearClause *, 4> LCs;
- SmallVector<const OMPClauseWithPreInit *, 4> PICs;
- // This is required for proper codegen.
- for (OMPClause *Clause : Clauses) {
- if (isOpenMPTaskingDirective(DSAStack->getCurrentDirective()) &&
- Clause->getClauseKind() == OMPC_in_reduction) {
- // Capture taskgroup task_reduction descriptors inside the tasking regions
- // with the corresponding in_reduction items.
- auto *IRC = cast<OMPInReductionClause>(Clause);
- for (Expr *E : IRC->taskgroup_descriptors())
- if (E)
- MarkDeclarationsReferencedInExpr(E);
- }
- if (isOpenMPPrivate(Clause->getClauseKind()) ||
- Clause->getClauseKind() == OMPC_copyprivate ||
- (getLangOpts().OpenMPUseTLS &&
- getASTContext().getTargetInfo().isTLSSupported() &&
- Clause->getClauseKind() == OMPC_copyin)) {
- DSAStack->setForceVarCapturing(Clause->getClauseKind() == OMPC_copyin);
- // Mark all variables in private list clauses as used in inner region.
- for (Stmt *VarRef : Clause->children()) {
- if (auto *E = cast_or_null<Expr>(VarRef)) {
- MarkDeclarationsReferencedInExpr(E);
- }
- }
- DSAStack->setForceVarCapturing(/*V=*/false);
- } else if (CaptureRegions.size() > 1 ||
- CaptureRegions.back() != OMPD_unknown) {
- if (auto *C = OMPClauseWithPreInit::get(Clause))
- PICs.push_back(C);
- if (auto *C = OMPClauseWithPostUpdate::get(Clause)) {
- if (Expr *E = C->getPostUpdateExpr())
- MarkDeclarationsReferencedInExpr(E);
- }
- }
- if (Clause->getClauseKind() == OMPC_schedule)
- SC = cast<OMPScheduleClause>(Clause);
- else if (Clause->getClauseKind() == OMPC_ordered)
- OC = cast<OMPOrderedClause>(Clause);
- else if (Clause->getClauseKind() == OMPC_linear)
- LCs.push_back(cast<OMPLinearClause>(Clause));
- }
- // OpenMP, 2.7.1 Loop Construct, Restrictions
- // The nonmonotonic modifier cannot be specified if an ordered clause is
- // specified.
- if (SC &&
- (SC->getFirstScheduleModifier() == OMPC_SCHEDULE_MODIFIER_nonmonotonic ||
- SC->getSecondScheduleModifier() ==
- OMPC_SCHEDULE_MODIFIER_nonmonotonic) &&
- OC) {
- Diag(SC->getFirstScheduleModifier() == OMPC_SCHEDULE_MODIFIER_nonmonotonic
- ? SC->getFirstScheduleModifierLoc()
- : SC->getSecondScheduleModifierLoc(),
- diag::err_omp_schedule_nonmonotonic_ordered)
- << SourceRange(OC->getBeginLoc(), OC->getEndLoc());
- ErrorFound = true;
- }
- if (!LCs.empty() && OC && OC->getNumForLoops()) {
- for (const OMPLinearClause *C : LCs) {
- Diag(C->getBeginLoc(), diag::err_omp_linear_ordered)
- << SourceRange(OC->getBeginLoc(), OC->getEndLoc());
- }
- ErrorFound = true;
- }
- if (isOpenMPWorksharingDirective(DSAStack->getCurrentDirective()) &&
- isOpenMPSimdDirective(DSAStack->getCurrentDirective()) && OC &&
- OC->getNumForLoops()) {
- Diag(OC->getBeginLoc(), diag::err_omp_ordered_simd)
- << getOpenMPDirectiveName(DSAStack->getCurrentDirective());
- ErrorFound = true;
- }
- if (ErrorFound) {
- return StmtError();
- }
- StmtResult SR = S;
- for (OpenMPDirectiveKind ThisCaptureRegion : llvm::reverse(CaptureRegions)) {
- // Mark all variables in private list clauses as used in inner region.
- // Required for proper codegen of combined directives.
- // TODO: add processing for other clauses.
- if (ThisCaptureRegion != OMPD_unknown) {
- for (const clang::OMPClauseWithPreInit *C : PICs) {
- OpenMPDirectiveKind CaptureRegion = C->getCaptureRegion();
- // Find the particular capture region for the clause if the
- // directive is a combined one with multiple capture regions.
- // If the directive is not a combined one, the capture region
- // associated with the clause is OMPD_unknown and is generated
- // only once.
- if (CaptureRegion == ThisCaptureRegion ||
- CaptureRegion == OMPD_unknown) {
- if (auto *DS = cast_or_null<DeclStmt>(C->getPreInitStmt())) {
- for (Decl *D : DS->decls())
- MarkVariableReferenced(D->getLocation(), cast<VarDecl>(D));
- }
- }
- }
- }
- SR = ActOnCapturedRegionEnd(SR.get());
- }
- return SR;
- }
- static bool checkCancelRegion(Sema &SemaRef, OpenMPDirectiveKind CurrentRegion,
- OpenMPDirectiveKind CancelRegion,
- SourceLocation StartLoc) {
- // CancelRegion is only needed for cancel and cancellation_point.
- if (CurrentRegion != OMPD_cancel && CurrentRegion != OMPD_cancellation_point)
- return false;
- if (CancelRegion == OMPD_parallel || CancelRegion == OMPD_for ||
- CancelRegion == OMPD_sections || CancelRegion == OMPD_taskgroup)
- return false;
- SemaRef.Diag(StartLoc, diag::err_omp_wrong_cancel_region)
- << getOpenMPDirectiveName(CancelRegion);
- return true;
- }
- static bool checkNestingOfRegions(Sema &SemaRef, const DSAStackTy *Stack,
- OpenMPDirectiveKind CurrentRegion,
- const DeclarationNameInfo &CurrentName,
- OpenMPDirectiveKind CancelRegion,
- SourceLocation StartLoc) {
- if (Stack->getCurScope()) {
- OpenMPDirectiveKind ParentRegion = Stack->getParentDirective();
- OpenMPDirectiveKind OffendingRegion = ParentRegion;
- bool NestingProhibited = false;
- bool CloseNesting = true;
- bool OrphanSeen = false;
- enum {
- NoRecommend,
- ShouldBeInParallelRegion,
- ShouldBeInOrderedRegion,
- ShouldBeInTargetRegion,
- ShouldBeInTeamsRegion
- } Recommend = NoRecommend;
- if (isOpenMPSimdDirective(ParentRegion) && CurrentRegion != OMPD_ordered) {
- // OpenMP [2.16, Nesting of Regions]
- // OpenMP constructs may not be nested inside a simd region.
- // OpenMP [2.8.1,simd Construct, Restrictions]
- // An ordered construct with the simd clause is the only OpenMP
- // construct that can appear in the simd region.
- // Allowing a SIMD construct nested in another SIMD construct is an
- // extension. The OpenMP 4.5 spec does not allow it. Issue a warning
- // message.
- SemaRef.Diag(StartLoc, (CurrentRegion != OMPD_simd)
- ? diag::err_omp_prohibited_region_simd
- : diag::warn_omp_nesting_simd);
- return CurrentRegion != OMPD_simd;
- }
- if (ParentRegion == OMPD_atomic) {
- // OpenMP [2.16, Nesting of Regions]
- // OpenMP constructs may not be nested inside an atomic region.
- SemaRef.Diag(StartLoc, diag::err_omp_prohibited_region_atomic);
- return true;
- }
- if (CurrentRegion == OMPD_section) {
- // OpenMP [2.7.2, sections Construct, Restrictions]
- // Orphaned section directives are prohibited. That is, the section
- // directives must appear within the sections construct and must not be
- // encountered elsewhere in the sections region.
- if (ParentRegion != OMPD_sections &&
- ParentRegion != OMPD_parallel_sections) {
- SemaRef.Diag(StartLoc, diag::err_omp_orphaned_section_directive)
- << (ParentRegion != OMPD_unknown)
- << getOpenMPDirectiveName(ParentRegion);
- return true;
- }
- return false;
- }
- // Allow some constructs (except teams) to be orphaned (they could be
- // used in functions, called from OpenMP regions with the required
- // preconditions).
- if (ParentRegion == OMPD_unknown &&
- !isOpenMPNestingTeamsDirective(CurrentRegion))
- return false;
- if (CurrentRegion == OMPD_cancellation_point ||
- CurrentRegion == OMPD_cancel) {
- // OpenMP [2.16, Nesting of Regions]
- // A cancellation point construct for which construct-type-clause is
- // taskgroup must be nested inside a task construct. A cancellation
- // point construct for which construct-type-clause is not taskgroup must
- // be closely nested inside an OpenMP construct that matches the type
- // specified in construct-type-clause.
- // A cancel construct for which construct-type-clause is taskgroup must be
- // nested inside a task construct. A cancel construct for which
- // construct-type-clause is not taskgroup must be closely nested inside an
- // OpenMP construct that matches the type specified in
- // construct-type-clause.
- NestingProhibited =
- !((CancelRegion == OMPD_parallel &&
- (ParentRegion == OMPD_parallel ||
- ParentRegion == OMPD_target_parallel)) ||
- (CancelRegion == OMPD_for &&
- (ParentRegion == OMPD_for || ParentRegion == OMPD_parallel_for ||
- ParentRegion == OMPD_target_parallel_for ||
- ParentRegion == OMPD_distribute_parallel_for ||
- ParentRegion == OMPD_teams_distribute_parallel_for ||
- ParentRegion == OMPD_target_teams_distribute_parallel_for)) ||
- (CancelRegion == OMPD_taskgroup && ParentRegion == OMPD_task) ||
- (CancelRegion == OMPD_sections &&
- (ParentRegion == OMPD_section || ParentRegion == OMPD_sections ||
- ParentRegion == OMPD_parallel_sections)));
- } else if (CurrentRegion == OMPD_master) {
- // OpenMP [2.16, Nesting of Regions]
- // A master region may not be closely nested inside a worksharing,
- // atomic, or explicit task region.
- NestingProhibited = isOpenMPWorksharingDirective(ParentRegion) ||
- isOpenMPTaskingDirective(ParentRegion);
- } else if (CurrentRegion == OMPD_critical && CurrentName.getName()) {
- // OpenMP [2.16, Nesting of Regions]
- // A critical region may not be nested (closely or otherwise) inside a
- // critical region with the same name. Note that this restriction is not
- // sufficient to prevent deadlock.
- SourceLocation PreviousCriticalLoc;
- bool DeadLock = Stack->hasDirective(
- [CurrentName, &PreviousCriticalLoc](OpenMPDirectiveKind K,
- const DeclarationNameInfo &DNI,
- SourceLocation Loc) {
- if (K == OMPD_critical && DNI.getName() == CurrentName.getName()) {
- PreviousCriticalLoc = Loc;
- return true;
- }
- return false;
- },
- false /* skip top directive */);
- if (DeadLock) {
- SemaRef.Diag(StartLoc,
- diag::err_omp_prohibited_region_critical_same_name)
- << CurrentName.getName();
- if (PreviousCriticalLoc.isValid())
- SemaRef.Diag(PreviousCriticalLoc,
- diag::note_omp_previous_critical_region);
- return true;
- }
- } else if (CurrentRegion == OMPD_barrier) {
- // OpenMP [2.16, Nesting of Regions]
- // A barrier region may not be closely nested inside a worksharing,
- // explicit task, critical, ordered, atomic, or master region.
- NestingProhibited = isOpenMPWorksharingDirective(ParentRegion) ||
- isOpenMPTaskingDirective(ParentRegion) ||
- ParentRegion == OMPD_master ||
- ParentRegion == OMPD_critical ||
- ParentRegion == OMPD_ordered;
- } else if (isOpenMPWorksharingDirective(CurrentRegion) &&
- !isOpenMPParallelDirective(CurrentRegion) &&
- !isOpenMPTeamsDirective(CurrentRegion)) {
- // OpenMP [2.16, Nesting of Regions]
- // A worksharing region may not be closely nested inside a worksharing,
- // explicit task, critical, ordered, atomic, or master region.
- NestingProhibited = isOpenMPWorksharingDirective(ParentRegion) ||
- isOpenMPTaskingDirective(ParentRegion) ||
- ParentRegion == OMPD_master ||
- ParentRegion == OMPD_critical ||
- ParentRegion == OMPD_ordered;
- Recommend = ShouldBeInParallelRegion;
- } else if (CurrentRegion == OMPD_ordered) {
- // OpenMP [2.16, Nesting of Regions]
- // An ordered region may not be closely nested inside a critical,
- // atomic, or explicit task region.
- // An ordered region must be closely nested inside a loop region (or
- // parallel loop region) with an ordered clause.
- // OpenMP [2.8.1,simd Construct, Restrictions]
- // An ordered construct with the simd clause is the only OpenMP construct
- // that can appear in the simd region.
- NestingProhibited = ParentRegion == OMPD_critical ||
- isOpenMPTaskingDirective(ParentRegion) ||
- !(isOpenMPSimdDirective(ParentRegion) ||
- Stack->isParentOrderedRegion());
- Recommend = ShouldBeInOrderedRegion;
- } else if (isOpenMPNestingTeamsDirective(CurrentRegion)) {
- // OpenMP [2.16, Nesting of Regions]
- // If specified, a teams construct must be contained within a target
- // construct.
- NestingProhibited = ParentRegion != OMPD_target;
- OrphanSeen = ParentRegion == OMPD_unknown;
- Recommend = ShouldBeInTargetRegion;
- }
- if (!NestingProhibited &&
- !isOpenMPTargetExecutionDirective(CurrentRegion) &&
- !isOpenMPTargetDataManagementDirective(CurrentRegion) &&
- (ParentRegion == OMPD_teams || ParentRegion == OMPD_target_teams)) {
- // OpenMP [2.16, Nesting of Regions]
- // distribute, parallel, parallel sections, parallel workshare, and the
- // parallel loop and parallel loop SIMD constructs are the only OpenMP
- // constructs that can be closely nested in the teams region.
- NestingProhibited = !isOpenMPParallelDirective(CurrentRegion) &&
- !isOpenMPDistributeDirective(CurrentRegion);
- Recommend = ShouldBeInParallelRegion;
- }
- if (!NestingProhibited &&
- isOpenMPNestingDistributeDirective(CurrentRegion)) {
- // OpenMP 4.5 [2.17 Nesting of Regions]
- // The region associated with the distribute construct must be strictly
- // nested inside a teams region
- NestingProhibited =
- (ParentRegion != OMPD_teams && ParentRegion != OMPD_target_teams);
- Recommend = ShouldBeInTeamsRegion;
- }
- if (!NestingProhibited &&
- (isOpenMPTargetExecutionDirective(CurrentRegion) ||
- isOpenMPTargetDataManagementDirective(CurrentRegion))) {
- // OpenMP 4.5 [2.17 Nesting of Regions]
- // If a target, target update, target data, target enter data, or
- // target exit data construct is encountered during execution of a
- // target region, the behavior is unspecified.
- NestingProhibited = Stack->hasDirective(
- [&OffendingRegion](OpenMPDirectiveKind K, const DeclarationNameInfo &,
- SourceLocation) {
- if (isOpenMPTargetExecutionDirective(K)) {
- OffendingRegion = K;
- return true;
- }
- return false;
- },
- false /* don't skip top directive */);
- CloseNesting = false;
- }
- if (NestingProhibited) {
- if (OrphanSeen) {
- SemaRef.Diag(StartLoc, diag::err_omp_orphaned_device_directive)
- << getOpenMPDirectiveName(CurrentRegion) << Recommend;
- } else {
- SemaRef.Diag(StartLoc, diag::err_omp_prohibited_region)
- << CloseNesting << getOpenMPDirectiveName(OffendingRegion)
- << Recommend << getOpenMPDirectiveName(CurrentRegion);
- }
- return true;
- }
- }
- return false;
- }
- static bool checkIfClauses(Sema &S, OpenMPDirectiveKind Kind,
- ArrayRef<OMPClause *> Clauses,
- ArrayRef<OpenMPDirectiveKind> AllowedNameModifiers) {
- bool ErrorFound = false;
- unsigned NamedModifiersNumber = 0;
- SmallVector<const OMPIfClause *, OMPC_unknown + 1> FoundNameModifiers(
- OMPD_unknown + 1);
- SmallVector<SourceLocation, 4> NameModifierLoc;
- for (const OMPClause *C : Clauses) {
- if (const auto *IC = dyn_cast_or_null<OMPIfClause>(C)) {
- // At most one if clause without a directive-name-modifier can appear on
- // the directive.
- OpenMPDirectiveKind CurNM = IC->getNameModifier();
- if (FoundNameModifiers[CurNM]) {
- S.Diag(C->getBeginLoc(), diag::err_omp_more_one_clause)
- << getOpenMPDirectiveName(Kind) << getOpenMPClauseName(OMPC_if)
- << (CurNM != OMPD_unknown) << getOpenMPDirectiveName(CurNM);
- ErrorFound = true;
- } else if (CurNM != OMPD_unknown) {
- NameModifierLoc.push_back(IC->getNameModifierLoc());
- ++NamedModifiersNumber;
- }
- FoundNameModifiers[CurNM] = IC;
- if (CurNM == OMPD_unknown)
- continue;
- // Check if the specified name modifier is allowed for the current
- // directive.
- // At most one if clause with the particular directive-name-modifier can
- // appear on the directive.
- bool MatchFound = false;
- for (auto NM : AllowedNameModifiers) {
- if (CurNM == NM) {
- MatchFound = true;
- break;
- }
- }
- if (!MatchFound) {
- S.Diag(IC->getNameModifierLoc(),
- diag::err_omp_wrong_if_directive_name_modifier)
- << getOpenMPDirectiveName(CurNM) << getOpenMPDirectiveName(Kind);
- ErrorFound = true;
- }
- }
- }
- // If any if clause on the directive includes a directive-name-modifier then
- // all if clauses on the directive must include a directive-name-modifier.
- if (FoundNameModifiers[OMPD_unknown] && NamedModifiersNumber > 0) {
- if (NamedModifiersNumber == AllowedNameModifiers.size()) {
- S.Diag(FoundNameModifiers[OMPD_unknown]->getBeginLoc(),
- diag::err_omp_no_more_if_clause);
- } else {
- std::string Values;
- std::string Sep(", ");
- unsigned AllowedCnt = 0;
- unsigned TotalAllowedNum =
- AllowedNameModifiers.size() - NamedModifiersNumber;
- for (unsigned Cnt = 0, End = AllowedNameModifiers.size(); Cnt < End;
- ++Cnt) {
- OpenMPDirectiveKind NM = AllowedNameModifiers[Cnt];
- if (!FoundNameModifiers[NM]) {
- Values += "'";
- Values += getOpenMPDirectiveName(NM);
- Values += "'";
- if (AllowedCnt + 2 == TotalAllowedNum)
- Values += " or ";
- else if (AllowedCnt + 1 != TotalAllowedNum)
- Values += Sep;
- ++AllowedCnt;
- }
- }
- S.Diag(FoundNameModifiers[OMPD_unknown]->getCondition()->getBeginLoc(),
- diag::err_omp_unnamed_if_clause)
- << (TotalAllowedNum > 1) << Values;
- }
- for (SourceLocation Loc : NameModifierLoc) {
- S.Diag(Loc, diag::note_omp_previous_named_if_clause);
- }
- ErrorFound = true;
- }
- return ErrorFound;
- }
- StmtResult Sema::ActOnOpenMPExecutableDirective(
- OpenMPDirectiveKind Kind, const DeclarationNameInfo &DirName,
- OpenMPDirectiveKind CancelRegion, ArrayRef<OMPClause *> Clauses,
- Stmt *AStmt, SourceLocation StartLoc, SourceLocation EndLoc) {
- StmtResult Res = StmtError();
- // First check CancelRegion which is then used in checkNestingOfRegions.
- if (checkCancelRegion(*this, Kind, CancelRegion, StartLoc) ||
- checkNestingOfRegions(*this, DSAStack, Kind, DirName, CancelRegion,
- StartLoc))
- return StmtError();
- llvm::SmallVector<OMPClause *, 8> ClausesWithImplicit;
- VarsWithInheritedDSAType VarsWithInheritedDSA;
- bool ErrorFound = false;
- ClausesWithImplicit.append(Clauses.begin(), Clauses.end());
- if (AStmt && !CurContext->isDependentContext()) {
- assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
- // Check default data sharing attributes for referenced variables.
- DSAAttrChecker DSAChecker(DSAStack, *this, cast<CapturedStmt>(AStmt));
- int ThisCaptureLevel = getOpenMPCaptureLevels(Kind);
- Stmt *S = AStmt;
- while (--ThisCaptureLevel >= 0)
- S = cast<CapturedStmt>(S)->getCapturedStmt();
- DSAChecker.Visit(S);
- if (DSAChecker.isErrorFound())
- return StmtError();
- // Generate list of implicitly defined firstprivate variables.
- VarsWithInheritedDSA = DSAChecker.getVarsWithInheritedDSA();
- SmallVector<Expr *, 4> ImplicitFirstprivates(
- DSAChecker.getImplicitFirstprivate().begin(),
- DSAChecker.getImplicitFirstprivate().end());
- SmallVector<Expr *, 4> ImplicitMaps(DSAChecker.getImplicitMap().begin(),
- DSAChecker.getImplicitMap().end());
- // Mark taskgroup task_reduction descriptors as implicitly firstprivate.
- for (OMPClause *C : Clauses) {
- if (auto *IRC = dyn_cast<OMPInReductionClause>(C)) {
- for (Expr *E : IRC->taskgroup_descriptors())
- if (E)
- ImplicitFirstprivates.emplace_back(E);
- }
- }
- if (!ImplicitFirstprivates.empty()) {
- if (OMPClause *Implicit = ActOnOpenMPFirstprivateClause(
- ImplicitFirstprivates, SourceLocation(), SourceLocation(),
- SourceLocation())) {
- ClausesWithImplicit.push_back(Implicit);
- ErrorFound = cast<OMPFirstprivateClause>(Implicit)->varlist_size() !=
- ImplicitFirstprivates.size();
- } else {
- ErrorFound = true;
- }
- }
- if (!ImplicitMaps.empty()) {
- if (OMPClause *Implicit = ActOnOpenMPMapClause(
- llvm::None, llvm::None, OMPC_MAP_tofrom,
- /*IsMapTypeImplicit=*/true, SourceLocation(), SourceLocation(),
- ImplicitMaps, SourceLocation(), SourceLocation(),
- SourceLocation())) {
- ClausesWithImplicit.emplace_back(Implicit);
- ErrorFound |=
- cast<OMPMapClause>(Implicit)->varlist_size() != ImplicitMaps.size();
- } else {
- ErrorFound = true;
- }
- }
- }
- llvm::SmallVector<OpenMPDirectiveKind, 4> AllowedNameModifiers;
- switch (Kind) {
- case OMPD_parallel:
- Res = ActOnOpenMPParallelDirective(ClausesWithImplicit, AStmt, StartLoc,
- EndLoc);
- AllowedNameModifiers.push_back(OMPD_parallel);
- break;
- case OMPD_simd:
- Res = ActOnOpenMPSimdDirective(ClausesWithImplicit, AStmt, StartLoc, EndLoc,
- VarsWithInheritedDSA);
- break;
- case OMPD_for:
- Res = ActOnOpenMPForDirective(ClausesWithImplicit, AStmt, StartLoc, EndLoc,
- VarsWithInheritedDSA);
- break;
- case OMPD_for_simd:
- Res = ActOnOpenMPForSimdDirective(ClausesWithImplicit, AStmt, StartLoc,
- EndLoc, VarsWithInheritedDSA);
- break;
- case OMPD_sections:
- Res = ActOnOpenMPSectionsDirective(ClausesWithImplicit, AStmt, StartLoc,
- EndLoc);
- break;
- case OMPD_section:
- assert(ClausesWithImplicit.empty() &&
- "No clauses are allowed for 'omp section' directive");
- Res = ActOnOpenMPSectionDirective(AStmt, StartLoc, EndLoc);
- break;
- case OMPD_single:
- Res = ActOnOpenMPSingleDirective(ClausesWithImplicit, AStmt, StartLoc,
- EndLoc);
- break;
- case OMPD_master:
- assert(ClausesWithImplicit.empty() &&
- "No clauses are allowed for 'omp master' directive");
- Res = ActOnOpenMPMasterDirective(AStmt, StartLoc, EndLoc);
- break;
- case OMPD_critical:
- Res = ActOnOpenMPCriticalDirective(DirName, ClausesWithImplicit, AStmt,
- StartLoc, EndLoc);
- break;
- case OMPD_parallel_for:
- Res = ActOnOpenMPParallelForDirective(ClausesWithImplicit, AStmt, StartLoc,
- EndLoc, VarsWithInheritedDSA);
- AllowedNameModifiers.push_back(OMPD_parallel);
- break;
- case OMPD_parallel_for_simd:
- Res = ActOnOpenMPParallelForSimdDirective(
- ClausesWithImplicit, AStmt, StartLoc, EndLoc, VarsWithInheritedDSA);
- AllowedNameModifiers.push_back(OMPD_parallel);
- break;
- case OMPD_parallel_sections:
- Res = ActOnOpenMPParallelSectionsDirective(ClausesWithImplicit, AStmt,
- StartLoc, EndLoc);
- AllowedNameModifiers.push_back(OMPD_parallel);
- break;
- case OMPD_task:
- Res =
- ActOnOpenMPTaskDirective(ClausesWithImplicit, AStmt, StartLoc, EndLoc);
- AllowedNameModifiers.push_back(OMPD_task);
- break;
- case OMPD_taskyield:
- assert(ClausesWithImplicit.empty() &&
- "No clauses are allowed for 'omp taskyield' directive");
- assert(AStmt == nullptr &&
- "No associated statement allowed for 'omp taskyield' directive");
- Res = ActOnOpenMPTaskyieldDirective(StartLoc, EndLoc);
- break;
- case OMPD_barrier:
- assert(ClausesWithImplicit.empty() &&
- "No clauses are allowed for 'omp barrier' directive");
- assert(AStmt == nullptr &&
- "No associated statement allowed for 'omp barrier' directive");
- Res = ActOnOpenMPBarrierDirective(StartLoc, EndLoc);
- break;
- case OMPD_taskwait:
- assert(ClausesWithImplicit.empty() &&
- "No clauses are allowed for 'omp taskwait' directive");
- assert(AStmt == nullptr &&
- "No associated statement allowed for 'omp taskwait' directive");
- Res = ActOnOpenMPTaskwaitDirective(StartLoc, EndLoc);
- break;
- case OMPD_taskgroup:
- Res = ActOnOpenMPTaskgroupDirective(ClausesWithImplicit, AStmt, StartLoc,
- EndLoc);
- break;
- case OMPD_flush:
- assert(AStmt == nullptr &&
- "No associated statement allowed for 'omp flush' directive");
- Res = ActOnOpenMPFlushDirective(ClausesWithImplicit, StartLoc, EndLoc);
- break;
- case OMPD_ordered:
- Res = ActOnOpenMPOrderedDirective(ClausesWithImplicit, AStmt, StartLoc,
- EndLoc);
- break;
- case OMPD_atomic:
- Res = ActOnOpenMPAtomicDirective(ClausesWithImplicit, AStmt, StartLoc,
- EndLoc);
- break;
- case OMPD_teams:
- Res =
- ActOnOpenMPTeamsDirective(ClausesWithImplicit, AStmt, StartLoc, EndLoc);
- break;
- case OMPD_target:
- Res = ActOnOpenMPTargetDirective(ClausesWithImplicit, AStmt, StartLoc,
- EndLoc);
- AllowedNameModifiers.push_back(OMPD_target);
- break;
- case OMPD_target_parallel:
- Res = ActOnOpenMPTargetParallelDirective(ClausesWithImplicit, AStmt,
- StartLoc, EndLoc);
- AllowedNameModifiers.push_back(OMPD_target);
- AllowedNameModifiers.push_back(OMPD_parallel);
- break;
- case OMPD_target_parallel_for:
- Res = ActOnOpenMPTargetParallelForDirective(
- ClausesWithImplicit, AStmt, StartLoc, EndLoc, VarsWithInheritedDSA);
- AllowedNameModifiers.push_back(OMPD_target);
- AllowedNameModifiers.push_back(OMPD_parallel);
- break;
- case OMPD_cancellation_point:
- assert(ClausesWithImplicit.empty() &&
- "No clauses are allowed for 'omp cancellation point' directive");
- assert(AStmt == nullptr && "No associated statement allowed for 'omp "
- "cancellation point' directive");
- Res = ActOnOpenMPCancellationPointDirective(StartLoc, EndLoc, CancelRegion);
- break;
- case OMPD_cancel:
- assert(AStmt == nullptr &&
- "No associated statement allowed for 'omp cancel' directive");
- Res = ActOnOpenMPCancelDirective(ClausesWithImplicit, StartLoc, EndLoc,
- CancelRegion);
- AllowedNameModifiers.push_back(OMPD_cancel);
- break;
- case OMPD_target_data:
- Res = ActOnOpenMPTargetDataDirective(ClausesWithImplicit, AStmt, StartLoc,
- EndLoc);
- AllowedNameModifiers.push_back(OMPD_target_data);
- break;
- case OMPD_target_enter_data:
- Res = ActOnOpenMPTargetEnterDataDirective(ClausesWithImplicit, StartLoc,
- EndLoc, AStmt);
- AllowedNameModifiers.push_back(OMPD_target_enter_data);
- break;
- case OMPD_target_exit_data:
- Res = ActOnOpenMPTargetExitDataDirective(ClausesWithImplicit, StartLoc,
- EndLoc, AStmt);
- AllowedNameModifiers.push_back(OMPD_target_exit_data);
- break;
- case OMPD_taskloop:
- Res = ActOnOpenMPTaskLoopDirective(ClausesWithImplicit, AStmt, StartLoc,
- EndLoc, VarsWithInheritedDSA);
- AllowedNameModifiers.push_back(OMPD_taskloop);
- break;
- case OMPD_taskloop_simd:
- Res = ActOnOpenMPTaskLoopSimdDirective(ClausesWithImplicit, AStmt, StartLoc,
- EndLoc, VarsWithInheritedDSA);
- AllowedNameModifiers.push_back(OMPD_taskloop);
- break;
- case OMPD_distribute:
- Res = ActOnOpenMPDistributeDirective(ClausesWithImplicit, AStmt, StartLoc,
- EndLoc, VarsWithInheritedDSA);
- break;
- case OMPD_target_update:
- Res = ActOnOpenMPTargetUpdateDirective(ClausesWithImplicit, StartLoc,
- EndLoc, AStmt);
- AllowedNameModifiers.push_back(OMPD_target_update);
- break;
- case OMPD_distribute_parallel_for:
- Res = ActOnOpenMPDistributeParallelForDirective(
- ClausesWithImplicit, AStmt, StartLoc, EndLoc, VarsWithInheritedDSA);
- AllowedNameModifiers.push_back(OMPD_parallel);
- break;
- case OMPD_distribute_parallel_for_simd:
- Res = ActOnOpenMPDistributeParallelForSimdDirective(
- ClausesWithImplicit, AStmt, StartLoc, EndLoc, VarsWithInheritedDSA);
- AllowedNameModifiers.push_back(OMPD_parallel);
- break;
- case OMPD_distribute_simd:
- Res = ActOnOpenMPDistributeSimdDirective(
- ClausesWithImplicit, AStmt, StartLoc, EndLoc, VarsWithInheritedDSA);
- break;
- case OMPD_target_parallel_for_simd:
- Res = ActOnOpenMPTargetParallelForSimdDirective(
- ClausesWithImplicit, AStmt, StartLoc, EndLoc, VarsWithInheritedDSA);
- AllowedNameModifiers.push_back(OMPD_target);
- AllowedNameModifiers.push_back(OMPD_parallel);
- break;
- case OMPD_target_simd:
- Res = ActOnOpenMPTargetSimdDirective(ClausesWithImplicit, AStmt, StartLoc,
- EndLoc, VarsWithInheritedDSA);
- AllowedNameModifiers.push_back(OMPD_target);
- break;
- case OMPD_teams_distribute:
- Res = ActOnOpenMPTeamsDistributeDirective(
- ClausesWithImplicit, AStmt, StartLoc, EndLoc, VarsWithInheritedDSA);
- break;
- case OMPD_teams_distribute_simd:
- Res = ActOnOpenMPTeamsDistributeSimdDirective(
- ClausesWithImplicit, AStmt, StartLoc, EndLoc, VarsWithInheritedDSA);
- break;
- case OMPD_teams_distribute_parallel_for_simd:
- Res = ActOnOpenMPTeamsDistributeParallelForSimdDirective(
- ClausesWithImplicit, AStmt, StartLoc, EndLoc, VarsWithInheritedDSA);
- AllowedNameModifiers.push_back(OMPD_parallel);
- break;
- case OMPD_teams_distribute_parallel_for:
- Res = ActOnOpenMPTeamsDistributeParallelForDirective(
- ClausesWithImplicit, AStmt, StartLoc, EndLoc, VarsWithInheritedDSA);
- AllowedNameModifiers.push_back(OMPD_parallel);
- break;
- case OMPD_target_teams:
- Res = ActOnOpenMPTargetTeamsDirective(ClausesWithImplicit, AStmt, StartLoc,
- EndLoc);
- AllowedNameModifiers.push_back(OMPD_target);
- break;
- case OMPD_target_teams_distribute:
- Res = ActOnOpenMPTargetTeamsDistributeDirective(
- ClausesWithImplicit, AStmt, StartLoc, EndLoc, VarsWithInheritedDSA);
- AllowedNameModifiers.push_back(OMPD_target);
- break;
- case OMPD_target_teams_distribute_parallel_for:
- Res = ActOnOpenMPTargetTeamsDistributeParallelForDirective(
- ClausesWithImplicit, AStmt, StartLoc, EndLoc, VarsWithInheritedDSA);
- AllowedNameModifiers.push_back(OMPD_target);
- AllowedNameModifiers.push_back(OMPD_parallel);
- break;
- case OMPD_target_teams_distribute_parallel_for_simd:
- Res = ActOnOpenMPTargetTeamsDistributeParallelForSimdDirective(
- ClausesWithImplicit, AStmt, StartLoc, EndLoc, VarsWithInheritedDSA);
- AllowedNameModifiers.push_back(OMPD_target);
- AllowedNameModifiers.push_back(OMPD_parallel);
- break;
- case OMPD_target_teams_distribute_simd:
- Res = ActOnOpenMPTargetTeamsDistributeSimdDirective(
- ClausesWithImplicit, AStmt, StartLoc, EndLoc, VarsWithInheritedDSA);
- AllowedNameModifiers.push_back(OMPD_target);
- break;
- case OMPD_declare_target:
- case OMPD_end_declare_target:
- case OMPD_threadprivate:
- case OMPD_declare_reduction:
- case OMPD_declare_simd:
- case OMPD_requires:
- llvm_unreachable("OpenMP Directive is not allowed");
- case OMPD_unknown:
- llvm_unreachable("Unknown OpenMP directive");
- }
- for (const auto &P : VarsWithInheritedDSA) {
- Diag(P.second->getExprLoc(), diag::err_omp_no_dsa_for_variable)
- << P.first << P.second->getSourceRange();
- }
- ErrorFound = !VarsWithInheritedDSA.empty() || ErrorFound;
- if (!AllowedNameModifiers.empty())
- ErrorFound = checkIfClauses(*this, Kind, Clauses, AllowedNameModifiers) ||
- ErrorFound;
- if (ErrorFound)
- return StmtError();
- return Res;
- }
- Sema::DeclGroupPtrTy Sema::ActOnOpenMPDeclareSimdDirective(
- DeclGroupPtrTy DG, OMPDeclareSimdDeclAttr::BranchStateTy BS, Expr *Simdlen,
- ArrayRef<Expr *> Uniforms, ArrayRef<Expr *> Aligneds,
- ArrayRef<Expr *> Alignments, ArrayRef<Expr *> Linears,
- ArrayRef<unsigned> LinModifiers, ArrayRef<Expr *> Steps, SourceRange SR) {
- assert(Aligneds.size() == Alignments.size());
- assert(Linears.size() == LinModifiers.size());
- assert(Linears.size() == Steps.size());
- if (!DG || DG.get().isNull())
- return DeclGroupPtrTy();
- if (!DG.get().isSingleDecl()) {
- Diag(SR.getBegin(), diag::err_omp_single_decl_in_declare_simd);
- return DG;
- }
- Decl *ADecl = DG.get().getSingleDecl();
- if (auto *FTD = dyn_cast<FunctionTemplateDecl>(ADecl))
- ADecl = FTD->getTemplatedDecl();
- auto *FD = dyn_cast<FunctionDecl>(ADecl);
- if (!FD) {
- Diag(ADecl->getLocation(), diag::err_omp_function_expected);
- return DeclGroupPtrTy();
- }
- // OpenMP [2.8.2, declare simd construct, Description]
- // The parameter of the simdlen clause must be a constant positive integer
- // expression.
- ExprResult SL;
- if (Simdlen)
- SL = VerifyPositiveIntegerConstantInClause(Simdlen, OMPC_simdlen);
- // OpenMP [2.8.2, declare simd construct, Description]
- // The special this pointer can be used as if was one of the arguments to the
- // function in any of the linear, aligned, or uniform clauses.
- // The uniform clause declares one or more arguments to have an invariant
- // value for all concurrent invocations of the function in the execution of a
- // single SIMD loop.
- llvm::DenseMap<const Decl *, const Expr *> UniformedArgs;
- const Expr *UniformedLinearThis = nullptr;
- for (const Expr *E : Uniforms) {
- E = E->IgnoreParenImpCasts();
- if (const auto *DRE = dyn_cast<DeclRefExpr>(E))
- if (const auto *PVD = dyn_cast<ParmVarDecl>(DRE->getDecl()))
- if (FD->getNumParams() > PVD->getFunctionScopeIndex() &&
- FD->getParamDecl(PVD->getFunctionScopeIndex())
- ->getCanonicalDecl() == PVD->getCanonicalDecl()) {
- UniformedArgs.try_emplace(PVD->getCanonicalDecl(), E);
- continue;
- }
- if (isa<CXXThisExpr>(E)) {
- UniformedLinearThis = E;
- continue;
- }
- Diag(E->getExprLoc(), diag::err_omp_param_or_this_in_clause)
- << FD->getDeclName() << (isa<CXXMethodDecl>(ADecl) ? 1 : 0);
- }
- // OpenMP [2.8.2, declare simd construct, Description]
- // The aligned clause declares that the object to which each list item points
- // is aligned to the number of bytes expressed in the optional parameter of
- // the aligned clause.
- // The special this pointer can be used as if was one of the arguments to the
- // function in any of the linear, aligned, or uniform clauses.
- // The type of list items appearing in the aligned clause must be array,
- // pointer, reference to array, or reference to pointer.
- llvm::DenseMap<const Decl *, const Expr *> AlignedArgs;
- const Expr *AlignedThis = nullptr;
- for (const Expr *E : Aligneds) {
- E = E->IgnoreParenImpCasts();
- if (const auto *DRE = dyn_cast<DeclRefExpr>(E))
- if (const auto *PVD = dyn_cast<ParmVarDecl>(DRE->getDecl())) {
- const VarDecl *CanonPVD = PVD->getCanonicalDecl();
- if (FD->getNumParams() > PVD->getFunctionScopeIndex() &&
- FD->getParamDecl(PVD->getFunctionScopeIndex())
- ->getCanonicalDecl() == CanonPVD) {
- // OpenMP [2.8.1, simd construct, Restrictions]
- // A list-item cannot appear in more than one aligned clause.
- if (AlignedArgs.count(CanonPVD) > 0) {
- Diag(E->getExprLoc(), diag::err_omp_aligned_twice)
- << 1 << E->getSourceRange();
- Diag(AlignedArgs[CanonPVD]->getExprLoc(),
- diag::note_omp_explicit_dsa)
- << getOpenMPClauseName(OMPC_aligned);
- continue;
- }
- AlignedArgs[CanonPVD] = E;
- QualType QTy = PVD->getType()
- .getNonReferenceType()
- .getUnqualifiedType()
- .getCanonicalType();
- const Type *Ty = QTy.getTypePtrOrNull();
- if (!Ty || (!Ty->isArrayType() && !Ty->isPointerType())) {
- Diag(E->getExprLoc(), diag::err_omp_aligned_expected_array_or_ptr)
- << QTy << getLangOpts().CPlusPlus << E->getSourceRange();
- Diag(PVD->getLocation(), diag::note_previous_decl) << PVD;
- }
- continue;
- }
- }
- if (isa<CXXThisExpr>(E)) {
- if (AlignedThis) {
- Diag(E->getExprLoc(), diag::err_omp_aligned_twice)
- << 2 << E->getSourceRange();
- Diag(AlignedThis->getExprLoc(), diag::note_omp_explicit_dsa)
- << getOpenMPClauseName(OMPC_aligned);
- }
- AlignedThis = E;
- continue;
- }
- Diag(E->getExprLoc(), diag::err_omp_param_or_this_in_clause)
- << FD->getDeclName() << (isa<CXXMethodDecl>(ADecl) ? 1 : 0);
- }
- // The optional parameter of the aligned clause, alignment, must be a constant
- // positive integer expression. If no optional parameter is specified,
- // implementation-defined default alignments for SIMD instructions on the
- // target platforms are assumed.
- SmallVector<const Expr *, 4> NewAligns;
- for (Expr *E : Alignments) {
- ExprResult Align;
- if (E)
- Align = VerifyPositiveIntegerConstantInClause(E, OMPC_aligned);
- NewAligns.push_back(Align.get());
- }
- // OpenMP [2.8.2, declare simd construct, Description]
- // The linear clause declares one or more list items to be private to a SIMD
- // lane and to have a linear relationship with respect to the iteration space
- // of a loop.
- // The special this pointer can be used as if was one of the arguments to the
- // function in any of the linear, aligned, or uniform clauses.
- // When a linear-step expression is specified in a linear clause it must be
- // either a constant integer expression or an integer-typed parameter that is
- // specified in a uniform clause on the directive.
- llvm::DenseMap<const Decl *, const Expr *> LinearArgs;
- const bool IsUniformedThis = UniformedLinearThis != nullptr;
- auto MI = LinModifiers.begin();
- for (const Expr *E : Linears) {
- auto LinKind = static_cast<OpenMPLinearClauseKind>(*MI);
- ++MI;
- E = E->IgnoreParenImpCasts();
- if (const auto *DRE = dyn_cast<DeclRefExpr>(E))
- if (const auto *PVD = dyn_cast<ParmVarDecl>(DRE->getDecl())) {
- const VarDecl *CanonPVD = PVD->getCanonicalDecl();
- if (FD->getNumParams() > PVD->getFunctionScopeIndex() &&
- FD->getParamDecl(PVD->getFunctionScopeIndex())
- ->getCanonicalDecl() == CanonPVD) {
- // OpenMP [2.15.3.7, linear Clause, Restrictions]
- // A list-item cannot appear in more than one linear clause.
- if (LinearArgs.count(CanonPVD) > 0) {
- Diag(E->getExprLoc(), diag::err_omp_wrong_dsa)
- << getOpenMPClauseName(OMPC_linear)
- << getOpenMPClauseName(OMPC_linear) << E->getSourceRange();
- Diag(LinearArgs[CanonPVD]->getExprLoc(),
- diag::note_omp_explicit_dsa)
- << getOpenMPClauseName(OMPC_linear);
- continue;
- }
- // Each argument can appear in at most one uniform or linear clause.
- if (UniformedArgs.count(CanonPVD) > 0) {
- Diag(E->getExprLoc(), diag::err_omp_wrong_dsa)
- << getOpenMPClauseName(OMPC_linear)
- << getOpenMPClauseName(OMPC_uniform) << E->getSourceRange();
- Diag(UniformedArgs[CanonPVD]->getExprLoc(),
- diag::note_omp_explicit_dsa)
- << getOpenMPClauseName(OMPC_uniform);
- continue;
- }
- LinearArgs[CanonPVD] = E;
- if (E->isValueDependent() || E->isTypeDependent() ||
- E->isInstantiationDependent() ||
- E->containsUnexpandedParameterPack())
- continue;
- (void)CheckOpenMPLinearDecl(CanonPVD, E->getExprLoc(), LinKind,
- PVD->getOriginalType());
- continue;
- }
- }
- if (isa<CXXThisExpr>(E)) {
- if (UniformedLinearThis) {
- Diag(E->getExprLoc(), diag::err_omp_wrong_dsa)
- << getOpenMPClauseName(OMPC_linear)
- << getOpenMPClauseName(IsUniformedThis ? OMPC_uniform : OMPC_linear)
- << E->getSourceRange();
- Diag(UniformedLinearThis->getExprLoc(), diag::note_omp_explicit_dsa)
- << getOpenMPClauseName(IsUniformedThis ? OMPC_uniform
- : OMPC_linear);
- continue;
- }
- UniformedLinearThis = E;
- if (E->isValueDependent() || E->isTypeDependent() ||
- E->isInstantiationDependent() || E->containsUnexpandedParameterPack())
- continue;
- (void)CheckOpenMPLinearDecl(/*D=*/nullptr, E->getExprLoc(), LinKind,
- E->getType());
- continue;
- }
- Diag(E->getExprLoc(), diag::err_omp_param_or_this_in_clause)
- << FD->getDeclName() << (isa<CXXMethodDecl>(ADecl) ? 1 : 0);
- }
- Expr *Step = nullptr;
- Expr *NewStep = nullptr;
- SmallVector<Expr *, 4> NewSteps;
- for (Expr *E : Steps) {
- // Skip the same step expression, it was checked already.
- if (Step == E || !E) {
- NewSteps.push_back(E ? NewStep : nullptr);
- continue;
- }
- Step = E;
- if (const auto *DRE = dyn_cast<DeclRefExpr>(Step))
- if (const auto *PVD = dyn_cast<ParmVarDecl>(DRE->getDecl())) {
- const VarDecl *CanonPVD = PVD->getCanonicalDecl();
- if (UniformedArgs.count(CanonPVD) == 0) {
- Diag(Step->getExprLoc(), diag::err_omp_expected_uniform_param)
- << Step->getSourceRange();
- } else if (E->isValueDependent() || E->isTypeDependent() ||
- E->isInstantiationDependent() ||
- E->containsUnexpandedParameterPack() ||
- CanonPVD->getType()->hasIntegerRepresentation()) {
- NewSteps.push_back(Step);
- } else {
- Diag(Step->getExprLoc(), diag::err_omp_expected_int_param)
- << Step->getSourceRange();
- }
- continue;
- }
- NewStep = Step;
- if (Step && !Step->isValueDependent() && !Step->isTypeDependent() &&
- !Step->isInstantiationDependent() &&
- !Step->containsUnexpandedParameterPack()) {
- NewStep = PerformOpenMPImplicitIntegerConversion(Step->getExprLoc(), Step)
- .get();
- if (NewStep)
- NewStep = VerifyIntegerConstantExpression(NewStep).get();
- }
- NewSteps.push_back(NewStep);
- }
- auto *NewAttr = OMPDeclareSimdDeclAttr::CreateImplicit(
- Context, BS, SL.get(), const_cast<Expr **>(Uniforms.data()),
- Uniforms.size(), const_cast<Expr **>(Aligneds.data()), Aligneds.size(),
- const_cast<Expr **>(NewAligns.data()), NewAligns.size(),
- const_cast<Expr **>(Linears.data()), Linears.size(),
- const_cast<unsigned *>(LinModifiers.data()), LinModifiers.size(),
- NewSteps.data(), NewSteps.size(), SR);
- ADecl->addAttr(NewAttr);
- return ConvertDeclToDeclGroup(ADecl);
- }
- StmtResult Sema::ActOnOpenMPParallelDirective(ArrayRef<OMPClause *> Clauses,
- Stmt *AStmt,
- SourceLocation StartLoc,
- SourceLocation EndLoc) {
- if (!AStmt)
- return StmtError();
- auto *CS = cast<CapturedStmt>(AStmt);
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- setFunctionHasBranchProtectedScope();
- return OMPParallelDirective::Create(Context, StartLoc, EndLoc, Clauses, AStmt,
- DSAStack->isCancelRegion());
- }
- namespace {
- /// Helper class for checking canonical form of the OpenMP loops and
- /// extracting iteration space of each loop in the loop nest, that will be used
- /// for IR generation.
- class OpenMPIterationSpaceChecker {
- /// Reference to Sema.
- Sema &SemaRef;
- /// A location for diagnostics (when there is no some better location).
- SourceLocation DefaultLoc;
- /// A location for diagnostics (when increment is not compatible).
- SourceLocation ConditionLoc;
- /// A source location for referring to loop init later.
- SourceRange InitSrcRange;
- /// A source location for referring to condition later.
- SourceRange ConditionSrcRange;
- /// A source location for referring to increment later.
- SourceRange IncrementSrcRange;
- /// Loop variable.
- ValueDecl *LCDecl = nullptr;
- /// Reference to loop variable.
- Expr *LCRef = nullptr;
- /// Lower bound (initializer for the var).
- Expr *LB = nullptr;
- /// Upper bound.
- Expr *UB = nullptr;
- /// Loop step (increment).
- Expr *Step = nullptr;
- /// This flag is true when condition is one of:
- /// Var < UB
- /// Var <= UB
- /// UB > Var
- /// UB >= Var
- /// This will have no value when the condition is !=
- llvm::Optional<bool> TestIsLessOp;
- /// This flag is true when condition is strict ( < or > ).
- bool TestIsStrictOp = false;
- /// This flag is true when step is subtracted on each iteration.
- bool SubtractStep = false;
- public:
- OpenMPIterationSpaceChecker(Sema &SemaRef, SourceLocation DefaultLoc)
- : SemaRef(SemaRef), DefaultLoc(DefaultLoc), ConditionLoc(DefaultLoc) {}
- /// Check init-expr for canonical loop form and save loop counter
- /// variable - #Var and its initialization value - #LB.
- bool checkAndSetInit(Stmt *S, bool EmitDiags = true);
- /// Check test-expr for canonical form, save upper-bound (#UB), flags
- /// for less/greater and for strict/non-strict comparison.
- bool checkAndSetCond(Expr *S);
- /// Check incr-expr for canonical loop form and return true if it
- /// does not conform, otherwise save loop step (#Step).
- bool checkAndSetInc(Expr *S);
- /// Return the loop counter variable.
- ValueDecl *getLoopDecl() const { return LCDecl; }
- /// Return the reference expression to loop counter variable.
- Expr *getLoopDeclRefExpr() const { return LCRef; }
- /// Source range of the loop init.
- SourceRange getInitSrcRange() const { return InitSrcRange; }
- /// Source range of the loop condition.
- SourceRange getConditionSrcRange() const { return ConditionSrcRange; }
- /// Source range of the loop increment.
- SourceRange getIncrementSrcRange() const { return IncrementSrcRange; }
- /// True if the step should be subtracted.
- bool shouldSubtractStep() const { return SubtractStep; }
- /// Build the expression to calculate the number of iterations.
- Expr *buildNumIterations(
- Scope *S, const bool LimitedType,
- llvm::MapVector<const Expr *, DeclRefExpr *> &Captures) const;
- /// Build the precondition expression for the loops.
- Expr *
- buildPreCond(Scope *S, Expr *Cond,
- llvm::MapVector<const Expr *, DeclRefExpr *> &Captures) const;
- /// Build reference expression to the counter be used for codegen.
- DeclRefExpr *
- buildCounterVar(llvm::MapVector<const Expr *, DeclRefExpr *> &Captures,
- DSAStackTy &DSA) const;
- /// Build reference expression to the private counter be used for
- /// codegen.
- Expr *buildPrivateCounterVar() const;
- /// Build initialization of the counter be used for codegen.
- Expr *buildCounterInit() const;
- /// Build step of the counter be used for codegen.
- Expr *buildCounterStep() const;
- /// Build loop data with counter value for depend clauses in ordered
- /// directives.
- Expr *
- buildOrderedLoopData(Scope *S, Expr *Counter,
- llvm::MapVector<const Expr *, DeclRefExpr *> &Captures,
- SourceLocation Loc, Expr *Inc = nullptr,
- OverloadedOperatorKind OOK = OO_Amp);
- /// Return true if any expression is dependent.
- bool dependent() const;
- private:
- /// Check the right-hand side of an assignment in the increment
- /// expression.
- bool checkAndSetIncRHS(Expr *RHS);
- /// Helper to set loop counter variable and its initializer.
- bool setLCDeclAndLB(ValueDecl *NewLCDecl, Expr *NewDeclRefExpr, Expr *NewLB);
- /// Helper to set upper bound.
- bool setUB(Expr *NewUB, llvm::Optional<bool> LessOp, bool StrictOp,
- SourceRange SR, SourceLocation SL);
- /// Helper to set loop increment.
- bool setStep(Expr *NewStep, bool Subtract);
- };
- bool OpenMPIterationSpaceChecker::dependent() const {
- if (!LCDecl) {
- assert(!LB && !UB && !Step);
- return false;
- }
- return LCDecl->getType()->isDependentType() ||
- (LB && LB->isValueDependent()) || (UB && UB->isValueDependent()) ||
- (Step && Step->isValueDependent());
- }
- bool OpenMPIterationSpaceChecker::setLCDeclAndLB(ValueDecl *NewLCDecl,
- Expr *NewLCRefExpr,
- Expr *NewLB) {
- // State consistency checking to ensure correct usage.
- assert(LCDecl == nullptr && LB == nullptr && LCRef == nullptr &&
- UB == nullptr && Step == nullptr && !TestIsLessOp && !TestIsStrictOp);
- if (!NewLCDecl || !NewLB)
- return true;
- LCDecl = getCanonicalDecl(NewLCDecl);
- LCRef = NewLCRefExpr;
- if (auto *CE = dyn_cast_or_null<CXXConstructExpr>(NewLB))
- if (const CXXConstructorDecl *Ctor = CE->getConstructor())
- if ((Ctor->isCopyOrMoveConstructor() ||
- Ctor->isConvertingConstructor(/*AllowExplicit=*/false)) &&
- CE->getNumArgs() > 0 && CE->getArg(0) != nullptr)
- NewLB = CE->getArg(0)->IgnoreParenImpCasts();
- LB = NewLB;
- return false;
- }
- bool OpenMPIterationSpaceChecker::setUB(Expr *NewUB, llvm::Optional<bool> LessOp,
- bool StrictOp, SourceRange SR,
- SourceLocation SL) {
- // State consistency checking to ensure correct usage.
- assert(LCDecl != nullptr && LB != nullptr && UB == nullptr &&
- Step == nullptr && !TestIsLessOp && !TestIsStrictOp);
- if (!NewUB)
- return true;
- UB = NewUB;
- if (LessOp)
- TestIsLessOp = LessOp;
- TestIsStrictOp = StrictOp;
- ConditionSrcRange = SR;
- ConditionLoc = SL;
- return false;
- }
- bool OpenMPIterationSpaceChecker::setStep(Expr *NewStep, bool Subtract) {
- // State consistency checking to ensure correct usage.
- assert(LCDecl != nullptr && LB != nullptr && Step == nullptr);
- if (!NewStep)
- return true;
- if (!NewStep->isValueDependent()) {
- // Check that the step is integer expression.
- SourceLocation StepLoc = NewStep->getBeginLoc();
- ExprResult Val = SemaRef.PerformOpenMPImplicitIntegerConversion(
- StepLoc, getExprAsWritten(NewStep));
- if (Val.isInvalid())
- return true;
- NewStep = Val.get();
- // OpenMP [2.6, Canonical Loop Form, Restrictions]
- // If test-expr is of form var relational-op b and relational-op is < or
- // <= then incr-expr must cause var to increase on each iteration of the
- // loop. If test-expr is of form var relational-op b and relational-op is
- // > or >= then incr-expr must cause var to decrease on each iteration of
- // the loop.
- // If test-expr is of form b relational-op var and relational-op is < or
- // <= then incr-expr must cause var to decrease on each iteration of the
- // loop. If test-expr is of form b relational-op var and relational-op is
- // > or >= then incr-expr must cause var to increase on each iteration of
- // the loop.
- llvm::APSInt Result;
- bool IsConstant = NewStep->isIntegerConstantExpr(Result, SemaRef.Context);
- bool IsUnsigned = !NewStep->getType()->hasSignedIntegerRepresentation();
- bool IsConstNeg =
- IsConstant && Result.isSigned() && (Subtract != Result.isNegative());
- bool IsConstPos =
- IsConstant && Result.isSigned() && (Subtract == Result.isNegative());
- bool IsConstZero = IsConstant && !Result.getBoolValue();
- // != with increment is treated as <; != with decrement is treated as >
- if (!TestIsLessOp.hasValue())
- TestIsLessOp = IsConstPos || (IsUnsigned && !Subtract);
- if (UB && (IsConstZero ||
- (TestIsLessOp.getValue() ?
- (IsConstNeg || (IsUnsigned && Subtract)) :
- (IsConstPos || (IsUnsigned && !Subtract))))) {
- SemaRef.Diag(NewStep->getExprLoc(),
- diag::err_omp_loop_incr_not_compatible)
- << LCDecl << TestIsLessOp.getValue() << NewStep->getSourceRange();
- SemaRef.Diag(ConditionLoc,
- diag::note_omp_loop_cond_requres_compatible_incr)
- << TestIsLessOp.getValue() << ConditionSrcRange;
- return true;
- }
- if (TestIsLessOp.getValue() == Subtract) {
- NewStep =
- SemaRef.CreateBuiltinUnaryOp(NewStep->getExprLoc(), UO_Minus, NewStep)
- .get();
- Subtract = !Subtract;
- }
- }
- Step = NewStep;
- SubtractStep = Subtract;
- return false;
- }
- bool OpenMPIterationSpaceChecker::checkAndSetInit(Stmt *S, bool EmitDiags) {
- // Check init-expr for canonical loop form and save loop counter
- // variable - #Var and its initialization value - #LB.
- // OpenMP [2.6] Canonical loop form. init-expr may be one of the following:
- // var = lb
- // integer-type var = lb
- // random-access-iterator-type var = lb
- // pointer-type var = lb
- //
- if (!S) {
- if (EmitDiags) {
- SemaRef.Diag(DefaultLoc, diag::err_omp_loop_not_canonical_init);
- }
- return true;
- }
- if (auto *ExprTemp = dyn_cast<ExprWithCleanups>(S))
- if (!ExprTemp->cleanupsHaveSideEffects())
- S = ExprTemp->getSubExpr();
- InitSrcRange = S->getSourceRange();
- if (Expr *E = dyn_cast<Expr>(S))
- S = E->IgnoreParens();
- if (auto *BO = dyn_cast<BinaryOperator>(S)) {
- if (BO->getOpcode() == BO_Assign) {
- Expr *LHS = BO->getLHS()->IgnoreParens();
- if (auto *DRE = dyn_cast<DeclRefExpr>(LHS)) {
- if (auto *CED = dyn_cast<OMPCapturedExprDecl>(DRE->getDecl()))
- if (auto *ME = dyn_cast<MemberExpr>(getExprAsWritten(CED->getInit())))
- return setLCDeclAndLB(ME->getMemberDecl(), ME, BO->getRHS());
- return setLCDeclAndLB(DRE->getDecl(), DRE, BO->getRHS());
- }
- if (auto *ME = dyn_cast<MemberExpr>(LHS)) {
- if (ME->isArrow() &&
- isa<CXXThisExpr>(ME->getBase()->IgnoreParenImpCasts()))
- return setLCDeclAndLB(ME->getMemberDecl(), ME, BO->getRHS());
- }
- }
- } else if (auto *DS = dyn_cast<DeclStmt>(S)) {
- if (DS->isSingleDecl()) {
- if (auto *Var = dyn_cast_or_null<VarDecl>(DS->getSingleDecl())) {
- if (Var->hasInit() && !Var->getType()->isReferenceType()) {
- // Accept non-canonical init form here but emit ext. warning.
- if (Var->getInitStyle() != VarDecl::CInit && EmitDiags)
- SemaRef.Diag(S->getBeginLoc(),
- diag::ext_omp_loop_not_canonical_init)
- << S->getSourceRange();
- return setLCDeclAndLB(
- Var,
- buildDeclRefExpr(SemaRef, Var,
- Var->getType().getNonReferenceType(),
- DS->getBeginLoc()),
- Var->getInit());
- }
- }
- }
- } else if (auto *CE = dyn_cast<CXXOperatorCallExpr>(S)) {
- if (CE->getOperator() == OO_Equal) {
- Expr *LHS = CE->getArg(0);
- if (auto *DRE = dyn_cast<DeclRefExpr>(LHS)) {
- if (auto *CED = dyn_cast<OMPCapturedExprDecl>(DRE->getDecl()))
- if (auto *ME = dyn_cast<MemberExpr>(getExprAsWritten(CED->getInit())))
- return setLCDeclAndLB(ME->getMemberDecl(), ME, BO->getRHS());
- return setLCDeclAndLB(DRE->getDecl(), DRE, CE->getArg(1));
- }
- if (auto *ME = dyn_cast<MemberExpr>(LHS)) {
- if (ME->isArrow() &&
- isa<CXXThisExpr>(ME->getBase()->IgnoreParenImpCasts()))
- return setLCDeclAndLB(ME->getMemberDecl(), ME, BO->getRHS());
- }
- }
- }
- if (dependent() || SemaRef.CurContext->isDependentContext())
- return false;
- if (EmitDiags) {
- SemaRef.Diag(S->getBeginLoc(), diag::err_omp_loop_not_canonical_init)
- << S->getSourceRange();
- }
- return true;
- }
- /// Ignore parenthesizes, implicit casts, copy constructor and return the
- /// variable (which may be the loop variable) if possible.
- static const ValueDecl *getInitLCDecl(const Expr *E) {
- if (!E)
- return nullptr;
- E = getExprAsWritten(E);
- if (const auto *CE = dyn_cast_or_null<CXXConstructExpr>(E))
- if (const CXXConstructorDecl *Ctor = CE->getConstructor())
- if ((Ctor->isCopyOrMoveConstructor() ||
- Ctor->isConvertingConstructor(/*AllowExplicit=*/false)) &&
- CE->getNumArgs() > 0 && CE->getArg(0) != nullptr)
- E = CE->getArg(0)->IgnoreParenImpCasts();
- if (const auto *DRE = dyn_cast_or_null<DeclRefExpr>(E)) {
- if (const auto *VD = dyn_cast<VarDecl>(DRE->getDecl()))
- return getCanonicalDecl(VD);
- }
- if (const auto *ME = dyn_cast_or_null<MemberExpr>(E))
- if (ME->isArrow() && isa<CXXThisExpr>(ME->getBase()->IgnoreParenImpCasts()))
- return getCanonicalDecl(ME->getMemberDecl());
- return nullptr;
- }
- bool OpenMPIterationSpaceChecker::checkAndSetCond(Expr *S) {
- // Check test-expr for canonical form, save upper-bound UB, flags for
- // less/greater and for strict/non-strict comparison.
- // OpenMP [2.6] Canonical loop form. Test-expr may be one of the following:
- // var relational-op b
- // b relational-op var
- //
- if (!S) {
- SemaRef.Diag(DefaultLoc, diag::err_omp_loop_not_canonical_cond) << LCDecl;
- return true;
- }
- S = getExprAsWritten(S);
- SourceLocation CondLoc = S->getBeginLoc();
- if (auto *BO = dyn_cast<BinaryOperator>(S)) {
- if (BO->isRelationalOp()) {
- if (getInitLCDecl(BO->getLHS()) == LCDecl)
- return setUB(BO->getRHS(),
- (BO->getOpcode() == BO_LT || BO->getOpcode() == BO_LE),
- (BO->getOpcode() == BO_LT || BO->getOpcode() == BO_GT),
- BO->getSourceRange(), BO->getOperatorLoc());
- if (getInitLCDecl(BO->getRHS()) == LCDecl)
- return setUB(BO->getLHS(),
- (BO->getOpcode() == BO_GT || BO->getOpcode() == BO_GE),
- (BO->getOpcode() == BO_LT || BO->getOpcode() == BO_GT),
- BO->getSourceRange(), BO->getOperatorLoc());
- } else if (BO->getOpcode() == BO_NE)
- return setUB(getInitLCDecl(BO->getLHS()) == LCDecl ?
- BO->getRHS() : BO->getLHS(),
- /*LessOp=*/llvm::None,
- /*StrictOp=*/true,
- BO->getSourceRange(), BO->getOperatorLoc());
- } else if (auto *CE = dyn_cast<CXXOperatorCallExpr>(S)) {
- if (CE->getNumArgs() == 2) {
- auto Op = CE->getOperator();
- switch (Op) {
- case OO_Greater:
- case OO_GreaterEqual:
- case OO_Less:
- case OO_LessEqual:
- if (getInitLCDecl(CE->getArg(0)) == LCDecl)
- return setUB(CE->getArg(1), Op == OO_Less || Op == OO_LessEqual,
- Op == OO_Less || Op == OO_Greater, CE->getSourceRange(),
- CE->getOperatorLoc());
- if (getInitLCDecl(CE->getArg(1)) == LCDecl)
- return setUB(CE->getArg(0), Op == OO_Greater || Op == OO_GreaterEqual,
- Op == OO_Less || Op == OO_Greater, CE->getSourceRange(),
- CE->getOperatorLoc());
- break;
- case OO_ExclaimEqual:
- return setUB(getInitLCDecl(CE->getArg(0)) == LCDecl ?
- CE->getArg(1) : CE->getArg(0),
- /*LessOp=*/llvm::None,
- /*StrictOp=*/true,
- CE->getSourceRange(),
- CE->getOperatorLoc());
- break;
- default:
- break;
- }
- }
- }
- if (dependent() || SemaRef.CurContext->isDependentContext())
- return false;
- SemaRef.Diag(CondLoc, diag::err_omp_loop_not_canonical_cond)
- << S->getSourceRange() << LCDecl;
- return true;
- }
- bool OpenMPIterationSpaceChecker::checkAndSetIncRHS(Expr *RHS) {
- // RHS of canonical loop form increment can be:
- // var + incr
- // incr + var
- // var - incr
- //
- RHS = RHS->IgnoreParenImpCasts();
- if (auto *BO = dyn_cast<BinaryOperator>(RHS)) {
- if (BO->isAdditiveOp()) {
- bool IsAdd = BO->getOpcode() == BO_Add;
- if (getInitLCDecl(BO->getLHS()) == LCDecl)
- return setStep(BO->getRHS(), !IsAdd);
- if (IsAdd && getInitLCDecl(BO->getRHS()) == LCDecl)
- return setStep(BO->getLHS(), /*Subtract=*/false);
- }
- } else if (auto *CE = dyn_cast<CXXOperatorCallExpr>(RHS)) {
- bool IsAdd = CE->getOperator() == OO_Plus;
- if ((IsAdd || CE->getOperator() == OO_Minus) && CE->getNumArgs() == 2) {
- if (getInitLCDecl(CE->getArg(0)) == LCDecl)
- return setStep(CE->getArg(1), !IsAdd);
- if (IsAdd && getInitLCDecl(CE->getArg(1)) == LCDecl)
- return setStep(CE->getArg(0), /*Subtract=*/false);
- }
- }
- if (dependent() || SemaRef.CurContext->isDependentContext())
- return false;
- SemaRef.Diag(RHS->getBeginLoc(), diag::err_omp_loop_not_canonical_incr)
- << RHS->getSourceRange() << LCDecl;
- return true;
- }
- bool OpenMPIterationSpaceChecker::checkAndSetInc(Expr *S) {
- // Check incr-expr for canonical loop form and return true if it
- // does not conform.
- // OpenMP [2.6] Canonical loop form. Test-expr may be one of the following:
- // ++var
- // var++
- // --var
- // var--
- // var += incr
- // var -= incr
- // var = var + incr
- // var = incr + var
- // var = var - incr
- //
- if (!S) {
- SemaRef.Diag(DefaultLoc, diag::err_omp_loop_not_canonical_incr) << LCDecl;
- return true;
- }
- if (auto *ExprTemp = dyn_cast<ExprWithCleanups>(S))
- if (!ExprTemp->cleanupsHaveSideEffects())
- S = ExprTemp->getSubExpr();
- IncrementSrcRange = S->getSourceRange();
- S = S->IgnoreParens();
- if (auto *UO = dyn_cast<UnaryOperator>(S)) {
- if (UO->isIncrementDecrementOp() &&
- getInitLCDecl(UO->getSubExpr()) == LCDecl)
- return setStep(SemaRef
- .ActOnIntegerConstant(UO->getBeginLoc(),
- (UO->isDecrementOp() ? -1 : 1))
- .get(),
- /*Subtract=*/false);
- } else if (auto *BO = dyn_cast<BinaryOperator>(S)) {
- switch (BO->getOpcode()) {
- case BO_AddAssign:
- case BO_SubAssign:
- if (getInitLCDecl(BO->getLHS()) == LCDecl)
- return setStep(BO->getRHS(), BO->getOpcode() == BO_SubAssign);
- break;
- case BO_Assign:
- if (getInitLCDecl(BO->getLHS()) == LCDecl)
- return checkAndSetIncRHS(BO->getRHS());
- break;
- default:
- break;
- }
- } else if (auto *CE = dyn_cast<CXXOperatorCallExpr>(S)) {
- switch (CE->getOperator()) {
- case OO_PlusPlus:
- case OO_MinusMinus:
- if (getInitLCDecl(CE->getArg(0)) == LCDecl)
- return setStep(SemaRef
- .ActOnIntegerConstant(
- CE->getBeginLoc(),
- ((CE->getOperator() == OO_MinusMinus) ? -1 : 1))
- .get(),
- /*Subtract=*/false);
- break;
- case OO_PlusEqual:
- case OO_MinusEqual:
- if (getInitLCDecl(CE->getArg(0)) == LCDecl)
- return setStep(CE->getArg(1), CE->getOperator() == OO_MinusEqual);
- break;
- case OO_Equal:
- if (getInitLCDecl(CE->getArg(0)) == LCDecl)
- return checkAndSetIncRHS(CE->getArg(1));
- break;
- default:
- break;
- }
- }
- if (dependent() || SemaRef.CurContext->isDependentContext())
- return false;
- SemaRef.Diag(S->getBeginLoc(), diag::err_omp_loop_not_canonical_incr)
- << S->getSourceRange() << LCDecl;
- return true;
- }
- static ExprResult
- tryBuildCapture(Sema &SemaRef, Expr *Capture,
- llvm::MapVector<const Expr *, DeclRefExpr *> &Captures) {
- if (SemaRef.CurContext->isDependentContext())
- return ExprResult(Capture);
- if (Capture->isEvaluatable(SemaRef.Context, Expr::SE_AllowSideEffects))
- return SemaRef.PerformImplicitConversion(
- Capture->IgnoreImpCasts(), Capture->getType(), Sema::AA_Converting,
- /*AllowExplicit=*/true);
- auto I = Captures.find(Capture);
- if (I != Captures.end())
- return buildCapture(SemaRef, Capture, I->second);
- DeclRefExpr *Ref = nullptr;
- ExprResult Res = buildCapture(SemaRef, Capture, Ref);
- Captures[Capture] = Ref;
- return Res;
- }
- /// Build the expression to calculate the number of iterations.
- Expr *OpenMPIterationSpaceChecker::buildNumIterations(
- Scope *S, const bool LimitedType,
- llvm::MapVector<const Expr *, DeclRefExpr *> &Captures) const {
- ExprResult Diff;
- QualType VarType = LCDecl->getType().getNonReferenceType();
- if (VarType->isIntegerType() || VarType->isPointerType() ||
- SemaRef.getLangOpts().CPlusPlus) {
- // Upper - Lower
- Expr *UBExpr = TestIsLessOp.getValue() ? UB : LB;
- Expr *LBExpr = TestIsLessOp.getValue() ? LB : UB;
- Expr *Upper = tryBuildCapture(SemaRef, UBExpr, Captures).get();
- Expr *Lower = tryBuildCapture(SemaRef, LBExpr, Captures).get();
- if (!Upper || !Lower)
- return nullptr;
- Diff = SemaRef.BuildBinOp(S, DefaultLoc, BO_Sub, Upper, Lower);
- if (!Diff.isUsable() && VarType->getAsCXXRecordDecl()) {
- // BuildBinOp already emitted error, this one is to point user to upper
- // and lower bound, and to tell what is passed to 'operator-'.
- SemaRef.Diag(Upper->getBeginLoc(), diag::err_omp_loop_diff_cxx)
- << Upper->getSourceRange() << Lower->getSourceRange();
- return nullptr;
- }
- }
- if (!Diff.isUsable())
- return nullptr;
- // Upper - Lower [- 1]
- if (TestIsStrictOp)
- Diff = SemaRef.BuildBinOp(
- S, DefaultLoc, BO_Sub, Diff.get(),
- SemaRef.ActOnIntegerConstant(SourceLocation(), 1).get());
- if (!Diff.isUsable())
- return nullptr;
- // Upper - Lower [- 1] + Step
- ExprResult NewStep = tryBuildCapture(SemaRef, Step, Captures);
- if (!NewStep.isUsable())
- return nullptr;
- Diff = SemaRef.BuildBinOp(S, DefaultLoc, BO_Add, Diff.get(), NewStep.get());
- if (!Diff.isUsable())
- return nullptr;
- // Parentheses (for dumping/debugging purposes only).
- Diff = SemaRef.ActOnParenExpr(DefaultLoc, DefaultLoc, Diff.get());
- if (!Diff.isUsable())
- return nullptr;
- // (Upper - Lower [- 1] + Step) / Step
- Diff = SemaRef.BuildBinOp(S, DefaultLoc, BO_Div, Diff.get(), NewStep.get());
- if (!Diff.isUsable())
- return nullptr;
- // OpenMP runtime requires 32-bit or 64-bit loop variables.
- QualType Type = Diff.get()->getType();
- ASTContext &C = SemaRef.Context;
- bool UseVarType = VarType->hasIntegerRepresentation() &&
- C.getTypeSize(Type) > C.getTypeSize(VarType);
- if (!Type->isIntegerType() || UseVarType) {
- unsigned NewSize =
- UseVarType ? C.getTypeSize(VarType) : C.getTypeSize(Type);
- bool IsSigned = UseVarType ? VarType->hasSignedIntegerRepresentation()
- : Type->hasSignedIntegerRepresentation();
- Type = C.getIntTypeForBitwidth(NewSize, IsSigned);
- if (!SemaRef.Context.hasSameType(Diff.get()->getType(), Type)) {
- Diff = SemaRef.PerformImplicitConversion(
- Diff.get(), Type, Sema::AA_Converting, /*AllowExplicit=*/true);
- if (!Diff.isUsable())
- return nullptr;
- }
- }
- if (LimitedType) {
- unsigned NewSize = (C.getTypeSize(Type) > 32) ? 64 : 32;
- if (NewSize != C.getTypeSize(Type)) {
- if (NewSize < C.getTypeSize(Type)) {
- assert(NewSize == 64 && "incorrect loop var size");
- SemaRef.Diag(DefaultLoc, diag::warn_omp_loop_64_bit_var)
- << InitSrcRange << ConditionSrcRange;
- }
- QualType NewType = C.getIntTypeForBitwidth(
- NewSize, Type->hasSignedIntegerRepresentation() ||
- C.getTypeSize(Type) < NewSize);
- if (!SemaRef.Context.hasSameType(Diff.get()->getType(), NewType)) {
- Diff = SemaRef.PerformImplicitConversion(Diff.get(), NewType,
- Sema::AA_Converting, true);
- if (!Diff.isUsable())
- return nullptr;
- }
- }
- }
- return Diff.get();
- }
- Expr *OpenMPIterationSpaceChecker::buildPreCond(
- Scope *S, Expr *Cond,
- llvm::MapVector<const Expr *, DeclRefExpr *> &Captures) const {
- // Try to build LB <op> UB, where <op> is <, >, <=, or >=.
- bool Suppress = SemaRef.getDiagnostics().getSuppressAllDiagnostics();
- SemaRef.getDiagnostics().setSuppressAllDiagnostics(/*Val=*/true);
- ExprResult NewLB = tryBuildCapture(SemaRef, LB, Captures);
- ExprResult NewUB = tryBuildCapture(SemaRef, UB, Captures);
- if (!NewLB.isUsable() || !NewUB.isUsable())
- return nullptr;
- ExprResult CondExpr =
- SemaRef.BuildBinOp(S, DefaultLoc,
- TestIsLessOp.getValue() ?
- (TestIsStrictOp ? BO_LT : BO_LE) :
- (TestIsStrictOp ? BO_GT : BO_GE),
- NewLB.get(), NewUB.get());
- if (CondExpr.isUsable()) {
- if (!SemaRef.Context.hasSameUnqualifiedType(CondExpr.get()->getType(),
- SemaRef.Context.BoolTy))
- CondExpr = SemaRef.PerformImplicitConversion(
- CondExpr.get(), SemaRef.Context.BoolTy, /*Action=*/Sema::AA_Casting,
- /*AllowExplicit=*/true);
- }
- SemaRef.getDiagnostics().setSuppressAllDiagnostics(Suppress);
- // Otherwise use original loop conditon and evaluate it in runtime.
- return CondExpr.isUsable() ? CondExpr.get() : Cond;
- }
- /// Build reference expression to the counter be used for codegen.
- DeclRefExpr *OpenMPIterationSpaceChecker::buildCounterVar(
- llvm::MapVector<const Expr *, DeclRefExpr *> &Captures,
- DSAStackTy &DSA) const {
- auto *VD = dyn_cast<VarDecl>(LCDecl);
- if (!VD) {
- VD = SemaRef.isOpenMPCapturedDecl(LCDecl);
- DeclRefExpr *Ref = buildDeclRefExpr(
- SemaRef, VD, VD->getType().getNonReferenceType(), DefaultLoc);
- const DSAStackTy::DSAVarData Data =
- DSA.getTopDSA(LCDecl, /*FromParent=*/false);
- // If the loop control decl is explicitly marked as private, do not mark it
- // as captured again.
- if (!isOpenMPPrivate(Data.CKind) || !Data.RefExpr)
- Captures.insert(std::make_pair(LCRef, Ref));
- return Ref;
- }
- return buildDeclRefExpr(SemaRef, VD, VD->getType().getNonReferenceType(),
- DefaultLoc);
- }
- Expr *OpenMPIterationSpaceChecker::buildPrivateCounterVar() const {
- if (LCDecl && !LCDecl->isInvalidDecl()) {
- QualType Type = LCDecl->getType().getNonReferenceType();
- VarDecl *PrivateVar = buildVarDecl(
- SemaRef, DefaultLoc, Type, LCDecl->getName(),
- LCDecl->hasAttrs() ? &LCDecl->getAttrs() : nullptr,
- isa<VarDecl>(LCDecl)
- ? buildDeclRefExpr(SemaRef, cast<VarDecl>(LCDecl), Type, DefaultLoc)
- : nullptr);
- if (PrivateVar->isInvalidDecl())
- return nullptr;
- return buildDeclRefExpr(SemaRef, PrivateVar, Type, DefaultLoc);
- }
- return nullptr;
- }
- /// Build initialization of the counter to be used for codegen.
- Expr *OpenMPIterationSpaceChecker::buildCounterInit() const { return LB; }
- /// Build step of the counter be used for codegen.
- Expr *OpenMPIterationSpaceChecker::buildCounterStep() const { return Step; }
- Expr *OpenMPIterationSpaceChecker::buildOrderedLoopData(
- Scope *S, Expr *Counter,
- llvm::MapVector<const Expr *, DeclRefExpr *> &Captures, SourceLocation Loc,
- Expr *Inc, OverloadedOperatorKind OOK) {
- Expr *Cnt = SemaRef.DefaultLvalueConversion(Counter).get();
- if (!Cnt)
- return nullptr;
- if (Inc) {
- assert((OOK == OO_Plus || OOK == OO_Minus) &&
- "Expected only + or - operations for depend clauses.");
- BinaryOperatorKind BOK = (OOK == OO_Plus) ? BO_Add : BO_Sub;
- Cnt = SemaRef.BuildBinOp(S, Loc, BOK, Cnt, Inc).get();
- if (!Cnt)
- return nullptr;
- }
- ExprResult Diff;
- QualType VarType = LCDecl->getType().getNonReferenceType();
- if (VarType->isIntegerType() || VarType->isPointerType() ||
- SemaRef.getLangOpts().CPlusPlus) {
- // Upper - Lower
- Expr *Upper =
- TestIsLessOp.getValue() ? Cnt : tryBuildCapture(SemaRef, UB, Captures).get();
- Expr *Lower =
- TestIsLessOp.getValue() ? tryBuildCapture(SemaRef, LB, Captures).get() : Cnt;
- if (!Upper || !Lower)
- return nullptr;
- Diff = SemaRef.BuildBinOp(S, DefaultLoc, BO_Sub, Upper, Lower);
- if (!Diff.isUsable() && VarType->getAsCXXRecordDecl()) {
- // BuildBinOp already emitted error, this one is to point user to upper
- // and lower bound, and to tell what is passed to 'operator-'.
- SemaRef.Diag(Upper->getBeginLoc(), diag::err_omp_loop_diff_cxx)
- << Upper->getSourceRange() << Lower->getSourceRange();
- return nullptr;
- }
- }
- if (!Diff.isUsable())
- return nullptr;
- // Parentheses (for dumping/debugging purposes only).
- Diff = SemaRef.ActOnParenExpr(DefaultLoc, DefaultLoc, Diff.get());
- if (!Diff.isUsable())
- return nullptr;
- ExprResult NewStep = tryBuildCapture(SemaRef, Step, Captures);
- if (!NewStep.isUsable())
- return nullptr;
- // (Upper - Lower) / Step
- Diff = SemaRef.BuildBinOp(S, DefaultLoc, BO_Div, Diff.get(), NewStep.get());
- if (!Diff.isUsable())
- return nullptr;
- return Diff.get();
- }
- /// Iteration space of a single for loop.
- struct LoopIterationSpace final {
- /// Condition of the loop.
- Expr *PreCond = nullptr;
- /// This expression calculates the number of iterations in the loop.
- /// It is always possible to calculate it before starting the loop.
- Expr *NumIterations = nullptr;
- /// The loop counter variable.
- Expr *CounterVar = nullptr;
- /// Private loop counter variable.
- Expr *PrivateCounterVar = nullptr;
- /// This is initializer for the initial value of #CounterVar.
- Expr *CounterInit = nullptr;
- /// This is step for the #CounterVar used to generate its update:
- /// #CounterVar = #CounterInit + #CounterStep * CurrentIteration.
- Expr *CounterStep = nullptr;
- /// Should step be subtracted?
- bool Subtract = false;
- /// Source range of the loop init.
- SourceRange InitSrcRange;
- /// Source range of the loop condition.
- SourceRange CondSrcRange;
- /// Source range of the loop increment.
- SourceRange IncSrcRange;
- };
- } // namespace
- void Sema::ActOnOpenMPLoopInitialization(SourceLocation ForLoc, Stmt *Init) {
- assert(getLangOpts().OpenMP && "OpenMP is not active.");
- assert(Init && "Expected loop in canonical form.");
- unsigned AssociatedLoops = DSAStack->getAssociatedLoops();
- if (AssociatedLoops > 0 &&
- isOpenMPLoopDirective(DSAStack->getCurrentDirective())) {
- OpenMPIterationSpaceChecker ISC(*this, ForLoc);
- if (!ISC.checkAndSetInit(Init, /*EmitDiags=*/false)) {
- if (ValueDecl *D = ISC.getLoopDecl()) {
- auto *VD = dyn_cast<VarDecl>(D);
- if (!VD) {
- if (VarDecl *Private = isOpenMPCapturedDecl(D)) {
- VD = Private;
- } else {
- DeclRefExpr *Ref = buildCapture(*this, D, ISC.getLoopDeclRefExpr(),
- /*WithInit=*/false);
- VD = cast<VarDecl>(Ref->getDecl());
- }
- }
- DSAStack->addLoopControlVariable(D, VD);
- const Decl *LD = DSAStack->getPossiblyLoopCunter();
- if (LD != D->getCanonicalDecl()) {
- DSAStack->resetPossibleLoopCounter();
- if (auto *Var = dyn_cast_or_null<VarDecl>(LD))
- MarkDeclarationsReferencedInExpr(
- buildDeclRefExpr(*this, const_cast<VarDecl *>(Var),
- Var->getType().getNonLValueExprType(Context),
- ForLoc, /*RefersToCapture=*/true));
- }
- }
- }
- DSAStack->setAssociatedLoops(AssociatedLoops - 1);
- }
- }
- /// Called on a for stmt to check and extract its iteration space
- /// for further processing (such as collapsing).
- static bool checkOpenMPIterationSpace(
- OpenMPDirectiveKind DKind, Stmt *S, Sema &SemaRef, DSAStackTy &DSA,
- unsigned CurrentNestedLoopCount, unsigned NestedLoopCount,
- unsigned TotalNestedLoopCount, Expr *CollapseLoopCountExpr,
- Expr *OrderedLoopCountExpr,
- Sema::VarsWithInheritedDSAType &VarsWithImplicitDSA,
- LoopIterationSpace &ResultIterSpace,
- llvm::MapVector<const Expr *, DeclRefExpr *> &Captures) {
- // OpenMP [2.6, Canonical Loop Form]
- // for (init-expr; test-expr; incr-expr) structured-block
- auto *For = dyn_cast_or_null<ForStmt>(S);
- if (!For) {
- SemaRef.Diag(S->getBeginLoc(), diag::err_omp_not_for)
- << (CollapseLoopCountExpr != nullptr || OrderedLoopCountExpr != nullptr)
- << getOpenMPDirectiveName(DKind) << TotalNestedLoopCount
- << (CurrentNestedLoopCount > 0) << CurrentNestedLoopCount;
- if (TotalNestedLoopCount > 1) {
- if (CollapseLoopCountExpr && OrderedLoopCountExpr)
- SemaRef.Diag(DSA.getConstructLoc(),
- diag::note_omp_collapse_ordered_expr)
- << 2 << CollapseLoopCountExpr->getSourceRange()
- << OrderedLoopCountExpr->getSourceRange();
- else if (CollapseLoopCountExpr)
- SemaRef.Diag(CollapseLoopCountExpr->getExprLoc(),
- diag::note_omp_collapse_ordered_expr)
- << 0 << CollapseLoopCountExpr->getSourceRange();
- else
- SemaRef.Diag(OrderedLoopCountExpr->getExprLoc(),
- diag::note_omp_collapse_ordered_expr)
- << 1 << OrderedLoopCountExpr->getSourceRange();
- }
- return true;
- }
- assert(For->getBody());
- OpenMPIterationSpaceChecker ISC(SemaRef, For->getForLoc());
- // Check init.
- Stmt *Init = For->getInit();
- if (ISC.checkAndSetInit(Init))
- return true;
- bool HasErrors = false;
- // Check loop variable's type.
- if (ValueDecl *LCDecl = ISC.getLoopDecl()) {
- Expr *LoopDeclRefExpr = ISC.getLoopDeclRefExpr();
- // OpenMP [2.6, Canonical Loop Form]
- // Var is one of the following:
- // A variable of signed or unsigned integer type.
- // For C++, a variable of a random access iterator type.
- // For C, a variable of a pointer type.
- QualType VarType = LCDecl->getType().getNonReferenceType();
- if (!VarType->isDependentType() && !VarType->isIntegerType() &&
- !VarType->isPointerType() &&
- !(SemaRef.getLangOpts().CPlusPlus && VarType->isOverloadableType())) {
- SemaRef.Diag(Init->getBeginLoc(), diag::err_omp_loop_variable_type)
- << SemaRef.getLangOpts().CPlusPlus;
- HasErrors = true;
- }
- // OpenMP, 2.14.1.1 Data-sharing Attribute Rules for Variables Referenced in
- // a Construct
- // The loop iteration variable(s) in the associated for-loop(s) of a for or
- // parallel for construct is (are) private.
- // The loop iteration variable in the associated for-loop of a simd
- // construct with just one associated for-loop is linear with a
- // constant-linear-step that is the increment of the associated for-loop.
- // Exclude loop var from the list of variables with implicitly defined data
- // sharing attributes.
- VarsWithImplicitDSA.erase(LCDecl);
- // OpenMP [2.14.1.1, Data-sharing Attribute Rules for Variables Referenced
- // in a Construct, C/C++].
- // The loop iteration variable in the associated for-loop of a simd
- // construct with just one associated for-loop may be listed in a linear
- // clause with a constant-linear-step that is the increment of the
- // associated for-loop.
- // The loop iteration variable(s) in the associated for-loop(s) of a for or
- // parallel for construct may be listed in a private or lastprivate clause.
- DSAStackTy::DSAVarData DVar = DSA.getTopDSA(LCDecl, false);
- // If LoopVarRefExpr is nullptr it means the corresponding loop variable is
- // declared in the loop and it is predetermined as a private.
- OpenMPClauseKind PredeterminedCKind =
- isOpenMPSimdDirective(DKind)
- ? ((NestedLoopCount == 1) ? OMPC_linear : OMPC_lastprivate)
- : OMPC_private;
- if (((isOpenMPSimdDirective(DKind) && DVar.CKind != OMPC_unknown &&
- DVar.CKind != PredeterminedCKind) ||
- ((isOpenMPWorksharingDirective(DKind) || DKind == OMPD_taskloop ||
- isOpenMPDistributeDirective(DKind)) &&
- !isOpenMPSimdDirective(DKind) && DVar.CKind != OMPC_unknown &&
- DVar.CKind != OMPC_private && DVar.CKind != OMPC_lastprivate)) &&
- (DVar.CKind != OMPC_private || DVar.RefExpr != nullptr)) {
- SemaRef.Diag(Init->getBeginLoc(), diag::err_omp_loop_var_dsa)
- << getOpenMPClauseName(DVar.CKind) << getOpenMPDirectiveName(DKind)
- << getOpenMPClauseName(PredeterminedCKind);
- if (DVar.RefExpr == nullptr)
- DVar.CKind = PredeterminedCKind;
- reportOriginalDsa(SemaRef, &DSA, LCDecl, DVar, /*IsLoopIterVar=*/true);
- HasErrors = true;
- } else if (LoopDeclRefExpr != nullptr) {
- // Make the loop iteration variable private (for worksharing constructs),
- // linear (for simd directives with the only one associated loop) or
- // lastprivate (for simd directives with several collapsed or ordered
- // loops).
- if (DVar.CKind == OMPC_unknown)
- DVar = DSA.hasDSA(LCDecl, isOpenMPPrivate,
- [](OpenMPDirectiveKind) -> bool { return true; },
- /*FromParent=*/false);
- DSA.addDSA(LCDecl, LoopDeclRefExpr, PredeterminedCKind);
- }
- assert(isOpenMPLoopDirective(DKind) && "DSA for non-loop vars");
- // Check test-expr.
- HasErrors |= ISC.checkAndSetCond(For->getCond());
- // Check incr-expr.
- HasErrors |= ISC.checkAndSetInc(For->getInc());
- }
- if (ISC.dependent() || SemaRef.CurContext->isDependentContext() || HasErrors)
- return HasErrors;
- // Build the loop's iteration space representation.
- ResultIterSpace.PreCond =
- ISC.buildPreCond(DSA.getCurScope(), For->getCond(), Captures);
- ResultIterSpace.NumIterations = ISC.buildNumIterations(
- DSA.getCurScope(),
- (isOpenMPWorksharingDirective(DKind) ||
- isOpenMPTaskLoopDirective(DKind) || isOpenMPDistributeDirective(DKind)),
- Captures);
- ResultIterSpace.CounterVar = ISC.buildCounterVar(Captures, DSA);
- ResultIterSpace.PrivateCounterVar = ISC.buildPrivateCounterVar();
- ResultIterSpace.CounterInit = ISC.buildCounterInit();
- ResultIterSpace.CounterStep = ISC.buildCounterStep();
- ResultIterSpace.InitSrcRange = ISC.getInitSrcRange();
- ResultIterSpace.CondSrcRange = ISC.getConditionSrcRange();
- ResultIterSpace.IncSrcRange = ISC.getIncrementSrcRange();
- ResultIterSpace.Subtract = ISC.shouldSubtractStep();
- HasErrors |= (ResultIterSpace.PreCond == nullptr ||
- ResultIterSpace.NumIterations == nullptr ||
- ResultIterSpace.CounterVar == nullptr ||
- ResultIterSpace.PrivateCounterVar == nullptr ||
- ResultIterSpace.CounterInit == nullptr ||
- ResultIterSpace.CounterStep == nullptr);
- if (!HasErrors && DSA.isOrderedRegion()) {
- if (DSA.getOrderedRegionParam().second->getNumForLoops()) {
- if (CurrentNestedLoopCount <
- DSA.getOrderedRegionParam().second->getLoopNumIterations().size()) {
- DSA.getOrderedRegionParam().second->setLoopNumIterations(
- CurrentNestedLoopCount, ResultIterSpace.NumIterations);
- DSA.getOrderedRegionParam().second->setLoopCounter(
- CurrentNestedLoopCount, ResultIterSpace.CounterVar);
- }
- }
- for (auto &Pair : DSA.getDoacrossDependClauses()) {
- if (CurrentNestedLoopCount >= Pair.first->getNumLoops()) {
- // Erroneous case - clause has some problems.
- continue;
- }
- if (Pair.first->getDependencyKind() == OMPC_DEPEND_sink &&
- Pair.second.size() <= CurrentNestedLoopCount) {
- // Erroneous case - clause has some problems.
- Pair.first->setLoopData(CurrentNestedLoopCount, nullptr);
- continue;
- }
- Expr *CntValue;
- if (Pair.first->getDependencyKind() == OMPC_DEPEND_source)
- CntValue = ISC.buildOrderedLoopData(
- DSA.getCurScope(), ResultIterSpace.CounterVar, Captures,
- Pair.first->getDependencyLoc());
- else
- CntValue = ISC.buildOrderedLoopData(
- DSA.getCurScope(), ResultIterSpace.CounterVar, Captures,
- Pair.first->getDependencyLoc(),
- Pair.second[CurrentNestedLoopCount].first,
- Pair.second[CurrentNestedLoopCount].second);
- Pair.first->setLoopData(CurrentNestedLoopCount, CntValue);
- }
- }
- return HasErrors;
- }
- /// Build 'VarRef = Start.
- static ExprResult
- buildCounterInit(Sema &SemaRef, Scope *S, SourceLocation Loc, ExprResult VarRef,
- ExprResult Start,
- llvm::MapVector<const Expr *, DeclRefExpr *> &Captures) {
- // Build 'VarRef = Start.
- ExprResult NewStart = tryBuildCapture(SemaRef, Start.get(), Captures);
- if (!NewStart.isUsable())
- return ExprError();
- if (!SemaRef.Context.hasSameType(NewStart.get()->getType(),
- VarRef.get()->getType())) {
- NewStart = SemaRef.PerformImplicitConversion(
- NewStart.get(), VarRef.get()->getType(), Sema::AA_Converting,
- /*AllowExplicit=*/true);
- if (!NewStart.isUsable())
- return ExprError();
- }
- ExprResult Init =
- SemaRef.BuildBinOp(S, Loc, BO_Assign, VarRef.get(), NewStart.get());
- return Init;
- }
- /// Build 'VarRef = Start + Iter * Step'.
- static ExprResult buildCounterUpdate(
- Sema &SemaRef, Scope *S, SourceLocation Loc, ExprResult VarRef,
- ExprResult Start, ExprResult Iter, ExprResult Step, bool Subtract,
- llvm::MapVector<const Expr *, DeclRefExpr *> *Captures = nullptr) {
- // Add parentheses (for debugging purposes only).
- Iter = SemaRef.ActOnParenExpr(Loc, Loc, Iter.get());
- if (!VarRef.isUsable() || !Start.isUsable() || !Iter.isUsable() ||
- !Step.isUsable())
- return ExprError();
- ExprResult NewStep = Step;
- if (Captures)
- NewStep = tryBuildCapture(SemaRef, Step.get(), *Captures);
- if (NewStep.isInvalid())
- return ExprError();
- ExprResult Update =
- SemaRef.BuildBinOp(S, Loc, BO_Mul, Iter.get(), NewStep.get());
- if (!Update.isUsable())
- return ExprError();
- // Try to build 'VarRef = Start, VarRef (+|-)= Iter * Step' or
- // 'VarRef = Start (+|-) Iter * Step'.
- ExprResult NewStart = Start;
- if (Captures)
- NewStart = tryBuildCapture(SemaRef, Start.get(), *Captures);
- if (NewStart.isInvalid())
- return ExprError();
- // First attempt: try to build 'VarRef = Start, VarRef += Iter * Step'.
- ExprResult SavedUpdate = Update;
- ExprResult UpdateVal;
- if (VarRef.get()->getType()->isOverloadableType() ||
- NewStart.get()->getType()->isOverloadableType() ||
- Update.get()->getType()->isOverloadableType()) {
- bool Suppress = SemaRef.getDiagnostics().getSuppressAllDiagnostics();
- SemaRef.getDiagnostics().setSuppressAllDiagnostics(/*Val=*/true);
- Update =
- SemaRef.BuildBinOp(S, Loc, BO_Assign, VarRef.get(), NewStart.get());
- if (Update.isUsable()) {
- UpdateVal =
- SemaRef.BuildBinOp(S, Loc, Subtract ? BO_SubAssign : BO_AddAssign,
- VarRef.get(), SavedUpdate.get());
- if (UpdateVal.isUsable()) {
- Update = SemaRef.CreateBuiltinBinOp(Loc, BO_Comma, Update.get(),
- UpdateVal.get());
- }
- }
- SemaRef.getDiagnostics().setSuppressAllDiagnostics(Suppress);
- }
- // Second attempt: try to build 'VarRef = Start (+|-) Iter * Step'.
- if (!Update.isUsable() || !UpdateVal.isUsable()) {
- Update = SemaRef.BuildBinOp(S, Loc, Subtract ? BO_Sub : BO_Add,
- NewStart.get(), SavedUpdate.get());
- if (!Update.isUsable())
- return ExprError();
- if (!SemaRef.Context.hasSameType(Update.get()->getType(),
- VarRef.get()->getType())) {
- Update = SemaRef.PerformImplicitConversion(
- Update.get(), VarRef.get()->getType(), Sema::AA_Converting, true);
- if (!Update.isUsable())
- return ExprError();
- }
- Update = SemaRef.BuildBinOp(S, Loc, BO_Assign, VarRef.get(), Update.get());
- }
- return Update;
- }
- /// Convert integer expression \a E to make it have at least \a Bits
- /// bits.
- static ExprResult widenIterationCount(unsigned Bits, Expr *E, Sema &SemaRef) {
- if (E == nullptr)
- return ExprError();
- ASTContext &C = SemaRef.Context;
- QualType OldType = E->getType();
- unsigned HasBits = C.getTypeSize(OldType);
- if (HasBits >= Bits)
- return ExprResult(E);
- // OK to convert to signed, because new type has more bits than old.
- QualType NewType = C.getIntTypeForBitwidth(Bits, /* Signed */ true);
- return SemaRef.PerformImplicitConversion(E, NewType, Sema::AA_Converting,
- true);
- }
- /// Check if the given expression \a E is a constant integer that fits
- /// into \a Bits bits.
- static bool fitsInto(unsigned Bits, bool Signed, const Expr *E, Sema &SemaRef) {
- if (E == nullptr)
- return false;
- llvm::APSInt Result;
- if (E->isIntegerConstantExpr(Result, SemaRef.Context))
- return Signed ? Result.isSignedIntN(Bits) : Result.isIntN(Bits);
- return false;
- }
- /// Build preinits statement for the given declarations.
- static Stmt *buildPreInits(ASTContext &Context,
- MutableArrayRef<Decl *> PreInits) {
- if (!PreInits.empty()) {
- return new (Context) DeclStmt(
- DeclGroupRef::Create(Context, PreInits.begin(), PreInits.size()),
- SourceLocation(), SourceLocation());
- }
- return nullptr;
- }
- /// Build preinits statement for the given declarations.
- static Stmt *
- buildPreInits(ASTContext &Context,
- const llvm::MapVector<const Expr *, DeclRefExpr *> &Captures) {
- if (!Captures.empty()) {
- SmallVector<Decl *, 16> PreInits;
- for (const auto &Pair : Captures)
- PreInits.push_back(Pair.second->getDecl());
- return buildPreInits(Context, PreInits);
- }
- return nullptr;
- }
- /// Build postupdate expression for the given list of postupdates expressions.
- static Expr *buildPostUpdate(Sema &S, ArrayRef<Expr *> PostUpdates) {
- Expr *PostUpdate = nullptr;
- if (!PostUpdates.empty()) {
- for (Expr *E : PostUpdates) {
- Expr *ConvE = S.BuildCStyleCastExpr(
- E->getExprLoc(),
- S.Context.getTrivialTypeSourceInfo(S.Context.VoidTy),
- E->getExprLoc(), E)
- .get();
- PostUpdate = PostUpdate
- ? S.CreateBuiltinBinOp(ConvE->getExprLoc(), BO_Comma,
- PostUpdate, ConvE)
- .get()
- : ConvE;
- }
- }
- return PostUpdate;
- }
- /// Called on a for stmt to check itself and nested loops (if any).
- /// \return Returns 0 if one of the collapsed stmts is not canonical for loop,
- /// number of collapsed loops otherwise.
- static unsigned
- checkOpenMPLoop(OpenMPDirectiveKind DKind, Expr *CollapseLoopCountExpr,
- Expr *OrderedLoopCountExpr, Stmt *AStmt, Sema &SemaRef,
- DSAStackTy &DSA,
- Sema::VarsWithInheritedDSAType &VarsWithImplicitDSA,
- OMPLoopDirective::HelperExprs &Built) {
- unsigned NestedLoopCount = 1;
- if (CollapseLoopCountExpr) {
- // Found 'collapse' clause - calculate collapse number.
- Expr::EvalResult Result;
- if (CollapseLoopCountExpr->EvaluateAsInt(Result, SemaRef.getASTContext()))
- NestedLoopCount = Result.Val.getInt().getLimitedValue();
- }
- unsigned OrderedLoopCount = 1;
- if (OrderedLoopCountExpr) {
- // Found 'ordered' clause - calculate collapse number.
- Expr::EvalResult EVResult;
- if (OrderedLoopCountExpr->EvaluateAsInt(EVResult, SemaRef.getASTContext())) {
- llvm::APSInt Result = EVResult.Val.getInt();
- if (Result.getLimitedValue() < NestedLoopCount) {
- SemaRef.Diag(OrderedLoopCountExpr->getExprLoc(),
- diag::err_omp_wrong_ordered_loop_count)
- << OrderedLoopCountExpr->getSourceRange();
- SemaRef.Diag(CollapseLoopCountExpr->getExprLoc(),
- diag::note_collapse_loop_count)
- << CollapseLoopCountExpr->getSourceRange();
- }
- OrderedLoopCount = Result.getLimitedValue();
- }
- }
- // This is helper routine for loop directives (e.g., 'for', 'simd',
- // 'for simd', etc.).
- llvm::MapVector<const Expr *, DeclRefExpr *> Captures;
- SmallVector<LoopIterationSpace, 4> IterSpaces;
- IterSpaces.resize(std::max(OrderedLoopCount, NestedLoopCount));
- Stmt *CurStmt = AStmt->IgnoreContainers(/* IgnoreCaptured */ true);
- for (unsigned Cnt = 0; Cnt < NestedLoopCount; ++Cnt) {
- if (checkOpenMPIterationSpace(
- DKind, CurStmt, SemaRef, DSA, Cnt, NestedLoopCount,
- std::max(OrderedLoopCount, NestedLoopCount), CollapseLoopCountExpr,
- OrderedLoopCountExpr, VarsWithImplicitDSA, IterSpaces[Cnt],
- Captures))
- return 0;
- // Move on to the next nested for loop, or to the loop body.
- // OpenMP [2.8.1, simd construct, Restrictions]
- // All loops associated with the construct must be perfectly nested; that
- // is, there must be no intervening code nor any OpenMP directive between
- // any two loops.
- CurStmt = cast<ForStmt>(CurStmt)->getBody()->IgnoreContainers();
- }
- for (unsigned Cnt = NestedLoopCount; Cnt < OrderedLoopCount; ++Cnt) {
- if (checkOpenMPIterationSpace(
- DKind, CurStmt, SemaRef, DSA, Cnt, NestedLoopCount,
- std::max(OrderedLoopCount, NestedLoopCount), CollapseLoopCountExpr,
- OrderedLoopCountExpr, VarsWithImplicitDSA, IterSpaces[Cnt],
- Captures))
- return 0;
- if (Cnt > 0 && IterSpaces[Cnt].CounterVar) {
- // Handle initialization of captured loop iterator variables.
- auto *DRE = cast<DeclRefExpr>(IterSpaces[Cnt].CounterVar);
- if (isa<OMPCapturedExprDecl>(DRE->getDecl())) {
- Captures[DRE] = DRE;
- }
- }
- // Move on to the next nested for loop, or to the loop body.
- // OpenMP [2.8.1, simd construct, Restrictions]
- // All loops associated with the construct must be perfectly nested; that
- // is, there must be no intervening code nor any OpenMP directive between
- // any two loops.
- CurStmt = cast<ForStmt>(CurStmt)->getBody()->IgnoreContainers();
- }
- Built.clear(/* size */ NestedLoopCount);
- if (SemaRef.CurContext->isDependentContext())
- return NestedLoopCount;
- // An example of what is generated for the following code:
- //
- // #pragma omp simd collapse(2) ordered(2)
- // for (i = 0; i < NI; ++i)
- // for (k = 0; k < NK; ++k)
- // for (j = J0; j < NJ; j+=2) {
- // <loop body>
- // }
- //
- // We generate the code below.
- // Note: the loop body may be outlined in CodeGen.
- // Note: some counters may be C++ classes, operator- is used to find number of
- // iterations and operator+= to calculate counter value.
- // Note: decltype(NumIterations) must be integer type (in 'omp for', only i32
- // or i64 is currently supported).
- //
- // #define NumIterations (NI * ((NJ - J0 - 1 + 2) / 2))
- // for (int[32|64]_t IV = 0; IV < NumIterations; ++IV ) {
- // .local.i = IV / ((NJ - J0 - 1 + 2) / 2);
- // .local.j = J0 + (IV % ((NJ - J0 - 1 + 2) / 2)) * 2;
- // // similar updates for vars in clauses (e.g. 'linear')
- // <loop body (using local i and j)>
- // }
- // i = NI; // assign final values of counters
- // j = NJ;
- //
- // Last iteration number is (I1 * I2 * ... In) - 1, where I1, I2 ... In are
- // the iteration counts of the collapsed for loops.
- // Precondition tests if there is at least one iteration (all conditions are
- // true).
- auto PreCond = ExprResult(IterSpaces[0].PreCond);
- Expr *N0 = IterSpaces[0].NumIterations;
- ExprResult LastIteration32 =
- widenIterationCount(/*Bits=*/32,
- SemaRef
- .PerformImplicitConversion(
- N0->IgnoreImpCasts(), N0->getType(),
- Sema::AA_Converting, /*AllowExplicit=*/true)
- .get(),
- SemaRef);
- ExprResult LastIteration64 = widenIterationCount(
- /*Bits=*/64,
- SemaRef
- .PerformImplicitConversion(N0->IgnoreImpCasts(), N0->getType(),
- Sema::AA_Converting,
- /*AllowExplicit=*/true)
- .get(),
- SemaRef);
- if (!LastIteration32.isUsable() || !LastIteration64.isUsable())
- return NestedLoopCount;
- ASTContext &C = SemaRef.Context;
- bool AllCountsNeedLessThan32Bits = C.getTypeSize(N0->getType()) < 32;
- Scope *CurScope = DSA.getCurScope();
- for (unsigned Cnt = 1; Cnt < NestedLoopCount; ++Cnt) {
- if (PreCond.isUsable()) {
- PreCond =
- SemaRef.BuildBinOp(CurScope, PreCond.get()->getExprLoc(), BO_LAnd,
- PreCond.get(), IterSpaces[Cnt].PreCond);
- }
- Expr *N = IterSpaces[Cnt].NumIterations;
- SourceLocation Loc = N->getExprLoc();
- AllCountsNeedLessThan32Bits &= C.getTypeSize(N->getType()) < 32;
- if (LastIteration32.isUsable())
- LastIteration32 = SemaRef.BuildBinOp(
- CurScope, Loc, BO_Mul, LastIteration32.get(),
- SemaRef
- .PerformImplicitConversion(N->IgnoreImpCasts(), N->getType(),
- Sema::AA_Converting,
- /*AllowExplicit=*/true)
- .get());
- if (LastIteration64.isUsable())
- LastIteration64 = SemaRef.BuildBinOp(
- CurScope, Loc, BO_Mul, LastIteration64.get(),
- SemaRef
- .PerformImplicitConversion(N->IgnoreImpCasts(), N->getType(),
- Sema::AA_Converting,
- /*AllowExplicit=*/true)
- .get());
- }
- // Choose either the 32-bit or 64-bit version.
- ExprResult LastIteration = LastIteration64;
- if (LastIteration32.isUsable() &&
- C.getTypeSize(LastIteration32.get()->getType()) == 32 &&
- (AllCountsNeedLessThan32Bits || NestedLoopCount == 1 ||
- fitsInto(
- /*Bits=*/32,
- LastIteration32.get()->getType()->hasSignedIntegerRepresentation(),
- LastIteration64.get(), SemaRef)))
- LastIteration = LastIteration32;
- QualType VType = LastIteration.get()->getType();
- QualType RealVType = VType;
- QualType StrideVType = VType;
- if (isOpenMPTaskLoopDirective(DKind)) {
- VType =
- SemaRef.Context.getIntTypeForBitwidth(/*DestWidth=*/64, /*Signed=*/0);
- StrideVType =
- SemaRef.Context.getIntTypeForBitwidth(/*DestWidth=*/64, /*Signed=*/1);
- }
- if (!LastIteration.isUsable())
- return 0;
- // Save the number of iterations.
- ExprResult NumIterations = LastIteration;
- {
- LastIteration = SemaRef.BuildBinOp(
- CurScope, LastIteration.get()->getExprLoc(), BO_Sub,
- LastIteration.get(),
- SemaRef.ActOnIntegerConstant(SourceLocation(), 1).get());
- if (!LastIteration.isUsable())
- return 0;
- }
- // Calculate the last iteration number beforehand instead of doing this on
- // each iteration. Do not do this if the number of iterations may be kfold-ed.
- llvm::APSInt Result;
- bool IsConstant =
- LastIteration.get()->isIntegerConstantExpr(Result, SemaRef.Context);
- ExprResult CalcLastIteration;
- if (!IsConstant) {
- ExprResult SaveRef =
- tryBuildCapture(SemaRef, LastIteration.get(), Captures);
- LastIteration = SaveRef;
- // Prepare SaveRef + 1.
- NumIterations = SemaRef.BuildBinOp(
- CurScope, SaveRef.get()->getExprLoc(), BO_Add, SaveRef.get(),
- SemaRef.ActOnIntegerConstant(SourceLocation(), 1).get());
- if (!NumIterations.isUsable())
- return 0;
- }
- SourceLocation InitLoc = IterSpaces[0].InitSrcRange.getBegin();
- // Build variables passed into runtime, necessary for worksharing directives.
- ExprResult LB, UB, IL, ST, EUB, CombLB, CombUB, PrevLB, PrevUB, CombEUB;
- if (isOpenMPWorksharingDirective(DKind) || isOpenMPTaskLoopDirective(DKind) ||
- isOpenMPDistributeDirective(DKind)) {
- // Lower bound variable, initialized with zero.
- VarDecl *LBDecl = buildVarDecl(SemaRef, InitLoc, VType, ".omp.lb");
- LB = buildDeclRefExpr(SemaRef, LBDecl, VType, InitLoc);
- SemaRef.AddInitializerToDecl(LBDecl,
- SemaRef.ActOnIntegerConstant(InitLoc, 0).get(),
- /*DirectInit*/ false);
- // Upper bound variable, initialized with last iteration number.
- VarDecl *UBDecl = buildVarDecl(SemaRef, InitLoc, VType, ".omp.ub");
- UB = buildDeclRefExpr(SemaRef, UBDecl, VType, InitLoc);
- SemaRef.AddInitializerToDecl(UBDecl, LastIteration.get(),
- /*DirectInit*/ false);
- // A 32-bit variable-flag where runtime returns 1 for the last iteration.
- // This will be used to implement clause 'lastprivate'.
- QualType Int32Ty = SemaRef.Context.getIntTypeForBitwidth(32, true);
- VarDecl *ILDecl = buildVarDecl(SemaRef, InitLoc, Int32Ty, ".omp.is_last");
- IL = buildDeclRefExpr(SemaRef, ILDecl, Int32Ty, InitLoc);
- SemaRef.AddInitializerToDecl(ILDecl,
- SemaRef.ActOnIntegerConstant(InitLoc, 0).get(),
- /*DirectInit*/ false);
- // Stride variable returned by runtime (we initialize it to 1 by default).
- VarDecl *STDecl =
- buildVarDecl(SemaRef, InitLoc, StrideVType, ".omp.stride");
- ST = buildDeclRefExpr(SemaRef, STDecl, StrideVType, InitLoc);
- SemaRef.AddInitializerToDecl(STDecl,
- SemaRef.ActOnIntegerConstant(InitLoc, 1).get(),
- /*DirectInit*/ false);
- // Build expression: UB = min(UB, LastIteration)
- // It is necessary for CodeGen of directives with static scheduling.
- ExprResult IsUBGreater = SemaRef.BuildBinOp(CurScope, InitLoc, BO_GT,
- UB.get(), LastIteration.get());
- ExprResult CondOp = SemaRef.ActOnConditionalOp(
- LastIteration.get()->getExprLoc(), InitLoc, IsUBGreater.get(),
- LastIteration.get(), UB.get());
- EUB = SemaRef.BuildBinOp(CurScope, InitLoc, BO_Assign, UB.get(),
- CondOp.get());
- EUB = SemaRef.ActOnFinishFullExpr(EUB.get());
- // If we have a combined directive that combines 'distribute', 'for' or
- // 'simd' we need to be able to access the bounds of the schedule of the
- // enclosing region. E.g. in 'distribute parallel for' the bounds obtained
- // by scheduling 'distribute' have to be passed to the schedule of 'for'.
- if (isOpenMPLoopBoundSharingDirective(DKind)) {
- // Lower bound variable, initialized with zero.
- VarDecl *CombLBDecl =
- buildVarDecl(SemaRef, InitLoc, VType, ".omp.comb.lb");
- CombLB = buildDeclRefExpr(SemaRef, CombLBDecl, VType, InitLoc);
- SemaRef.AddInitializerToDecl(
- CombLBDecl, SemaRef.ActOnIntegerConstant(InitLoc, 0).get(),
- /*DirectInit*/ false);
- // Upper bound variable, initialized with last iteration number.
- VarDecl *CombUBDecl =
- buildVarDecl(SemaRef, InitLoc, VType, ".omp.comb.ub");
- CombUB = buildDeclRefExpr(SemaRef, CombUBDecl, VType, InitLoc);
- SemaRef.AddInitializerToDecl(CombUBDecl, LastIteration.get(),
- /*DirectInit*/ false);
- ExprResult CombIsUBGreater = SemaRef.BuildBinOp(
- CurScope, InitLoc, BO_GT, CombUB.get(), LastIteration.get());
- ExprResult CombCondOp =
- SemaRef.ActOnConditionalOp(InitLoc, InitLoc, CombIsUBGreater.get(),
- LastIteration.get(), CombUB.get());
- CombEUB = SemaRef.BuildBinOp(CurScope, InitLoc, BO_Assign, CombUB.get(),
- CombCondOp.get());
- CombEUB = SemaRef.ActOnFinishFullExpr(CombEUB.get());
- const CapturedDecl *CD = cast<CapturedStmt>(AStmt)->getCapturedDecl();
- // We expect to have at least 2 more parameters than the 'parallel'
- // directive does - the lower and upper bounds of the previous schedule.
- assert(CD->getNumParams() >= 4 &&
- "Unexpected number of parameters in loop combined directive");
- // Set the proper type for the bounds given what we learned from the
- // enclosed loops.
- ImplicitParamDecl *PrevLBDecl = CD->getParam(/*PrevLB=*/2);
- ImplicitParamDecl *PrevUBDecl = CD->getParam(/*PrevUB=*/3);
- // Previous lower and upper bounds are obtained from the region
- // parameters.
- PrevLB =
- buildDeclRefExpr(SemaRef, PrevLBDecl, PrevLBDecl->getType(), InitLoc);
- PrevUB =
- buildDeclRefExpr(SemaRef, PrevUBDecl, PrevUBDecl->getType(), InitLoc);
- }
- }
- // Build the iteration variable and its initialization before loop.
- ExprResult IV;
- ExprResult Init, CombInit;
- {
- VarDecl *IVDecl = buildVarDecl(SemaRef, InitLoc, RealVType, ".omp.iv");
- IV = buildDeclRefExpr(SemaRef, IVDecl, RealVType, InitLoc);
- Expr *RHS =
- (isOpenMPWorksharingDirective(DKind) ||
- isOpenMPTaskLoopDirective(DKind) || isOpenMPDistributeDirective(DKind))
- ? LB.get()
- : SemaRef.ActOnIntegerConstant(SourceLocation(), 0).get();
- Init = SemaRef.BuildBinOp(CurScope, InitLoc, BO_Assign, IV.get(), RHS);
- Init = SemaRef.ActOnFinishFullExpr(Init.get());
- if (isOpenMPLoopBoundSharingDirective(DKind)) {
- Expr *CombRHS =
- (isOpenMPWorksharingDirective(DKind) ||
- isOpenMPTaskLoopDirective(DKind) ||
- isOpenMPDistributeDirective(DKind))
- ? CombLB.get()
- : SemaRef.ActOnIntegerConstant(SourceLocation(), 0).get();
- CombInit =
- SemaRef.BuildBinOp(CurScope, InitLoc, BO_Assign, IV.get(), CombRHS);
- CombInit = SemaRef.ActOnFinishFullExpr(CombInit.get());
- }
- }
- // Loop condition (IV < NumIterations) or (IV <= UB) for worksharing loops.
- SourceLocation CondLoc = AStmt->getBeginLoc();
- ExprResult Cond =
- (isOpenMPWorksharingDirective(DKind) ||
- isOpenMPTaskLoopDirective(DKind) || isOpenMPDistributeDirective(DKind))
- ? SemaRef.BuildBinOp(CurScope, CondLoc, BO_LE, IV.get(), UB.get())
- : SemaRef.BuildBinOp(CurScope, CondLoc, BO_LT, IV.get(),
- NumIterations.get());
- ExprResult CombDistCond;
- if (isOpenMPLoopBoundSharingDirective(DKind)) {
- CombDistCond =
- SemaRef.BuildBinOp(
- CurScope, CondLoc, BO_LT, IV.get(), NumIterations.get());
- }
- ExprResult CombCond;
- if (isOpenMPLoopBoundSharingDirective(DKind)) {
- CombCond =
- SemaRef.BuildBinOp(CurScope, CondLoc, BO_LE, IV.get(), CombUB.get());
- }
- // Loop increment (IV = IV + 1)
- SourceLocation IncLoc = AStmt->getBeginLoc();
- ExprResult Inc =
- SemaRef.BuildBinOp(CurScope, IncLoc, BO_Add, IV.get(),
- SemaRef.ActOnIntegerConstant(IncLoc, 1).get());
- if (!Inc.isUsable())
- return 0;
- Inc = SemaRef.BuildBinOp(CurScope, IncLoc, BO_Assign, IV.get(), Inc.get());
- Inc = SemaRef.ActOnFinishFullExpr(Inc.get());
- if (!Inc.isUsable())
- return 0;
- // Increments for worksharing loops (LB = LB + ST; UB = UB + ST).
- // Used for directives with static scheduling.
- // In combined construct, add combined version that use CombLB and CombUB
- // base variables for the update
- ExprResult NextLB, NextUB, CombNextLB, CombNextUB;
- if (isOpenMPWorksharingDirective(DKind) || isOpenMPTaskLoopDirective(DKind) ||
- isOpenMPDistributeDirective(DKind)) {
- // LB + ST
- NextLB = SemaRef.BuildBinOp(CurScope, IncLoc, BO_Add, LB.get(), ST.get());
- if (!NextLB.isUsable())
- return 0;
- // LB = LB + ST
- NextLB =
- SemaRef.BuildBinOp(CurScope, IncLoc, BO_Assign, LB.get(), NextLB.get());
- NextLB = SemaRef.ActOnFinishFullExpr(NextLB.get());
- if (!NextLB.isUsable())
- return 0;
- // UB + ST
- NextUB = SemaRef.BuildBinOp(CurScope, IncLoc, BO_Add, UB.get(), ST.get());
- if (!NextUB.isUsable())
- return 0;
- // UB = UB + ST
- NextUB =
- SemaRef.BuildBinOp(CurScope, IncLoc, BO_Assign, UB.get(), NextUB.get());
- NextUB = SemaRef.ActOnFinishFullExpr(NextUB.get());
- if (!NextUB.isUsable())
- return 0;
- if (isOpenMPLoopBoundSharingDirective(DKind)) {
- CombNextLB =
- SemaRef.BuildBinOp(CurScope, IncLoc, BO_Add, CombLB.get(), ST.get());
- if (!NextLB.isUsable())
- return 0;
- // LB = LB + ST
- CombNextLB = SemaRef.BuildBinOp(CurScope, IncLoc, BO_Assign, CombLB.get(),
- CombNextLB.get());
- CombNextLB = SemaRef.ActOnFinishFullExpr(CombNextLB.get());
- if (!CombNextLB.isUsable())
- return 0;
- // UB + ST
- CombNextUB =
- SemaRef.BuildBinOp(CurScope, IncLoc, BO_Add, CombUB.get(), ST.get());
- if (!CombNextUB.isUsable())
- return 0;
- // UB = UB + ST
- CombNextUB = SemaRef.BuildBinOp(CurScope, IncLoc, BO_Assign, CombUB.get(),
- CombNextUB.get());
- CombNextUB = SemaRef.ActOnFinishFullExpr(CombNextUB.get());
- if (!CombNextUB.isUsable())
- return 0;
- }
- }
- // Create increment expression for distribute loop when combined in a same
- // directive with for as IV = IV + ST; ensure upper bound expression based
- // on PrevUB instead of NumIterations - used to implement 'for' when found
- // in combination with 'distribute', like in 'distribute parallel for'
- SourceLocation DistIncLoc = AStmt->getBeginLoc();
- ExprResult DistCond, DistInc, PrevEUB, ParForInDistCond;
- if (isOpenMPLoopBoundSharingDirective(DKind)) {
- DistCond = SemaRef.BuildBinOp(CurScope, CondLoc, BO_LE, IV.get(), UB.get());
- assert(DistCond.isUsable() && "distribute cond expr was not built");
- DistInc =
- SemaRef.BuildBinOp(CurScope, DistIncLoc, BO_Add, IV.get(), ST.get());
- assert(DistInc.isUsable() && "distribute inc expr was not built");
- DistInc = SemaRef.BuildBinOp(CurScope, DistIncLoc, BO_Assign, IV.get(),
- DistInc.get());
- DistInc = SemaRef.ActOnFinishFullExpr(DistInc.get());
- assert(DistInc.isUsable() && "distribute inc expr was not built");
- // Build expression: UB = min(UB, prevUB) for #for in composite or combined
- // construct
- SourceLocation DistEUBLoc = AStmt->getBeginLoc();
- ExprResult IsUBGreater =
- SemaRef.BuildBinOp(CurScope, DistEUBLoc, BO_GT, UB.get(), PrevUB.get());
- ExprResult CondOp = SemaRef.ActOnConditionalOp(
- DistEUBLoc, DistEUBLoc, IsUBGreater.get(), PrevUB.get(), UB.get());
- PrevEUB = SemaRef.BuildBinOp(CurScope, DistIncLoc, BO_Assign, UB.get(),
- CondOp.get());
- PrevEUB = SemaRef.ActOnFinishFullExpr(PrevEUB.get());
- // Build IV <= PrevUB to be used in parallel for is in combination with
- // a distribute directive with schedule(static, 1)
- ParForInDistCond =
- SemaRef.BuildBinOp(CurScope, CondLoc, BO_LE, IV.get(), PrevUB.get());
- }
- // Build updates and final values of the loop counters.
- bool HasErrors = false;
- Built.Counters.resize(NestedLoopCount);
- Built.Inits.resize(NestedLoopCount);
- Built.Updates.resize(NestedLoopCount);
- Built.Finals.resize(NestedLoopCount);
- {
- ExprResult Div;
- // Go from inner nested loop to outer.
- for (int Cnt = NestedLoopCount - 1; Cnt >= 0; --Cnt) {
- LoopIterationSpace &IS = IterSpaces[Cnt];
- SourceLocation UpdLoc = IS.IncSrcRange.getBegin();
- // Build: Iter = (IV / Div) % IS.NumIters
- // where Div is product of previous iterations' IS.NumIters.
- ExprResult Iter;
- if (Div.isUsable()) {
- Iter =
- SemaRef.BuildBinOp(CurScope, UpdLoc, BO_Div, IV.get(), Div.get());
- } else {
- Iter = IV;
- assert((Cnt == (int)NestedLoopCount - 1) &&
- "unusable div expected on first iteration only");
- }
- if (Cnt != 0 && Iter.isUsable())
- Iter = SemaRef.BuildBinOp(CurScope, UpdLoc, BO_Rem, Iter.get(),
- IS.NumIterations);
- if (!Iter.isUsable()) {
- HasErrors = true;
- break;
- }
- // Build update: IS.CounterVar(Private) = IS.Start + Iter * IS.Step
- auto *VD = cast<VarDecl>(cast<DeclRefExpr>(IS.CounterVar)->getDecl());
- DeclRefExpr *CounterVar = buildDeclRefExpr(
- SemaRef, VD, IS.CounterVar->getType(), IS.CounterVar->getExprLoc(),
- /*RefersToCapture=*/true);
- ExprResult Init = buildCounterInit(SemaRef, CurScope, UpdLoc, CounterVar,
- IS.CounterInit, Captures);
- if (!Init.isUsable()) {
- HasErrors = true;
- break;
- }
- ExprResult Update = buildCounterUpdate(
- SemaRef, CurScope, UpdLoc, CounterVar, IS.CounterInit, Iter,
- IS.CounterStep, IS.Subtract, &Captures);
- if (!Update.isUsable()) {
- HasErrors = true;
- break;
- }
- // Build final: IS.CounterVar = IS.Start + IS.NumIters * IS.Step
- ExprResult Final = buildCounterUpdate(
- SemaRef, CurScope, UpdLoc, CounterVar, IS.CounterInit,
- IS.NumIterations, IS.CounterStep, IS.Subtract, &Captures);
- if (!Final.isUsable()) {
- HasErrors = true;
- break;
- }
- // Build Div for the next iteration: Div <- Div * IS.NumIters
- if (Cnt != 0) {
- if (Div.isUnset())
- Div = IS.NumIterations;
- else
- Div = SemaRef.BuildBinOp(CurScope, UpdLoc, BO_Mul, Div.get(),
- IS.NumIterations);
- // Add parentheses (for debugging purposes only).
- if (Div.isUsable())
- Div = tryBuildCapture(SemaRef, Div.get(), Captures);
- if (!Div.isUsable()) {
- HasErrors = true;
- break;
- }
- }
- if (!Update.isUsable() || !Final.isUsable()) {
- HasErrors = true;
- break;
- }
- // Save results
- Built.Counters[Cnt] = IS.CounterVar;
- Built.PrivateCounters[Cnt] = IS.PrivateCounterVar;
- Built.Inits[Cnt] = Init.get();
- Built.Updates[Cnt] = Update.get();
- Built.Finals[Cnt] = Final.get();
- }
- }
- if (HasErrors)
- return 0;
- // Save results
- Built.IterationVarRef = IV.get();
- Built.LastIteration = LastIteration.get();
- Built.NumIterations = NumIterations.get();
- Built.CalcLastIteration =
- SemaRef.ActOnFinishFullExpr(CalcLastIteration.get()).get();
- Built.PreCond = PreCond.get();
- Built.PreInits = buildPreInits(C, Captures);
- Built.Cond = Cond.get();
- Built.Init = Init.get();
- Built.Inc = Inc.get();
- Built.LB = LB.get();
- Built.UB = UB.get();
- Built.IL = IL.get();
- Built.ST = ST.get();
- Built.EUB = EUB.get();
- Built.NLB = NextLB.get();
- Built.NUB = NextUB.get();
- Built.PrevLB = PrevLB.get();
- Built.PrevUB = PrevUB.get();
- Built.DistInc = DistInc.get();
- Built.PrevEUB = PrevEUB.get();
- Built.DistCombinedFields.LB = CombLB.get();
- Built.DistCombinedFields.UB = CombUB.get();
- Built.DistCombinedFields.EUB = CombEUB.get();
- Built.DistCombinedFields.Init = CombInit.get();
- Built.DistCombinedFields.Cond = CombCond.get();
- Built.DistCombinedFields.NLB = CombNextLB.get();
- Built.DistCombinedFields.NUB = CombNextUB.get();
- Built.DistCombinedFields.DistCond = CombDistCond.get();
- Built.DistCombinedFields.ParForInDistCond = ParForInDistCond.get();
- return NestedLoopCount;
- }
- static Expr *getCollapseNumberExpr(ArrayRef<OMPClause *> Clauses) {
- auto CollapseClauses =
- OMPExecutableDirective::getClausesOfKind<OMPCollapseClause>(Clauses);
- if (CollapseClauses.begin() != CollapseClauses.end())
- return (*CollapseClauses.begin())->getNumForLoops();
- return nullptr;
- }
- static Expr *getOrderedNumberExpr(ArrayRef<OMPClause *> Clauses) {
- auto OrderedClauses =
- OMPExecutableDirective::getClausesOfKind<OMPOrderedClause>(Clauses);
- if (OrderedClauses.begin() != OrderedClauses.end())
- return (*OrderedClauses.begin())->getNumForLoops();
- return nullptr;
- }
- static bool checkSimdlenSafelenSpecified(Sema &S,
- const ArrayRef<OMPClause *> Clauses) {
- const OMPSafelenClause *Safelen = nullptr;
- const OMPSimdlenClause *Simdlen = nullptr;
- for (const OMPClause *Clause : Clauses) {
- if (Clause->getClauseKind() == OMPC_safelen)
- Safelen = cast<OMPSafelenClause>(Clause);
- else if (Clause->getClauseKind() == OMPC_simdlen)
- Simdlen = cast<OMPSimdlenClause>(Clause);
- if (Safelen && Simdlen)
- break;
- }
- if (Simdlen && Safelen) {
- const Expr *SimdlenLength = Simdlen->getSimdlen();
- const Expr *SafelenLength = Safelen->getSafelen();
- if (SimdlenLength->isValueDependent() || SimdlenLength->isTypeDependent() ||
- SimdlenLength->isInstantiationDependent() ||
- SimdlenLength->containsUnexpandedParameterPack())
- return false;
- if (SafelenLength->isValueDependent() || SafelenLength->isTypeDependent() ||
- SafelenLength->isInstantiationDependent() ||
- SafelenLength->containsUnexpandedParameterPack())
- return false;
- Expr::EvalResult SimdlenResult, SafelenResult;
- SimdlenLength->EvaluateAsInt(SimdlenResult, S.Context);
- SafelenLength->EvaluateAsInt(SafelenResult, S.Context);
- llvm::APSInt SimdlenRes = SimdlenResult.Val.getInt();
- llvm::APSInt SafelenRes = SafelenResult.Val.getInt();
- // OpenMP 4.5 [2.8.1, simd Construct, Restrictions]
- // If both simdlen and safelen clauses are specified, the value of the
- // simdlen parameter must be less than or equal to the value of the safelen
- // parameter.
- if (SimdlenRes > SafelenRes) {
- S.Diag(SimdlenLength->getExprLoc(),
- diag::err_omp_wrong_simdlen_safelen_values)
- << SimdlenLength->getSourceRange() << SafelenLength->getSourceRange();
- return true;
- }
- }
- return false;
- }
- StmtResult
- Sema::ActOnOpenMPSimdDirective(ArrayRef<OMPClause *> Clauses, Stmt *AStmt,
- SourceLocation StartLoc, SourceLocation EndLoc,
- VarsWithInheritedDSAType &VarsWithImplicitDSA) {
- if (!AStmt)
- return StmtError();
- assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
- OMPLoopDirective::HelperExprs B;
- // In presence of clause 'collapse' or 'ordered' with number of loops, it will
- // define the nested loops number.
- unsigned NestedLoopCount = checkOpenMPLoop(
- OMPD_simd, getCollapseNumberExpr(Clauses), getOrderedNumberExpr(Clauses),
- AStmt, *this, *DSAStack, VarsWithImplicitDSA, B);
- if (NestedLoopCount == 0)
- return StmtError();
- assert((CurContext->isDependentContext() || B.builtAll()) &&
- "omp simd loop exprs were not built");
- if (!CurContext->isDependentContext()) {
- // Finalize the clauses that need pre-built expressions for CodeGen.
- for (OMPClause *C : Clauses) {
- if (auto *LC = dyn_cast<OMPLinearClause>(C))
- if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef),
- B.NumIterations, *this, CurScope,
- DSAStack))
- return StmtError();
- }
- }
- if (checkSimdlenSafelenSpecified(*this, Clauses))
- return StmtError();
- setFunctionHasBranchProtectedScope();
- return OMPSimdDirective::Create(Context, StartLoc, EndLoc, NestedLoopCount,
- Clauses, AStmt, B);
- }
- StmtResult
- Sema::ActOnOpenMPForDirective(ArrayRef<OMPClause *> Clauses, Stmt *AStmt,
- SourceLocation StartLoc, SourceLocation EndLoc,
- VarsWithInheritedDSAType &VarsWithImplicitDSA) {
- if (!AStmt)
- return StmtError();
- assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
- OMPLoopDirective::HelperExprs B;
- // In presence of clause 'collapse' or 'ordered' with number of loops, it will
- // define the nested loops number.
- unsigned NestedLoopCount = checkOpenMPLoop(
- OMPD_for, getCollapseNumberExpr(Clauses), getOrderedNumberExpr(Clauses),
- AStmt, *this, *DSAStack, VarsWithImplicitDSA, B);
- if (NestedLoopCount == 0)
- return StmtError();
- assert((CurContext->isDependentContext() || B.builtAll()) &&
- "omp for loop exprs were not built");
- if (!CurContext->isDependentContext()) {
- // Finalize the clauses that need pre-built expressions for CodeGen.
- for (OMPClause *C : Clauses) {
- if (auto *LC = dyn_cast<OMPLinearClause>(C))
- if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef),
- B.NumIterations, *this, CurScope,
- DSAStack))
- return StmtError();
- }
- }
- setFunctionHasBranchProtectedScope();
- return OMPForDirective::Create(Context, StartLoc, EndLoc, NestedLoopCount,
- Clauses, AStmt, B, DSAStack->isCancelRegion());
- }
- StmtResult Sema::ActOnOpenMPForSimdDirective(
- ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc,
- SourceLocation EndLoc, VarsWithInheritedDSAType &VarsWithImplicitDSA) {
- if (!AStmt)
- return StmtError();
- assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
- OMPLoopDirective::HelperExprs B;
- // In presence of clause 'collapse' or 'ordered' with number of loops, it will
- // define the nested loops number.
- unsigned NestedLoopCount =
- checkOpenMPLoop(OMPD_for_simd, getCollapseNumberExpr(Clauses),
- getOrderedNumberExpr(Clauses), AStmt, *this, *DSAStack,
- VarsWithImplicitDSA, B);
- if (NestedLoopCount == 0)
- return StmtError();
- assert((CurContext->isDependentContext() || B.builtAll()) &&
- "omp for simd loop exprs were not built");
- if (!CurContext->isDependentContext()) {
- // Finalize the clauses that need pre-built expressions for CodeGen.
- for (OMPClause *C : Clauses) {
- if (auto *LC = dyn_cast<OMPLinearClause>(C))
- if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef),
- B.NumIterations, *this, CurScope,
- DSAStack))
- return StmtError();
- }
- }
- if (checkSimdlenSafelenSpecified(*this, Clauses))
- return StmtError();
- setFunctionHasBranchProtectedScope();
- return OMPForSimdDirective::Create(Context, StartLoc, EndLoc, NestedLoopCount,
- Clauses, AStmt, B);
- }
- StmtResult Sema::ActOnOpenMPSectionsDirective(ArrayRef<OMPClause *> Clauses,
- Stmt *AStmt,
- SourceLocation StartLoc,
- SourceLocation EndLoc) {
- if (!AStmt)
- return StmtError();
- assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
- auto BaseStmt = AStmt;
- while (auto *CS = dyn_cast_or_null<CapturedStmt>(BaseStmt))
- BaseStmt = CS->getCapturedStmt();
- if (auto *C = dyn_cast_or_null<CompoundStmt>(BaseStmt)) {
- auto S = C->children();
- if (S.begin() == S.end())
- return StmtError();
- // All associated statements must be '#pragma omp section' except for
- // the first one.
- for (Stmt *SectionStmt : llvm::make_range(std::next(S.begin()), S.end())) {
- if (!SectionStmt || !isa<OMPSectionDirective>(SectionStmt)) {
- if (SectionStmt)
- Diag(SectionStmt->getBeginLoc(),
- diag::err_omp_sections_substmt_not_section);
- return StmtError();
- }
- cast<OMPSectionDirective>(SectionStmt)
- ->setHasCancel(DSAStack->isCancelRegion());
- }
- } else {
- Diag(AStmt->getBeginLoc(), diag::err_omp_sections_not_compound_stmt);
- return StmtError();
- }
- setFunctionHasBranchProtectedScope();
- return OMPSectionsDirective::Create(Context, StartLoc, EndLoc, Clauses, AStmt,
- DSAStack->isCancelRegion());
- }
- StmtResult Sema::ActOnOpenMPSectionDirective(Stmt *AStmt,
- SourceLocation StartLoc,
- SourceLocation EndLoc) {
- if (!AStmt)
- return StmtError();
- assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
- setFunctionHasBranchProtectedScope();
- DSAStack->setParentCancelRegion(DSAStack->isCancelRegion());
- return OMPSectionDirective::Create(Context, StartLoc, EndLoc, AStmt,
- DSAStack->isCancelRegion());
- }
- StmtResult Sema::ActOnOpenMPSingleDirective(ArrayRef<OMPClause *> Clauses,
- Stmt *AStmt,
- SourceLocation StartLoc,
- SourceLocation EndLoc) {
- if (!AStmt)
- return StmtError();
- assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
- setFunctionHasBranchProtectedScope();
- // OpenMP [2.7.3, single Construct, Restrictions]
- // The copyprivate clause must not be used with the nowait clause.
- const OMPClause *Nowait = nullptr;
- const OMPClause *Copyprivate = nullptr;
- for (const OMPClause *Clause : Clauses) {
- if (Clause->getClauseKind() == OMPC_nowait)
- Nowait = Clause;
- else if (Clause->getClauseKind() == OMPC_copyprivate)
- Copyprivate = Clause;
- if (Copyprivate && Nowait) {
- Diag(Copyprivate->getBeginLoc(),
- diag::err_omp_single_copyprivate_with_nowait);
- Diag(Nowait->getBeginLoc(), diag::note_omp_nowait_clause_here);
- return StmtError();
- }
- }
- return OMPSingleDirective::Create(Context, StartLoc, EndLoc, Clauses, AStmt);
- }
- StmtResult Sema::ActOnOpenMPMasterDirective(Stmt *AStmt,
- SourceLocation StartLoc,
- SourceLocation EndLoc) {
- if (!AStmt)
- return StmtError();
- assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
- setFunctionHasBranchProtectedScope();
- return OMPMasterDirective::Create(Context, StartLoc, EndLoc, AStmt);
- }
- StmtResult Sema::ActOnOpenMPCriticalDirective(
- const DeclarationNameInfo &DirName, ArrayRef<OMPClause *> Clauses,
- Stmt *AStmt, SourceLocation StartLoc, SourceLocation EndLoc) {
- if (!AStmt)
- return StmtError();
- assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
- bool ErrorFound = false;
- llvm::APSInt Hint;
- SourceLocation HintLoc;
- bool DependentHint = false;
- for (const OMPClause *C : Clauses) {
- if (C->getClauseKind() == OMPC_hint) {
- if (!DirName.getName()) {
- Diag(C->getBeginLoc(), diag::err_omp_hint_clause_no_name);
- ErrorFound = true;
- }
- Expr *E = cast<OMPHintClause>(C)->getHint();
- if (E->isTypeDependent() || E->isValueDependent() ||
- E->isInstantiationDependent()) {
- DependentHint = true;
- } else {
- Hint = E->EvaluateKnownConstInt(Context);
- HintLoc = C->getBeginLoc();
- }
- }
- }
- if (ErrorFound)
- return StmtError();
- const auto Pair = DSAStack->getCriticalWithHint(DirName);
- if (Pair.first && DirName.getName() && !DependentHint) {
- if (llvm::APSInt::compareValues(Hint, Pair.second) != 0) {
- Diag(StartLoc, diag::err_omp_critical_with_hint);
- if (HintLoc.isValid())
- Diag(HintLoc, diag::note_omp_critical_hint_here)
- << 0 << Hint.toString(/*Radix=*/10, /*Signed=*/false);
- else
- Diag(StartLoc, diag::note_omp_critical_no_hint) << 0;
- if (const auto *C = Pair.first->getSingleClause<OMPHintClause>()) {
- Diag(C->getBeginLoc(), diag::note_omp_critical_hint_here)
- << 1
- << C->getHint()->EvaluateKnownConstInt(Context).toString(
- /*Radix=*/10, /*Signed=*/false);
- } else {
- Diag(Pair.first->getBeginLoc(), diag::note_omp_critical_no_hint) << 1;
- }
- }
- }
- setFunctionHasBranchProtectedScope();
- auto *Dir = OMPCriticalDirective::Create(Context, DirName, StartLoc, EndLoc,
- Clauses, AStmt);
- if (!Pair.first && DirName.getName() && !DependentHint)
- DSAStack->addCriticalWithHint(Dir, Hint);
- return Dir;
- }
- StmtResult Sema::ActOnOpenMPParallelForDirective(
- ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc,
- SourceLocation EndLoc, VarsWithInheritedDSAType &VarsWithImplicitDSA) {
- if (!AStmt)
- return StmtError();
- auto *CS = cast<CapturedStmt>(AStmt);
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- OMPLoopDirective::HelperExprs B;
- // In presence of clause 'collapse' or 'ordered' with number of loops, it will
- // define the nested loops number.
- unsigned NestedLoopCount =
- checkOpenMPLoop(OMPD_parallel_for, getCollapseNumberExpr(Clauses),
- getOrderedNumberExpr(Clauses), AStmt, *this, *DSAStack,
- VarsWithImplicitDSA, B);
- if (NestedLoopCount == 0)
- return StmtError();
- assert((CurContext->isDependentContext() || B.builtAll()) &&
- "omp parallel for loop exprs were not built");
- if (!CurContext->isDependentContext()) {
- // Finalize the clauses that need pre-built expressions for CodeGen.
- for (OMPClause *C : Clauses) {
- if (auto *LC = dyn_cast<OMPLinearClause>(C))
- if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef),
- B.NumIterations, *this, CurScope,
- DSAStack))
- return StmtError();
- }
- }
- setFunctionHasBranchProtectedScope();
- return OMPParallelForDirective::Create(Context, StartLoc, EndLoc,
- NestedLoopCount, Clauses, AStmt, B,
- DSAStack->isCancelRegion());
- }
- StmtResult Sema::ActOnOpenMPParallelForSimdDirective(
- ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc,
- SourceLocation EndLoc, VarsWithInheritedDSAType &VarsWithImplicitDSA) {
- if (!AStmt)
- return StmtError();
- auto *CS = cast<CapturedStmt>(AStmt);
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- OMPLoopDirective::HelperExprs B;
- // In presence of clause 'collapse' or 'ordered' with number of loops, it will
- // define the nested loops number.
- unsigned NestedLoopCount =
- checkOpenMPLoop(OMPD_parallel_for_simd, getCollapseNumberExpr(Clauses),
- getOrderedNumberExpr(Clauses), AStmt, *this, *DSAStack,
- VarsWithImplicitDSA, B);
- if (NestedLoopCount == 0)
- return StmtError();
- if (!CurContext->isDependentContext()) {
- // Finalize the clauses that need pre-built expressions for CodeGen.
- for (OMPClause *C : Clauses) {
- if (auto *LC = dyn_cast<OMPLinearClause>(C))
- if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef),
- B.NumIterations, *this, CurScope,
- DSAStack))
- return StmtError();
- }
- }
- if (checkSimdlenSafelenSpecified(*this, Clauses))
- return StmtError();
- setFunctionHasBranchProtectedScope();
- return OMPParallelForSimdDirective::Create(
- Context, StartLoc, EndLoc, NestedLoopCount, Clauses, AStmt, B);
- }
- StmtResult
- Sema::ActOnOpenMPParallelSectionsDirective(ArrayRef<OMPClause *> Clauses,
- Stmt *AStmt, SourceLocation StartLoc,
- SourceLocation EndLoc) {
- if (!AStmt)
- return StmtError();
- assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
- auto BaseStmt = AStmt;
- while (auto *CS = dyn_cast_or_null<CapturedStmt>(BaseStmt))
- BaseStmt = CS->getCapturedStmt();
- if (auto *C = dyn_cast_or_null<CompoundStmt>(BaseStmt)) {
- auto S = C->children();
- if (S.begin() == S.end())
- return StmtError();
- // All associated statements must be '#pragma omp section' except for
- // the first one.
- for (Stmt *SectionStmt : llvm::make_range(std::next(S.begin()), S.end())) {
- if (!SectionStmt || !isa<OMPSectionDirective>(SectionStmt)) {
- if (SectionStmt)
- Diag(SectionStmt->getBeginLoc(),
- diag::err_omp_parallel_sections_substmt_not_section);
- return StmtError();
- }
- cast<OMPSectionDirective>(SectionStmt)
- ->setHasCancel(DSAStack->isCancelRegion());
- }
- } else {
- Diag(AStmt->getBeginLoc(),
- diag::err_omp_parallel_sections_not_compound_stmt);
- return StmtError();
- }
- setFunctionHasBranchProtectedScope();
- return OMPParallelSectionsDirective::Create(
- Context, StartLoc, EndLoc, Clauses, AStmt, DSAStack->isCancelRegion());
- }
- StmtResult Sema::ActOnOpenMPTaskDirective(ArrayRef<OMPClause *> Clauses,
- Stmt *AStmt, SourceLocation StartLoc,
- SourceLocation EndLoc) {
- if (!AStmt)
- return StmtError();
- auto *CS = cast<CapturedStmt>(AStmt);
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- setFunctionHasBranchProtectedScope();
- return OMPTaskDirective::Create(Context, StartLoc, EndLoc, Clauses, AStmt,
- DSAStack->isCancelRegion());
- }
- StmtResult Sema::ActOnOpenMPTaskyieldDirective(SourceLocation StartLoc,
- SourceLocation EndLoc) {
- return OMPTaskyieldDirective::Create(Context, StartLoc, EndLoc);
- }
- StmtResult Sema::ActOnOpenMPBarrierDirective(SourceLocation StartLoc,
- SourceLocation EndLoc) {
- return OMPBarrierDirective::Create(Context, StartLoc, EndLoc);
- }
- StmtResult Sema::ActOnOpenMPTaskwaitDirective(SourceLocation StartLoc,
- SourceLocation EndLoc) {
- return OMPTaskwaitDirective::Create(Context, StartLoc, EndLoc);
- }
- StmtResult Sema::ActOnOpenMPTaskgroupDirective(ArrayRef<OMPClause *> Clauses,
- Stmt *AStmt,
- SourceLocation StartLoc,
- SourceLocation EndLoc) {
- if (!AStmt)
- return StmtError();
- assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
- setFunctionHasBranchProtectedScope();
- return OMPTaskgroupDirective::Create(Context, StartLoc, EndLoc, Clauses,
- AStmt,
- DSAStack->getTaskgroupReductionRef());
- }
- StmtResult Sema::ActOnOpenMPFlushDirective(ArrayRef<OMPClause *> Clauses,
- SourceLocation StartLoc,
- SourceLocation EndLoc) {
- assert(Clauses.size() <= 1 && "Extra clauses in flush directive");
- return OMPFlushDirective::Create(Context, StartLoc, EndLoc, Clauses);
- }
- StmtResult Sema::ActOnOpenMPOrderedDirective(ArrayRef<OMPClause *> Clauses,
- Stmt *AStmt,
- SourceLocation StartLoc,
- SourceLocation EndLoc) {
- const OMPClause *DependFound = nullptr;
- const OMPClause *DependSourceClause = nullptr;
- const OMPClause *DependSinkClause = nullptr;
- bool ErrorFound = false;
- const OMPThreadsClause *TC = nullptr;
- const OMPSIMDClause *SC = nullptr;
- for (const OMPClause *C : Clauses) {
- if (auto *DC = dyn_cast<OMPDependClause>(C)) {
- DependFound = C;
- if (DC->getDependencyKind() == OMPC_DEPEND_source) {
- if (DependSourceClause) {
- Diag(C->getBeginLoc(), diag::err_omp_more_one_clause)
- << getOpenMPDirectiveName(OMPD_ordered)
- << getOpenMPClauseName(OMPC_depend) << 2;
- ErrorFound = true;
- } else {
- DependSourceClause = C;
- }
- if (DependSinkClause) {
- Diag(C->getBeginLoc(), diag::err_omp_depend_sink_source_not_allowed)
- << 0;
- ErrorFound = true;
- }
- } else if (DC->getDependencyKind() == OMPC_DEPEND_sink) {
- if (DependSourceClause) {
- Diag(C->getBeginLoc(), diag::err_omp_depend_sink_source_not_allowed)
- << 1;
- ErrorFound = true;
- }
- DependSinkClause = C;
- }
- } else if (C->getClauseKind() == OMPC_threads) {
- TC = cast<OMPThreadsClause>(C);
- } else if (C->getClauseKind() == OMPC_simd) {
- SC = cast<OMPSIMDClause>(C);
- }
- }
- if (!ErrorFound && !SC &&
- isOpenMPSimdDirective(DSAStack->getParentDirective())) {
- // OpenMP [2.8.1,simd Construct, Restrictions]
- // An ordered construct with the simd clause is the only OpenMP construct
- // that can appear in the simd region.
- Diag(StartLoc, diag::err_omp_prohibited_region_simd);
- ErrorFound = true;
- } else if (DependFound && (TC || SC)) {
- Diag(DependFound->getBeginLoc(), diag::err_omp_depend_clause_thread_simd)
- << getOpenMPClauseName(TC ? TC->getClauseKind() : SC->getClauseKind());
- ErrorFound = true;
- } else if (DependFound && !DSAStack->getParentOrderedRegionParam().first) {
- Diag(DependFound->getBeginLoc(),
- diag::err_omp_ordered_directive_without_param);
- ErrorFound = true;
- } else if (TC || Clauses.empty()) {
- if (const Expr *Param = DSAStack->getParentOrderedRegionParam().first) {
- SourceLocation ErrLoc = TC ? TC->getBeginLoc() : StartLoc;
- Diag(ErrLoc, diag::err_omp_ordered_directive_with_param)
- << (TC != nullptr);
- Diag(Param->getBeginLoc(), diag::note_omp_ordered_param);
- ErrorFound = true;
- }
- }
- if ((!AStmt && !DependFound) || ErrorFound)
- return StmtError();
- if (AStmt) {
- assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
- setFunctionHasBranchProtectedScope();
- }
- return OMPOrderedDirective::Create(Context, StartLoc, EndLoc, Clauses, AStmt);
- }
- namespace {
- /// Helper class for checking expression in 'omp atomic [update]'
- /// construct.
- class OpenMPAtomicUpdateChecker {
- /// Error results for atomic update expressions.
- enum ExprAnalysisErrorCode {
- /// A statement is not an expression statement.
- NotAnExpression,
- /// Expression is not builtin binary or unary operation.
- NotABinaryOrUnaryExpression,
- /// Unary operation is not post-/pre- increment/decrement operation.
- NotAnUnaryIncDecExpression,
- /// An expression is not of scalar type.
- NotAScalarType,
- /// A binary operation is not an assignment operation.
- NotAnAssignmentOp,
- /// RHS part of the binary operation is not a binary expression.
- NotABinaryExpression,
- /// RHS part is not additive/multiplicative/shift/biwise binary
- /// expression.
- NotABinaryOperator,
- /// RHS binary operation does not have reference to the updated LHS
- /// part.
- NotAnUpdateExpression,
- /// No errors is found.
- NoError
- };
- /// Reference to Sema.
- Sema &SemaRef;
- /// A location for note diagnostics (when error is found).
- SourceLocation NoteLoc;
- /// 'x' lvalue part of the source atomic expression.
- Expr *X;
- /// 'expr' rvalue part of the source atomic expression.
- Expr *E;
- /// Helper expression of the form
- /// 'OpaqueValueExpr(x) binop OpaqueValueExpr(expr)' or
- /// 'OpaqueValueExpr(expr) binop OpaqueValueExpr(x)'.
- Expr *UpdateExpr;
- /// Is 'x' a LHS in a RHS part of full update expression. It is
- /// important for non-associative operations.
- bool IsXLHSInRHSPart;
- BinaryOperatorKind Op;
- SourceLocation OpLoc;
- /// true if the source expression is a postfix unary operation, false
- /// if it is a prefix unary operation.
- bool IsPostfixUpdate;
- public:
- OpenMPAtomicUpdateChecker(Sema &SemaRef)
- : SemaRef(SemaRef), X(nullptr), E(nullptr), UpdateExpr(nullptr),
- IsXLHSInRHSPart(false), Op(BO_PtrMemD), IsPostfixUpdate(false) {}
- /// Check specified statement that it is suitable for 'atomic update'
- /// constructs and extract 'x', 'expr' and Operation from the original
- /// expression. If DiagId and NoteId == 0, then only check is performed
- /// without error notification.
- /// \param DiagId Diagnostic which should be emitted if error is found.
- /// \param NoteId Diagnostic note for the main error message.
- /// \return true if statement is not an update expression, false otherwise.
- bool checkStatement(Stmt *S, unsigned DiagId = 0, unsigned NoteId = 0);
- /// Return the 'x' lvalue part of the source atomic expression.
- Expr *getX() const { return X; }
- /// Return the 'expr' rvalue part of the source atomic expression.
- Expr *getExpr() const { return E; }
- /// Return the update expression used in calculation of the updated
- /// value. Always has form 'OpaqueValueExpr(x) binop OpaqueValueExpr(expr)' or
- /// 'OpaqueValueExpr(expr) binop OpaqueValueExpr(x)'.
- Expr *getUpdateExpr() const { return UpdateExpr; }
- /// Return true if 'x' is LHS in RHS part of full update expression,
- /// false otherwise.
- bool isXLHSInRHSPart() const { return IsXLHSInRHSPart; }
- /// true if the source expression is a postfix unary operation, false
- /// if it is a prefix unary operation.
- bool isPostfixUpdate() const { return IsPostfixUpdate; }
- private:
- bool checkBinaryOperation(BinaryOperator *AtomicBinOp, unsigned DiagId = 0,
- unsigned NoteId = 0);
- };
- } // namespace
- bool OpenMPAtomicUpdateChecker::checkBinaryOperation(
- BinaryOperator *AtomicBinOp, unsigned DiagId, unsigned NoteId) {
- ExprAnalysisErrorCode ErrorFound = NoError;
- SourceLocation ErrorLoc, NoteLoc;
- SourceRange ErrorRange, NoteRange;
- // Allowed constructs are:
- // x = x binop expr;
- // x = expr binop x;
- if (AtomicBinOp->getOpcode() == BO_Assign) {
- X = AtomicBinOp->getLHS();
- if (const auto *AtomicInnerBinOp = dyn_cast<BinaryOperator>(
- AtomicBinOp->getRHS()->IgnoreParenImpCasts())) {
- if (AtomicInnerBinOp->isMultiplicativeOp() ||
- AtomicInnerBinOp->isAdditiveOp() || AtomicInnerBinOp->isShiftOp() ||
- AtomicInnerBinOp->isBitwiseOp()) {
- Op = AtomicInnerBinOp->getOpcode();
- OpLoc = AtomicInnerBinOp->getOperatorLoc();
- Expr *LHS = AtomicInnerBinOp->getLHS();
- Expr *RHS = AtomicInnerBinOp->getRHS();
- llvm::FoldingSetNodeID XId, LHSId, RHSId;
- X->IgnoreParenImpCasts()->Profile(XId, SemaRef.getASTContext(),
- /*Canonical=*/true);
- LHS->IgnoreParenImpCasts()->Profile(LHSId, SemaRef.getASTContext(),
- /*Canonical=*/true);
- RHS->IgnoreParenImpCasts()->Profile(RHSId, SemaRef.getASTContext(),
- /*Canonical=*/true);
- if (XId == LHSId) {
- E = RHS;
- IsXLHSInRHSPart = true;
- } else if (XId == RHSId) {
- E = LHS;
- IsXLHSInRHSPart = false;
- } else {
- ErrorLoc = AtomicInnerBinOp->getExprLoc();
- ErrorRange = AtomicInnerBinOp->getSourceRange();
- NoteLoc = X->getExprLoc();
- NoteRange = X->getSourceRange();
- ErrorFound = NotAnUpdateExpression;
- }
- } else {
- ErrorLoc = AtomicInnerBinOp->getExprLoc();
- ErrorRange = AtomicInnerBinOp->getSourceRange();
- NoteLoc = AtomicInnerBinOp->getOperatorLoc();
- NoteRange = SourceRange(NoteLoc, NoteLoc);
- ErrorFound = NotABinaryOperator;
- }
- } else {
- NoteLoc = ErrorLoc = AtomicBinOp->getRHS()->getExprLoc();
- NoteRange = ErrorRange = AtomicBinOp->getRHS()->getSourceRange();
- ErrorFound = NotABinaryExpression;
- }
- } else {
- ErrorLoc = AtomicBinOp->getExprLoc();
- ErrorRange = AtomicBinOp->getSourceRange();
- NoteLoc = AtomicBinOp->getOperatorLoc();
- NoteRange = SourceRange(NoteLoc, NoteLoc);
- ErrorFound = NotAnAssignmentOp;
- }
- if (ErrorFound != NoError && DiagId != 0 && NoteId != 0) {
- SemaRef.Diag(ErrorLoc, DiagId) << ErrorRange;
- SemaRef.Diag(NoteLoc, NoteId) << ErrorFound << NoteRange;
- return true;
- }
- if (SemaRef.CurContext->isDependentContext())
- E = X = UpdateExpr = nullptr;
- return ErrorFound != NoError;
- }
- bool OpenMPAtomicUpdateChecker::checkStatement(Stmt *S, unsigned DiagId,
- unsigned NoteId) {
- ExprAnalysisErrorCode ErrorFound = NoError;
- SourceLocation ErrorLoc, NoteLoc;
- SourceRange ErrorRange, NoteRange;
- // Allowed constructs are:
- // x++;
- // x--;
- // ++x;
- // --x;
- // x binop= expr;
- // x = x binop expr;
- // x = expr binop x;
- if (auto *AtomicBody = dyn_cast<Expr>(S)) {
- AtomicBody = AtomicBody->IgnoreParenImpCasts();
- if (AtomicBody->getType()->isScalarType() ||
- AtomicBody->isInstantiationDependent()) {
- if (const auto *AtomicCompAssignOp = dyn_cast<CompoundAssignOperator>(
- AtomicBody->IgnoreParenImpCasts())) {
- // Check for Compound Assignment Operation
- Op = BinaryOperator::getOpForCompoundAssignment(
- AtomicCompAssignOp->getOpcode());
- OpLoc = AtomicCompAssignOp->getOperatorLoc();
- E = AtomicCompAssignOp->getRHS();
- X = AtomicCompAssignOp->getLHS()->IgnoreParens();
- IsXLHSInRHSPart = true;
- } else if (auto *AtomicBinOp = dyn_cast<BinaryOperator>(
- AtomicBody->IgnoreParenImpCasts())) {
- // Check for Binary Operation
- if (checkBinaryOperation(AtomicBinOp, DiagId, NoteId))
- return true;
- } else if (const auto *AtomicUnaryOp = dyn_cast<UnaryOperator>(
- AtomicBody->IgnoreParenImpCasts())) {
- // Check for Unary Operation
- if (AtomicUnaryOp->isIncrementDecrementOp()) {
- IsPostfixUpdate = AtomicUnaryOp->isPostfix();
- Op = AtomicUnaryOp->isIncrementOp() ? BO_Add : BO_Sub;
- OpLoc = AtomicUnaryOp->getOperatorLoc();
- X = AtomicUnaryOp->getSubExpr()->IgnoreParens();
- E = SemaRef.ActOnIntegerConstant(OpLoc, /*uint64_t Val=*/1).get();
- IsXLHSInRHSPart = true;
- } else {
- ErrorFound = NotAnUnaryIncDecExpression;
- ErrorLoc = AtomicUnaryOp->getExprLoc();
- ErrorRange = AtomicUnaryOp->getSourceRange();
- NoteLoc = AtomicUnaryOp->getOperatorLoc();
- NoteRange = SourceRange(NoteLoc, NoteLoc);
- }
- } else if (!AtomicBody->isInstantiationDependent()) {
- ErrorFound = NotABinaryOrUnaryExpression;
- NoteLoc = ErrorLoc = AtomicBody->getExprLoc();
- NoteRange = ErrorRange = AtomicBody->getSourceRange();
- }
- } else {
- ErrorFound = NotAScalarType;
- NoteLoc = ErrorLoc = AtomicBody->getBeginLoc();
- NoteRange = ErrorRange = SourceRange(NoteLoc, NoteLoc);
- }
- } else {
- ErrorFound = NotAnExpression;
- NoteLoc = ErrorLoc = S->getBeginLoc();
- NoteRange = ErrorRange = SourceRange(NoteLoc, NoteLoc);
- }
- if (ErrorFound != NoError && DiagId != 0 && NoteId != 0) {
- SemaRef.Diag(ErrorLoc, DiagId) << ErrorRange;
- SemaRef.Diag(NoteLoc, NoteId) << ErrorFound << NoteRange;
- return true;
- }
- if (SemaRef.CurContext->isDependentContext())
- E = X = UpdateExpr = nullptr;
- if (ErrorFound == NoError && E && X) {
- // Build an update expression of form 'OpaqueValueExpr(x) binop
- // OpaqueValueExpr(expr)' or 'OpaqueValueExpr(expr) binop
- // OpaqueValueExpr(x)' and then cast it to the type of the 'x' expression.
- auto *OVEX = new (SemaRef.getASTContext())
- OpaqueValueExpr(X->getExprLoc(), X->getType(), VK_RValue);
- auto *OVEExpr = new (SemaRef.getASTContext())
- OpaqueValueExpr(E->getExprLoc(), E->getType(), VK_RValue);
- ExprResult Update =
- SemaRef.CreateBuiltinBinOp(OpLoc, Op, IsXLHSInRHSPart ? OVEX : OVEExpr,
- IsXLHSInRHSPart ? OVEExpr : OVEX);
- if (Update.isInvalid())
- return true;
- Update = SemaRef.PerformImplicitConversion(Update.get(), X->getType(),
- Sema::AA_Casting);
- if (Update.isInvalid())
- return true;
- UpdateExpr = Update.get();
- }
- return ErrorFound != NoError;
- }
- StmtResult Sema::ActOnOpenMPAtomicDirective(ArrayRef<OMPClause *> Clauses,
- Stmt *AStmt,
- SourceLocation StartLoc,
- SourceLocation EndLoc) {
- if (!AStmt)
- return StmtError();
- auto *CS = cast<CapturedStmt>(AStmt);
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- OpenMPClauseKind AtomicKind = OMPC_unknown;
- SourceLocation AtomicKindLoc;
- for (const OMPClause *C : Clauses) {
- if (C->getClauseKind() == OMPC_read || C->getClauseKind() == OMPC_write ||
- C->getClauseKind() == OMPC_update ||
- C->getClauseKind() == OMPC_capture) {
- if (AtomicKind != OMPC_unknown) {
- Diag(C->getBeginLoc(), diag::err_omp_atomic_several_clauses)
- << SourceRange(C->getBeginLoc(), C->getEndLoc());
- Diag(AtomicKindLoc, diag::note_omp_atomic_previous_clause)
- << getOpenMPClauseName(AtomicKind);
- } else {
- AtomicKind = C->getClauseKind();
- AtomicKindLoc = C->getBeginLoc();
- }
- }
- }
- Stmt *Body = CS->getCapturedStmt();
- if (auto *EWC = dyn_cast<ExprWithCleanups>(Body))
- Body = EWC->getSubExpr();
- Expr *X = nullptr;
- Expr *V = nullptr;
- Expr *E = nullptr;
- Expr *UE = nullptr;
- bool IsXLHSInRHSPart = false;
- bool IsPostfixUpdate = false;
- // OpenMP [2.12.6, atomic Construct]
- // In the next expressions:
- // * x and v (as applicable) are both l-value expressions with scalar type.
- // * During the execution of an atomic region, multiple syntactic
- // occurrences of x must designate the same storage location.
- // * Neither of v and expr (as applicable) may access the storage location
- // designated by x.
- // * Neither of x and expr (as applicable) may access the storage location
- // designated by v.
- // * expr is an expression with scalar type.
- // * binop is one of +, *, -, /, &, ^, |, <<, or >>.
- // * binop, binop=, ++, and -- are not overloaded operators.
- // * The expression x binop expr must be numerically equivalent to x binop
- // (expr). This requirement is satisfied if the operators in expr have
- // precedence greater than binop, or by using parentheses around expr or
- // subexpressions of expr.
- // * The expression expr binop x must be numerically equivalent to (expr)
- // binop x. This requirement is satisfied if the operators in expr have
- // precedence equal to or greater than binop, or by using parentheses around
- // expr or subexpressions of expr.
- // * For forms that allow multiple occurrences of x, the number of times
- // that x is evaluated is unspecified.
- if (AtomicKind == OMPC_read) {
- enum {
- NotAnExpression,
- NotAnAssignmentOp,
- NotAScalarType,
- NotAnLValue,
- NoError
- } ErrorFound = NoError;
- SourceLocation ErrorLoc, NoteLoc;
- SourceRange ErrorRange, NoteRange;
- // If clause is read:
- // v = x;
- if (const auto *AtomicBody = dyn_cast<Expr>(Body)) {
- const auto *AtomicBinOp =
- dyn_cast<BinaryOperator>(AtomicBody->IgnoreParenImpCasts());
- if (AtomicBinOp && AtomicBinOp->getOpcode() == BO_Assign) {
- X = AtomicBinOp->getRHS()->IgnoreParenImpCasts();
- V = AtomicBinOp->getLHS()->IgnoreParenImpCasts();
- if ((X->isInstantiationDependent() || X->getType()->isScalarType()) &&
- (V->isInstantiationDependent() || V->getType()->isScalarType())) {
- if (!X->isLValue() || !V->isLValue()) {
- const Expr *NotLValueExpr = X->isLValue() ? V : X;
- ErrorFound = NotAnLValue;
- ErrorLoc = AtomicBinOp->getExprLoc();
- ErrorRange = AtomicBinOp->getSourceRange();
- NoteLoc = NotLValueExpr->getExprLoc();
- NoteRange = NotLValueExpr->getSourceRange();
- }
- } else if (!X->isInstantiationDependent() ||
- !V->isInstantiationDependent()) {
- const Expr *NotScalarExpr =
- (X->isInstantiationDependent() || X->getType()->isScalarType())
- ? V
- : X;
- ErrorFound = NotAScalarType;
- ErrorLoc = AtomicBinOp->getExprLoc();
- ErrorRange = AtomicBinOp->getSourceRange();
- NoteLoc = NotScalarExpr->getExprLoc();
- NoteRange = NotScalarExpr->getSourceRange();
- }
- } else if (!AtomicBody->isInstantiationDependent()) {
- ErrorFound = NotAnAssignmentOp;
- ErrorLoc = AtomicBody->getExprLoc();
- ErrorRange = AtomicBody->getSourceRange();
- NoteLoc = AtomicBinOp ? AtomicBinOp->getOperatorLoc()
- : AtomicBody->getExprLoc();
- NoteRange = AtomicBinOp ? AtomicBinOp->getSourceRange()
- : AtomicBody->getSourceRange();
- }
- } else {
- ErrorFound = NotAnExpression;
- NoteLoc = ErrorLoc = Body->getBeginLoc();
- NoteRange = ErrorRange = SourceRange(NoteLoc, NoteLoc);
- }
- if (ErrorFound != NoError) {
- Diag(ErrorLoc, diag::err_omp_atomic_read_not_expression_statement)
- << ErrorRange;
- Diag(NoteLoc, diag::note_omp_atomic_read_write) << ErrorFound
- << NoteRange;
- return StmtError();
- }
- if (CurContext->isDependentContext())
- V = X = nullptr;
- } else if (AtomicKind == OMPC_write) {
- enum {
- NotAnExpression,
- NotAnAssignmentOp,
- NotAScalarType,
- NotAnLValue,
- NoError
- } ErrorFound = NoError;
- SourceLocation ErrorLoc, NoteLoc;
- SourceRange ErrorRange, NoteRange;
- // If clause is write:
- // x = expr;
- if (const auto *AtomicBody = dyn_cast<Expr>(Body)) {
- const auto *AtomicBinOp =
- dyn_cast<BinaryOperator>(AtomicBody->IgnoreParenImpCasts());
- if (AtomicBinOp && AtomicBinOp->getOpcode() == BO_Assign) {
- X = AtomicBinOp->getLHS();
- E = AtomicBinOp->getRHS();
- if ((X->isInstantiationDependent() || X->getType()->isScalarType()) &&
- (E->isInstantiationDependent() || E->getType()->isScalarType())) {
- if (!X->isLValue()) {
- ErrorFound = NotAnLValue;
- ErrorLoc = AtomicBinOp->getExprLoc();
- ErrorRange = AtomicBinOp->getSourceRange();
- NoteLoc = X->getExprLoc();
- NoteRange = X->getSourceRange();
- }
- } else if (!X->isInstantiationDependent() ||
- !E->isInstantiationDependent()) {
- const Expr *NotScalarExpr =
- (X->isInstantiationDependent() || X->getType()->isScalarType())
- ? E
- : X;
- ErrorFound = NotAScalarType;
- ErrorLoc = AtomicBinOp->getExprLoc();
- ErrorRange = AtomicBinOp->getSourceRange();
- NoteLoc = NotScalarExpr->getExprLoc();
- NoteRange = NotScalarExpr->getSourceRange();
- }
- } else if (!AtomicBody->isInstantiationDependent()) {
- ErrorFound = NotAnAssignmentOp;
- ErrorLoc = AtomicBody->getExprLoc();
- ErrorRange = AtomicBody->getSourceRange();
- NoteLoc = AtomicBinOp ? AtomicBinOp->getOperatorLoc()
- : AtomicBody->getExprLoc();
- NoteRange = AtomicBinOp ? AtomicBinOp->getSourceRange()
- : AtomicBody->getSourceRange();
- }
- } else {
- ErrorFound = NotAnExpression;
- NoteLoc = ErrorLoc = Body->getBeginLoc();
- NoteRange = ErrorRange = SourceRange(NoteLoc, NoteLoc);
- }
- if (ErrorFound != NoError) {
- Diag(ErrorLoc, diag::err_omp_atomic_write_not_expression_statement)
- << ErrorRange;
- Diag(NoteLoc, diag::note_omp_atomic_read_write) << ErrorFound
- << NoteRange;
- return StmtError();
- }
- if (CurContext->isDependentContext())
- E = X = nullptr;
- } else if (AtomicKind == OMPC_update || AtomicKind == OMPC_unknown) {
- // If clause is update:
- // x++;
- // x--;
- // ++x;
- // --x;
- // x binop= expr;
- // x = x binop expr;
- // x = expr binop x;
- OpenMPAtomicUpdateChecker Checker(*this);
- if (Checker.checkStatement(
- Body, (AtomicKind == OMPC_update)
- ? diag::err_omp_atomic_update_not_expression_statement
- : diag::err_omp_atomic_not_expression_statement,
- diag::note_omp_atomic_update))
- return StmtError();
- if (!CurContext->isDependentContext()) {
- E = Checker.getExpr();
- X = Checker.getX();
- UE = Checker.getUpdateExpr();
- IsXLHSInRHSPart = Checker.isXLHSInRHSPart();
- }
- } else if (AtomicKind == OMPC_capture) {
- enum {
- NotAnAssignmentOp,
- NotACompoundStatement,
- NotTwoSubstatements,
- NotASpecificExpression,
- NoError
- } ErrorFound = NoError;
- SourceLocation ErrorLoc, NoteLoc;
- SourceRange ErrorRange, NoteRange;
- if (const auto *AtomicBody = dyn_cast<Expr>(Body)) {
- // If clause is a capture:
- // v = x++;
- // v = x--;
- // v = ++x;
- // v = --x;
- // v = x binop= expr;
- // v = x = x binop expr;
- // v = x = expr binop x;
- const auto *AtomicBinOp =
- dyn_cast<BinaryOperator>(AtomicBody->IgnoreParenImpCasts());
- if (AtomicBinOp && AtomicBinOp->getOpcode() == BO_Assign) {
- V = AtomicBinOp->getLHS();
- Body = AtomicBinOp->getRHS()->IgnoreParenImpCasts();
- OpenMPAtomicUpdateChecker Checker(*this);
- if (Checker.checkStatement(
- Body, diag::err_omp_atomic_capture_not_expression_statement,
- diag::note_omp_atomic_update))
- return StmtError();
- E = Checker.getExpr();
- X = Checker.getX();
- UE = Checker.getUpdateExpr();
- IsXLHSInRHSPart = Checker.isXLHSInRHSPart();
- IsPostfixUpdate = Checker.isPostfixUpdate();
- } else if (!AtomicBody->isInstantiationDependent()) {
- ErrorLoc = AtomicBody->getExprLoc();
- ErrorRange = AtomicBody->getSourceRange();
- NoteLoc = AtomicBinOp ? AtomicBinOp->getOperatorLoc()
- : AtomicBody->getExprLoc();
- NoteRange = AtomicBinOp ? AtomicBinOp->getSourceRange()
- : AtomicBody->getSourceRange();
- ErrorFound = NotAnAssignmentOp;
- }
- if (ErrorFound != NoError) {
- Diag(ErrorLoc, diag::err_omp_atomic_capture_not_expression_statement)
- << ErrorRange;
- Diag(NoteLoc, diag::note_omp_atomic_capture) << ErrorFound << NoteRange;
- return StmtError();
- }
- if (CurContext->isDependentContext())
- UE = V = E = X = nullptr;
- } else {
- // If clause is a capture:
- // { v = x; x = expr; }
- // { v = x; x++; }
- // { v = x; x--; }
- // { v = x; ++x; }
- // { v = x; --x; }
- // { v = x; x binop= expr; }
- // { v = x; x = x binop expr; }
- // { v = x; x = expr binop x; }
- // { x++; v = x; }
- // { x--; v = x; }
- // { ++x; v = x; }
- // { --x; v = x; }
- // { x binop= expr; v = x; }
- // { x = x binop expr; v = x; }
- // { x = expr binop x; v = x; }
- if (auto *CS = dyn_cast<CompoundStmt>(Body)) {
- // Check that this is { expr1; expr2; }
- if (CS->size() == 2) {
- Stmt *First = CS->body_front();
- Stmt *Second = CS->body_back();
- if (auto *EWC = dyn_cast<ExprWithCleanups>(First))
- First = EWC->getSubExpr()->IgnoreParenImpCasts();
- if (auto *EWC = dyn_cast<ExprWithCleanups>(Second))
- Second = EWC->getSubExpr()->IgnoreParenImpCasts();
- // Need to find what subexpression is 'v' and what is 'x'.
- OpenMPAtomicUpdateChecker Checker(*this);
- bool IsUpdateExprFound = !Checker.checkStatement(Second);
- BinaryOperator *BinOp = nullptr;
- if (IsUpdateExprFound) {
- BinOp = dyn_cast<BinaryOperator>(First);
- IsUpdateExprFound = BinOp && BinOp->getOpcode() == BO_Assign;
- }
- if (IsUpdateExprFound && !CurContext->isDependentContext()) {
- // { v = x; x++; }
- // { v = x; x--; }
- // { v = x; ++x; }
- // { v = x; --x; }
- // { v = x; x binop= expr; }
- // { v = x; x = x binop expr; }
- // { v = x; x = expr binop x; }
- // Check that the first expression has form v = x.
- Expr *PossibleX = BinOp->getRHS()->IgnoreParenImpCasts();
- llvm::FoldingSetNodeID XId, PossibleXId;
- Checker.getX()->Profile(XId, Context, /*Canonical=*/true);
- PossibleX->Profile(PossibleXId, Context, /*Canonical=*/true);
- IsUpdateExprFound = XId == PossibleXId;
- if (IsUpdateExprFound) {
- V = BinOp->getLHS();
- X = Checker.getX();
- E = Checker.getExpr();
- UE = Checker.getUpdateExpr();
- IsXLHSInRHSPart = Checker.isXLHSInRHSPart();
- IsPostfixUpdate = true;
- }
- }
- if (!IsUpdateExprFound) {
- IsUpdateExprFound = !Checker.checkStatement(First);
- BinOp = nullptr;
- if (IsUpdateExprFound) {
- BinOp = dyn_cast<BinaryOperator>(Second);
- IsUpdateExprFound = BinOp && BinOp->getOpcode() == BO_Assign;
- }
- if (IsUpdateExprFound && !CurContext->isDependentContext()) {
- // { x++; v = x; }
- // { x--; v = x; }
- // { ++x; v = x; }
- // { --x; v = x; }
- // { x binop= expr; v = x; }
- // { x = x binop expr; v = x; }
- // { x = expr binop x; v = x; }
- // Check that the second expression has form v = x.
- Expr *PossibleX = BinOp->getRHS()->IgnoreParenImpCasts();
- llvm::FoldingSetNodeID XId, PossibleXId;
- Checker.getX()->Profile(XId, Context, /*Canonical=*/true);
- PossibleX->Profile(PossibleXId, Context, /*Canonical=*/true);
- IsUpdateExprFound = XId == PossibleXId;
- if (IsUpdateExprFound) {
- V = BinOp->getLHS();
- X = Checker.getX();
- E = Checker.getExpr();
- UE = Checker.getUpdateExpr();
- IsXLHSInRHSPart = Checker.isXLHSInRHSPart();
- IsPostfixUpdate = false;
- }
- }
- }
- if (!IsUpdateExprFound) {
- // { v = x; x = expr; }
- auto *FirstExpr = dyn_cast<Expr>(First);
- auto *SecondExpr = dyn_cast<Expr>(Second);
- if (!FirstExpr || !SecondExpr ||
- !(FirstExpr->isInstantiationDependent() ||
- SecondExpr->isInstantiationDependent())) {
- auto *FirstBinOp = dyn_cast<BinaryOperator>(First);
- if (!FirstBinOp || FirstBinOp->getOpcode() != BO_Assign) {
- ErrorFound = NotAnAssignmentOp;
- NoteLoc = ErrorLoc = FirstBinOp ? FirstBinOp->getOperatorLoc()
- : First->getBeginLoc();
- NoteRange = ErrorRange = FirstBinOp
- ? FirstBinOp->getSourceRange()
- : SourceRange(ErrorLoc, ErrorLoc);
- } else {
- auto *SecondBinOp = dyn_cast<BinaryOperator>(Second);
- if (!SecondBinOp || SecondBinOp->getOpcode() != BO_Assign) {
- ErrorFound = NotAnAssignmentOp;
- NoteLoc = ErrorLoc = SecondBinOp
- ? SecondBinOp->getOperatorLoc()
- : Second->getBeginLoc();
- NoteRange = ErrorRange =
- SecondBinOp ? SecondBinOp->getSourceRange()
- : SourceRange(ErrorLoc, ErrorLoc);
- } else {
- Expr *PossibleXRHSInFirst =
- FirstBinOp->getRHS()->IgnoreParenImpCasts();
- Expr *PossibleXLHSInSecond =
- SecondBinOp->getLHS()->IgnoreParenImpCasts();
- llvm::FoldingSetNodeID X1Id, X2Id;
- PossibleXRHSInFirst->Profile(X1Id, Context,
- /*Canonical=*/true);
- PossibleXLHSInSecond->Profile(X2Id, Context,
- /*Canonical=*/true);
- IsUpdateExprFound = X1Id == X2Id;
- if (IsUpdateExprFound) {
- V = FirstBinOp->getLHS();
- X = SecondBinOp->getLHS();
- E = SecondBinOp->getRHS();
- UE = nullptr;
- IsXLHSInRHSPart = false;
- IsPostfixUpdate = true;
- } else {
- ErrorFound = NotASpecificExpression;
- ErrorLoc = FirstBinOp->getExprLoc();
- ErrorRange = FirstBinOp->getSourceRange();
- NoteLoc = SecondBinOp->getLHS()->getExprLoc();
- NoteRange = SecondBinOp->getRHS()->getSourceRange();
- }
- }
- }
- }
- }
- } else {
- NoteLoc = ErrorLoc = Body->getBeginLoc();
- NoteRange = ErrorRange =
- SourceRange(Body->getBeginLoc(), Body->getBeginLoc());
- ErrorFound = NotTwoSubstatements;
- }
- } else {
- NoteLoc = ErrorLoc = Body->getBeginLoc();
- NoteRange = ErrorRange =
- SourceRange(Body->getBeginLoc(), Body->getBeginLoc());
- ErrorFound = NotACompoundStatement;
- }
- if (ErrorFound != NoError) {
- Diag(ErrorLoc, diag::err_omp_atomic_capture_not_compound_statement)
- << ErrorRange;
- Diag(NoteLoc, diag::note_omp_atomic_capture) << ErrorFound << NoteRange;
- return StmtError();
- }
- if (CurContext->isDependentContext())
- UE = V = E = X = nullptr;
- }
- }
- setFunctionHasBranchProtectedScope();
- return OMPAtomicDirective::Create(Context, StartLoc, EndLoc, Clauses, AStmt,
- X, V, E, UE, IsXLHSInRHSPart,
- IsPostfixUpdate);
- }
- StmtResult Sema::ActOnOpenMPTargetDirective(ArrayRef<OMPClause *> Clauses,
- Stmt *AStmt,
- SourceLocation StartLoc,
- SourceLocation EndLoc) {
- if (!AStmt)
- return StmtError();
- auto *CS = cast<CapturedStmt>(AStmt);
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- for (int ThisCaptureLevel = getOpenMPCaptureLevels(OMPD_target);
- ThisCaptureLevel > 1; --ThisCaptureLevel) {
- CS = cast<CapturedStmt>(CS->getCapturedStmt());
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- }
- // OpenMP [2.16, Nesting of Regions]
- // If specified, a teams construct must be contained within a target
- // construct. That target construct must contain no statements or directives
- // outside of the teams construct.
- if (DSAStack->hasInnerTeamsRegion()) {
- const Stmt *S = CS->IgnoreContainers(/*IgnoreCaptured=*/true);
- bool OMPTeamsFound = true;
- if (const auto *CS = dyn_cast<CompoundStmt>(S)) {
- auto I = CS->body_begin();
- while (I != CS->body_end()) {
- const auto *OED = dyn_cast<OMPExecutableDirective>(*I);
- if (!OED || !isOpenMPTeamsDirective(OED->getDirectiveKind())) {
- OMPTeamsFound = false;
- break;
- }
- ++I;
- }
- assert(I != CS->body_end() && "Not found statement");
- S = *I;
- } else {
- const auto *OED = dyn_cast<OMPExecutableDirective>(S);
- OMPTeamsFound = OED && isOpenMPTeamsDirective(OED->getDirectiveKind());
- }
- if (!OMPTeamsFound) {
- Diag(StartLoc, diag::err_omp_target_contains_not_only_teams);
- Diag(DSAStack->getInnerTeamsRegionLoc(),
- diag::note_omp_nested_teams_construct_here);
- Diag(S->getBeginLoc(), diag::note_omp_nested_statement_here)
- << isa<OMPExecutableDirective>(S);
- return StmtError();
- }
- }
- setFunctionHasBranchProtectedScope();
- return OMPTargetDirective::Create(Context, StartLoc, EndLoc, Clauses, AStmt);
- }
- StmtResult
- Sema::ActOnOpenMPTargetParallelDirective(ArrayRef<OMPClause *> Clauses,
- Stmt *AStmt, SourceLocation StartLoc,
- SourceLocation EndLoc) {
- if (!AStmt)
- return StmtError();
- auto *CS = cast<CapturedStmt>(AStmt);
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- for (int ThisCaptureLevel = getOpenMPCaptureLevels(OMPD_target_parallel);
- ThisCaptureLevel > 1; --ThisCaptureLevel) {
- CS = cast<CapturedStmt>(CS->getCapturedStmt());
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- }
- setFunctionHasBranchProtectedScope();
- return OMPTargetParallelDirective::Create(Context, StartLoc, EndLoc, Clauses,
- AStmt);
- }
- StmtResult Sema::ActOnOpenMPTargetParallelForDirective(
- ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc,
- SourceLocation EndLoc, VarsWithInheritedDSAType &VarsWithImplicitDSA) {
- if (!AStmt)
- return StmtError();
- auto *CS = cast<CapturedStmt>(AStmt);
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- for (int ThisCaptureLevel = getOpenMPCaptureLevels(OMPD_target_parallel_for);
- ThisCaptureLevel > 1; --ThisCaptureLevel) {
- CS = cast<CapturedStmt>(CS->getCapturedStmt());
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- }
- OMPLoopDirective::HelperExprs B;
- // In presence of clause 'collapse' or 'ordered' with number of loops, it will
- // define the nested loops number.
- unsigned NestedLoopCount =
- checkOpenMPLoop(OMPD_target_parallel_for, getCollapseNumberExpr(Clauses),
- getOrderedNumberExpr(Clauses), CS, *this, *DSAStack,
- VarsWithImplicitDSA, B);
- if (NestedLoopCount == 0)
- return StmtError();
- assert((CurContext->isDependentContext() || B.builtAll()) &&
- "omp target parallel for loop exprs were not built");
- if (!CurContext->isDependentContext()) {
- // Finalize the clauses that need pre-built expressions for CodeGen.
- for (OMPClause *C : Clauses) {
- if (auto *LC = dyn_cast<OMPLinearClause>(C))
- if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef),
- B.NumIterations, *this, CurScope,
- DSAStack))
- return StmtError();
- }
- }
- setFunctionHasBranchProtectedScope();
- return OMPTargetParallelForDirective::Create(Context, StartLoc, EndLoc,
- NestedLoopCount, Clauses, AStmt,
- B, DSAStack->isCancelRegion());
- }
- /// Check for existence of a map clause in the list of clauses.
- static bool hasClauses(ArrayRef<OMPClause *> Clauses,
- const OpenMPClauseKind K) {
- return llvm::any_of(
- Clauses, [K](const OMPClause *C) { return C->getClauseKind() == K; });
- }
- template <typename... Params>
- static bool hasClauses(ArrayRef<OMPClause *> Clauses, const OpenMPClauseKind K,
- const Params... ClauseTypes) {
- return hasClauses(Clauses, K) || hasClauses(Clauses, ClauseTypes...);
- }
- StmtResult Sema::ActOnOpenMPTargetDataDirective(ArrayRef<OMPClause *> Clauses,
- Stmt *AStmt,
- SourceLocation StartLoc,
- SourceLocation EndLoc) {
- if (!AStmt)
- return StmtError();
- assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
- // OpenMP [2.10.1, Restrictions, p. 97]
- // At least one map clause must appear on the directive.
- if (!hasClauses(Clauses, OMPC_map, OMPC_use_device_ptr)) {
- Diag(StartLoc, diag::err_omp_no_clause_for_directive)
- << "'map' or 'use_device_ptr'"
- << getOpenMPDirectiveName(OMPD_target_data);
- return StmtError();
- }
- setFunctionHasBranchProtectedScope();
- return OMPTargetDataDirective::Create(Context, StartLoc, EndLoc, Clauses,
- AStmt);
- }
- StmtResult
- Sema::ActOnOpenMPTargetEnterDataDirective(ArrayRef<OMPClause *> Clauses,
- SourceLocation StartLoc,
- SourceLocation EndLoc, Stmt *AStmt) {
- if (!AStmt)
- return StmtError();
- auto *CS = cast<CapturedStmt>(AStmt);
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- for (int ThisCaptureLevel = getOpenMPCaptureLevels(OMPD_target_enter_data);
- ThisCaptureLevel > 1; --ThisCaptureLevel) {
- CS = cast<CapturedStmt>(CS->getCapturedStmt());
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- }
- // OpenMP [2.10.2, Restrictions, p. 99]
- // At least one map clause must appear on the directive.
- if (!hasClauses(Clauses, OMPC_map)) {
- Diag(StartLoc, diag::err_omp_no_clause_for_directive)
- << "'map'" << getOpenMPDirectiveName(OMPD_target_enter_data);
- return StmtError();
- }
- return OMPTargetEnterDataDirective::Create(Context, StartLoc, EndLoc, Clauses,
- AStmt);
- }
- StmtResult
- Sema::ActOnOpenMPTargetExitDataDirective(ArrayRef<OMPClause *> Clauses,
- SourceLocation StartLoc,
- SourceLocation EndLoc, Stmt *AStmt) {
- if (!AStmt)
- return StmtError();
- auto *CS = cast<CapturedStmt>(AStmt);
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- for (int ThisCaptureLevel = getOpenMPCaptureLevels(OMPD_target_exit_data);
- ThisCaptureLevel > 1; --ThisCaptureLevel) {
- CS = cast<CapturedStmt>(CS->getCapturedStmt());
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- }
- // OpenMP [2.10.3, Restrictions, p. 102]
- // At least one map clause must appear on the directive.
- if (!hasClauses(Clauses, OMPC_map)) {
- Diag(StartLoc, diag::err_omp_no_clause_for_directive)
- << "'map'" << getOpenMPDirectiveName(OMPD_target_exit_data);
- return StmtError();
- }
- return OMPTargetExitDataDirective::Create(Context, StartLoc, EndLoc, Clauses,
- AStmt);
- }
- StmtResult Sema::ActOnOpenMPTargetUpdateDirective(ArrayRef<OMPClause *> Clauses,
- SourceLocation StartLoc,
- SourceLocation EndLoc,
- Stmt *AStmt) {
- if (!AStmt)
- return StmtError();
- auto *CS = cast<CapturedStmt>(AStmt);
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- for (int ThisCaptureLevel = getOpenMPCaptureLevels(OMPD_target_update);
- ThisCaptureLevel > 1; --ThisCaptureLevel) {
- CS = cast<CapturedStmt>(CS->getCapturedStmt());
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- }
- if (!hasClauses(Clauses, OMPC_to, OMPC_from)) {
- Diag(StartLoc, diag::err_omp_at_least_one_motion_clause_required);
- return StmtError();
- }
- return OMPTargetUpdateDirective::Create(Context, StartLoc, EndLoc, Clauses,
- AStmt);
- }
- StmtResult Sema::ActOnOpenMPTeamsDirective(ArrayRef<OMPClause *> Clauses,
- Stmt *AStmt, SourceLocation StartLoc,
- SourceLocation EndLoc) {
- if (!AStmt)
- return StmtError();
- auto *CS = cast<CapturedStmt>(AStmt);
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- setFunctionHasBranchProtectedScope();
- DSAStack->setParentTeamsRegionLoc(StartLoc);
- return OMPTeamsDirective::Create(Context, StartLoc, EndLoc, Clauses, AStmt);
- }
- StmtResult
- Sema::ActOnOpenMPCancellationPointDirective(SourceLocation StartLoc,
- SourceLocation EndLoc,
- OpenMPDirectiveKind CancelRegion) {
- if (DSAStack->isParentNowaitRegion()) {
- Diag(StartLoc, diag::err_omp_parent_cancel_region_nowait) << 0;
- return StmtError();
- }
- if (DSAStack->isParentOrderedRegion()) {
- Diag(StartLoc, diag::err_omp_parent_cancel_region_ordered) << 0;
- return StmtError();
- }
- return OMPCancellationPointDirective::Create(Context, StartLoc, EndLoc,
- CancelRegion);
- }
- StmtResult Sema::ActOnOpenMPCancelDirective(ArrayRef<OMPClause *> Clauses,
- SourceLocation StartLoc,
- SourceLocation EndLoc,
- OpenMPDirectiveKind CancelRegion) {
- if (DSAStack->isParentNowaitRegion()) {
- Diag(StartLoc, diag::err_omp_parent_cancel_region_nowait) << 1;
- return StmtError();
- }
- if (DSAStack->isParentOrderedRegion()) {
- Diag(StartLoc, diag::err_omp_parent_cancel_region_ordered) << 1;
- return StmtError();
- }
- DSAStack->setParentCancelRegion(/*Cancel=*/true);
- return OMPCancelDirective::Create(Context, StartLoc, EndLoc, Clauses,
- CancelRegion);
- }
- static bool checkGrainsizeNumTasksClauses(Sema &S,
- ArrayRef<OMPClause *> Clauses) {
- const OMPClause *PrevClause = nullptr;
- bool ErrorFound = false;
- for (const OMPClause *C : Clauses) {
- if (C->getClauseKind() == OMPC_grainsize ||
- C->getClauseKind() == OMPC_num_tasks) {
- if (!PrevClause)
- PrevClause = C;
- else if (PrevClause->getClauseKind() != C->getClauseKind()) {
- S.Diag(C->getBeginLoc(),
- diag::err_omp_grainsize_num_tasks_mutually_exclusive)
- << getOpenMPClauseName(C->getClauseKind())
- << getOpenMPClauseName(PrevClause->getClauseKind());
- S.Diag(PrevClause->getBeginLoc(),
- diag::note_omp_previous_grainsize_num_tasks)
- << getOpenMPClauseName(PrevClause->getClauseKind());
- ErrorFound = true;
- }
- }
- }
- return ErrorFound;
- }
- static bool checkReductionClauseWithNogroup(Sema &S,
- ArrayRef<OMPClause *> Clauses) {
- const OMPClause *ReductionClause = nullptr;
- const OMPClause *NogroupClause = nullptr;
- for (const OMPClause *C : Clauses) {
- if (C->getClauseKind() == OMPC_reduction) {
- ReductionClause = C;
- if (NogroupClause)
- break;
- continue;
- }
- if (C->getClauseKind() == OMPC_nogroup) {
- NogroupClause = C;
- if (ReductionClause)
- break;
- continue;
- }
- }
- if (ReductionClause && NogroupClause) {
- S.Diag(ReductionClause->getBeginLoc(), diag::err_omp_reduction_with_nogroup)
- << SourceRange(NogroupClause->getBeginLoc(),
- NogroupClause->getEndLoc());
- return true;
- }
- return false;
- }
- StmtResult Sema::ActOnOpenMPTaskLoopDirective(
- ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc,
- SourceLocation EndLoc, VarsWithInheritedDSAType &VarsWithImplicitDSA) {
- if (!AStmt)
- return StmtError();
- assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
- OMPLoopDirective::HelperExprs B;
- // In presence of clause 'collapse' or 'ordered' with number of loops, it will
- // define the nested loops number.
- unsigned NestedLoopCount =
- checkOpenMPLoop(OMPD_taskloop, getCollapseNumberExpr(Clauses),
- /*OrderedLoopCountExpr=*/nullptr, AStmt, *this, *DSAStack,
- VarsWithImplicitDSA, B);
- if (NestedLoopCount == 0)
- return StmtError();
- assert((CurContext->isDependentContext() || B.builtAll()) &&
- "omp for loop exprs were not built");
- // OpenMP, [2.9.2 taskloop Construct, Restrictions]
- // The grainsize clause and num_tasks clause are mutually exclusive and may
- // not appear on the same taskloop directive.
- if (checkGrainsizeNumTasksClauses(*this, Clauses))
- return StmtError();
- // OpenMP, [2.9.2 taskloop Construct, Restrictions]
- // If a reduction clause is present on the taskloop directive, the nogroup
- // clause must not be specified.
- if (checkReductionClauseWithNogroup(*this, Clauses))
- return StmtError();
- setFunctionHasBranchProtectedScope();
- return OMPTaskLoopDirective::Create(Context, StartLoc, EndLoc,
- NestedLoopCount, Clauses, AStmt, B);
- }
- StmtResult Sema::ActOnOpenMPTaskLoopSimdDirective(
- ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc,
- SourceLocation EndLoc, VarsWithInheritedDSAType &VarsWithImplicitDSA) {
- if (!AStmt)
- return StmtError();
- assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
- OMPLoopDirective::HelperExprs B;
- // In presence of clause 'collapse' or 'ordered' with number of loops, it will
- // define the nested loops number.
- unsigned NestedLoopCount =
- checkOpenMPLoop(OMPD_taskloop_simd, getCollapseNumberExpr(Clauses),
- /*OrderedLoopCountExpr=*/nullptr, AStmt, *this, *DSAStack,
- VarsWithImplicitDSA, B);
- if (NestedLoopCount == 0)
- return StmtError();
- assert((CurContext->isDependentContext() || B.builtAll()) &&
- "omp for loop exprs were not built");
- if (!CurContext->isDependentContext()) {
- // Finalize the clauses that need pre-built expressions for CodeGen.
- for (OMPClause *C : Clauses) {
- if (auto *LC = dyn_cast<OMPLinearClause>(C))
- if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef),
- B.NumIterations, *this, CurScope,
- DSAStack))
- return StmtError();
- }
- }
- // OpenMP, [2.9.2 taskloop Construct, Restrictions]
- // The grainsize clause and num_tasks clause are mutually exclusive and may
- // not appear on the same taskloop directive.
- if (checkGrainsizeNumTasksClauses(*this, Clauses))
- return StmtError();
- // OpenMP, [2.9.2 taskloop Construct, Restrictions]
- // If a reduction clause is present on the taskloop directive, the nogroup
- // clause must not be specified.
- if (checkReductionClauseWithNogroup(*this, Clauses))
- return StmtError();
- if (checkSimdlenSafelenSpecified(*this, Clauses))
- return StmtError();
- setFunctionHasBranchProtectedScope();
- return OMPTaskLoopSimdDirective::Create(Context, StartLoc, EndLoc,
- NestedLoopCount, Clauses, AStmt, B);
- }
- StmtResult Sema::ActOnOpenMPDistributeDirective(
- ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc,
- SourceLocation EndLoc, VarsWithInheritedDSAType &VarsWithImplicitDSA) {
- if (!AStmt)
- return StmtError();
- assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
- OMPLoopDirective::HelperExprs B;
- // In presence of clause 'collapse' with number of loops, it will
- // define the nested loops number.
- unsigned NestedLoopCount =
- checkOpenMPLoop(OMPD_distribute, getCollapseNumberExpr(Clauses),
- nullptr /*ordered not a clause on distribute*/, AStmt,
- *this, *DSAStack, VarsWithImplicitDSA, B);
- if (NestedLoopCount == 0)
- return StmtError();
- assert((CurContext->isDependentContext() || B.builtAll()) &&
- "omp for loop exprs were not built");
- setFunctionHasBranchProtectedScope();
- return OMPDistributeDirective::Create(Context, StartLoc, EndLoc,
- NestedLoopCount, Clauses, AStmt, B);
- }
- StmtResult Sema::ActOnOpenMPDistributeParallelForDirective(
- ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc,
- SourceLocation EndLoc, VarsWithInheritedDSAType &VarsWithImplicitDSA) {
- if (!AStmt)
- return StmtError();
- auto *CS = cast<CapturedStmt>(AStmt);
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- for (int ThisCaptureLevel =
- getOpenMPCaptureLevels(OMPD_distribute_parallel_for);
- ThisCaptureLevel > 1; --ThisCaptureLevel) {
- CS = cast<CapturedStmt>(CS->getCapturedStmt());
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- }
- OMPLoopDirective::HelperExprs B;
- // In presence of clause 'collapse' with number of loops, it will
- // define the nested loops number.
- unsigned NestedLoopCount = checkOpenMPLoop(
- OMPD_distribute_parallel_for, getCollapseNumberExpr(Clauses),
- nullptr /*ordered not a clause on distribute*/, CS, *this, *DSAStack,
- VarsWithImplicitDSA, B);
- if (NestedLoopCount == 0)
- return StmtError();
- assert((CurContext->isDependentContext() || B.builtAll()) &&
- "omp for loop exprs were not built");
- setFunctionHasBranchProtectedScope();
- return OMPDistributeParallelForDirective::Create(
- Context, StartLoc, EndLoc, NestedLoopCount, Clauses, AStmt, B,
- DSAStack->isCancelRegion());
- }
- StmtResult Sema::ActOnOpenMPDistributeParallelForSimdDirective(
- ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc,
- SourceLocation EndLoc, VarsWithInheritedDSAType &VarsWithImplicitDSA) {
- if (!AStmt)
- return StmtError();
- auto *CS = cast<CapturedStmt>(AStmt);
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- for (int ThisCaptureLevel =
- getOpenMPCaptureLevels(OMPD_distribute_parallel_for_simd);
- ThisCaptureLevel > 1; --ThisCaptureLevel) {
- CS = cast<CapturedStmt>(CS->getCapturedStmt());
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- }
- OMPLoopDirective::HelperExprs B;
- // In presence of clause 'collapse' with number of loops, it will
- // define the nested loops number.
- unsigned NestedLoopCount = checkOpenMPLoop(
- OMPD_distribute_parallel_for_simd, getCollapseNumberExpr(Clauses),
- nullptr /*ordered not a clause on distribute*/, CS, *this, *DSAStack,
- VarsWithImplicitDSA, B);
- if (NestedLoopCount == 0)
- return StmtError();
- assert((CurContext->isDependentContext() || B.builtAll()) &&
- "omp for loop exprs were not built");
- if (!CurContext->isDependentContext()) {
- // Finalize the clauses that need pre-built expressions for CodeGen.
- for (OMPClause *C : Clauses) {
- if (auto *LC = dyn_cast<OMPLinearClause>(C))
- if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef),
- B.NumIterations, *this, CurScope,
- DSAStack))
- return StmtError();
- }
- }
- if (checkSimdlenSafelenSpecified(*this, Clauses))
- return StmtError();
- setFunctionHasBranchProtectedScope();
- return OMPDistributeParallelForSimdDirective::Create(
- Context, StartLoc, EndLoc, NestedLoopCount, Clauses, AStmt, B);
- }
- StmtResult Sema::ActOnOpenMPDistributeSimdDirective(
- ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc,
- SourceLocation EndLoc, VarsWithInheritedDSAType &VarsWithImplicitDSA) {
- if (!AStmt)
- return StmtError();
- auto *CS = cast<CapturedStmt>(AStmt);
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- for (int ThisCaptureLevel = getOpenMPCaptureLevels(OMPD_distribute_simd);
- ThisCaptureLevel > 1; --ThisCaptureLevel) {
- CS = cast<CapturedStmt>(CS->getCapturedStmt());
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- }
- OMPLoopDirective::HelperExprs B;
- // In presence of clause 'collapse' with number of loops, it will
- // define the nested loops number.
- unsigned NestedLoopCount =
- checkOpenMPLoop(OMPD_distribute_simd, getCollapseNumberExpr(Clauses),
- nullptr /*ordered not a clause on distribute*/, CS, *this,
- *DSAStack, VarsWithImplicitDSA, B);
- if (NestedLoopCount == 0)
- return StmtError();
- assert((CurContext->isDependentContext() || B.builtAll()) &&
- "omp for loop exprs were not built");
- if (!CurContext->isDependentContext()) {
- // Finalize the clauses that need pre-built expressions for CodeGen.
- for (OMPClause *C : Clauses) {
- if (auto *LC = dyn_cast<OMPLinearClause>(C))
- if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef),
- B.NumIterations, *this, CurScope,
- DSAStack))
- return StmtError();
- }
- }
- if (checkSimdlenSafelenSpecified(*this, Clauses))
- return StmtError();
- setFunctionHasBranchProtectedScope();
- return OMPDistributeSimdDirective::Create(Context, StartLoc, EndLoc,
- NestedLoopCount, Clauses, AStmt, B);
- }
- StmtResult Sema::ActOnOpenMPTargetParallelForSimdDirective(
- ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc,
- SourceLocation EndLoc, VarsWithInheritedDSAType &VarsWithImplicitDSA) {
- if (!AStmt)
- return StmtError();
- auto *CS = cast<CapturedStmt>(AStmt);
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- for (int ThisCaptureLevel = getOpenMPCaptureLevels(OMPD_target_parallel_for);
- ThisCaptureLevel > 1; --ThisCaptureLevel) {
- CS = cast<CapturedStmt>(CS->getCapturedStmt());
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- }
- OMPLoopDirective::HelperExprs B;
- // In presence of clause 'collapse' or 'ordered' with number of loops, it will
- // define the nested loops number.
- unsigned NestedLoopCount = checkOpenMPLoop(
- OMPD_target_parallel_for_simd, getCollapseNumberExpr(Clauses),
- getOrderedNumberExpr(Clauses), CS, *this, *DSAStack,
- VarsWithImplicitDSA, B);
- if (NestedLoopCount == 0)
- return StmtError();
- assert((CurContext->isDependentContext() || B.builtAll()) &&
- "omp target parallel for simd loop exprs were not built");
- if (!CurContext->isDependentContext()) {
- // Finalize the clauses that need pre-built expressions for CodeGen.
- for (OMPClause *C : Clauses) {
- if (auto *LC = dyn_cast<OMPLinearClause>(C))
- if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef),
- B.NumIterations, *this, CurScope,
- DSAStack))
- return StmtError();
- }
- }
- if (checkSimdlenSafelenSpecified(*this, Clauses))
- return StmtError();
- setFunctionHasBranchProtectedScope();
- return OMPTargetParallelForSimdDirective::Create(
- Context, StartLoc, EndLoc, NestedLoopCount, Clauses, AStmt, B);
- }
- StmtResult Sema::ActOnOpenMPTargetSimdDirective(
- ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc,
- SourceLocation EndLoc, VarsWithInheritedDSAType &VarsWithImplicitDSA) {
- if (!AStmt)
- return StmtError();
- auto *CS = cast<CapturedStmt>(AStmt);
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- for (int ThisCaptureLevel = getOpenMPCaptureLevels(OMPD_target_simd);
- ThisCaptureLevel > 1; --ThisCaptureLevel) {
- CS = cast<CapturedStmt>(CS->getCapturedStmt());
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- }
- OMPLoopDirective::HelperExprs B;
- // In presence of clause 'collapse' with number of loops, it will define the
- // nested loops number.
- unsigned NestedLoopCount =
- checkOpenMPLoop(OMPD_target_simd, getCollapseNumberExpr(Clauses),
- getOrderedNumberExpr(Clauses), CS, *this, *DSAStack,
- VarsWithImplicitDSA, B);
- if (NestedLoopCount == 0)
- return StmtError();
- assert((CurContext->isDependentContext() || B.builtAll()) &&
- "omp target simd loop exprs were not built");
- if (!CurContext->isDependentContext()) {
- // Finalize the clauses that need pre-built expressions for CodeGen.
- for (OMPClause *C : Clauses) {
- if (auto *LC = dyn_cast<OMPLinearClause>(C))
- if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef),
- B.NumIterations, *this, CurScope,
- DSAStack))
- return StmtError();
- }
- }
- if (checkSimdlenSafelenSpecified(*this, Clauses))
- return StmtError();
- setFunctionHasBranchProtectedScope();
- return OMPTargetSimdDirective::Create(Context, StartLoc, EndLoc,
- NestedLoopCount, Clauses, AStmt, B);
- }
- StmtResult Sema::ActOnOpenMPTeamsDistributeDirective(
- ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc,
- SourceLocation EndLoc, VarsWithInheritedDSAType &VarsWithImplicitDSA) {
- if (!AStmt)
- return StmtError();
- auto *CS = cast<CapturedStmt>(AStmt);
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- for (int ThisCaptureLevel = getOpenMPCaptureLevels(OMPD_teams_distribute);
- ThisCaptureLevel > 1; --ThisCaptureLevel) {
- CS = cast<CapturedStmt>(CS->getCapturedStmt());
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- }
- OMPLoopDirective::HelperExprs B;
- // In presence of clause 'collapse' with number of loops, it will
- // define the nested loops number.
- unsigned NestedLoopCount =
- checkOpenMPLoop(OMPD_teams_distribute, getCollapseNumberExpr(Clauses),
- nullptr /*ordered not a clause on distribute*/, CS, *this,
- *DSAStack, VarsWithImplicitDSA, B);
- if (NestedLoopCount == 0)
- return StmtError();
- assert((CurContext->isDependentContext() || B.builtAll()) &&
- "omp teams distribute loop exprs were not built");
- setFunctionHasBranchProtectedScope();
- DSAStack->setParentTeamsRegionLoc(StartLoc);
- return OMPTeamsDistributeDirective::Create(
- Context, StartLoc, EndLoc, NestedLoopCount, Clauses, AStmt, B);
- }
- StmtResult Sema::ActOnOpenMPTeamsDistributeSimdDirective(
- ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc,
- SourceLocation EndLoc, VarsWithInheritedDSAType &VarsWithImplicitDSA) {
- if (!AStmt)
- return StmtError();
- auto *CS = cast<CapturedStmt>(AStmt);
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- for (int ThisCaptureLevel =
- getOpenMPCaptureLevels(OMPD_teams_distribute_simd);
- ThisCaptureLevel > 1; --ThisCaptureLevel) {
- CS = cast<CapturedStmt>(CS->getCapturedStmt());
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- }
- OMPLoopDirective::HelperExprs B;
- // In presence of clause 'collapse' with number of loops, it will
- // define the nested loops number.
- unsigned NestedLoopCount = checkOpenMPLoop(
- OMPD_teams_distribute_simd, getCollapseNumberExpr(Clauses),
- nullptr /*ordered not a clause on distribute*/, CS, *this, *DSAStack,
- VarsWithImplicitDSA, B);
- if (NestedLoopCount == 0)
- return StmtError();
- assert((CurContext->isDependentContext() || B.builtAll()) &&
- "omp teams distribute simd loop exprs were not built");
- if (!CurContext->isDependentContext()) {
- // Finalize the clauses that need pre-built expressions for CodeGen.
- for (OMPClause *C : Clauses) {
- if (auto *LC = dyn_cast<OMPLinearClause>(C))
- if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef),
- B.NumIterations, *this, CurScope,
- DSAStack))
- return StmtError();
- }
- }
- if (checkSimdlenSafelenSpecified(*this, Clauses))
- return StmtError();
- setFunctionHasBranchProtectedScope();
- DSAStack->setParentTeamsRegionLoc(StartLoc);
- return OMPTeamsDistributeSimdDirective::Create(
- Context, StartLoc, EndLoc, NestedLoopCount, Clauses, AStmt, B);
- }
- StmtResult Sema::ActOnOpenMPTeamsDistributeParallelForSimdDirective(
- ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc,
- SourceLocation EndLoc, VarsWithInheritedDSAType &VarsWithImplicitDSA) {
- if (!AStmt)
- return StmtError();
- auto *CS = cast<CapturedStmt>(AStmt);
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- for (int ThisCaptureLevel =
- getOpenMPCaptureLevels(OMPD_teams_distribute_parallel_for_simd);
- ThisCaptureLevel > 1; --ThisCaptureLevel) {
- CS = cast<CapturedStmt>(CS->getCapturedStmt());
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- }
- OMPLoopDirective::HelperExprs B;
- // In presence of clause 'collapse' with number of loops, it will
- // define the nested loops number.
- unsigned NestedLoopCount = checkOpenMPLoop(
- OMPD_teams_distribute_parallel_for_simd, getCollapseNumberExpr(Clauses),
- nullptr /*ordered not a clause on distribute*/, CS, *this, *DSAStack,
- VarsWithImplicitDSA, B);
- if (NestedLoopCount == 0)
- return StmtError();
- assert((CurContext->isDependentContext() || B.builtAll()) &&
- "omp for loop exprs were not built");
- if (!CurContext->isDependentContext()) {
- // Finalize the clauses that need pre-built expressions for CodeGen.
- for (OMPClause *C : Clauses) {
- if (auto *LC = dyn_cast<OMPLinearClause>(C))
- if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef),
- B.NumIterations, *this, CurScope,
- DSAStack))
- return StmtError();
- }
- }
- if (checkSimdlenSafelenSpecified(*this, Clauses))
- return StmtError();
- setFunctionHasBranchProtectedScope();
- DSAStack->setParentTeamsRegionLoc(StartLoc);
- return OMPTeamsDistributeParallelForSimdDirective::Create(
- Context, StartLoc, EndLoc, NestedLoopCount, Clauses, AStmt, B);
- }
- StmtResult Sema::ActOnOpenMPTeamsDistributeParallelForDirective(
- ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc,
- SourceLocation EndLoc, VarsWithInheritedDSAType &VarsWithImplicitDSA) {
- if (!AStmt)
- return StmtError();
- auto *CS = cast<CapturedStmt>(AStmt);
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- for (int ThisCaptureLevel =
- getOpenMPCaptureLevels(OMPD_teams_distribute_parallel_for);
- ThisCaptureLevel > 1; --ThisCaptureLevel) {
- CS = cast<CapturedStmt>(CS->getCapturedStmt());
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- }
- OMPLoopDirective::HelperExprs B;
- // In presence of clause 'collapse' with number of loops, it will
- // define the nested loops number.
- unsigned NestedLoopCount = checkOpenMPLoop(
- OMPD_teams_distribute_parallel_for, getCollapseNumberExpr(Clauses),
- nullptr /*ordered not a clause on distribute*/, CS, *this, *DSAStack,
- VarsWithImplicitDSA, B);
- if (NestedLoopCount == 0)
- return StmtError();
- assert((CurContext->isDependentContext() || B.builtAll()) &&
- "omp for loop exprs were not built");
- setFunctionHasBranchProtectedScope();
- DSAStack->setParentTeamsRegionLoc(StartLoc);
- return OMPTeamsDistributeParallelForDirective::Create(
- Context, StartLoc, EndLoc, NestedLoopCount, Clauses, AStmt, B,
- DSAStack->isCancelRegion());
- }
- StmtResult Sema::ActOnOpenMPTargetTeamsDirective(ArrayRef<OMPClause *> Clauses,
- Stmt *AStmt,
- SourceLocation StartLoc,
- SourceLocation EndLoc) {
- if (!AStmt)
- return StmtError();
- auto *CS = cast<CapturedStmt>(AStmt);
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- for (int ThisCaptureLevel = getOpenMPCaptureLevels(OMPD_target_teams);
- ThisCaptureLevel > 1; --ThisCaptureLevel) {
- CS = cast<CapturedStmt>(CS->getCapturedStmt());
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- }
- setFunctionHasBranchProtectedScope();
- return OMPTargetTeamsDirective::Create(Context, StartLoc, EndLoc, Clauses,
- AStmt);
- }
- StmtResult Sema::ActOnOpenMPTargetTeamsDistributeDirective(
- ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc,
- SourceLocation EndLoc, VarsWithInheritedDSAType &VarsWithImplicitDSA) {
- if (!AStmt)
- return StmtError();
- auto *CS = cast<CapturedStmt>(AStmt);
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- for (int ThisCaptureLevel =
- getOpenMPCaptureLevels(OMPD_target_teams_distribute);
- ThisCaptureLevel > 1; --ThisCaptureLevel) {
- CS = cast<CapturedStmt>(CS->getCapturedStmt());
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- }
- OMPLoopDirective::HelperExprs B;
- // In presence of clause 'collapse' with number of loops, it will
- // define the nested loops number.
- unsigned NestedLoopCount = checkOpenMPLoop(
- OMPD_target_teams_distribute, getCollapseNumberExpr(Clauses),
- nullptr /*ordered not a clause on distribute*/, CS, *this, *DSAStack,
- VarsWithImplicitDSA, B);
- if (NestedLoopCount == 0)
- return StmtError();
- assert((CurContext->isDependentContext() || B.builtAll()) &&
- "omp target teams distribute loop exprs were not built");
- setFunctionHasBranchProtectedScope();
- return OMPTargetTeamsDistributeDirective::Create(
- Context, StartLoc, EndLoc, NestedLoopCount, Clauses, AStmt, B);
- }
- StmtResult Sema::ActOnOpenMPTargetTeamsDistributeParallelForDirective(
- ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc,
- SourceLocation EndLoc, VarsWithInheritedDSAType &VarsWithImplicitDSA) {
- if (!AStmt)
- return StmtError();
- auto *CS = cast<CapturedStmt>(AStmt);
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- for (int ThisCaptureLevel =
- getOpenMPCaptureLevels(OMPD_target_teams_distribute_parallel_for);
- ThisCaptureLevel > 1; --ThisCaptureLevel) {
- CS = cast<CapturedStmt>(CS->getCapturedStmt());
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- }
- OMPLoopDirective::HelperExprs B;
- // In presence of clause 'collapse' with number of loops, it will
- // define the nested loops number.
- unsigned NestedLoopCount = checkOpenMPLoop(
- OMPD_target_teams_distribute_parallel_for, getCollapseNumberExpr(Clauses),
- nullptr /*ordered not a clause on distribute*/, CS, *this, *DSAStack,
- VarsWithImplicitDSA, B);
- if (NestedLoopCount == 0)
- return StmtError();
- assert((CurContext->isDependentContext() || B.builtAll()) &&
- "omp target teams distribute parallel for loop exprs were not built");
- if (!CurContext->isDependentContext()) {
- // Finalize the clauses that need pre-built expressions for CodeGen.
- for (OMPClause *C : Clauses) {
- if (auto *LC = dyn_cast<OMPLinearClause>(C))
- if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef),
- B.NumIterations, *this, CurScope,
- DSAStack))
- return StmtError();
- }
- }
- setFunctionHasBranchProtectedScope();
- return OMPTargetTeamsDistributeParallelForDirective::Create(
- Context, StartLoc, EndLoc, NestedLoopCount, Clauses, AStmt, B,
- DSAStack->isCancelRegion());
- }
- StmtResult Sema::ActOnOpenMPTargetTeamsDistributeParallelForSimdDirective(
- ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc,
- SourceLocation EndLoc, VarsWithInheritedDSAType &VarsWithImplicitDSA) {
- if (!AStmt)
- return StmtError();
- auto *CS = cast<CapturedStmt>(AStmt);
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- for (int ThisCaptureLevel = getOpenMPCaptureLevels(
- OMPD_target_teams_distribute_parallel_for_simd);
- ThisCaptureLevel > 1; --ThisCaptureLevel) {
- CS = cast<CapturedStmt>(CS->getCapturedStmt());
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- }
- OMPLoopDirective::HelperExprs B;
- // In presence of clause 'collapse' with number of loops, it will
- // define the nested loops number.
- unsigned NestedLoopCount =
- checkOpenMPLoop(OMPD_target_teams_distribute_parallel_for_simd,
- getCollapseNumberExpr(Clauses),
- nullptr /*ordered not a clause on distribute*/, CS, *this,
- *DSAStack, VarsWithImplicitDSA, B);
- if (NestedLoopCount == 0)
- return StmtError();
- assert((CurContext->isDependentContext() || B.builtAll()) &&
- "omp target teams distribute parallel for simd loop exprs were not "
- "built");
- if (!CurContext->isDependentContext()) {
- // Finalize the clauses that need pre-built expressions for CodeGen.
- for (OMPClause *C : Clauses) {
- if (auto *LC = dyn_cast<OMPLinearClause>(C))
- if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef),
- B.NumIterations, *this, CurScope,
- DSAStack))
- return StmtError();
- }
- }
- if (checkSimdlenSafelenSpecified(*this, Clauses))
- return StmtError();
- setFunctionHasBranchProtectedScope();
- return OMPTargetTeamsDistributeParallelForSimdDirective::Create(
- Context, StartLoc, EndLoc, NestedLoopCount, Clauses, AStmt, B);
- }
- StmtResult Sema::ActOnOpenMPTargetTeamsDistributeSimdDirective(
- ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc,
- SourceLocation EndLoc, VarsWithInheritedDSAType &VarsWithImplicitDSA) {
- if (!AStmt)
- return StmtError();
- auto *CS = cast<CapturedStmt>(AStmt);
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- for (int ThisCaptureLevel =
- getOpenMPCaptureLevels(OMPD_target_teams_distribute_simd);
- ThisCaptureLevel > 1; --ThisCaptureLevel) {
- CS = cast<CapturedStmt>(CS->getCapturedStmt());
- // 1.2.2 OpenMP Language Terminology
- // Structured block - An executable statement with a single entry at the
- // top and a single exit at the bottom.
- // The point of exit cannot be a branch out of the structured block.
- // longjmp() and throw() must not violate the entry/exit criteria.
- CS->getCapturedDecl()->setNothrow();
- }
- OMPLoopDirective::HelperExprs B;
- // In presence of clause 'collapse' with number of loops, it will
- // define the nested loops number.
- unsigned NestedLoopCount = checkOpenMPLoop(
- OMPD_target_teams_distribute_simd, getCollapseNumberExpr(Clauses),
- nullptr /*ordered not a clause on distribute*/, CS, *this, *DSAStack,
- VarsWithImplicitDSA, B);
- if (NestedLoopCount == 0)
- return StmtError();
- assert((CurContext->isDependentContext() || B.builtAll()) &&
- "omp target teams distribute simd loop exprs were not built");
- if (!CurContext->isDependentContext()) {
- // Finalize the clauses that need pre-built expressions for CodeGen.
- for (OMPClause *C : Clauses) {
- if (auto *LC = dyn_cast<OMPLinearClause>(C))
- if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef),
- B.NumIterations, *this, CurScope,
- DSAStack))
- return StmtError();
- }
- }
- if (checkSimdlenSafelenSpecified(*this, Clauses))
- return StmtError();
- setFunctionHasBranchProtectedScope();
- return OMPTargetTeamsDistributeSimdDirective::Create(
- Context, StartLoc, EndLoc, NestedLoopCount, Clauses, AStmt, B);
- }
- OMPClause *Sema::ActOnOpenMPSingleExprClause(OpenMPClauseKind Kind, Expr *Expr,
- SourceLocation StartLoc,
- SourceLocation LParenLoc,
- SourceLocation EndLoc) {
- OMPClause *Res = nullptr;
- switch (Kind) {
- case OMPC_final:
- Res = ActOnOpenMPFinalClause(Expr, StartLoc, LParenLoc, EndLoc);
- break;
- case OMPC_num_threads:
- Res = ActOnOpenMPNumThreadsClause(Expr, StartLoc, LParenLoc, EndLoc);
- break;
- case OMPC_safelen:
- Res = ActOnOpenMPSafelenClause(Expr, StartLoc, LParenLoc, EndLoc);
- break;
- case OMPC_simdlen:
- Res = ActOnOpenMPSimdlenClause(Expr, StartLoc, LParenLoc, EndLoc);
- break;
- case OMPC_collapse:
- Res = ActOnOpenMPCollapseClause(Expr, StartLoc, LParenLoc, EndLoc);
- break;
- case OMPC_ordered:
- Res = ActOnOpenMPOrderedClause(StartLoc, EndLoc, LParenLoc, Expr);
- break;
- case OMPC_device:
- Res = ActOnOpenMPDeviceClause(Expr, StartLoc, LParenLoc, EndLoc);
- break;
- case OMPC_num_teams:
- Res = ActOnOpenMPNumTeamsClause(Expr, StartLoc, LParenLoc, EndLoc);
- break;
- case OMPC_thread_limit:
- Res = ActOnOpenMPThreadLimitClause(Expr, StartLoc, LParenLoc, EndLoc);
- break;
- case OMPC_priority:
- Res = ActOnOpenMPPriorityClause(Expr, StartLoc, LParenLoc, EndLoc);
- break;
- case OMPC_grainsize:
- Res = ActOnOpenMPGrainsizeClause(Expr, StartLoc, LParenLoc, EndLoc);
- break;
- case OMPC_num_tasks:
- Res = ActOnOpenMPNumTasksClause(Expr, StartLoc, LParenLoc, EndLoc);
- break;
- case OMPC_hint:
- Res = ActOnOpenMPHintClause(Expr, StartLoc, LParenLoc, EndLoc);
- break;
- case OMPC_if:
- case OMPC_default:
- case OMPC_proc_bind:
- case OMPC_schedule:
- 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_nowait:
- case OMPC_untied:
- case OMPC_mergeable:
- case OMPC_threadprivate:
- case OMPC_flush:
- case OMPC_read:
- case OMPC_write:
- case OMPC_update:
- case OMPC_capture:
- case OMPC_seq_cst:
- case OMPC_depend:
- case OMPC_threads:
- case OMPC_simd:
- case OMPC_map:
- case OMPC_nogroup:
- case OMPC_dist_schedule:
- case OMPC_defaultmap:
- case OMPC_unknown:
- case OMPC_uniform:
- case OMPC_to:
- case OMPC_from:
- case OMPC_use_device_ptr:
- case OMPC_is_device_ptr:
- case OMPC_unified_address:
- case OMPC_unified_shared_memory:
- case OMPC_reverse_offload:
- case OMPC_dynamic_allocators:
- case OMPC_atomic_default_mem_order:
- llvm_unreachable("Clause is not allowed.");
- }
- return Res;
- }
- // An OpenMP directive such as 'target parallel' has two captured regions:
- // for the 'target' and 'parallel' respectively. This function returns
- // the region in which to capture expressions associated with a clause.
- // A return value of OMPD_unknown signifies that the expression should not
- // be captured.
- static OpenMPDirectiveKind getOpenMPCaptureRegionForClause(
- OpenMPDirectiveKind DKind, OpenMPClauseKind CKind,
- OpenMPDirectiveKind NameModifier = OMPD_unknown) {
- OpenMPDirectiveKind CaptureRegion = OMPD_unknown;
- switch (CKind) {
- case OMPC_if:
- switch (DKind) {
- case OMPD_target_parallel:
- case OMPD_target_parallel_for:
- case OMPD_target_parallel_for_simd:
- // If this clause applies to the nested 'parallel' region, capture within
- // the 'target' region, otherwise do not capture.
- if (NameModifier == OMPD_unknown || NameModifier == OMPD_parallel)
- CaptureRegion = OMPD_target;
- break;
- case OMPD_target_teams_distribute_parallel_for:
- case OMPD_target_teams_distribute_parallel_for_simd:
- // If this clause applies to the nested 'parallel' region, capture within
- // the 'teams' region, otherwise do not capture.
- if (NameModifier == OMPD_unknown || NameModifier == OMPD_parallel)
- CaptureRegion = OMPD_teams;
- break;
- case OMPD_teams_distribute_parallel_for:
- case OMPD_teams_distribute_parallel_for_simd:
- CaptureRegion = OMPD_teams;
- break;
- case OMPD_target_update:
- case OMPD_target_enter_data:
- case OMPD_target_exit_data:
- CaptureRegion = OMPD_task;
- break;
- case OMPD_cancel:
- case OMPD_parallel:
- case OMPD_parallel_sections:
- case OMPD_parallel_for:
- case OMPD_parallel_for_simd:
- case OMPD_target:
- case OMPD_target_simd:
- case OMPD_target_teams:
- case OMPD_target_teams_distribute:
- case OMPD_target_teams_distribute_simd:
- case OMPD_distribute_parallel_for:
- case OMPD_distribute_parallel_for_simd:
- case OMPD_task:
- case OMPD_taskloop:
- case OMPD_taskloop_simd:
- case OMPD_target_data:
- // Do not capture if-clause expressions.
- break;
- case OMPD_threadprivate:
- case OMPD_taskyield:
- case OMPD_barrier:
- case OMPD_taskwait:
- case OMPD_cancellation_point:
- case OMPD_flush:
- case OMPD_declare_reduction:
- case OMPD_declare_simd:
- case OMPD_declare_target:
- case OMPD_end_declare_target:
- case OMPD_teams:
- case OMPD_simd:
- case OMPD_for:
- case OMPD_for_simd:
- case OMPD_sections:
- case OMPD_section:
- case OMPD_single:
- case OMPD_master:
- case OMPD_critical:
- case OMPD_taskgroup:
- case OMPD_distribute:
- case OMPD_ordered:
- case OMPD_atomic:
- case OMPD_distribute_simd:
- case OMPD_teams_distribute:
- case OMPD_teams_distribute_simd:
- case OMPD_requires:
- llvm_unreachable("Unexpected OpenMP directive with if-clause");
- case OMPD_unknown:
- llvm_unreachable("Unknown OpenMP directive");
- }
- break;
- case OMPC_num_threads:
- switch (DKind) {
- case OMPD_target_parallel:
- case OMPD_target_parallel_for:
- case OMPD_target_parallel_for_simd:
- CaptureRegion = OMPD_target;
- break;
- case OMPD_teams_distribute_parallel_for:
- case OMPD_teams_distribute_parallel_for_simd:
- case OMPD_target_teams_distribute_parallel_for:
- case OMPD_target_teams_distribute_parallel_for_simd:
- CaptureRegion = OMPD_teams;
- break;
- case OMPD_parallel:
- case OMPD_parallel_sections:
- case OMPD_parallel_for:
- case OMPD_parallel_for_simd:
- case OMPD_distribute_parallel_for:
- case OMPD_distribute_parallel_for_simd:
- // Do not capture num_threads-clause expressions.
- break;
- case OMPD_target_data:
- case OMPD_target_enter_data:
- case OMPD_target_exit_data:
- case OMPD_target_update:
- case OMPD_target:
- case OMPD_target_simd:
- case OMPD_target_teams:
- case OMPD_target_teams_distribute:
- case OMPD_target_teams_distribute_simd:
- case OMPD_cancel:
- case OMPD_task:
- case OMPD_taskloop:
- case OMPD_taskloop_simd:
- case OMPD_threadprivate:
- case OMPD_taskyield:
- case OMPD_barrier:
- case OMPD_taskwait:
- case OMPD_cancellation_point:
- case OMPD_flush:
- case OMPD_declare_reduction:
- case OMPD_declare_simd:
- case OMPD_declare_target:
- case OMPD_end_declare_target:
- case OMPD_teams:
- case OMPD_simd:
- case OMPD_for:
- case OMPD_for_simd:
- case OMPD_sections:
- case OMPD_section:
- case OMPD_single:
- case OMPD_master:
- case OMPD_critical:
- case OMPD_taskgroup:
- case OMPD_distribute:
- case OMPD_ordered:
- case OMPD_atomic:
- case OMPD_distribute_simd:
- case OMPD_teams_distribute:
- case OMPD_teams_distribute_simd:
- case OMPD_requires:
- llvm_unreachable("Unexpected OpenMP directive with num_threads-clause");
- case OMPD_unknown:
- llvm_unreachable("Unknown OpenMP directive");
- }
- break;
- case OMPC_num_teams:
- switch (DKind) {
- case OMPD_target_teams:
- case OMPD_target_teams_distribute:
- case OMPD_target_teams_distribute_simd:
- case OMPD_target_teams_distribute_parallel_for:
- case OMPD_target_teams_distribute_parallel_for_simd:
- CaptureRegion = OMPD_target;
- break;
- case OMPD_teams_distribute_parallel_for:
- case OMPD_teams_distribute_parallel_for_simd:
- case OMPD_teams:
- case OMPD_teams_distribute:
- case OMPD_teams_distribute_simd:
- // Do not capture num_teams-clause expressions.
- break;
- case OMPD_distribute_parallel_for:
- case OMPD_distribute_parallel_for_simd:
- case OMPD_task:
- case OMPD_taskloop:
- case OMPD_taskloop_simd:
- case OMPD_target_data:
- case OMPD_target_enter_data:
- case OMPD_target_exit_data:
- case OMPD_target_update:
- case OMPD_cancel:
- case OMPD_parallel:
- case OMPD_parallel_sections:
- case OMPD_parallel_for:
- case OMPD_parallel_for_simd:
- case OMPD_target:
- case OMPD_target_simd:
- case OMPD_target_parallel:
- case OMPD_target_parallel_for:
- case OMPD_target_parallel_for_simd:
- case OMPD_threadprivate:
- case OMPD_taskyield:
- case OMPD_barrier:
- case OMPD_taskwait:
- case OMPD_cancellation_point:
- case OMPD_flush:
- case OMPD_declare_reduction:
- case OMPD_declare_simd:
- case OMPD_declare_target:
- case OMPD_end_declare_target:
- case OMPD_simd:
- case OMPD_for:
- case OMPD_for_simd:
- case OMPD_sections:
- case OMPD_section:
- case OMPD_single:
- case OMPD_master:
- case OMPD_critical:
- case OMPD_taskgroup:
- case OMPD_distribute:
- case OMPD_ordered:
- case OMPD_atomic:
- case OMPD_distribute_simd:
- case OMPD_requires:
- llvm_unreachable("Unexpected OpenMP directive with num_teams-clause");
- case OMPD_unknown:
- llvm_unreachable("Unknown OpenMP directive");
- }
- break;
- case OMPC_thread_limit:
- switch (DKind) {
- case OMPD_target_teams:
- case OMPD_target_teams_distribute:
- case OMPD_target_teams_distribute_simd:
- case OMPD_target_teams_distribute_parallel_for:
- case OMPD_target_teams_distribute_parallel_for_simd:
- CaptureRegion = OMPD_target;
- break;
- case OMPD_teams_distribute_parallel_for:
- case OMPD_teams_distribute_parallel_for_simd:
- case OMPD_teams:
- case OMPD_teams_distribute:
- case OMPD_teams_distribute_simd:
- // Do not capture thread_limit-clause expressions.
- break;
- case OMPD_distribute_parallel_for:
- case OMPD_distribute_parallel_for_simd:
- case OMPD_task:
- case OMPD_taskloop:
- case OMPD_taskloop_simd:
- case OMPD_target_data:
- case OMPD_target_enter_data:
- case OMPD_target_exit_data:
- case OMPD_target_update:
- case OMPD_cancel:
- case OMPD_parallel:
- case OMPD_parallel_sections:
- case OMPD_parallel_for:
- case OMPD_parallel_for_simd:
- case OMPD_target:
- case OMPD_target_simd:
- case OMPD_target_parallel:
- case OMPD_target_parallel_for:
- case OMPD_target_parallel_for_simd:
- case OMPD_threadprivate:
- case OMPD_taskyield:
- case OMPD_barrier:
- case OMPD_taskwait:
- case OMPD_cancellation_point:
- case OMPD_flush:
- case OMPD_declare_reduction:
- case OMPD_declare_simd:
- case OMPD_declare_target:
- case OMPD_end_declare_target:
- case OMPD_simd:
- case OMPD_for:
- case OMPD_for_simd:
- case OMPD_sections:
- case OMPD_section:
- case OMPD_single:
- case OMPD_master:
- case OMPD_critical:
- case OMPD_taskgroup:
- case OMPD_distribute:
- case OMPD_ordered:
- case OMPD_atomic:
- case OMPD_distribute_simd:
- case OMPD_requires:
- llvm_unreachable("Unexpected OpenMP directive with thread_limit-clause");
- case OMPD_unknown:
- llvm_unreachable("Unknown OpenMP directive");
- }
- break;
- case OMPC_schedule:
- switch (DKind) {
- case OMPD_parallel_for:
- case OMPD_parallel_for_simd:
- case OMPD_distribute_parallel_for:
- case OMPD_distribute_parallel_for_simd:
- case OMPD_teams_distribute_parallel_for:
- case OMPD_teams_distribute_parallel_for_simd:
- case OMPD_target_parallel_for:
- case OMPD_target_parallel_for_simd:
- case OMPD_target_teams_distribute_parallel_for:
- case OMPD_target_teams_distribute_parallel_for_simd:
- CaptureRegion = OMPD_parallel;
- break;
- case OMPD_for:
- case OMPD_for_simd:
- // Do not capture schedule-clause expressions.
- break;
- case OMPD_task:
- case OMPD_taskloop:
- case OMPD_taskloop_simd:
- case OMPD_target_data:
- case OMPD_target_enter_data:
- case OMPD_target_exit_data:
- case OMPD_target_update:
- case OMPD_teams:
- case OMPD_teams_distribute:
- case OMPD_teams_distribute_simd:
- case OMPD_target_teams_distribute:
- case OMPD_target_teams_distribute_simd:
- case OMPD_target:
- case OMPD_target_simd:
- case OMPD_target_parallel:
- case OMPD_cancel:
- case OMPD_parallel:
- case OMPD_parallel_sections:
- case OMPD_threadprivate:
- case OMPD_taskyield:
- case OMPD_barrier:
- case OMPD_taskwait:
- case OMPD_cancellation_point:
- case OMPD_flush:
- case OMPD_declare_reduction:
- case OMPD_declare_simd:
- case OMPD_declare_target:
- case OMPD_end_declare_target:
- case OMPD_simd:
- case OMPD_sections:
- case OMPD_section:
- case OMPD_single:
- case OMPD_master:
- case OMPD_critical:
- case OMPD_taskgroup:
- case OMPD_distribute:
- case OMPD_ordered:
- case OMPD_atomic:
- case OMPD_distribute_simd:
- case OMPD_target_teams:
- case OMPD_requires:
- llvm_unreachable("Unexpected OpenMP directive with schedule clause");
- case OMPD_unknown:
- llvm_unreachable("Unknown OpenMP directive");
- }
- break;
- case OMPC_dist_schedule:
- switch (DKind) {
- case OMPD_teams_distribute_parallel_for:
- case OMPD_teams_distribute_parallel_for_simd:
- case OMPD_teams_distribute:
- case OMPD_teams_distribute_simd:
- case OMPD_target_teams_distribute_parallel_for:
- case OMPD_target_teams_distribute_parallel_for_simd:
- case OMPD_target_teams_distribute:
- case OMPD_target_teams_distribute_simd:
- CaptureRegion = OMPD_teams;
- break;
- case OMPD_distribute_parallel_for:
- case OMPD_distribute_parallel_for_simd:
- case OMPD_distribute:
- case OMPD_distribute_simd:
- // Do not capture thread_limit-clause expressions.
- break;
- case OMPD_parallel_for:
- case OMPD_parallel_for_simd:
- case OMPD_target_parallel_for_simd:
- case OMPD_target_parallel_for:
- case OMPD_task:
- case OMPD_taskloop:
- case OMPD_taskloop_simd:
- case OMPD_target_data:
- case OMPD_target_enter_data:
- case OMPD_target_exit_data:
- case OMPD_target_update:
- case OMPD_teams:
- case OMPD_target:
- case OMPD_target_simd:
- case OMPD_target_parallel:
- case OMPD_cancel:
- case OMPD_parallel:
- case OMPD_parallel_sections:
- case OMPD_threadprivate:
- case OMPD_taskyield:
- case OMPD_barrier:
- case OMPD_taskwait:
- case OMPD_cancellation_point:
- case OMPD_flush:
- case OMPD_declare_reduction:
- case OMPD_declare_simd:
- case OMPD_declare_target:
- case OMPD_end_declare_target:
- case OMPD_simd:
- case OMPD_for:
- case OMPD_for_simd:
- case OMPD_sections:
- case OMPD_section:
- case OMPD_single:
- case OMPD_master:
- case OMPD_critical:
- case OMPD_taskgroup:
- case OMPD_ordered:
- case OMPD_atomic:
- case OMPD_target_teams:
- case OMPD_requires:
- llvm_unreachable("Unexpected OpenMP directive with schedule clause");
- case OMPD_unknown:
- llvm_unreachable("Unknown OpenMP directive");
- }
- break;
- case OMPC_device:
- switch (DKind) {
- case OMPD_target_update:
- case OMPD_target_enter_data:
- case OMPD_target_exit_data:
- case OMPD_target:
- case OMPD_target_simd:
- case OMPD_target_teams:
- case OMPD_target_parallel:
- case OMPD_target_teams_distribute:
- case OMPD_target_teams_distribute_simd:
- case OMPD_target_parallel_for:
- case OMPD_target_parallel_for_simd:
- case OMPD_target_teams_distribute_parallel_for:
- case OMPD_target_teams_distribute_parallel_for_simd:
- CaptureRegion = OMPD_task;
- break;
- case OMPD_target_data:
- // Do not capture device-clause expressions.
- break;
- case OMPD_teams_distribute_parallel_for:
- case OMPD_teams_distribute_parallel_for_simd:
- case OMPD_teams:
- case OMPD_teams_distribute:
- case OMPD_teams_distribute_simd:
- case OMPD_distribute_parallel_for:
- case OMPD_distribute_parallel_for_simd:
- case OMPD_task:
- case OMPD_taskloop:
- case OMPD_taskloop_simd:
- case OMPD_cancel:
- case OMPD_parallel:
- case OMPD_parallel_sections:
- case OMPD_parallel_for:
- case OMPD_parallel_for_simd:
- case OMPD_threadprivate:
- case OMPD_taskyield:
- case OMPD_barrier:
- case OMPD_taskwait:
- case OMPD_cancellation_point:
- case OMPD_flush:
- case OMPD_declare_reduction:
- case OMPD_declare_simd:
- case OMPD_declare_target:
- case OMPD_end_declare_target:
- case OMPD_simd:
- case OMPD_for:
- case OMPD_for_simd:
- case OMPD_sections:
- case OMPD_section:
- case OMPD_single:
- case OMPD_master:
- case OMPD_critical:
- case OMPD_taskgroup:
- case OMPD_distribute:
- case OMPD_ordered:
- case OMPD_atomic:
- case OMPD_distribute_simd:
- case OMPD_requires:
- llvm_unreachable("Unexpected OpenMP directive with num_teams-clause");
- case OMPD_unknown:
- llvm_unreachable("Unknown OpenMP directive");
- }
- break;
- case OMPC_firstprivate:
- case OMPC_lastprivate:
- case OMPC_reduction:
- case OMPC_task_reduction:
- case OMPC_in_reduction:
- case OMPC_linear:
- case OMPC_default:
- case OMPC_proc_bind:
- case OMPC_final:
- case OMPC_safelen:
- case OMPC_simdlen:
- case OMPC_collapse:
- case OMPC_private:
- case OMPC_shared:
- case OMPC_aligned:
- case OMPC_copyin:
- case OMPC_copyprivate:
- case OMPC_ordered:
- case OMPC_nowait:
- case OMPC_untied:
- case OMPC_mergeable:
- case OMPC_threadprivate:
- case OMPC_flush:
- case OMPC_read:
- case OMPC_write:
- case OMPC_update:
- case OMPC_capture:
- case OMPC_seq_cst:
- case OMPC_depend:
- case OMPC_threads:
- case OMPC_simd:
- case OMPC_map:
- case OMPC_priority:
- case OMPC_grainsize:
- case OMPC_nogroup:
- case OMPC_num_tasks:
- case OMPC_hint:
- case OMPC_defaultmap:
- case OMPC_unknown:
- case OMPC_uniform:
- case OMPC_to:
- case OMPC_from:
- case OMPC_use_device_ptr:
- case OMPC_is_device_ptr:
- case OMPC_unified_address:
- case OMPC_unified_shared_memory:
- case OMPC_reverse_offload:
- case OMPC_dynamic_allocators:
- case OMPC_atomic_default_mem_order:
- llvm_unreachable("Unexpected OpenMP clause.");
- }
- return CaptureRegion;
- }
- OMPClause *Sema::ActOnOpenMPIfClause(OpenMPDirectiveKind NameModifier,
- Expr *Condition, SourceLocation StartLoc,
- SourceLocation LParenLoc,
- SourceLocation NameModifierLoc,
- SourceLocation ColonLoc,
- SourceLocation EndLoc) {
- Expr *ValExpr = Condition;
- Stmt *HelperValStmt = nullptr;
- OpenMPDirectiveKind CaptureRegion = OMPD_unknown;
- if (!Condition->isValueDependent() && !Condition->isTypeDependent() &&
- !Condition->isInstantiationDependent() &&
- !Condition->containsUnexpandedParameterPack()) {
- ExprResult Val = CheckBooleanCondition(StartLoc, Condition);
- if (Val.isInvalid())
- return nullptr;
- ValExpr = Val.get();
- OpenMPDirectiveKind DKind = DSAStack->getCurrentDirective();
- CaptureRegion =
- getOpenMPCaptureRegionForClause(DKind, OMPC_if, NameModifier);
- if (CaptureRegion != OMPD_unknown && !CurContext->isDependentContext()) {
- ValExpr = MakeFullExpr(ValExpr).get();
- llvm::MapVector<const Expr *, DeclRefExpr *> Captures;
- ValExpr = tryBuildCapture(*this, ValExpr, Captures).get();
- HelperValStmt = buildPreInits(Context, Captures);
- }
- }
- return new (Context)
- OMPIfClause(NameModifier, ValExpr, HelperValStmt, CaptureRegion, StartLoc,
- LParenLoc, NameModifierLoc, ColonLoc, EndLoc);
- }
- OMPClause *Sema::ActOnOpenMPFinalClause(Expr *Condition,
- SourceLocation StartLoc,
- SourceLocation LParenLoc,
- SourceLocation EndLoc) {
- Expr *ValExpr = Condition;
- if (!Condition->isValueDependent() && !Condition->isTypeDependent() &&
- !Condition->isInstantiationDependent() &&
- !Condition->containsUnexpandedParameterPack()) {
- ExprResult Val = CheckBooleanCondition(StartLoc, Condition);
- if (Val.isInvalid())
- return nullptr;
- ValExpr = MakeFullExpr(Val.get()).get();
- }
- return new (Context) OMPFinalClause(ValExpr, StartLoc, LParenLoc, EndLoc);
- }
- ExprResult Sema::PerformOpenMPImplicitIntegerConversion(SourceLocation Loc,
- Expr *Op) {
- if (!Op)
- return ExprError();
- class IntConvertDiagnoser : public ICEConvertDiagnoser {
- public:
- IntConvertDiagnoser()
- : ICEConvertDiagnoser(/*AllowScopedEnumerations*/ false, false, true) {}
- SemaDiagnosticBuilder diagnoseNotInt(Sema &S, SourceLocation Loc,
- QualType T) override {
- return S.Diag(Loc, diag::err_omp_not_integral) << T;
- }
- SemaDiagnosticBuilder diagnoseIncomplete(Sema &S, SourceLocation Loc,
- QualType T) override {
- return S.Diag(Loc, diag::err_omp_incomplete_type) << T;
- }
- SemaDiagnosticBuilder diagnoseExplicitConv(Sema &S, SourceLocation Loc,
- QualType T,
- QualType ConvTy) override {
- return S.Diag(Loc, diag::err_omp_explicit_conversion) << T << ConvTy;
- }
- SemaDiagnosticBuilder noteExplicitConv(Sema &S, CXXConversionDecl *Conv,
- QualType ConvTy) override {
- return S.Diag(Conv->getLocation(), diag::note_omp_conversion_here)
- << ConvTy->isEnumeralType() << ConvTy;
- }
- SemaDiagnosticBuilder diagnoseAmbiguous(Sema &S, SourceLocation Loc,
- QualType T) override {
- return S.Diag(Loc, diag::err_omp_ambiguous_conversion) << T;
- }
- SemaDiagnosticBuilder noteAmbiguous(Sema &S, CXXConversionDecl *Conv,
- QualType ConvTy) override {
- return S.Diag(Conv->getLocation(), diag::note_omp_conversion_here)
- << ConvTy->isEnumeralType() << ConvTy;
- }
- SemaDiagnosticBuilder diagnoseConversion(Sema &, SourceLocation, QualType,
- QualType) override {
- llvm_unreachable("conversion functions are permitted");
- }
- } ConvertDiagnoser;
- return PerformContextualImplicitConversion(Loc, Op, ConvertDiagnoser);
- }
- static bool isNonNegativeIntegerValue(Expr *&ValExpr, Sema &SemaRef,
- OpenMPClauseKind CKind,
- bool StrictlyPositive) {
- if (!ValExpr->isTypeDependent() && !ValExpr->isValueDependent() &&
- !ValExpr->isInstantiationDependent()) {
- SourceLocation Loc = ValExpr->getExprLoc();
- ExprResult Value =
- SemaRef.PerformOpenMPImplicitIntegerConversion(Loc, ValExpr);
- if (Value.isInvalid())
- return false;
- ValExpr = Value.get();
- // The expression must evaluate to a non-negative integer value.
- llvm::APSInt Result;
- if (ValExpr->isIntegerConstantExpr(Result, SemaRef.Context) &&
- Result.isSigned() &&
- !((!StrictlyPositive && Result.isNonNegative()) ||
- (StrictlyPositive && Result.isStrictlyPositive()))) {
- SemaRef.Diag(Loc, diag::err_omp_negative_expression_in_clause)
- << getOpenMPClauseName(CKind) << (StrictlyPositive ? 1 : 0)
- << ValExpr->getSourceRange();
- return false;
- }
- }
- return true;
- }
- OMPClause *Sema::ActOnOpenMPNumThreadsClause(Expr *NumThreads,
- SourceLocation StartLoc,
- SourceLocation LParenLoc,
- SourceLocation EndLoc) {
- Expr *ValExpr = NumThreads;
- Stmt *HelperValStmt = nullptr;
- // OpenMP [2.5, Restrictions]
- // The num_threads expression must evaluate to a positive integer value.
- if (!isNonNegativeIntegerValue(ValExpr, *this, OMPC_num_threads,
- /*StrictlyPositive=*/true))
- return nullptr;
- OpenMPDirectiveKind DKind = DSAStack->getCurrentDirective();
- OpenMPDirectiveKind CaptureRegion =
- getOpenMPCaptureRegionForClause(DKind, OMPC_num_threads);
- if (CaptureRegion != OMPD_unknown && !CurContext->isDependentContext()) {
- ValExpr = MakeFullExpr(ValExpr).get();
- llvm::MapVector<const Expr *, DeclRefExpr *> Captures;
- ValExpr = tryBuildCapture(*this, ValExpr, Captures).get();
- HelperValStmt = buildPreInits(Context, Captures);
- }
- return new (Context) OMPNumThreadsClause(
- ValExpr, HelperValStmt, CaptureRegion, StartLoc, LParenLoc, EndLoc);
- }
- ExprResult Sema::VerifyPositiveIntegerConstantInClause(Expr *E,
- OpenMPClauseKind CKind,
- bool StrictlyPositive) {
- if (!E)
- return ExprError();
- if (E->isValueDependent() || E->isTypeDependent() ||
- E->isInstantiationDependent() || E->containsUnexpandedParameterPack())
- return E;
- llvm::APSInt Result;
- ExprResult ICE = VerifyIntegerConstantExpression(E, &Result);
- if (ICE.isInvalid())
- return ExprError();
- if ((StrictlyPositive && !Result.isStrictlyPositive()) ||
- (!StrictlyPositive && !Result.isNonNegative())) {
- Diag(E->getExprLoc(), diag::err_omp_negative_expression_in_clause)
- << getOpenMPClauseName(CKind) << (StrictlyPositive ? 1 : 0)
- << E->getSourceRange();
- return ExprError();
- }
- if (CKind == OMPC_aligned && !Result.isPowerOf2()) {
- Diag(E->getExprLoc(), diag::warn_omp_alignment_not_power_of_two)
- << E->getSourceRange();
- return ExprError();
- }
- if (CKind == OMPC_collapse && DSAStack->getAssociatedLoops() == 1)
- DSAStack->setAssociatedLoops(Result.getExtValue());
- else if (CKind == OMPC_ordered)
- DSAStack->setAssociatedLoops(Result.getExtValue());
- return ICE;
- }
- OMPClause *Sema::ActOnOpenMPSafelenClause(Expr *Len, SourceLocation StartLoc,
- SourceLocation LParenLoc,
- SourceLocation EndLoc) {
- // OpenMP [2.8.1, simd construct, Description]
- // The parameter of the safelen clause must be a constant
- // positive integer expression.
- ExprResult Safelen = VerifyPositiveIntegerConstantInClause(Len, OMPC_safelen);
- if (Safelen.isInvalid())
- return nullptr;
- return new (Context)
- OMPSafelenClause(Safelen.get(), StartLoc, LParenLoc, EndLoc);
- }
- OMPClause *Sema::ActOnOpenMPSimdlenClause(Expr *Len, SourceLocation StartLoc,
- SourceLocation LParenLoc,
- SourceLocation EndLoc) {
- // OpenMP [2.8.1, simd construct, Description]
- // The parameter of the simdlen clause must be a constant
- // positive integer expression.
- ExprResult Simdlen = VerifyPositiveIntegerConstantInClause(Len, OMPC_simdlen);
- if (Simdlen.isInvalid())
- return nullptr;
- return new (Context)
- OMPSimdlenClause(Simdlen.get(), StartLoc, LParenLoc, EndLoc);
- }
- OMPClause *Sema::ActOnOpenMPCollapseClause(Expr *NumForLoops,
- SourceLocation StartLoc,
- SourceLocation LParenLoc,
- SourceLocation EndLoc) {
- // OpenMP [2.7.1, loop construct, Description]
- // OpenMP [2.8.1, simd construct, Description]
- // OpenMP [2.9.6, distribute construct, Description]
- // The parameter of the collapse clause must be a constant
- // positive integer expression.
- ExprResult NumForLoopsResult =
- VerifyPositiveIntegerConstantInClause(NumForLoops, OMPC_collapse);
- if (NumForLoopsResult.isInvalid())
- return nullptr;
- return new (Context)
- OMPCollapseClause(NumForLoopsResult.get(), StartLoc, LParenLoc, EndLoc);
- }
- OMPClause *Sema::ActOnOpenMPOrderedClause(SourceLocation StartLoc,
- SourceLocation EndLoc,
- SourceLocation LParenLoc,
- Expr *NumForLoops) {
- // OpenMP [2.7.1, loop construct, Description]
- // OpenMP [2.8.1, simd construct, Description]
- // OpenMP [2.9.6, distribute construct, Description]
- // The parameter of the ordered clause must be a constant
- // positive integer expression if any.
- if (NumForLoops && LParenLoc.isValid()) {
- ExprResult NumForLoopsResult =
- VerifyPositiveIntegerConstantInClause(NumForLoops, OMPC_ordered);
- if (NumForLoopsResult.isInvalid())
- return nullptr;
- NumForLoops = NumForLoopsResult.get();
- } else {
- NumForLoops = nullptr;
- }
- auto *Clause = OMPOrderedClause::Create(
- Context, NumForLoops, NumForLoops ? DSAStack->getAssociatedLoops() : 0,
- StartLoc, LParenLoc, EndLoc);
- DSAStack->setOrderedRegion(/*IsOrdered=*/true, NumForLoops, Clause);
- return Clause;
- }
- OMPClause *Sema::ActOnOpenMPSimpleClause(
- OpenMPClauseKind Kind, unsigned Argument, SourceLocation ArgumentLoc,
- SourceLocation StartLoc, SourceLocation LParenLoc, SourceLocation EndLoc) {
- OMPClause *Res = nullptr;
- switch (Kind) {
- case OMPC_default:
- Res =
- ActOnOpenMPDefaultClause(static_cast<OpenMPDefaultClauseKind>(Argument),
- ArgumentLoc, StartLoc, LParenLoc, EndLoc);
- break;
- case OMPC_proc_bind:
- Res = ActOnOpenMPProcBindClause(
- static_cast<OpenMPProcBindClauseKind>(Argument), ArgumentLoc, StartLoc,
- LParenLoc, EndLoc);
- break;
- case OMPC_atomic_default_mem_order:
- Res = ActOnOpenMPAtomicDefaultMemOrderClause(
- static_cast<OpenMPAtomicDefaultMemOrderClauseKind>(Argument),
- ArgumentLoc, StartLoc, LParenLoc, EndLoc);
- break;
- case OMPC_if:
- case OMPC_final:
- case OMPC_num_threads:
- case OMPC_safelen:
- case OMPC_simdlen:
- case OMPC_collapse:
- case OMPC_schedule:
- 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_ordered:
- case OMPC_nowait:
- case OMPC_untied:
- case OMPC_mergeable:
- case OMPC_threadprivate:
- case OMPC_flush:
- case OMPC_read:
- case OMPC_write:
- case OMPC_update:
- case OMPC_capture:
- case OMPC_seq_cst:
- case OMPC_depend:
- case OMPC_device:
- case OMPC_threads:
- case OMPC_simd:
- case OMPC_map:
- case OMPC_num_teams:
- case OMPC_thread_limit:
- case OMPC_priority:
- case OMPC_grainsize:
- case OMPC_nogroup:
- case OMPC_num_tasks:
- case OMPC_hint:
- case OMPC_dist_schedule:
- case OMPC_defaultmap:
- case OMPC_unknown:
- case OMPC_uniform:
- case OMPC_to:
- case OMPC_from:
- case OMPC_use_device_ptr:
- case OMPC_is_device_ptr:
- case OMPC_unified_address:
- case OMPC_unified_shared_memory:
- case OMPC_reverse_offload:
- case OMPC_dynamic_allocators:
- llvm_unreachable("Clause is not allowed.");
- }
- return Res;
- }
- static std::string
- getListOfPossibleValues(OpenMPClauseKind K, unsigned First, unsigned Last,
- ArrayRef<unsigned> Exclude = llvm::None) {
- SmallString<256> Buffer;
- llvm::raw_svector_ostream Out(Buffer);
- unsigned Bound = Last >= 2 ? Last - 2 : 0;
- unsigned Skipped = Exclude.size();
- auto S = Exclude.begin(), E = Exclude.end();
- for (unsigned I = First; I < Last; ++I) {
- if (std::find(S, E, I) != E) {
- --Skipped;
- continue;
- }
- Out << "'" << getOpenMPSimpleClauseTypeName(K, I) << "'";
- if (I == Bound - Skipped)
- Out << " or ";
- else if (I != Bound + 1 - Skipped)
- Out << ", ";
- }
- return Out.str();
- }
- OMPClause *Sema::ActOnOpenMPDefaultClause(OpenMPDefaultClauseKind Kind,
- SourceLocation KindKwLoc,
- SourceLocation StartLoc,
- SourceLocation LParenLoc,
- SourceLocation EndLoc) {
- if (Kind == OMPC_DEFAULT_unknown) {
- static_assert(OMPC_DEFAULT_unknown > 0,
- "OMPC_DEFAULT_unknown not greater than 0");
- Diag(KindKwLoc, diag::err_omp_unexpected_clause_value)
- << getListOfPossibleValues(OMPC_default, /*First=*/0,
- /*Last=*/OMPC_DEFAULT_unknown)
- << getOpenMPClauseName(OMPC_default);
- return nullptr;
- }
- switch (Kind) {
- case OMPC_DEFAULT_none:
- DSAStack->setDefaultDSANone(KindKwLoc);
- break;
- case OMPC_DEFAULT_shared:
- DSAStack->setDefaultDSAShared(KindKwLoc);
- break;
- case OMPC_DEFAULT_unknown:
- llvm_unreachable("Clause kind is not allowed.");
- break;
- }
- return new (Context)
- OMPDefaultClause(Kind, KindKwLoc, StartLoc, LParenLoc, EndLoc);
- }
- OMPClause *Sema::ActOnOpenMPProcBindClause(OpenMPProcBindClauseKind Kind,
- SourceLocation KindKwLoc,
- SourceLocation StartLoc,
- SourceLocation LParenLoc,
- SourceLocation EndLoc) {
- if (Kind == OMPC_PROC_BIND_unknown) {
- Diag(KindKwLoc, diag::err_omp_unexpected_clause_value)
- << getListOfPossibleValues(OMPC_proc_bind, /*First=*/0,
- /*Last=*/OMPC_PROC_BIND_unknown)
- << getOpenMPClauseName(OMPC_proc_bind);
- return nullptr;
- }
- return new (Context)
- OMPProcBindClause(Kind, KindKwLoc, StartLoc, LParenLoc, EndLoc);
- }
- OMPClause *Sema::ActOnOpenMPAtomicDefaultMemOrderClause(
- OpenMPAtomicDefaultMemOrderClauseKind Kind, SourceLocation KindKwLoc,
- SourceLocation StartLoc, SourceLocation LParenLoc, SourceLocation EndLoc) {
- if (Kind == OMPC_ATOMIC_DEFAULT_MEM_ORDER_unknown) {
- Diag(KindKwLoc, diag::err_omp_unexpected_clause_value)
- << getListOfPossibleValues(
- OMPC_atomic_default_mem_order, /*First=*/0,
- /*Last=*/OMPC_ATOMIC_DEFAULT_MEM_ORDER_unknown)
- << getOpenMPClauseName(OMPC_atomic_default_mem_order);
- return nullptr;
- }
- return new (Context) OMPAtomicDefaultMemOrderClause(Kind, KindKwLoc, StartLoc,
- LParenLoc, EndLoc);
- }
- OMPClause *Sema::ActOnOpenMPSingleExprWithArgClause(
- OpenMPClauseKind Kind, ArrayRef<unsigned> Argument, Expr *Expr,
- SourceLocation StartLoc, SourceLocation LParenLoc,
- ArrayRef<SourceLocation> ArgumentLoc, SourceLocation DelimLoc,
- SourceLocation EndLoc) {
- OMPClause *Res = nullptr;
- switch (Kind) {
- case OMPC_schedule:
- enum { Modifier1, Modifier2, ScheduleKind, NumberOfElements };
- assert(Argument.size() == NumberOfElements &&
- ArgumentLoc.size() == NumberOfElements);
- Res = ActOnOpenMPScheduleClause(
- static_cast<OpenMPScheduleClauseModifier>(Argument[Modifier1]),
- static_cast<OpenMPScheduleClauseModifier>(Argument[Modifier2]),
- static_cast<OpenMPScheduleClauseKind>(Argument[ScheduleKind]), Expr,
- StartLoc, LParenLoc, ArgumentLoc[Modifier1], ArgumentLoc[Modifier2],
- ArgumentLoc[ScheduleKind], DelimLoc, EndLoc);
- break;
- case OMPC_if:
- assert(Argument.size() == 1 && ArgumentLoc.size() == 1);
- Res = ActOnOpenMPIfClause(static_cast<OpenMPDirectiveKind>(Argument.back()),
- Expr, StartLoc, LParenLoc, ArgumentLoc.back(),
- DelimLoc, EndLoc);
- break;
- case OMPC_dist_schedule:
- Res = ActOnOpenMPDistScheduleClause(
- static_cast<OpenMPDistScheduleClauseKind>(Argument.back()), Expr,
- StartLoc, LParenLoc, ArgumentLoc.back(), DelimLoc, EndLoc);
- break;
- case OMPC_defaultmap:
- enum { Modifier, DefaultmapKind };
- Res = ActOnOpenMPDefaultmapClause(
- static_cast<OpenMPDefaultmapClauseModifier>(Argument[Modifier]),
- static_cast<OpenMPDefaultmapClauseKind>(Argument[DefaultmapKind]),
- StartLoc, LParenLoc, ArgumentLoc[Modifier], ArgumentLoc[DefaultmapKind],
- EndLoc);
- break;
- case OMPC_final:
- case OMPC_num_threads:
- case OMPC_safelen:
- case OMPC_simdlen:
- case OMPC_collapse:
- case OMPC_default:
- case OMPC_proc_bind:
- 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_ordered:
- case OMPC_nowait:
- case OMPC_untied:
- case OMPC_mergeable:
- case OMPC_threadprivate:
- case OMPC_flush:
- case OMPC_read:
- case OMPC_write:
- case OMPC_update:
- case OMPC_capture:
- case OMPC_seq_cst:
- case OMPC_depend:
- case OMPC_device:
- case OMPC_threads:
- case OMPC_simd:
- case OMPC_map:
- case OMPC_num_teams:
- case OMPC_thread_limit:
- case OMPC_priority:
- case OMPC_grainsize:
- case OMPC_nogroup:
- case OMPC_num_tasks:
- case OMPC_hint:
- case OMPC_unknown:
- case OMPC_uniform:
- case OMPC_to:
- case OMPC_from:
- case OMPC_use_device_ptr:
- case OMPC_is_device_ptr:
- case OMPC_unified_address:
- case OMPC_unified_shared_memory:
- case OMPC_reverse_offload:
- case OMPC_dynamic_allocators:
- case OMPC_atomic_default_mem_order:
- llvm_unreachable("Clause is not allowed.");
- }
- return Res;
- }
- static bool checkScheduleModifiers(Sema &S, OpenMPScheduleClauseModifier M1,
- OpenMPScheduleClauseModifier M2,
- SourceLocation M1Loc, SourceLocation M2Loc) {
- if (M1 == OMPC_SCHEDULE_MODIFIER_unknown && M1Loc.isValid()) {
- SmallVector<unsigned, 2> Excluded;
- if (M2 != OMPC_SCHEDULE_MODIFIER_unknown)
- Excluded.push_back(M2);
- if (M2 == OMPC_SCHEDULE_MODIFIER_nonmonotonic)
- Excluded.push_back(OMPC_SCHEDULE_MODIFIER_monotonic);
- if (M2 == OMPC_SCHEDULE_MODIFIER_monotonic)
- Excluded.push_back(OMPC_SCHEDULE_MODIFIER_nonmonotonic);
- S.Diag(M1Loc, diag::err_omp_unexpected_clause_value)
- << getListOfPossibleValues(OMPC_schedule,
- /*First=*/OMPC_SCHEDULE_MODIFIER_unknown + 1,
- /*Last=*/OMPC_SCHEDULE_MODIFIER_last,
- Excluded)
- << getOpenMPClauseName(OMPC_schedule);
- return true;
- }
- return false;
- }
- OMPClause *Sema::ActOnOpenMPScheduleClause(
- OpenMPScheduleClauseModifier M1, OpenMPScheduleClauseModifier M2,
- OpenMPScheduleClauseKind Kind, Expr *ChunkSize, SourceLocation StartLoc,
- SourceLocation LParenLoc, SourceLocation M1Loc, SourceLocation M2Loc,
- SourceLocation KindLoc, SourceLocation CommaLoc, SourceLocation EndLoc) {
- if (checkScheduleModifiers(*this, M1, M2, M1Loc, M2Loc) ||
- checkScheduleModifiers(*this, M2, M1, M2Loc, M1Loc))
- return nullptr;
- // OpenMP, 2.7.1, Loop Construct, Restrictions
- // Either the monotonic modifier or the nonmonotonic modifier can be specified
- // but not both.
- if ((M1 == M2 && M1 != OMPC_SCHEDULE_MODIFIER_unknown) ||
- (M1 == OMPC_SCHEDULE_MODIFIER_monotonic &&
- M2 == OMPC_SCHEDULE_MODIFIER_nonmonotonic) ||
- (M1 == OMPC_SCHEDULE_MODIFIER_nonmonotonic &&
- M2 == OMPC_SCHEDULE_MODIFIER_monotonic)) {
- Diag(M2Loc, diag::err_omp_unexpected_schedule_modifier)
- << getOpenMPSimpleClauseTypeName(OMPC_schedule, M2)
- << getOpenMPSimpleClauseTypeName(OMPC_schedule, M1);
- return nullptr;
- }
- if (Kind == OMPC_SCHEDULE_unknown) {
- std::string Values;
- if (M1Loc.isInvalid() && M2Loc.isInvalid()) {
- unsigned Exclude[] = {OMPC_SCHEDULE_unknown};
- Values = getListOfPossibleValues(OMPC_schedule, /*First=*/0,
- /*Last=*/OMPC_SCHEDULE_MODIFIER_last,
- Exclude);
- } else {
- Values = getListOfPossibleValues(OMPC_schedule, /*First=*/0,
- /*Last=*/OMPC_SCHEDULE_unknown);
- }
- Diag(KindLoc, diag::err_omp_unexpected_clause_value)
- << Values << getOpenMPClauseName(OMPC_schedule);
- return nullptr;
- }
- // OpenMP, 2.7.1, Loop Construct, Restrictions
- // The nonmonotonic modifier can only be specified with schedule(dynamic) or
- // schedule(guided).
- if ((M1 == OMPC_SCHEDULE_MODIFIER_nonmonotonic ||
- M2 == OMPC_SCHEDULE_MODIFIER_nonmonotonic) &&
- Kind != OMPC_SCHEDULE_dynamic && Kind != OMPC_SCHEDULE_guided) {
- Diag(M1 == OMPC_SCHEDULE_MODIFIER_nonmonotonic ? M1Loc : M2Loc,
- diag::err_omp_schedule_nonmonotonic_static);
- return nullptr;
- }
- Expr *ValExpr = ChunkSize;
- Stmt *HelperValStmt = nullptr;
- if (ChunkSize) {
- if (!ChunkSize->isValueDependent() && !ChunkSize->isTypeDependent() &&
- !ChunkSize->isInstantiationDependent() &&
- !ChunkSize->containsUnexpandedParameterPack()) {
- SourceLocation ChunkSizeLoc = ChunkSize->getBeginLoc();
- ExprResult Val =
- PerformOpenMPImplicitIntegerConversion(ChunkSizeLoc, ChunkSize);
- if (Val.isInvalid())
- return nullptr;
- ValExpr = Val.get();
- // OpenMP [2.7.1, Restrictions]
- // chunk_size must be a loop invariant integer expression with a positive
- // value.
- llvm::APSInt Result;
- if (ValExpr->isIntegerConstantExpr(Result, Context)) {
- if (Result.isSigned() && !Result.isStrictlyPositive()) {
- Diag(ChunkSizeLoc, diag::err_omp_negative_expression_in_clause)
- << "schedule" << 1 << ChunkSize->getSourceRange();
- return nullptr;
- }
- } else if (getOpenMPCaptureRegionForClause(
- DSAStack->getCurrentDirective(), OMPC_schedule) !=
- OMPD_unknown &&
- !CurContext->isDependentContext()) {
- ValExpr = MakeFullExpr(ValExpr).get();
- llvm::MapVector<const Expr *, DeclRefExpr *> Captures;
- ValExpr = tryBuildCapture(*this, ValExpr, Captures).get();
- HelperValStmt = buildPreInits(Context, Captures);
- }
- }
- }
- return new (Context)
- OMPScheduleClause(StartLoc, LParenLoc, KindLoc, CommaLoc, EndLoc, Kind,
- ValExpr, HelperValStmt, M1, M1Loc, M2, M2Loc);
- }
- OMPClause *Sema::ActOnOpenMPClause(OpenMPClauseKind Kind,
- SourceLocation StartLoc,
- SourceLocation EndLoc) {
- OMPClause *Res = nullptr;
- switch (Kind) {
- case OMPC_ordered:
- Res = ActOnOpenMPOrderedClause(StartLoc, EndLoc);
- break;
- case OMPC_nowait:
- Res = ActOnOpenMPNowaitClause(StartLoc, EndLoc);
- break;
- case OMPC_untied:
- Res = ActOnOpenMPUntiedClause(StartLoc, EndLoc);
- break;
- case OMPC_mergeable:
- Res = ActOnOpenMPMergeableClause(StartLoc, EndLoc);
- break;
- case OMPC_read:
- Res = ActOnOpenMPReadClause(StartLoc, EndLoc);
- break;
- case OMPC_write:
- Res = ActOnOpenMPWriteClause(StartLoc, EndLoc);
- break;
- case OMPC_update:
- Res = ActOnOpenMPUpdateClause(StartLoc, EndLoc);
- break;
- case OMPC_capture:
- Res = ActOnOpenMPCaptureClause(StartLoc, EndLoc);
- break;
- case OMPC_seq_cst:
- Res = ActOnOpenMPSeqCstClause(StartLoc, EndLoc);
- break;
- case OMPC_threads:
- Res = ActOnOpenMPThreadsClause(StartLoc, EndLoc);
- break;
- case OMPC_simd:
- Res = ActOnOpenMPSIMDClause(StartLoc, EndLoc);
- break;
- case OMPC_nogroup:
- Res = ActOnOpenMPNogroupClause(StartLoc, EndLoc);
- break;
- case OMPC_unified_address:
- Res = ActOnOpenMPUnifiedAddressClause(StartLoc, EndLoc);
- break;
- case OMPC_unified_shared_memory:
- Res = ActOnOpenMPUnifiedSharedMemoryClause(StartLoc, EndLoc);
- break;
- case OMPC_reverse_offload:
- Res = ActOnOpenMPReverseOffloadClause(StartLoc, EndLoc);
- break;
- case OMPC_dynamic_allocators:
- Res = ActOnOpenMPDynamicAllocatorsClause(StartLoc, EndLoc);
- break;
- case OMPC_if:
- case OMPC_final:
- case OMPC_num_threads:
- case OMPC_safelen:
- case OMPC_simdlen:
- case OMPC_collapse:
- case OMPC_schedule:
- 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_default:
- case OMPC_proc_bind:
- case OMPC_threadprivate:
- case OMPC_flush:
- case OMPC_depend:
- case OMPC_device:
- case OMPC_map:
- case OMPC_num_teams:
- case OMPC_thread_limit:
- case OMPC_priority:
- case OMPC_grainsize:
- case OMPC_num_tasks:
- case OMPC_hint:
- case OMPC_dist_schedule:
- case OMPC_defaultmap:
- case OMPC_unknown:
- case OMPC_uniform:
- case OMPC_to:
- case OMPC_from:
- case OMPC_use_device_ptr:
- case OMPC_is_device_ptr:
- case OMPC_atomic_default_mem_order:
- llvm_unreachable("Clause is not allowed.");
- }
- return Res;
- }
- OMPClause *Sema::ActOnOpenMPNowaitClause(SourceLocation StartLoc,
- SourceLocation EndLoc) {
- DSAStack->setNowaitRegion();
- return new (Context) OMPNowaitClause(StartLoc, EndLoc);
- }
- OMPClause *Sema::ActOnOpenMPUntiedClause(SourceLocation StartLoc,
- SourceLocation EndLoc) {
- return new (Context) OMPUntiedClause(StartLoc, EndLoc);
- }
- OMPClause *Sema::ActOnOpenMPMergeableClause(SourceLocation StartLoc,
- SourceLocation EndLoc) {
- return new (Context) OMPMergeableClause(StartLoc, EndLoc);
- }
- OMPClause *Sema::ActOnOpenMPReadClause(SourceLocation StartLoc,
- SourceLocation EndLoc) {
- return new (Context) OMPReadClause(StartLoc, EndLoc);
- }
- OMPClause *Sema::ActOnOpenMPWriteClause(SourceLocation StartLoc,
- SourceLocation EndLoc) {
- return new (Context) OMPWriteClause(StartLoc, EndLoc);
- }
- OMPClause *Sema::ActOnOpenMPUpdateClause(SourceLocation StartLoc,
- SourceLocation EndLoc) {
- return new (Context) OMPUpdateClause(StartLoc, EndLoc);
- }
- OMPClause *Sema::ActOnOpenMPCaptureClause(SourceLocation StartLoc,
- SourceLocation EndLoc) {
- return new (Context) OMPCaptureClause(StartLoc, EndLoc);
- }
- OMPClause *Sema::ActOnOpenMPSeqCstClause(SourceLocation StartLoc,
- SourceLocation EndLoc) {
- return new (Context) OMPSeqCstClause(StartLoc, EndLoc);
- }
- OMPClause *Sema::ActOnOpenMPThreadsClause(SourceLocation StartLoc,
- SourceLocation EndLoc) {
- return new (Context) OMPThreadsClause(StartLoc, EndLoc);
- }
- OMPClause *Sema::ActOnOpenMPSIMDClause(SourceLocation StartLoc,
- SourceLocation EndLoc) {
- return new (Context) OMPSIMDClause(StartLoc, EndLoc);
- }
- OMPClause *Sema::ActOnOpenMPNogroupClause(SourceLocation StartLoc,
- SourceLocation EndLoc) {
- return new (Context) OMPNogroupClause(StartLoc, EndLoc);
- }
- OMPClause *Sema::ActOnOpenMPUnifiedAddressClause(SourceLocation StartLoc,
- SourceLocation EndLoc) {
- return new (Context) OMPUnifiedAddressClause(StartLoc, EndLoc);
- }
- OMPClause *Sema::ActOnOpenMPUnifiedSharedMemoryClause(SourceLocation StartLoc,
- SourceLocation EndLoc) {
- return new (Context) OMPUnifiedSharedMemoryClause(StartLoc, EndLoc);
- }
- OMPClause *Sema::ActOnOpenMPReverseOffloadClause(SourceLocation StartLoc,
- SourceLocation EndLoc) {
- return new (Context) OMPReverseOffloadClause(StartLoc, EndLoc);
- }
- OMPClause *Sema::ActOnOpenMPDynamicAllocatorsClause(SourceLocation StartLoc,
- SourceLocation EndLoc) {
- return new (Context) OMPDynamicAllocatorsClause(StartLoc, EndLoc);
- }
- OMPClause *Sema::ActOnOpenMPVarListClause(
- OpenMPClauseKind Kind, ArrayRef<Expr *> VarList, Expr *TailExpr,
- SourceLocation StartLoc, SourceLocation LParenLoc, SourceLocation ColonLoc,
- SourceLocation EndLoc, CXXScopeSpec &ReductionIdScopeSpec,
- const DeclarationNameInfo &ReductionId, OpenMPDependClauseKind DepKind,
- OpenMPLinearClauseKind LinKind,
- ArrayRef<OpenMPMapModifierKind> MapTypeModifiers,
- ArrayRef<SourceLocation> MapTypeModifiersLoc,
- OpenMPMapClauseKind MapType, bool IsMapTypeImplicit,
- SourceLocation DepLinMapLoc) {
- OMPClause *Res = nullptr;
- switch (Kind) {
- case OMPC_private:
- Res = ActOnOpenMPPrivateClause(VarList, StartLoc, LParenLoc, EndLoc);
- break;
- case OMPC_firstprivate:
- Res = ActOnOpenMPFirstprivateClause(VarList, StartLoc, LParenLoc, EndLoc);
- break;
- case OMPC_lastprivate:
- Res = ActOnOpenMPLastprivateClause(VarList, StartLoc, LParenLoc, EndLoc);
- break;
- case OMPC_shared:
- Res = ActOnOpenMPSharedClause(VarList, StartLoc, LParenLoc, EndLoc);
- break;
- case OMPC_reduction:
- Res = ActOnOpenMPReductionClause(VarList, StartLoc, LParenLoc, ColonLoc,
- EndLoc, ReductionIdScopeSpec, ReductionId);
- break;
- case OMPC_task_reduction:
- Res = ActOnOpenMPTaskReductionClause(VarList, StartLoc, LParenLoc, ColonLoc,
- EndLoc, ReductionIdScopeSpec,
- ReductionId);
- break;
- case OMPC_in_reduction:
- Res =
- ActOnOpenMPInReductionClause(VarList, StartLoc, LParenLoc, ColonLoc,
- EndLoc, ReductionIdScopeSpec, ReductionId);
- break;
- case OMPC_linear:
- Res = ActOnOpenMPLinearClause(VarList, TailExpr, StartLoc, LParenLoc,
- LinKind, DepLinMapLoc, ColonLoc, EndLoc);
- break;
- case OMPC_aligned:
- Res = ActOnOpenMPAlignedClause(VarList, TailExpr, StartLoc, LParenLoc,
- ColonLoc, EndLoc);
- break;
- case OMPC_copyin:
- Res = ActOnOpenMPCopyinClause(VarList, StartLoc, LParenLoc, EndLoc);
- break;
- case OMPC_copyprivate:
- Res = ActOnOpenMPCopyprivateClause(VarList, StartLoc, LParenLoc, EndLoc);
- break;
- case OMPC_flush:
- Res = ActOnOpenMPFlushClause(VarList, StartLoc, LParenLoc, EndLoc);
- break;
- case OMPC_depend:
- Res = ActOnOpenMPDependClause(DepKind, DepLinMapLoc, ColonLoc, VarList,
- StartLoc, LParenLoc, EndLoc);
- break;
- case OMPC_map:
- Res = ActOnOpenMPMapClause(MapTypeModifiers, MapTypeModifiersLoc, MapType,
- IsMapTypeImplicit, DepLinMapLoc, ColonLoc,
- VarList, StartLoc, LParenLoc, EndLoc);
- break;
- case OMPC_to:
- Res = ActOnOpenMPToClause(VarList, StartLoc, LParenLoc, EndLoc);
- break;
- case OMPC_from:
- Res = ActOnOpenMPFromClause(VarList, StartLoc, LParenLoc, EndLoc);
- break;
- case OMPC_use_device_ptr:
- Res = ActOnOpenMPUseDevicePtrClause(VarList, StartLoc, LParenLoc, EndLoc);
- break;
- case OMPC_is_device_ptr:
- Res = ActOnOpenMPIsDevicePtrClause(VarList, StartLoc, LParenLoc, EndLoc);
- break;
- case OMPC_if:
- case OMPC_final:
- case OMPC_num_threads:
- case OMPC_safelen:
- case OMPC_simdlen:
- case OMPC_collapse:
- case OMPC_default:
- case OMPC_proc_bind:
- case OMPC_schedule:
- case OMPC_ordered:
- case OMPC_nowait:
- case OMPC_untied:
- case OMPC_mergeable:
- case OMPC_threadprivate:
- case OMPC_read:
- case OMPC_write:
- case OMPC_update:
- case OMPC_capture:
- case OMPC_seq_cst:
- case OMPC_device:
- case OMPC_threads:
- case OMPC_simd:
- case OMPC_num_teams:
- case OMPC_thread_limit:
- case OMPC_priority:
- case OMPC_grainsize:
- case OMPC_nogroup:
- case OMPC_num_tasks:
- case OMPC_hint:
- case OMPC_dist_schedule:
- case OMPC_defaultmap:
- case OMPC_unknown:
- case OMPC_uniform:
- case OMPC_unified_address:
- case OMPC_unified_shared_memory:
- case OMPC_reverse_offload:
- case OMPC_dynamic_allocators:
- case OMPC_atomic_default_mem_order:
- llvm_unreachable("Clause is not allowed.");
- }
- return Res;
- }
- ExprResult Sema::getOpenMPCapturedExpr(VarDecl *Capture, ExprValueKind VK,
- ExprObjectKind OK, SourceLocation Loc) {
- ExprResult Res = BuildDeclRefExpr(
- Capture, Capture->getType().getNonReferenceType(), VK_LValue, Loc);
- if (!Res.isUsable())
- return ExprError();
- if (OK == OK_Ordinary && !getLangOpts().CPlusPlus) {
- Res = CreateBuiltinUnaryOp(Loc, UO_Deref, Res.get());
- if (!Res.isUsable())
- return ExprError();
- }
- if (VK != VK_LValue && Res.get()->isGLValue()) {
- Res = DefaultLvalueConversion(Res.get());
- if (!Res.isUsable())
- return ExprError();
- }
- return Res;
- }
- static std::pair<ValueDecl *, bool>
- getPrivateItem(Sema &S, Expr *&RefExpr, SourceLocation &ELoc,
- SourceRange &ERange, bool AllowArraySection = false) {
- if (RefExpr->isTypeDependent() || RefExpr->isValueDependent() ||
- RefExpr->containsUnexpandedParameterPack())
- return std::make_pair(nullptr, true);
- // OpenMP [3.1, C/C++]
- // A list item is a variable name.
- // OpenMP [2.9.3.3, Restrictions, p.1]
- // A variable that is part of another variable (as an array or
- // structure element) cannot appear in a private clause.
- RefExpr = RefExpr->IgnoreParens();
- enum {
- NoArrayExpr = -1,
- ArraySubscript = 0,
- OMPArraySection = 1
- } IsArrayExpr = NoArrayExpr;
- if (AllowArraySection) {
- if (auto *ASE = dyn_cast_or_null<ArraySubscriptExpr>(RefExpr)) {
- Expr *Base = ASE->getBase()->IgnoreParenImpCasts();
- while (auto *TempASE = dyn_cast<ArraySubscriptExpr>(Base))
- Base = TempASE->getBase()->IgnoreParenImpCasts();
- RefExpr = Base;
- IsArrayExpr = ArraySubscript;
- } else if (auto *OASE = dyn_cast_or_null<OMPArraySectionExpr>(RefExpr)) {
- Expr *Base = OASE->getBase()->IgnoreParenImpCasts();
- while (auto *TempOASE = dyn_cast<OMPArraySectionExpr>(Base))
- Base = TempOASE->getBase()->IgnoreParenImpCasts();
- while (auto *TempASE = dyn_cast<ArraySubscriptExpr>(Base))
- Base = TempASE->getBase()->IgnoreParenImpCasts();
- RefExpr = Base;
- IsArrayExpr = OMPArraySection;
- }
- }
- ELoc = RefExpr->getExprLoc();
- ERange = RefExpr->getSourceRange();
- RefExpr = RefExpr->IgnoreParenImpCasts();
- auto *DE = dyn_cast_or_null<DeclRefExpr>(RefExpr);
- auto *ME = dyn_cast_or_null<MemberExpr>(RefExpr);
- if ((!DE || !isa<VarDecl>(DE->getDecl())) &&
- (S.getCurrentThisType().isNull() || !ME ||
- !isa<CXXThisExpr>(ME->getBase()->IgnoreParenImpCasts()) ||
- !isa<FieldDecl>(ME->getMemberDecl()))) {
- if (IsArrayExpr != NoArrayExpr) {
- S.Diag(ELoc, diag::err_omp_expected_base_var_name) << IsArrayExpr
- << ERange;
- } else {
- S.Diag(ELoc,
- AllowArraySection
- ? diag::err_omp_expected_var_name_member_expr_or_array_item
- : diag::err_omp_expected_var_name_member_expr)
- << (S.getCurrentThisType().isNull() ? 0 : 1) << ERange;
- }
- return std::make_pair(nullptr, false);
- }
- return std::make_pair(
- getCanonicalDecl(DE ? DE->getDecl() : ME->getMemberDecl()), false);
- }
- OMPClause *Sema::ActOnOpenMPPrivateClause(ArrayRef<Expr *> VarList,
- SourceLocation StartLoc,
- SourceLocation LParenLoc,
- SourceLocation EndLoc) {
- SmallVector<Expr *, 8> Vars;
- SmallVector<Expr *, 8> PrivateCopies;
- for (Expr *RefExpr : VarList) {
- assert(RefExpr && "NULL expr in OpenMP private clause.");
- SourceLocation ELoc;
- SourceRange ERange;
- Expr *SimpleRefExpr = RefExpr;
- auto Res = getPrivateItem(*this, SimpleRefExpr, ELoc, ERange);
- if (Res.second) {
- // It will be analyzed later.
- Vars.push_back(RefExpr);
- PrivateCopies.push_back(nullptr);
- }
- ValueDecl *D = Res.first;
- if (!D)
- continue;
- QualType Type = D->getType();
- auto *VD = dyn_cast<VarDecl>(D);
- // OpenMP [2.9.3.3, Restrictions, C/C++, p.3]
- // A variable that appears in a private clause must not have an incomplete
- // type or a reference type.
- if (RequireCompleteType(ELoc, Type, diag::err_omp_private_incomplete_type))
- continue;
- Type = Type.getNonReferenceType();
- // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced
- // in a Construct]
- // Variables with the predetermined data-sharing attributes may not be
- // listed in data-sharing attributes clauses, except for the cases
- // listed below. For these exceptions only, listing a predetermined
- // variable in a data-sharing attribute clause is allowed and overrides
- // the variable's predetermined data-sharing attributes.
- DSAStackTy::DSAVarData DVar = DSAStack->getTopDSA(D, /*FromParent=*/false);
- if (DVar.CKind != OMPC_unknown && DVar.CKind != OMPC_private) {
- Diag(ELoc, diag::err_omp_wrong_dsa) << getOpenMPClauseName(DVar.CKind)
- << getOpenMPClauseName(OMPC_private);
- reportOriginalDsa(*this, DSAStack, D, DVar);
- continue;
- }
- OpenMPDirectiveKind CurrDir = DSAStack->getCurrentDirective();
- // Variably modified types are not supported for tasks.
- if (!Type->isAnyPointerType() && Type->isVariablyModifiedType() &&
- isOpenMPTaskingDirective(CurrDir)) {
- Diag(ELoc, diag::err_omp_variably_modified_type_not_supported)
- << getOpenMPClauseName(OMPC_private) << Type
- << getOpenMPDirectiveName(CurrDir);
- bool IsDecl =
- !VD ||
- VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly;
- Diag(D->getLocation(),
- IsDecl ? diag::note_previous_decl : diag::note_defined_here)
- << D;
- continue;
- }
- // OpenMP 4.5 [2.15.5.1, Restrictions, p.3]
- // A list item cannot appear in both a map clause and a data-sharing
- // attribute clause on the same construct
- if (isOpenMPTargetExecutionDirective(CurrDir)) {
- OpenMPClauseKind ConflictKind;
- if (DSAStack->checkMappableExprComponentListsForDecl(
- VD, /*CurrentRegionOnly=*/true,
- [&](OMPClauseMappableExprCommon::MappableExprComponentListRef,
- OpenMPClauseKind WhereFoundClauseKind) -> bool {
- ConflictKind = WhereFoundClauseKind;
- return true;
- })) {
- Diag(ELoc, diag::err_omp_variable_in_given_clause_and_dsa)
- << getOpenMPClauseName(OMPC_private)
- << getOpenMPClauseName(ConflictKind)
- << getOpenMPDirectiveName(CurrDir);
- reportOriginalDsa(*this, DSAStack, D, DVar);
- continue;
- }
- }
- // OpenMP [2.9.3.3, Restrictions, C/C++, p.1]
- // A variable of class type (or array thereof) that appears in a private
- // clause requires an accessible, unambiguous default constructor for the
- // class type.
- // Generate helper private variable and initialize it with the default
- // value. The address of the original variable is replaced by the address of
- // the new private variable in CodeGen. This new variable is not added to
- // IdResolver, so the code in the OpenMP region uses original variable for
- // proper diagnostics.
- Type = Type.getUnqualifiedType();
- VarDecl *VDPrivate =
- buildVarDecl(*this, ELoc, Type, D->getName(),
- D->hasAttrs() ? &D->getAttrs() : nullptr,
- VD ? cast<DeclRefExpr>(SimpleRefExpr) : nullptr);
- ActOnUninitializedDecl(VDPrivate);
- if (VDPrivate->isInvalidDecl())
- continue;
- DeclRefExpr *VDPrivateRefExpr = buildDeclRefExpr(
- *this, VDPrivate, RefExpr->getType().getUnqualifiedType(), ELoc);
- DeclRefExpr *Ref = nullptr;
- if (!VD && !CurContext->isDependentContext())
- Ref = buildCapture(*this, D, SimpleRefExpr, /*WithInit=*/false);
- DSAStack->addDSA(D, RefExpr->IgnoreParens(), OMPC_private, Ref);
- Vars.push_back((VD || CurContext->isDependentContext())
- ? RefExpr->IgnoreParens()
- : Ref);
- PrivateCopies.push_back(VDPrivateRefExpr);
- }
- if (Vars.empty())
- return nullptr;
- return OMPPrivateClause::Create(Context, StartLoc, LParenLoc, EndLoc, Vars,
- PrivateCopies);
- }
- namespace {
- class DiagsUninitializedSeveretyRAII {
- private:
- DiagnosticsEngine &Diags;
- SourceLocation SavedLoc;
- bool IsIgnored = false;
- public:
- DiagsUninitializedSeveretyRAII(DiagnosticsEngine &Diags, SourceLocation Loc,
- bool IsIgnored)
- : Diags(Diags), SavedLoc(Loc), IsIgnored(IsIgnored) {
- if (!IsIgnored) {
- Diags.setSeverity(/*Diag*/ diag::warn_uninit_self_reference_in_init,
- /*Map*/ diag::Severity::Ignored, Loc);
- }
- }
- ~DiagsUninitializedSeveretyRAII() {
- if (!IsIgnored)
- Diags.popMappings(SavedLoc);
- }
- };
- }
- OMPClause *Sema::ActOnOpenMPFirstprivateClause(ArrayRef<Expr *> VarList,
- SourceLocation StartLoc,
- SourceLocation LParenLoc,
- SourceLocation EndLoc) {
- SmallVector<Expr *, 8> Vars;
- SmallVector<Expr *, 8> PrivateCopies;
- SmallVector<Expr *, 8> Inits;
- SmallVector<Decl *, 4> ExprCaptures;
- bool IsImplicitClause =
- StartLoc.isInvalid() && LParenLoc.isInvalid() && EndLoc.isInvalid();
- SourceLocation ImplicitClauseLoc = DSAStack->getConstructLoc();
- for (Expr *RefExpr : VarList) {
- assert(RefExpr && "NULL expr in OpenMP firstprivate clause.");
- SourceLocation ELoc;
- SourceRange ERange;
- Expr *SimpleRefExpr = RefExpr;
- auto Res = getPrivateItem(*this, SimpleRefExpr, ELoc, ERange);
- if (Res.second) {
- // It will be analyzed later.
- Vars.push_back(RefExpr);
- PrivateCopies.push_back(nullptr);
- Inits.push_back(nullptr);
- }
- ValueDecl *D = Res.first;
- if (!D)
- continue;
- ELoc = IsImplicitClause ? ImplicitClauseLoc : ELoc;
- QualType Type = D->getType();
- auto *VD = dyn_cast<VarDecl>(D);
- // OpenMP [2.9.3.3, Restrictions, C/C++, p.3]
- // A variable that appears in a private clause must not have an incomplete
- // type or a reference type.
- if (RequireCompleteType(ELoc, Type,
- diag::err_omp_firstprivate_incomplete_type))
- continue;
- Type = Type.getNonReferenceType();
- // OpenMP [2.9.3.4, Restrictions, C/C++, p.1]
- // A variable of class type (or array thereof) that appears in a private
- // clause requires an accessible, unambiguous copy constructor for the
- // class type.
- QualType ElemType = Context.getBaseElementType(Type).getNonReferenceType();
- // If an implicit firstprivate variable found it was checked already.
- DSAStackTy::DSAVarData TopDVar;
- if (!IsImplicitClause) {
- DSAStackTy::DSAVarData DVar =
- DSAStack->getTopDSA(D, /*FromParent=*/false);
- TopDVar = DVar;
- OpenMPDirectiveKind CurrDir = DSAStack->getCurrentDirective();
- bool IsConstant = ElemType.isConstant(Context);
- // OpenMP [2.4.13, Data-sharing Attribute Clauses]
- // A list item that specifies a given variable may not appear in more
- // than one clause on the same directive, except that a variable may be
- // specified in both firstprivate and lastprivate clauses.
- // OpenMP 4.5 [2.10.8, Distribute Construct, p.3]
- // A list item may appear in a firstprivate or lastprivate clause but not
- // both.
- if (DVar.CKind != OMPC_unknown && DVar.CKind != OMPC_firstprivate &&
- (isOpenMPDistributeDirective(CurrDir) ||
- DVar.CKind != OMPC_lastprivate) &&
- DVar.RefExpr) {
- Diag(ELoc, diag::err_omp_wrong_dsa)
- << getOpenMPClauseName(DVar.CKind)
- << getOpenMPClauseName(OMPC_firstprivate);
- reportOriginalDsa(*this, DSAStack, D, DVar);
- continue;
- }
- // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced
- // in a Construct]
- // Variables with the predetermined data-sharing attributes may not be
- // listed in data-sharing attributes clauses, except for the cases
- // listed below. For these exceptions only, listing a predetermined
- // variable in a data-sharing attribute clause is allowed and overrides
- // the variable's predetermined data-sharing attributes.
- // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced
- // in a Construct, C/C++, p.2]
- // Variables with const-qualified type having no mutable member may be
- // listed in a firstprivate clause, even if they are static data members.
- if (!(IsConstant || (VD && VD->isStaticDataMember())) && !DVar.RefExpr &&
- DVar.CKind != OMPC_unknown && DVar.CKind != OMPC_shared) {
- Diag(ELoc, diag::err_omp_wrong_dsa)
- << getOpenMPClauseName(DVar.CKind)
- << getOpenMPClauseName(OMPC_firstprivate);
- reportOriginalDsa(*this, DSAStack, D, DVar);
- continue;
- }
- // OpenMP [2.9.3.4, Restrictions, p.2]
- // A list item that is private within a parallel region must not appear
- // in a firstprivate clause on a worksharing construct if any of the
- // worksharing regions arising from the worksharing construct ever bind
- // to any of the parallel regions arising from the parallel construct.
- // OpenMP 4.5 [2.15.3.4, Restrictions, p.3]
- // A list item that is private within a teams region must not appear in a
- // firstprivate clause on a distribute construct if any of the distribute
- // regions arising from the distribute construct ever bind to any of the
- // teams regions arising from the teams construct.
- // OpenMP 4.5 [2.15.3.4, Restrictions, p.3]
- // A list item that appears in a reduction clause of a teams construct
- // must not appear in a firstprivate clause on a distribute construct if
- // any of the distribute regions arising from the distribute construct
- // ever bind to any of the teams regions arising from the teams construct.
- if ((isOpenMPWorksharingDirective(CurrDir) ||
- isOpenMPDistributeDirective(CurrDir)) &&
- !isOpenMPParallelDirective(CurrDir) &&
- !isOpenMPTeamsDirective(CurrDir)) {
- DVar = DSAStack->getImplicitDSA(D, true);
- if (DVar.CKind != OMPC_shared &&
- (isOpenMPParallelDirective(DVar.DKind) ||
- isOpenMPTeamsDirective(DVar.DKind) ||
- DVar.DKind == OMPD_unknown)) {
- Diag(ELoc, diag::err_omp_required_access)
- << getOpenMPClauseName(OMPC_firstprivate)
- << getOpenMPClauseName(OMPC_shared);
- reportOriginalDsa(*this, DSAStack, D, DVar);
- continue;
- }
- }
- // OpenMP [2.9.3.4, Restrictions, p.3]
- // A list item that appears in a reduction clause of a parallel construct
- // must not appear in a firstprivate clause on a worksharing or task
- // construct if any of the worksharing or task regions arising from the
- // worksharing or task construct ever bind to any of the parallel regions
- // arising from the parallel construct.
- // OpenMP [2.9.3.4, Restrictions, p.4]
- // A list item that appears in a reduction clause in worksharing
- // construct must not appear in a firstprivate clause in a task construct
- // encountered during execution of any of the worksharing regions arising
- // from the worksharing construct.
- if (isOpenMPTaskingDirective(CurrDir)) {
- DVar = DSAStack->hasInnermostDSA(
- D, [](OpenMPClauseKind C) { return C == OMPC_reduction; },
- [](OpenMPDirectiveKind K) {
- return isOpenMPParallelDirective(K) ||
- isOpenMPWorksharingDirective(K) ||
- isOpenMPTeamsDirective(K);
- },
- /*FromParent=*/true);
- if (DVar.CKind == OMPC_reduction &&
- (isOpenMPParallelDirective(DVar.DKind) ||
- isOpenMPWorksharingDirective(DVar.DKind) ||
- isOpenMPTeamsDirective(DVar.DKind))) {
- Diag(ELoc, diag::err_omp_parallel_reduction_in_task_firstprivate)
- << getOpenMPDirectiveName(DVar.DKind);
- reportOriginalDsa(*this, DSAStack, D, DVar);
- continue;
- }
- }
- // OpenMP 4.5 [2.15.5.1, Restrictions, p.3]
- // A list item cannot appear in both a map clause and a data-sharing
- // attribute clause on the same construct
- if (isOpenMPTargetExecutionDirective(CurrDir)) {
- OpenMPClauseKind ConflictKind;
- if (DSAStack->checkMappableExprComponentListsForDecl(
- VD, /*CurrentRegionOnly=*/true,
- [&ConflictKind](
- OMPClauseMappableExprCommon::MappableExprComponentListRef,
- OpenMPClauseKind WhereFoundClauseKind) {
- ConflictKind = WhereFoundClauseKind;
- return true;
- })) {
- Diag(ELoc, diag::err_omp_variable_in_given_clause_and_dsa)
- << getOpenMPClauseName(OMPC_firstprivate)
- << getOpenMPClauseName(ConflictKind)
- << getOpenMPDirectiveName(DSAStack->getCurrentDirective());
- reportOriginalDsa(*this, DSAStack, D, DVar);
- continue;
- }
- }
- }
- // Variably modified types are not supported for tasks.
- if (!Type->isAnyPointerType() && Type->isVariablyModifiedType() &&
- isOpenMPTaskingDirective(DSAStack->getCurrentDirective())) {
- Diag(ELoc, diag::err_omp_variably_modified_type_not_supported)
- << getOpenMPClauseName(OMPC_firstprivate) << Type
- << getOpenMPDirectiveName(DSAStack->getCurrentDirective());
- bool IsDecl =
- !VD ||
- VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly;
- Diag(D->getLocation(),
- IsDecl ? diag::note_previous_decl : diag::note_defined_here)
- << D;
- continue;
- }
- Type = Type.getUnqualifiedType();
- VarDecl *VDPrivate =
- buildVarDecl(*this, ELoc, Type, D->getName(),
- D->hasAttrs() ? &D->getAttrs() : nullptr,
- VD ? cast<DeclRefExpr>(SimpleRefExpr) : nullptr);
- // Generate helper private variable and initialize it with the value of the
- // original variable. The address of the original variable is replaced by
- // the address of the new private variable in the CodeGen. This new variable
- // is not added to IdResolver, so the code in the OpenMP region uses
- // original variable for proper diagnostics and variable capturing.
- Expr *VDInitRefExpr = nullptr;
- // For arrays generate initializer for single element and replace it by the
- // original array element in CodeGen.
- if (Type->isArrayType()) {
- VarDecl *VDInit =
- buildVarDecl(*this, RefExpr->getExprLoc(), ElemType, D->getName());
- VDInitRefExpr = buildDeclRefExpr(*this, VDInit, ElemType, ELoc);
- Expr *Init = DefaultLvalueConversion(VDInitRefExpr).get();
- ElemType = ElemType.getUnqualifiedType();
- VarDecl *VDInitTemp = buildVarDecl(*this, RefExpr->getExprLoc(), ElemType,
- ".firstprivate.temp");
- InitializedEntity Entity =
- InitializedEntity::InitializeVariable(VDInitTemp);
- InitializationKind Kind = InitializationKind::CreateCopy(ELoc, ELoc);
- InitializationSequence InitSeq(*this, Entity, Kind, Init);
- ExprResult Result = InitSeq.Perform(*this, Entity, Kind, Init);
- if (Result.isInvalid())
- VDPrivate->setInvalidDecl();
- else
- VDPrivate->setInit(Result.getAs<Expr>());
- // Remove temp variable declaration.
- Context.Deallocate(VDInitTemp);
- } else {
- VarDecl *VDInit = buildVarDecl(*this, RefExpr->getExprLoc(), Type,
- ".firstprivate.temp");
- VDInitRefExpr = buildDeclRefExpr(*this, VDInit, RefExpr->getType(),
- RefExpr->getExprLoc());
- AddInitializerToDecl(VDPrivate,
- DefaultLvalueConversion(VDInitRefExpr).get(),
- /*DirectInit=*/false);
- }
- if (VDPrivate->isInvalidDecl()) {
- if (IsImplicitClause) {
- Diag(RefExpr->getExprLoc(),
- diag::note_omp_task_predetermined_firstprivate_here);
- }
- continue;
- }
- CurContext->addDecl(VDPrivate);
- DeclRefExpr *VDPrivateRefExpr = buildDeclRefExpr(
- *this, VDPrivate, RefExpr->getType().getUnqualifiedType(),
- RefExpr->getExprLoc());
- DeclRefExpr *Ref = nullptr;
- if (!VD && !CurContext->isDependentContext()) {
- if (TopDVar.CKind == OMPC_lastprivate) {
- Ref = TopDVar.PrivateCopy;
- } else {
- Ref = buildCapture(*this, D, SimpleRefExpr, /*WithInit=*/true);
- if (!isOpenMPCapturedDecl(D))
- ExprCaptures.push_back(Ref->getDecl());
- }
- }
- DSAStack->addDSA(D, RefExpr->IgnoreParens(), OMPC_firstprivate, Ref);
- Vars.push_back((VD || CurContext->isDependentContext())
- ? RefExpr->IgnoreParens()
- : Ref);
- PrivateCopies.push_back(VDPrivateRefExpr);
- Inits.push_back(VDInitRefExpr);
- }
- if (Vars.empty())
- return nullptr;
- return OMPFirstprivateClause::Create(Context, StartLoc, LParenLoc, EndLoc,
- Vars, PrivateCopies, Inits,
- buildPreInits(Context, ExprCaptures));
- }
- OMPClause *Sema::ActOnOpenMPLastprivateClause(ArrayRef<Expr *> VarList,
- SourceLocation StartLoc,
- SourceLocation LParenLoc,
- SourceLocation EndLoc) {
- SmallVector<Expr *, 8> Vars;
- SmallVector<Expr *, 8> SrcExprs;
- SmallVector<Expr *, 8> DstExprs;
- SmallVector<Expr *, 8> AssignmentOps;
- SmallVector<Decl *, 4> ExprCaptures;
- SmallVector<Expr *, 4> ExprPostUpdates;
- for (Expr *RefExpr : VarList) {
- assert(RefExpr && "NULL expr in OpenMP lastprivate clause.");
- SourceLocation ELoc;
- SourceRange ERange;
- Expr *SimpleRefExpr = RefExpr;
- auto Res = getPrivateItem(*this, SimpleRefExpr, ELoc, ERange);
- if (Res.second) {
- // It will be analyzed later.
- Vars.push_back(RefExpr);
- SrcExprs.push_back(nullptr);
- DstExprs.push_back(nullptr);
- AssignmentOps.push_back(nullptr);
- }
- ValueDecl *D = Res.first;
- if (!D)
- continue;
- QualType Type = D->getType();
- auto *VD = dyn_cast<VarDecl>(D);
- // OpenMP [2.14.3.5, Restrictions, C/C++, p.2]
- // A variable that appears in a lastprivate clause must not have an
- // incomplete type or a reference type.
- if (RequireCompleteType(ELoc, Type,
- diag::err_omp_lastprivate_incomplete_type))
- continue;
- Type = Type.getNonReferenceType();
- OpenMPDirectiveKind CurrDir = DSAStack->getCurrentDirective();
- // OpenMP [2.14.1.1, Data-sharing Attribute Rules for Variables Referenced
- // in a Construct]
- // Variables with the predetermined data-sharing attributes may not be
- // listed in data-sharing attributes clauses, except for the cases
- // listed below.
- // OpenMP 4.5 [2.10.8, Distribute Construct, p.3]
- // A list item may appear in a firstprivate or lastprivate clause but not
- // both.
- DSAStackTy::DSAVarData DVar = DSAStack->getTopDSA(D, /*FromParent=*/false);
- if (DVar.CKind != OMPC_unknown && DVar.CKind != OMPC_lastprivate &&
- (isOpenMPDistributeDirective(CurrDir) ||
- DVar.CKind != OMPC_firstprivate) &&
- (DVar.CKind != OMPC_private || DVar.RefExpr != nullptr)) {
- Diag(ELoc, diag::err_omp_wrong_dsa)
- << getOpenMPClauseName(DVar.CKind)
- << getOpenMPClauseName(OMPC_lastprivate);
- reportOriginalDsa(*this, DSAStack, D, DVar);
- continue;
- }
- // OpenMP [2.14.3.5, Restrictions, p.2]
- // A list item that is private within a parallel region, or that appears in
- // the reduction clause of a parallel construct, must not appear in a
- // lastprivate clause on a worksharing construct if any of the corresponding
- // worksharing regions ever binds to any of the corresponding parallel
- // regions.
- DSAStackTy::DSAVarData TopDVar = DVar;
- if (isOpenMPWorksharingDirective(CurrDir) &&
- !isOpenMPParallelDirective(CurrDir) &&
- !isOpenMPTeamsDirective(CurrDir)) {
- DVar = DSAStack->getImplicitDSA(D, true);
- if (DVar.CKind != OMPC_shared) {
- Diag(ELoc, diag::err_omp_required_access)
- << getOpenMPClauseName(OMPC_lastprivate)
- << getOpenMPClauseName(OMPC_shared);
- reportOriginalDsa(*this, DSAStack, D, DVar);
- continue;
- }
- }
- // OpenMP [2.14.3.5, Restrictions, C++, p.1,2]
- // A variable of class type (or array thereof) that appears in a
- // lastprivate clause requires an accessible, unambiguous default
- // constructor for the class type, unless the list item is also specified
- // in a firstprivate clause.
- // A variable of class type (or array thereof) that appears in a
- // lastprivate clause requires an accessible, unambiguous copy assignment
- // operator for the class type.
- Type = Context.getBaseElementType(Type).getNonReferenceType();
- VarDecl *SrcVD = buildVarDecl(*this, ERange.getBegin(),
- Type.getUnqualifiedType(), ".lastprivate.src",
- D->hasAttrs() ? &D->getAttrs() : nullptr);
- DeclRefExpr *PseudoSrcExpr =
- buildDeclRefExpr(*this, SrcVD, Type.getUnqualifiedType(), ELoc);
- VarDecl *DstVD =
- buildVarDecl(*this, ERange.getBegin(), Type, ".lastprivate.dst",
- D->hasAttrs() ? &D->getAttrs() : nullptr);
- DeclRefExpr *PseudoDstExpr = buildDeclRefExpr(*this, DstVD, Type, ELoc);
- // For arrays generate assignment operation for single element and replace
- // it by the original array element in CodeGen.
- ExprResult AssignmentOp = BuildBinOp(/*S=*/nullptr, ELoc, BO_Assign,
- PseudoDstExpr, PseudoSrcExpr);
- if (AssignmentOp.isInvalid())
- continue;
- AssignmentOp = ActOnFinishFullExpr(AssignmentOp.get(), ELoc,
- /*DiscardedValue=*/true);
- if (AssignmentOp.isInvalid())
- continue;
- DeclRefExpr *Ref = nullptr;
- if (!VD && !CurContext->isDependentContext()) {
- if (TopDVar.CKind == OMPC_firstprivate) {
- Ref = TopDVar.PrivateCopy;
- } else {
- Ref = buildCapture(*this, D, SimpleRefExpr, /*WithInit=*/false);
- if (!isOpenMPCapturedDecl(D))
- ExprCaptures.push_back(Ref->getDecl());
- }
- if (TopDVar.CKind == OMPC_firstprivate ||
- (!isOpenMPCapturedDecl(D) &&
- Ref->getDecl()->hasAttr<OMPCaptureNoInitAttr>())) {
- ExprResult RefRes = DefaultLvalueConversion(Ref);
- if (!RefRes.isUsable())
- continue;
- ExprResult PostUpdateRes =
- BuildBinOp(DSAStack->getCurScope(), ELoc, BO_Assign, SimpleRefExpr,
- RefRes.get());
- if (!PostUpdateRes.isUsable())
- continue;
- ExprPostUpdates.push_back(
- IgnoredValueConversions(PostUpdateRes.get()).get());
- }
- }
- DSAStack->addDSA(D, RefExpr->IgnoreParens(), OMPC_lastprivate, Ref);
- Vars.push_back((VD || CurContext->isDependentContext())
- ? RefExpr->IgnoreParens()
- : Ref);
- SrcExprs.push_back(PseudoSrcExpr);
- DstExprs.push_back(PseudoDstExpr);
- AssignmentOps.push_back(AssignmentOp.get());
- }
- if (Vars.empty())
- return nullptr;
- return OMPLastprivateClause::Create(Context, StartLoc, LParenLoc, EndLoc,
- Vars, SrcExprs, DstExprs, AssignmentOps,
- buildPreInits(Context, ExprCaptures),
- buildPostUpdate(*this, ExprPostUpdates));
- }
- OMPClause *Sema::ActOnOpenMPSharedClause(ArrayRef<Expr *> VarList,
- SourceLocation StartLoc,
- SourceLocation LParenLoc,
- SourceLocation EndLoc) {
- SmallVector<Expr *, 8> Vars;
- for (Expr *RefExpr : VarList) {
- assert(RefExpr && "NULL expr in OpenMP lastprivate clause.");
- SourceLocation ELoc;
- SourceRange ERange;
- Expr *SimpleRefExpr = RefExpr;
- auto Res = getPrivateItem(*this, SimpleRefExpr, ELoc, ERange);
- if (Res.second) {
- // It will be analyzed later.
- Vars.push_back(RefExpr);
- }
- ValueDecl *D = Res.first;
- if (!D)
- continue;
- auto *VD = dyn_cast<VarDecl>(D);
- // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced
- // in a Construct]
- // Variables with the predetermined data-sharing attributes may not be
- // listed in data-sharing attributes clauses, except for the cases
- // listed below. For these exceptions only, listing a predetermined
- // variable in a data-sharing attribute clause is allowed and overrides
- // the variable's predetermined data-sharing attributes.
- DSAStackTy::DSAVarData DVar = DSAStack->getTopDSA(D, /*FromParent=*/false);
- if (DVar.CKind != OMPC_unknown && DVar.CKind != OMPC_shared &&
- DVar.RefExpr) {
- Diag(ELoc, diag::err_omp_wrong_dsa) << getOpenMPClauseName(DVar.CKind)
- << getOpenMPClauseName(OMPC_shared);
- reportOriginalDsa(*this, DSAStack, D, DVar);
- continue;
- }
- DeclRefExpr *Ref = nullptr;
- if (!VD && isOpenMPCapturedDecl(D) && !CurContext->isDependentContext())
- Ref = buildCapture(*this, D, SimpleRefExpr, /*WithInit=*/true);
- DSAStack->addDSA(D, RefExpr->IgnoreParens(), OMPC_shared, Ref);
- Vars.push_back((VD || !Ref || CurContext->isDependentContext())
- ? RefExpr->IgnoreParens()
- : Ref);
- }
- if (Vars.empty())
- return nullptr;
- return OMPSharedClause::Create(Context, StartLoc, LParenLoc, EndLoc, Vars);
- }
- namespace {
- class DSARefChecker : public StmtVisitor<DSARefChecker, bool> {
- DSAStackTy *Stack;
- public:
- bool VisitDeclRefExpr(DeclRefExpr *E) {
- if (auto *VD = dyn_cast<VarDecl>(E->getDecl())) {
- DSAStackTy::DSAVarData DVar = Stack->getTopDSA(VD, /*FromParent=*/false);
- if (DVar.CKind == OMPC_shared && !DVar.RefExpr)
- return false;
- if (DVar.CKind != OMPC_unknown)
- return true;
- DSAStackTy::DSAVarData DVarPrivate = Stack->hasDSA(
- VD, isOpenMPPrivate, [](OpenMPDirectiveKind) { return true; },
- /*FromParent=*/true);
- return DVarPrivate.CKind != OMPC_unknown;
- }
- return false;
- }
- bool VisitStmt(Stmt *S) {
- for (Stmt *Child : S->children()) {
- if (Child && Visit(Child))
- return true;
- }
- return false;
- }
- explicit DSARefChecker(DSAStackTy *S) : Stack(S) {}
- };
- } // namespace
- namespace {
- // Transform MemberExpression for specified FieldDecl of current class to
- // DeclRefExpr to specified OMPCapturedExprDecl.
- class TransformExprToCaptures : public TreeTransform<TransformExprToCaptures> {
- typedef TreeTransform<TransformExprToCaptures> BaseTransform;
- ValueDecl *Field = nullptr;
- DeclRefExpr *CapturedExpr = nullptr;
- public:
- TransformExprToCaptures(Sema &SemaRef, ValueDecl *FieldDecl)
- : BaseTransform(SemaRef), Field(FieldDecl), CapturedExpr(nullptr) {}
- ExprResult TransformMemberExpr(MemberExpr *E) {
- if (isa<CXXThisExpr>(E->getBase()->IgnoreParenImpCasts()) &&
- E->getMemberDecl() == Field) {
- CapturedExpr = buildCapture(SemaRef, Field, E, /*WithInit=*/false);
- return CapturedExpr;
- }
- return BaseTransform::TransformMemberExpr(E);
- }
- DeclRefExpr *getCapturedExpr() { return CapturedExpr; }
- };
- } // namespace
- template <typename T, typename U>
- static T filterLookupForUDR(SmallVectorImpl<U> &Lookups,
- const llvm::function_ref<T(ValueDecl *)> Gen) {
- for (U &Set : Lookups) {
- for (auto *D : Set) {
- if (T Res = Gen(cast<ValueDecl>(D)))
- return Res;
- }
- }
- return T();
- }
- static NamedDecl *findAcceptableDecl(Sema &SemaRef, NamedDecl *D) {
- assert(!LookupResult::isVisible(SemaRef, D) && "not in slow case");
- for (auto RD : D->redecls()) {
- // Don't bother with extra checks if we already know this one isn't visible.
- if (RD == D)
- continue;
- auto ND = cast<NamedDecl>(RD);
- if (LookupResult::isVisible(SemaRef, ND))
- return ND;
- }
- return nullptr;
- }
- static void
- argumentDependentLookup(Sema &SemaRef, const DeclarationNameInfo &ReductionId,
- SourceLocation Loc, QualType Ty,
- SmallVectorImpl<UnresolvedSet<8>> &Lookups) {
- // Find all of the associated namespaces and classes based on the
- // arguments we have.
- Sema::AssociatedNamespaceSet AssociatedNamespaces;
- Sema::AssociatedClassSet AssociatedClasses;
- OpaqueValueExpr OVE(Loc, Ty, VK_LValue);
- SemaRef.FindAssociatedClassesAndNamespaces(Loc, &OVE, AssociatedNamespaces,
- AssociatedClasses);
- // C++ [basic.lookup.argdep]p3:
- // Let X be the lookup set produced by unqualified lookup (3.4.1)
- // and let Y be the lookup set produced by argument dependent
- // lookup (defined as follows). If X contains [...] then Y is
- // empty. Otherwise Y is the set of declarations found in the
- // namespaces associated with the argument types as described
- // below. The set of declarations found by the lookup of the name
- // is the union of X and Y.
- //
- // Here, we compute Y and add its members to the overloaded
- // candidate set.
- for (auto *NS : AssociatedNamespaces) {
- // When considering an associated namespace, the lookup is the
- // same as the lookup performed when the associated namespace is
- // used as a qualifier (3.4.3.2) except that:
- //
- // -- Any using-directives in the associated namespace are
- // ignored.
- //
- // -- Any namespace-scope friend functions declared in
- // associated classes are visible within their respective
- // namespaces even if they are not visible during an ordinary
- // lookup (11.4).
- DeclContext::lookup_result R = NS->lookup(ReductionId.getName());
- for (auto *D : R) {
- auto *Underlying = D;
- if (auto *USD = dyn_cast<UsingShadowDecl>(D))
- Underlying = USD->getTargetDecl();
- if (!isa<OMPDeclareReductionDecl>(Underlying))
- continue;
- if (!SemaRef.isVisible(D)) {
- D = findAcceptableDecl(SemaRef, D);
- if (!D)
- continue;
- if (auto *USD = dyn_cast<UsingShadowDecl>(D))
- Underlying = USD->getTargetDecl();
- }
- Lookups.emplace_back();
- Lookups.back().addDecl(Underlying);
- }
- }
- }
- static ExprResult
- buildDeclareReductionRef(Sema &SemaRef, SourceLocation Loc, SourceRange Range,
- Scope *S, CXXScopeSpec &ReductionIdScopeSpec,
- const DeclarationNameInfo &ReductionId, QualType Ty,
- CXXCastPath &BasePath, Expr *UnresolvedReduction) {
- if (ReductionIdScopeSpec.isInvalid())
- return ExprError();
- SmallVector<UnresolvedSet<8>, 4> Lookups;
- if (S) {
- LookupResult Lookup(SemaRef, ReductionId, Sema::LookupOMPReductionName);
- Lookup.suppressDiagnostics();
- while (S && SemaRef.LookupParsedName(Lookup, S, &ReductionIdScopeSpec)) {
- NamedDecl *D = Lookup.getRepresentativeDecl();
- do {
- S = S->getParent();
- } while (S && !S->isDeclScope(D));
- if (S)
- S = S->getParent();
- Lookups.emplace_back();
- Lookups.back().append(Lookup.begin(), Lookup.end());
- Lookup.clear();
- }
- } else if (auto *ULE =
- cast_or_null<UnresolvedLookupExpr>(UnresolvedReduction)) {
- Lookups.push_back(UnresolvedSet<8>());
- Decl *PrevD = nullptr;
- for (NamedDecl *D : ULE->decls()) {
- if (D == PrevD)
- Lookups.push_back(UnresolvedSet<8>());
- else if (auto *DRD = cast<OMPDeclareReductionDecl>(D))
- Lookups.back().addDecl(DRD);
- PrevD = D;
- }
- }
- if (SemaRef.CurContext->isDependentContext() || Ty->isDependentType() ||
- Ty->isInstantiationDependentType() ||
- Ty->containsUnexpandedParameterPack() ||
- filterLookupForUDR<bool>(Lookups, [](ValueDecl *D) {
- return !D->isInvalidDecl() &&
- (D->getType()->isDependentType() ||
- D->getType()->isInstantiationDependentType() ||
- D->getType()->containsUnexpandedParameterPack());
- })) {
- UnresolvedSet<8> ResSet;
- for (const UnresolvedSet<8> &Set : Lookups) {
- if (Set.empty())
- continue;
- ResSet.append(Set.begin(), Set.end());
- // The last item marks the end of all declarations at the specified scope.
- ResSet.addDecl(Set[Set.size() - 1]);
- }
- return UnresolvedLookupExpr::Create(
- SemaRef.Context, /*NamingClass=*/nullptr,
- ReductionIdScopeSpec.getWithLocInContext(SemaRef.Context), ReductionId,
- /*ADL=*/true, /*Overloaded=*/true, ResSet.begin(), ResSet.end());
- }
- // Lookup inside the classes.
- // C++ [over.match.oper]p3:
- // For a unary operator @ with an operand of a type whose
- // cv-unqualified version is T1, and for a binary operator @ with
- // a left operand of a type whose cv-unqualified version is T1 and
- // a right operand of a type whose cv-unqualified version is T2,
- // three sets of candidate functions, designated member
- // candidates, non-member candidates and built-in candidates, are
- // constructed as follows:
- // -- If T1 is a complete class type or a class currently being
- // defined, the set of member candidates is the result of the
- // qualified lookup of T1::operator@ (13.3.1.1.1); otherwise,
- // the set of member candidates is empty.
- LookupResult Lookup(SemaRef, ReductionId, Sema::LookupOMPReductionName);
- Lookup.suppressDiagnostics();
- if (const auto *TyRec = Ty->getAs<RecordType>()) {
- // Complete the type if it can be completed.
- // If the type is neither complete nor being defined, bail out now.
- if (SemaRef.isCompleteType(Loc, Ty) || TyRec->isBeingDefined() ||
- TyRec->getDecl()->getDefinition()) {
- Lookup.clear();
- SemaRef.LookupQualifiedName(Lookup, TyRec->getDecl());
- if (Lookup.empty()) {
- Lookups.emplace_back();
- Lookups.back().append(Lookup.begin(), Lookup.end());
- }
- }
- }
- // Perform ADL.
- argumentDependentLookup(SemaRef, ReductionId, Loc, Ty, Lookups);
- if (auto *VD = filterLookupForUDR<ValueDecl *>(
- Lookups, [&SemaRef, Ty](ValueDecl *D) -> ValueDecl * {
- if (!D->isInvalidDecl() &&
- SemaRef.Context.hasSameType(D->getType(), Ty))
- return D;
- return nullptr;
- }))
- return SemaRef.BuildDeclRefExpr(VD, Ty, VK_LValue, Loc);
- if (auto *VD = filterLookupForUDR<ValueDecl *>(
- Lookups, [&SemaRef, Ty, Loc](ValueDecl *D) -> ValueDecl * {
- if (!D->isInvalidDecl() &&
- SemaRef.IsDerivedFrom(Loc, Ty, D->getType()) &&
- !Ty.isMoreQualifiedThan(D->getType()))
- return D;
- return nullptr;
- })) {
- CXXBasePaths Paths(/*FindAmbiguities=*/true, /*RecordPaths=*/true,
- /*DetectVirtual=*/false);
- if (SemaRef.IsDerivedFrom(Loc, Ty, VD->getType(), Paths)) {
- if (!Paths.isAmbiguous(SemaRef.Context.getCanonicalType(
- VD->getType().getUnqualifiedType()))) {
- if (SemaRef.CheckBaseClassAccess(Loc, VD->getType(), Ty, Paths.front(),
- /*DiagID=*/0) !=
- Sema::AR_inaccessible) {
- SemaRef.BuildBasePathArray(Paths, BasePath);
- return SemaRef.BuildDeclRefExpr(VD, Ty, VK_LValue, Loc);
- }
- }
- }
- }
- if (ReductionIdScopeSpec.isSet()) {
- SemaRef.Diag(Loc, diag::err_omp_not_resolved_reduction_identifier) << Range;
- return ExprError();
- }
- return ExprEmpty();
- }
- namespace {
- /// Data for the reduction-based clauses.
- struct ReductionData {
- /// List of original reduction items.
- SmallVector<Expr *, 8> Vars;
- /// List of private copies of the reduction items.
- SmallVector<Expr *, 8> Privates;
- /// LHS expressions for the reduction_op expressions.
- SmallVector<Expr *, 8> LHSs;
- /// RHS expressions for the reduction_op expressions.
- SmallVector<Expr *, 8> RHSs;
- /// Reduction operation expression.
- SmallVector<Expr *, 8> ReductionOps;
- /// Taskgroup descriptors for the corresponding reduction items in
- /// in_reduction clauses.
- SmallVector<Expr *, 8> TaskgroupDescriptors;
- /// List of captures for clause.
- SmallVector<Decl *, 4> ExprCaptures;
- /// List of postupdate expressions.
- SmallVector<Expr *, 4> ExprPostUpdates;
- ReductionData() = delete;
- /// Reserves required memory for the reduction data.
- ReductionData(unsigned Size) {
- Vars.reserve(Size);
- Privates.reserve(Size);
- LHSs.reserve(Size);
- RHSs.reserve(Size);
- ReductionOps.reserve(Size);
- TaskgroupDescriptors.reserve(Size);
- ExprCaptures.reserve(Size);
- ExprPostUpdates.reserve(Size);
- }
- /// Stores reduction item and reduction operation only (required for dependent
- /// reduction item).
- void push(Expr *Item, Expr *ReductionOp) {
- Vars.emplace_back(Item);
- Privates.emplace_back(nullptr);
- LHSs.emplace_back(nullptr);
- RHSs.emplace_back(nullptr);
- ReductionOps.emplace_back(ReductionOp);
- TaskgroupDescriptors.emplace_back(nullptr);
- }
- /// Stores reduction data.
- void push(Expr *Item, Expr *Private, Expr *LHS, Expr *RHS, Expr *ReductionOp,
- Expr *TaskgroupDescriptor) {
- Vars.emplace_back(Item);
- Privates.emplace_back(Private);
- LHSs.emplace_back(LHS);
- RHSs.emplace_back(RHS);
- ReductionOps.emplace_back(ReductionOp);
- TaskgroupDescriptors.emplace_back(TaskgroupDescriptor);
- }
- };
- } // namespace
- static bool checkOMPArraySectionConstantForReduction(
- ASTContext &Context, const OMPArraySectionExpr *OASE, bool &SingleElement,
- SmallVectorImpl<llvm::APSInt> &ArraySizes) {
- const Expr *Length = OASE->getLength();
- if (Length == nullptr) {
- // For array sections of the form [1:] or [:], we would need to analyze
- // the lower bound...
- if (OASE->getColonLoc().isValid())
- return false;
- // This is an array subscript which has implicit length 1!
- SingleElement = true;
- ArraySizes.push_back(llvm::APSInt::get(1));
- } else {
- Expr::EvalResult Result;
- if (!Length->EvaluateAsInt(Result, Context))
- return false;
- llvm::APSInt ConstantLengthValue = Result.Val.getInt();
- SingleElement = (ConstantLengthValue.getSExtValue() == 1);
- ArraySizes.push_back(ConstantLengthValue);
- }
- // Get the base of this array section and walk up from there.
- const Expr *Base = OASE->getBase()->IgnoreParenImpCasts();
- // We require length = 1 for all array sections except the right-most to
- // guarantee that the memory region is contiguous and has no holes in it.
- while (const auto *TempOASE = dyn_cast<OMPArraySectionExpr>(Base)) {
- Length = TempOASE->getLength();
- if (Length == nullptr) {
- // For array sections of the form [1:] or [:], we would need to analyze
- // the lower bound...
- if (OASE->getColonLoc().isValid())
- return false;
- // This is an array subscript which has implicit length 1!
- ArraySizes.push_back(llvm::APSInt::get(1));
- } else {
- Expr::EvalResult Result;
- if (!Length->EvaluateAsInt(Result, Context))
- return false;
- llvm::APSInt ConstantLengthValue = Result.Val.getInt();
- if (ConstantLengthValue.getSExtValue() != 1)
- return false;
- ArraySizes.push_back(ConstantLengthValue);
- }
- Base = TempOASE->getBase()->IgnoreParenImpCasts();
- }
- // If we have a single element, we don't need to add the implicit lengths.
- if (!SingleElement) {
- while (const auto *TempASE = dyn_cast<ArraySubscriptExpr>(Base)) {
- // Has implicit length 1!
- ArraySizes.push_back(llvm::APSInt::get(1));
- Base = TempASE->getBase()->IgnoreParenImpCasts();
- }
- }
- // This array section can be privatized as a single value or as a constant
- // sized array.
- return true;
- }
- static bool actOnOMPReductionKindClause(
- Sema &S, DSAStackTy *Stack, OpenMPClauseKind ClauseKind,
- ArrayRef<Expr *> VarList, SourceLocation StartLoc, SourceLocation LParenLoc,
- SourceLocation ColonLoc, SourceLocation EndLoc,
- CXXScopeSpec &ReductionIdScopeSpec, const DeclarationNameInfo &ReductionId,
- ArrayRef<Expr *> UnresolvedReductions, ReductionData &RD) {
- DeclarationName DN = ReductionId.getName();
- OverloadedOperatorKind OOK = DN.getCXXOverloadedOperator();
- BinaryOperatorKind BOK = BO_Comma;
- ASTContext &Context = S.Context;
- // OpenMP [2.14.3.6, reduction clause]
- // C
- // reduction-identifier is either an identifier or one of the following
- // operators: +, -, *, &, |, ^, && and ||
- // C++
- // reduction-identifier is either an id-expression or one of the following
- // operators: +, -, *, &, |, ^, && and ||
- switch (OOK) {
- case OO_Plus:
- case OO_Minus:
- BOK = BO_Add;
- break;
- case OO_Star:
- BOK = BO_Mul;
- break;
- case OO_Amp:
- BOK = BO_And;
- break;
- case OO_Pipe:
- BOK = BO_Or;
- break;
- case OO_Caret:
- BOK = BO_Xor;
- break;
- case OO_AmpAmp:
- BOK = BO_LAnd;
- break;
- case OO_PipePipe:
- BOK = BO_LOr;
- break;
- case OO_New:
- case OO_Delete:
- case OO_Array_New:
- case OO_Array_Delete:
- case OO_Slash:
- case OO_Percent:
- case OO_Tilde:
- case OO_Exclaim:
- case OO_Equal:
- case OO_Less:
- case OO_Greater:
- case OO_LessEqual:
- case OO_GreaterEqual:
- case OO_PlusEqual:
- case OO_MinusEqual:
- case OO_StarEqual:
- case OO_SlashEqual:
- case OO_PercentEqual:
- case OO_CaretEqual:
- case OO_AmpEqual:
- case OO_PipeEqual:
- case OO_LessLess:
- case OO_GreaterGreater:
- case OO_LessLessEqual:
- case OO_GreaterGreaterEqual:
- case OO_EqualEqual:
- case OO_ExclaimEqual:
- case OO_Spaceship:
- case OO_PlusPlus:
- case OO_MinusMinus:
- case OO_Comma:
- case OO_ArrowStar:
- case OO_Arrow:
- case OO_Call:
- case OO_Subscript:
- case OO_Conditional:
- case OO_Coawait:
- case NUM_OVERLOADED_OPERATORS:
- llvm_unreachable("Unexpected reduction identifier");
- case OO_None:
- if (IdentifierInfo *II = DN.getAsIdentifierInfo()) {
- if (II->isStr("max"))
- BOK = BO_GT;
- else if (II->isStr("min"))
- BOK = BO_LT;
- }
- break;
- }
- SourceRange ReductionIdRange;
- if (ReductionIdScopeSpec.isValid())
- ReductionIdRange.setBegin(ReductionIdScopeSpec.getBeginLoc());
- else
- ReductionIdRange.setBegin(ReductionId.getBeginLoc());
- ReductionIdRange.setEnd(ReductionId.getEndLoc());
- auto IR = UnresolvedReductions.begin(), ER = UnresolvedReductions.end();
- bool FirstIter = true;
- for (Expr *RefExpr : VarList) {
- assert(RefExpr && "nullptr expr in OpenMP reduction clause.");
- // OpenMP [2.1, C/C++]
- // A list item is a variable or array section, subject to the restrictions
- // specified in Section 2.4 on page 42 and in each of the sections
- // describing clauses and directives for which a list appears.
- // OpenMP [2.14.3.3, Restrictions, p.1]
- // A variable that is part of another variable (as an array or
- // structure element) cannot appear in a private clause.
- if (!FirstIter && IR != ER)
- ++IR;
- FirstIter = false;
- SourceLocation ELoc;
- SourceRange ERange;
- Expr *SimpleRefExpr = RefExpr;
- auto Res = getPrivateItem(S, SimpleRefExpr, ELoc, ERange,
- /*AllowArraySection=*/true);
- if (Res.second) {
- // Try to find 'declare reduction' corresponding construct before using
- // builtin/overloaded operators.
- QualType Type = Context.DependentTy;
- CXXCastPath BasePath;
- ExprResult DeclareReductionRef = buildDeclareReductionRef(
- S, ELoc, ERange, Stack->getCurScope(), ReductionIdScopeSpec,
- ReductionId, Type, BasePath, IR == ER ? nullptr : *IR);
- Expr *ReductionOp = nullptr;
- if (S.CurContext->isDependentContext() &&
- (DeclareReductionRef.isUnset() ||
- isa<UnresolvedLookupExpr>(DeclareReductionRef.get())))
- ReductionOp = DeclareReductionRef.get();
- // It will be analyzed later.
- RD.push(RefExpr, ReductionOp);
- }
- ValueDecl *D = Res.first;
- if (!D)
- continue;
- Expr *TaskgroupDescriptor = nullptr;
- QualType Type;
- auto *ASE = dyn_cast<ArraySubscriptExpr>(RefExpr->IgnoreParens());
- auto *OASE = dyn_cast<OMPArraySectionExpr>(RefExpr->IgnoreParens());
- if (ASE) {
- Type = ASE->getType().getNonReferenceType();
- } else if (OASE) {
- QualType BaseType =
- OMPArraySectionExpr::getBaseOriginalType(OASE->getBase());
- if (const auto *ATy = BaseType->getAsArrayTypeUnsafe())
- Type = ATy->getElementType();
- else
- Type = BaseType->getPointeeType();
- Type = Type.getNonReferenceType();
- } else {
- Type = Context.getBaseElementType(D->getType().getNonReferenceType());
- }
- auto *VD = dyn_cast<VarDecl>(D);
- // OpenMP [2.9.3.3, Restrictions, C/C++, p.3]
- // A variable that appears in a private clause must not have an incomplete
- // type or a reference type.
- if (S.RequireCompleteType(ELoc, D->getType(),
- diag::err_omp_reduction_incomplete_type))
- continue;
- // OpenMP [2.14.3.6, reduction clause, Restrictions]
- // A list item that appears in a reduction clause must not be
- // const-qualified.
- if (Type.getNonReferenceType().isConstant(Context)) {
- S.Diag(ELoc, diag::err_omp_const_reduction_list_item) << ERange;
- if (!ASE && !OASE) {
- bool IsDecl = !VD || VD->isThisDeclarationADefinition(Context) ==
- VarDecl::DeclarationOnly;
- S.Diag(D->getLocation(),
- IsDecl ? diag::note_previous_decl : diag::note_defined_here)
- << D;
- }
- continue;
- }
- OpenMPDirectiveKind CurrDir = Stack->getCurrentDirective();
- // OpenMP [2.9.3.6, Restrictions, C/C++, p.4]
- // If a list-item is a reference type then it must bind to the same object
- // for all threads of the team.
- if (!ASE && !OASE) {
- if (VD) {
- VarDecl *VDDef = VD->getDefinition();
- if (VD->getType()->isReferenceType() && VDDef && VDDef->hasInit()) {
- DSARefChecker Check(Stack);
- if (Check.Visit(VDDef->getInit())) {
- S.Diag(ELoc, diag::err_omp_reduction_ref_type_arg)
- << getOpenMPClauseName(ClauseKind) << ERange;
- S.Diag(VDDef->getLocation(), diag::note_defined_here) << VDDef;
- continue;
- }
- }
- }
- // OpenMP [2.14.1.1, Data-sharing Attribute Rules for Variables Referenced
- // in a Construct]
- // Variables with the predetermined data-sharing attributes may not be
- // listed in data-sharing attributes clauses, except for the cases
- // listed below. For these exceptions only, listing a predetermined
- // variable in a data-sharing attribute clause is allowed and overrides
- // the variable's predetermined data-sharing attributes.
- // OpenMP [2.14.3.6, Restrictions, p.3]
- // Any number of reduction clauses can be specified on the directive,
- // but a list item can appear only once in the reduction clauses for that
- // directive.
- DSAStackTy::DSAVarData DVar = Stack->getTopDSA(D, /*FromParent=*/false);
- if (DVar.CKind == OMPC_reduction) {
- S.Diag(ELoc, diag::err_omp_once_referenced)
- << getOpenMPClauseName(ClauseKind);
- if (DVar.RefExpr)
- S.Diag(DVar.RefExpr->getExprLoc(), diag::note_omp_referenced);
- continue;
- }
- if (DVar.CKind != OMPC_unknown) {
- S.Diag(ELoc, diag::err_omp_wrong_dsa)
- << getOpenMPClauseName(DVar.CKind)
- << getOpenMPClauseName(OMPC_reduction);
- reportOriginalDsa(S, Stack, D, DVar);
- continue;
- }
- // OpenMP [2.14.3.6, Restrictions, p.1]
- // A list item that appears in a reduction clause of a worksharing
- // construct must be shared in the parallel regions to which any of the
- // worksharing regions arising from the worksharing construct bind.
- if (isOpenMPWorksharingDirective(CurrDir) &&
- !isOpenMPParallelDirective(CurrDir) &&
- !isOpenMPTeamsDirective(CurrDir)) {
- DVar = Stack->getImplicitDSA(D, true);
- if (DVar.CKind != OMPC_shared) {
- S.Diag(ELoc, diag::err_omp_required_access)
- << getOpenMPClauseName(OMPC_reduction)
- << getOpenMPClauseName(OMPC_shared);
- reportOriginalDsa(S, Stack, D, DVar);
- continue;
- }
- }
- }
- // Try to find 'declare reduction' corresponding construct before using
- // builtin/overloaded operators.
- CXXCastPath BasePath;
- ExprResult DeclareReductionRef = buildDeclareReductionRef(
- S, ELoc, ERange, Stack->getCurScope(), ReductionIdScopeSpec,
- ReductionId, Type, BasePath, IR == ER ? nullptr : *IR);
- if (DeclareReductionRef.isInvalid())
- continue;
- if (S.CurContext->isDependentContext() &&
- (DeclareReductionRef.isUnset() ||
- isa<UnresolvedLookupExpr>(DeclareReductionRef.get()))) {
- RD.push(RefExpr, DeclareReductionRef.get());
- continue;
- }
- if (BOK == BO_Comma && DeclareReductionRef.isUnset()) {
- // Not allowed reduction identifier is found.
- S.Diag(ReductionId.getBeginLoc(),
- diag::err_omp_unknown_reduction_identifier)
- << Type << ReductionIdRange;
- continue;
- }
- // OpenMP [2.14.3.6, reduction clause, Restrictions]
- // The type of a list item that appears in a reduction clause must be valid
- // for the reduction-identifier. For a max or min reduction in C, the type
- // of the list item must be an allowed arithmetic data type: char, int,
- // float, double, or _Bool, possibly modified with long, short, signed, or
- // unsigned. For a max or min reduction in C++, the type of the list item
- // must be an allowed arithmetic data type: char, wchar_t, int, float,
- // double, or bool, possibly modified with long, short, signed, or unsigned.
- if (DeclareReductionRef.isUnset()) {
- if ((BOK == BO_GT || BOK == BO_LT) &&
- !(Type->isScalarType() ||
- (S.getLangOpts().CPlusPlus && Type->isArithmeticType()))) {
- S.Diag(ELoc, diag::err_omp_clause_not_arithmetic_type_arg)
- << getOpenMPClauseName(ClauseKind) << S.getLangOpts().CPlusPlus;
- if (!ASE && !OASE) {
- bool IsDecl = !VD || VD->isThisDeclarationADefinition(Context) ==
- VarDecl::DeclarationOnly;
- S.Diag(D->getLocation(),
- IsDecl ? diag::note_previous_decl : diag::note_defined_here)
- << D;
- }
- continue;
- }
- if ((BOK == BO_OrAssign || BOK == BO_AndAssign || BOK == BO_XorAssign) &&
- !S.getLangOpts().CPlusPlus && Type->isFloatingType()) {
- S.Diag(ELoc, diag::err_omp_clause_floating_type_arg)
- << getOpenMPClauseName(ClauseKind);
- if (!ASE && !OASE) {
- bool IsDecl = !VD || VD->isThisDeclarationADefinition(Context) ==
- VarDecl::DeclarationOnly;
- S.Diag(D->getLocation(),
- IsDecl ? diag::note_previous_decl : diag::note_defined_here)
- << D;
- }
- continue;
- }
- }
- Type = Type.getNonLValueExprType(Context).getUnqualifiedType();
- VarDecl *LHSVD = buildVarDecl(S, ELoc, Type, ".reduction.lhs",
- D->hasAttrs() ? &D->getAttrs() : nullptr);
- VarDecl *RHSVD = buildVarDecl(S, ELoc, Type, D->getName(),
- D->hasAttrs() ? &D->getAttrs() : nullptr);
- QualType PrivateTy = Type;
- // Try if we can determine constant lengths for all array sections and avoid
- // the VLA.
- bool ConstantLengthOASE = false;
- if (OASE) {
- bool SingleElement;
- llvm::SmallVector<llvm::APSInt, 4> ArraySizes;
- ConstantLengthOASE = checkOMPArraySectionConstantForReduction(
- Context, OASE, SingleElement, ArraySizes);
- // If we don't have a single element, we must emit a constant array type.
- if (ConstantLengthOASE && !SingleElement) {
- for (llvm::APSInt &Size : ArraySizes)
- PrivateTy = Context.getConstantArrayType(
- PrivateTy, Size, ArrayType::Normal, /*IndexTypeQuals=*/0);
- }
- }
- if ((OASE && !ConstantLengthOASE) ||
- (!OASE && !ASE &&
- D->getType().getNonReferenceType()->isVariablyModifiedType())) {
- if (!Context.getTargetInfo().isVLASupported() &&
- S.shouldDiagnoseTargetSupportFromOpenMP()) {
- S.Diag(ELoc, diag::err_omp_reduction_vla_unsupported) << !!OASE;
- S.Diag(ELoc, diag::note_vla_unsupported);
- continue;
- }
- // For arrays/array sections only:
- // Create pseudo array type for private copy. The size for this array will
- // be generated during codegen.
- // For array subscripts or single variables Private Ty is the same as Type
- // (type of the variable or single array element).
- PrivateTy = Context.getVariableArrayType(
- Type,
- new (Context) OpaqueValueExpr(ELoc, Context.getSizeType(), VK_RValue),
- ArrayType::Normal, /*IndexTypeQuals=*/0, SourceRange());
- } else if (!ASE && !OASE &&
- Context.getAsArrayType(D->getType().getNonReferenceType())) {
- PrivateTy = D->getType().getNonReferenceType();
- }
- // Private copy.
- VarDecl *PrivateVD =
- buildVarDecl(S, ELoc, PrivateTy, D->getName(),
- D->hasAttrs() ? &D->getAttrs() : nullptr,
- VD ? cast<DeclRefExpr>(SimpleRefExpr) : nullptr);
- // Add initializer for private variable.
- Expr *Init = nullptr;
- DeclRefExpr *LHSDRE = buildDeclRefExpr(S, LHSVD, Type, ELoc);
- DeclRefExpr *RHSDRE = buildDeclRefExpr(S, RHSVD, Type, ELoc);
- if (DeclareReductionRef.isUsable()) {
- auto *DRDRef = DeclareReductionRef.getAs<DeclRefExpr>();
- auto *DRD = cast<OMPDeclareReductionDecl>(DRDRef->getDecl());
- if (DRD->getInitializer()) {
- Init = DRDRef;
- RHSVD->setInit(DRDRef);
- RHSVD->setInitStyle(VarDecl::CallInit);
- }
- } else {
- switch (BOK) {
- case BO_Add:
- case BO_Xor:
- case BO_Or:
- case BO_LOr:
- // '+', '-', '^', '|', '||' reduction ops - initializer is '0'.
- if (Type->isScalarType() || Type->isAnyComplexType())
- Init = S.ActOnIntegerConstant(ELoc, /*Val=*/0).get();
- break;
- case BO_Mul:
- case BO_LAnd:
- if (Type->isScalarType() || Type->isAnyComplexType()) {
- // '*' and '&&' reduction ops - initializer is '1'.
- Init = S.ActOnIntegerConstant(ELoc, /*Val=*/1).get();
- }
- break;
- case BO_And: {
- // '&' reduction op - initializer is '~0'.
- QualType OrigType = Type;
- if (auto *ComplexTy = OrigType->getAs<ComplexType>())
- Type = ComplexTy->getElementType();
- if (Type->isRealFloatingType()) {
- llvm::APFloat InitValue =
- llvm::APFloat::getAllOnesValue(Context.getTypeSize(Type),
- /*isIEEE=*/true);
- Init = FloatingLiteral::Create(Context, InitValue, /*isexact=*/true,
- Type, ELoc);
- } else if (Type->isScalarType()) {
- uint64_t Size = Context.getTypeSize(Type);
- QualType IntTy = Context.getIntTypeForBitwidth(Size, /*Signed=*/0);
- llvm::APInt InitValue = llvm::APInt::getAllOnesValue(Size);
- Init = IntegerLiteral::Create(Context, InitValue, IntTy, ELoc);
- }
- if (Init && OrigType->isAnyComplexType()) {
- // Init = 0xFFFF + 0xFFFFi;
- auto *Im = new (Context) ImaginaryLiteral(Init, OrigType);
- Init = S.CreateBuiltinBinOp(ELoc, BO_Add, Init, Im).get();
- }
- Type = OrigType;
- break;
- }
- case BO_LT:
- case BO_GT: {
- // 'min' reduction op - initializer is 'Largest representable number in
- // the reduction list item type'.
- // 'max' reduction op - initializer is 'Least representable number in
- // the reduction list item type'.
- if (Type->isIntegerType() || Type->isPointerType()) {
- bool IsSigned = Type->hasSignedIntegerRepresentation();
- uint64_t Size = Context.getTypeSize(Type);
- QualType IntTy =
- Context.getIntTypeForBitwidth(Size, /*Signed=*/IsSigned);
- llvm::APInt InitValue =
- (BOK != BO_LT) ? IsSigned ? llvm::APInt::getSignedMinValue(Size)
- : llvm::APInt::getMinValue(Size)
- : IsSigned ? llvm::APInt::getSignedMaxValue(Size)
- : llvm::APInt::getMaxValue(Size);
- Init = IntegerLiteral::Create(Context, InitValue, IntTy, ELoc);
- if (Type->isPointerType()) {
- // Cast to pointer type.
- ExprResult CastExpr = S.BuildCStyleCastExpr(
- ELoc, Context.getTrivialTypeSourceInfo(Type, ELoc), ELoc, Init);
- if (CastExpr.isInvalid())
- continue;
- Init = CastExpr.get();
- }
- } else if (Type->isRealFloatingType()) {
- llvm::APFloat InitValue = llvm::APFloat::getLargest(
- Context.getFloatTypeSemantics(Type), BOK != BO_LT);
- Init = FloatingLiteral::Create(Context, InitValue, /*isexact=*/true,
- Type, ELoc);
- }
- break;
- }
- case BO_PtrMemD:
- case BO_PtrMemI:
- case BO_MulAssign:
- case BO_Div:
- case BO_Rem:
- case BO_Sub:
- case BO_Shl:
- case BO_Shr:
- case BO_LE:
- case BO_GE:
- case BO_EQ:
- case BO_NE:
- case BO_Cmp:
- case BO_AndAssign:
- case BO_XorAssign:
- case BO_OrAssign:
- case BO_Assign:
- case BO_AddAssign:
- case BO_SubAssign:
- case BO_DivAssign:
- case BO_RemAssign:
- case BO_ShlAssign:
- case BO_ShrAssign:
- case BO_Comma:
- llvm_unreachable("Unexpected reduction operation");
- }
- }
- if (Init && DeclareReductionRef.isUnset())
- S.AddInitializerToDecl(RHSVD, Init, /*DirectInit=*/false);
- else if (!Init)
- S.ActOnUninitializedDecl(RHSVD);
- if (RHSVD->isInvalidDecl())
- continue;
- if (!RHSVD->hasInit() && DeclareReductionRef.isUnset()) {
- S.Diag(ELoc, diag::err_omp_reduction_id_not_compatible)
- << Type << ReductionIdRange;
- bool IsDecl = !VD || VD->isThisDeclarationADefinition(Context) ==
- VarDecl::DeclarationOnly;
- S.Diag(D->getLocation(),
- IsDecl ? diag::note_previous_decl : diag::note_defined_here)
- << D;
- continue;
- }
- // Store initializer for single element in private copy. Will be used during
- // codegen.
- PrivateVD->setInit(RHSVD->getInit());
- PrivateVD->setInitStyle(RHSVD->getInitStyle());
- DeclRefExpr *PrivateDRE = buildDeclRefExpr(S, PrivateVD, PrivateTy, ELoc);
- ExprResult ReductionOp;
- if (DeclareReductionRef.isUsable()) {
- QualType RedTy = DeclareReductionRef.get()->getType();
- QualType PtrRedTy = Context.getPointerType(RedTy);
- ExprResult LHS = S.CreateBuiltinUnaryOp(ELoc, UO_AddrOf, LHSDRE);
- ExprResult RHS = S.CreateBuiltinUnaryOp(ELoc, UO_AddrOf, RHSDRE);
- if (!BasePath.empty()) {
- LHS = S.DefaultLvalueConversion(LHS.get());
- RHS = S.DefaultLvalueConversion(RHS.get());
- LHS = ImplicitCastExpr::Create(Context, PtrRedTy,
- CK_UncheckedDerivedToBase, LHS.get(),
- &BasePath, LHS.get()->getValueKind());
- RHS = ImplicitCastExpr::Create(Context, PtrRedTy,
- CK_UncheckedDerivedToBase, RHS.get(),
- &BasePath, RHS.get()->getValueKind());
- }
- FunctionProtoType::ExtProtoInfo EPI;
- QualType Params[] = {PtrRedTy, PtrRedTy};
- QualType FnTy = Context.getFunctionType(Context.VoidTy, Params, EPI);
- auto *OVE = new (Context) OpaqueValueExpr(
- ELoc, Context.getPointerType(FnTy), VK_RValue, OK_Ordinary,
- S.DefaultLvalueConversion(DeclareReductionRef.get()).get());
- Expr *Args[] = {LHS.get(), RHS.get()};
- ReductionOp = new (Context)
- CallExpr(Context, OVE, Args, Context.VoidTy, VK_RValue, ELoc);
- } else {
- ReductionOp = S.BuildBinOp(
- Stack->getCurScope(), ReductionId.getBeginLoc(), BOK, LHSDRE, RHSDRE);
- if (ReductionOp.isUsable()) {
- if (BOK != BO_LT && BOK != BO_GT) {
- ReductionOp =
- S.BuildBinOp(Stack->getCurScope(), ReductionId.getBeginLoc(),
- BO_Assign, LHSDRE, ReductionOp.get());
- } else {
- auto *ConditionalOp = new (Context)
- ConditionalOperator(ReductionOp.get(), ELoc, LHSDRE, ELoc, RHSDRE,
- Type, VK_LValue, OK_Ordinary);
- ReductionOp =
- S.BuildBinOp(Stack->getCurScope(), ReductionId.getBeginLoc(),
- BO_Assign, LHSDRE, ConditionalOp);
- }
- if (ReductionOp.isUsable())
- ReductionOp = S.ActOnFinishFullExpr(ReductionOp.get());
- }
- if (!ReductionOp.isUsable())
- continue;
- }
- // OpenMP [2.15.4.6, Restrictions, p.2]
- // A list item that appears in an in_reduction clause of a task construct
- // must appear in a task_reduction clause of a construct associated with a
- // taskgroup region that includes the participating task in its taskgroup
- // set. The construct associated with the innermost region that meets this
- // condition must specify the same reduction-identifier as the in_reduction
- // clause.
- if (ClauseKind == OMPC_in_reduction) {
- SourceRange ParentSR;
- BinaryOperatorKind ParentBOK;
- const Expr *ParentReductionOp;
- Expr *ParentBOKTD, *ParentReductionOpTD;
- DSAStackTy::DSAVarData ParentBOKDSA =
- Stack->getTopMostTaskgroupReductionData(D, ParentSR, ParentBOK,
- ParentBOKTD);
- DSAStackTy::DSAVarData ParentReductionOpDSA =
- Stack->getTopMostTaskgroupReductionData(
- D, ParentSR, ParentReductionOp, ParentReductionOpTD);
- bool IsParentBOK = ParentBOKDSA.DKind != OMPD_unknown;
- bool IsParentReductionOp = ParentReductionOpDSA.DKind != OMPD_unknown;
- if (!IsParentBOK && !IsParentReductionOp) {
- S.Diag(ELoc, diag::err_omp_in_reduction_not_task_reduction);
- continue;
- }
- if ((DeclareReductionRef.isUnset() && IsParentReductionOp) ||
- (DeclareReductionRef.isUsable() && IsParentBOK) || BOK != ParentBOK ||
- IsParentReductionOp) {
- bool EmitError = true;
- if (IsParentReductionOp && DeclareReductionRef.isUsable()) {
- llvm::FoldingSetNodeID RedId, ParentRedId;
- ParentReductionOp->Profile(ParentRedId, Context, /*Canonical=*/true);
- DeclareReductionRef.get()->Profile(RedId, Context,
- /*Canonical=*/true);
- EmitError = RedId != ParentRedId;
- }
- if (EmitError) {
- S.Diag(ReductionId.getBeginLoc(),
- diag::err_omp_reduction_identifier_mismatch)
- << ReductionIdRange << RefExpr->getSourceRange();
- S.Diag(ParentSR.getBegin(),
- diag::note_omp_previous_reduction_identifier)
- << ParentSR
- << (IsParentBOK ? ParentBOKDSA.RefExpr
- : ParentReductionOpDSA.RefExpr)
- ->getSourceRange();
- continue;
- }
- }
- TaskgroupDescriptor = IsParentBOK ? ParentBOKTD : ParentReductionOpTD;
- assert(TaskgroupDescriptor && "Taskgroup descriptor must be defined.");
- }
- DeclRefExpr *Ref = nullptr;
- Expr *VarsExpr = RefExpr->IgnoreParens();
- if (!VD && !S.CurContext->isDependentContext()) {
- if (ASE || OASE) {
- TransformExprToCaptures RebuildToCapture(S, D);
- VarsExpr =
- RebuildToCapture.TransformExpr(RefExpr->IgnoreParens()).get();
- Ref = RebuildToCapture.getCapturedExpr();
- } else {
- VarsExpr = Ref = buildCapture(S, D, SimpleRefExpr, /*WithInit=*/false);
- }
- if (!S.isOpenMPCapturedDecl(D)) {
- RD.ExprCaptures.emplace_back(Ref->getDecl());
- if (Ref->getDecl()->hasAttr<OMPCaptureNoInitAttr>()) {
- ExprResult RefRes = S.DefaultLvalueConversion(Ref);
- if (!RefRes.isUsable())
- continue;
- ExprResult PostUpdateRes =
- S.BuildBinOp(Stack->getCurScope(), ELoc, BO_Assign, SimpleRefExpr,
- RefRes.get());
- if (!PostUpdateRes.isUsable())
- continue;
- if (isOpenMPTaskingDirective(Stack->getCurrentDirective()) ||
- Stack->getCurrentDirective() == OMPD_taskgroup) {
- S.Diag(RefExpr->getExprLoc(),
- diag::err_omp_reduction_non_addressable_expression)
- << RefExpr->getSourceRange();
- continue;
- }
- RD.ExprPostUpdates.emplace_back(
- S.IgnoredValueConversions(PostUpdateRes.get()).get());
- }
- }
- }
- // All reduction items are still marked as reduction (to do not increase
- // code base size).
- Stack->addDSA(D, RefExpr->IgnoreParens(), OMPC_reduction, Ref);
- if (CurrDir == OMPD_taskgroup) {
- if (DeclareReductionRef.isUsable())
- Stack->addTaskgroupReductionData(D, ReductionIdRange,
- DeclareReductionRef.get());
- else
- Stack->addTaskgroupReductionData(D, ReductionIdRange, BOK);
- }
- RD.push(VarsExpr, PrivateDRE, LHSDRE, RHSDRE, ReductionOp.get(),
- TaskgroupDescriptor);
- }
- return RD.Vars.empty();
- }
- OMPClause *Sema::ActOnOpenMPReductionClause(
- ArrayRef<Expr *> VarList, SourceLocation StartLoc, SourceLocation LParenLoc,
- SourceLocation ColonLoc, SourceLocation EndLoc,
- CXXScopeSpec &ReductionIdScopeSpec, const DeclarationNameInfo &ReductionId,
- ArrayRef<Expr *> UnresolvedReductions) {
- ReductionData RD(VarList.size());
- if (actOnOMPReductionKindClause(*this, DSAStack, OMPC_reduction, VarList,
- StartLoc, LParenLoc, ColonLoc, EndLoc,
- ReductionIdScopeSpec, ReductionId,
- UnresolvedReductions, RD))
- return nullptr;
- return OMPReductionClause::Create(
- Context, StartLoc, LParenLoc, ColonLoc, EndLoc, RD.Vars,
- ReductionIdScopeSpec.getWithLocInContext(Context), ReductionId,
- RD.Privates, RD.LHSs, RD.RHSs, RD.ReductionOps,
- buildPreInits(Context, RD.ExprCaptures),
- buildPostUpdate(*this, RD.ExprPostUpdates));
- }
- OMPClause *Sema::ActOnOpenMPTaskReductionClause(
- ArrayRef<Expr *> VarList, SourceLocation StartLoc, SourceLocation LParenLoc,
- SourceLocation ColonLoc, SourceLocation EndLoc,
- CXXScopeSpec &ReductionIdScopeSpec, const DeclarationNameInfo &ReductionId,
- ArrayRef<Expr *> UnresolvedReductions) {
- ReductionData RD(VarList.size());
- if (actOnOMPReductionKindClause(*this, DSAStack, OMPC_task_reduction, VarList,
- StartLoc, LParenLoc, ColonLoc, EndLoc,
- ReductionIdScopeSpec, ReductionId,
- UnresolvedReductions, RD))
- return nullptr;
- return OMPTaskReductionClause::Create(
- Context, StartLoc, LParenLoc, ColonLoc, EndLoc, RD.Vars,
- ReductionIdScopeSpec.getWithLocInContext(Context), ReductionId,
- RD.Privates, RD.LHSs, RD.RHSs, RD.ReductionOps,
- buildPreInits(Context, RD.ExprCaptures),
- buildPostUpdate(*this, RD.ExprPostUpdates));
- }
- OMPClause *Sema::ActOnOpenMPInReductionClause(
- ArrayRef<Expr *> VarList, SourceLocation StartLoc, SourceLocation LParenLoc,
- SourceLocation ColonLoc, SourceLocation EndLoc,
- CXXScopeSpec &ReductionIdScopeSpec, const DeclarationNameInfo &ReductionId,
- ArrayRef<Expr *> UnresolvedReductions) {
- ReductionData RD(VarList.size());
- if (actOnOMPReductionKindClause(*this, DSAStack, OMPC_in_reduction, VarList,
- StartLoc, LParenLoc, ColonLoc, EndLoc,
- ReductionIdScopeSpec, ReductionId,
- UnresolvedReductions, RD))
- return nullptr;
- return OMPInReductionClause::Create(
- Context, StartLoc, LParenLoc, ColonLoc, EndLoc, RD.Vars,
- ReductionIdScopeSpec.getWithLocInContext(Context), ReductionId,
- RD.Privates, RD.LHSs, RD.RHSs, RD.ReductionOps, RD.TaskgroupDescriptors,
- buildPreInits(Context, RD.ExprCaptures),
- buildPostUpdate(*this, RD.ExprPostUpdates));
- }
- bool Sema::CheckOpenMPLinearModifier(OpenMPLinearClauseKind LinKind,
- SourceLocation LinLoc) {
- if ((!LangOpts.CPlusPlus && LinKind != OMPC_LINEAR_val) ||
- LinKind == OMPC_LINEAR_unknown) {
- Diag(LinLoc, diag::err_omp_wrong_linear_modifier) << LangOpts.CPlusPlus;
- return true;
- }
- return false;
- }
- bool Sema::CheckOpenMPLinearDecl(const ValueDecl *D, SourceLocation ELoc,
- OpenMPLinearClauseKind LinKind,
- QualType Type) {
- const auto *VD = dyn_cast_or_null<VarDecl>(D);
- // A variable must not have an incomplete type or a reference type.
- if (RequireCompleteType(ELoc, Type, diag::err_omp_linear_incomplete_type))
- return true;
- if ((LinKind == OMPC_LINEAR_uval || LinKind == OMPC_LINEAR_ref) &&
- !Type->isReferenceType()) {
- Diag(ELoc, diag::err_omp_wrong_linear_modifier_non_reference)
- << Type << getOpenMPSimpleClauseTypeName(OMPC_linear, LinKind);
- return true;
- }
- Type = Type.getNonReferenceType();
- // A list item must not be const-qualified.
- if (Type.isConstant(Context)) {
- Diag(ELoc, diag::err_omp_const_variable)
- << getOpenMPClauseName(OMPC_linear);
- if (D) {
- bool IsDecl =
- !VD ||
- VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly;
- Diag(D->getLocation(),
- IsDecl ? diag::note_previous_decl : diag::note_defined_here)
- << D;
- }
- return true;
- }
- // A list item must be of integral or pointer type.
- Type = Type.getUnqualifiedType().getCanonicalType();
- const auto *Ty = Type.getTypePtrOrNull();
- if (!Ty || (!Ty->isDependentType() && !Ty->isIntegralType(Context) &&
- !Ty->isPointerType())) {
- Diag(ELoc, diag::err_omp_linear_expected_int_or_ptr) << Type;
- if (D) {
- bool IsDecl =
- !VD ||
- VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly;
- Diag(D->getLocation(),
- IsDecl ? diag::note_previous_decl : diag::note_defined_here)
- << D;
- }
- return true;
- }
- return false;
- }
- OMPClause *Sema::ActOnOpenMPLinearClause(
- ArrayRef<Expr *> VarList, Expr *Step, SourceLocation StartLoc,
- SourceLocation LParenLoc, OpenMPLinearClauseKind LinKind,
- SourceLocation LinLoc, SourceLocation ColonLoc, SourceLocation EndLoc) {
- SmallVector<Expr *, 8> Vars;
- SmallVector<Expr *, 8> Privates;
- SmallVector<Expr *, 8> Inits;
- SmallVector<Decl *, 4> ExprCaptures;
- SmallVector<Expr *, 4> ExprPostUpdates;
- if (CheckOpenMPLinearModifier(LinKind, LinLoc))
- LinKind = OMPC_LINEAR_val;
- for (Expr *RefExpr : VarList) {
- assert(RefExpr && "NULL expr in OpenMP linear clause.");
- SourceLocation ELoc;
- SourceRange ERange;
- Expr *SimpleRefExpr = RefExpr;
- auto Res = getPrivateItem(*this, SimpleRefExpr, ELoc, ERange);
- if (Res.second) {
- // It will be analyzed later.
- Vars.push_back(RefExpr);
- Privates.push_back(nullptr);
- Inits.push_back(nullptr);
- }
- ValueDecl *D = Res.first;
- if (!D)
- continue;
- QualType Type = D->getType();
- auto *VD = dyn_cast<VarDecl>(D);
- // OpenMP [2.14.3.7, linear clause]
- // A list-item cannot appear in more than one linear clause.
- // A list-item that appears in a linear clause cannot appear in any
- // other data-sharing attribute clause.
- DSAStackTy::DSAVarData DVar = DSAStack->getTopDSA(D, /*FromParent=*/false);
- if (DVar.RefExpr) {
- Diag(ELoc, diag::err_omp_wrong_dsa) << getOpenMPClauseName(DVar.CKind)
- << getOpenMPClauseName(OMPC_linear);
- reportOriginalDsa(*this, DSAStack, D, DVar);
- continue;
- }
- if (CheckOpenMPLinearDecl(D, ELoc, LinKind, Type))
- continue;
- Type = Type.getNonReferenceType().getUnqualifiedType().getCanonicalType();
- // Build private copy of original var.
- VarDecl *Private =
- buildVarDecl(*this, ELoc, Type, D->getName(),
- D->hasAttrs() ? &D->getAttrs() : nullptr,
- VD ? cast<DeclRefExpr>(SimpleRefExpr) : nullptr);
- DeclRefExpr *PrivateRef = buildDeclRefExpr(*this, Private, Type, ELoc);
- // Build var to save initial value.
- VarDecl *Init = buildVarDecl(*this, ELoc, Type, ".linear.start");
- Expr *InitExpr;
- DeclRefExpr *Ref = nullptr;
- if (!VD && !CurContext->isDependentContext()) {
- Ref = buildCapture(*this, D, SimpleRefExpr, /*WithInit=*/false);
- if (!isOpenMPCapturedDecl(D)) {
- ExprCaptures.push_back(Ref->getDecl());
- if (Ref->getDecl()->hasAttr<OMPCaptureNoInitAttr>()) {
- ExprResult RefRes = DefaultLvalueConversion(Ref);
- if (!RefRes.isUsable())
- continue;
- ExprResult PostUpdateRes =
- BuildBinOp(DSAStack->getCurScope(), ELoc, BO_Assign,
- SimpleRefExpr, RefRes.get());
- if (!PostUpdateRes.isUsable())
- continue;
- ExprPostUpdates.push_back(
- IgnoredValueConversions(PostUpdateRes.get()).get());
- }
- }
- }
- if (LinKind == OMPC_LINEAR_uval)
- InitExpr = VD ? VD->getInit() : SimpleRefExpr;
- else
- InitExpr = VD ? SimpleRefExpr : Ref;
- AddInitializerToDecl(Init, DefaultLvalueConversion(InitExpr).get(),
- /*DirectInit=*/false);
- DeclRefExpr *InitRef = buildDeclRefExpr(*this, Init, Type, ELoc);
- DSAStack->addDSA(D, RefExpr->IgnoreParens(), OMPC_linear, Ref);
- Vars.push_back((VD || CurContext->isDependentContext())
- ? RefExpr->IgnoreParens()
- : Ref);
- Privates.push_back(PrivateRef);
- Inits.push_back(InitRef);
- }
- if (Vars.empty())
- return nullptr;
- Expr *StepExpr = Step;
- Expr *CalcStepExpr = nullptr;
- if (Step && !Step->isValueDependent() && !Step->isTypeDependent() &&
- !Step->isInstantiationDependent() &&
- !Step->containsUnexpandedParameterPack()) {
- SourceLocation StepLoc = Step->getBeginLoc();
- ExprResult Val = PerformOpenMPImplicitIntegerConversion(StepLoc, Step);
- if (Val.isInvalid())
- return nullptr;
- StepExpr = Val.get();
- // Build var to save the step value.
- VarDecl *SaveVar =
- buildVarDecl(*this, StepLoc, StepExpr->getType(), ".linear.step");
- ExprResult SaveRef =
- buildDeclRefExpr(*this, SaveVar, StepExpr->getType(), StepLoc);
- ExprResult CalcStep =
- BuildBinOp(CurScope, StepLoc, BO_Assign, SaveRef.get(), StepExpr);
- CalcStep = ActOnFinishFullExpr(CalcStep.get());
- // Warn about zero linear step (it would be probably better specified as
- // making corresponding variables 'const').
- llvm::APSInt Result;
- bool IsConstant = StepExpr->isIntegerConstantExpr(Result, Context);
- if (IsConstant && !Result.isNegative() && !Result.isStrictlyPositive())
- Diag(StepLoc, diag::warn_omp_linear_step_zero) << Vars[0]
- << (Vars.size() > 1);
- if (!IsConstant && CalcStep.isUsable()) {
- // Calculate the step beforehand instead of doing this on each iteration.
- // (This is not used if the number of iterations may be kfold-ed).
- CalcStepExpr = CalcStep.get();
- }
- }
- return OMPLinearClause::Create(Context, StartLoc, LParenLoc, LinKind, LinLoc,
- ColonLoc, EndLoc, Vars, Privates, Inits,
- StepExpr, CalcStepExpr,
- buildPreInits(Context, ExprCaptures),
- buildPostUpdate(*this, ExprPostUpdates));
- }
- static bool FinishOpenMPLinearClause(OMPLinearClause &Clause, DeclRefExpr *IV,
- Expr *NumIterations, Sema &SemaRef,
- Scope *S, DSAStackTy *Stack) {
- // Walk the vars and build update/final expressions for the CodeGen.
- SmallVector<Expr *, 8> Updates;
- SmallVector<Expr *, 8> Finals;
- Expr *Step = Clause.getStep();
- Expr *CalcStep = Clause.getCalcStep();
- // OpenMP [2.14.3.7, linear clause]
- // If linear-step is not specified it is assumed to be 1.
- if (!Step)
- Step = SemaRef.ActOnIntegerConstant(SourceLocation(), 1).get();
- else if (CalcStep)
- Step = cast<BinaryOperator>(CalcStep)->getLHS();
- bool HasErrors = false;
- auto CurInit = Clause.inits().begin();
- auto CurPrivate = Clause.privates().begin();
- OpenMPLinearClauseKind LinKind = Clause.getModifier();
- for (Expr *RefExpr : Clause.varlists()) {
- SourceLocation ELoc;
- SourceRange ERange;
- Expr *SimpleRefExpr = RefExpr;
- auto Res = getPrivateItem(SemaRef, SimpleRefExpr, ELoc, ERange);
- ValueDecl *D = Res.first;
- if (Res.second || !D) {
- Updates.push_back(nullptr);
- Finals.push_back(nullptr);
- HasErrors = true;
- continue;
- }
- auto &&Info = Stack->isLoopControlVariable(D);
- // OpenMP [2.15.11, distribute simd Construct]
- // A list item may not appear in a linear clause, unless it is the loop
- // iteration variable.
- if (isOpenMPDistributeDirective(Stack->getCurrentDirective()) &&
- isOpenMPSimdDirective(Stack->getCurrentDirective()) && !Info.first) {
- SemaRef.Diag(ELoc,
- diag::err_omp_linear_distribute_var_non_loop_iteration);
- Updates.push_back(nullptr);
- Finals.push_back(nullptr);
- HasErrors = true;
- continue;
- }
- Expr *InitExpr = *CurInit;
- // Build privatized reference to the current linear var.
- auto *DE = cast<DeclRefExpr>(SimpleRefExpr);
- Expr *CapturedRef;
- if (LinKind == OMPC_LINEAR_uval)
- CapturedRef = cast<VarDecl>(DE->getDecl())->getInit();
- else
- CapturedRef =
- buildDeclRefExpr(SemaRef, cast<VarDecl>(DE->getDecl()),
- DE->getType().getUnqualifiedType(), DE->getExprLoc(),
- /*RefersToCapture=*/true);
- // Build update: Var = InitExpr + IV * Step
- ExprResult Update;
- if (!Info.first)
- Update =
- buildCounterUpdate(SemaRef, S, RefExpr->getExprLoc(), *CurPrivate,
- InitExpr, IV, Step, /* Subtract */ false);
- else
- Update = *CurPrivate;
- Update = SemaRef.ActOnFinishFullExpr(Update.get(), DE->getBeginLoc(),
- /*DiscardedValue=*/true);
- // Build final: Var = InitExpr + NumIterations * Step
- ExprResult Final;
- if (!Info.first)
- Final =
- buildCounterUpdate(SemaRef, S, RefExpr->getExprLoc(), CapturedRef,
- InitExpr, NumIterations, Step, /*Subtract=*/false);
- else
- Final = *CurPrivate;
- Final = SemaRef.ActOnFinishFullExpr(Final.get(), DE->getBeginLoc(),
- /*DiscardedValue=*/true);
- if (!Update.isUsable() || !Final.isUsable()) {
- Updates.push_back(nullptr);
- Finals.push_back(nullptr);
- HasErrors = true;
- } else {
- Updates.push_back(Update.get());
- Finals.push_back(Final.get());
- }
- ++CurInit;
- ++CurPrivate;
- }
- Clause.setUpdates(Updates);
- Clause.setFinals(Finals);
- return HasErrors;
- }
- OMPClause *Sema::ActOnOpenMPAlignedClause(
- ArrayRef<Expr *> VarList, Expr *Alignment, SourceLocation StartLoc,
- SourceLocation LParenLoc, SourceLocation ColonLoc, SourceLocation EndLoc) {
- SmallVector<Expr *, 8> Vars;
- for (Expr *RefExpr : VarList) {
- assert(RefExpr && "NULL expr in OpenMP linear clause.");
- SourceLocation ELoc;
- SourceRange ERange;
- Expr *SimpleRefExpr = RefExpr;
- auto Res = getPrivateItem(*this, SimpleRefExpr, ELoc, ERange);
- if (Res.second) {
- // It will be analyzed later.
- Vars.push_back(RefExpr);
- }
- ValueDecl *D = Res.first;
- if (!D)
- continue;
- QualType QType = D->getType();
- auto *VD = dyn_cast<VarDecl>(D);
- // OpenMP [2.8.1, simd construct, Restrictions]
- // The type of list items appearing in the aligned clause must be
- // array, pointer, reference to array, or reference to pointer.
- QType = QType.getNonReferenceType().getUnqualifiedType().getCanonicalType();
- const Type *Ty = QType.getTypePtrOrNull();
- if (!Ty || (!Ty->isArrayType() && !Ty->isPointerType())) {
- Diag(ELoc, diag::err_omp_aligned_expected_array_or_ptr)
- << QType << getLangOpts().CPlusPlus << ERange;
- bool IsDecl =
- !VD ||
- VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly;
- Diag(D->getLocation(),
- IsDecl ? diag::note_previous_decl : diag::note_defined_here)
- << D;
- continue;
- }
- // OpenMP [2.8.1, simd construct, Restrictions]
- // A list-item cannot appear in more than one aligned clause.
- if (const Expr *PrevRef = DSAStack->addUniqueAligned(D, SimpleRefExpr)) {
- Diag(ELoc, diag::err_omp_aligned_twice) << 0 << ERange;
- Diag(PrevRef->getExprLoc(), diag::note_omp_explicit_dsa)
- << getOpenMPClauseName(OMPC_aligned);
- continue;
- }
- DeclRefExpr *Ref = nullptr;
- if (!VD && isOpenMPCapturedDecl(D))
- Ref = buildCapture(*this, D, SimpleRefExpr, /*WithInit=*/true);
- Vars.push_back(DefaultFunctionArrayConversion(
- (VD || !Ref) ? RefExpr->IgnoreParens() : Ref)
- .get());
- }
- // OpenMP [2.8.1, simd construct, Description]
- // The parameter of the aligned clause, alignment, must be a constant
- // positive integer expression.
- // If no optional parameter is specified, implementation-defined default
- // alignments for SIMD instructions on the target platforms are assumed.
- if (Alignment != nullptr) {
- ExprResult AlignResult =
- VerifyPositiveIntegerConstantInClause(Alignment, OMPC_aligned);
- if (AlignResult.isInvalid())
- return nullptr;
- Alignment = AlignResult.get();
- }
- if (Vars.empty())
- return nullptr;
- return OMPAlignedClause::Create(Context, StartLoc, LParenLoc, ColonLoc,
- EndLoc, Vars, Alignment);
- }
- OMPClause *Sema::ActOnOpenMPCopyinClause(ArrayRef<Expr *> VarList,
- SourceLocation StartLoc,
- SourceLocation LParenLoc,
- SourceLocation EndLoc) {
- SmallVector<Expr *, 8> Vars;
- SmallVector<Expr *, 8> SrcExprs;
- SmallVector<Expr *, 8> DstExprs;
- SmallVector<Expr *, 8> AssignmentOps;
- for (Expr *RefExpr : VarList) {
- assert(RefExpr && "NULL expr in OpenMP copyin clause.");
- if (isa<DependentScopeDeclRefExpr>(RefExpr)) {
- // It will be analyzed later.
- Vars.push_back(RefExpr);
- SrcExprs.push_back(nullptr);
- DstExprs.push_back(nullptr);
- AssignmentOps.push_back(nullptr);
- continue;
- }
- SourceLocation ELoc = RefExpr->getExprLoc();
- // OpenMP [2.1, C/C++]
- // A list item is a variable name.
- // OpenMP [2.14.4.1, Restrictions, p.1]
- // A list item that appears in a copyin clause must be threadprivate.
- auto *DE = dyn_cast<DeclRefExpr>(RefExpr);
- if (!DE || !isa<VarDecl>(DE->getDecl())) {
- Diag(ELoc, diag::err_omp_expected_var_name_member_expr)
- << 0 << RefExpr->getSourceRange();
- continue;
- }
- Decl *D = DE->getDecl();
- auto *VD = cast<VarDecl>(D);
- QualType Type = VD->getType();
- if (Type->isDependentType() || Type->isInstantiationDependentType()) {
- // It will be analyzed later.
- Vars.push_back(DE);
- SrcExprs.push_back(nullptr);
- DstExprs.push_back(nullptr);
- AssignmentOps.push_back(nullptr);
- continue;
- }
- // OpenMP [2.14.4.1, Restrictions, C/C++, p.1]
- // A list item that appears in a copyin clause must be threadprivate.
- if (!DSAStack->isThreadPrivate(VD)) {
- Diag(ELoc, diag::err_omp_required_access)
- << getOpenMPClauseName(OMPC_copyin)
- << getOpenMPDirectiveName(OMPD_threadprivate);
- continue;
- }
- // OpenMP [2.14.4.1, Restrictions, C/C++, p.2]
- // A variable of class type (or array thereof) that appears in a
- // copyin clause requires an accessible, unambiguous copy assignment
- // operator for the class type.
- QualType ElemType = Context.getBaseElementType(Type).getNonReferenceType();
- VarDecl *SrcVD =
- buildVarDecl(*this, DE->getBeginLoc(), ElemType.getUnqualifiedType(),
- ".copyin.src", VD->hasAttrs() ? &VD->getAttrs() : nullptr);
- DeclRefExpr *PseudoSrcExpr = buildDeclRefExpr(
- *this, SrcVD, ElemType.getUnqualifiedType(), DE->getExprLoc());
- VarDecl *DstVD =
- buildVarDecl(*this, DE->getBeginLoc(), ElemType, ".copyin.dst",
- VD->hasAttrs() ? &VD->getAttrs() : nullptr);
- DeclRefExpr *PseudoDstExpr =
- buildDeclRefExpr(*this, DstVD, ElemType, DE->getExprLoc());
- // For arrays generate assignment operation for single element and replace
- // it by the original array element in CodeGen.
- ExprResult AssignmentOp =
- BuildBinOp(/*S=*/nullptr, DE->getExprLoc(), BO_Assign, PseudoDstExpr,
- PseudoSrcExpr);
- if (AssignmentOp.isInvalid())
- continue;
- AssignmentOp = ActOnFinishFullExpr(AssignmentOp.get(), DE->getExprLoc(),
- /*DiscardedValue=*/true);
- if (AssignmentOp.isInvalid())
- continue;
- DSAStack->addDSA(VD, DE, OMPC_copyin);
- Vars.push_back(DE);
- SrcExprs.push_back(PseudoSrcExpr);
- DstExprs.push_back(PseudoDstExpr);
- AssignmentOps.push_back(AssignmentOp.get());
- }
- if (Vars.empty())
- return nullptr;
- return OMPCopyinClause::Create(Context, StartLoc, LParenLoc, EndLoc, Vars,
- SrcExprs, DstExprs, AssignmentOps);
- }
- OMPClause *Sema::ActOnOpenMPCopyprivateClause(ArrayRef<Expr *> VarList,
- SourceLocation StartLoc,
- SourceLocation LParenLoc,
- SourceLocation EndLoc) {
- SmallVector<Expr *, 8> Vars;
- SmallVector<Expr *, 8> SrcExprs;
- SmallVector<Expr *, 8> DstExprs;
- SmallVector<Expr *, 8> AssignmentOps;
- for (Expr *RefExpr : VarList) {
- assert(RefExpr && "NULL expr in OpenMP linear clause.");
- SourceLocation ELoc;
- SourceRange ERange;
- Expr *SimpleRefExpr = RefExpr;
- auto Res = getPrivateItem(*this, SimpleRefExpr, ELoc, ERange);
- if (Res.second) {
- // It will be analyzed later.
- Vars.push_back(RefExpr);
- SrcExprs.push_back(nullptr);
- DstExprs.push_back(nullptr);
- AssignmentOps.push_back(nullptr);
- }
- ValueDecl *D = Res.first;
- if (!D)
- continue;
- QualType Type = D->getType();
- auto *VD = dyn_cast<VarDecl>(D);
- // OpenMP [2.14.4.2, Restrictions, p.2]
- // A list item that appears in a copyprivate clause may not appear in a
- // private or firstprivate clause on the single construct.
- if (!VD || !DSAStack->isThreadPrivate(VD)) {
- DSAStackTy::DSAVarData DVar =
- DSAStack->getTopDSA(D, /*FromParent=*/false);
- if (DVar.CKind != OMPC_unknown && DVar.CKind != OMPC_copyprivate &&
- DVar.RefExpr) {
- Diag(ELoc, diag::err_omp_wrong_dsa)
- << getOpenMPClauseName(DVar.CKind)
- << getOpenMPClauseName(OMPC_copyprivate);
- reportOriginalDsa(*this, DSAStack, D, DVar);
- continue;
- }
- // OpenMP [2.11.4.2, Restrictions, p.1]
- // All list items that appear in a copyprivate clause must be either
- // threadprivate or private in the enclosing context.
- if (DVar.CKind == OMPC_unknown) {
- DVar = DSAStack->getImplicitDSA(D, false);
- if (DVar.CKind == OMPC_shared) {
- Diag(ELoc, diag::err_omp_required_access)
- << getOpenMPClauseName(OMPC_copyprivate)
- << "threadprivate or private in the enclosing context";
- reportOriginalDsa(*this, DSAStack, D, DVar);
- continue;
- }
- }
- }
- // Variably modified types are not supported.
- if (!Type->isAnyPointerType() && Type->isVariablyModifiedType()) {
- Diag(ELoc, diag::err_omp_variably_modified_type_not_supported)
- << getOpenMPClauseName(OMPC_copyprivate) << Type
- << getOpenMPDirectiveName(DSAStack->getCurrentDirective());
- bool IsDecl =
- !VD ||
- VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly;
- Diag(D->getLocation(),
- IsDecl ? diag::note_previous_decl : diag::note_defined_here)
- << D;
- continue;
- }
- // OpenMP [2.14.4.1, Restrictions, C/C++, p.2]
- // A variable of class type (or array thereof) that appears in a
- // copyin clause requires an accessible, unambiguous copy assignment
- // operator for the class type.
- Type = Context.getBaseElementType(Type.getNonReferenceType())
- .getUnqualifiedType();
- VarDecl *SrcVD =
- buildVarDecl(*this, RefExpr->getBeginLoc(), Type, ".copyprivate.src",
- D->hasAttrs() ? &D->getAttrs() : nullptr);
- DeclRefExpr *PseudoSrcExpr = buildDeclRefExpr(*this, SrcVD, Type, ELoc);
- VarDecl *DstVD =
- buildVarDecl(*this, RefExpr->getBeginLoc(), Type, ".copyprivate.dst",
- D->hasAttrs() ? &D->getAttrs() : nullptr);
- DeclRefExpr *PseudoDstExpr = buildDeclRefExpr(*this, DstVD, Type, ELoc);
- ExprResult AssignmentOp = BuildBinOp(
- DSAStack->getCurScope(), ELoc, BO_Assign, PseudoDstExpr, PseudoSrcExpr);
- if (AssignmentOp.isInvalid())
- continue;
- AssignmentOp = ActOnFinishFullExpr(AssignmentOp.get(), ELoc,
- /*DiscardedValue=*/true);
- if (AssignmentOp.isInvalid())
- continue;
- // No need to mark vars as copyprivate, they are already threadprivate or
- // implicitly private.
- assert(VD || isOpenMPCapturedDecl(D));
- Vars.push_back(
- VD ? RefExpr->IgnoreParens()
- : buildCapture(*this, D, SimpleRefExpr, /*WithInit=*/false));
- SrcExprs.push_back(PseudoSrcExpr);
- DstExprs.push_back(PseudoDstExpr);
- AssignmentOps.push_back(AssignmentOp.get());
- }
- if (Vars.empty())
- return nullptr;
- return OMPCopyprivateClause::Create(Context, StartLoc, LParenLoc, EndLoc,
- Vars, SrcExprs, DstExprs, AssignmentOps);
- }
- OMPClause *Sema::ActOnOpenMPFlushClause(ArrayRef<Expr *> VarList,
- SourceLocation StartLoc,
- SourceLocation LParenLoc,
- SourceLocation EndLoc) {
- if (VarList.empty())
- return nullptr;
- return OMPFlushClause::Create(Context, StartLoc, LParenLoc, EndLoc, VarList);
- }
- OMPClause *
- Sema::ActOnOpenMPDependClause(OpenMPDependClauseKind DepKind,
- SourceLocation DepLoc, SourceLocation ColonLoc,
- ArrayRef<Expr *> VarList, SourceLocation StartLoc,
- SourceLocation LParenLoc, SourceLocation EndLoc) {
- if (DSAStack->getCurrentDirective() == OMPD_ordered &&
- DepKind != OMPC_DEPEND_source && DepKind != OMPC_DEPEND_sink) {
- Diag(DepLoc, diag::err_omp_unexpected_clause_value)
- << "'source' or 'sink'" << getOpenMPClauseName(OMPC_depend);
- return nullptr;
- }
- if (DSAStack->getCurrentDirective() != OMPD_ordered &&
- (DepKind == OMPC_DEPEND_unknown || DepKind == OMPC_DEPEND_source ||
- DepKind == OMPC_DEPEND_sink)) {
- unsigned Except[] = {OMPC_DEPEND_source, OMPC_DEPEND_sink};
- Diag(DepLoc, diag::err_omp_unexpected_clause_value)
- << getListOfPossibleValues(OMPC_depend, /*First=*/0,
- /*Last=*/OMPC_DEPEND_unknown, Except)
- << getOpenMPClauseName(OMPC_depend);
- return nullptr;
- }
- SmallVector<Expr *, 8> Vars;
- DSAStackTy::OperatorOffsetTy OpsOffs;
- llvm::APSInt DepCounter(/*BitWidth=*/32);
- llvm::APSInt TotalDepCount(/*BitWidth=*/32);
- if (DepKind == OMPC_DEPEND_sink || DepKind == OMPC_DEPEND_source) {
- if (const Expr *OrderedCountExpr =
- DSAStack->getParentOrderedRegionParam().first) {
- TotalDepCount = OrderedCountExpr->EvaluateKnownConstInt(Context);
- TotalDepCount.setIsUnsigned(/*Val=*/true);
- }
- }
- for (Expr *RefExpr : VarList) {
- assert(RefExpr && "NULL expr in OpenMP shared clause.");
- if (isa<DependentScopeDeclRefExpr>(RefExpr)) {
- // It will be analyzed later.
- Vars.push_back(RefExpr);
- continue;
- }
- SourceLocation ELoc = RefExpr->getExprLoc();
- Expr *SimpleExpr = RefExpr->IgnoreParenCasts();
- if (DepKind == OMPC_DEPEND_sink) {
- if (DSAStack->getParentOrderedRegionParam().first &&
- DepCounter >= TotalDepCount) {
- Diag(ELoc, diag::err_omp_depend_sink_unexpected_expr);
- continue;
- }
- ++DepCounter;
- // OpenMP [2.13.9, Summary]
- // depend(dependence-type : vec), where dependence-type is:
- // 'sink' and where vec is the iteration vector, which has the form:
- // x1 [+- d1], x2 [+- d2 ], . . . , xn [+- dn]
- // where n is the value specified by the ordered clause in the loop
- // directive, xi denotes the loop iteration variable of the i-th nested
- // loop associated with the loop directive, and di is a constant
- // non-negative integer.
- if (CurContext->isDependentContext()) {
- // It will be analyzed later.
- Vars.push_back(RefExpr);
- continue;
- }
- SimpleExpr = SimpleExpr->IgnoreImplicit();
- OverloadedOperatorKind OOK = OO_None;
- SourceLocation OOLoc;
- Expr *LHS = SimpleExpr;
- Expr *RHS = nullptr;
- if (auto *BO = dyn_cast<BinaryOperator>(SimpleExpr)) {
- OOK = BinaryOperator::getOverloadedOperator(BO->getOpcode());
- OOLoc = BO->getOperatorLoc();
- LHS = BO->getLHS()->IgnoreParenImpCasts();
- RHS = BO->getRHS()->IgnoreParenImpCasts();
- } else if (auto *OCE = dyn_cast<CXXOperatorCallExpr>(SimpleExpr)) {
- OOK = OCE->getOperator();
- OOLoc = OCE->getOperatorLoc();
- LHS = OCE->getArg(/*Arg=*/0)->IgnoreParenImpCasts();
- RHS = OCE->getArg(/*Arg=*/1)->IgnoreParenImpCasts();
- } else if (auto *MCE = dyn_cast<CXXMemberCallExpr>(SimpleExpr)) {
- OOK = MCE->getMethodDecl()
- ->getNameInfo()
- .getName()
- .getCXXOverloadedOperator();
- OOLoc = MCE->getCallee()->getExprLoc();
- LHS = MCE->getImplicitObjectArgument()->IgnoreParenImpCasts();
- RHS = MCE->getArg(/*Arg=*/0)->IgnoreParenImpCasts();
- }
- SourceLocation ELoc;
- SourceRange ERange;
- auto Res = getPrivateItem(*this, LHS, ELoc, ERange);
- if (Res.second) {
- // It will be analyzed later.
- Vars.push_back(RefExpr);
- }
- ValueDecl *D = Res.first;
- if (!D)
- continue;
- if (OOK != OO_Plus && OOK != OO_Minus && (RHS || OOK != OO_None)) {
- Diag(OOLoc, diag::err_omp_depend_sink_expected_plus_minus);
- continue;
- }
- if (RHS) {
- ExprResult RHSRes = VerifyPositiveIntegerConstantInClause(
- RHS, OMPC_depend, /*StrictlyPositive=*/false);
- if (RHSRes.isInvalid())
- continue;
- }
- if (!CurContext->isDependentContext() &&
- DSAStack->getParentOrderedRegionParam().first &&
- DepCounter != DSAStack->isParentLoopControlVariable(D).first) {
- const ValueDecl *VD =
- DSAStack->getParentLoopControlVariable(DepCounter.getZExtValue());
- if (VD)
- Diag(ELoc, diag::err_omp_depend_sink_expected_loop_iteration)
- << 1 << VD;
- else
- Diag(ELoc, diag::err_omp_depend_sink_expected_loop_iteration) << 0;
- continue;
- }
- OpsOffs.emplace_back(RHS, OOK);
- } else {
- auto *ASE = dyn_cast<ArraySubscriptExpr>(SimpleExpr);
- if (!RefExpr->IgnoreParenImpCasts()->isLValue() ||
- (ASE &&
- !ASE->getBase()->getType().getNonReferenceType()->isPointerType() &&
- !ASE->getBase()->getType().getNonReferenceType()->isArrayType())) {
- Diag(ELoc, diag::err_omp_expected_addressable_lvalue_or_array_item)
- << RefExpr->getSourceRange();
- continue;
- }
- bool Suppress = getDiagnostics().getSuppressAllDiagnostics();
- getDiagnostics().setSuppressAllDiagnostics(/*Val=*/true);
- ExprResult Res =
- CreateBuiltinUnaryOp(ELoc, UO_AddrOf, RefExpr->IgnoreParenImpCasts());
- getDiagnostics().setSuppressAllDiagnostics(Suppress);
- if (!Res.isUsable() && !isa<OMPArraySectionExpr>(SimpleExpr)) {
- Diag(ELoc, diag::err_omp_expected_addressable_lvalue_or_array_item)
- << RefExpr->getSourceRange();
- continue;
- }
- }
- Vars.push_back(RefExpr->IgnoreParenImpCasts());
- }
- if (!CurContext->isDependentContext() && DepKind == OMPC_DEPEND_sink &&
- TotalDepCount > VarList.size() &&
- DSAStack->getParentOrderedRegionParam().first &&
- DSAStack->getParentLoopControlVariable(VarList.size() + 1)) {
- Diag(EndLoc, diag::err_omp_depend_sink_expected_loop_iteration)
- << 1 << DSAStack->getParentLoopControlVariable(VarList.size() + 1);
- }
- if (DepKind != OMPC_DEPEND_source && DepKind != OMPC_DEPEND_sink &&
- Vars.empty())
- return nullptr;
- auto *C = OMPDependClause::Create(Context, StartLoc, LParenLoc, EndLoc,
- DepKind, DepLoc, ColonLoc, Vars,
- TotalDepCount.getZExtValue());
- if ((DepKind == OMPC_DEPEND_sink || DepKind == OMPC_DEPEND_source) &&
- DSAStack->isParentOrderedRegion())
- DSAStack->addDoacrossDependClause(C, OpsOffs);
- return C;
- }
- OMPClause *Sema::ActOnOpenMPDeviceClause(Expr *Device, SourceLocation StartLoc,
- SourceLocation LParenLoc,
- SourceLocation EndLoc) {
- Expr *ValExpr = Device;
- Stmt *HelperValStmt = nullptr;
- // OpenMP [2.9.1, Restrictions]
- // The device expression must evaluate to a non-negative integer value.
- if (!isNonNegativeIntegerValue(ValExpr, *this, OMPC_device,
- /*StrictlyPositive=*/false))
- return nullptr;
- OpenMPDirectiveKind DKind = DSAStack->getCurrentDirective();
- OpenMPDirectiveKind CaptureRegion =
- getOpenMPCaptureRegionForClause(DKind, OMPC_device);
- if (CaptureRegion != OMPD_unknown && !CurContext->isDependentContext()) {
- ValExpr = MakeFullExpr(ValExpr).get();
- llvm::MapVector<const Expr *, DeclRefExpr *> Captures;
- ValExpr = tryBuildCapture(*this, ValExpr, Captures).get();
- HelperValStmt = buildPreInits(Context, Captures);
- }
- return new (Context) OMPDeviceClause(ValExpr, HelperValStmt, CaptureRegion,
- StartLoc, LParenLoc, EndLoc);
- }
- static bool checkTypeMappable(SourceLocation SL, SourceRange SR, Sema &SemaRef,
- DSAStackTy *Stack, QualType QTy,
- bool FullCheck = true) {
- NamedDecl *ND;
- if (QTy->isIncompleteType(&ND)) {
- SemaRef.Diag(SL, diag::err_incomplete_type) << QTy << SR;
- return false;
- }
- if (FullCheck && !SemaRef.CurContext->isDependentContext() &&
- !QTy.isTrivialType(SemaRef.Context))
- SemaRef.Diag(SL, diag::warn_omp_non_trivial_type_mapped) << QTy << SR;
- return true;
- }
- /// Return true if it can be proven that the provided array expression
- /// (array section or array subscript) does NOT specify the whole size of the
- /// array whose base type is \a BaseQTy.
- static bool checkArrayExpressionDoesNotReferToWholeSize(Sema &SemaRef,
- const Expr *E,
- QualType BaseQTy) {
- const auto *OASE = dyn_cast<OMPArraySectionExpr>(E);
- // If this is an array subscript, it refers to the whole size if the size of
- // the dimension is constant and equals 1. Also, an array section assumes the
- // format of an array subscript if no colon is used.
- if (isa<ArraySubscriptExpr>(E) || (OASE && OASE->getColonLoc().isInvalid())) {
- if (const auto *ATy = dyn_cast<ConstantArrayType>(BaseQTy.getTypePtr()))
- return ATy->getSize().getSExtValue() != 1;
- // Size can't be evaluated statically.
- return false;
- }
- assert(OASE && "Expecting array section if not an array subscript.");
- const Expr *LowerBound = OASE->getLowerBound();
- const Expr *Length = OASE->getLength();
- // If there is a lower bound that does not evaluates to zero, we are not
- // covering the whole dimension.
- if (LowerBound) {
- Expr::EvalResult Result;
- if (!LowerBound->EvaluateAsInt(Result, SemaRef.getASTContext()))
- return false; // Can't get the integer value as a constant.
- llvm::APSInt ConstLowerBound = Result.Val.getInt();
- if (ConstLowerBound.getSExtValue())
- return true;
- }
- // If we don't have a length we covering the whole dimension.
- if (!Length)
- return false;
- // If the base is a pointer, we don't have a way to get the size of the
- // pointee.
- if (BaseQTy->isPointerType())
- return false;
- // We can only check if the length is the same as the size of the dimension
- // if we have a constant array.
- const auto *CATy = dyn_cast<ConstantArrayType>(BaseQTy.getTypePtr());
- if (!CATy)
- return false;
- Expr::EvalResult Result;
- if (!Length->EvaluateAsInt(Result, SemaRef.getASTContext()))
- return false; // Can't get the integer value as a constant.
- llvm::APSInt ConstLength = Result.Val.getInt();
- return CATy->getSize().getSExtValue() != ConstLength.getSExtValue();
- }
- // Return true if it can be proven that the provided array expression (array
- // section or array subscript) does NOT specify a single element of the array
- // whose base type is \a BaseQTy.
- static bool checkArrayExpressionDoesNotReferToUnitySize(Sema &SemaRef,
- const Expr *E,
- QualType BaseQTy) {
- const auto *OASE = dyn_cast<OMPArraySectionExpr>(E);
- // An array subscript always refer to a single element. Also, an array section
- // assumes the format of an array subscript if no colon is used.
- if (isa<ArraySubscriptExpr>(E) || (OASE && OASE->getColonLoc().isInvalid()))
- return false;
- assert(OASE && "Expecting array section if not an array subscript.");
- const Expr *Length = OASE->getLength();
- // If we don't have a length we have to check if the array has unitary size
- // for this dimension. Also, we should always expect a length if the base type
- // is pointer.
- if (!Length) {
- if (const auto *ATy = dyn_cast<ConstantArrayType>(BaseQTy.getTypePtr()))
- return ATy->getSize().getSExtValue() != 1;
- // We cannot assume anything.
- return false;
- }
- // Check if the length evaluates to 1.
- Expr::EvalResult Result;
- if (!Length->EvaluateAsInt(Result, SemaRef.getASTContext()))
- return false; // Can't get the integer value as a constant.
- llvm::APSInt ConstLength = Result.Val.getInt();
- return ConstLength.getSExtValue() != 1;
- }
- // Return the expression of the base of the mappable expression or null if it
- // cannot be determined and do all the necessary checks to see if the expression
- // is valid as a standalone mappable expression. In the process, record all the
- // components of the expression.
- static const Expr *checkMapClauseExpressionBase(
- Sema &SemaRef, Expr *E,
- OMPClauseMappableExprCommon::MappableExprComponentList &CurComponents,
- OpenMPClauseKind CKind, bool NoDiagnose) {
- SourceLocation ELoc = E->getExprLoc();
- SourceRange ERange = E->getSourceRange();
- // The base of elements of list in a map clause have to be either:
- // - a reference to variable or field.
- // - a member expression.
- // - an array expression.
- //
- // E.g. if we have the expression 'r.S.Arr[:12]', we want to retrieve the
- // reference to 'r'.
- //
- // If we have:
- //
- // struct SS {
- // Bla S;
- // foo() {
- // #pragma omp target map (S.Arr[:12]);
- // }
- // }
- //
- // We want to retrieve the member expression 'this->S';
- const Expr *RelevantExpr = nullptr;
- // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, p.2]
- // If a list item is an array section, it must specify contiguous storage.
- //
- // For this restriction it is sufficient that we make sure only references
- // to variables or fields and array expressions, and that no array sections
- // exist except in the rightmost expression (unless they cover the whole
- // dimension of the array). E.g. these would be invalid:
- //
- // r.ArrS[3:5].Arr[6:7]
- //
- // r.ArrS[3:5].x
- //
- // but these would be valid:
- // r.ArrS[3].Arr[6:7]
- //
- // r.ArrS[3].x
- bool AllowUnitySizeArraySection = true;
- bool AllowWholeSizeArraySection = true;
- while (!RelevantExpr) {
- E = E->IgnoreParenImpCasts();
- if (auto *CurE = dyn_cast<DeclRefExpr>(E)) {
- if (!isa<VarDecl>(CurE->getDecl()))
- return nullptr;
- RelevantExpr = CurE;
- // If we got a reference to a declaration, we should not expect any array
- // section before that.
- AllowUnitySizeArraySection = false;
- AllowWholeSizeArraySection = false;
- // Record the component.
- CurComponents.emplace_back(CurE, CurE->getDecl());
- } else if (auto *CurE = dyn_cast<MemberExpr>(E)) {
- Expr *BaseE = CurE->getBase()->IgnoreParenImpCasts();
- if (isa<CXXThisExpr>(BaseE))
- // We found a base expression: this->Val.
- RelevantExpr = CurE;
- else
- E = BaseE;
- if (!isa<FieldDecl>(CurE->getMemberDecl())) {
- if (!NoDiagnose) {
- SemaRef.Diag(ELoc, diag::err_omp_expected_access_to_data_field)
- << CurE->getSourceRange();
- return nullptr;
- }
- if (RelevantExpr)
- return nullptr;
- continue;
- }
- auto *FD = cast<FieldDecl>(CurE->getMemberDecl());
- // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, C/C++, p.3]
- // A bit-field cannot appear in a map clause.
- //
- if (FD->isBitField()) {
- if (!NoDiagnose) {
- SemaRef.Diag(ELoc, diag::err_omp_bit_fields_forbidden_in_clause)
- << CurE->getSourceRange() << getOpenMPClauseName(CKind);
- return nullptr;
- }
- if (RelevantExpr)
- return nullptr;
- continue;
- }
- // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, C++, p.1]
- // If the type of a list item is a reference to a type T then the type
- // will be considered to be T for all purposes of this clause.
- QualType CurType = BaseE->getType().getNonReferenceType();
- // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, C/C++, p.2]
- // A list item cannot be a variable that is a member of a structure with
- // a union type.
- //
- if (CurType->isUnionType()) {
- if (!NoDiagnose) {
- SemaRef.Diag(ELoc, diag::err_omp_union_type_not_allowed)
- << CurE->getSourceRange();
- return nullptr;
- }
- continue;
- }
- // If we got a member expression, we should not expect any array section
- // before that:
- //
- // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, p.7]
- // If a list item is an element of a structure, only the rightmost symbol
- // of the variable reference can be an array section.
- //
- AllowUnitySizeArraySection = false;
- AllowWholeSizeArraySection = false;
- // Record the component.
- CurComponents.emplace_back(CurE, FD);
- } else if (auto *CurE = dyn_cast<ArraySubscriptExpr>(E)) {
- E = CurE->getBase()->IgnoreParenImpCasts();
- if (!E->getType()->isAnyPointerType() && !E->getType()->isArrayType()) {
- if (!NoDiagnose) {
- SemaRef.Diag(ELoc, diag::err_omp_expected_base_var_name)
- << 0 << CurE->getSourceRange();
- return nullptr;
- }
- continue;
- }
- // If we got an array subscript that express the whole dimension we
- // can have any array expressions before. If it only expressing part of
- // the dimension, we can only have unitary-size array expressions.
- if (checkArrayExpressionDoesNotReferToWholeSize(SemaRef, CurE,
- E->getType()))
- AllowWholeSizeArraySection = false;
- // Record the component - we don't have any declaration associated.
- CurComponents.emplace_back(CurE, nullptr);
- } else if (auto *CurE = dyn_cast<OMPArraySectionExpr>(E)) {
- assert(!NoDiagnose && "Array sections cannot be implicitly mapped.");
- E = CurE->getBase()->IgnoreParenImpCasts();
- QualType CurType =
- OMPArraySectionExpr::getBaseOriginalType(E).getCanonicalType();
- // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, C++, p.1]
- // If the type of a list item is a reference to a type T then the type
- // will be considered to be T for all purposes of this clause.
- if (CurType->isReferenceType())
- CurType = CurType->getPointeeType();
- bool IsPointer = CurType->isAnyPointerType();
- if (!IsPointer && !CurType->isArrayType()) {
- SemaRef.Diag(ELoc, diag::err_omp_expected_base_var_name)
- << 0 << CurE->getSourceRange();
- return nullptr;
- }
- bool NotWhole =
- checkArrayExpressionDoesNotReferToWholeSize(SemaRef, CurE, CurType);
- bool NotUnity =
- checkArrayExpressionDoesNotReferToUnitySize(SemaRef, CurE, CurType);
- if (AllowWholeSizeArraySection) {
- // Any array section is currently allowed. Allowing a whole size array
- // section implies allowing a unity array section as well.
- //
- // If this array section refers to the whole dimension we can still
- // accept other array sections before this one, except if the base is a
- // pointer. Otherwise, only unitary sections are accepted.
- if (NotWhole || IsPointer)
- AllowWholeSizeArraySection = false;
- } else if (AllowUnitySizeArraySection && NotUnity) {
- // A unity or whole array section is not allowed and that is not
- // compatible with the properties of the current array section.
- SemaRef.Diag(
- ELoc, diag::err_array_section_does_not_specify_contiguous_storage)
- << CurE->getSourceRange();
- return nullptr;
- }
- // Record the component - we don't have any declaration associated.
- CurComponents.emplace_back(CurE, nullptr);
- } else {
- if (!NoDiagnose) {
- // If nothing else worked, this is not a valid map clause expression.
- SemaRef.Diag(
- ELoc, diag::err_omp_expected_named_var_member_or_array_expression)
- << ERange;
- }
- return nullptr;
- }
- }
- return RelevantExpr;
- }
- // Return true if expression E associated with value VD has conflicts with other
- // map information.
- static bool checkMapConflicts(
- Sema &SemaRef, DSAStackTy *DSAS, const ValueDecl *VD, const Expr *E,
- bool CurrentRegionOnly,
- OMPClauseMappableExprCommon::MappableExprComponentListRef CurComponents,
- OpenMPClauseKind CKind) {
- assert(VD && E);
- SourceLocation ELoc = E->getExprLoc();
- SourceRange ERange = E->getSourceRange();
- // In order to easily check the conflicts we need to match each component of
- // the expression under test with the components of the expressions that are
- // already in the stack.
- assert(!CurComponents.empty() && "Map clause expression with no components!");
- assert(CurComponents.back().getAssociatedDeclaration() == VD &&
- "Map clause expression with unexpected base!");
- // Variables to help detecting enclosing problems in data environment nests.
- bool IsEnclosedByDataEnvironmentExpr = false;
- const Expr *EnclosingExpr = nullptr;
- bool FoundError = DSAS->checkMappableExprComponentListsForDecl(
- VD, CurrentRegionOnly,
- [&IsEnclosedByDataEnvironmentExpr, &SemaRef, VD, CurrentRegionOnly, ELoc,
- ERange, CKind, &EnclosingExpr,
- CurComponents](OMPClauseMappableExprCommon::MappableExprComponentListRef
- StackComponents,
- OpenMPClauseKind) {
- assert(!StackComponents.empty() &&
- "Map clause expression with no components!");
- assert(StackComponents.back().getAssociatedDeclaration() == VD &&
- "Map clause expression with unexpected base!");
- (void)VD;
- // The whole expression in the stack.
- const Expr *RE = StackComponents.front().getAssociatedExpression();
- // Expressions must start from the same base. Here we detect at which
- // point both expressions diverge from each other and see if we can
- // detect if the memory referred to both expressions is contiguous and
- // do not overlap.
- auto CI = CurComponents.rbegin();
- auto CE = CurComponents.rend();
- auto SI = StackComponents.rbegin();
- auto SE = StackComponents.rend();
- for (; CI != CE && SI != SE; ++CI, ++SI) {
- // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, p.3]
- // At most one list item can be an array item derived from a given
- // variable in map clauses of the same construct.
- if (CurrentRegionOnly &&
- (isa<ArraySubscriptExpr>(CI->getAssociatedExpression()) ||
- isa<OMPArraySectionExpr>(CI->getAssociatedExpression())) &&
- (isa<ArraySubscriptExpr>(SI->getAssociatedExpression()) ||
- isa<OMPArraySectionExpr>(SI->getAssociatedExpression()))) {
- SemaRef.Diag(CI->getAssociatedExpression()->getExprLoc(),
- diag::err_omp_multiple_array_items_in_map_clause)
- << CI->getAssociatedExpression()->getSourceRange();
- SemaRef.Diag(SI->getAssociatedExpression()->getExprLoc(),
- diag::note_used_here)
- << SI->getAssociatedExpression()->getSourceRange();
- return true;
- }
- // Do both expressions have the same kind?
- if (CI->getAssociatedExpression()->getStmtClass() !=
- SI->getAssociatedExpression()->getStmtClass())
- break;
- // Are we dealing with different variables/fields?
- if (CI->getAssociatedDeclaration() != SI->getAssociatedDeclaration())
- break;
- }
- // Check if the extra components of the expressions in the enclosing
- // data environment are redundant for the current base declaration.
- // If they are, the maps completely overlap, which is legal.
- for (; SI != SE; ++SI) {
- QualType Type;
- if (const auto *ASE =
- dyn_cast<ArraySubscriptExpr>(SI->getAssociatedExpression())) {
- Type = ASE->getBase()->IgnoreParenImpCasts()->getType();
- } else if (const auto *OASE = dyn_cast<OMPArraySectionExpr>(
- SI->getAssociatedExpression())) {
- const Expr *E = OASE->getBase()->IgnoreParenImpCasts();
- Type =
- OMPArraySectionExpr::getBaseOriginalType(E).getCanonicalType();
- }
- if (Type.isNull() || Type->isAnyPointerType() ||
- checkArrayExpressionDoesNotReferToWholeSize(
- SemaRef, SI->getAssociatedExpression(), Type))
- break;
- }
- // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, p.4]
- // List items of map clauses in the same construct must not share
- // original storage.
- //
- // If the expressions are exactly the same or one is a subset of the
- // other, it means they are sharing storage.
- if (CI == CE && SI == SE) {
- if (CurrentRegionOnly) {
- if (CKind == OMPC_map) {
- SemaRef.Diag(ELoc, diag::err_omp_map_shared_storage) << ERange;
- } else {
- assert(CKind == OMPC_to || CKind == OMPC_from);
- SemaRef.Diag(ELoc, diag::err_omp_once_referenced_in_target_update)
- << ERange;
- }
- SemaRef.Diag(RE->getExprLoc(), diag::note_used_here)
- << RE->getSourceRange();
- return true;
- }
- // If we find the same expression in the enclosing data environment,
- // that is legal.
- IsEnclosedByDataEnvironmentExpr = true;
- return false;
- }
- QualType DerivedType =
- std::prev(CI)->getAssociatedDeclaration()->getType();
- SourceLocation DerivedLoc =
- std::prev(CI)->getAssociatedExpression()->getExprLoc();
- // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, C++, p.1]
- // If the type of a list item is a reference to a type T then the type
- // will be considered to be T for all purposes of this clause.
- DerivedType = DerivedType.getNonReferenceType();
- // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, C/C++, p.1]
- // A variable for which the type is pointer and an array section
- // derived from that variable must not appear as list items of map
- // clauses of the same construct.
- //
- // Also, cover one of the cases in:
- // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, p.5]
- // If any part of the original storage of a list item has corresponding
- // storage in the device data environment, all of the original storage
- // must have corresponding storage in the device data environment.
- //
- if (DerivedType->isAnyPointerType()) {
- if (CI == CE || SI == SE) {
- SemaRef.Diag(
- DerivedLoc,
- diag::err_omp_pointer_mapped_along_with_derived_section)
- << DerivedLoc;
- SemaRef.Diag(RE->getExprLoc(), diag::note_used_here)
- << RE->getSourceRange();
- return true;
- }
- if (CI->getAssociatedExpression()->getStmtClass() !=
- SI->getAssociatedExpression()->getStmtClass() ||
- CI->getAssociatedDeclaration()->getCanonicalDecl() ==
- SI->getAssociatedDeclaration()->getCanonicalDecl()) {
- assert(CI != CE && SI != SE);
- SemaRef.Diag(DerivedLoc, diag::err_omp_same_pointer_dereferenced)
- << DerivedLoc;
- SemaRef.Diag(RE->getExprLoc(), diag::note_used_here)
- << RE->getSourceRange();
- return true;
- }
- }
- // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, p.4]
- // List items of map clauses in the same construct must not share
- // original storage.
- //
- // An expression is a subset of the other.
- if (CurrentRegionOnly && (CI == CE || SI == SE)) {
- if (CKind == OMPC_map) {
- if (CI != CE || SI != SE) {
- // Allow constructs like this: map(s, s.ptr[0:1]), where s.ptr is
- // a pointer.
- auto Begin =
- CI != CE ? CurComponents.begin() : StackComponents.begin();
- auto End = CI != CE ? CurComponents.end() : StackComponents.end();
- auto It = Begin;
- while (It != End && !It->getAssociatedDeclaration())
- std::advance(It, 1);
- assert(It != End &&
- "Expected at least one component with the declaration.");
- if (It != Begin && It->getAssociatedDeclaration()
- ->getType()
- .getCanonicalType()
- ->isAnyPointerType()) {
- IsEnclosedByDataEnvironmentExpr = false;
- EnclosingExpr = nullptr;
- return false;
- }
- }
- SemaRef.Diag(ELoc, diag::err_omp_map_shared_storage) << ERange;
- } else {
- assert(CKind == OMPC_to || CKind == OMPC_from);
- SemaRef.Diag(ELoc, diag::err_omp_once_referenced_in_target_update)
- << ERange;
- }
- SemaRef.Diag(RE->getExprLoc(), diag::note_used_here)
- << RE->getSourceRange();
- return true;
- }
- // The current expression uses the same base as other expression in the
- // data environment but does not contain it completely.
- if (!CurrentRegionOnly && SI != SE)
- EnclosingExpr = RE;
- // The current expression is a subset of the expression in the data
- // environment.
- IsEnclosedByDataEnvironmentExpr |=
- (!CurrentRegionOnly && CI != CE && SI == SE);
- return false;
- });
- if (CurrentRegionOnly)
- return FoundError;
- // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, p.5]
- // If any part of the original storage of a list item has corresponding
- // storage in the device data environment, all of the original storage must
- // have corresponding storage in the device data environment.
- // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, p.6]
- // If a list item is an element of a structure, and a different element of
- // the structure has a corresponding list item in the device data environment
- // prior to a task encountering the construct associated with the map clause,
- // then the list item must also have a corresponding list item in the device
- // data environment prior to the task encountering the construct.
- //
- if (EnclosingExpr && !IsEnclosedByDataEnvironmentExpr) {
- SemaRef.Diag(ELoc,
- diag::err_omp_original_storage_is_shared_and_does_not_contain)
- << ERange;
- SemaRef.Diag(EnclosingExpr->getExprLoc(), diag::note_used_here)
- << EnclosingExpr->getSourceRange();
- return true;
- }
- return FoundError;
- }
- namespace {
- // Utility struct that gathers all the related lists associated with a mappable
- // expression.
- struct MappableVarListInfo {
- // The list of expressions.
- ArrayRef<Expr *> VarList;
- // The list of processed expressions.
- SmallVector<Expr *, 16> ProcessedVarList;
- // The mappble components for each expression.
- OMPClauseMappableExprCommon::MappableExprComponentLists VarComponents;
- // The base declaration of the variable.
- SmallVector<ValueDecl *, 16> VarBaseDeclarations;
- MappableVarListInfo(ArrayRef<Expr *> VarList) : VarList(VarList) {
- // We have a list of components and base declarations for each entry in the
- // variable list.
- VarComponents.reserve(VarList.size());
- VarBaseDeclarations.reserve(VarList.size());
- }
- };
- }
- // Check the validity of the provided variable list for the provided clause kind
- // \a CKind. In the check process the valid expressions, and mappable expression
- // components and variables are extracted and used to fill \a Vars,
- // \a ClauseComponents, and \a ClauseBaseDeclarations. \a MapType and
- // \a IsMapTypeImplicit are expected to be valid if the clause kind is 'map'.
- static void
- checkMappableExpressionList(Sema &SemaRef, DSAStackTy *DSAS,
- OpenMPClauseKind CKind, MappableVarListInfo &MVLI,
- SourceLocation StartLoc,
- OpenMPMapClauseKind MapType = OMPC_MAP_unknown,
- bool IsMapTypeImplicit = false) {
- // We only expect mappable expressions in 'to', 'from', and 'map' clauses.
- assert((CKind == OMPC_map || CKind == OMPC_to || CKind == OMPC_from) &&
- "Unexpected clause kind with mappable expressions!");
- // Keep track of the mappable components and base declarations in this clause.
- // Each entry in the list is going to have a list of components associated. We
- // record each set of the components so that we can build the clause later on.
- // In the end we should have the same amount of declarations and component
- // lists.
- for (Expr *RE : MVLI.VarList) {
- assert(RE && "Null expr in omp to/from/map clause");
- SourceLocation ELoc = RE->getExprLoc();
- const Expr *VE = RE->IgnoreParenLValueCasts();
- if (VE->isValueDependent() || VE->isTypeDependent() ||
- VE->isInstantiationDependent() ||
- VE->containsUnexpandedParameterPack()) {
- // We can only analyze this information once the missing information is
- // resolved.
- MVLI.ProcessedVarList.push_back(RE);
- continue;
- }
- Expr *SimpleExpr = RE->IgnoreParenCasts();
- if (!RE->IgnoreParenImpCasts()->isLValue()) {
- SemaRef.Diag(ELoc,
- diag::err_omp_expected_named_var_member_or_array_expression)
- << RE->getSourceRange();
- continue;
- }
- OMPClauseMappableExprCommon::MappableExprComponentList CurComponents;
- ValueDecl *CurDeclaration = nullptr;
- // Obtain the array or member expression bases if required. Also, fill the
- // components array with all the components identified in the process.
- const Expr *BE = checkMapClauseExpressionBase(
- SemaRef, SimpleExpr, CurComponents, CKind, /*NoDiagnose=*/false);
- if (!BE)
- continue;
- assert(!CurComponents.empty() &&
- "Invalid mappable expression information.");
- // For the following checks, we rely on the base declaration which is
- // expected to be associated with the last component. The declaration is
- // expected to be a variable or a field (if 'this' is being mapped).
- CurDeclaration = CurComponents.back().getAssociatedDeclaration();
- assert(CurDeclaration && "Null decl on map clause.");
- assert(
- CurDeclaration->isCanonicalDecl() &&
- "Expecting components to have associated only canonical declarations.");
- auto *VD = dyn_cast<VarDecl>(CurDeclaration);
- const auto *FD = dyn_cast<FieldDecl>(CurDeclaration);
- assert((VD || FD) && "Only variables or fields are expected here!");
- (void)FD;
- // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, p.10]
- // threadprivate variables cannot appear in a map clause.
- // OpenMP 4.5 [2.10.5, target update Construct]
- // threadprivate variables cannot appear in a from clause.
- if (VD && DSAS->isThreadPrivate(VD)) {
- DSAStackTy::DSAVarData DVar = DSAS->getTopDSA(VD, /*FromParent=*/false);
- SemaRef.Diag(ELoc, diag::err_omp_threadprivate_in_clause)
- << getOpenMPClauseName(CKind);
- reportOriginalDsa(SemaRef, DSAS, VD, DVar);
- continue;
- }
- // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, p.9]
- // A list item cannot appear in both a map clause and a data-sharing
- // attribute clause on the same construct.
- // Check conflicts with other map clause expressions. We check the conflicts
- // with the current construct separately from the enclosing data
- // environment, because the restrictions are different. We only have to
- // check conflicts across regions for the map clauses.
- if (checkMapConflicts(SemaRef, DSAS, CurDeclaration, SimpleExpr,
- /*CurrentRegionOnly=*/true, CurComponents, CKind))
- break;
- if (CKind == OMPC_map &&
- checkMapConflicts(SemaRef, DSAS, CurDeclaration, SimpleExpr,
- /*CurrentRegionOnly=*/false, CurComponents, CKind))
- break;
- // OpenMP 4.5 [2.10.5, target update Construct]
- // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, C++, p.1]
- // If the type of a list item is a reference to a type T then the type will
- // be considered to be T for all purposes of this clause.
- auto I = llvm::find_if(
- CurComponents,
- [](const OMPClauseMappableExprCommon::MappableComponent &MC) {
- return MC.getAssociatedDeclaration();
- });
- assert(I != CurComponents.end() && "Null decl on map clause.");
- QualType Type =
- I->getAssociatedDeclaration()->getType().getNonReferenceType();
- // OpenMP 4.5 [2.10.5, target update Construct, Restrictions, p.4]
- // A list item in a to or from clause must have a mappable type.
- // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, p.9]
- // A list item must have a mappable type.
- if (!checkTypeMappable(VE->getExprLoc(), VE->getSourceRange(), SemaRef,
- DSAS, Type))
- continue;
- if (CKind == OMPC_map) {
- // target enter data
- // OpenMP [2.10.2, Restrictions, p. 99]
- // A map-type must be specified in all map clauses and must be either
- // to or alloc.
- OpenMPDirectiveKind DKind = DSAS->getCurrentDirective();
- if (DKind == OMPD_target_enter_data &&
- !(MapType == OMPC_MAP_to || MapType == OMPC_MAP_alloc)) {
- SemaRef.Diag(StartLoc, diag::err_omp_invalid_map_type_for_directive)
- << (IsMapTypeImplicit ? 1 : 0)
- << getOpenMPSimpleClauseTypeName(OMPC_map, MapType)
- << getOpenMPDirectiveName(DKind);
- continue;
- }
- // target exit_data
- // OpenMP [2.10.3, Restrictions, p. 102]
- // A map-type must be specified in all map clauses and must be either
- // from, release, or delete.
- if (DKind == OMPD_target_exit_data &&
- !(MapType == OMPC_MAP_from || MapType == OMPC_MAP_release ||
- MapType == OMPC_MAP_delete)) {
- SemaRef.Diag(StartLoc, diag::err_omp_invalid_map_type_for_directive)
- << (IsMapTypeImplicit ? 1 : 0)
- << getOpenMPSimpleClauseTypeName(OMPC_map, MapType)
- << getOpenMPDirectiveName(DKind);
- continue;
- }
- // OpenMP 4.5 [2.15.5.1, Restrictions, p.3]
- // A list item cannot appear in both a map clause and a data-sharing
- // attribute clause on the same construct
- if (VD && isOpenMPTargetExecutionDirective(DKind)) {
- DSAStackTy::DSAVarData DVar = DSAS->getTopDSA(VD, /*FromParent=*/false);
- if (isOpenMPPrivate(DVar.CKind)) {
- SemaRef.Diag(ELoc, diag::err_omp_variable_in_given_clause_and_dsa)
- << getOpenMPClauseName(DVar.CKind)
- << getOpenMPClauseName(OMPC_map)
- << getOpenMPDirectiveName(DSAS->getCurrentDirective());
- reportOriginalDsa(SemaRef, DSAS, CurDeclaration, DVar);
- continue;
- }
- }
- }
- // Save the current expression.
- MVLI.ProcessedVarList.push_back(RE);
- // Store the components in the stack so that they can be used to check
- // against other clauses later on.
- DSAS->addMappableExpressionComponents(CurDeclaration, CurComponents,
- /*WhereFoundClauseKind=*/OMPC_map);
- // Save the components and declaration to create the clause. For purposes of
- // the clause creation, any component list that has has base 'this' uses
- // null as base declaration.
- MVLI.VarComponents.resize(MVLI.VarComponents.size() + 1);
- MVLI.VarComponents.back().append(CurComponents.begin(),
- CurComponents.end());
- MVLI.VarBaseDeclarations.push_back(isa<MemberExpr>(BE) ? nullptr
- : CurDeclaration);
- }
- }
- OMPClause *
- Sema::ActOnOpenMPMapClause(ArrayRef<OpenMPMapModifierKind> MapTypeModifiers,
- ArrayRef<SourceLocation> MapTypeModifiersLoc,
- OpenMPMapClauseKind MapType, bool IsMapTypeImplicit,
- SourceLocation MapLoc, SourceLocation ColonLoc,
- ArrayRef<Expr *> VarList, SourceLocation StartLoc,
- SourceLocation LParenLoc, SourceLocation EndLoc) {
- MappableVarListInfo MVLI(VarList);
- checkMappableExpressionList(*this, DSAStack, OMPC_map, MVLI, StartLoc,
- MapType, IsMapTypeImplicit);
- OpenMPMapModifierKind Modifiers[] = { OMPC_MAP_MODIFIER_unknown,
- OMPC_MAP_MODIFIER_unknown };
- SourceLocation ModifiersLoc[OMPMapClause::NumberOfModifiers];
- // Process map-type-modifiers, flag errors for duplicate modifiers.
- unsigned Count = 0;
- for (unsigned I = 0, E = MapTypeModifiers.size(); I < E; ++I) {
- if (MapTypeModifiers[I] != OMPC_MAP_MODIFIER_unknown &&
- llvm::find(Modifiers, MapTypeModifiers[I]) != std::end(Modifiers)) {
- Diag(MapTypeModifiersLoc[I], diag::err_omp_duplicate_map_type_modifier);
- continue;
- }
- assert(Count < OMPMapClause::NumberOfModifiers &&
- "Modifiers exceed the allowed number of map type modifiers");
- Modifiers[Count] = MapTypeModifiers[I];
- ModifiersLoc[Count] = MapTypeModifiersLoc[I];
- ++Count;
- }
- // We need to produce a map clause even if we don't have variables so that
- // other diagnostics related with non-existing map clauses are accurate.
- return OMPMapClause::Create(Context, StartLoc, LParenLoc, EndLoc,
- MVLI.ProcessedVarList, MVLI.VarBaseDeclarations,
- MVLI.VarComponents, Modifiers, ModifiersLoc,
- MapType, IsMapTypeImplicit, MapLoc);
- }
- QualType Sema::ActOnOpenMPDeclareReductionType(SourceLocation TyLoc,
- TypeResult ParsedType) {
- assert(ParsedType.isUsable());
- QualType ReductionType = GetTypeFromParser(ParsedType.get());
- if (ReductionType.isNull())
- return QualType();
- // [OpenMP 4.0], 2.15 declare reduction Directive, Restrictions, C\C++
- // A type name in a declare reduction directive cannot be a function type, an
- // array type, a reference type, or a type qualified with const, volatile or
- // restrict.
- if (ReductionType.hasQualifiers()) {
- Diag(TyLoc, diag::err_omp_reduction_wrong_type) << 0;
- return QualType();
- }
- if (ReductionType->isFunctionType()) {
- Diag(TyLoc, diag::err_omp_reduction_wrong_type) << 1;
- return QualType();
- }
- if (ReductionType->isReferenceType()) {
- Diag(TyLoc, diag::err_omp_reduction_wrong_type) << 2;
- return QualType();
- }
- if (ReductionType->isArrayType()) {
- Diag(TyLoc, diag::err_omp_reduction_wrong_type) << 3;
- return QualType();
- }
- return ReductionType;
- }
- Sema::DeclGroupPtrTy Sema::ActOnOpenMPDeclareReductionDirectiveStart(
- Scope *S, DeclContext *DC, DeclarationName Name,
- ArrayRef<std::pair<QualType, SourceLocation>> ReductionTypes,
- AccessSpecifier AS, Decl *PrevDeclInScope) {
- SmallVector<Decl *, 8> Decls;
- Decls.reserve(ReductionTypes.size());
- LookupResult Lookup(*this, Name, SourceLocation(), LookupOMPReductionName,
- forRedeclarationInCurContext());
- // [OpenMP 4.0], 2.15 declare reduction Directive, Restrictions
- // A reduction-identifier may not be re-declared in the current scope for the
- // same type or for a type that is compatible according to the base language
- // rules.
- llvm::DenseMap<QualType, SourceLocation> PreviousRedeclTypes;
- OMPDeclareReductionDecl *PrevDRD = nullptr;
- bool InCompoundScope = true;
- if (S != nullptr) {
- // Find previous declaration with the same name not referenced in other
- // declarations.
- FunctionScopeInfo *ParentFn = getEnclosingFunction();
- InCompoundScope =
- (ParentFn != nullptr) && !ParentFn->CompoundScopes.empty();
- LookupName(Lookup, S);
- FilterLookupForScope(Lookup, DC, S, /*ConsiderLinkage=*/false,
- /*AllowInlineNamespace=*/false);
- llvm::DenseMap<OMPDeclareReductionDecl *, bool> UsedAsPrevious;
- LookupResult::Filter Filter = Lookup.makeFilter();
- while (Filter.hasNext()) {
- auto *PrevDecl = cast<OMPDeclareReductionDecl>(Filter.next());
- if (InCompoundScope) {
- auto I = UsedAsPrevious.find(PrevDecl);
- if (I == UsedAsPrevious.end())
- UsedAsPrevious[PrevDecl] = false;
- if (OMPDeclareReductionDecl *D = PrevDecl->getPrevDeclInScope())
- UsedAsPrevious[D] = true;
- }
- PreviousRedeclTypes[PrevDecl->getType().getCanonicalType()] =
- PrevDecl->getLocation();
- }
- Filter.done();
- if (InCompoundScope) {
- for (const auto &PrevData : UsedAsPrevious) {
- if (!PrevData.second) {
- PrevDRD = PrevData.first;
- break;
- }
- }
- }
- } else if (PrevDeclInScope != nullptr) {
- auto *PrevDRDInScope = PrevDRD =
- cast<OMPDeclareReductionDecl>(PrevDeclInScope);
- do {
- PreviousRedeclTypes[PrevDRDInScope->getType().getCanonicalType()] =
- PrevDRDInScope->getLocation();
- PrevDRDInScope = PrevDRDInScope->getPrevDeclInScope();
- } while (PrevDRDInScope != nullptr);
- }
- for (const auto &TyData : ReductionTypes) {
- const auto I = PreviousRedeclTypes.find(TyData.first.getCanonicalType());
- bool Invalid = false;
- if (I != PreviousRedeclTypes.end()) {
- Diag(TyData.second, diag::err_omp_declare_reduction_redefinition)
- << TyData.first;
- Diag(I->second, diag::note_previous_definition);
- Invalid = true;
- }
- PreviousRedeclTypes[TyData.first.getCanonicalType()] = TyData.second;
- auto *DRD = OMPDeclareReductionDecl::Create(Context, DC, TyData.second,
- Name, TyData.first, PrevDRD);
- DC->addDecl(DRD);
- DRD->setAccess(AS);
- Decls.push_back(DRD);
- if (Invalid)
- DRD->setInvalidDecl();
- else
- PrevDRD = DRD;
- }
- return DeclGroupPtrTy::make(
- DeclGroupRef::Create(Context, Decls.begin(), Decls.size()));
- }
- void Sema::ActOnOpenMPDeclareReductionCombinerStart(Scope *S, Decl *D) {
- auto *DRD = cast<OMPDeclareReductionDecl>(D);
- // Enter new function scope.
- PushFunctionScope();
- setFunctionHasBranchProtectedScope();
- getCurFunction()->setHasOMPDeclareReductionCombiner();
- if (S != nullptr)
- PushDeclContext(S, DRD);
- else
- CurContext = DRD;
- PushExpressionEvaluationContext(
- ExpressionEvaluationContext::PotentiallyEvaluated);
- QualType ReductionType = DRD->getType();
- // Create 'T* omp_parm;T omp_in;'. All references to 'omp_in' will
- // be replaced by '*omp_parm' during codegen. This required because 'omp_in'
- // uses semantics of argument handles by value, but it should be passed by
- // reference. C lang does not support references, so pass all parameters as
- // pointers.
- // Create 'T omp_in;' variable.
- VarDecl *OmpInParm =
- buildVarDecl(*this, D->getLocation(), ReductionType, "omp_in");
- // Create 'T* omp_parm;T omp_out;'. All references to 'omp_out' will
- // be replaced by '*omp_parm' during codegen. This required because 'omp_out'
- // uses semantics of argument handles by value, but it should be passed by
- // reference. C lang does not support references, so pass all parameters as
- // pointers.
- // Create 'T omp_out;' variable.
- VarDecl *OmpOutParm =
- buildVarDecl(*this, D->getLocation(), ReductionType, "omp_out");
- if (S != nullptr) {
- PushOnScopeChains(OmpInParm, S);
- PushOnScopeChains(OmpOutParm, S);
- } else {
- DRD->addDecl(OmpInParm);
- DRD->addDecl(OmpOutParm);
- }
- Expr *InE =
- ::buildDeclRefExpr(*this, OmpInParm, ReductionType, D->getLocation());
- Expr *OutE =
- ::buildDeclRefExpr(*this, OmpOutParm, ReductionType, D->getLocation());
- DRD->setCombinerData(InE, OutE);
- }
- void Sema::ActOnOpenMPDeclareReductionCombinerEnd(Decl *D, Expr *Combiner) {
- auto *DRD = cast<OMPDeclareReductionDecl>(D);
- DiscardCleanupsInEvaluationContext();
- PopExpressionEvaluationContext();
- PopDeclContext();
- PopFunctionScopeInfo();
- if (Combiner != nullptr)
- DRD->setCombiner(Combiner);
- else
- DRD->setInvalidDecl();
- }
- VarDecl *Sema::ActOnOpenMPDeclareReductionInitializerStart(Scope *S, Decl *D) {
- auto *DRD = cast<OMPDeclareReductionDecl>(D);
- // Enter new function scope.
- PushFunctionScope();
- setFunctionHasBranchProtectedScope();
- if (S != nullptr)
- PushDeclContext(S, DRD);
- else
- CurContext = DRD;
- PushExpressionEvaluationContext(
- ExpressionEvaluationContext::PotentiallyEvaluated);
- QualType ReductionType = DRD->getType();
- // Create 'T* omp_parm;T omp_priv;'. All references to 'omp_priv' will
- // be replaced by '*omp_parm' during codegen. This required because 'omp_priv'
- // uses semantics of argument handles by value, but it should be passed by
- // reference. C lang does not support references, so pass all parameters as
- // pointers.
- // Create 'T omp_priv;' variable.
- VarDecl *OmpPrivParm =
- buildVarDecl(*this, D->getLocation(), ReductionType, "omp_priv");
- // Create 'T* omp_parm;T omp_orig;'. All references to 'omp_orig' will
- // be replaced by '*omp_parm' during codegen. This required because 'omp_orig'
- // uses semantics of argument handles by value, but it should be passed by
- // reference. C lang does not support references, so pass all parameters as
- // pointers.
- // Create 'T omp_orig;' variable.
- VarDecl *OmpOrigParm =
- buildVarDecl(*this, D->getLocation(), ReductionType, "omp_orig");
- if (S != nullptr) {
- PushOnScopeChains(OmpPrivParm, S);
- PushOnScopeChains(OmpOrigParm, S);
- } else {
- DRD->addDecl(OmpPrivParm);
- DRD->addDecl(OmpOrigParm);
- }
- Expr *OrigE =
- ::buildDeclRefExpr(*this, OmpOrigParm, ReductionType, D->getLocation());
- Expr *PrivE =
- ::buildDeclRefExpr(*this, OmpPrivParm, ReductionType, D->getLocation());
- DRD->setInitializerData(OrigE, PrivE);
- return OmpPrivParm;
- }
- void Sema::ActOnOpenMPDeclareReductionInitializerEnd(Decl *D, Expr *Initializer,
- VarDecl *OmpPrivParm) {
- auto *DRD = cast<OMPDeclareReductionDecl>(D);
- DiscardCleanupsInEvaluationContext();
- PopExpressionEvaluationContext();
- PopDeclContext();
- PopFunctionScopeInfo();
- if (Initializer != nullptr) {
- DRD->setInitializer(Initializer, OMPDeclareReductionDecl::CallInit);
- } else if (OmpPrivParm->hasInit()) {
- DRD->setInitializer(OmpPrivParm->getInit(),
- OmpPrivParm->isDirectInit()
- ? OMPDeclareReductionDecl::DirectInit
- : OMPDeclareReductionDecl::CopyInit);
- } else {
- DRD->setInvalidDecl();
- }
- }
- Sema::DeclGroupPtrTy Sema::ActOnOpenMPDeclareReductionDirectiveEnd(
- Scope *S, DeclGroupPtrTy DeclReductions, bool IsValid) {
- for (Decl *D : DeclReductions.get()) {
- if (IsValid) {
- if (S)
- PushOnScopeChains(cast<OMPDeclareReductionDecl>(D), S,
- /*AddToContext=*/false);
- } else {
- D->setInvalidDecl();
- }
- }
- return DeclReductions;
- }
- OMPClause *Sema::ActOnOpenMPNumTeamsClause(Expr *NumTeams,
- SourceLocation StartLoc,
- SourceLocation LParenLoc,
- SourceLocation EndLoc) {
- Expr *ValExpr = NumTeams;
- Stmt *HelperValStmt = nullptr;
- // OpenMP [teams Constrcut, Restrictions]
- // The num_teams expression must evaluate to a positive integer value.
- if (!isNonNegativeIntegerValue(ValExpr, *this, OMPC_num_teams,
- /*StrictlyPositive=*/true))
- return nullptr;
- OpenMPDirectiveKind DKind = DSAStack->getCurrentDirective();
- OpenMPDirectiveKind CaptureRegion =
- getOpenMPCaptureRegionForClause(DKind, OMPC_num_teams);
- if (CaptureRegion != OMPD_unknown && !CurContext->isDependentContext()) {
- ValExpr = MakeFullExpr(ValExpr).get();
- llvm::MapVector<const Expr *, DeclRefExpr *> Captures;
- ValExpr = tryBuildCapture(*this, ValExpr, Captures).get();
- HelperValStmt = buildPreInits(Context, Captures);
- }
- return new (Context) OMPNumTeamsClause(ValExpr, HelperValStmt, CaptureRegion,
- StartLoc, LParenLoc, EndLoc);
- }
- OMPClause *Sema::ActOnOpenMPThreadLimitClause(Expr *ThreadLimit,
- SourceLocation StartLoc,
- SourceLocation LParenLoc,
- SourceLocation EndLoc) {
- Expr *ValExpr = ThreadLimit;
- Stmt *HelperValStmt = nullptr;
- // OpenMP [teams Constrcut, Restrictions]
- // The thread_limit expression must evaluate to a positive integer value.
- if (!isNonNegativeIntegerValue(ValExpr, *this, OMPC_thread_limit,
- /*StrictlyPositive=*/true))
- return nullptr;
- OpenMPDirectiveKind DKind = DSAStack->getCurrentDirective();
- OpenMPDirectiveKind CaptureRegion =
- getOpenMPCaptureRegionForClause(DKind, OMPC_thread_limit);
- if (CaptureRegion != OMPD_unknown && !CurContext->isDependentContext()) {
- ValExpr = MakeFullExpr(ValExpr).get();
- llvm::MapVector<const Expr *, DeclRefExpr *> Captures;
- ValExpr = tryBuildCapture(*this, ValExpr, Captures).get();
- HelperValStmt = buildPreInits(Context, Captures);
- }
- return new (Context) OMPThreadLimitClause(
- ValExpr, HelperValStmt, CaptureRegion, StartLoc, LParenLoc, EndLoc);
- }
- OMPClause *Sema::ActOnOpenMPPriorityClause(Expr *Priority,
- SourceLocation StartLoc,
- SourceLocation LParenLoc,
- SourceLocation EndLoc) {
- Expr *ValExpr = Priority;
- // OpenMP [2.9.1, task Constrcut]
- // The priority-value is a non-negative numerical scalar expression.
- if (!isNonNegativeIntegerValue(ValExpr, *this, OMPC_priority,
- /*StrictlyPositive=*/false))
- return nullptr;
- return new (Context) OMPPriorityClause(ValExpr, StartLoc, LParenLoc, EndLoc);
- }
- OMPClause *Sema::ActOnOpenMPGrainsizeClause(Expr *Grainsize,
- SourceLocation StartLoc,
- SourceLocation LParenLoc,
- SourceLocation EndLoc) {
- Expr *ValExpr = Grainsize;
- // OpenMP [2.9.2, taskloop Constrcut]
- // The parameter of the grainsize clause must be a positive integer
- // expression.
- if (!isNonNegativeIntegerValue(ValExpr, *this, OMPC_grainsize,
- /*StrictlyPositive=*/true))
- return nullptr;
- return new (Context) OMPGrainsizeClause(ValExpr, StartLoc, LParenLoc, EndLoc);
- }
- OMPClause *Sema::ActOnOpenMPNumTasksClause(Expr *NumTasks,
- SourceLocation StartLoc,
- SourceLocation LParenLoc,
- SourceLocation EndLoc) {
- Expr *ValExpr = NumTasks;
- // OpenMP [2.9.2, taskloop Constrcut]
- // The parameter of the num_tasks clause must be a positive integer
- // expression.
- if (!isNonNegativeIntegerValue(ValExpr, *this, OMPC_num_tasks,
- /*StrictlyPositive=*/true))
- return nullptr;
- return new (Context) OMPNumTasksClause(ValExpr, StartLoc, LParenLoc, EndLoc);
- }
- OMPClause *Sema::ActOnOpenMPHintClause(Expr *Hint, SourceLocation StartLoc,
- SourceLocation LParenLoc,
- SourceLocation EndLoc) {
- // OpenMP [2.13.2, critical construct, Description]
- // ... where hint-expression is an integer constant expression that evaluates
- // to a valid lock hint.
- ExprResult HintExpr = VerifyPositiveIntegerConstantInClause(Hint, OMPC_hint);
- if (HintExpr.isInvalid())
- return nullptr;
- return new (Context)
- OMPHintClause(HintExpr.get(), StartLoc, LParenLoc, EndLoc);
- }
- OMPClause *Sema::ActOnOpenMPDistScheduleClause(
- OpenMPDistScheduleClauseKind Kind, Expr *ChunkSize, SourceLocation StartLoc,
- SourceLocation LParenLoc, SourceLocation KindLoc, SourceLocation CommaLoc,
- SourceLocation EndLoc) {
- if (Kind == OMPC_DIST_SCHEDULE_unknown) {
- std::string Values;
- Values += "'";
- Values += getOpenMPSimpleClauseTypeName(OMPC_dist_schedule, 0);
- Values += "'";
- Diag(KindLoc, diag::err_omp_unexpected_clause_value)
- << Values << getOpenMPClauseName(OMPC_dist_schedule);
- return nullptr;
- }
- Expr *ValExpr = ChunkSize;
- Stmt *HelperValStmt = nullptr;
- if (ChunkSize) {
- if (!ChunkSize->isValueDependent() && !ChunkSize->isTypeDependent() &&
- !ChunkSize->isInstantiationDependent() &&
- !ChunkSize->containsUnexpandedParameterPack()) {
- SourceLocation ChunkSizeLoc = ChunkSize->getBeginLoc();
- ExprResult Val =
- PerformOpenMPImplicitIntegerConversion(ChunkSizeLoc, ChunkSize);
- if (Val.isInvalid())
- return nullptr;
- ValExpr = Val.get();
- // OpenMP [2.7.1, Restrictions]
- // chunk_size must be a loop invariant integer expression with a positive
- // value.
- llvm::APSInt Result;
- if (ValExpr->isIntegerConstantExpr(Result, Context)) {
- if (Result.isSigned() && !Result.isStrictlyPositive()) {
- Diag(ChunkSizeLoc, diag::err_omp_negative_expression_in_clause)
- << "dist_schedule" << ChunkSize->getSourceRange();
- return nullptr;
- }
- } else if (getOpenMPCaptureRegionForClause(
- DSAStack->getCurrentDirective(), OMPC_dist_schedule) !=
- OMPD_unknown &&
- !CurContext->isDependentContext()) {
- ValExpr = MakeFullExpr(ValExpr).get();
- llvm::MapVector<const Expr *, DeclRefExpr *> Captures;
- ValExpr = tryBuildCapture(*this, ValExpr, Captures).get();
- HelperValStmt = buildPreInits(Context, Captures);
- }
- }
- }
- return new (Context)
- OMPDistScheduleClause(StartLoc, LParenLoc, KindLoc, CommaLoc, EndLoc,
- Kind, ValExpr, HelperValStmt);
- }
- OMPClause *Sema::ActOnOpenMPDefaultmapClause(
- OpenMPDefaultmapClauseModifier M, OpenMPDefaultmapClauseKind Kind,
- SourceLocation StartLoc, SourceLocation LParenLoc, SourceLocation MLoc,
- SourceLocation KindLoc, SourceLocation EndLoc) {
- // OpenMP 4.5 only supports 'defaultmap(tofrom: scalar)'
- if (M != OMPC_DEFAULTMAP_MODIFIER_tofrom || Kind != OMPC_DEFAULTMAP_scalar) {
- std::string Value;
- SourceLocation Loc;
- Value += "'";
- if (M != OMPC_DEFAULTMAP_MODIFIER_tofrom) {
- Value += getOpenMPSimpleClauseTypeName(OMPC_defaultmap,
- OMPC_DEFAULTMAP_MODIFIER_tofrom);
- Loc = MLoc;
- } else {
- Value += getOpenMPSimpleClauseTypeName(OMPC_defaultmap,
- OMPC_DEFAULTMAP_scalar);
- Loc = KindLoc;
- }
- Value += "'";
- Diag(Loc, diag::err_omp_unexpected_clause_value)
- << Value << getOpenMPClauseName(OMPC_defaultmap);
- return nullptr;
- }
- DSAStack->setDefaultDMAToFromScalar(StartLoc);
- return new (Context)
- OMPDefaultmapClause(StartLoc, LParenLoc, MLoc, KindLoc, EndLoc, Kind, M);
- }
- bool Sema::ActOnStartOpenMPDeclareTargetDirective(SourceLocation Loc) {
- DeclContext *CurLexicalContext = getCurLexicalContext();
- if (!CurLexicalContext->isFileContext() &&
- !CurLexicalContext->isExternCContext() &&
- !CurLexicalContext->isExternCXXContext() &&
- !isa<CXXRecordDecl>(CurLexicalContext) &&
- !isa<ClassTemplateDecl>(CurLexicalContext) &&
- !isa<ClassTemplatePartialSpecializationDecl>(CurLexicalContext) &&
- !isa<ClassTemplateSpecializationDecl>(CurLexicalContext)) {
- Diag(Loc, diag::err_omp_region_not_file_context);
- return false;
- }
- ++DeclareTargetNestingLevel;
- return true;
- }
- void Sema::ActOnFinishOpenMPDeclareTargetDirective() {
- assert(DeclareTargetNestingLevel > 0 &&
- "Unexpected ActOnFinishOpenMPDeclareTargetDirective");
- --DeclareTargetNestingLevel;
- }
- void Sema::ActOnOpenMPDeclareTargetName(Scope *CurScope,
- CXXScopeSpec &ScopeSpec,
- const DeclarationNameInfo &Id,
- OMPDeclareTargetDeclAttr::MapTypeTy MT,
- NamedDeclSetType &SameDirectiveDecls) {
- LookupResult Lookup(*this, Id, LookupOrdinaryName);
- LookupParsedName(Lookup, CurScope, &ScopeSpec, true);
- if (Lookup.isAmbiguous())
- return;
- Lookup.suppressDiagnostics();
- if (!Lookup.isSingleResult()) {
- if (TypoCorrection Corrected =
- CorrectTypo(Id, LookupOrdinaryName, CurScope, nullptr,
- llvm::make_unique<VarOrFuncDeclFilterCCC>(*this),
- CTK_ErrorRecovery)) {
- diagnoseTypo(Corrected, PDiag(diag::err_undeclared_var_use_suggest)
- << Id.getName());
- checkDeclIsAllowedInOpenMPTarget(nullptr, Corrected.getCorrectionDecl());
- return;
- }
- Diag(Id.getLoc(), diag::err_undeclared_var_use) << Id.getName();
- return;
- }
- NamedDecl *ND = Lookup.getAsSingle<NamedDecl>();
- if (isa<VarDecl>(ND) || isa<FunctionDecl>(ND) ||
- isa<FunctionTemplateDecl>(ND)) {
- if (!SameDirectiveDecls.insert(cast<NamedDecl>(ND->getCanonicalDecl())))
- Diag(Id.getLoc(), diag::err_omp_declare_target_multiple) << Id.getName();
- llvm::Optional<OMPDeclareTargetDeclAttr::MapTypeTy> Res =
- OMPDeclareTargetDeclAttr::isDeclareTargetDeclaration(
- cast<ValueDecl>(ND));
- if (!Res) {
- auto *A = OMPDeclareTargetDeclAttr::CreateImplicit(Context, MT);
- ND->addAttr(A);
- if (ASTMutationListener *ML = Context.getASTMutationListener())
- ML->DeclarationMarkedOpenMPDeclareTarget(ND, A);
- checkDeclIsAllowedInOpenMPTarget(nullptr, ND, Id.getLoc());
- } else if (*Res != MT) {
- Diag(Id.getLoc(), diag::err_omp_declare_target_to_and_link)
- << Id.getName();
- }
- } else {
- Diag(Id.getLoc(), diag::err_omp_invalid_target_decl) << Id.getName();
- }
- }
- static void checkDeclInTargetContext(SourceLocation SL, SourceRange SR,
- Sema &SemaRef, Decl *D) {
- if (!D || !isa<VarDecl>(D))
- return;
- auto *VD = cast<VarDecl>(D);
- if (OMPDeclareTargetDeclAttr::isDeclareTargetDeclaration(VD))
- return;
- SemaRef.Diag(VD->getLocation(), diag::warn_omp_not_in_target_context);
- SemaRef.Diag(SL, diag::note_used_here) << SR;
- }
- static bool checkValueDeclInTarget(SourceLocation SL, SourceRange SR,
- Sema &SemaRef, DSAStackTy *Stack,
- ValueDecl *VD) {
- return VD->hasAttr<OMPDeclareTargetDeclAttr>() ||
- checkTypeMappable(SL, SR, SemaRef, Stack, VD->getType(),
- /*FullCheck=*/false);
- }
- void Sema::checkDeclIsAllowedInOpenMPTarget(Expr *E, Decl *D,
- SourceLocation IdLoc) {
- if (!D || D->isInvalidDecl())
- return;
- SourceRange SR = E ? E->getSourceRange() : D->getSourceRange();
- SourceLocation SL = E ? E->getBeginLoc() : D->getLocation();
- if (auto *VD = dyn_cast<VarDecl>(D)) {
- // Only global variables can be marked as declare target.
- if (!VD->isFileVarDecl() && !VD->isStaticLocal() &&
- !VD->isStaticDataMember())
- return;
- // 2.10.6: threadprivate variable cannot appear in a declare target
- // directive.
- if (DSAStack->isThreadPrivate(VD)) {
- Diag(SL, diag::err_omp_threadprivate_in_target);
- reportOriginalDsa(*this, DSAStack, VD, DSAStack->getTopDSA(VD, false));
- return;
- }
- }
- if (const auto *FTD = dyn_cast<FunctionTemplateDecl>(D))
- D = FTD->getTemplatedDecl();
- if (const auto *FD = dyn_cast<FunctionDecl>(D)) {
- llvm::Optional<OMPDeclareTargetDeclAttr::MapTypeTy> Res =
- OMPDeclareTargetDeclAttr::isDeclareTargetDeclaration(FD);
- if (Res && *Res == OMPDeclareTargetDeclAttr::MT_Link) {
- assert(IdLoc.isValid() && "Source location is expected");
- Diag(IdLoc, diag::err_omp_function_in_link_clause);
- Diag(FD->getLocation(), diag::note_defined_here) << FD;
- return;
- }
- }
- if (auto *VD = dyn_cast<ValueDecl>(D)) {
- // Problem if any with var declared with incomplete type will be reported
- // as normal, so no need to check it here.
- if ((E || !VD->getType()->isIncompleteType()) &&
- !checkValueDeclInTarget(SL, SR, *this, DSAStack, VD))
- return;
- if (!E && !OMPDeclareTargetDeclAttr::isDeclareTargetDeclaration(VD)) {
- // Checking declaration inside declare target region.
- if (isa<VarDecl>(D) || isa<FunctionDecl>(D) ||
- isa<FunctionTemplateDecl>(D)) {
- auto *A = OMPDeclareTargetDeclAttr::CreateImplicit(
- Context, OMPDeclareTargetDeclAttr::MT_To);
- D->addAttr(A);
- if (ASTMutationListener *ML = Context.getASTMutationListener())
- ML->DeclarationMarkedOpenMPDeclareTarget(D, A);
- }
- return;
- }
- }
- if (!E)
- return;
- checkDeclInTargetContext(E->getExprLoc(), E->getSourceRange(), *this, D);
- }
- OMPClause *Sema::ActOnOpenMPToClause(ArrayRef<Expr *> VarList,
- SourceLocation StartLoc,
- SourceLocation LParenLoc,
- SourceLocation EndLoc) {
- MappableVarListInfo MVLI(VarList);
- checkMappableExpressionList(*this, DSAStack, OMPC_to, MVLI, StartLoc);
- if (MVLI.ProcessedVarList.empty())
- return nullptr;
- return OMPToClause::Create(Context, StartLoc, LParenLoc, EndLoc,
- MVLI.ProcessedVarList, MVLI.VarBaseDeclarations,
- MVLI.VarComponents);
- }
- OMPClause *Sema::ActOnOpenMPFromClause(ArrayRef<Expr *> VarList,
- SourceLocation StartLoc,
- SourceLocation LParenLoc,
- SourceLocation EndLoc) {
- MappableVarListInfo MVLI(VarList);
- checkMappableExpressionList(*this, DSAStack, OMPC_from, MVLI, StartLoc);
- if (MVLI.ProcessedVarList.empty())
- return nullptr;
- return OMPFromClause::Create(Context, StartLoc, LParenLoc, EndLoc,
- MVLI.ProcessedVarList, MVLI.VarBaseDeclarations,
- MVLI.VarComponents);
- }
- OMPClause *Sema::ActOnOpenMPUseDevicePtrClause(ArrayRef<Expr *> VarList,
- SourceLocation StartLoc,
- SourceLocation LParenLoc,
- SourceLocation EndLoc) {
- MappableVarListInfo MVLI(VarList);
- SmallVector<Expr *, 8> PrivateCopies;
- SmallVector<Expr *, 8> Inits;
- for (Expr *RefExpr : VarList) {
- assert(RefExpr && "NULL expr in OpenMP use_device_ptr clause.");
- SourceLocation ELoc;
- SourceRange ERange;
- Expr *SimpleRefExpr = RefExpr;
- auto Res = getPrivateItem(*this, SimpleRefExpr, ELoc, ERange);
- if (Res.second) {
- // It will be analyzed later.
- MVLI.ProcessedVarList.push_back(RefExpr);
- PrivateCopies.push_back(nullptr);
- Inits.push_back(nullptr);
- }
- ValueDecl *D = Res.first;
- if (!D)
- continue;
- QualType Type = D->getType();
- Type = Type.getNonReferenceType().getUnqualifiedType();
- auto *VD = dyn_cast<VarDecl>(D);
- // Item should be a pointer or reference to pointer.
- if (!Type->isPointerType()) {
- Diag(ELoc, diag::err_omp_usedeviceptr_not_a_pointer)
- << 0 << RefExpr->getSourceRange();
- continue;
- }
- // Build the private variable and the expression that refers to it.
- auto VDPrivate =
- buildVarDecl(*this, ELoc, Type, D->getName(),
- D->hasAttrs() ? &D->getAttrs() : nullptr,
- VD ? cast<DeclRefExpr>(SimpleRefExpr) : nullptr);
- if (VDPrivate->isInvalidDecl())
- continue;
- CurContext->addDecl(VDPrivate);
- DeclRefExpr *VDPrivateRefExpr = buildDeclRefExpr(
- *this, VDPrivate, RefExpr->getType().getUnqualifiedType(), ELoc);
- // Add temporary variable to initialize the private copy of the pointer.
- VarDecl *VDInit =
- buildVarDecl(*this, RefExpr->getExprLoc(), Type, ".devptr.temp");
- DeclRefExpr *VDInitRefExpr = buildDeclRefExpr(
- *this, VDInit, RefExpr->getType(), RefExpr->getExprLoc());
- AddInitializerToDecl(VDPrivate,
- DefaultLvalueConversion(VDInitRefExpr).get(),
- /*DirectInit=*/false);
- // If required, build a capture to implement the privatization initialized
- // with the current list item value.
- DeclRefExpr *Ref = nullptr;
- if (!VD)
- Ref = buildCapture(*this, D, SimpleRefExpr, /*WithInit=*/true);
- MVLI.ProcessedVarList.push_back(VD ? RefExpr->IgnoreParens() : Ref);
- PrivateCopies.push_back(VDPrivateRefExpr);
- Inits.push_back(VDInitRefExpr);
- // We need to add a data sharing attribute for this variable to make sure it
- // is correctly captured. A variable that shows up in a use_device_ptr has
- // similar properties of a first private variable.
- DSAStack->addDSA(D, RefExpr->IgnoreParens(), OMPC_firstprivate, Ref);
- // Create a mappable component for the list item. List items in this clause
- // only need a component.
- MVLI.VarBaseDeclarations.push_back(D);
- MVLI.VarComponents.resize(MVLI.VarComponents.size() + 1);
- MVLI.VarComponents.back().push_back(
- OMPClauseMappableExprCommon::MappableComponent(SimpleRefExpr, D));
- }
- if (MVLI.ProcessedVarList.empty())
- return nullptr;
- return OMPUseDevicePtrClause::Create(
- Context, StartLoc, LParenLoc, EndLoc, MVLI.ProcessedVarList,
- PrivateCopies, Inits, MVLI.VarBaseDeclarations, MVLI.VarComponents);
- }
- OMPClause *Sema::ActOnOpenMPIsDevicePtrClause(ArrayRef<Expr *> VarList,
- SourceLocation StartLoc,
- SourceLocation LParenLoc,
- SourceLocation EndLoc) {
- MappableVarListInfo MVLI(VarList);
- for (Expr *RefExpr : VarList) {
- assert(RefExpr && "NULL expr in OpenMP is_device_ptr clause.");
- SourceLocation ELoc;
- SourceRange ERange;
- Expr *SimpleRefExpr = RefExpr;
- auto Res = getPrivateItem(*this, SimpleRefExpr, ELoc, ERange);
- if (Res.second) {
- // It will be analyzed later.
- MVLI.ProcessedVarList.push_back(RefExpr);
- }
- ValueDecl *D = Res.first;
- if (!D)
- continue;
- QualType Type = D->getType();
- // item should be a pointer or array or reference to pointer or array
- if (!Type.getNonReferenceType()->isPointerType() &&
- !Type.getNonReferenceType()->isArrayType()) {
- Diag(ELoc, diag::err_omp_argument_type_isdeviceptr)
- << 0 << RefExpr->getSourceRange();
- continue;
- }
- // Check if the declaration in the clause does not show up in any data
- // sharing attribute.
- DSAStackTy::DSAVarData DVar = DSAStack->getTopDSA(D, /*FromParent=*/false);
- if (isOpenMPPrivate(DVar.CKind)) {
- Diag(ELoc, diag::err_omp_variable_in_given_clause_and_dsa)
- << getOpenMPClauseName(DVar.CKind)
- << getOpenMPClauseName(OMPC_is_device_ptr)
- << getOpenMPDirectiveName(DSAStack->getCurrentDirective());
- reportOriginalDsa(*this, DSAStack, D, DVar);
- continue;
- }
- const Expr *ConflictExpr;
- if (DSAStack->checkMappableExprComponentListsForDecl(
- D, /*CurrentRegionOnly=*/true,
- [&ConflictExpr](
- OMPClauseMappableExprCommon::MappableExprComponentListRef R,
- OpenMPClauseKind) -> bool {
- ConflictExpr = R.front().getAssociatedExpression();
- return true;
- })) {
- Diag(ELoc, diag::err_omp_map_shared_storage) << RefExpr->getSourceRange();
- Diag(ConflictExpr->getExprLoc(), diag::note_used_here)
- << ConflictExpr->getSourceRange();
- continue;
- }
- // Store the components in the stack so that they can be used to check
- // against other clauses later on.
- OMPClauseMappableExprCommon::MappableComponent MC(SimpleRefExpr, D);
- DSAStack->addMappableExpressionComponents(
- D, MC, /*WhereFoundClauseKind=*/OMPC_is_device_ptr);
- // Record the expression we've just processed.
- MVLI.ProcessedVarList.push_back(SimpleRefExpr);
- // Create a mappable component for the list item. List items in this clause
- // only need a component. We use a null declaration to signal fields in
- // 'this'.
- assert((isa<DeclRefExpr>(SimpleRefExpr) ||
- isa<CXXThisExpr>(cast<MemberExpr>(SimpleRefExpr)->getBase())) &&
- "Unexpected device pointer expression!");
- MVLI.VarBaseDeclarations.push_back(
- isa<DeclRefExpr>(SimpleRefExpr) ? D : nullptr);
- MVLI.VarComponents.resize(MVLI.VarComponents.size() + 1);
- MVLI.VarComponents.back().push_back(MC);
- }
- if (MVLI.ProcessedVarList.empty())
- return nullptr;
- return OMPIsDevicePtrClause::Create(
- Context, StartLoc, LParenLoc, EndLoc, MVLI.ProcessedVarList,
- MVLI.VarBaseDeclarations, MVLI.VarComponents);
- }
|