12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284328532863287328832893290329132923293329432953296329732983299330033013302330333043305330633073308330933103311331233133314331533163317331833193320332133223323332433253326332733283329333033313332333333343335333633373338333933403341334233433344334533463347334833493350335133523353335433553356335733583359336033613362336333643365336633673368336933703371337233733374337533763377337833793380338133823383338433853386338733883389339033913392339333943395339633973398339934003401340234033404340534063407340834093410341134123413341434153416341734183419342034213422342334243425342634273428342934303431343234333434343534363437343834393440344134423443344434453446344734483449345034513452345334543455345634573458345934603461346234633464346534663467346834693470347134723473347434753476347734783479348034813482348334843485348634873488348934903491349234933494349534963497349834993500350135023503350435053506350735083509351035113512351335143515351635173518351935203521352235233524352535263527352835293530353135323533353435353536353735383539354035413542354335443545354635473548354935503551355235533554355535563557355835593560356135623563356435653566356735683569357035713572357335743575357635773578357935803581358235833584358535863587358835893590359135923593359435953596359735983599360036013602360336043605360636073608360936103611361236133614361536163617361836193620362136223623362436253626362736283629363036313632363336343635363636373638363936403641364236433644364536463647364836493650365136523653365436553656365736583659366036613662366336643665366636673668366936703671367236733674367536763677367836793680368136823683368436853686368736883689369036913692369336943695369636973698369937003701370237033704370537063707370837093710371137123713371437153716371737183719372037213722372337243725372637273728372937303731373237333734373537363737373837393740374137423743374437453746374737483749375037513752375337543755375637573758375937603761376237633764376537663767376837693770377137723773377437753776377737783779378037813782378337843785378637873788378937903791379237933794379537963797379837993800380138023803380438053806380738083809381038113812381338143815381638173818381938203821382238233824382538263827382838293830383138323833383438353836383738383839384038413842384338443845384638473848384938503851385238533854385538563857385838593860386138623863386438653866386738683869387038713872387338743875387638773878387938803881388238833884388538863887388838893890389138923893389438953896389738983899390039013902390339043905390639073908390939103911391239133914391539163917391839193920392139223923392439253926392739283929393039313932393339343935393639373938393939403941394239433944394539463947394839493950395139523953395439553956395739583959396039613962396339643965396639673968396939703971397239733974397539763977397839793980398139823983398439853986398739883989399039913992399339943995399639973998399940004001400240034004400540064007400840094010401140124013401440154016401740184019402040214022402340244025402640274028402940304031403240334034403540364037403840394040404140424043404440454046404740484049405040514052405340544055405640574058405940604061406240634064406540664067406840694070407140724073407440754076407740784079408040814082408340844085408640874088408940904091409240934094409540964097409840994100410141024103410441054106410741084109411041114112411341144115411641174118411941204121412241234124412541264127412841294130413141324133413441354136413741384139414041414142414341444145414641474148414941504151415241534154415541564157415841594160416141624163416441654166416741684169417041714172417341744175417641774178417941804181418241834184418541864187418841894190419141924193419441954196419741984199420042014202420342044205420642074208420942104211421242134214421542164217421842194220422142224223422442254226422742284229423042314232423342344235423642374238423942404241424242434244424542464247424842494250425142524253425442554256425742584259426042614262426342644265426642674268426942704271427242734274427542764277427842794280428142824283428442854286428742884289429042914292429342944295429642974298429943004301430243034304430543064307430843094310431143124313431443154316431743184319432043214322432343244325432643274328432943304331433243334334433543364337433843394340434143424343434443454346434743484349435043514352435343544355435643574358435943604361436243634364436543664367436843694370437143724373437443754376437743784379438043814382438343844385438643874388438943904391439243934394439543964397439843994400440144024403440444054406440744084409441044114412441344144415441644174418441944204421442244234424442544264427442844294430443144324433443444354436443744384439444044414442444344444445444644474448444944504451445244534454445544564457445844594460446144624463446444654466446744684469447044714472447344744475447644774478447944804481448244834484448544864487448844894490449144924493449444954496449744984499450045014502450345044505450645074508450945104511451245134514451545164517451845194520452145224523452445254526452745284529453045314532453345344535453645374538453945404541454245434544454545464547454845494550455145524553455445554556455745584559456045614562456345644565456645674568456945704571457245734574457545764577457845794580458145824583458445854586458745884589459045914592459345944595459645974598459946004601460246034604460546064607460846094610461146124613461446154616461746184619462046214622462346244625462646274628462946304631463246334634463546364637463846394640464146424643464446454646464746484649465046514652465346544655465646574658465946604661466246634664466546664667466846694670467146724673467446754676467746784679468046814682468346844685468646874688468946904691469246934694469546964697469846994700470147024703470447054706470747084709471047114712471347144715471647174718471947204721472247234724472547264727472847294730473147324733473447354736473747384739474047414742474347444745474647474748474947504751475247534754475547564757475847594760476147624763476447654766476747684769477047714772477347744775477647774778477947804781478247834784478547864787478847894790479147924793479447954796479747984799480048014802480348044805480648074808480948104811481248134814481548164817481848194820482148224823482448254826482748284829483048314832483348344835483648374838483948404841484248434844484548464847484848494850485148524853485448554856485748584859486048614862486348644865486648674868486948704871487248734874487548764877487848794880488148824883488448854886488748884889489048914892489348944895489648974898489949004901490249034904490549064907490849094910491149124913491449154916491749184919492049214922492349244925492649274928492949304931493249334934493549364937493849394940494149424943494449454946494749484949495049514952495349544955495649574958495949604961496249634964496549664967496849694970497149724973497449754976497749784979498049814982498349844985498649874988498949904991499249934994499549964997499849995000500150025003500450055006500750085009501050115012501350145015501650175018501950205021502250235024502550265027502850295030503150325033503450355036503750385039504050415042504350445045504650475048504950505051505250535054505550565057505850595060506150625063506450655066506750685069507050715072507350745075507650775078507950805081508250835084508550865087508850895090509150925093509450955096509750985099510051015102510351045105510651075108510951105111511251135114511551165117511851195120512151225123512451255126512751285129513051315132513351345135513651375138513951405141514251435144514551465147514851495150515151525153515451555156515751585159516051615162516351645165516651675168516951705171517251735174517551765177517851795180518151825183518451855186518751885189519051915192519351945195519651975198519952005201520252035204520552065207520852095210521152125213521452155216521752185219522052215222522352245225522652275228522952305231523252335234523552365237523852395240524152425243524452455246524752485249525052515252525352545255525652575258525952605261526252635264526552665267526852695270527152725273527452755276527752785279528052815282528352845285528652875288528952905291529252935294529552965297529852995300530153025303530453055306530753085309531053115312531353145315531653175318531953205321532253235324532553265327532853295330533153325333533453355336533753385339534053415342534353445345534653475348534953505351535253535354535553565357535853595360536153625363536453655366536753685369537053715372537353745375537653775378537953805381538253835384538553865387538853895390539153925393539453955396539753985399540054015402540354045405540654075408540954105411541254135414541554165417541854195420542154225423542454255426542754285429543054315432543354345435543654375438543954405441544254435444544554465447544854495450545154525453545454555456545754585459546054615462546354645465546654675468546954705471547254735474547554765477547854795480548154825483548454855486548754885489549054915492549354945495549654975498549955005501550255035504550555065507550855095510551155125513551455155516551755185519552055215522552355245525552655275528552955305531553255335534553555365537553855395540554155425543554455455546554755485549555055515552555355545555555655575558555955605561556255635564556555665567556855695570557155725573557455755576557755785579558055815582558355845585558655875588558955905591559255935594559555965597559855995600560156025603560456055606560756085609561056115612561356145615561656175618561956205621562256235624562556265627562856295630563156325633563456355636563756385639564056415642564356445645564656475648564956505651565256535654565556565657565856595660566156625663566456655666566756685669567056715672567356745675567656775678567956805681568256835684568556865687568856895690569156925693569456955696569756985699570057015702570357045705570657075708570957105711571257135714571557165717571857195720572157225723572457255726572757285729573057315732573357345735573657375738573957405741574257435744574557465747574857495750575157525753575457555756575757585759576057615762576357645765576657675768576957705771577257735774577557765777577857795780578157825783578457855786578757885789579057915792579357945795579657975798579958005801580258035804580558065807580858095810581158125813581458155816581758185819582058215822582358245825582658275828582958305831583258335834583558365837583858395840584158425843584458455846584758485849585058515852585358545855585658575858585958605861586258635864586558665867586858695870587158725873587458755876587758785879588058815882588358845885588658875888588958905891589258935894589558965897589858995900590159025903590459055906590759085909591059115912591359145915591659175918591959205921592259235924592559265927592859295930593159325933593459355936593759385939594059415942594359445945594659475948594959505951595259535954595559565957595859595960596159625963596459655966596759685969597059715972597359745975597659775978597959805981598259835984598559865987598859895990599159925993599459955996599759985999600060016002600360046005600660076008600960106011601260136014601560166017601860196020602160226023602460256026602760286029603060316032603360346035603660376038603960406041604260436044604560466047604860496050605160526053605460556056605760586059606060616062606360646065606660676068606960706071607260736074607560766077607860796080608160826083608460856086608760886089609060916092609360946095609660976098609961006101610261036104610561066107610861096110611161126113611461156116611761186119612061216122612361246125612661276128612961306131613261336134613561366137613861396140614161426143614461456146614761486149615061516152615361546155615661576158615961606161616261636164616561666167616861696170617161726173617461756176617761786179618061816182618361846185618661876188618961906191619261936194619561966197619861996200620162026203620462056206620762086209621062116212621362146215621662176218621962206221622262236224622562266227622862296230623162326233623462356236623762386239624062416242624362446245624662476248624962506251625262536254625562566257625862596260626162626263626462656266626762686269627062716272627362746275627662776278627962806281628262836284628562866287628862896290629162926293629462956296629762986299630063016302630363046305630663076308630963106311631263136314631563166317631863196320632163226323632463256326632763286329633063316332633363346335633663376338633963406341634263436344634563466347634863496350635163526353635463556356635763586359636063616362636363646365636663676368636963706371637263736374637563766377637863796380638163826383638463856386638763886389639063916392639363946395639663976398639964006401640264036404640564066407640864096410641164126413641464156416641764186419642064216422642364246425642664276428642964306431643264336434643564366437643864396440644164426443644464456446644764486449645064516452645364546455645664576458645964606461646264636464646564666467646864696470647164726473647464756476647764786479648064816482648364846485648664876488648964906491649264936494649564966497649864996500650165026503650465056506650765086509651065116512651365146515651665176518651965206521652265236524652565266527652865296530653165326533653465356536653765386539654065416542654365446545654665476548654965506551655265536554655565566557655865596560656165626563656465656566656765686569657065716572657365746575657665776578657965806581658265836584658565866587658865896590659165926593659465956596659765986599660066016602660366046605660666076608660966106611661266136614661566166617661866196620662166226623662466256626662766286629663066316632663366346635663666376638663966406641664266436644664566466647664866496650665166526653665466556656665766586659666066616662666366646665666666676668666966706671667266736674667566766677667866796680668166826683668466856686668766886689669066916692669366946695669666976698669967006701670267036704670567066707670867096710671167126713671467156716671767186719672067216722672367246725672667276728672967306731673267336734673567366737673867396740674167426743674467456746674767486749675067516752675367546755675667576758675967606761676267636764676567666767676867696770677167726773677467756776677767786779678067816782678367846785678667876788678967906791679267936794679567966797679867996800680168026803680468056806680768086809681068116812681368146815681668176818681968206821682268236824682568266827682868296830683168326833683468356836683768386839684068416842684368446845684668476848684968506851685268536854685568566857685868596860686168626863686468656866686768686869687068716872687368746875687668776878687968806881688268836884688568866887688868896890689168926893689468956896689768986899690069016902690369046905690669076908690969106911691269136914691569166917691869196920692169226923692469256926692769286929693069316932693369346935693669376938693969406941694269436944694569466947694869496950695169526953695469556956695769586959696069616962696369646965696669676968696969706971697269736974697569766977697869796980698169826983698469856986698769886989699069916992699369946995699669976998699970007001700270037004700570067007700870097010701170127013701470157016701770187019702070217022702370247025702670277028702970307031703270337034703570367037703870397040704170427043704470457046704770487049705070517052705370547055705670577058705970607061706270637064706570667067706870697070707170727073707470757076707770787079708070817082708370847085708670877088708970907091709270937094709570967097709870997100710171027103710471057106710771087109711071117112711371147115711671177118711971207121712271237124712571267127712871297130713171327133713471357136713771387139714071417142714371447145714671477148714971507151715271537154715571567157715871597160716171627163716471657166716771687169717071717172717371747175717671777178717971807181718271837184718571867187718871897190719171927193719471957196719771987199720072017202720372047205720672077208720972107211721272137214721572167217721872197220722172227223722472257226722772287229723072317232723372347235723672377238723972407241724272437244724572467247724872497250725172527253725472557256725772587259726072617262726372647265726672677268726972707271727272737274727572767277727872797280728172827283728472857286728772887289729072917292729372947295729672977298729973007301730273037304730573067307730873097310731173127313731473157316731773187319732073217322732373247325732673277328732973307331733273337334733573367337733873397340734173427343734473457346734773487349735073517352735373547355735673577358735973607361736273637364736573667367736873697370737173727373737473757376737773787379738073817382738373847385738673877388738973907391739273937394739573967397739873997400740174027403740474057406740774087409741074117412741374147415741674177418741974207421742274237424742574267427742874297430743174327433743474357436743774387439744074417442744374447445744674477448744974507451745274537454745574567457745874597460746174627463746474657466746774687469747074717472747374747475747674777478747974807481748274837484748574867487748874897490749174927493749474957496749774987499750075017502750375047505750675077508750975107511751275137514751575167517751875197520752175227523752475257526752775287529753075317532753375347535753675377538753975407541754275437544754575467547754875497550755175527553755475557556755775587559756075617562756375647565756675677568756975707571757275737574757575767577757875797580758175827583758475857586758775887589759075917592759375947595759675977598759976007601760276037604760576067607760876097610761176127613761476157616761776187619762076217622762376247625762676277628762976307631763276337634763576367637763876397640764176427643764476457646764776487649765076517652765376547655765676577658765976607661766276637664766576667667766876697670767176727673767476757676767776787679768076817682768376847685768676877688768976907691769276937694769576967697769876997700770177027703770477057706770777087709771077117712771377147715771677177718771977207721772277237724772577267727772877297730773177327733773477357736773777387739774077417742774377447745774677477748774977507751775277537754775577567757775877597760776177627763776477657766776777687769777077717772777377747775777677777778777977807781778277837784778577867787778877897790779177927793779477957796779777987799780078017802780378047805780678077808780978107811781278137814781578167817781878197820782178227823782478257826782778287829783078317832783378347835783678377838783978407841784278437844784578467847784878497850785178527853785478557856785778587859786078617862786378647865786678677868786978707871787278737874787578767877787878797880788178827883788478857886788778887889789078917892789378947895789678977898789979007901790279037904790579067907790879097910791179127913791479157916791779187919792079217922792379247925792679277928792979307931793279337934793579367937793879397940794179427943794479457946794779487949795079517952795379547955795679577958795979607961796279637964796579667967796879697970797179727973797479757976797779787979798079817982798379847985798679877988798979907991799279937994799579967997799879998000800180028003800480058006800780088009801080118012801380148015801680178018801980208021802280238024802580268027802880298030803180328033803480358036803780388039804080418042804380448045804680478048804980508051805280538054805580568057805880598060806180628063806480658066806780688069807080718072807380748075807680778078807980808081808280838084808580868087808880898090809180928093809480958096809780988099810081018102810381048105810681078108810981108111811281138114811581168117811881198120812181228123812481258126812781288129813081318132813381348135813681378138813981408141814281438144814581468147814881498150815181528153815481558156815781588159816081618162816381648165816681678168816981708171817281738174817581768177817881798180818181828183818481858186818781888189819081918192819381948195819681978198819982008201820282038204820582068207820882098210821182128213821482158216821782188219822082218222822382248225822682278228822982308231823282338234823582368237823882398240824182428243824482458246824782488249825082518252825382548255825682578258825982608261826282638264826582668267826882698270827182728273827482758276827782788279828082818282828382848285828682878288828982908291829282938294829582968297829882998300830183028303830483058306830783088309831083118312831383148315831683178318831983208321832283238324832583268327832883298330833183328333833483358336833783388339834083418342834383448345834683478348834983508351835283538354835583568357835883598360836183628363836483658366836783688369837083718372837383748375837683778378837983808381838283838384838583868387838883898390839183928393839483958396839783988399840084018402840384048405840684078408840984108411841284138414841584168417841884198420842184228423842484258426842784288429843084318432843384348435843684378438843984408441844284438444844584468447844884498450845184528453845484558456845784588459846084618462846384648465846684678468846984708471847284738474847584768477847884798480848184828483848484858486848784888489849084918492849384948495849684978498849985008501850285038504850585068507850885098510851185128513851485158516851785188519852085218522852385248525852685278528852985308531853285338534853585368537853885398540854185428543854485458546854785488549855085518552855385548555855685578558855985608561856285638564856585668567856885698570857185728573857485758576857785788579858085818582858385848585858685878588858985908591859285938594859585968597859885998600860186028603860486058606860786088609861086118612861386148615861686178618861986208621862286238624862586268627862886298630863186328633863486358636863786388639864086418642864386448645864686478648864986508651865286538654865586568657865886598660866186628663866486658666866786688669867086718672867386748675867686778678867986808681868286838684868586868687868886898690869186928693869486958696869786988699870087018702870387048705870687078708870987108711871287138714871587168717871887198720872187228723872487258726872787288729873087318732873387348735873687378738873987408741874287438744874587468747874887498750875187528753875487558756875787588759876087618762876387648765876687678768876987708771877287738774877587768777877887798780878187828783878487858786878787888789879087918792879387948795879687978798879988008801880288038804880588068807880888098810881188128813881488158816881788188819882088218822882388248825882688278828882988308831883288338834883588368837883888398840884188428843884488458846884788488849885088518852885388548855885688578858885988608861886288638864886588668867886888698870887188728873887488758876887788788879888088818882888388848885888688878888888988908891889288938894889588968897889888998900890189028903890489058906890789088909891089118912891389148915891689178918891989208921892289238924892589268927892889298930893189328933893489358936893789388939894089418942894389448945894689478948894989508951895289538954895589568957895889598960896189628963896489658966896789688969897089718972897389748975897689778978897989808981898289838984898589868987898889898990899189928993899489958996899789988999900090019002900390049005900690079008900990109011901290139014901590169017901890199020902190229023902490259026902790289029903090319032903390349035903690379038903990409041904290439044904590469047904890499050905190529053905490559056905790589059906090619062906390649065906690679068906990709071907290739074907590769077907890799080908190829083908490859086908790889089909090919092909390949095909690979098909991009101910291039104910591069107910891099110911191129113911491159116911791189119912091219122912391249125912691279128912991309131913291339134913591369137913891399140914191429143914491459146914791489149915091519152915391549155915691579158915991609161916291639164916591669167916891699170917191729173917491759176917791789179918091819182918391849185918691879188918991909191919291939194919591969197919891999200920192029203920492059206920792089209921092119212921392149215921692179218921992209221922292239224922592269227922892299230923192329233923492359236923792389239924092419242924392449245924692479248924992509251925292539254925592569257925892599260926192629263926492659266926792689269927092719272927392749275927692779278927992809281928292839284928592869287928892899290929192929293929492959296929792989299930093019302930393049305930693079308930993109311931293139314931593169317931893199320932193229323932493259326932793289329933093319332933393349335933693379338933993409341934293439344934593469347934893499350935193529353935493559356935793589359936093619362936393649365936693679368936993709371937293739374937593769377937893799380938193829383938493859386938793889389939093919392939393949395939693979398939994009401940294039404940594069407940894099410941194129413941494159416941794189419942094219422942394249425942694279428942994309431943294339434943594369437943894399440944194429443944494459446944794489449945094519452945394549455945694579458945994609461946294639464946594669467946894699470947194729473947494759476947794789479948094819482948394849485948694879488948994909491949294939494949594969497949894999500950195029503950495059506950795089509951095119512951395149515951695179518951995209521952295239524952595269527952895299530953195329533953495359536953795389539954095419542954395449545954695479548954995509551955295539554955595569557955895599560956195629563956495659566956795689569957095719572957395749575957695779578957995809581958295839584958595869587958895899590959195929593959495959596959795989599960096019602960396049605960696079608960996109611961296139614961596169617961896199620962196229623962496259626962796289629963096319632963396349635963696379638963996409641964296439644964596469647964896499650965196529653965496559656965796589659966096619662966396649665966696679668966996709671967296739674967596769677967896799680968196829683968496859686968796889689969096919692969396949695969696979698969997009701970297039704970597069707970897099710971197129713971497159716971797189719972097219722972397249725972697279728972997309731973297339734973597369737973897399740974197429743974497459746974797489749975097519752975397549755975697579758975997609761976297639764976597669767976897699770977197729773977497759776977797789779978097819782978397849785978697879788978997909791979297939794979597969797979897999800980198029803980498059806980798089809981098119812981398149815981698179818981998209821982298239824982598269827982898299830983198329833983498359836983798389839984098419842984398449845984698479848984998509851985298539854985598569857985898599860986198629863986498659866986798689869987098719872987398749875987698779878987998809881988298839884988598869887988898899890989198929893989498959896989798989899990099019902990399049905990699079908990999109911991299139914991599169917991899199920992199229923992499259926992799289929993099319932993399349935993699379938993999409941994299439944994599469947994899499950995199529953995499559956995799589959996099619962996399649965996699679968996999709971997299739974997599769977997899799980998199829983998499859986998799889989999099919992999399949995999699979998999910000100011000210003100041000510006100071000810009100101001110012100131001410015100161001710018100191002010021100221002310024100251002610027100281002910030100311003210033100341003510036100371003810039100401004110042100431004410045100461004710048100491005010051100521005310054100551005610057100581005910060100611006210063100641006510066100671006810069100701007110072100731007410075100761007710078100791008010081100821008310084100851008610087100881008910090100911009210093100941009510096100971009810099101001010110102101031010410105101061010710108101091011010111101121011310114101151011610117101181011910120101211012210123101241012510126101271012810129101301013110132101331013410135101361013710138101391014010141101421014310144101451014610147101481014910150101511015210153101541015510156101571015810159101601016110162101631016410165101661016710168101691017010171101721017310174101751017610177101781017910180101811018210183101841018510186101871018810189101901019110192101931019410195101961019710198101991020010201102021020310204102051020610207102081020910210102111021210213102141021510216102171021810219102201022110222102231022410225102261022710228102291023010231102321023310234102351023610237102381023910240102411024210243102441024510246102471024810249102501025110252102531025410255102561025710258102591026010261102621026310264102651026610267102681026910270102711027210273102741027510276102771027810279102801028110282102831028410285102861028710288102891029010291102921029310294102951029610297102981029910300103011030210303103041030510306103071030810309103101031110312103131031410315103161031710318103191032010321103221032310324103251032610327103281032910330103311033210333103341033510336103371033810339103401034110342103431034410345103461034710348103491035010351103521035310354103551035610357103581035910360103611036210363103641036510366103671036810369103701037110372103731037410375103761037710378103791038010381103821038310384103851038610387103881038910390103911039210393103941039510396103971039810399104001040110402104031040410405104061040710408104091041010411104121041310414104151041610417104181041910420104211042210423104241042510426104271042810429104301043110432104331043410435104361043710438104391044010441104421044310444104451044610447104481044910450104511045210453104541045510456104571045810459104601046110462104631046410465104661046710468104691047010471104721047310474104751047610477104781047910480104811048210483104841048510486104871048810489104901049110492104931049410495104961049710498104991050010501105021050310504105051050610507105081050910510105111051210513105141051510516105171051810519105201052110522105231052410525105261052710528105291053010531105321053310534105351053610537105381053910540105411054210543105441054510546105471054810549105501055110552105531055410555105561055710558105591056010561105621056310564105651056610567105681056910570105711057210573105741057510576105771057810579105801058110582105831058410585105861058710588105891059010591105921059310594105951059610597105981059910600106011060210603106041060510606106071060810609106101061110612106131061410615106161061710618106191062010621106221062310624106251062610627106281062910630106311063210633106341063510636106371063810639106401064110642106431064410645106461064710648106491065010651106521065310654106551065610657106581065910660106611066210663106641066510666106671066810669106701067110672106731067410675106761067710678106791068010681106821068310684106851068610687106881068910690106911069210693106941069510696106971069810699107001070110702107031070410705107061070710708107091071010711107121071310714107151071610717107181071910720107211072210723107241072510726107271072810729107301073110732107331073410735107361073710738107391074010741107421074310744107451074610747107481074910750107511075210753107541075510756107571075810759107601076110762107631076410765107661076710768107691077010771107721077310774107751077610777107781077910780107811078210783107841078510786107871078810789107901079110792107931079410795107961079710798107991080010801108021080310804108051080610807108081080910810108111081210813108141081510816108171081810819108201082110822108231082410825108261082710828108291083010831108321083310834108351083610837108381083910840108411084210843108441084510846108471084810849108501085110852108531085410855108561085710858108591086010861108621086310864108651086610867108681086910870108711087210873108741087510876108771087810879108801088110882108831088410885108861088710888108891089010891108921089310894108951089610897108981089910900109011090210903109041090510906109071090810909109101091110912109131091410915109161091710918109191092010921109221092310924109251092610927109281092910930109311093210933109341093510936109371093810939109401094110942109431094410945109461094710948109491095010951109521095310954109551095610957109581095910960109611096210963109641096510966109671096810969109701097110972109731097410975109761097710978109791098010981109821098310984109851098610987109881098910990109911099210993109941099510996109971099810999110001100111002110031100411005110061100711008110091101011011110121101311014110151101611017110181101911020110211102211023110241102511026110271102811029110301103111032110331103411035110361103711038110391104011041110421104311044110451104611047110481104911050110511105211053110541105511056110571105811059110601106111062110631106411065110661106711068110691107011071110721107311074110751107611077110781107911080110811108211083110841108511086110871108811089110901109111092110931109411095110961109711098110991110011101111021110311104111051110611107111081110911110111111111211113111141111511116111171111811119111201112111122111231112411125111261112711128111291113011131111321113311134111351113611137111381113911140111411114211143111441114511146111471114811149111501115111152111531115411155111561115711158111591116011161111621116311164111651116611167111681116911170111711117211173111741117511176111771117811179111801118111182111831118411185111861118711188111891119011191111921119311194111951119611197111981119911200112011120211203112041120511206112071120811209112101121111212112131121411215112161121711218112191122011221112221122311224112251122611227112281122911230112311123211233112341123511236112371123811239112401124111242112431124411245112461124711248112491125011251112521125311254112551125611257112581125911260112611126211263112641126511266112671126811269112701127111272112731127411275112761127711278112791128011281112821128311284112851128611287112881128911290112911129211293112941129511296112971129811299113001130111302113031130411305113061130711308113091131011311113121131311314113151131611317113181131911320113211132211323113241132511326113271132811329113301133111332113331133411335113361133711338113391134011341113421134311344113451134611347113481134911350113511135211353113541135511356113571135811359113601136111362113631136411365113661136711368113691137011371113721137311374113751137611377113781137911380113811138211383113841138511386113871138811389113901139111392113931139411395113961139711398113991140011401114021140311404114051140611407114081140911410114111141211413114141141511416114171141811419114201142111422114231142411425114261142711428114291143011431114321143311434114351143611437114381143911440114411144211443114441144511446114471144811449114501145111452114531145411455114561145711458114591146011461114621146311464114651146611467114681146911470114711147211473114741147511476114771147811479114801148111482114831148411485114861148711488114891149011491114921149311494114951149611497114981149911500115011150211503115041150511506115071150811509115101151111512115131151411515115161151711518115191152011521115221152311524115251152611527115281152911530115311153211533115341153511536115371153811539115401154111542115431154411545115461154711548115491155011551115521155311554115551155611557115581155911560115611156211563115641156511566115671156811569115701157111572115731157411575115761157711578115791158011581115821158311584115851158611587115881158911590115911159211593115941159511596115971159811599116001160111602116031160411605116061160711608116091161011611116121161311614116151161611617116181161911620116211162211623116241162511626116271162811629116301163111632116331163411635116361163711638116391164011641116421164311644116451164611647116481164911650116511165211653116541165511656116571165811659116601166111662116631166411665116661166711668116691167011671116721167311674116751167611677116781167911680116811168211683116841168511686116871168811689116901169111692116931169411695116961169711698116991170011701117021170311704117051170611707117081170911710117111171211713117141171511716117171171811719117201172111722117231172411725117261172711728117291173011731117321173311734117351173611737117381173911740117411174211743117441174511746117471174811749117501175111752117531175411755117561175711758117591176011761117621176311764117651176611767117681176911770117711177211773117741177511776117771177811779117801178111782117831178411785117861178711788117891179011791117921179311794117951179611797117981179911800118011180211803118041180511806118071180811809118101181111812118131181411815118161181711818118191182011821118221182311824118251182611827118281182911830118311183211833118341183511836118371183811839118401184111842118431184411845118461184711848118491185011851118521185311854118551185611857118581185911860118611186211863118641186511866118671186811869118701187111872118731187411875118761187711878118791188011881118821188311884118851188611887118881188911890118911189211893118941189511896118971189811899119001190111902119031190411905119061190711908119091191011911119121191311914119151191611917119181191911920119211192211923119241192511926119271192811929119301193111932119331193411935119361193711938119391194011941119421194311944119451194611947119481194911950119511195211953119541195511956119571195811959119601196111962119631196411965119661196711968119691197011971119721197311974119751197611977119781197911980119811198211983119841198511986119871198811989119901199111992119931199411995119961199711998119991200012001120021200312004120051200612007120081200912010120111201212013120141201512016120171201812019120201202112022120231202412025120261202712028120291203012031120321203312034120351203612037120381203912040120411204212043120441204512046120471204812049120501205112052120531205412055120561205712058120591206012061120621206312064120651206612067120681206912070120711207212073120741207512076120771207812079120801208112082120831208412085120861208712088120891209012091120921209312094120951209612097120981209912100121011210212103121041210512106121071210812109121101211112112121131211412115121161211712118121191212012121121221212312124121251212612127121281212912130121311213212133121341213512136121371213812139121401214112142121431214412145121461214712148121491215012151121521215312154121551215612157121581215912160121611216212163121641216512166121671216812169121701217112172121731217412175121761217712178121791218012181121821218312184121851218612187121881218912190121911219212193121941219512196121971219812199122001220112202122031220412205122061220712208122091221012211122121221312214122151221612217122181221912220122211222212223122241222512226122271222812229122301223112232122331223412235122361223712238122391224012241122421224312244122451224612247122481224912250122511225212253122541225512256122571225812259122601226112262122631226412265122661226712268122691227012271122721227312274122751227612277122781227912280122811228212283122841228512286122871228812289122901229112292122931229412295122961229712298122991230012301123021230312304123051230612307123081230912310123111231212313123141231512316123171231812319123201232112322123231232412325123261232712328123291233012331123321233312334123351233612337123381233912340123411234212343123441234512346123471234812349123501235112352123531235412355123561235712358123591236012361123621236312364123651236612367123681236912370123711237212373123741237512376123771237812379123801238112382123831238412385123861238712388123891239012391123921239312394123951239612397123981239912400124011240212403124041240512406124071240812409124101241112412124131241412415124161241712418124191242012421124221242312424124251242612427124281242912430124311243212433124341243512436124371243812439124401244112442124431244412445124461244712448124491245012451124521245312454124551245612457124581245912460124611246212463124641246512466124671246812469124701247112472124731247412475124761247712478124791248012481124821248312484124851248612487124881248912490124911249212493124941249512496124971249812499125001250112502125031250412505125061250712508125091251012511125121251312514125151251612517125181251912520125211252212523125241252512526125271252812529125301253112532125331253412535125361253712538125391254012541125421254312544125451254612547125481254912550125511255212553125541255512556125571255812559125601256112562125631256412565125661256712568125691257012571125721257312574125751257612577125781257912580125811258212583125841258512586125871258812589125901259112592125931259412595125961259712598125991260012601126021260312604126051260612607126081260912610126111261212613126141261512616126171261812619126201262112622126231262412625126261262712628126291263012631126321263312634126351263612637126381263912640126411264212643126441264512646126471264812649126501265112652126531265412655126561265712658126591266012661126621266312664126651266612667126681266912670126711267212673126741267512676126771267812679126801268112682126831268412685126861268712688126891269012691126921269312694126951269612697126981269912700127011270212703127041270512706127071270812709127101271112712127131271412715127161271712718127191272012721127221272312724127251272612727127281272912730127311273212733127341273512736127371273812739127401274112742127431274412745127461274712748127491275012751127521275312754127551275612757127581275912760127611276212763127641276512766127671276812769127701277112772127731277412775127761277712778127791278012781127821278312784127851278612787127881278912790127911279212793127941279512796127971279812799128001280112802128031280412805128061280712808128091281012811128121281312814128151281612817128181281912820128211282212823128241282512826128271282812829128301283112832128331283412835128361283712838128391284012841128421284312844128451284612847128481284912850128511285212853128541285512856128571285812859128601286112862128631286412865128661286712868128691287012871128721287312874128751287612877128781287912880128811288212883128841288512886128871288812889128901289112892128931289412895128961289712898128991290012901129021290312904129051290612907129081290912910129111291212913129141291512916129171291812919129201292112922129231292412925129261292712928129291293012931129321293312934129351293612937129381293912940129411294212943129441294512946129471294812949129501295112952129531295412955129561295712958129591296012961129621296312964129651296612967129681296912970129711297212973129741297512976129771297812979129801298112982129831298412985129861298712988129891299012991129921299312994129951299612997129981299913000130011300213003130041300513006130071300813009130101301113012130131301413015130161301713018130191302013021130221302313024130251302613027130281302913030130311303213033130341303513036130371303813039130401304113042130431304413045130461304713048130491305013051130521305313054130551305613057130581305913060130611306213063130641306513066130671306813069130701307113072130731307413075130761307713078130791308013081130821308313084130851308613087130881308913090130911309213093130941309513096130971309813099131001310113102131031310413105131061310713108131091311013111131121311313114131151311613117131181311913120131211312213123131241312513126131271312813129131301313113132131331313413135131361313713138131391314013141131421314313144131451314613147131481314913150131511315213153131541315513156131571315813159131601316113162131631316413165131661316713168131691317013171131721317313174131751317613177131781317913180131811318213183131841318513186131871318813189131901319113192131931319413195131961319713198131991320013201132021320313204132051320613207132081320913210132111321213213132141321513216132171321813219132201322113222132231322413225132261322713228132291323013231132321323313234132351323613237132381323913240132411324213243132441324513246132471324813249132501325113252132531325413255132561325713258132591326013261132621326313264132651326613267132681326913270132711327213273132741327513276132771327813279132801328113282132831328413285132861328713288132891329013291132921329313294132951329613297132981329913300133011330213303133041330513306133071330813309133101331113312133131331413315133161331713318133191332013321133221332313324133251332613327133281332913330133311333213333133341333513336133371333813339133401334113342133431334413345133461334713348133491335013351133521335313354133551335613357133581335913360133611336213363133641336513366133671336813369133701337113372133731337413375133761337713378133791338013381133821338313384133851338613387133881338913390133911339213393133941339513396133971339813399134001340113402134031340413405134061340713408134091341013411134121341313414134151341613417134181341913420134211342213423134241342513426134271342813429134301343113432134331343413435134361343713438134391344013441134421344313444134451344613447134481344913450134511345213453134541345513456134571345813459134601346113462134631346413465134661346713468134691347013471134721347313474134751347613477134781347913480134811348213483134841348513486134871348813489134901349113492134931349413495134961349713498134991350013501135021350313504135051350613507135081350913510135111351213513135141351513516135171351813519135201352113522135231352413525135261352713528135291353013531135321353313534135351353613537135381353913540135411354213543135441354513546135471354813549135501355113552135531355413555135561355713558135591356013561135621356313564135651356613567135681356913570135711357213573135741357513576135771357813579135801358113582135831358413585135861358713588135891359013591135921359313594135951359613597135981359913600136011360213603136041360513606136071360813609136101361113612136131361413615136161361713618136191362013621136221362313624136251362613627136281362913630136311363213633136341363513636136371363813639136401364113642136431364413645136461364713648136491365013651136521365313654136551365613657136581365913660136611366213663136641366513666136671366813669136701367113672136731367413675136761367713678136791368013681136821368313684136851368613687136881368913690136911369213693136941369513696136971369813699137001370113702137031370413705137061370713708137091371013711137121371313714137151371613717137181371913720137211372213723137241372513726137271372813729137301373113732137331373413735137361373713738137391374013741137421374313744137451374613747137481374913750137511375213753137541375513756137571375813759137601376113762137631376413765137661376713768137691377013771137721377313774137751377613777137781377913780137811378213783137841378513786137871378813789137901379113792137931379413795137961379713798137991380013801138021380313804138051380613807138081380913810138111381213813138141381513816138171381813819138201382113822138231382413825138261382713828138291383013831138321383313834138351383613837138381383913840138411384213843138441384513846138471384813849138501385113852138531385413855138561385713858138591386013861138621386313864138651386613867138681386913870138711387213873138741387513876138771387813879138801388113882138831388413885138861388713888138891389013891138921389313894138951389613897138981389913900139011390213903139041390513906139071390813909139101391113912139131391413915139161391713918139191392013921139221392313924139251392613927139281392913930139311393213933139341393513936139371393813939139401394113942139431394413945139461394713948139491395013951139521395313954139551395613957139581395913960139611396213963139641396513966139671396813969139701397113972139731397413975139761397713978139791398013981139821398313984139851398613987139881398913990139911399213993139941399513996139971399813999140001400114002140031400414005140061400714008140091401014011140121401314014140151401614017140181401914020140211402214023140241402514026140271402814029140301403114032140331403414035140361403714038140391404014041140421404314044140451404614047140481404914050140511405214053140541405514056140571405814059140601406114062140631406414065140661406714068140691407014071140721407314074140751407614077140781407914080140811408214083140841408514086140871408814089140901409114092140931409414095140961409714098140991410014101141021410314104141051410614107141081410914110141111411214113141141411514116141171411814119141201412114122141231412414125141261412714128141291413014131141321413314134141351413614137141381413914140141411414214143141441414514146141471414814149141501415114152141531415414155141561415714158141591416014161141621416314164141651416614167141681416914170141711417214173141741417514176141771417814179141801418114182141831418414185141861418714188141891419014191141921419314194141951419614197141981419914200142011420214203142041420514206142071420814209142101421114212142131421414215142161421714218142191422014221142221422314224142251422614227142281422914230142311423214233142341423514236142371423814239142401424114242142431424414245142461424714248142491425014251142521425314254142551425614257142581425914260142611426214263142641426514266142671426814269142701427114272142731427414275142761427714278142791428014281142821428314284142851428614287142881428914290142911429214293142941429514296142971429814299143001430114302143031430414305143061430714308143091431014311143121431314314143151431614317143181431914320143211432214323143241432514326143271432814329143301433114332143331433414335143361433714338143391434014341143421434314344143451434614347143481434914350143511435214353143541435514356143571435814359143601436114362143631436414365143661436714368143691437014371143721437314374143751437614377143781437914380143811438214383143841438514386143871438814389143901439114392143931439414395143961439714398143991440014401144021440314404144051440614407144081440914410144111441214413144141441514416144171441814419144201442114422144231442414425144261442714428144291443014431144321443314434144351443614437144381443914440144411444214443144441444514446144471444814449144501445114452144531445414455144561445714458144591446014461144621446314464144651446614467144681446914470144711447214473144741447514476144771447814479144801448114482144831448414485144861448714488144891449014491144921449314494144951449614497144981449914500145011450214503145041450514506145071450814509145101451114512145131451414515145161451714518145191452014521145221452314524145251452614527145281452914530145311453214533145341453514536145371453814539145401454114542145431454414545145461454714548145491455014551145521455314554145551455614557145581455914560145611456214563145641456514566145671456814569145701457114572145731457414575145761457714578145791458014581145821458314584145851458614587145881458914590145911459214593145941459514596145971459814599146001460114602146031460414605146061460714608146091461014611146121461314614146151461614617146181461914620146211462214623146241462514626146271462814629146301463114632146331463414635146361463714638146391464014641146421464314644146451464614647146481464914650146511465214653146541465514656146571465814659146601466114662146631466414665146661466714668146691467014671146721467314674146751467614677146781467914680146811468214683146841468514686146871468814689146901469114692146931469414695146961469714698146991470014701147021470314704147051470614707147081470914710147111471214713147141471514716147171471814719147201472114722147231472414725147261472714728147291473014731147321473314734147351473614737147381473914740147411474214743147441474514746147471474814749147501475114752147531475414755147561475714758147591476014761147621476314764147651476614767147681476914770147711477214773147741477514776147771477814779147801478114782147831478414785147861478714788147891479014791147921479314794147951479614797147981479914800148011480214803148041480514806148071480814809148101481114812148131481414815148161481714818148191482014821148221482314824148251482614827148281482914830148311483214833148341483514836148371483814839148401484114842148431484414845148461484714848148491485014851148521485314854148551485614857148581485914860148611486214863148641486514866148671486814869148701487114872148731487414875148761487714878148791488014881148821488314884148851488614887148881488914890148911489214893148941489514896148971489814899149001490114902149031490414905149061490714908149091491014911149121491314914149151491614917149181491914920149211492214923149241492514926149271492814929149301493114932149331493414935149361493714938149391494014941149421494314944149451494614947149481494914950149511495214953149541495514956149571495814959149601496114962149631496414965149661496714968149691497014971149721497314974149751497614977149781497914980149811498214983149841498514986149871498814989149901499114992149931499414995149961499714998149991500015001150021500315004150051500615007150081500915010150111501215013150141501515016150171501815019150201502115022150231502415025150261502715028150291503015031150321503315034150351503615037150381503915040150411504215043150441504515046150471504815049150501505115052150531505415055150561505715058150591506015061150621506315064150651506615067150681506915070150711507215073150741507515076150771507815079150801508115082150831508415085150861508715088150891509015091150921509315094150951509615097150981509915100151011510215103151041510515106151071510815109151101511115112151131511415115151161511715118151191512015121151221512315124151251512615127151281512915130151311513215133151341513515136151371513815139151401514115142151431514415145151461514715148151491515015151151521515315154151551515615157151581515915160151611516215163151641516515166151671516815169151701517115172151731517415175151761517715178151791518015181151821518315184151851518615187151881518915190151911519215193151941519515196151971519815199152001520115202152031520415205152061520715208152091521015211152121521315214152151521615217152181521915220152211522215223152241522515226152271522815229152301523115232152331523415235152361523715238152391524015241152421524315244152451524615247152481524915250152511525215253152541525515256152571525815259152601526115262152631526415265152661526715268152691527015271152721527315274152751527615277152781527915280152811528215283152841528515286152871528815289152901529115292152931529415295152961529715298152991530015301153021530315304153051530615307153081530915310153111531215313153141531515316153171531815319153201532115322153231532415325153261532715328153291533015331153321533315334153351533615337153381533915340153411534215343153441534515346153471534815349153501535115352153531535415355153561535715358153591536015361153621536315364153651536615367153681536915370153711537215373153741537515376153771537815379153801538115382153831538415385153861538715388153891539015391153921539315394153951539615397153981539915400154011540215403154041540515406154071540815409154101541115412154131541415415154161541715418154191542015421154221542315424154251542615427154281542915430154311543215433154341543515436154371543815439154401544115442154431544415445154461544715448154491545015451154521545315454154551545615457154581545915460154611546215463154641546515466154671546815469154701547115472154731547415475154761547715478154791548015481154821548315484154851548615487154881548915490154911549215493154941549515496154971549815499155001550115502155031550415505155061550715508155091551015511155121551315514155151551615517155181551915520155211552215523155241552515526155271552815529155301553115532155331553415535155361553715538155391554015541155421554315544155451554615547155481554915550155511555215553155541555515556155571555815559155601556115562155631556415565155661556715568155691557015571155721557315574155751557615577155781557915580155811558215583155841558515586155871558815589155901559115592155931559415595155961559715598155991560015601156021560315604156051560615607156081560915610156111561215613156141561515616156171561815619156201562115622156231562415625156261562715628156291563015631156321563315634156351563615637156381563915640156411564215643156441564515646156471564815649156501565115652156531565415655156561565715658156591566015661156621566315664156651566615667156681566915670156711567215673156741567515676156771567815679156801568115682156831568415685156861568715688156891569015691156921569315694156951569615697156981569915700157011570215703157041570515706157071570815709157101571115712157131571415715157161571715718157191572015721157221572315724157251572615727157281572915730157311573215733157341573515736157371573815739157401574115742157431574415745157461574715748157491575015751157521575315754157551575615757157581575915760157611576215763157641576515766157671576815769157701577115772157731577415775157761577715778157791578015781157821578315784157851578615787157881578915790157911579215793157941579515796157971579815799158001580115802158031580415805158061580715808158091581015811158121581315814158151581615817158181581915820158211582215823158241582515826158271582815829158301583115832158331583415835158361583715838158391584015841158421584315844158451584615847158481584915850158511585215853158541585515856158571585815859158601586115862158631586415865158661586715868158691587015871158721587315874158751587615877158781587915880158811588215883158841588515886158871588815889158901589115892158931589415895158961589715898158991590015901159021590315904159051590615907159081590915910159111591215913159141591515916159171591815919159201592115922159231592415925159261592715928159291593015931159321593315934159351593615937159381593915940159411594215943159441594515946159471594815949159501595115952159531595415955159561595715958159591596015961159621596315964159651596615967159681596915970159711597215973159741597515976159771597815979159801598115982159831598415985159861598715988159891599015991159921599315994159951599615997159981599916000160011600216003160041600516006160071600816009160101601116012160131601416015160161601716018160191602016021160221602316024160251602616027160281602916030160311603216033160341603516036160371603816039160401604116042160431604416045160461604716048160491605016051160521605316054160551605616057160581605916060160611606216063160641606516066160671606816069160701607116072160731607416075160761607716078160791608016081160821608316084160851608616087160881608916090160911609216093160941609516096160971609816099161001610116102161031610416105161061610716108161091611016111161121611316114161151611616117161181611916120161211612216123161241612516126161271612816129161301613116132161331613416135161361613716138161391614016141161421614316144161451614616147161481614916150161511615216153161541615516156161571615816159161601616116162161631616416165161661616716168161691617016171161721617316174161751617616177161781617916180161811618216183161841618516186161871618816189161901619116192161931619416195161961619716198161991620016201162021620316204162051620616207162081620916210162111621216213162141621516216162171621816219162201622116222162231622416225162261622716228162291623016231162321623316234162351623616237162381623916240162411624216243162441624516246162471624816249162501625116252162531625416255162561625716258162591626016261162621626316264162651626616267162681626916270162711627216273162741627516276162771627816279162801628116282 |
- //===--- SemaOpenMP.cpp - Semantic Analysis for OpenMP constructs ---------===//
- //
- // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
- // See https://llvm.org/LICENSE.txt for license information.
- // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
- //
- //===----------------------------------------------------------------------===//
- /// \file
- /// This file implements 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/AST/TypeOrdering.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;
- bool HasMutipleLoops = false;
- const Decl *PossiblyLoopCounter = nullptr;
- bool NowaitRegion = false;
- bool CancelRegion = false;
- bool LoopStart = false;
- bool BodyComplete = false;
- SourceLocation InnerTeamsRegionLoc;
- /// Reference to the taskgroup task_reduction reference expression.
- Expr *TaskgroupReductionRef = nullptr;
- llvm::DenseSet<QualType> MappedClassesQualTypes;
- /// List of globals marked as declare target link in this target region
- /// (isOpenMPTargetExecutionDirective(Directive) == true).
- llvm::SmallVector<DeclRefExpr *, 4> DeclareTargetLinkVarDecls;
- 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 variables in the target executable directives must be
- /// captured by reference.
- bool ForceCaptureByReferenceInTargetExecutable = false;
- CriticalsWithHintsTy Criticals;
- unsigned IgnoredStackElements = 0;
- /// Iterators over the stack iterate in order from innermost to outermost
- /// directive.
- using const_iterator = StackTy::const_reverse_iterator;
- const_iterator begin() const {
- return Stack.empty() ? const_iterator()
- : Stack.back().first.rbegin() + IgnoredStackElements;
- }
- const_iterator end() const {
- return Stack.empty() ? const_iterator() : Stack.back().first.rend();
- }
- using iterator = StackTy::reverse_iterator;
- iterator begin() {
- return Stack.empty() ? iterator()
- : Stack.back().first.rbegin() + IgnoredStackElements;
- }
- iterator end() {
- return Stack.empty() ? iterator() : Stack.back().first.rend();
- }
- // Convenience operations to get at the elements of the stack.
- bool isStackEmpty() const {
- return Stack.empty() ||
- Stack.back().second != CurrentNonCapturingFunctionScope ||
- Stack.back().first.size() <= IgnoredStackElements;
- }
- size_t getStackSize() const {
- return isStackEmpty() ? 0
- : Stack.back().first.size() - IgnoredStackElements;
- }
- SharingMapTy *getTopOfStackOrNull() {
- size_t Size = getStackSize();
- if (Size == 0)
- return nullptr;
- return &Stack.back().first[Size - 1];
- }
- const SharingMapTy *getTopOfStackOrNull() const {
- return const_cast<DSAStackTy&>(*this).getTopOfStackOrNull();
- }
- SharingMapTy &getTopOfStack() {
- assert(!isStackEmpty() && "no current directive");
- return *getTopOfStackOrNull();
- }
- const SharingMapTy &getTopOfStack() const {
- return const_cast<DSAStackTy&>(*this).getTopOfStack();
- }
- SharingMapTy *getSecondOnStackOrNull() {
- size_t Size = getStackSize();
- if (Size <= 1)
- return nullptr;
- return &Stack.back().first[Size - 2];
- }
- const SharingMapTy *getSecondOnStackOrNull() const {
- return const_cast<DSAStackTy&>(*this).getSecondOnStackOrNull();
- }
- /// Get the stack element at a certain level (previously returned by
- /// \c getNestingLevel).
- ///
- /// Note that nesting levels count from outermost to innermost, and this is
- /// the reverse of our iteration order where new inner levels are pushed at
- /// the front of the stack.
- SharingMapTy &getStackElemAtLevel(unsigned Level) {
- assert(Level < getStackSize() && "no such stack element");
- return Stack.back().first[Level];
- }
- const SharingMapTy &getStackElemAtLevel(unsigned Level) const {
- return const_cast<DSAStackTy&>(*this).getStackElemAtLevel(Level);
- }
- DSAVarData getDSA(const_iterator &Iter, ValueDecl *D) const;
- /// Checks if the variable is a local for OpenMP region.
- bool isOpenMPLocal(VarDecl *D, const_iterator Iter) const;
- /// Vector of previously declared requires directives
- SmallVector<const OMPRequiresDecl *, 2> RequiresDecls;
- /// omp_allocator_handle_t type.
- QualType OMPAllocatorHandleT;
- /// Expression for the predefined allocators.
- Expr *OMPPredefinedAllocators[OMPAllocateDeclAttr::OMPUserDefinedMemAlloc] = {
- nullptr};
- /// Vector of previously encountered target directives
- SmallVector<SourceLocation, 2> TargetLocations;
- public:
- explicit DSAStackTy(Sema &S) : SemaRef(S) {}
- /// Sets omp_allocator_handle_t type.
- void setOMPAllocatorHandleT(QualType Ty) { OMPAllocatorHandleT = Ty; }
- /// Gets omp_allocator_handle_t type.
- QualType getOMPAllocatorHandleT() const { return OMPAllocatorHandleT; }
- /// Sets the given default allocator.
- void setAllocator(OMPAllocateDeclAttr::AllocatorTypeTy AllocatorKind,
- Expr *Allocator) {
- OMPPredefinedAllocators[AllocatorKind] = Allocator;
- }
- /// Returns the specified default allocator.
- Expr *getAllocator(OMPAllocateDeclAttr::AllocatorTypeTy AllocatorKind) const {
- return OMPPredefinedAllocators[AllocatorKind];
- }
- 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 isBodyComplete() const {
- const SharingMapTy *Top = getTopOfStackOrNull();
- return Top && Top->BodyComplete;
- }
- void setBodyComplete() {
- getTopOfStack().BodyComplete = true;
- }
- 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) {
- assert(!IgnoredStackElements &&
- "cannot change stack while ignoring elements");
- 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(!IgnoredStackElements &&
- "cannot change stack while ignoring elements");
- assert(!Stack.back().first.empty() &&
- "Data-sharing attributes stack is empty!");
- Stack.back().first.pop_back();
- }
- /// RAII object to temporarily leave the scope of a directive when we want to
- /// logically operate in its parent.
- class ParentDirectiveScope {
- DSAStackTy &Self;
- bool Active;
- public:
- ParentDirectiveScope(DSAStackTy &Self, bool Activate)
- : Self(Self), Active(false) {
- if (Activate)
- enable();
- }
- ~ParentDirectiveScope() { disable(); }
- void disable() {
- if (Active) {
- --Self.IgnoredStackElements;
- Active = false;
- }
- }
- void enable() {
- if (!Active) {
- ++Self.IgnoredStackElements;
- Active = true;
- }
- }
- };
- /// Marks that we're started loop parsing.
- void loopInit() {
- assert(isOpenMPLoopDirective(getCurrentDirective()) &&
- "Expected loop-based directive.");
- getTopOfStack().LoopStart = true;
- }
- /// Start capturing of the variables in the loop context.
- void loopStart() {
- assert(isOpenMPLoopDirective(getCurrentDirective()) &&
- "Expected loop-based directive.");
- getTopOfStack().LoopStart = false;
- }
- /// true, if variables are captured, false otherwise.
- bool isLoopStarted() const {
- assert(isOpenMPLoopDirective(getCurrentDirective()) &&
- "Expected loop-based directive.");
- return !getTopOfStack().LoopStart;
- }
- /// Marks (or clears) declaration as possibly loop counter.
- void resetPossibleLoopCounter(const Decl *D = nullptr) {
- getTopOfStack().PossiblyLoopCounter =
- D ? D->getCanonicalDecl() : D;
- }
- /// Gets the possible loop counter decl.
- const Decl *getPossiblyLoopCunter() const {
- return getTopOfStack().PossiblyLoopCounter;
- }
- /// Start new OpenMP region stack in new non-capturing function.
- void pushFunction() {
- assert(!IgnoredStackElements &&
- "cannot change stack while ignoring elements");
- const FunctionScopeInfo *CurFnScope = SemaRef.getCurFunction();
- assert(!isa<CapturingScopeInfo>(CurFnScope));
- CurrentNonCapturingFunctionScope = CurFnScope;
- }
- /// Pop region stack for non-capturing function.
- void popFunction(const FunctionScopeInfo *OldFSI) {
- assert(!IgnoredStackElements &&
- "cannot change stack while ignoring elements");
- 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(getTopOfStack().Directive == OMPD_taskgroup &&
- "taskgroup reference expression requested for non taskgroup "
- "directive.");
- return getTopOfStack().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 getStackElemAtLevel(Level).TaskgroupReductionRef &&
- cast<DeclRefExpr>(getStackElemAtLevel(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 {
- const SharingMapTy *Top = getTopOfStackOrNull();
- return Top ? Top->Directive : OMPD_unknown;
- }
- /// Returns directive kind at specified level.
- OpenMPDirectiveKind getDirective(unsigned Level) const {
- assert(!isStackEmpty() && "No directive at specified level.");
- return getStackElemAtLevel(Level).Directive;
- }
- /// Returns the capture region at the specified level.
- OpenMPDirectiveKind getCaptureRegion(unsigned Level,
- unsigned OpenMPCaptureLevel) const {
- SmallVector<OpenMPDirectiveKind, 4> CaptureRegions;
- getOpenMPCaptureRegions(CaptureRegions, getDirective(Level));
- return CaptureRegions[OpenMPCaptureLevel];
- }
- /// Returns parent directive.
- OpenMPDirectiveKind getParentDirective() const {
- const SharingMapTy *Parent = getSecondOnStackOrNull();
- return Parent ? Parent->Directive : OMPD_unknown;
- }
- /// Add requires decl to internal vector
- void addRequiresDecl(OMPRequiresDecl *RD) {
- RequiresDecls.push_back(RD);
- }
- /// Checks if the defined 'requires' directive has specified type of clause.
- template <typename ClauseType>
- bool hasRequiresDeclWithClause() {
- return llvm::any_of(RequiresDecls, [](const OMPRequiresDecl *D) {
- return llvm::any_of(D->clauselists(), [](const OMPClause *C) {
- return isa<ClauseType>(C);
- });
- });
- }
- /// 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;
- }
- /// Add location of previously encountered target to internal vector
- void addTargetDirLocation(SourceLocation LocStart) {
- TargetLocations.push_back(LocStart);
- }
- // Return previously encountered target region locations.
- ArrayRef<SourceLocation> getEncounteredTargetLocs() const {
- return TargetLocations;
- }
- /// Set default data sharing attribute to none.
- void setDefaultDSANone(SourceLocation Loc) {
- getTopOfStack().DefaultAttr = DSA_none;
- getTopOfStack().DefaultAttrLoc = Loc;
- }
- /// Set default data sharing attribute to shared.
- void setDefaultDSAShared(SourceLocation Loc) {
- getTopOfStack().DefaultAttr = DSA_shared;
- getTopOfStack().DefaultAttrLoc = Loc;
- }
- /// Set default data mapping attribute to 'tofrom:scalar'.
- void setDefaultDMAToFromScalar(SourceLocation Loc) {
- getTopOfStack().DefaultMapAttr = DMA_tofrom_scalar;
- getTopOfStack().DefaultMapAttrLoc = Loc;
- }
- DefaultDataSharingAttributes getDefaultDSA() const {
- return isStackEmpty() ? DSA_unspecified
- : getTopOfStack().DefaultAttr;
- }
- SourceLocation getDefaultDSALocation() const {
- return isStackEmpty() ? SourceLocation()
- : getTopOfStack().DefaultAttrLoc;
- }
- DefaultMapAttributes getDefaultDMA() const {
- return isStackEmpty() ? DMA_unspecified
- : getTopOfStack().DefaultMapAttr;
- }
- DefaultMapAttributes getDefaultDMAAtLevel(unsigned Level) const {
- return getStackElemAtLevel(Level).DefaultMapAttr;
- }
- SourceLocation getDefaultDMALocation() const {
- return isStackEmpty() ? SourceLocation()
- : getTopOfStack().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) {
- if (IsOrdered)
- getTopOfStack().OrderedRegion.emplace(Param, Clause);
- else
- getTopOfStack().OrderedRegion.reset();
- }
- /// Returns true, if region is ordered (has associated 'ordered' clause),
- /// false - otherwise.
- bool isOrderedRegion() const {
- if (const SharingMapTy *Top = getTopOfStackOrNull())
- return Top->OrderedRegion.hasValue();
- return false;
- }
- /// Returns optional parameter for the ordered region.
- std::pair<const Expr *, OMPOrderedClause *> getOrderedRegionParam() const {
- if (const SharingMapTy *Top = getTopOfStackOrNull())
- if (Top->OrderedRegion.hasValue())
- return Top->OrderedRegion.getValue();
- return std::make_pair(nullptr, nullptr);
- }
- /// Returns true, if parent region is ordered (has associated
- /// 'ordered' clause), false - otherwise.
- bool isParentOrderedRegion() const {
- if (const SharingMapTy *Parent = getSecondOnStackOrNull())
- return Parent->OrderedRegion.hasValue();
- return false;
- }
- /// Returns optional parameter for the ordered region.
- std::pair<const Expr *, OMPOrderedClause *>
- getParentOrderedRegionParam() const {
- if (const SharingMapTy *Parent = getSecondOnStackOrNull())
- if (Parent->OrderedRegion.hasValue())
- return Parent->OrderedRegion.getValue();
- return std::make_pair(nullptr, nullptr);
- }
- /// Marks current region as nowait (it has a 'nowait' clause).
- void setNowaitRegion(bool IsNowait = true) {
- getTopOfStack().NowaitRegion = IsNowait;
- }
- /// Returns true, if parent region is nowait (has associated
- /// 'nowait' clause), false - otherwise.
- bool isParentNowaitRegion() const {
- if (const SharingMapTy *Parent = getSecondOnStackOrNull())
- return Parent->NowaitRegion;
- return false;
- }
- /// Marks parent region as cancel region.
- void setParentCancelRegion(bool Cancel = true) {
- if (SharingMapTy *Parent = getSecondOnStackOrNull())
- Parent->CancelRegion |= Cancel;
- }
- /// Return true if current region has inner cancel construct.
- bool isCancelRegion() const {
- const SharingMapTy *Top = getTopOfStackOrNull();
- return Top ? Top->CancelRegion : false;
- }
- /// Set collapse value for the region.
- void setAssociatedLoops(unsigned Val) {
- getTopOfStack().AssociatedLoops = Val;
- if (Val > 1)
- getTopOfStack().HasMutipleLoops = true;
- }
- /// Return collapse value for region.
- unsigned getAssociatedLoops() const {
- const SharingMapTy *Top = getTopOfStackOrNull();
- return Top ? Top->AssociatedLoops : 0;
- }
- /// Returns true if the construct is associated with multiple loops.
- bool hasMutipleLoops() const {
- const SharingMapTy *Top = getTopOfStackOrNull();
- return Top ? Top->HasMutipleLoops : false;
- }
- /// Marks current target region as one with closely nested teams
- /// region.
- void setParentTeamsRegionLoc(SourceLocation TeamsRegionLoc) {
- if (SharingMapTy *Parent = getSecondOnStackOrNull())
- Parent->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 {
- const SharingMapTy *Top = getTopOfStackOrNull();
- return Top ? Top->InnerTeamsRegionLoc : SourceLocation();
- }
- Scope *getCurScope() const {
- const SharingMapTy *Top = getTopOfStackOrNull();
- return Top ? Top->CurScope : nullptr;
- }
- SourceLocation getConstructLoc() const {
- const SharingMapTy *Top = getTopOfStackOrNull();
- return Top ? Top->ConstructLoc : SourceLocation();
- }
- /// 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 = begin();
- auto SE = end();
- 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 (getStackSize() <= Level)
- return false;
- const SharingMapTy &StackElem = getStackElemAtLevel(Level);
- auto MI = StackElem.MappedExprComponents.find(VD);
- if (MI != StackElem.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) {
- MappedExprComponentTy &MEC = getTopOfStack().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 getStackSize() - 1;
- }
- void addDoacrossDependClause(OMPDependClause *C,
- const OperatorOffsetTy &OpsOffs) {
- SharingMapTy *Parent = getSecondOnStackOrNull();
- assert(Parent && isOpenMPWorksharingDirective(Parent->Directive));
- Parent->DoacrossDepends.try_emplace(C, OpsOffs);
- }
- llvm::iterator_range<DoacrossDependMapTy::const_iterator>
- getDoacrossDependClauses() const {
- const SharingMapTy &StackElem = getTopOfStack();
- 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());
- }
- // Store types of classes which have been explicitly mapped
- void addMappedClassesQualTypes(QualType QT) {
- SharingMapTy &StackElem = getTopOfStack();
- StackElem.MappedClassesQualTypes.insert(QT);
- }
- // Return set of mapped classes types
- bool isClassPreviouslyMapped(QualType QT) const {
- const SharingMapTy &StackElem = getTopOfStack();
- return StackElem.MappedClassesQualTypes.count(QT) != 0;
- }
- /// Adds global declare target to the parent target region.
- void addToParentTargetRegionLinkGlobals(DeclRefExpr *E) {
- assert(*OMPDeclareTargetDeclAttr::isDeclareTargetDeclaration(
- E->getDecl()) == OMPDeclareTargetDeclAttr::MT_Link &&
- "Expected declare target link global.");
- for (auto &Elem : *this) {
- if (isOpenMPTargetExecutionDirective(Elem.Directive)) {
- Elem.DeclareTargetLinkVarDecls.push_back(E);
- return;
- }
- }
- }
- /// Returns the list of globals with declare target link if current directive
- /// is target.
- ArrayRef<DeclRefExpr *> getLinkGlobals() const {
- assert(isOpenMPTargetExecutionDirective(getCurrentDirective()) &&
- "Expected target executable directive.");
- return getTopOfStack().DeclareTargetLinkVarDecls;
- }
- };
- bool isImplicitTaskingRegion(OpenMPDirectiveKind DKind) {
- return isOpenMPParallelDirective(DKind) || isOpenMPTeamsDirective(DKind);
- }
- bool isImplicitOrExplicitTaskingRegion(OpenMPDirectiveKind DKind) {
- return isImplicitTaskingRegion(DKind) || isOpenMPTaskingDirective(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(const_iterator &Iter,
- ValueDecl *D) const {
- D = getCanonicalDecl(D);
- auto *VD = dyn_cast<VarDecl>(D);
- const auto *FD = dyn_cast<FieldDecl>(D);
- DSAVarData DVar;
- if (Iter == end()) {
- // 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;
- const_iterator I = Iter, E = end();
- 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 && !isImplicitTaskingRegion(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 = getTopOfStack();
- 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 = getTopOfStack();
- 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 = getTopOfStack();
- 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 {
- const SharingMapTy *Parent = getSecondOnStackOrNull();
- assert(Parent && "Data-sharing attributes stack is empty");
- D = getCanonicalDecl(D);
- auto It = Parent->LCVMap.find(D);
- if (It != Parent->LCVMap.end())
- return It->second;
- return {0, nullptr};
- }
- const ValueDecl *DSAStackTy::getParentLoopControlVariable(unsigned I) const {
- const SharingMapTy *Parent = getSecondOnStackOrNull();
- assert(Parent && "Data-sharing attributes stack is empty");
- if (Parent->LCVMap.size() < I)
- return nullptr;
- for (const auto &Pair : Parent->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 {
- DSAInfo &Data = getTopOfStack().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 = getTopOfStack().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(
- getTopOfStack().SharingMap[D].Attributes == OMPC_reduction &&
- "Additional reduction info may be specified only for reduction items.");
- ReductionData &ReductionData = getTopOfStack().ReductionMap[D];
- assert(ReductionData.ReductionRange.isInvalid() &&
- getTopOfStack().Directive == OMPD_taskgroup &&
- "Additional reduction info may be specified only once for reduction "
- "items.");
- ReductionData.set(BOK, SR);
- Expr *&TaskgroupReductionRef =
- getTopOfStack().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(
- getTopOfStack().SharingMap[D].Attributes == OMPC_reduction &&
- "Additional reduction info may be specified only for reduction items.");
- ReductionData &ReductionData = getTopOfStack().ReductionMap[D];
- assert(ReductionData.ReductionRange.isInvalid() &&
- getTopOfStack().Directive == OMPD_taskgroup &&
- "Additional reduction info may be specified only once for reduction "
- "items.");
- ReductionData.set(ReductionRef, SR);
- Expr *&TaskgroupReductionRef =
- getTopOfStack().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.");
- for (const_iterator I = begin() + 1, E = end(); I != E; ++I) {
- 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.");
- for (const_iterator I = begin() + 1, E = end(); I != E; ++I) {
- 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, const_iterator I) const {
- D = D->getCanonicalDecl();
- for (const_iterator E = end(); I != E; ++I) {
- if (isImplicitOrExplicitTaskingRegion(I->Directive) ||
- isOpenMPTargetExecutionDirective(I->Directive)) {
- Scope *TopScope = I->CurScope ? I->CurScope->getParent() : nullptr;
- Scope *CurScope = getCurScope();
- while (CurScope && CurScope != TopScope && !CurScope->isDeclScope(D))
- CurScope = CurScope->getParent();
- return CurScope != TopScope;
- }
- }
- return false;
- }
- static bool isConstNotMutableType(Sema &SemaRef, QualType Type,
- bool AcceptIfMutable = true,
- bool *IsClassType = nullptr) {
- ASTContext &Context = SemaRef.getASTContext();
- Type = Type.getNonReferenceType().getCanonicalType();
- bool IsConstant = Type.isConstant(Context);
- Type = Context.getBaseElementType(Type);
- const CXXRecordDecl *RD = AcceptIfMutable && 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 (IsClassType)
- *IsClassType = RD;
- return IsConstant && !(SemaRef.getLangOpts().CPlusPlus && RD &&
- RD->hasDefinition() && RD->hasMutableFields());
- }
- static bool rejectConstNotMutableType(Sema &SemaRef, const ValueDecl *D,
- QualType Type, OpenMPClauseKind CKind,
- SourceLocation ELoc,
- bool AcceptIfMutable = true,
- bool ListItemNotVar = false) {
- ASTContext &Context = SemaRef.getASTContext();
- bool IsClassType;
- if (isConstNotMutableType(SemaRef, Type, AcceptIfMutable, &IsClassType)) {
- unsigned Diag = ListItemNotVar
- ? diag::err_omp_const_list_item
- : IsClassType ? diag::err_omp_const_not_mutable_variable
- : diag::err_omp_const_variable;
- SemaRef.Diag(ELoc, Diag) << getOpenMPClauseName(CKind);
- if (!ListItemNotVar && D) {
- const VarDecl *VD = dyn_cast<VarDecl>(D);
- bool IsDecl = !VD || VD->isThisDeclarationADefinition(Context) ==
- VarDecl::DeclarationOnly;
- SemaRef.Diag(D->getLocation(),
- IsDecl ? diag::note_previous_decl : diag::note_defined_here)
- << D;
- }
- return true;
- }
- 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) {
- const_iterator IterTarget =
- std::find_if(begin(), end(), [](const SharingMapTy &Data) {
- return isOpenMPTargetExecutionDirective(Data.Directive);
- });
- if (IterTarget != end()) {
- const_iterator ParentIterTarget = IterTarget + 1;
- for (const_iterator Iter = begin();
- Iter != ParentIterTarget; ++Iter) {
- if (isOpenMPLocal(VD, Iter)) {
- DVar.RefExpr =
- buildDeclRefExpr(SemaRef, VD, D->getType().getNonReferenceType(),
- D->getLocation());
- DVar.CKind = OMPC_threadprivate;
- return DVar;
- }
- }
- if (!isClauseParsingMode() || IterTarget != begin()) {
- 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;
- }
- const_iterator End = end();
- if (!SemaRef.isOpenMPCapturedByRef(
- D, std::distance(ParentIterTarget, End),
- /*OpenMPCaptureLevel=*/0)) {
- 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.
- if (VD && VD->isStaticDataMember()) {
- // Check for explicitly specified attributes.
- const_iterator I = begin();
- const_iterator EndI = end();
- if (FromParent && I != EndI)
- ++I;
- 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;
- }
- DVar.CKind = OMPC_shared;
- return DVar;
- }
- auto &&MatchesAlways = [](OpenMPDirectiveKind) { return true; };
- // The predetermined shared attribute for const-qualified types having no
- // mutable members was removed after OpenMP 3.1.
- if (SemaRef.LangOpts.OpenMP <= 31) {
- // 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.
- if (isConstNotMutableType(SemaRef, D->getType())) {
- // 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 = hasInnermostDSA(
- D,
- [](OpenMPClauseKind C) {
- return C == OMPC_firstprivate || C == OMPC_shared;
- },
- MatchesAlways, FromParent);
- if (DVarTemp.CKind != OMPC_unknown && DVarTemp.RefExpr)
- return DVarTemp;
- DVar.CKind = OMPC_shared;
- return DVar;
- }
- }
- // Explicitly specified attributes and local variables with predetermined
- // attributes.
- const_iterator I = begin();
- const_iterator EndI = end();
- if (FromParent && I != EndI)
- ++I;
- 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()) {
- const_iterator I;
- return getDSA(I, D);
- }
- D = getCanonicalDecl(D);
- const_iterator StartI = begin();
- const_iterator EndI = end();
- if (FromParent && StartI != EndI)
- ++StartI;
- 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);
- const_iterator I = begin();
- const_iterator EndI = end();
- if (FromParent && I != EndI)
- ++I;
- for (; I != EndI; ++I) {
- if (!DPred(I->Directive) &&
- !isImplicitOrExplicitTaskingRegion(I->Directive))
- continue;
- const_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);
- const_iterator StartI = begin();
- const_iterator EndI = end();
- if (FromParent && StartI != EndI)
- ++StartI;
- if (StartI == EndI || !DPred(StartI->Directive))
- return {};
- const_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 (getStackSize() <= Level)
- return false;
- D = getCanonicalDecl(D);
- const SharingMapTy &StackElem = getStackElemAtLevel(Level);
- auto I = StackElem.SharingMap.find(D);
- if (I != StackElem.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 = StackElem.LCVMap.find(D);
- if (LI != StackElem.LCVMap.end())
- return CPred(OMPC_private);
- return false;
- }
- bool DSAStackTy::hasExplicitDirective(
- const llvm::function_ref<bool(OpenMPDirectiveKind)> DPred,
- unsigned Level) const {
- if (getStackSize() <= Level)
- return false;
- const SharingMapTy &StackElem = getStackElemAtLevel(Level);
- return DPred(StackElem.Directive);
- }
- bool DSAStackTy::hasDirective(
- const llvm::function_ref<bool(OpenMPDirectiveKind,
- const DeclarationNameInfo &, SourceLocation)>
- DPred,
- bool FromParent) const {
- // We look only in the enclosing region.
- size_t Skip = FromParent ? 2 : 1;
- for (const_iterator I = begin() + std::min(Skip, getStackSize()), E = end();
- I != E; ++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);
- }
- static bool isOpenMPDeviceDelayedContext(Sema &S) {
- assert(S.LangOpts.OpenMP && S.LangOpts.OpenMPIsDevice &&
- "Expected OpenMP device compilation.");
- return !S.isInOpenMPTargetExecutionDirective() &&
- !S.isInOpenMPDeclareTargetContext();
- }
- namespace {
- /// Status of the function emission on the host/device.
- enum class FunctionEmissionStatus {
- Emitted,
- Discarded,
- Unknown,
- };
- } // anonymous namespace
- /// Do we know that we will eventually codegen the given function?
- static FunctionEmissionStatus isKnownDeviceEmitted(Sema &S, FunctionDecl *FD) {
- assert(S.LangOpts.OpenMP && S.LangOpts.OpenMPIsDevice &&
- "Expected OpenMP device compilation.");
- // Templates are emitted when they're instantiated.
- if (FD->isDependentContext())
- return FunctionEmissionStatus::Discarded;
- Optional<OMPDeclareTargetDeclAttr::DevTypeTy> DevTy =
- OMPDeclareTargetDeclAttr::getDeviceType(FD->getCanonicalDecl());
- if (DevTy.hasValue())
- return (*DevTy == OMPDeclareTargetDeclAttr::DT_Host)
- ? FunctionEmissionStatus::Discarded
- : FunctionEmissionStatus::Emitted;
- // Otherwise, the function is known-emitted if it's in our set of
- // known-emitted functions.
- return (S.DeviceKnownEmittedFns.count(FD) > 0)
- ? FunctionEmissionStatus::Emitted
- : FunctionEmissionStatus::Unknown;
- }
- Sema::DeviceDiagBuilder Sema::diagIfOpenMPDeviceCode(SourceLocation Loc,
- unsigned DiagID) {
- assert(LangOpts.OpenMP && LangOpts.OpenMPIsDevice &&
- "Expected OpenMP device compilation.");
- FunctionEmissionStatus FES =
- isKnownDeviceEmitted(*this, getCurFunctionDecl());
- DeviceDiagBuilder::Kind Kind = DeviceDiagBuilder::K_Nop;
- switch (FES) {
- case FunctionEmissionStatus::Emitted:
- Kind = DeviceDiagBuilder::K_Immediate;
- break;
- case FunctionEmissionStatus::Unknown:
- Kind = isOpenMPDeviceDelayedContext(*this) ? DeviceDiagBuilder::K_Deferred
- : DeviceDiagBuilder::K_Immediate;
- break;
- case FunctionEmissionStatus::Discarded:
- Kind = DeviceDiagBuilder::K_Nop;
- break;
- }
- return DeviceDiagBuilder(Kind, Loc, DiagID, getCurFunctionDecl(), *this);
- }
- /// Do we know that we will eventually codegen the given function?
- static FunctionEmissionStatus isKnownHostEmitted(Sema &S, FunctionDecl *FD) {
- assert(S.LangOpts.OpenMP && !S.LangOpts.OpenMPIsDevice &&
- "Expected OpenMP host compilation.");
- // In OpenMP 4.5 all the functions are host functions.
- if (S.LangOpts.OpenMP <= 45)
- return FunctionEmissionStatus::Emitted;
- Optional<OMPDeclareTargetDeclAttr::DevTypeTy> DevTy =
- OMPDeclareTargetDeclAttr::getDeviceType(FD->getCanonicalDecl());
- if (DevTy.hasValue())
- return (*DevTy == OMPDeclareTargetDeclAttr::DT_NoHost)
- ? FunctionEmissionStatus::Discarded
- : FunctionEmissionStatus::Emitted;
- // Otherwise, the function is known-emitted if it's in our set of
- // known-emitted functions.
- return (S.DeviceKnownEmittedFns.count(FD) > 0)
- ? FunctionEmissionStatus::Emitted
- : FunctionEmissionStatus::Unknown;
- }
- Sema::DeviceDiagBuilder Sema::diagIfOpenMPHostCode(SourceLocation Loc,
- unsigned DiagID) {
- assert(LangOpts.OpenMP && !LangOpts.OpenMPIsDevice &&
- "Expected OpenMP host compilation.");
- FunctionEmissionStatus FES =
- isKnownHostEmitted(*this, getCurFunctionDecl());
- DeviceDiagBuilder::Kind Kind = DeviceDiagBuilder::K_Nop;
- switch (FES) {
- case FunctionEmissionStatus::Emitted:
- Kind = DeviceDiagBuilder::K_Immediate;
- break;
- case FunctionEmissionStatus::Unknown:
- Kind = DeviceDiagBuilder::K_Deferred;
- break;
- case FunctionEmissionStatus::Discarded:
- Kind = DeviceDiagBuilder::K_Nop;
- break;
- }
- return DeviceDiagBuilder(Kind, Loc, DiagID, getCurFunctionDecl(), *this);
- }
- void Sema::checkOpenMPDeviceFunction(SourceLocation Loc, FunctionDecl *Callee,
- bool CheckForDelayedContext) {
- assert(LangOpts.OpenMP && LangOpts.OpenMPIsDevice &&
- "Expected OpenMP device compilation.");
- assert(Callee && "Callee may not be null.");
- Callee = Callee->getMostRecentDecl();
- FunctionDecl *Caller = getCurFunctionDecl();
- // host only function are not available on the device.
- if (Caller &&
- (isKnownDeviceEmitted(*this, Caller) == FunctionEmissionStatus::Emitted ||
- (!isOpenMPDeviceDelayedContext(*this) &&
- isKnownDeviceEmitted(*this, Caller) ==
- FunctionEmissionStatus::Unknown)) &&
- isKnownDeviceEmitted(*this, Callee) ==
- FunctionEmissionStatus::Discarded) {
- StringRef HostDevTy =
- getOpenMPSimpleClauseTypeName(OMPC_device_type, OMPC_DEVICE_TYPE_host);
- Diag(Loc, diag::err_omp_wrong_device_function_call) << HostDevTy << 0;
- Diag(Callee->getAttr<OMPDeclareTargetDeclAttr>()->getLocation(),
- diag::note_omp_marked_device_type_here)
- << HostDevTy;
- return;
- }
- // If the caller is known-emitted, mark the callee as known-emitted.
- // Otherwise, mark the call in our call graph so we can traverse it later.
- if ((CheckForDelayedContext && !isOpenMPDeviceDelayedContext(*this)) ||
- (!Caller && !CheckForDelayedContext) ||
- (Caller &&
- isKnownDeviceEmitted(*this, Caller) == FunctionEmissionStatus::Emitted))
- markKnownEmitted(*this, Caller, Callee, Loc,
- [CheckForDelayedContext](Sema &S, FunctionDecl *FD) {
- return CheckForDelayedContext &&
- isKnownDeviceEmitted(S, FD) ==
- FunctionEmissionStatus::Emitted;
- });
- else if (Caller)
- DeviceCallGraph[Caller].insert({Callee, Loc});
- }
- void Sema::checkOpenMPHostFunction(SourceLocation Loc, FunctionDecl *Callee,
- bool CheckCaller) {
- assert(LangOpts.OpenMP && !LangOpts.OpenMPIsDevice &&
- "Expected OpenMP host compilation.");
- assert(Callee && "Callee may not be null.");
- Callee = Callee->getMostRecentDecl();
- FunctionDecl *Caller = getCurFunctionDecl();
- // device only function are not available on the host.
- if (Caller &&
- isKnownHostEmitted(*this, Caller) == FunctionEmissionStatus::Emitted &&
- isKnownHostEmitted(*this, Callee) == FunctionEmissionStatus::Discarded) {
- StringRef NoHostDevTy = getOpenMPSimpleClauseTypeName(
- OMPC_device_type, OMPC_DEVICE_TYPE_nohost);
- Diag(Loc, diag::err_omp_wrong_device_function_call) << NoHostDevTy << 1;
- Diag(Callee->getAttr<OMPDeclareTargetDeclAttr>()->getLocation(),
- diag::note_omp_marked_device_type_here)
- << NoHostDevTy;
- return;
- }
- // If the caller is known-emitted, mark the callee as known-emitted.
- // Otherwise, mark the call in our call graph so we can traverse it later.
- if ((!CheckCaller && !Caller) ||
- (Caller &&
- isKnownHostEmitted(*this, Caller) == FunctionEmissionStatus::Emitted))
- markKnownEmitted(
- *this, Caller, Callee, Loc, [CheckCaller](Sema &S, FunctionDecl *FD) {
- return CheckCaller &&
- isKnownHostEmitted(S, FD) == FunctionEmissionStatus::Emitted;
- });
- else if (Caller)
- DeviceCallGraph[Caller].insert({Callee, Loc});
- }
- void Sema::checkOpenMPDeviceExpr(const Expr *E) {
- assert(getLangOpts().OpenMP && getLangOpts().OpenMPIsDevice &&
- "OpenMP device compilation mode is expected.");
- QualType Ty = E->getType();
- if ((Ty->isFloat16Type() && !Context.getTargetInfo().hasFloat16Type()) ||
- ((Ty->isFloat128Type() ||
- (Ty->isRealFloatingType() && Context.getTypeSize(Ty) == 128)) &&
- !Context.getTargetInfo().hasFloat128Type()) ||
- (Ty->isIntegerType() && Context.getTypeSize(Ty) == 128 &&
- !Context.getTargetInfo().hasInt128Type()))
- targetDiag(E->getExprLoc(), diag::err_omp_unsupported_type)
- << static_cast<unsigned>(Context.getTypeSize(Ty)) << Ty
- << Context.getTargetInfo().getTriple().str() << E->getSourceRange();
- }
- bool Sema::isOpenMPCapturedByRef(const ValueDecl *D, unsigned Level,
- unsigned OpenMPCaptureLevel) 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();
- bool IsVariableUsedInMapClause = false;
- 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 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 =
- ((IsVariableUsedInMapClause &&
- DSAStack->getCaptureRegion(Level, OpenMPCaptureLevel) ==
- OMPD_target) ||
- !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, bool CheckScopeInfo,
- unsigned StopAt) {
- assert(LangOpts.OpenMP && "OpenMP is not allowed");
- D = getCanonicalDecl(D);
- // If we want to determine whether the variable should be captured from the
- // perspective of the current capturing scope, and we've already left all the
- // capturing scopes of the top directive on the stack, check from the
- // perspective of its parent directive (if any) instead.
- DSAStackTy::ParentDirectiveScope InParentDirectiveRAII(
- *DSAStack, CheckScopeInfo && DSAStack->isBodyComplete());
- // 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() &&
- (getCurCapturedRegion() || getCurBlock() || getCurLambda())) {
- if (isInOpenMPDeclareTargetContext()) {
- // Try to mark variable as declare target if it is used in capturing
- // regions.
- if (LangOpts.OpenMP <= 45 &&
- !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;
- }
- }
- if (CheckScopeInfo) {
- bool OpenMPFound = false;
- for (unsigned I = StopAt + 1; I > 0; --I) {
- FunctionScopeInfo *FSI = FunctionScopes[I - 1];
- if(!isa<CapturingScopeInfo>(FSI))
- return nullptr;
- if (auto *RSI = dyn_cast<CapturedRegionScopeInfo>(FSI))
- if (RSI->CapRegionKind == CR_OpenMP) {
- OpenMPFound = true;
- break;
- }
- }
- if (!OpenMPFound)
- return nullptr;
- }
- if (DSAStack->getCurrentDirective() != OMPD_unknown &&
- (!DSAStack->isClauseParsingMode() ||
- DSAStack->getParentDirective() != OMPD_unknown)) {
- auto &&Info = DSAStack->isLoopControlVariable(D);
- if (Info.first ||
- (VD && VD->hasLocalStorage() &&
- isImplicitOrExplicitTaskingRegion(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());
- // Threadprivate variables must not be captured.
- if (isOpenMPThreadPrivate(DVarPrivate.CKind))
- return nullptr;
- // The variable is not private or it is the variable in the directive with
- // default(none) clause and not used in any clause.
- DVarPrivate = DSAStack->hasDSA(D, isOpenMPPrivate,
- [](OpenMPDirectiveKind) { return true; },
- DSAStack->isClauseParsingMode());
- if (DVarPrivate.CKind != OMPC_unknown ||
- (VD && DSAStack->getDefaultDSA() == DSA_none))
- 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();
- }
- void Sema::startOpenMPCXXRangeFor() {
- assert(LangOpts.OpenMP && "OpenMP must be enabled.");
- if (isOpenMPLoopDirective(DSAStack->getCurrentDirective())) {
- DSAStack->resetPossibleLoopCounter();
- DSAStack->loopStart();
- }
- }
- 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;
- }
- if (const auto *VD = dyn_cast<VarDecl>(D)) {
- if (DSAStack->isThreadPrivate(const_cast<VarDecl *>(VD)) &&
- DSAStack->isForceVarCapturing() &&
- !DSAStack->hasExplicitDSA(
- D, [](OpenMPClauseKind K) { return K == OMPC_copyin; }, Level))
- 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::finalizeOpenMPDelayedAnalysis() {
- assert(LangOpts.OpenMP && "Expected OpenMP compilation mode.");
- // Diagnose implicit declare target functions and their callees.
- for (const auto &CallerCallees : DeviceCallGraph) {
- Optional<OMPDeclareTargetDeclAttr::DevTypeTy> DevTy =
- OMPDeclareTargetDeclAttr::getDeviceType(
- CallerCallees.getFirst()->getMostRecentDecl());
- // Ignore host functions during device analyzis.
- if (LangOpts.OpenMPIsDevice && DevTy &&
- *DevTy == OMPDeclareTargetDeclAttr::DT_Host)
- continue;
- // Ignore nohost functions during host analyzis.
- if (!LangOpts.OpenMPIsDevice && DevTy &&
- *DevTy == OMPDeclareTargetDeclAttr::DT_NoHost)
- continue;
- for (const std::pair<CanonicalDeclPtr<FunctionDecl>, SourceLocation>
- &Callee : CallerCallees.getSecond()) {
- const FunctionDecl *FD = Callee.first->getMostRecentDecl();
- Optional<OMPDeclareTargetDeclAttr::DevTypeTy> DevTy =
- OMPDeclareTargetDeclAttr::getDeviceType(FD);
- if (LangOpts.OpenMPIsDevice && DevTy &&
- *DevTy == OMPDeclareTargetDeclAttr::DT_Host) {
- // Diagnose host function called during device codegen.
- StringRef HostDevTy = getOpenMPSimpleClauseTypeName(
- OMPC_device_type, OMPC_DEVICE_TYPE_host);
- Diag(Callee.second, diag::err_omp_wrong_device_function_call)
- << HostDevTy << 0;
- Diag(FD->getAttr<OMPDeclareTargetDeclAttr>()->getLocation(),
- diag::note_omp_marked_device_type_here)
- << HostDevTy;
- continue;
- }
- if (!LangOpts.OpenMPIsDevice && DevTy &&
- *DevTy == OMPDeclareTargetDeclAttr::DT_NoHost) {
- // Diagnose nohost function called during host codegen.
- StringRef NoHostDevTy = getOpenMPSimpleClauseTypeName(
- OMPC_device_type, OMPC_DEVICE_TYPE_nohost);
- Diag(Callee.second, diag::err_omp_wrong_device_function_call)
- << NoHostDevTy << 1;
- Diag(FD->getAttr<OMPDeclareTargetDeclAttr>()->getLocation(),
- diag::note_omp_marked_device_type_here)
- << NoHostDevTy;
- continue;
- }
- }
- }
- }
- 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);
- }
- static void checkAllocateClauses(Sema &S, DSAStackTy *Stack,
- ArrayRef<OMPClause *> Clauses);
- 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()) {
- PrivateCopies.push_back(nullptr);
- 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);
- }
- }
- Clause->setPrivateCopies(PrivateCopies);
- }
- }
- // Check allocate clauses.
- if (!CurContext->isDependentContext())
- checkAllocateClauses(*this, DSAStack, D->clauses());
- }
- 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;
- }
- std::unique_ptr<CorrectionCandidateCallback> clone() override {
- return std::make_unique<VarDeclFilterCCC>(*this);
- }
- };
- 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) && ND->getKind() == Decl::Var) ||
- isa<FunctionDecl>(ND))) {
- return SemaRef.isDeclInScope(ND, SemaRef.getCurLexicalContext(),
- SemaRef.getCurScope());
- }
- return false;
- }
- std::unique_ptr<CorrectionCandidateCallback> clone() override {
- return std::make_unique<VarOrFuncDeclFilterCCC>(*this);
- }
- };
- } // namespace
- ExprResult Sema::ActOnOpenMPIdExpression(Scope *CurScope,
- CXXScopeSpec &ScopeSpec,
- const DeclarationNameInfo &Id,
- OpenMPDirectiveKind Kind) {
- LookupResult Lookup(*this, Id, LookupOrdinaryName);
- LookupParsedName(Lookup, CurScope, &ScopeSpec, true);
- if (Lookup.isAmbiguous())
- return ExprError();
- VarDecl *VD;
- if (!Lookup.isSingleResult()) {
- VarDeclFilterCCC CCC(*this);
- if (TypoCorrection Corrected =
- CorrectTypo(Id, LookupOrdinaryName, CurScope, nullptr, CCC,
- 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 (Kind == OMPD_threadprivate && !VD->hasGlobalStorage()) {
- Diag(Id.getLoc(), diag::err_omp_global_var_arg)
- << getOpenMPDirectiveName(Kind) << !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(Kind) << 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(Kind) << 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(Kind) << 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->isLocalVarDecl() && CurScope &&
- !isDeclInScope(ND, getCurLexicalContext(), CurScope)) {
- Diag(Id.getLoc(), diag::err_omp_var_scope)
- << getOpenMPDirectiveName(Kind) << 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 (Kind == OMPD_threadprivate && VD->isUsed() &&
- !DSAStack->isThreadPrivate(VD)) {
- Diag(Id.getLoc(), diag::err_omp_var_used)
- << getOpenMPDirectiveName(Kind) << 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;
- }
- static OMPAllocateDeclAttr::AllocatorTypeTy
- getAllocatorKind(Sema &S, DSAStackTy *Stack, Expr *Allocator) {
- if (!Allocator)
- return OMPAllocateDeclAttr::OMPDefaultMemAlloc;
- if (Allocator->isTypeDependent() || Allocator->isValueDependent() ||
- Allocator->isInstantiationDependent() ||
- Allocator->containsUnexpandedParameterPack())
- return OMPAllocateDeclAttr::OMPUserDefinedMemAlloc;
- auto AllocatorKindRes = OMPAllocateDeclAttr::OMPUserDefinedMemAlloc;
- const Expr *AE = Allocator->IgnoreParenImpCasts();
- for (int I = OMPAllocateDeclAttr::OMPDefaultMemAlloc;
- I < OMPAllocateDeclAttr::OMPUserDefinedMemAlloc; ++I) {
- auto AllocatorKind = static_cast<OMPAllocateDeclAttr::AllocatorTypeTy>(I);
- const Expr *DefAllocator = Stack->getAllocator(AllocatorKind);
- llvm::FoldingSetNodeID AEId, DAEId;
- AE->Profile(AEId, S.getASTContext(), /*Canonical=*/true);
- DefAllocator->Profile(DAEId, S.getASTContext(), /*Canonical=*/true);
- if (AEId == DAEId) {
- AllocatorKindRes = AllocatorKind;
- break;
- }
- }
- return AllocatorKindRes;
- }
- static bool checkPreviousOMPAllocateAttribute(
- Sema &S, DSAStackTy *Stack, Expr *RefExpr, VarDecl *VD,
- OMPAllocateDeclAttr::AllocatorTypeTy AllocatorKind, Expr *Allocator) {
- if (!VD->hasAttr<OMPAllocateDeclAttr>())
- return false;
- const auto *A = VD->getAttr<OMPAllocateDeclAttr>();
- Expr *PrevAllocator = A->getAllocator();
- OMPAllocateDeclAttr::AllocatorTypeTy PrevAllocatorKind =
- getAllocatorKind(S, Stack, PrevAllocator);
- bool AllocatorsMatch = AllocatorKind == PrevAllocatorKind;
- if (AllocatorsMatch &&
- AllocatorKind == OMPAllocateDeclAttr::OMPUserDefinedMemAlloc &&
- Allocator && PrevAllocator) {
- const Expr *AE = Allocator->IgnoreParenImpCasts();
- const Expr *PAE = PrevAllocator->IgnoreParenImpCasts();
- llvm::FoldingSetNodeID AEId, PAEId;
- AE->Profile(AEId, S.Context, /*Canonical=*/true);
- PAE->Profile(PAEId, S.Context, /*Canonical=*/true);
- AllocatorsMatch = AEId == PAEId;
- }
- if (!AllocatorsMatch) {
- SmallString<256> AllocatorBuffer;
- llvm::raw_svector_ostream AllocatorStream(AllocatorBuffer);
- if (Allocator)
- Allocator->printPretty(AllocatorStream, nullptr, S.getPrintingPolicy());
- SmallString<256> PrevAllocatorBuffer;
- llvm::raw_svector_ostream PrevAllocatorStream(PrevAllocatorBuffer);
- if (PrevAllocator)
- PrevAllocator->printPretty(PrevAllocatorStream, nullptr,
- S.getPrintingPolicy());
- SourceLocation AllocatorLoc =
- Allocator ? Allocator->getExprLoc() : RefExpr->getExprLoc();
- SourceRange AllocatorRange =
- Allocator ? Allocator->getSourceRange() : RefExpr->getSourceRange();
- SourceLocation PrevAllocatorLoc =
- PrevAllocator ? PrevAllocator->getExprLoc() : A->getLocation();
- SourceRange PrevAllocatorRange =
- PrevAllocator ? PrevAllocator->getSourceRange() : A->getRange();
- S.Diag(AllocatorLoc, diag::warn_omp_used_different_allocator)
- << (Allocator ? 1 : 0) << AllocatorStream.str()
- << (PrevAllocator ? 1 : 0) << PrevAllocatorStream.str()
- << AllocatorRange;
- S.Diag(PrevAllocatorLoc, diag::note_omp_previous_allocator)
- << PrevAllocatorRange;
- return true;
- }
- return false;
- }
- static void
- applyOMPAllocateAttribute(Sema &S, VarDecl *VD,
- OMPAllocateDeclAttr::AllocatorTypeTy AllocatorKind,
- Expr *Allocator, SourceRange SR) {
- if (VD->hasAttr<OMPAllocateDeclAttr>())
- return;
- if (Allocator &&
- (Allocator->isTypeDependent() || Allocator->isValueDependent() ||
- Allocator->isInstantiationDependent() ||
- Allocator->containsUnexpandedParameterPack()))
- return;
- auto *A = OMPAllocateDeclAttr::CreateImplicit(S.Context, AllocatorKind,
- Allocator, SR);
- VD->addAttr(A);
- if (ASTMutationListener *ML = S.Context.getASTMutationListener())
- ML->DeclarationMarkedOpenMPAllocate(VD, A);
- }
- Sema::DeclGroupPtrTy Sema::ActOnOpenMPAllocateDirective(
- SourceLocation Loc, ArrayRef<Expr *> VarList,
- ArrayRef<OMPClause *> Clauses, DeclContext *Owner) {
- assert(Clauses.size() <= 1 && "Expected at most one clause.");
- Expr *Allocator = nullptr;
- if (Clauses.empty()) {
- // OpenMP 5.0, 2.11.3 allocate Directive, Restrictions.
- // allocate directives that appear in a target region must specify an
- // allocator clause unless a requires directive with the dynamic_allocators
- // clause is present in the same compilation unit.
- if (LangOpts.OpenMPIsDevice &&
- !DSAStack->hasRequiresDeclWithClause<OMPDynamicAllocatorsClause>())
- targetDiag(Loc, diag::err_expected_allocator_clause);
- } else {
- Allocator = cast<OMPAllocatorClause>(Clauses.back())->getAllocator();
- }
- OMPAllocateDeclAttr::AllocatorTypeTy AllocatorKind =
- getAllocatorKind(*this, DSAStack, Allocator);
- SmallVector<Expr *, 8> Vars;
- for (Expr *RefExpr : VarList) {
- auto *DE = cast<DeclRefExpr>(RefExpr);
- auto *VD = cast<VarDecl>(DE->getDecl());
- // Check if this is a TLS variable or global register.
- if (VD->getTLSKind() != VarDecl::TLS_None ||
- VD->hasAttr<OMPThreadPrivateDeclAttr>() ||
- (VD->getStorageClass() == SC_Register && VD->hasAttr<AsmLabelAttr>() &&
- !VD->isLocalVarDecl()))
- continue;
- // If the used several times in the allocate directive, the same allocator
- // must be used.
- if (checkPreviousOMPAllocateAttribute(*this, DSAStack, RefExpr, VD,
- AllocatorKind, Allocator))
- continue;
- // OpenMP, 2.11.3 allocate Directive, Restrictions, C / C++
- // If a list item has a static storage type, the allocator expression in the
- // allocator clause must be a constant expression that evaluates to one of
- // the predefined memory allocator values.
- if (Allocator && VD->hasGlobalStorage()) {
- if (AllocatorKind == OMPAllocateDeclAttr::OMPUserDefinedMemAlloc) {
- Diag(Allocator->getExprLoc(),
- diag::err_omp_expected_predefined_allocator)
- << Allocator->getSourceRange();
- bool IsDecl = VD->isThisDeclarationADefinition(Context) ==
- VarDecl::DeclarationOnly;
- Diag(VD->getLocation(),
- IsDecl ? diag::note_previous_decl : diag::note_defined_here)
- << VD;
- continue;
- }
- }
- Vars.push_back(RefExpr);
- applyOMPAllocateAttribute(*this, VD, AllocatorKind, Allocator,
- DE->getSourceRange());
- }
- if (Vars.empty())
- return nullptr;
- if (!Owner)
- Owner = getCurLexicalContext();
- auto *D = OMPAllocateDecl::Create(Context, Owner, Loc, Vars, Clauses);
- D->setAccess(AS_public);
- Owner->addDecl(D);
- return DeclGroupPtrTy::make(DeclGroupRef(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) {
- /// For target specific clauses, the requires directive cannot be
- /// specified after the handling of any of the target regions in the
- /// current compilation unit.
- ArrayRef<SourceLocation> TargetLocations =
- DSAStack->getEncounteredTargetLocs();
- if (!TargetLocations.empty()) {
- for (const OMPClause *CNew : ClauseList) {
- // Check if any of the requires clauses affect target regions.
- if (isa<OMPUnifiedSharedMemoryClause>(CNew) ||
- isa<OMPUnifiedAddressClause>(CNew) ||
- isa<OMPReverseOffloadClause>(CNew) ||
- isa<OMPDynamicAllocatorsClause>(CNew)) {
- Diag(Loc, diag::err_omp_target_before_requires)
- << getOpenMPClauseName(CNew->getClauseKind());
- for (SourceLocation TargetLoc : TargetLocations) {
- Diag(TargetLoc, diag::note_omp_requires_encountered_target);
- }
- }
- }
- }
- 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;
- visitSubCaptures(S->getInnermostCapturedStmt());
- }
- public:
- void VisitDeclRefExpr(DeclRefExpr *E) {
- if (E->isTypeDependent() || E->isValueDependent() ||
- E->containsUnexpandedParameterPack() || E->isInstantiationDependent())
- return;
- if (auto *VD = dyn_cast<VarDecl>(E->getDecl())) {
- // Check the datasharing rules for the expressions in the clauses.
- if (!CS) {
- if (auto *CED = dyn_cast<OMPCapturedExprDecl>(VD))
- if (!CED->hasAttr<OMPCaptureNoInitAttr>()) {
- Visit(CED->getInit());
- return;
- }
- } else if (VD->isImplicit() || isa<OMPCapturedExprDecl>(VD))
- // Do not analyze internal variables and do not enclose them into
- // implicit clauses.
- return;
- VD = VD->getCanonicalDecl();
- // Skip internally declared variables.
- if (VD->hasLocalStorage() && CS && !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 && !CS->capturesVariable(VD) &&
- (Stack->hasRequiresDeclWithClause<OMPUnifiedSharedMemoryClause>() ||
- !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 &&
- isImplicitOrExplicitTaskingRegion(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);
- return;
- }
- // Store implicitly used globals with declare target link for parent
- // target.
- if (!isOpenMPTargetExecutionDirective(DKind) && Res &&
- *Res == OMPDeclareTargetDeclAttr::MT_Link) {
- Stack->addToParentTargetRegionLinkGlobals(E);
- return;
- }
- }
- }
- 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 (auto *TE = dyn_cast<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;
- // Check to see if the member expression is referencing a class that
- // has already been explicitly mapped
- if (Stack->isClassPreviouslyMapped(TE->getType()))
- 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) {
- // Check implicitly captured variables in the task-based directives to
- // check if they must be firstprivatized.
- Visit(C);
- }
- }
- }
- void visitSubCaptures(CapturedStmt *S) {
- for (const CapturedStmt::Capture &Cap : S->captures()) {
- if (!Cap.capturesVariable() && !Cap.capturesVariableByCopy())
- 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);
- }
- }
- 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) {
- // Process declare target link variables for the target directives.
- if (isOpenMPTargetExecutionDirective(S->getCurrentDirective())) {
- for (DeclRefExpr *E : Stack->getLinkGlobals())
- Visit(E);
- }
- }
- };
- } // 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, /*OpenMPCaptureLevel=*/0);
- // Mark this captured region as inlined, because we don't use outlined
- // function directly.
- getCurCapturedRegion()->TheCapturedDecl->addAttr(
- AlwaysInlineAttr::CreateImplicit(
- Context, {}, AttributeCommonInfo::AS_Keyword,
- 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, /*OpenMPCaptureLevel=*/1);
- 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, /*OpenMPCaptureLevel=*/2);
- 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, /*OpenMPCaptureLevel=*/0);
- // Mark this captured region as inlined, because we don't use outlined
- // function directly.
- getCurCapturedRegion()->TheCapturedDecl->addAttr(
- AlwaysInlineAttr::CreateImplicit(
- Context, {}, AttributeCommonInfo::AS_Keyword,
- AlwaysInlineAttr::Keyword_forceinline));
- ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
- std::make_pair(StringRef(), QualType()),
- /*OpenMPCaptureLevel=*/1);
- 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, {}, AttributeCommonInfo::AS_Keyword,
- 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, {}, AttributeCommonInfo::AS_Keyword,
- 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, /*OpenMPCaptureLevel=*/0);
- // Mark this captured region as inlined, because we don't use outlined
- // function directly.
- getCurCapturedRegion()->TheCapturedDecl->addAttr(
- AlwaysInlineAttr::CreateImplicit(
- Context, {}, AttributeCommonInfo::AS_Keyword,
- 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, /*OpenMPCaptureLevel=*/1);
- 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, /*OpenMPCaptureLevel=*/2);
- 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, /*OpenMPCaptureLevel=*/3);
- 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, /*OpenMPCaptureLevel=*/0);
- 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, /*OpenMPCaptureLevel=*/1);
- 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, {}, AttributeCommonInfo::AS_Keyword,
- AlwaysInlineAttr::Keyword_forceinline));
- break;
- }
- case OMPD_threadprivate:
- case OMPD_allocate:
- 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_mapper:
- case OMPD_declare_simd:
- case OMPD_declare_target:
- case OMPD_end_declare_target:
- case OMPD_requires:
- case OMPD_declare_variant:
- 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
- void Sema::tryCaptureOpenMPLambdas(ValueDecl *V) {
- // Capture variables captured by reference in lambdas for target-based
- // directives.
- if (!CurContext->isDependentContext() &&
- (isOpenMPTargetExecutionDirective(DSAStack->getCurrentDirective()) ||
- isOpenMPTargetDataManagementDirective(
- DSAStack->getCurrentDirective()))) {
- QualType Type = V->getType();
- if (const auto *RD = Type.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;
- MarkVariableReferenced(LC.getLocation(), VD);
- } else if (LC.getCaptureKind() == LCK_This) {
- QualType ThisTy = getCurrentThisType();
- if (!ThisTy.isNull() &&
- Context.typesAreCompatible(ThisTy, ThisCapture->getType()))
- CheckCXXThisCapture(LC.getLocation());
- }
- }
- }
- DSAStack->setForceCaptureByReferenceInTargetExecutable(
- SavedForceCaptureByReferenceInTargetExecutable);
- }
- }
- }
- 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;
- unsigned CompletedRegions = 0;
- 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));
- }
- }
- }
- }
- if (++CompletedRegions == CaptureRegions.size())
- DSAStack->setBodyComplete();
- 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 and cancellation constructs) to be
- // orphaned (they could be used in functions, called from OpenMP regions
- // with the required preconditions).
- if (ParentRegion == OMPD_unknown &&
- !isOpenMPNestingTeamsDirective(CurrentRegion) &&
- CurrentRegion != OMPD_cancellation_point &&
- CurrentRegion != OMPD_cancel)
- 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)));
- OrphanSeen = ParentRegion == OMPD_unknown;
- } 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 =
- (SemaRef.LangOpts.OpenMP <= 45 && ParentRegion != OMPD_target) ||
- (SemaRef.LangOpts.OpenMP >= 50 && ParentRegion != OMPD_unknown &&
- 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;
- }
- 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);
- }
- static void checkAllocateClauses(Sema &S, DSAStackTy *Stack,
- ArrayRef<OMPClause *> Clauses) {
- assert(!S.CurContext->isDependentContext() &&
- "Expected non-dependent context.");
- auto AllocateRange =
- llvm::make_filter_range(Clauses, OMPAllocateClause::classof);
- llvm::DenseMap<CanonicalDeclPtr<Decl>, CanonicalDeclPtr<VarDecl>>
- DeclToCopy;
- auto PrivateRange = llvm::make_filter_range(Clauses, [](const OMPClause *C) {
- return isOpenMPPrivate(C->getClauseKind());
- });
- for (OMPClause *Cl : PrivateRange) {
- MutableArrayRef<Expr *>::iterator I, It, Et;
- if (Cl->getClauseKind() == OMPC_private) {
- auto *PC = cast<OMPPrivateClause>(Cl);
- I = PC->private_copies().begin();
- It = PC->varlist_begin();
- Et = PC->varlist_end();
- } else if (Cl->getClauseKind() == OMPC_firstprivate) {
- auto *PC = cast<OMPFirstprivateClause>(Cl);
- I = PC->private_copies().begin();
- It = PC->varlist_begin();
- Et = PC->varlist_end();
- } else if (Cl->getClauseKind() == OMPC_lastprivate) {
- auto *PC = cast<OMPLastprivateClause>(Cl);
- I = PC->private_copies().begin();
- It = PC->varlist_begin();
- Et = PC->varlist_end();
- } else if (Cl->getClauseKind() == OMPC_linear) {
- auto *PC = cast<OMPLinearClause>(Cl);
- I = PC->privates().begin();
- It = PC->varlist_begin();
- Et = PC->varlist_end();
- } else if (Cl->getClauseKind() == OMPC_reduction) {
- auto *PC = cast<OMPReductionClause>(Cl);
- I = PC->privates().begin();
- It = PC->varlist_begin();
- Et = PC->varlist_end();
- } else if (Cl->getClauseKind() == OMPC_task_reduction) {
- auto *PC = cast<OMPTaskReductionClause>(Cl);
- I = PC->privates().begin();
- It = PC->varlist_begin();
- Et = PC->varlist_end();
- } else if (Cl->getClauseKind() == OMPC_in_reduction) {
- auto *PC = cast<OMPInReductionClause>(Cl);
- I = PC->privates().begin();
- It = PC->varlist_begin();
- Et = PC->varlist_end();
- } else {
- llvm_unreachable("Expected private clause.");
- }
- for (Expr *E : llvm::make_range(It, Et)) {
- if (!*I) {
- ++I;
- continue;
- }
- SourceLocation ELoc;
- SourceRange ERange;
- Expr *SimpleRefExpr = E;
- auto Res = getPrivateItem(S, SimpleRefExpr, ELoc, ERange,
- /*AllowArraySection=*/true);
- DeclToCopy.try_emplace(Res.first,
- cast<VarDecl>(cast<DeclRefExpr>(*I)->getDecl()));
- ++I;
- }
- }
- for (OMPClause *C : AllocateRange) {
- auto *AC = cast<OMPAllocateClause>(C);
- OMPAllocateDeclAttr::AllocatorTypeTy AllocatorKind =
- getAllocatorKind(S, Stack, AC->getAllocator());
- // OpenMP, 2.11.4 allocate Clause, Restrictions.
- // For task, taskloop or target directives, allocation requests to memory
- // allocators with the trait access set to thread result in unspecified
- // behavior.
- if (AllocatorKind == OMPAllocateDeclAttr::OMPThreadMemAlloc &&
- (isOpenMPTaskingDirective(Stack->getCurrentDirective()) ||
- isOpenMPTargetExecutionDirective(Stack->getCurrentDirective()))) {
- S.Diag(AC->getAllocator()->getExprLoc(),
- diag::warn_omp_allocate_thread_on_task_target_directive)
- << getOpenMPDirectiveName(Stack->getCurrentDirective());
- }
- for (Expr *E : AC->varlists()) {
- SourceLocation ELoc;
- SourceRange ERange;
- Expr *SimpleRefExpr = E;
- auto Res = getPrivateItem(S, SimpleRefExpr, ELoc, ERange);
- ValueDecl *VD = Res.first;
- DSAStackTy::DSAVarData Data = Stack->getTopDSA(VD, /*FromParent=*/false);
- if (!isOpenMPPrivate(Data.CKind)) {
- S.Diag(E->getExprLoc(),
- diag::err_omp_expected_private_copy_for_allocate);
- continue;
- }
- VarDecl *PrivateVD = DeclToCopy[VD];
- if (checkPreviousOMPAllocateAttribute(S, Stack, E, PrivateVD,
- AllocatorKind, AC->getAllocator()))
- continue;
- applyOMPAllocateAttribute(S, PrivateVD, AllocatorKind, AC->getAllocator(),
- E->getSourceRange());
- }
- }
- }
- 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 (!isOpenMPTargetDataManagementDirective(Kind) &&
- !isOpenMPTaskingDirective(Kind)) {
- // Visit subcaptures to generate implicit clauses for captured vars.
- auto *CS = cast<CapturedStmt>(AStmt);
- SmallVector<OpenMPDirectiveKind, 4> CaptureRegions;
- getOpenMPCaptureRegions(CaptureRegions, Kind);
- // Ignore outer tasking regions for target directives.
- if (CaptureRegions.size() > 1 && CaptureRegions.front() == OMPD_task)
- CS = cast<CapturedStmt>(CS->getCapturedStmt());
- DSAChecker.visitSubCaptures(CS);
- }
- 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()) {
- CXXScopeSpec MapperIdScopeSpec;
- DeclarationNameInfo MapperId;
- if (OMPClause *Implicit = ActOnOpenMPMapClause(
- llvm::None, llvm::None, MapperIdScopeSpec, MapperId,
- OMPC_MAP_tofrom, /*IsMapTypeImplicit=*/true, SourceLocation(),
- SourceLocation(), ImplicitMaps, OMPVarListLocTy())) {
- 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_allocate:
- case OMPD_declare_reduction:
- case OMPD_declare_mapper:
- case OMPD_declare_simd:
- case OMPD_requires:
- case OMPD_declare_variant:
- llvm_unreachable("OpenMP Directive is not allowed");
- case OMPD_unknown:
- llvm_unreachable("Unknown OpenMP directive");
- }
- ErrorFound = Res.isInvalid() || ErrorFound;
- // Check variables in the clauses if default(none) was specified.
- if (DSAStack->getDefaultDSA() == DSA_none) {
- DSAAttrChecker DSAChecker(DSAStack, *this, nullptr);
- for (OMPClause *C : Clauses) {
- switch (C->getClauseKind()) {
- case OMPC_num_threads:
- case OMPC_dist_schedule:
- // Do not analyse if no parent teams directive.
- if (isOpenMPTeamsDirective(DSAStack->getCurrentDirective()))
- break;
- continue;
- case OMPC_if:
- if (isOpenMPTeamsDirective(DSAStack->getCurrentDirective()) &&
- cast<OMPIfClause>(C)->getNameModifier() != OMPD_target)
- break;
- continue;
- case OMPC_schedule:
- break;
- case OMPC_ordered:
- case OMPC_device:
- case OMPC_num_teams:
- case OMPC_thread_limit:
- case OMPC_priority:
- case OMPC_grainsize:
- case OMPC_num_tasks:
- case OMPC_hint:
- case OMPC_collapse:
- case OMPC_safelen:
- case OMPC_simdlen:
- case OMPC_final:
- 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_nowait:
- case OMPC_untied:
- case OMPC_mergeable:
- case OMPC_allocate:
- 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_defaultmap:
- case OMPC_to:
- case OMPC_from:
- case OMPC_use_device_ptr:
- case OMPC_is_device_ptr:
- continue;
- case OMPC_allocator:
- case OMPC_flush:
- case OMPC_threadprivate:
- case OMPC_uniform:
- case OMPC_unknown:
- case OMPC_unified_address:
- case OMPC_unified_shared_memory:
- case OMPC_reverse_offload:
- case OMPC_dynamic_allocators:
- case OMPC_atomic_default_mem_order:
- case OMPC_device_type:
- case OMPC_match:
- llvm_unreachable("Unexpected clause");
- }
- for (Stmt *CC : C->children()) {
- if (CC)
- DSAChecker.Visit(CC);
- }
- }
- for (auto &P : DSAChecker.getVarsWithInheritedDSA())
- VarsWithInheritedDSA[P.getFirst()] = P.getSecond();
- }
- for (const auto &P : VarsWithInheritedDSA) {
- if (P.getFirst()->isImplicit() || isa<OMPCapturedExprDecl>(P.getFirst()))
- continue;
- ErrorFound = true;
- Diag(P.second->getExprLoc(), diag::err_omp_no_dsa_for_variable)
- << P.first << P.second->getSourceRange();
- Diag(DSAStack->getDefaultDSALocation(), diag::note_omp_default_dsa_none);
- }
- if (!AllowedNameModifiers.empty())
- ErrorFound = checkIfClauses(*this, Kind, Clauses, AllowedNameModifiers) ||
- ErrorFound;
- if (ErrorFound)
- return StmtError();
- if (!(Res.getAs<OMPExecutableDirective>()->isStandaloneDirective())) {
- Res.getAs<OMPExecutableDirective>()
- ->getStructuredBlock()
- ->setIsOMPStructuredBlock(true);
- }
- if (!CurContext->isDependentContext() &&
- isOpenMPTargetExecutionDirective(Kind) &&
- !(DSAStack->hasRequiresDeclWithClause<OMPUnifiedSharedMemoryClause>() ||
- DSAStack->hasRequiresDeclWithClause<OMPUnifiedAddressClause>() ||
- DSAStack->hasRequiresDeclWithClause<OMPReverseOffloadClause>() ||
- DSAStack->hasRequiresDeclWithClause<OMPDynamicAllocatorsClause>())) {
- // Register target to DSA Stack.
- DSAStack->addTargetDirLocation(StartLoc);
- }
- 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();
- const int SimdId = 0;
- if (!DG.get().isSingleDecl()) {
- Diag(SR.getBegin(), diag::err_omp_single_decl_in_declare_simd_variant)
- << SimdId;
- 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) << SimdId;
- 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 DG;
- }
- Optional<std::pair<FunctionDecl *, Expr *>>
- Sema::checkOpenMPDeclareVariantFunction(Sema::DeclGroupPtrTy DG,
- Expr *VariantRef, SourceRange SR) {
- if (!DG || DG.get().isNull())
- return None;
- const int VariantId = 1;
- // Must be applied only to single decl.
- if (!DG.get().isSingleDecl()) {
- Diag(SR.getBegin(), diag::err_omp_single_decl_in_declare_simd_variant)
- << VariantId << SR;
- return None;
- }
- Decl *ADecl = DG.get().getSingleDecl();
- if (auto *FTD = dyn_cast<FunctionTemplateDecl>(ADecl))
- ADecl = FTD->getTemplatedDecl();
- // Decl must be a function.
- auto *FD = dyn_cast<FunctionDecl>(ADecl);
- if (!FD) {
- Diag(ADecl->getLocation(), diag::err_omp_function_expected)
- << VariantId << SR;
- return None;
- }
- auto &&HasMultiVersionAttributes = [](const FunctionDecl *FD) {
- return FD->hasAttrs() &&
- (FD->hasAttr<CPUDispatchAttr>() || FD->hasAttr<CPUSpecificAttr>() ||
- FD->hasAttr<TargetAttr>());
- };
- // OpenMP is not compatible with CPU-specific attributes.
- if (HasMultiVersionAttributes(FD)) {
- Diag(FD->getLocation(), diag::err_omp_declare_variant_incompat_attributes)
- << SR;
- return None;
- }
- // Allow #pragma omp declare variant only if the function is not used.
- if (FD->isUsed(false))
- Diag(SR.getBegin(), diag::warn_omp_declare_variant_after_used)
- << FD->getLocation();
- // Check if the function was emitted already.
- const FunctionDecl *Definition;
- if (!FD->isThisDeclarationADefinition() && FD->isDefined(Definition) &&
- (LangOpts.EmitAllDecls || Context.DeclMustBeEmitted(Definition)))
- Diag(SR.getBegin(), diag::warn_omp_declare_variant_after_emitted)
- << FD->getLocation();
- // The VariantRef must point to function.
- if (!VariantRef) {
- Diag(SR.getBegin(), diag::err_omp_function_expected) << VariantId;
- return None;
- }
- // Do not check templates, wait until instantiation.
- if (VariantRef->isTypeDependent() || VariantRef->isValueDependent() ||
- VariantRef->containsUnexpandedParameterPack() ||
- VariantRef->isInstantiationDependent() || FD->isDependentContext())
- return std::make_pair(FD, VariantRef);
- // Convert VariantRef expression to the type of the original function to
- // resolve possible conflicts.
- ExprResult VariantRefCast;
- if (LangOpts.CPlusPlus) {
- QualType FnPtrType;
- auto *Method = dyn_cast<CXXMethodDecl>(FD);
- if (Method && !Method->isStatic()) {
- const Type *ClassType =
- Context.getTypeDeclType(Method->getParent()).getTypePtr();
- FnPtrType = Context.getMemberPointerType(FD->getType(), ClassType);
- ExprResult ER;
- {
- // Build adrr_of unary op to correctly handle type checks for member
- // functions.
- Sema::TentativeAnalysisScope Trap(*this);
- ER = CreateBuiltinUnaryOp(VariantRef->getBeginLoc(), UO_AddrOf,
- VariantRef);
- }
- if (!ER.isUsable()) {
- Diag(VariantRef->getExprLoc(), diag::err_omp_function_expected)
- << VariantId << VariantRef->getSourceRange();
- return None;
- }
- VariantRef = ER.get();
- } else {
- FnPtrType = Context.getPointerType(FD->getType());
- }
- ImplicitConversionSequence ICS =
- TryImplicitConversion(VariantRef, FnPtrType.getUnqualifiedType(),
- /*SuppressUserConversions=*/false,
- /*AllowExplicit=*/false,
- /*InOverloadResolution=*/false,
- /*CStyle=*/false,
- /*AllowObjCWritebackConversion=*/false);
- if (ICS.isFailure()) {
- Diag(VariantRef->getExprLoc(),
- diag::err_omp_declare_variant_incompat_types)
- << VariantRef->getType() << FnPtrType << VariantRef->getSourceRange();
- return None;
- }
- VariantRefCast = PerformImplicitConversion(
- VariantRef, FnPtrType.getUnqualifiedType(), AA_Converting);
- if (!VariantRefCast.isUsable())
- return None;
- // Drop previously built artificial addr_of unary op for member functions.
- if (Method && !Method->isStatic()) {
- Expr *PossibleAddrOfVariantRef = VariantRefCast.get();
- if (auto *UO = dyn_cast<UnaryOperator>(
- PossibleAddrOfVariantRef->IgnoreImplicit()))
- VariantRefCast = UO->getSubExpr();
- }
- } else {
- VariantRefCast = VariantRef;
- }
- ExprResult ER = CheckPlaceholderExpr(VariantRefCast.get());
- if (!ER.isUsable() ||
- !ER.get()->IgnoreParenImpCasts()->getType()->isFunctionType()) {
- Diag(VariantRef->getExprLoc(), diag::err_omp_function_expected)
- << VariantId << VariantRef->getSourceRange();
- return None;
- }
- // The VariantRef must point to function.
- auto *DRE = dyn_cast<DeclRefExpr>(ER.get()->IgnoreParenImpCasts());
- if (!DRE) {
- Diag(VariantRef->getExprLoc(), diag::err_omp_function_expected)
- << VariantId << VariantRef->getSourceRange();
- return None;
- }
- auto *NewFD = dyn_cast_or_null<FunctionDecl>(DRE->getDecl());
- if (!NewFD) {
- Diag(VariantRef->getExprLoc(), diag::err_omp_function_expected)
- << VariantId << VariantRef->getSourceRange();
- return None;
- }
- // Check if variant function is not marked with declare variant directive.
- if (NewFD->hasAttrs() && NewFD->hasAttr<OMPDeclareVariantAttr>()) {
- Diag(VariantRef->getExprLoc(),
- diag::warn_omp_declare_variant_marked_as_declare_variant)
- << VariantRef->getSourceRange();
- SourceRange SR =
- NewFD->specific_attr_begin<OMPDeclareVariantAttr>()->getRange();
- Diag(SR.getBegin(), diag::note_omp_marked_declare_variant_here) << SR;
- return None;
- }
- enum DoesntSupport {
- VirtFuncs = 1,
- Constructors = 3,
- Destructors = 4,
- DeletedFuncs = 5,
- DefaultedFuncs = 6,
- ConstexprFuncs = 7,
- ConstevalFuncs = 8,
- };
- if (const auto *CXXFD = dyn_cast<CXXMethodDecl>(FD)) {
- if (CXXFD->isVirtual()) {
- Diag(FD->getLocation(), diag::err_omp_declare_variant_doesnt_support)
- << VirtFuncs;
- return None;
- }
- if (isa<CXXConstructorDecl>(FD)) {
- Diag(FD->getLocation(), diag::err_omp_declare_variant_doesnt_support)
- << Constructors;
- return None;
- }
- if (isa<CXXDestructorDecl>(FD)) {
- Diag(FD->getLocation(), diag::err_omp_declare_variant_doesnt_support)
- << Destructors;
- return None;
- }
- }
- if (FD->isDeleted()) {
- Diag(FD->getLocation(), diag::err_omp_declare_variant_doesnt_support)
- << DeletedFuncs;
- return None;
- }
- if (FD->isDefaulted()) {
- Diag(FD->getLocation(), diag::err_omp_declare_variant_doesnt_support)
- << DefaultedFuncs;
- return None;
- }
- if (FD->isConstexpr()) {
- Diag(FD->getLocation(), diag::err_omp_declare_variant_doesnt_support)
- << (NewFD->isConsteval() ? ConstevalFuncs : ConstexprFuncs);
- return None;
- }
- // Check general compatibility.
- if (areMultiversionVariantFunctionsCompatible(
- FD, NewFD, PDiag(diag::err_omp_declare_variant_noproto),
- PartialDiagnosticAt(
- SR.getBegin(),
- PDiag(diag::note_omp_declare_variant_specified_here) << SR),
- PartialDiagnosticAt(
- VariantRef->getExprLoc(),
- PDiag(diag::err_omp_declare_variant_doesnt_support)),
- PartialDiagnosticAt(VariantRef->getExprLoc(),
- PDiag(diag::err_omp_declare_variant_diff)
- << FD->getLocation()),
- /*TemplatesSupported=*/true, /*ConstexprSupported=*/false,
- /*CLinkageMayDiffer=*/true))
- return None;
- return std::make_pair(FD, cast<Expr>(DRE));
- }
- void Sema::ActOnOpenMPDeclareVariantDirective(
- FunctionDecl *FD, Expr *VariantRef, SourceRange SR,
- const Sema::OpenMPDeclareVariantCtsSelectorData &Data) {
- if (Data.CtxSet == OMPDeclareVariantAttr::CtxSetUnknown ||
- Data.Ctx == OMPDeclareVariantAttr::CtxUnknown)
- return;
- Expr *Score = nullptr;
- OMPDeclareVariantAttr::ScoreType ST = OMPDeclareVariantAttr::ScoreUnknown;
- if (Data.CtxScore.isUsable()) {
- ST = OMPDeclareVariantAttr::ScoreSpecified;
- Score = Data.CtxScore.get();
- if (!Score->isTypeDependent() && !Score->isValueDependent() &&
- !Score->isInstantiationDependent() &&
- !Score->containsUnexpandedParameterPack()) {
- llvm::APSInt Result;
- ExprResult ICE = VerifyIntegerConstantExpression(Score, &Result);
- if (ICE.isInvalid())
- return;
- }
- }
- auto *NewAttr = OMPDeclareVariantAttr::CreateImplicit(
- Context, VariantRef, Score, Data.CtxSet, ST, Data.Ctx,
- Data.ImplVendors.begin(), Data.ImplVendors.size(), SR);
- FD->addAttr(NewAttr);
- }
- void Sema::markOpenMPDeclareVariantFuncsReferenced(SourceLocation Loc,
- FunctionDecl *Func,
- bool MightBeOdrUse) {
- assert(LangOpts.OpenMP && "Expected OpenMP mode.");
- if (!Func->isDependentContext() && Func->hasAttrs()) {
- for (OMPDeclareVariantAttr *A :
- Func->specific_attrs<OMPDeclareVariantAttr>()) {
- // TODO: add checks for active OpenMP context where possible.
- Expr *VariantRef = A->getVariantFuncRef();
- auto *DRE = dyn_cast<DeclRefExpr>(VariantRef->IgnoreParenImpCasts());
- auto *F = cast<FunctionDecl>(DRE->getDecl());
- if (!F->isDefined() && F->isTemplateInstantiation())
- InstantiateFunctionDefinition(Loc, F->getFirstDecl());
- MarkFunctionReferenced(Loc, F, MightBeOdrUse);
- }
- }
- }
- 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 {
- /// Iteration space of a single for loop.
- struct LoopIterationSpace final {
- /// True if the condition operator is the strict compare operator (<, > or
- /// !=).
- bool IsStrictCompare = false;
- /// 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;
- /// Minimum value that can have the loop control variable. Used to support
- /// non-rectangular loops. Applied only for LCV with the non-iterator types,
- /// since only such variables can be used in non-loop invariant expressions.
- Expr *MinValue = nullptr;
- /// Maximum value that can have the loop control variable. Used to support
- /// non-rectangular loops. Applied only for LCV with the non-iterator type,
- /// since only such variables can be used in non-loop invariant expressions.
- Expr *MaxValue = nullptr;
- /// true, if the lower bound depends on the outer loop control var.
- bool IsNonRectangularLB = false;
- /// true, if the upper bound depends on the outer loop control var.
- bool IsNonRectangularUB = false;
- /// Index of the loop this loop depends on and forms non-rectangular loop
- /// nest.
- unsigned LoopDependentIdx = 0;
- /// Final condition for the non-rectangular loop nest support. It is used to
- /// check that the number of iterations for this particular counter must be
- /// finished.
- Expr *FinalCondition = nullptr;
- };
- /// 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;
- /// Data-sharing stack.
- DSAStackTy &Stack;
- /// 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;
- /// The outer loop counter this loop depends on (if any).
- const ValueDecl *DepDecl = nullptr;
- /// Contains number of loop (starts from 1) on which loop counter init
- /// expression of this loop depends on.
- Optional<unsigned> InitDependOnLC;
- /// Contains number of loop (starts from 1) on which loop counter condition
- /// expression of this loop depends on.
- Optional<unsigned> CondDependOnLC;
- /// Checks if the provide statement depends on the loop counter.
- Optional<unsigned> doesDependOnLoopCounter(const Stmt *S, bool IsInitializer);
- /// Original condition required for checking of the exit condition for
- /// non-rectangular loop.
- Expr *Condition = nullptr;
- public:
- OpenMPIterationSpaceChecker(Sema &SemaRef, DSAStackTy &Stack,
- SourceLocation DefaultLoc)
- : SemaRef(SemaRef), Stack(Stack), 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; }
- /// True, if the compare operator is strict (<, > or !=).
- bool isStrictTestOp() const { return TestIsStrictOp; }
- /// Build the expression to calculate the number of iterations.
- Expr *buildNumIterations(
- Scope *S, ArrayRef<LoopIterationSpace> ResultIterSpaces, 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);
- /// Builds the minimum value for the loop counter.
- std::pair<Expr *, Expr *> buildMinMaxValues(
- Scope *S, llvm::MapVector<const Expr *, DeclRefExpr *> &Captures) const;
- /// Builds final condition for the non-rectangular loops.
- Expr *buildFinalCondition(Scope *S) const;
- /// Return true if any expression is dependent.
- bool dependent() const;
- /// Returns true if the initializer forms non-rectangular loop.
- bool doesInitDependOnLC() const { return InitDependOnLC.hasValue(); }
- /// Returns true if the condition forms non-rectangular loop.
- bool doesCondDependOnLC() const { return CondDependOnLC.hasValue(); }
- /// Returns index of the loop we depend on (starting from 1), or 0 otherwise.
- unsigned getLoopDependentIdx() const {
- return InitDependOnLC.getValueOr(CondDependOnLC.getValueOr(0));
- }
- 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,
- bool EmitDiags);
- /// 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, bool EmitDiags) {
- // 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;
- if (EmitDiags)
- InitDependOnLC = doesDependOnLoopCounter(LB, /*IsInitializer=*/true);
- 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;
- CondDependOnLC = doesDependOnLoopCounter(UB, /*IsInitializer=*/false);
- 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;
- }
- namespace {
- /// Checker for the non-rectangular loops. Checks if the initializer or
- /// condition expression references loop counter variable.
- class LoopCounterRefChecker final
- : public ConstStmtVisitor<LoopCounterRefChecker, bool> {
- Sema &SemaRef;
- DSAStackTy &Stack;
- const ValueDecl *CurLCDecl = nullptr;
- const ValueDecl *DepDecl = nullptr;
- const ValueDecl *PrevDepDecl = nullptr;
- bool IsInitializer = true;
- unsigned BaseLoopId = 0;
- bool checkDecl(const Expr *E, const ValueDecl *VD) {
- if (getCanonicalDecl(VD) == getCanonicalDecl(CurLCDecl)) {
- SemaRef.Diag(E->getExprLoc(), diag::err_omp_stmt_depends_on_loop_counter)
- << (IsInitializer ? 0 : 1);
- return false;
- }
- const auto &&Data = Stack.isLoopControlVariable(VD);
- // OpenMP, 2.9.1 Canonical Loop Form, Restrictions.
- // The type of the loop iterator on which we depend may not have a random
- // access iterator type.
- if (Data.first && VD->getType()->isRecordType()) {
- SmallString<128> Name;
- llvm::raw_svector_ostream OS(Name);
- VD->getNameForDiagnostic(OS, SemaRef.getPrintingPolicy(),
- /*Qualified=*/true);
- SemaRef.Diag(E->getExprLoc(),
- diag::err_omp_wrong_dependency_iterator_type)
- << OS.str();
- SemaRef.Diag(VD->getLocation(), diag::note_previous_decl) << VD;
- return false;
- }
- if (Data.first &&
- (DepDecl || (PrevDepDecl &&
- getCanonicalDecl(VD) != getCanonicalDecl(PrevDepDecl)))) {
- if (!DepDecl && PrevDepDecl)
- DepDecl = PrevDepDecl;
- SmallString<128> Name;
- llvm::raw_svector_ostream OS(Name);
- DepDecl->getNameForDiagnostic(OS, SemaRef.getPrintingPolicy(),
- /*Qualified=*/true);
- SemaRef.Diag(E->getExprLoc(),
- diag::err_omp_invariant_or_linear_dependency)
- << OS.str();
- return false;
- }
- if (Data.first) {
- DepDecl = VD;
- BaseLoopId = Data.first;
- }
- return Data.first;
- }
- public:
- bool VisitDeclRefExpr(const DeclRefExpr *E) {
- const ValueDecl *VD = E->getDecl();
- if (isa<VarDecl>(VD))
- return checkDecl(E, VD);
- return false;
- }
- bool VisitMemberExpr(const MemberExpr *E) {
- if (isa<CXXThisExpr>(E->getBase()->IgnoreParens())) {
- const ValueDecl *VD = E->getMemberDecl();
- if (isa<VarDecl>(VD) || isa<FieldDecl>(VD))
- return checkDecl(E, VD);
- }
- return false;
- }
- bool VisitStmt(const Stmt *S) {
- bool Res = false;
- for (const Stmt *Child : S->children())
- Res = (Child && Visit(Child)) || Res;
- return Res;
- }
- explicit LoopCounterRefChecker(Sema &SemaRef, DSAStackTy &Stack,
- const ValueDecl *CurLCDecl, bool IsInitializer,
- const ValueDecl *PrevDepDecl = nullptr)
- : SemaRef(SemaRef), Stack(Stack), CurLCDecl(CurLCDecl),
- PrevDepDecl(PrevDepDecl), IsInitializer(IsInitializer) {}
- unsigned getBaseLoopId() const {
- assert(CurLCDecl && "Expected loop dependency.");
- return BaseLoopId;
- }
- const ValueDecl *getDepDecl() const {
- assert(CurLCDecl && "Expected loop dependency.");
- return DepDecl;
- }
- };
- } // namespace
- Optional<unsigned>
- OpenMPIterationSpaceChecker::doesDependOnLoopCounter(const Stmt *S,
- bool IsInitializer) {
- // Check for the non-rectangular loops.
- LoopCounterRefChecker LoopStmtChecker(SemaRef, Stack, LCDecl, IsInitializer,
- DepDecl);
- if (LoopStmtChecker.Visit(S)) {
- DepDecl = LoopStmtChecker.getDepDecl();
- return LoopStmtChecker.getBaseLoopId();
- }
- return llvm::None;
- }
- 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(),
- EmitDiags);
- return setLCDeclAndLB(DRE->getDecl(), DRE, BO->getRHS(), EmitDiags);
- }
- if (auto *ME = dyn_cast<MemberExpr>(LHS)) {
- if (ME->isArrow() &&
- isa<CXXThisExpr>(ME->getBase()->IgnoreParenImpCasts()))
- return setLCDeclAndLB(ME->getMemberDecl(), ME, BO->getRHS(),
- EmitDiags);
- }
- }
- } 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(), EmitDiags);
- }
- }
- }
- } 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(),
- EmitDiags);
- return setLCDeclAndLB(DRE->getDecl(), DRE, CE->getArg(1), EmitDiags);
- }
- if (auto *ME = dyn_cast<MemberExpr>(LHS)) {
- if (ME->isArrow() &&
- isa<CXXThisExpr>(ME->getBase()->IgnoreParenImpCasts()))
- return setLCDeclAndLB(ME->getMemberDecl(), ME, BO->getRHS(),
- EmitDiags);
- }
- }
- }
- 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.9] Canonical loop form. Test-expr may be one of the following:
- // var relational-op b
- // b relational-op var
- //
- bool IneqCondIsCanonical = SemaRef.getLangOpts().OpenMP >= 50;
- if (!S) {
- SemaRef.Diag(DefaultLoc, diag::err_omp_loop_not_canonical_cond)
- << (IneqCondIsCanonical ? 1 : 0) << LCDecl;
- return true;
- }
- Condition = S;
- 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 (IneqCondIsCanonical && 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:
- if (IneqCondIsCanonical)
- 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)
- << (IneqCondIsCanonical ? 1 : 0) << 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, ArrayRef<LoopIterationSpace> ResultIterSpaces, bool LimitedType,
- llvm::MapVector<const Expr *, DeclRefExpr *> &Captures) const {
- ExprResult Diff;
- QualType VarType = LCDecl->getType().getNonReferenceType();
- if (VarType->isIntegerType() || VarType->isPointerType() ||
- SemaRef.getLangOpts().CPlusPlus) {
- Expr *LBVal = LB;
- Expr *UBVal = UB;
- // LB = TestIsLessOp.getValue() ? min(LB(MinVal), LB(MaxVal)) :
- // max(LB(MinVal), LB(MaxVal))
- if (InitDependOnLC) {
- const LoopIterationSpace &IS =
- ResultIterSpaces[ResultIterSpaces.size() - 1 -
- InitDependOnLC.getValueOr(
- CondDependOnLC.getValueOr(0))];
- if (!IS.MinValue || !IS.MaxValue)
- return nullptr;
- // OuterVar = Min
- ExprResult MinValue =
- SemaRef.ActOnParenExpr(DefaultLoc, DefaultLoc, IS.MinValue);
- if (!MinValue.isUsable())
- return nullptr;
- ExprResult LBMinVal = SemaRef.BuildBinOp(S, DefaultLoc, BO_Assign,
- IS.CounterVar, MinValue.get());
- if (!LBMinVal.isUsable())
- return nullptr;
- // OuterVar = Min, LBVal
- LBMinVal =
- SemaRef.BuildBinOp(S, DefaultLoc, BO_Comma, LBMinVal.get(), LBVal);
- if (!LBMinVal.isUsable())
- return nullptr;
- // (OuterVar = Min, LBVal)
- LBMinVal = SemaRef.ActOnParenExpr(DefaultLoc, DefaultLoc, LBMinVal.get());
- if (!LBMinVal.isUsable())
- return nullptr;
- // OuterVar = Max
- ExprResult MaxValue =
- SemaRef.ActOnParenExpr(DefaultLoc, DefaultLoc, IS.MaxValue);
- if (!MaxValue.isUsable())
- return nullptr;
- ExprResult LBMaxVal = SemaRef.BuildBinOp(S, DefaultLoc, BO_Assign,
- IS.CounterVar, MaxValue.get());
- if (!LBMaxVal.isUsable())
- return nullptr;
- // OuterVar = Max, LBVal
- LBMaxVal =
- SemaRef.BuildBinOp(S, DefaultLoc, BO_Comma, LBMaxVal.get(), LBVal);
- if (!LBMaxVal.isUsable())
- return nullptr;
- // (OuterVar = Max, LBVal)
- LBMaxVal = SemaRef.ActOnParenExpr(DefaultLoc, DefaultLoc, LBMaxVal.get());
- if (!LBMaxVal.isUsable())
- return nullptr;
- Expr *LBMin = tryBuildCapture(SemaRef, LBMinVal.get(), Captures).get();
- Expr *LBMax = tryBuildCapture(SemaRef, LBMaxVal.get(), Captures).get();
- if (!LBMin || !LBMax)
- return nullptr;
- // LB(MinVal) < LB(MaxVal)
- ExprResult MinLessMaxRes =
- SemaRef.BuildBinOp(S, DefaultLoc, BO_LT, LBMin, LBMax);
- if (!MinLessMaxRes.isUsable())
- return nullptr;
- Expr *MinLessMax =
- tryBuildCapture(SemaRef, MinLessMaxRes.get(), Captures).get();
- if (!MinLessMax)
- return nullptr;
- if (TestIsLessOp.getValue()) {
- // LB(MinVal) < LB(MaxVal) ? LB(MinVal) : LB(MaxVal) - min(LB(MinVal),
- // LB(MaxVal))
- ExprResult MinLB = SemaRef.ActOnConditionalOp(DefaultLoc, DefaultLoc,
- MinLessMax, LBMin, LBMax);
- if (!MinLB.isUsable())
- return nullptr;
- LBVal = MinLB.get();
- } else {
- // LB(MinVal) < LB(MaxVal) ? LB(MaxVal) : LB(MinVal) - max(LB(MinVal),
- // LB(MaxVal))
- ExprResult MaxLB = SemaRef.ActOnConditionalOp(DefaultLoc, DefaultLoc,
- MinLessMax, LBMax, LBMin);
- if (!MaxLB.isUsable())
- return nullptr;
- LBVal = MaxLB.get();
- }
- }
- // UB = TestIsLessOp.getValue() ? max(UB(MinVal), UB(MaxVal)) :
- // min(UB(MinVal), UB(MaxVal))
- if (CondDependOnLC) {
- const LoopIterationSpace &IS =
- ResultIterSpaces[ResultIterSpaces.size() - 1 -
- InitDependOnLC.getValueOr(
- CondDependOnLC.getValueOr(0))];
- if (!IS.MinValue || !IS.MaxValue)
- return nullptr;
- // OuterVar = Min
- ExprResult MinValue =
- SemaRef.ActOnParenExpr(DefaultLoc, DefaultLoc, IS.MinValue);
- if (!MinValue.isUsable())
- return nullptr;
- ExprResult UBMinVal = SemaRef.BuildBinOp(S, DefaultLoc, BO_Assign,
- IS.CounterVar, MinValue.get());
- if (!UBMinVal.isUsable())
- return nullptr;
- // OuterVar = Min, UBVal
- UBMinVal =
- SemaRef.BuildBinOp(S, DefaultLoc, BO_Comma, UBMinVal.get(), UBVal);
- if (!UBMinVal.isUsable())
- return nullptr;
- // (OuterVar = Min, UBVal)
- UBMinVal = SemaRef.ActOnParenExpr(DefaultLoc, DefaultLoc, UBMinVal.get());
- if (!UBMinVal.isUsable())
- return nullptr;
- // OuterVar = Max
- ExprResult MaxValue =
- SemaRef.ActOnParenExpr(DefaultLoc, DefaultLoc, IS.MaxValue);
- if (!MaxValue.isUsable())
- return nullptr;
- ExprResult UBMaxVal = SemaRef.BuildBinOp(S, DefaultLoc, BO_Assign,
- IS.CounterVar, MaxValue.get());
- if (!UBMaxVal.isUsable())
- return nullptr;
- // OuterVar = Max, UBVal
- UBMaxVal =
- SemaRef.BuildBinOp(S, DefaultLoc, BO_Comma, UBMaxVal.get(), UBVal);
- if (!UBMaxVal.isUsable())
- return nullptr;
- // (OuterVar = Max, UBVal)
- UBMaxVal = SemaRef.ActOnParenExpr(DefaultLoc, DefaultLoc, UBMaxVal.get());
- if (!UBMaxVal.isUsable())
- return nullptr;
- Expr *UBMin = tryBuildCapture(SemaRef, UBMinVal.get(), Captures).get();
- Expr *UBMax = tryBuildCapture(SemaRef, UBMaxVal.get(), Captures).get();
- if (!UBMin || !UBMax)
- return nullptr;
- // UB(MinVal) > UB(MaxVal)
- ExprResult MinGreaterMaxRes =
- SemaRef.BuildBinOp(S, DefaultLoc, BO_GT, UBMin, UBMax);
- if (!MinGreaterMaxRes.isUsable())
- return nullptr;
- Expr *MinGreaterMax =
- tryBuildCapture(SemaRef, MinGreaterMaxRes.get(), Captures).get();
- if (!MinGreaterMax)
- return nullptr;
- if (TestIsLessOp.getValue()) {
- // UB(MinVal) > UB(MaxVal) ? UB(MinVal) : UB(MaxVal) - max(UB(MinVal),
- // UB(MaxVal))
- ExprResult MaxUB = SemaRef.ActOnConditionalOp(
- DefaultLoc, DefaultLoc, MinGreaterMax, UBMin, UBMax);
- if (!MaxUB.isUsable())
- return nullptr;
- UBVal = MaxUB.get();
- } else {
- // UB(MinVal) > UB(MaxVal) ? UB(MaxVal) : UB(MinVal) - min(UB(MinVal),
- // UB(MaxVal))
- ExprResult MinUB = SemaRef.ActOnConditionalOp(
- DefaultLoc, DefaultLoc, MinGreaterMax, UBMax, UBMin);
- if (!MinUB.isUsable())
- return nullptr;
- UBVal = MinUB.get();
- }
- }
- // Upper - Lower
- Expr *UBExpr = TestIsLessOp.getValue() ? UBVal : LBVal;
- Expr *LBExpr = TestIsLessOp.getValue() ? LBVal : UBVal;
- 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();
- }
- std::pair<Expr *, Expr *> OpenMPIterationSpaceChecker::buildMinMaxValues(
- Scope *S, llvm::MapVector<const Expr *, DeclRefExpr *> &Captures) const {
- // Do not build for iterators, they cannot be used in non-rectangular loop
- // nests.
- if (LCDecl->getType()->isRecordType())
- return std::make_pair(nullptr, nullptr);
- // If we subtract, the min is in the condition, otherwise the min is in the
- // init value.
- Expr *MinExpr = nullptr;
- Expr *MaxExpr = nullptr;
- Expr *LBExpr = TestIsLessOp.getValue() ? LB : UB;
- Expr *UBExpr = TestIsLessOp.getValue() ? UB : LB;
- bool LBNonRect = TestIsLessOp.getValue() ? InitDependOnLC.hasValue()
- : CondDependOnLC.hasValue();
- bool UBNonRect = TestIsLessOp.getValue() ? CondDependOnLC.hasValue()
- : InitDependOnLC.hasValue();
- Expr *Lower =
- LBNonRect ? LBExpr : tryBuildCapture(SemaRef, LBExpr, Captures).get();
- Expr *Upper =
- UBNonRect ? UBExpr : tryBuildCapture(SemaRef, UBExpr, Captures).get();
- if (!Upper || !Lower)
- return std::make_pair(nullptr, nullptr);
- if (TestIsLessOp.getValue())
- MinExpr = Lower;
- else
- MaxExpr = Upper;
- // Build minimum/maximum value based on number of iterations.
- ExprResult Diff;
- QualType VarType = LCDecl->getType().getNonReferenceType();
- Diff = SemaRef.BuildBinOp(S, DefaultLoc, BO_Sub, Upper, Lower);
- if (!Diff.isUsable())
- return std::make_pair(nullptr, nullptr);
- // Upper - Lower [- 1]
- if (TestIsStrictOp)
- Diff = SemaRef.BuildBinOp(
- S, DefaultLoc, BO_Sub, Diff.get(),
- SemaRef.ActOnIntegerConstant(SourceLocation(), 1).get());
- if (!Diff.isUsable())
- return std::make_pair(nullptr, nullptr);
- // Upper - Lower [- 1] + Step
- ExprResult NewStep = tryBuildCapture(SemaRef, Step, Captures);
- if (!NewStep.isUsable())
- return std::make_pair(nullptr, nullptr);
- // Parentheses (for dumping/debugging purposes only).
- Diff = SemaRef.ActOnParenExpr(DefaultLoc, DefaultLoc, Diff.get());
- if (!Diff.isUsable())
- return std::make_pair(nullptr, nullptr);
- // (Upper - Lower [- 1]) / Step
- Diff = SemaRef.BuildBinOp(S, DefaultLoc, BO_Div, Diff.get(), NewStep.get());
- if (!Diff.isUsable())
- return std::make_pair(nullptr, nullptr);
- // ((Upper - Lower [- 1]) / Step) * Step
- // Parentheses (for dumping/debugging purposes only).
- Diff = SemaRef.ActOnParenExpr(DefaultLoc, DefaultLoc, Diff.get());
- if (!Diff.isUsable())
- return std::make_pair(nullptr, nullptr);
- Diff = SemaRef.BuildBinOp(S, DefaultLoc, BO_Mul, Diff.get(), NewStep.get());
- if (!Diff.isUsable())
- return std::make_pair(nullptr, nullptr);
- // Convert to the original type or ptrdiff_t, if original type is pointer.
- if (!VarType->isAnyPointerType() &&
- !SemaRef.Context.hasSameType(Diff.get()->getType(), VarType)) {
- Diff = SemaRef.PerformImplicitConversion(
- Diff.get(), VarType, Sema::AA_Converting, /*AllowExplicit=*/true);
- } else if (VarType->isAnyPointerType() &&
- !SemaRef.Context.hasSameType(
- Diff.get()->getType(),
- SemaRef.Context.getUnsignedPointerDiffType())) {
- Diff = SemaRef.PerformImplicitConversion(
- Diff.get(), SemaRef.Context.getUnsignedPointerDiffType(),
- Sema::AA_Converting, /*AllowExplicit=*/true);
- }
- if (!Diff.isUsable())
- return std::make_pair(nullptr, nullptr);
- // Parentheses (for dumping/debugging purposes only).
- Diff = SemaRef.ActOnParenExpr(DefaultLoc, DefaultLoc, Diff.get());
- if (!Diff.isUsable())
- return std::make_pair(nullptr, nullptr);
- if (TestIsLessOp.getValue()) {
- // MinExpr = Lower;
- // MaxExpr = Lower + (((Upper - Lower [- 1]) / Step) * Step)
- Diff = SemaRef.BuildBinOp(S, DefaultLoc, BO_Add, Lower, Diff.get());
- if (!Diff.isUsable())
- return std::make_pair(nullptr, nullptr);
- Diff = SemaRef.ActOnFinishFullExpr(Diff.get(), /*DiscardedValue*/ false);
- if (!Diff.isUsable())
- return std::make_pair(nullptr, nullptr);
- MaxExpr = Diff.get();
- } else {
- // MaxExpr = Upper;
- // MinExpr = Upper - (((Upper - Lower [- 1]) / Step) * Step)
- Diff = SemaRef.BuildBinOp(S, DefaultLoc, BO_Sub, Upper, Diff.get());
- if (!Diff.isUsable())
- return std::make_pair(nullptr, nullptr);
- Diff = SemaRef.ActOnFinishFullExpr(Diff.get(), /*DiscardedValue*/ false);
- if (!Diff.isUsable())
- return std::make_pair(nullptr, nullptr);
- MinExpr = Diff.get();
- }
- return std::make_pair(MinExpr, MaxExpr);
- }
- Expr *OpenMPIterationSpaceChecker::buildFinalCondition(Scope *S) const {
- if (InitDependOnLC || CondDependOnLC)
- return Condition;
- return nullptr;
- }
- Expr *OpenMPIterationSpaceChecker::buildPreCond(
- Scope *S, Expr *Cond,
- llvm::MapVector<const Expr *, DeclRefExpr *> &Captures) const {
- // Do not build a precondition when the condition/initialization is dependent
- // to prevent pessimistic early loop exit.
- // TODO: this can be improved by calculating min/max values but not sure that
- // it will be very effective.
- if (CondDependOnLC || InitDependOnLC)
- return SemaRef.PerformImplicitConversion(
- SemaRef.ActOnIntegerConstant(SourceLocation(), 1).get(),
- SemaRef.Context.BoolTy, /*Action=*/Sema::AA_Casting,
- /*AllowExplicit=*/true).get();
- // Try to build LB <op> UB, where <op> is <, >, <=, or >=.
- Sema::TentativeAnalysisScope Trap(SemaRef);
- 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);
- }
- // Otherwise use original loop condition 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 cast<DeclRefExpr>(LCRef);
- }
- 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();
- }
- } // 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())) {
- DSAStack->loopStart();
- OpenMPIterationSpaceChecker ISC(*this, *DSAStack, ForLoc);
- if (!ISC.checkAndSetInit(Init, /*EmitDiags=*/false)) {
- if (ValueDecl *D = ISC.getLoopDecl()) {
- auto *VD = dyn_cast<VarDecl>(D);
- DeclRefExpr *PrivateRef = nullptr;
- if (!VD) {
- if (VarDecl *Private = isOpenMPCapturedDecl(D)) {
- VD = Private;
- } else {
- PrivateRef = buildCapture(*this, D, ISC.getLoopDeclRefExpr(),
- /*WithInit=*/false);
- VD = cast<VarDecl>(PrivateRef->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));
- }
- OpenMPDirectiveKind DKind = DSAStack->getCurrentDirective();
- // 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 =
- DSAStack->getTopDSA(D, /*FromParent=*/false);
- // If LoopVarRefExpr is nullptr it means the corresponding loop variable
- // is declared in the loop and it is predetermined as a private.
- Expr *LoopDeclRefExpr = ISC.getLoopDeclRefExpr();
- OpenMPClauseKind PredeterminedCKind =
- isOpenMPSimdDirective(DKind)
- ? (DSAStack->hasMutipleLoops() ? OMPC_lastprivate : OMPC_linear)
- : OMPC_private;
- if (((isOpenMPSimdDirective(DKind) && DVar.CKind != OMPC_unknown &&
- DVar.CKind != PredeterminedCKind && DVar.RefExpr &&
- (LangOpts.OpenMP <= 45 || (DVar.CKind != OMPC_lastprivate &&
- DVar.CKind != OMPC_private))) ||
- ((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)) {
- Diag(Init->getBeginLoc(), diag::err_omp_loop_var_dsa)
- << getOpenMPClauseName(DVar.CKind)
- << getOpenMPDirectiveName(DKind)
- << getOpenMPClauseName(PredeterminedCKind);
- if (DVar.RefExpr == nullptr)
- DVar.CKind = PredeterminedCKind;
- reportOriginalDsa(*this, DSAStack, D, DVar,
- /*IsLoopIterVar=*/true);
- } else if (LoopDeclRefExpr) {
- // 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)
- DSAStack->addDSA(D, LoopDeclRefExpr, PredeterminedCKind,
- PrivateRef);
- }
- }
- }
- 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,
- llvm::MutableArrayRef<LoopIterationSpace> ResultIterSpaces,
- llvm::MapVector<const Expr *, DeclRefExpr *> &Captures) {
- // OpenMP [2.9.1, Canonical Loop Form]
- // for (init-expr; test-expr; incr-expr) structured-block
- // for (range-decl: range-expr) structured-block
- auto *For = dyn_cast_or_null<ForStmt>(S);
- auto *CXXFor = dyn_cast_or_null<CXXForRangeStmt>(S);
- // Ranged for is supported only in OpenMP 5.0.
- if (!For && (SemaRef.LangOpts.OpenMP <= 45 || !CXXFor)) {
- 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 && For->getBody()) || (CXXFor && CXXFor->getBody())) &&
- "No loop body.");
- OpenMPIterationSpaceChecker ISC(SemaRef, DSA,
- For ? For->getForLoc() : CXXFor->getForLoc());
- // Check init.
- Stmt *Init = For ? For->getInit() : CXXFor->getBeginStmt();
- if (ISC.checkAndSetInit(Init))
- return true;
- bool HasErrors = false;
- // Check loop variable's type.
- if (ValueDecl *LCDecl = ISC.getLoopDecl()) {
- // 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);
- assert(isOpenMPLoopDirective(DKind) && "DSA for non-loop vars");
- // Check test-expr.
- HasErrors |= ISC.checkAndSetCond(For ? For->getCond() : CXXFor->getCond());
- // Check incr-expr.
- HasErrors |= ISC.checkAndSetInc(For ? For->getInc() : CXXFor->getInc());
- }
- if (ISC.dependent() || SemaRef.CurContext->isDependentContext() || HasErrors)
- return HasErrors;
- // Build the loop's iteration space representation.
- ResultIterSpaces[CurrentNestedLoopCount].PreCond = ISC.buildPreCond(
- DSA.getCurScope(), For ? For->getCond() : CXXFor->getCond(), Captures);
- ResultIterSpaces[CurrentNestedLoopCount].NumIterations =
- ISC.buildNumIterations(DSA.getCurScope(), ResultIterSpaces,
- (isOpenMPWorksharingDirective(DKind) ||
- isOpenMPTaskLoopDirective(DKind) ||
- isOpenMPDistributeDirective(DKind)),
- Captures);
- ResultIterSpaces[CurrentNestedLoopCount].CounterVar =
- ISC.buildCounterVar(Captures, DSA);
- ResultIterSpaces[CurrentNestedLoopCount].PrivateCounterVar =
- ISC.buildPrivateCounterVar();
- ResultIterSpaces[CurrentNestedLoopCount].CounterInit = ISC.buildCounterInit();
- ResultIterSpaces[CurrentNestedLoopCount].CounterStep = ISC.buildCounterStep();
- ResultIterSpaces[CurrentNestedLoopCount].InitSrcRange = ISC.getInitSrcRange();
- ResultIterSpaces[CurrentNestedLoopCount].CondSrcRange =
- ISC.getConditionSrcRange();
- ResultIterSpaces[CurrentNestedLoopCount].IncSrcRange =
- ISC.getIncrementSrcRange();
- ResultIterSpaces[CurrentNestedLoopCount].Subtract = ISC.shouldSubtractStep();
- ResultIterSpaces[CurrentNestedLoopCount].IsStrictCompare =
- ISC.isStrictTestOp();
- std::tie(ResultIterSpaces[CurrentNestedLoopCount].MinValue,
- ResultIterSpaces[CurrentNestedLoopCount].MaxValue) =
- ISC.buildMinMaxValues(DSA.getCurScope(), Captures);
- ResultIterSpaces[CurrentNestedLoopCount].FinalCondition =
- ISC.buildFinalCondition(DSA.getCurScope());
- ResultIterSpaces[CurrentNestedLoopCount].IsNonRectangularLB =
- ISC.doesInitDependOnLC();
- ResultIterSpaces[CurrentNestedLoopCount].IsNonRectangularUB =
- ISC.doesCondDependOnLC();
- ResultIterSpaces[CurrentNestedLoopCount].LoopDependentIdx =
- ISC.getLoopDependentIdx();
- HasErrors |=
- (ResultIterSpaces[CurrentNestedLoopCount].PreCond == nullptr ||
- ResultIterSpaces[CurrentNestedLoopCount].NumIterations == nullptr ||
- ResultIterSpaces[CurrentNestedLoopCount].CounterVar == nullptr ||
- ResultIterSpaces[CurrentNestedLoopCount].PrivateCounterVar == nullptr ||
- ResultIterSpaces[CurrentNestedLoopCount].CounterInit == nullptr ||
- ResultIterSpaces[CurrentNestedLoopCount].CounterStep == nullptr);
- if (!HasErrors && DSA.isOrderedRegion()) {
- if (DSA.getOrderedRegionParam().second->getNumForLoops()) {
- if (CurrentNestedLoopCount <
- DSA.getOrderedRegionParam().second->getLoopNumIterations().size()) {
- DSA.getOrderedRegionParam().second->setLoopNumIterations(
- CurrentNestedLoopCount,
- ResultIterSpaces[CurrentNestedLoopCount].NumIterations);
- DSA.getOrderedRegionParam().second->setLoopCounter(
- CurrentNestedLoopCount,
- ResultIterSpaces[CurrentNestedLoopCount].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(),
- ResultIterSpaces[CurrentNestedLoopCount].CounterVar, Captures,
- Pair.first->getDependencyLoc());
- else
- CntValue = ISC.buildOrderedLoopData(
- DSA.getCurScope(),
- ResultIterSpaces[CurrentNestedLoopCount].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, bool IsNonRectangularLB,
- llvm::MapVector<const Expr *, DeclRefExpr *> &Captures) {
- // Build 'VarRef = Start.
- ExprResult NewStart = IsNonRectangularLB
- ? Start.get()
- : 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,
- bool IsNonRectangularLB,
- 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'.
- if (!Start.isUsable())
- return ExprError();
- ExprResult NewStart = SemaRef.ActOnParenExpr(Loc, Loc, Start.get());
- if (!NewStart.isUsable())
- return ExprError();
- if (Captures && !IsNonRectangularLB)
- 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()) {
- Sema::TentativeAnalysisScope Trap(SemaRef);
- 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());
- }
- }
- }
- // 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->isValueDependent() &&
- CollapseLoopCountExpr->EvaluateAsInt(Result, SemaRef.getASTContext())) {
- NestedLoopCount = Result.Val.getInt().getLimitedValue();
- } else {
- Built.clear(/*Size=*/1);
- return 1;
- }
- }
- unsigned OrderedLoopCount = 1;
- if (OrderedLoopCountExpr) {
- // Found 'ordered' clause - calculate collapse number.
- Expr::EvalResult EVResult;
- if (!OrderedLoopCountExpr->isValueDependent() &&
- 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();
- } else {
- Built.clear(/*Size=*/1);
- return 1;
- }
- }
- // This is helper routine for loop directives (e.g., 'for', 'simd',
- // 'for simd', etc.).
- llvm::MapVector<const Expr *, DeclRefExpr *> Captures;
- SmallVector<LoopIterationSpace, 4> IterSpaces(
- 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, 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.
- if (auto *For = dyn_cast<ForStmt>(CurStmt)) {
- CurStmt = For->getBody();
- } else {
- assert(isa<CXXForRangeStmt>(CurStmt) &&
- "Expected canonical for or range-based for loops.");
- CurStmt = cast<CXXForRangeStmt>(CurStmt)->getBody();
- }
- CurStmt = CurStmt->IgnoreContainers();
- }
- for (unsigned Cnt = NestedLoopCount; Cnt < OrderedLoopCount; ++Cnt) {
- if (checkOpenMPIterationSpace(
- DKind, CurStmt, SemaRef, DSA, Cnt, NestedLoopCount,
- std::max(OrderedLoopCount, NestedLoopCount), CollapseLoopCountExpr,
- OrderedLoopCountExpr, VarsWithImplicitDSA, IterSpaces, 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.
- if (auto *For = dyn_cast<ForStmt>(CurStmt)) {
- CurStmt = For->getBody();
- } else {
- assert(isa<CXXForRangeStmt>(CurStmt) &&
- "Expected canonical for or range-based for loops.");
- CurStmt = cast<CXXForRangeStmt>(CurStmt)->getBody();
- }
- CurStmt = CurStmt->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 (SemaRef.getLangOpts().OpenMPOptimisticCollapse ||
- (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(), /*DiscardedValue*/ false);
- // 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(), /*DiscardedValue*/ false);
- 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(), /*DiscardedValue*/ false);
- 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(), /*DiscardedValue*/ false);
- }
- }
- bool UseStrictCompare =
- RealVType->hasUnsignedIntegerRepresentation() &&
- llvm::all_of(IterSpaces, [](const LoopIterationSpace &LIS) {
- return LIS.IsStrictCompare;
- });
- // Loop condition (IV < NumIterations) or (IV <= UB or IV < UB + 1 (for
- // unsigned IV)) for worksharing loops.
- SourceLocation CondLoc = AStmt->getBeginLoc();
- Expr *BoundUB = UB.get();
- if (UseStrictCompare) {
- BoundUB =
- SemaRef
- .BuildBinOp(CurScope, CondLoc, BO_Add, BoundUB,
- SemaRef.ActOnIntegerConstant(SourceLocation(), 1).get())
- .get();
- BoundUB =
- SemaRef.ActOnFinishFullExpr(BoundUB, /*DiscardedValue*/ false).get();
- }
- ExprResult Cond =
- (isOpenMPWorksharingDirective(DKind) ||
- isOpenMPTaskLoopDirective(DKind) || isOpenMPDistributeDirective(DKind))
- ? SemaRef.BuildBinOp(CurScope, CondLoc,
- UseStrictCompare ? BO_LT : BO_LE, IV.get(),
- BoundUB)
- : 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)) {
- Expr *BoundCombUB = CombUB.get();
- if (UseStrictCompare) {
- BoundCombUB =
- SemaRef
- .BuildBinOp(
- CurScope, CondLoc, BO_Add, BoundCombUB,
- SemaRef.ActOnIntegerConstant(SourceLocation(), 1).get())
- .get();
- BoundCombUB =
- SemaRef.ActOnFinishFullExpr(BoundCombUB, /*DiscardedValue*/ false)
- .get();
- }
- CombCond =
- SemaRef.BuildBinOp(CurScope, CondLoc, UseStrictCompare ? BO_LT : BO_LE,
- IV.get(), BoundCombUB);
- }
- // 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(), /*DiscardedValue*/ false);
- 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(), /*DiscardedValue*/ false);
- 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(), /*DiscardedValue*/ false);
- 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(),
- /*DiscardedValue*/ false);
- 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(),
- /*DiscardedValue*/ false);
- 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, UseStrictCompare ? BO_LT : BO_LE, IV.get(), BoundUB);
- 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(), /*DiscardedValue*/ false);
- 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(), /*DiscardedValue*/ false);
- // Build IV <= PrevUB or IV < PrevUB + 1 for unsigned IV to be used in
- // parallel for is in combination with a distribute directive with
- // schedule(static, 1)
- Expr *BoundPrevUB = PrevUB.get();
- if (UseStrictCompare) {
- BoundPrevUB =
- SemaRef
- .BuildBinOp(
- CurScope, CondLoc, BO_Add, BoundPrevUB,
- SemaRef.ActOnIntegerConstant(SourceLocation(), 1).get())
- .get();
- BoundPrevUB =
- SemaRef.ActOnFinishFullExpr(BoundPrevUB, /*DiscardedValue*/ false)
- .get();
- }
- ParForInDistCond =
- SemaRef.BuildBinOp(CurScope, CondLoc, UseStrictCompare ? BO_LT : BO_LE,
- IV.get(), BoundPrevUB);
- }
- // 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);
- Built.DependentCounters.resize(NestedLoopCount);
- Built.DependentInits.resize(NestedLoopCount);
- Built.FinalsConditions.resize(NestedLoopCount);
- {
- // We implement the following algorithm for obtaining the
- // original loop iteration variable values based on the
- // value of the collapsed loop iteration variable IV.
- //
- // Let n+1 be the number of collapsed loops in the nest.
- // Iteration variables (I0, I1, .... In)
- // Iteration counts (N0, N1, ... Nn)
- //
- // Acc = IV;
- //
- // To compute Ik for loop k, 0 <= k <= n, generate:
- // Prod = N(k+1) * N(k+2) * ... * Nn;
- // Ik = Acc / Prod;
- // Acc -= Ik * Prod;
- //
- ExprResult Acc = IV;
- for (unsigned int Cnt = 0; Cnt < NestedLoopCount; ++Cnt) {
- LoopIterationSpace &IS = IterSpaces[Cnt];
- SourceLocation UpdLoc = IS.IncSrcRange.getBegin();
- ExprResult Iter;
- // Compute prod
- ExprResult Prod =
- SemaRef.ActOnIntegerConstant(SourceLocation(), 1).get();
- for (unsigned int K = Cnt+1; K < NestedLoopCount; ++K)
- Prod = SemaRef.BuildBinOp(CurScope, UpdLoc, BO_Mul, Prod.get(),
- IterSpaces[K].NumIterations);
- // Iter = Acc / Prod
- // If there is at least one more inner loop to avoid
- // multiplication by 1.
- if (Cnt + 1 < NestedLoopCount)
- Iter = SemaRef.BuildBinOp(CurScope, UpdLoc, BO_Div,
- Acc.get(), Prod.get());
- else
- Iter = Acc;
- if (!Iter.isUsable()) {
- HasErrors = true;
- break;
- }
- // Update Acc:
- // Acc -= Iter * Prod
- // Check if there is at least one more inner loop to avoid
- // multiplication by 1.
- if (Cnt + 1 < NestedLoopCount)
- Prod = SemaRef.BuildBinOp(CurScope, UpdLoc, BO_Mul,
- Iter.get(), Prod.get());
- else
- Prod = Iter;
- Acc = SemaRef.BuildBinOp(CurScope, UpdLoc, BO_Sub,
- Acc.get(), Prod.get());
- // 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, IS.IsNonRectangularLB, Captures);
- if (!Init.isUsable()) {
- HasErrors = true;
- break;
- }
- ExprResult Update = buildCounterUpdate(
- SemaRef, CurScope, UpdLoc, CounterVar, IS.CounterInit, Iter,
- IS.CounterStep, IS.Subtract, IS.IsNonRectangularLB, &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, IS.IsNonRectangularLB, &Captures);
- if (!Final.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();
- Built.DependentCounters[Cnt] = nullptr;
- Built.DependentInits[Cnt] = nullptr;
- Built.FinalsConditions[Cnt] = nullptr;
- if (IS.IsNonRectangularLB || IS.IsNonRectangularUB) {
- Built.DependentCounters[Cnt] =
- Built.Counters[NestedLoopCount - 1 - IS.LoopDependentIdx];
- Built.DependentInits[Cnt] =
- Built.Inits[NestedLoopCount - 1 - IS.LoopDependentIdx];
- Built.FinalsConditions[Cnt] = IS.FinalCondition;
- }
- }
- }
- if (HasErrors)
- return 0;
- // Save results
- Built.IterationVarRef = IV.get();
- Built.LastIteration = LastIteration.get();
- Built.NumIterations = NumIterations.get();
- Built.CalcLastIteration = SemaRef
- .ActOnFinishFullExpr(CalcLastIteration.get(),
- /*DiscardedValue=*/false)
- .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) {
- 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_allocator:
- Res = ActOnOpenMPAllocatorClause(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_allocate:
- 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:
- case OMPC_device_type:
- case OMPC_match:
- 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_allocate:
- case OMPD_taskyield:
- case OMPD_barrier:
- case OMPD_taskwait:
- case OMPD_cancellation_point:
- case OMPD_flush:
- case OMPD_declare_reduction:
- case OMPD_declare_mapper:
- case OMPD_declare_simd:
- case OMPD_declare_variant:
- 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_allocate:
- case OMPD_taskyield:
- case OMPD_barrier:
- case OMPD_taskwait:
- case OMPD_cancellation_point:
- case OMPD_flush:
- case OMPD_declare_reduction:
- case OMPD_declare_mapper:
- case OMPD_declare_simd:
- case OMPD_declare_variant:
- 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_allocate:
- case OMPD_taskyield:
- case OMPD_barrier:
- case OMPD_taskwait:
- case OMPD_cancellation_point:
- case OMPD_flush:
- case OMPD_declare_reduction:
- case OMPD_declare_mapper:
- case OMPD_declare_simd:
- case OMPD_declare_variant:
- 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_allocate:
- case OMPD_taskyield:
- case OMPD_barrier:
- case OMPD_taskwait:
- case OMPD_cancellation_point:
- case OMPD_flush:
- case OMPD_declare_reduction:
- case OMPD_declare_mapper:
- case OMPD_declare_simd:
- case OMPD_declare_variant:
- 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_allocate:
- case OMPD_taskyield:
- case OMPD_barrier:
- case OMPD_taskwait:
- case OMPD_cancellation_point:
- case OMPD_flush:
- case OMPD_declare_reduction:
- case OMPD_declare_mapper:
- case OMPD_declare_simd:
- case OMPD_declare_variant:
- 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_allocate:
- case OMPD_taskyield:
- case OMPD_barrier:
- case OMPD_taskwait:
- case OMPD_cancellation_point:
- case OMPD_flush:
- case OMPD_declare_reduction:
- case OMPD_declare_mapper:
- case OMPD_declare_simd:
- case OMPD_declare_variant:
- 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_allocate:
- case OMPD_taskyield:
- case OMPD_barrier:
- case OMPD_taskwait:
- case OMPD_cancellation_point:
- case OMPD_flush:
- case OMPD_declare_reduction:
- case OMPD_declare_mapper:
- case OMPD_declare_simd:
- case OMPD_declare_variant:
- 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_allocator:
- 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_allocate:
- 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:
- case OMPC_device_type:
- case OMPC_match:
- 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);
- }
- /// Tries to find omp_allocator_handle_t type.
- static bool findOMPAllocatorHandleT(Sema &S, SourceLocation Loc,
- DSAStackTy *Stack) {
- QualType OMPAllocatorHandleT = Stack->getOMPAllocatorHandleT();
- if (!OMPAllocatorHandleT.isNull())
- return true;
- // Build the predefined allocator expressions.
- bool ErrorFound = false;
- for (int I = OMPAllocateDeclAttr::OMPDefaultMemAlloc;
- I < OMPAllocateDeclAttr::OMPUserDefinedMemAlloc; ++I) {
- auto AllocatorKind = static_cast<OMPAllocateDeclAttr::AllocatorTypeTy>(I);
- StringRef Allocator =
- OMPAllocateDeclAttr::ConvertAllocatorTypeTyToStr(AllocatorKind);
- DeclarationName AllocatorName = &S.getASTContext().Idents.get(Allocator);
- auto *VD = dyn_cast_or_null<ValueDecl>(
- S.LookupSingleName(S.TUScope, AllocatorName, Loc, Sema::LookupAnyName));
- if (!VD) {
- ErrorFound = true;
- break;
- }
- QualType AllocatorType =
- VD->getType().getNonLValueExprType(S.getASTContext());
- ExprResult Res = S.BuildDeclRefExpr(VD, AllocatorType, VK_LValue, Loc);
- if (!Res.isUsable()) {
- ErrorFound = true;
- break;
- }
- if (OMPAllocatorHandleT.isNull())
- OMPAllocatorHandleT = AllocatorType;
- if (!S.getASTContext().hasSameType(OMPAllocatorHandleT, AllocatorType)) {
- ErrorFound = true;
- break;
- }
- Stack->setAllocator(AllocatorKind, Res.get());
- }
- if (ErrorFound) {
- S.Diag(Loc, diag::err_implied_omp_allocator_handle_t_not_found);
- return false;
- }
- OMPAllocatorHandleT.addConst();
- Stack->setOMPAllocatorHandleT(OMPAllocatorHandleT);
- return true;
- }
- OMPClause *Sema::ActOnOpenMPAllocatorClause(Expr *A, SourceLocation StartLoc,
- SourceLocation LParenLoc,
- SourceLocation EndLoc) {
- // OpenMP [2.11.3, allocate Directive, Description]
- // allocator is an expression of omp_allocator_handle_t type.
- if (!findOMPAllocatorHandleT(*this, A->getExprLoc(), DSAStack))
- return nullptr;
- ExprResult Allocator = DefaultLvalueConversion(A);
- if (Allocator.isInvalid())
- return nullptr;
- Allocator = PerformImplicitConversion(Allocator.get(),
- DSAStack->getOMPAllocatorHandleT(),
- Sema::AA_Initializing,
- /*AllowExplicit=*/true);
- if (Allocator.isInvalid())
- return nullptr;
- return new (Context)
- OMPAllocatorClause(Allocator.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_allocator:
- 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_allocate:
- 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:
- case OMPC_device_type:
- case OMPC_match:
- 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_allocator:
- 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_allocate:
- 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:
- case OMPC_device_type:
- case OMPC_match:
- 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_allocator:
- 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_allocate:
- 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:
- case OMPC_device_type:
- case OMPC_match:
- 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,
- const OMPVarListLocTy &Locs, SourceLocation ColonLoc,
- CXXScopeSpec &ReductionOrMapperIdScopeSpec,
- DeclarationNameInfo &ReductionOrMapperId, OpenMPDependClauseKind DepKind,
- OpenMPLinearClauseKind LinKind,
- ArrayRef<OpenMPMapModifierKind> MapTypeModifiers,
- ArrayRef<SourceLocation> MapTypeModifiersLoc, OpenMPMapClauseKind MapType,
- bool IsMapTypeImplicit, SourceLocation DepLinMapLoc) {
- SourceLocation StartLoc = Locs.StartLoc;
- SourceLocation LParenLoc = Locs.LParenLoc;
- SourceLocation EndLoc = Locs.EndLoc;
- 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, ReductionOrMapperIdScopeSpec,
- ReductionOrMapperId);
- break;
- case OMPC_task_reduction:
- Res = ActOnOpenMPTaskReductionClause(VarList, StartLoc, LParenLoc, ColonLoc,
- EndLoc, ReductionOrMapperIdScopeSpec,
- ReductionOrMapperId);
- break;
- case OMPC_in_reduction:
- Res = ActOnOpenMPInReductionClause(VarList, StartLoc, LParenLoc, ColonLoc,
- EndLoc, ReductionOrMapperIdScopeSpec,
- ReductionOrMapperId);
- 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,
- ReductionOrMapperIdScopeSpec,
- ReductionOrMapperId, MapType, IsMapTypeImplicit,
- DepLinMapLoc, ColonLoc, VarList, Locs);
- break;
- case OMPC_to:
- Res = ActOnOpenMPToClause(VarList, ReductionOrMapperIdScopeSpec,
- ReductionOrMapperId, Locs);
- break;
- case OMPC_from:
- Res = ActOnOpenMPFromClause(VarList, ReductionOrMapperIdScopeSpec,
- ReductionOrMapperId, Locs);
- break;
- case OMPC_use_device_ptr:
- Res = ActOnOpenMPUseDevicePtrClause(VarList, Locs);
- break;
- case OMPC_is_device_ptr:
- Res = ActOnOpenMPIsDevicePtrClause(VarList, Locs);
- break;
- case OMPC_allocate:
- Res = ActOnOpenMPAllocateClause(TailExpr, VarList, StartLoc, LParenLoc,
- ColonLoc, EndLoc);
- break;
- case OMPC_if:
- case OMPC_final:
- case OMPC_num_threads:
- case OMPC_safelen:
- case OMPC_simdlen:
- case OMPC_allocator:
- 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:
- case OMPC_device_type:
- case OMPC_match:
- 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;
- }
- 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 5.0 [2.19.3, List Item Privatization, Restrictions]
- // A variable that is privatized must not have a const-qualified type
- // unless it is of class type with a mutable member. This restriction does
- // not apply to the firstprivate clause.
- //
- // OpenMP 3.1 [2.9.3.3, private clause, Restrictions]
- // A variable that appears in a private clause must not have a
- // const-qualified type unless it is of class type with a mutable member.
- if (rejectConstNotMutableType(*this, D, Type, OMPC_private, ELoc))
- 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.
- 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
- //
- // OpenMP 5.0 [2.19.7.1, Restrictions, p.7]
- // A list item cannot appear in both a map clause and a data-sharing
- // attribute clause on the same construct unless the construct is a
- // combined construct.
- if ((LangOpts.OpenMP <= 45 && isOpenMPTargetExecutionDirective(CurrDir)) ||
- CurrDir == OMPD_target) {
- 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
- //
- // OpenMP 5.0 [2.19.7.1, Restrictions, p.7]
- // A list item cannot appear in both a map clause and a data-sharing
- // attribute clause on the same construct unless the construct is a
- // combined construct.
- if ((LangOpts.OpenMP <= 45 &&
- isOpenMPTargetExecutionDirective(CurrDir)) ||
- CurrDir == OMPD_target) {
- 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();
- // OpenMP 5.0 [2.19.3, List Item Privatization, Restrictions]
- // A variable that is privatized must not have a const-qualified type
- // unless it is of class type with a mutable member. This restriction does
- // not apply to the firstprivate clause.
- //
- // OpenMP 3.1 [2.9.3.5, lastprivate clause, Restrictions]
- // A variable that appears in a lastprivate clause must not have a
- // const-qualified type unless it is of class type with a mutable member.
- if (rejectConstNotMutableType(*this, D, Type, OMPC_lastprivate, ELoc))
- continue;
- 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*/ false);
- 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 filterLookupForUDReductionAndMapper(
- 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 &Id,
- 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(Id.getName());
- for (auto *D : R) {
- auto *Underlying = D;
- if (auto *USD = dyn_cast<UsingShadowDecl>(D))
- Underlying = USD->getTargetDecl();
- if (!isa<OMPDeclareReductionDecl>(Underlying) &&
- !isa<OMPDeclareMapperDecl>(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 = dyn_cast<OMPDeclareReductionDecl>(D))
- Lookups.back().addDecl(DRD);
- PrevD = D;
- }
- }
- if (SemaRef.CurContext->isDependentContext() || Ty->isDependentType() ||
- Ty->isInstantiationDependentType() ||
- Ty->containsUnexpandedParameterPack() ||
- filterLookupForUDReductionAndMapper<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.
- if (SemaRef.getLangOpts().CPlusPlus)
- argumentDependentLookup(SemaRef, ReductionId, Loc, Ty, Lookups);
- if (auto *VD = filterLookupForUDReductionAndMapper<ValueDecl *>(
- Lookups, [&SemaRef, Ty](ValueDecl *D) -> ValueDecl * {
- if (!D->isInvalidDecl() &&
- SemaRef.Context.hasSameType(D->getType(), Ty))
- return D;
- return nullptr;
- }))
- return SemaRef.BuildDeclRefExpr(VD, VD->getType().getNonReferenceType(),
- VK_LValue, Loc);
- if (SemaRef.getLangOpts().CPlusPlus) {
- if (auto *VD = filterLookupForUDReductionAndMapper<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, VD->getType().getNonReferenceType(), 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 (rejectConstNotMutableType(S, D, Type, ClauseKind, ELoc,
- /*AcceptIfMutable*/ false, ASE || OASE))
- 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, nullptr,
- ArrayType::Normal,
- /*IndexTypeQuals=*/0);
- }
- }
- if ((OASE && !ConstantLengthOASE) ||
- (!OASE && !ASE &&
- D->getType().getNonReferenceType()->isVariablyModifiedType())) {
- if (!Context.getTargetInfo().isVLASupported()) {
- if (isOpenMPTargetExecutionDirective(Stack->getCurrentDirective())) {
- S.Diag(ELoc, diag::err_omp_reduction_vla_unsupported) << !!OASE;
- S.Diag(ELoc, diag::note_vla_unsupported);
- } else {
- S.targetDiag(ELoc, diag::err_omp_reduction_vla_unsupported) << !!OASE;
- S.targetDiag(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.LangOpts.CPlusPlus)) {
- 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 =
- CallExpr::Create(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(),
- /*DiscardedValue*/ false);
- }
- 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();
- // OpenMP 5.0 [2.19.3, List Item Privatization, Restrictions]
- // A variable that is privatized must not have a const-qualified type
- // unless it is of class type with a mutable member. This restriction does
- // not apply to the firstprivate clause.
- if (rejectConstNotMutableType(*this, D, Type, OMPC_linear, ELoc))
- 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(), /*DiscardedValue*/ false);
- // 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;
- SmallVector<Expr *, 8> UsedExprs;
- 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, /*IsNonRectangularLB=*/false);
- else
- Update = *CurPrivate;
- Update = SemaRef.ActOnFinishFullExpr(Update.get(), DE->getBeginLoc(),
- /*DiscardedValue*/ false);
- // Build final: Var = InitExpr + NumIterations * Step
- ExprResult Final;
- if (!Info.first)
- Final =
- buildCounterUpdate(SemaRef, S, RefExpr->getExprLoc(), CapturedRef,
- InitExpr, NumIterations, Step, /*Subtract=*/false,
- /*IsNonRectangularLB=*/false);
- else
- Final = *CurPrivate;
- Final = SemaRef.ActOnFinishFullExpr(Final.get(), DE->getBeginLoc(),
- /*DiscardedValue*/ false);
- if (!Update.isUsable() || !Final.isUsable()) {
- Updates.push_back(nullptr);
- Finals.push_back(nullptr);
- UsedExprs.push_back(nullptr);
- HasErrors = true;
- } else {
- Updates.push_back(Update.get());
- Finals.push_back(Final.get());
- if (!Info.first)
- UsedExprs.push_back(SimpleRefExpr);
- }
- ++CurInit;
- ++CurPrivate;
- }
- if (Expr *S = Clause.getStep())
- UsedExprs.push_back(S);
- // Fill the remaining part with the nullptr.
- UsedExprs.append(Clause.varlist_size() + 1 - UsedExprs.size(), nullptr);
- Clause.setUpdates(Updates);
- Clause.setFinals(Finals);
- Clause.setUsedExprs(UsedExprs);
- 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*/ false);
- 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*/ false);
- 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;
- }
- ExprResult Res;
- {
- Sema::TentativeAnalysisScope Trap(*this);
- Res = CreateBuiltinUnaryOp(ELoc, UO_AddrOf,
- RefExpr->IgnoreParenImpCasts());
- }
- 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;
- if (const auto *TE = dyn_cast<CXXThisExpr>(E)) {
- Expr::EvalResult Result;
- if (CurE->getIdx()->EvaluateAsInt(Result, SemaRef.getASTContext())) {
- if (!Result.Val.getInt().isNullValue()) {
- SemaRef.Diag(CurE->getIdx()->getExprLoc(),
- diag::err_omp_invalid_map_this_expr);
- SemaRef.Diag(CurE->getIdx()->getExprLoc(),
- diag::note_omp_invalid_subscript_on_this_ptr_map);
- }
- }
- RelevantExpr = TE;
- }
- // 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;
- }
- if (const auto *TE = dyn_cast<CXXThisExpr>(E)) {
- Expr::EvalResult ResultR;
- Expr::EvalResult ResultL;
- if (CurE->getLength()->EvaluateAsInt(ResultR,
- SemaRef.getASTContext())) {
- if (!ResultR.Val.getInt().isOneValue()) {
- SemaRef.Diag(CurE->getLength()->getExprLoc(),
- diag::err_omp_invalid_map_this_expr);
- SemaRef.Diag(CurE->getLength()->getExprLoc(),
- diag::note_omp_invalid_length_on_this_ptr_mapping);
- }
- }
- if (CurE->getLowerBound() && CurE->getLowerBound()->EvaluateAsInt(
- ResultL, SemaRef.getASTContext())) {
- if (!ResultL.Val.getInt().isNullValue()) {
- SemaRef.Diag(CurE->getLowerBound()->getExprLoc(),
- diag::err_omp_invalid_map_this_expr);
- SemaRef.Diag(CurE->getLowerBound()->getExprLoc(),
- diag::note_omp_invalid_lower_bound_on_this_ptr_mapping);
- }
- }
- RelevantExpr = TE;
- }
- // 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;
- }
- // Look up the user-defined mapper given the mapper name and mapped type, and
- // build a reference to it.
- static ExprResult buildUserDefinedMapperRef(Sema &SemaRef, Scope *S,
- CXXScopeSpec &MapperIdScopeSpec,
- const DeclarationNameInfo &MapperId,
- QualType Type,
- Expr *UnresolvedMapper) {
- if (MapperIdScopeSpec.isInvalid())
- return ExprError();
- // Get the actual type for the array type.
- if (Type->isArrayType()) {
- assert(Type->getAsArrayTypeUnsafe() && "Expect to get a valid array type");
- Type = Type->getAsArrayTypeUnsafe()->getElementType().getCanonicalType();
- }
- // Find all user-defined mappers with the given MapperId.
- SmallVector<UnresolvedSet<8>, 4> Lookups;
- LookupResult Lookup(SemaRef, MapperId, Sema::LookupOMPMapperName);
- Lookup.suppressDiagnostics();
- if (S) {
- while (S && SemaRef.LookupParsedName(Lookup, S, &MapperIdScopeSpec)) {
- NamedDecl *D = Lookup.getRepresentativeDecl();
- while (S && !S->isDeclScope(D))
- S = S->getParent();
- 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>(UnresolvedMapper)) {
- // Extract the user-defined mappers with the given MapperId.
- Lookups.push_back(UnresolvedSet<8>());
- for (NamedDecl *D : ULE->decls()) {
- auto *DMD = cast<OMPDeclareMapperDecl>(D);
- assert(DMD && "Expect valid OMPDeclareMapperDecl during instantiation.");
- Lookups.back().addDecl(DMD);
- }
- }
- // Defer the lookup for dependent types. The results will be passed through
- // UnresolvedMapper on instantiation.
- if (SemaRef.CurContext->isDependentContext() || Type->isDependentType() ||
- Type->isInstantiationDependentType() ||
- Type->containsUnexpandedParameterPack() ||
- filterLookupForUDReductionAndMapper<bool>(Lookups, [](ValueDecl *D) {
- return !D->isInvalidDecl() &&
- (D->getType()->isDependentType() ||
- D->getType()->isInstantiationDependentType() ||
- D->getType()->containsUnexpandedParameterPack());
- })) {
- UnresolvedSet<8> URS;
- for (const UnresolvedSet<8> &Set : Lookups) {
- if (Set.empty())
- continue;
- URS.append(Set.begin(), Set.end());
- }
- return UnresolvedLookupExpr::Create(
- SemaRef.Context, /*NamingClass=*/nullptr,
- MapperIdScopeSpec.getWithLocInContext(SemaRef.Context), MapperId,
- /*ADL=*/false, /*Overloaded=*/true, URS.begin(), URS.end());
- }
- SourceLocation Loc = MapperId.getLoc();
- // [OpenMP 5.0], 2.19.7.3 declare mapper Directive, Restrictions
- // The type must be of struct, union or class type in C and C++
- if (!Type->isStructureOrClassType() && !Type->isUnionType() &&
- (MapperIdScopeSpec.isSet() || MapperId.getAsString() != "default")) {
- SemaRef.Diag(Loc, diag::err_omp_mapper_wrong_type);
- return ExprError();
- }
- // Perform argument dependent lookup.
- if (SemaRef.getLangOpts().CPlusPlus && !MapperIdScopeSpec.isSet())
- argumentDependentLookup(SemaRef, MapperId, Loc, Type, Lookups);
- // Return the first user-defined mapper with the desired type.
- if (auto *VD = filterLookupForUDReductionAndMapper<ValueDecl *>(
- Lookups, [&SemaRef, Type](ValueDecl *D) -> ValueDecl * {
- if (!D->isInvalidDecl() &&
- SemaRef.Context.hasSameType(D->getType(), Type))
- return D;
- return nullptr;
- }))
- return SemaRef.BuildDeclRefExpr(VD, Type, VK_LValue, Loc);
- // Find the first user-defined mapper with a type derived from the desired
- // type.
- if (auto *VD = filterLookupForUDReductionAndMapper<ValueDecl *>(
- Lookups, [&SemaRef, Type, Loc](ValueDecl *D) -> ValueDecl * {
- if (!D->isInvalidDecl() &&
- SemaRef.IsDerivedFrom(Loc, Type, D->getType()) &&
- !Type.isMoreQualifiedThan(D->getType()))
- return D;
- return nullptr;
- })) {
- CXXBasePaths Paths(/*FindAmbiguities=*/true, /*RecordPaths=*/true,
- /*DetectVirtual=*/false);
- if (SemaRef.IsDerivedFrom(Loc, Type, VD->getType(), Paths)) {
- if (!Paths.isAmbiguous(SemaRef.Context.getCanonicalType(
- VD->getType().getUnqualifiedType()))) {
- if (SemaRef.CheckBaseClassAccess(
- Loc, VD->getType(), Type, Paths.front(),
- /*DiagID=*/0) != Sema::AR_inaccessible) {
- return SemaRef.BuildDeclRefExpr(VD, Type, VK_LValue, Loc);
- }
- }
- }
- }
- // Report error if a mapper is specified, but cannot be found.
- if (MapperIdScopeSpec.isSet() || MapperId.getAsString() != "default") {
- SemaRef.Diag(Loc, diag::err_omp_invalid_mapper)
- << Type << MapperId.getName();
- return ExprError();
- }
- return ExprEmpty();
- }
- 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;
- // The reference to the user-defined mapper associated with every expression.
- SmallVector<Expr *, 16> UDMapperList;
- 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, mappable expression
- // components, variables, and user-defined mappers are extracted and used to
- // fill \a ProcessedVarList, \a VarComponents, \a VarBaseDeclarations, and \a
- // UDMapperList in MVLI. \a MapType, \a IsMapTypeImplicit, \a MapperIdScopeSpec,
- // and \a MapperId are expected to be valid if the clause kind is 'map'.
- static void checkMappableExpressionList(
- Sema &SemaRef, DSAStackTy *DSAS, OpenMPClauseKind CKind,
- MappableVarListInfo &MVLI, SourceLocation StartLoc,
- CXXScopeSpec &MapperIdScopeSpec, DeclarationNameInfo MapperId,
- ArrayRef<Expr *> UnresolvedMappers,
- 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!");
- // If the identifier of user-defined mapper is not specified, it is "default".
- // We do not change the actual name in this clause to distinguish whether a
- // mapper is specified explicitly, i.e., it is not explicitly specified when
- // MapperId.getName() is empty.
- if (!MapperId.getName() || MapperId.getName().isEmpty()) {
- auto &DeclNames = SemaRef.getASTContext().DeclarationNames;
- MapperId.setName(DeclNames.getIdentifier(
- &SemaRef.getASTContext().Idents.get("default")));
- }
- // Iterators to find the current unresolved mapper expression.
- auto UMIt = UnresolvedMappers.begin(), UMEnd = UnresolvedMappers.end();
- bool UpdateUMIt = false;
- Expr *UnresolvedMapper = nullptr;
- // 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();
- // Find the current unresolved mapper expression.
- if (UpdateUMIt && UMIt != UMEnd) {
- UMIt++;
- assert(
- UMIt != UMEnd &&
- "Expect the size of UnresolvedMappers to match with that of VarList");
- }
- UpdateUMIt = true;
- if (UMIt != UMEnd)
- UnresolvedMapper = *UMIt;
- const Expr *VE = RE->IgnoreParenLValueCasts();
- if (VE->isValueDependent() || VE->isTypeDependent() ||
- VE->isInstantiationDependent() ||
- VE->containsUnexpandedParameterPack()) {
- // Try to find the associated user-defined mapper.
- ExprResult ER = buildUserDefinedMapperRef(
- SemaRef, DSAS->getCurScope(), MapperIdScopeSpec, MapperId,
- VE->getType().getCanonicalType(), UnresolvedMapper);
- if (ER.isInvalid())
- continue;
- MVLI.UDMapperList.push_back(ER.get());
- // 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.");
- if (const auto *TE = dyn_cast<CXXThisExpr>(BE)) {
- // Add store "this" pointer to class in DSAStackTy for future checking
- DSAS->addMappedClassesQualTypes(TE->getType());
- // Try to find the associated user-defined mapper.
- ExprResult ER = buildUserDefinedMapperRef(
- SemaRef, DSAS->getCurScope(), MapperIdScopeSpec, MapperId,
- VE->getType().getCanonicalType(), UnresolvedMapper);
- if (ER.isInvalid())
- continue;
- MVLI.UDMapperList.push_back(ER.get());
- // Skip restriction checking for variable or field declarations
- MVLI.ProcessedVarList.push_back(RE);
- MVLI.VarComponents.resize(MVLI.VarComponents.size() + 1);
- MVLI.VarComponents.back().append(CurComponents.begin(),
- CurComponents.end());
- MVLI.VarBaseDeclarations.push_back(nullptr);
- continue;
- }
- // 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
- //
- // OpenMP 5.0 [2.19.7.1, Restrictions, p.7]
- // A list item cannot appear in both a map clause and a data-sharing
- // attribute clause on the same construct unless the construct is a
- // combined construct.
- if (VD && ((SemaRef.LangOpts.OpenMP <= 45 &&
- isOpenMPTargetExecutionDirective(DKind)) ||
- DKind == OMPD_target)) {
- 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;
- }
- }
- }
- // Try to find the associated user-defined mapper.
- ExprResult ER = buildUserDefinedMapperRef(
- SemaRef, DSAS->getCurScope(), MapperIdScopeSpec, MapperId,
- Type.getCanonicalType(), UnresolvedMapper);
- if (ER.isInvalid())
- continue;
- MVLI.UDMapperList.push_back(ER.get());
- // 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,
- CXXScopeSpec &MapperIdScopeSpec, DeclarationNameInfo &MapperId,
- OpenMPMapClauseKind MapType, bool IsMapTypeImplicit, SourceLocation MapLoc,
- SourceLocation ColonLoc, ArrayRef<Expr *> VarList,
- const OMPVarListLocTy &Locs, ArrayRef<Expr *> UnresolvedMappers) {
- OpenMPMapModifierKind Modifiers[] = {OMPC_MAP_MODIFIER_unknown,
- 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;
- }
- MappableVarListInfo MVLI(VarList);
- checkMappableExpressionList(*this, DSAStack, OMPC_map, MVLI, Locs.StartLoc,
- MapperIdScopeSpec, MapperId, UnresolvedMappers,
- MapType, IsMapTypeImplicit);
- // 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, Locs, MVLI.ProcessedVarList,
- MVLI.VarBaseDeclarations, MVLI.VarComponents,
- MVLI.UDMapperList, Modifiers, ModifiersLoc,
- MapperIdScopeSpec.getWithLocInContext(Context),
- MapperId, 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;
- }
- TypeResult Sema::ActOnOpenMPDeclareMapperVarDecl(Scope *S, Declarator &D) {
- TypeSourceInfo *TInfo = GetTypeForDeclarator(D, S);
- QualType T = TInfo->getType();
- if (D.isInvalidType())
- return true;
- if (getLangOpts().CPlusPlus) {
- // Check that there are no default arguments (C++ only).
- CheckExtraCXXDefaultArguments(D);
- }
- return CreateParsedType(T, TInfo);
- }
- QualType Sema::ActOnOpenMPDeclareMapperType(SourceLocation TyLoc,
- TypeResult ParsedType) {
- assert(ParsedType.isUsable() && "Expect usable parsed mapper type");
- QualType MapperType = GetTypeFromParser(ParsedType.get());
- assert(!MapperType.isNull() && "Expect valid mapper type");
- // [OpenMP 5.0], 2.19.7.3 declare mapper Directive, Restrictions
- // The type must be of struct, union or class type in C and C++
- if (!MapperType->isStructureOrClassType() && !MapperType->isUnionType()) {
- Diag(TyLoc, diag::err_omp_mapper_wrong_type);
- return QualType();
- }
- return MapperType;
- }
- OMPDeclareMapperDecl *Sema::ActOnOpenMPDeclareMapperDirectiveStart(
- Scope *S, DeclContext *DC, DeclarationName Name, QualType MapperType,
- SourceLocation StartLoc, DeclarationName VN, AccessSpecifier AS,
- Decl *PrevDeclInScope) {
- LookupResult Lookup(*this, Name, SourceLocation(), LookupOMPMapperName,
- forRedeclarationInCurContext());
- // [OpenMP 5.0], 2.19.7.3 declare mapper Directive, Restrictions
- // A mapper-identifier may not be redeclared 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;
- OMPDeclareMapperDecl *PrevDMD = 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<OMPDeclareMapperDecl *, bool> UsedAsPrevious;
- LookupResult::Filter Filter = Lookup.makeFilter();
- while (Filter.hasNext()) {
- auto *PrevDecl = cast<OMPDeclareMapperDecl>(Filter.next());
- if (InCompoundScope) {
- auto I = UsedAsPrevious.find(PrevDecl);
- if (I == UsedAsPrevious.end())
- UsedAsPrevious[PrevDecl] = false;
- if (OMPDeclareMapperDecl *D = PrevDecl->getPrevDeclInScope())
- UsedAsPrevious[D] = true;
- }
- PreviousRedeclTypes[PrevDecl->getType().getCanonicalType()] =
- PrevDecl->getLocation();
- }
- Filter.done();
- if (InCompoundScope) {
- for (const auto &PrevData : UsedAsPrevious) {
- if (!PrevData.second) {
- PrevDMD = PrevData.first;
- break;
- }
- }
- }
- } else if (PrevDeclInScope) {
- auto *PrevDMDInScope = PrevDMD =
- cast<OMPDeclareMapperDecl>(PrevDeclInScope);
- do {
- PreviousRedeclTypes[PrevDMDInScope->getType().getCanonicalType()] =
- PrevDMDInScope->getLocation();
- PrevDMDInScope = PrevDMDInScope->getPrevDeclInScope();
- } while (PrevDMDInScope != nullptr);
- }
- const auto I = PreviousRedeclTypes.find(MapperType.getCanonicalType());
- bool Invalid = false;
- if (I != PreviousRedeclTypes.end()) {
- Diag(StartLoc, diag::err_omp_declare_mapper_redefinition)
- << MapperType << Name;
- Diag(I->second, diag::note_previous_definition);
- Invalid = true;
- }
- auto *DMD = OMPDeclareMapperDecl::Create(Context, DC, StartLoc, Name,
- MapperType, VN, PrevDMD);
- DC->addDecl(DMD);
- DMD->setAccess(AS);
- if (Invalid)
- DMD->setInvalidDecl();
- // Enter new function scope.
- PushFunctionScope();
- setFunctionHasBranchProtectedScope();
- CurContext = DMD;
- return DMD;
- }
- void Sema::ActOnOpenMPDeclareMapperDirectiveVarDecl(OMPDeclareMapperDecl *DMD,
- Scope *S,
- QualType MapperType,
- SourceLocation StartLoc,
- DeclarationName VN) {
- VarDecl *VD = buildVarDecl(*this, StartLoc, MapperType, VN.getAsString());
- if (S)
- PushOnScopeChains(VD, S);
- else
- DMD->addDecl(VD);
- Expr *MapperVarRefExpr = buildDeclRefExpr(*this, VD, MapperType, StartLoc);
- DMD->setMapperVarRef(MapperVarRefExpr);
- }
- Sema::DeclGroupPtrTy
- Sema::ActOnOpenMPDeclareMapperDirectiveEnd(OMPDeclareMapperDecl *D, Scope *S,
- ArrayRef<OMPClause *> ClauseList) {
- PopDeclContext();
- PopFunctionScopeInfo();
- if (D) {
- if (S)
- PushOnScopeChains(D, S, /*AddToContext=*/false);
- D->CreateClauses(Context, ClauseList);
- }
- return DeclGroupPtrTy::make(DeclGroupRef(D));
- }
- 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;
- }
- NamedDecl *
- Sema::lookupOpenMPDeclareTargetName(Scope *CurScope, CXXScopeSpec &ScopeSpec,
- const DeclarationNameInfo &Id,
- NamedDeclSetType &SameDirectiveDecls) {
- LookupResult Lookup(*this, Id, LookupOrdinaryName);
- LookupParsedName(Lookup, CurScope, &ScopeSpec, true);
- if (Lookup.isAmbiguous())
- return nullptr;
- Lookup.suppressDiagnostics();
- if (!Lookup.isSingleResult()) {
- VarOrFuncDeclFilterCCC CCC(*this);
- if (TypoCorrection Corrected =
- CorrectTypo(Id, LookupOrdinaryName, CurScope, nullptr, CCC,
- CTK_ErrorRecovery)) {
- diagnoseTypo(Corrected, PDiag(diag::err_undeclared_var_use_suggest)
- << Id.getName());
- checkDeclIsAllowedInOpenMPTarget(nullptr, Corrected.getCorrectionDecl());
- return nullptr;
- }
- Diag(Id.getLoc(), diag::err_undeclared_var_use) << Id.getName();
- return nullptr;
- }
- NamedDecl *ND = Lookup.getAsSingle<NamedDecl>();
- if (!isa<VarDecl>(ND) && !isa<FunctionDecl>(ND) &&
- !isa<FunctionTemplateDecl>(ND)) {
- Diag(Id.getLoc(), diag::err_omp_invalid_target_decl) << Id.getName();
- return nullptr;
- }
- if (!SameDirectiveDecls.insert(cast<NamedDecl>(ND->getCanonicalDecl())))
- Diag(Id.getLoc(), diag::err_omp_declare_target_multiple) << Id.getName();
- return ND;
- }
- void Sema::ActOnOpenMPDeclareTargetName(
- NamedDecl *ND, SourceLocation Loc, OMPDeclareTargetDeclAttr::MapTypeTy MT,
- OMPDeclareTargetDeclAttr::DevTypeTy DT) {
- assert((isa<VarDecl>(ND) || isa<FunctionDecl>(ND) ||
- isa<FunctionTemplateDecl>(ND)) &&
- "Expected variable, function or function template.");
- // Diagnose marking after use as it may lead to incorrect diagnosis and
- // codegen.
- if (LangOpts.OpenMP >= 50 &&
- (ND->isUsed(/*CheckUsedAttr=*/false) || ND->isReferenced()))
- Diag(Loc, diag::warn_omp_declare_target_after_first_use);
- Optional<OMPDeclareTargetDeclAttr::DevTypeTy> DevTy =
- OMPDeclareTargetDeclAttr::getDeviceType(cast<ValueDecl>(ND));
- if (DevTy.hasValue() && *DevTy != DT) {
- Diag(Loc, diag::err_omp_device_type_mismatch)
- << OMPDeclareTargetDeclAttr::ConvertDevTypeTyToStr(DT)
- << OMPDeclareTargetDeclAttr::ConvertDevTypeTyToStr(*DevTy);
- return;
- }
- Optional<OMPDeclareTargetDeclAttr::MapTypeTy> Res =
- OMPDeclareTargetDeclAttr::isDeclareTargetDeclaration(cast<ValueDecl>(ND));
- if (!Res) {
- auto *A = OMPDeclareTargetDeclAttr::CreateImplicit(Context, MT, DT,
- SourceRange(Loc, Loc));
- ND->addAttr(A);
- if (ASTMutationListener *ML = Context.getASTMutationListener())
- ML->DeclarationMarkedOpenMPDeclareTarget(ND, A);
- checkDeclIsAllowedInOpenMPTarget(nullptr, ND, Loc);
- } else if (*Res != MT) {
- Diag(Loc, diag::err_omp_declare_target_to_and_link) << ND;
- }
- }
- static void checkDeclInTargetContext(SourceLocation SL, SourceRange SR,
- Sema &SemaRef, Decl *D) {
- if (!D || !isa<VarDecl>(D))
- return;
- auto *VD = cast<VarDecl>(D);
- Optional<OMPDeclareTargetDeclAttr::MapTypeTy> MapTy =
- OMPDeclareTargetDeclAttr::isDeclareTargetDeclaration(VD);
- if (SemaRef.LangOpts.OpenMP >= 50 &&
- (SemaRef.getCurLambda(/*IgnoreNonLambdaCapturingScope=*/true) ||
- SemaRef.getCurBlock() || SemaRef.getCurCapturedRegion()) &&
- VD->hasGlobalStorage()) {
- llvm::Optional<OMPDeclareTargetDeclAttr::MapTypeTy> MapTy =
- OMPDeclareTargetDeclAttr::isDeclareTargetDeclaration(VD);
- if (!MapTy || *MapTy != OMPDeclareTargetDeclAttr::MT_To) {
- // OpenMP 5.0, 2.12.7 declare target Directive, Restrictions
- // If a lambda declaration and definition appears between a
- // declare target directive and the matching end declare target
- // directive, all variables that are captured by the lambda
- // expression must also appear in a to clause.
- SemaRef.Diag(VD->getLocation(),
- diag::err_omp_lambda_capture_in_declare_target_not_to);
- SemaRef.Diag(SL, diag::note_var_explicitly_captured_here)
- << VD << 0 << SR;
- return;
- }
- }
- if (MapTy.hasValue())
- 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 OMPDeclareTargetDeclAttr::isDeclareTargetDeclaration(VD) ||
- 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 (auto *FD = dyn_cast<FunctionDecl>(D)) {
- llvm::Optional<OMPDeclareTargetDeclAttr::MapTypeTy> Res =
- OMPDeclareTargetDeclAttr::isDeclareTargetDeclaration(FD);
- if (IdLoc.isValid() && Res && *Res == OMPDeclareTargetDeclAttr::MT_Link) {
- Diag(IdLoc, diag::err_omp_function_in_link_clause);
- Diag(FD->getLocation(), diag::note_defined_here) << FD;
- return;
- }
- // Mark the function as must be emitted for the device.
- Optional<OMPDeclareTargetDeclAttr::DevTypeTy> DevTy =
- OMPDeclareTargetDeclAttr::getDeviceType(FD);
- if (LangOpts.OpenMPIsDevice && Res.hasValue() && IdLoc.isValid() &&
- *DevTy != OMPDeclareTargetDeclAttr::DT_Host)
- checkOpenMPDeviceFunction(IdLoc, FD, /*CheckForDelayedContext=*/false);
- if (!LangOpts.OpenMPIsDevice && Res.hasValue() && IdLoc.isValid() &&
- *DevTy != OMPDeclareTargetDeclAttr::DT_NoHost)
- checkOpenMPHostFunction(IdLoc, FD, /*CheckCaller=*/false);
- }
- 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,
- OMPDeclareTargetDeclAttr::DT_Any, SourceRange(IdLoc, IdLoc));
- 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,
- CXXScopeSpec &MapperIdScopeSpec,
- DeclarationNameInfo &MapperId,
- const OMPVarListLocTy &Locs,
- ArrayRef<Expr *> UnresolvedMappers) {
- MappableVarListInfo MVLI(VarList);
- checkMappableExpressionList(*this, DSAStack, OMPC_to, MVLI, Locs.StartLoc,
- MapperIdScopeSpec, MapperId, UnresolvedMappers);
- if (MVLI.ProcessedVarList.empty())
- return nullptr;
- return OMPToClause::Create(
- Context, Locs, MVLI.ProcessedVarList, MVLI.VarBaseDeclarations,
- MVLI.VarComponents, MVLI.UDMapperList,
- MapperIdScopeSpec.getWithLocInContext(Context), MapperId);
- }
- OMPClause *Sema::ActOnOpenMPFromClause(ArrayRef<Expr *> VarList,
- CXXScopeSpec &MapperIdScopeSpec,
- DeclarationNameInfo &MapperId,
- const OMPVarListLocTy &Locs,
- ArrayRef<Expr *> UnresolvedMappers) {
- MappableVarListInfo MVLI(VarList);
- checkMappableExpressionList(*this, DSAStack, OMPC_from, MVLI, Locs.StartLoc,
- MapperIdScopeSpec, MapperId, UnresolvedMappers);
- if (MVLI.ProcessedVarList.empty())
- return nullptr;
- return OMPFromClause::Create(
- Context, Locs, MVLI.ProcessedVarList, MVLI.VarBaseDeclarations,
- MVLI.VarComponents, MVLI.UDMapperList,
- MapperIdScopeSpec.getWithLocInContext(Context), MapperId);
- }
- OMPClause *Sema::ActOnOpenMPUseDevicePtrClause(ArrayRef<Expr *> VarList,
- const OMPVarListLocTy &Locs) {
- 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, Locs, MVLI.ProcessedVarList, PrivateCopies, Inits,
- MVLI.VarBaseDeclarations, MVLI.VarComponents);
- }
- OMPClause *Sema::ActOnOpenMPIsDevicePtrClause(ArrayRef<Expr *> VarList,
- const OMPVarListLocTy &Locs) {
- 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, Locs, MVLI.ProcessedVarList,
- MVLI.VarBaseDeclarations,
- MVLI.VarComponents);
- }
- OMPClause *Sema::ActOnOpenMPAllocateClause(
- Expr *Allocator, ArrayRef<Expr *> VarList, SourceLocation StartLoc,
- SourceLocation ColonLoc, SourceLocation LParenLoc, SourceLocation EndLoc) {
- if (Allocator) {
- // OpenMP [2.11.4 allocate Clause, Description]
- // allocator is an expression of omp_allocator_handle_t type.
- if (!findOMPAllocatorHandleT(*this, Allocator->getExprLoc(), DSAStack))
- return nullptr;
- ExprResult AllocatorRes = DefaultLvalueConversion(Allocator);
- if (AllocatorRes.isInvalid())
- return nullptr;
- AllocatorRes = PerformImplicitConversion(AllocatorRes.get(),
- DSAStack->getOMPAllocatorHandleT(),
- Sema::AA_Initializing,
- /*AllowExplicit=*/true);
- if (AllocatorRes.isInvalid())
- return nullptr;
- Allocator = AllocatorRes.get();
- } else {
- // OpenMP 5.0, 2.11.4 allocate Clause, Restrictions.
- // allocate clauses that appear on a target construct or on constructs in a
- // target region must specify an allocator expression unless a requires
- // directive with the dynamic_allocators clause is present in the same
- // compilation unit.
- if (LangOpts.OpenMPIsDevice &&
- !DSAStack->hasRequiresDeclWithClause<OMPDynamicAllocatorsClause>())
- targetDiag(StartLoc, diag::err_expected_allocator_expression);
- }
- // Analyze and build list of variables.
- SmallVector<Expr *, 8> Vars;
- 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);
- }
- ValueDecl *D = Res.first;
- if (!D)
- continue;
- auto *VD = dyn_cast<VarDecl>(D);
- DeclRefExpr *Ref = nullptr;
- if (!VD && !CurContext->isDependentContext())
- Ref = buildCapture(*this, D, SimpleRefExpr, /*WithInit=*/false);
- Vars.push_back((VD || CurContext->isDependentContext())
- ? RefExpr->IgnoreParens()
- : Ref);
- }
- if (Vars.empty())
- return nullptr;
- return OMPAllocateClause::Create(Context, StartLoc, LParenLoc, Allocator,
- ColonLoc, EndLoc, Vars);
- }
|