1 //===--- SemaOpenMP.cpp - Semantic Analysis for OpenMP constructs ---------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 /// \file
10 /// \brief This file implements semantic analysis for OpenMP directives and
11 /// clauses.
12 ///
13 //===----------------------------------------------------------------------===//
14 
15 #include "TreeTransform.h"
16 #include "clang/AST/ASTContext.h"
17 #include "clang/AST/ASTMutationListener.h"
18 #include "clang/AST/Decl.h"
19 #include "clang/AST/DeclCXX.h"
20 #include "clang/AST/DeclOpenMP.h"
21 #include "clang/AST/StmtCXX.h"
22 #include "clang/AST/StmtOpenMP.h"
23 #include "clang/AST/StmtVisitor.h"
24 #include "clang/Basic/OpenMPKinds.h"
25 #include "clang/Basic/TargetInfo.h"
26 #include "clang/Lex/Preprocessor.h"
27 #include "clang/Sema/Initialization.h"
28 #include "clang/Sema/Lookup.h"
29 #include "clang/Sema/Scope.h"
30 #include "clang/Sema/ScopeInfo.h"
31 #include "clang/Sema/SemaInternal.h"
32 using namespace clang;
33 
34 //===----------------------------------------------------------------------===//
35 // Stack of data-sharing attributes for variables
36 //===----------------------------------------------------------------------===//
37 
38 namespace {
39 /// \brief Default data sharing attributes, which can be applied to directive.
40 enum DefaultDataSharingAttributes {
41   DSA_unspecified = 0, /// \brief Data sharing attribute not specified.
42   DSA_none = 1 << 0,   /// \brief Default data sharing attribute 'none'.
43   DSA_shared = 1 << 1  /// \brief Default data sharing attribute 'shared'.
44 };
45 
46 template <class T> struct MatchesAny {
MatchesAny__anon216cbd260111::MatchesAny47   explicit MatchesAny(ArrayRef<T> Arr) : Arr(std::move(Arr)) {}
operator ()__anon216cbd260111::MatchesAny48   bool operator()(T Kind) {
49     for (auto KindEl : Arr)
50       if (KindEl == Kind)
51         return true;
52     return false;
53   }
54 
55 private:
56   ArrayRef<T> Arr;
57 };
58 struct MatchesAlways {
MatchesAlways__anon216cbd260111::MatchesAlways59   MatchesAlways() {}
operator ()__anon216cbd260111::MatchesAlways60   template <class T> bool operator()(T) { return true; }
61 };
62 
63 typedef MatchesAny<OpenMPClauseKind> MatchesAnyClause;
64 typedef MatchesAny<OpenMPDirectiveKind> MatchesAnyDirective;
65 
66 /// \brief Stack for tracking declarations used in OpenMP directives and
67 /// clauses and their data-sharing attributes.
68 class DSAStackTy {
69 public:
70   struct DSAVarData {
71     OpenMPDirectiveKind DKind;
72     OpenMPClauseKind CKind;
73     DeclRefExpr *RefExpr;
74     SourceLocation ImplicitDSALoc;
DSAVarData__anon216cbd260111::DSAStackTy::DSAVarData75     DSAVarData()
76         : DKind(OMPD_unknown), CKind(OMPC_unknown), RefExpr(nullptr),
77           ImplicitDSALoc() {}
78   };
79 
80 public:
81   struct MapInfo {
82     Expr *RefExpr;
83   };
84 
85 private:
86   struct DSAInfo {
87     OpenMPClauseKind Attributes;
88     DeclRefExpr *RefExpr;
89   };
90   typedef llvm::SmallDenseMap<VarDecl *, DSAInfo, 64> DeclSAMapTy;
91   typedef llvm::SmallDenseMap<VarDecl *, DeclRefExpr *, 64> AlignedMapTy;
92   typedef llvm::DenseSet<VarDecl *> LoopControlVariablesSetTy;
93   typedef llvm::SmallDenseMap<VarDecl *, MapInfo, 64> MappedDeclsTy;
94   typedef llvm::StringMap<std::pair<OMPCriticalDirective *, llvm::APSInt>>
95       CriticalsWithHintsTy;
96 
97   struct SharingMapTy {
98     DeclSAMapTy SharingMap;
99     AlignedMapTy AlignedMap;
100     MappedDeclsTy MappedDecls;
101     LoopControlVariablesSetTy LCVSet;
102     DefaultDataSharingAttributes DefaultAttr;
103     SourceLocation DefaultAttrLoc;
104     OpenMPDirectiveKind Directive;
105     DeclarationNameInfo DirectiveName;
106     Scope *CurScope;
107     SourceLocation ConstructLoc;
108     /// \brief first argument (Expr *) contains optional argument of the
109     /// 'ordered' clause, the second one is true if the regions has 'ordered'
110     /// clause, false otherwise.
111     llvm::PointerIntPair<Expr *, 1, bool> OrderedRegion;
112     bool NowaitRegion;
113     bool CancelRegion;
114     unsigned CollapseNumber;
115     SourceLocation InnerTeamsRegionLoc;
SharingMapTy__anon216cbd260111::DSAStackTy::SharingMapTy116     SharingMapTy(OpenMPDirectiveKind DKind, DeclarationNameInfo Name,
117                  Scope *CurScope, SourceLocation Loc)
118         : SharingMap(), AlignedMap(), LCVSet(), DefaultAttr(DSA_unspecified),
119           Directive(DKind), DirectiveName(std::move(Name)), CurScope(CurScope),
120           ConstructLoc(Loc), OrderedRegion(), NowaitRegion(false),
121           CancelRegion(false), CollapseNumber(1), InnerTeamsRegionLoc() {}
SharingMapTy__anon216cbd260111::DSAStackTy::SharingMapTy122     SharingMapTy()
123         : SharingMap(), AlignedMap(), LCVSet(), DefaultAttr(DSA_unspecified),
124           Directive(OMPD_unknown), DirectiveName(), CurScope(nullptr),
125           ConstructLoc(), OrderedRegion(), NowaitRegion(false),
126           CancelRegion(false), CollapseNumber(1), InnerTeamsRegionLoc() {}
127   };
128 
129   typedef SmallVector<SharingMapTy, 64> StackTy;
130 
131   /// \brief Stack of used declaration and their data-sharing attributes.
132   StackTy Stack;
133   /// \brief true, if check for DSA must be from parent directive, false, if
134   /// from current directive.
135   OpenMPClauseKind ClauseKindMode;
136   Sema &SemaRef;
137   bool ForceCapturing;
138   CriticalsWithHintsTy Criticals;
139 
140   typedef SmallVector<SharingMapTy, 8>::reverse_iterator reverse_iterator;
141 
142   DSAVarData getDSA(StackTy::reverse_iterator Iter, VarDecl *D);
143 
144   /// \brief Checks if the variable is a local for OpenMP region.
145   bool isOpenMPLocal(VarDecl *D, StackTy::reverse_iterator Iter);
146 
147 public:
DSAStackTy(Sema & S)148   explicit DSAStackTy(Sema &S)
149       : Stack(1), ClauseKindMode(OMPC_unknown), SemaRef(S),
150         ForceCapturing(false) {}
151 
isClauseParsingMode() const152   bool isClauseParsingMode() const { return ClauseKindMode != OMPC_unknown; }
setClauseParsingMode(OpenMPClauseKind K)153   void setClauseParsingMode(OpenMPClauseKind K) { ClauseKindMode = K; }
154 
isForceVarCapturing() const155   bool isForceVarCapturing() const { return ForceCapturing; }
setForceVarCapturing(bool V)156   void setForceVarCapturing(bool V) { ForceCapturing = V; }
157 
push(OpenMPDirectiveKind DKind,const DeclarationNameInfo & DirName,Scope * CurScope,SourceLocation Loc)158   void push(OpenMPDirectiveKind DKind, const DeclarationNameInfo &DirName,
159             Scope *CurScope, SourceLocation Loc) {
160     Stack.push_back(SharingMapTy(DKind, DirName, CurScope, Loc));
161     Stack.back().DefaultAttrLoc = Loc;
162   }
163 
pop()164   void pop() {
165     assert(Stack.size() > 1 && "Data-sharing attributes stack is empty!");
166     Stack.pop_back();
167   }
168 
addCriticalWithHint(OMPCriticalDirective * D,llvm::APSInt Hint)169   void addCriticalWithHint(OMPCriticalDirective *D, llvm::APSInt Hint) {
170     Criticals[D->getDirectiveName().getAsString()] = std::make_pair(D, Hint);
171   }
172   const std::pair<OMPCriticalDirective *, llvm::APSInt>
getCriticalWithHint(const DeclarationNameInfo & Name) const173   getCriticalWithHint(const DeclarationNameInfo &Name) const {
174     auto I = Criticals.find(Name.getAsString());
175     if (I != Criticals.end())
176       return I->second;
177     return std::make_pair(nullptr, llvm::APSInt());
178   }
179   /// \brief If 'aligned' declaration for given variable \a D was not seen yet,
180   /// add it and return NULL; otherwise return previous occurrence's expression
181   /// for diagnostics.
182   DeclRefExpr *addUniqueAligned(VarDecl *D, DeclRefExpr *NewDE);
183 
184   /// \brief Register specified variable as loop control variable.
185   void addLoopControlVariable(VarDecl *D);
186   /// \brief Check if the specified variable is a loop control variable for
187   /// current region.
188   bool isLoopControlVariable(VarDecl *D);
189 
190   /// \brief Adds explicit data sharing attribute to the specified declaration.
191   void addDSA(VarDecl *D, DeclRefExpr *E, OpenMPClauseKind A);
192 
193   /// \brief Returns data sharing attributes from top of the stack for the
194   /// specified declaration.
195   DSAVarData getTopDSA(VarDecl *D, bool FromParent);
196   /// \brief Returns data-sharing attributes for the specified declaration.
197   DSAVarData getImplicitDSA(VarDecl *D, bool FromParent);
198   /// \brief Checks if the specified variables has data-sharing attributes which
199   /// match specified \a CPred predicate in any directive which matches \a DPred
200   /// predicate.
201   template <class ClausesPredicate, class DirectivesPredicate>
202   DSAVarData hasDSA(VarDecl *D, ClausesPredicate CPred,
203                     DirectivesPredicate DPred, bool FromParent);
204   /// \brief Checks if the specified variables has data-sharing attributes which
205   /// match specified \a CPred predicate in any innermost directive which
206   /// matches \a DPred predicate.
207   template <class ClausesPredicate, class DirectivesPredicate>
208   DSAVarData hasInnermostDSA(VarDecl *D, ClausesPredicate CPred,
209                              DirectivesPredicate DPred,
210                              bool FromParent);
211   /// \brief Checks if the specified variables has explicit data-sharing
212   /// attributes which match specified \a CPred predicate at the specified
213   /// OpenMP region.
214   bool hasExplicitDSA(VarDecl *D,
215                       const llvm::function_ref<bool(OpenMPClauseKind)> &CPred,
216                       unsigned Level);
217 
218   /// \brief Returns true if the directive at level \Level matches in the
219   /// specified \a DPred predicate.
220   bool hasExplicitDirective(
221       const llvm::function_ref<bool(OpenMPDirectiveKind)> &DPred,
222       unsigned Level);
223 
224   /// \brief Finds a directive which matches specified \a DPred predicate.
225   template <class NamedDirectivesPredicate>
226   bool hasDirective(NamedDirectivesPredicate DPred, bool FromParent);
227 
228   /// \brief Returns currently analyzed directive.
getCurrentDirective() const229   OpenMPDirectiveKind getCurrentDirective() const {
230     return Stack.back().Directive;
231   }
232   /// \brief Returns parent directive.
getParentDirective() const233   OpenMPDirectiveKind getParentDirective() const {
234     if (Stack.size() > 2)
235       return Stack[Stack.size() - 2].Directive;
236     return OMPD_unknown;
237   }
238   /// \brief Return the directive associated with the provided scope.
239   OpenMPDirectiveKind getDirectiveForScope(const Scope *S) const;
240 
241   /// \brief Set default data sharing attribute to none.
setDefaultDSANone(SourceLocation Loc)242   void setDefaultDSANone(SourceLocation Loc) {
243     Stack.back().DefaultAttr = DSA_none;
244     Stack.back().DefaultAttrLoc = Loc;
245   }
246   /// \brief Set default data sharing attribute to shared.
setDefaultDSAShared(SourceLocation Loc)247   void setDefaultDSAShared(SourceLocation Loc) {
248     Stack.back().DefaultAttr = DSA_shared;
249     Stack.back().DefaultAttrLoc = Loc;
250   }
251 
getDefaultDSA() const252   DefaultDataSharingAttributes getDefaultDSA() const {
253     return Stack.back().DefaultAttr;
254   }
getDefaultDSALocation() const255   SourceLocation getDefaultDSALocation() const {
256     return Stack.back().DefaultAttrLoc;
257   }
258 
259   /// \brief Checks if the specified variable is a threadprivate.
isThreadPrivate(VarDecl * D)260   bool isThreadPrivate(VarDecl *D) {
261     DSAVarData DVar = getTopDSA(D, false);
262     return isOpenMPThreadPrivate(DVar.CKind);
263   }
264 
265   /// \brief Marks current region as ordered (it has an 'ordered' clause).
setOrderedRegion(bool IsOrdered,Expr * Param)266   void setOrderedRegion(bool IsOrdered, Expr *Param) {
267     Stack.back().OrderedRegion.setInt(IsOrdered);
268     Stack.back().OrderedRegion.setPointer(Param);
269   }
270   /// \brief Returns true, if parent region is ordered (has associated
271   /// 'ordered' clause), false - otherwise.
isParentOrderedRegion() const272   bool isParentOrderedRegion() const {
273     if (Stack.size() > 2)
274       return Stack[Stack.size() - 2].OrderedRegion.getInt();
275     return false;
276   }
277   /// \brief Returns optional parameter for the ordered region.
getParentOrderedRegionParam() const278   Expr *getParentOrderedRegionParam() const {
279     if (Stack.size() > 2)
280       return Stack[Stack.size() - 2].OrderedRegion.getPointer();
281     return nullptr;
282   }
283   /// \brief Marks current region as nowait (it has a 'nowait' clause).
setNowaitRegion(bool IsNowait=true)284   void setNowaitRegion(bool IsNowait = true) {
285     Stack.back().NowaitRegion = IsNowait;
286   }
287   /// \brief Returns true, if parent region is nowait (has associated
288   /// 'nowait' clause), false - otherwise.
isParentNowaitRegion() const289   bool isParentNowaitRegion() const {
290     if (Stack.size() > 2)
291       return Stack[Stack.size() - 2].NowaitRegion;
292     return false;
293   }
294   /// \brief Marks parent region as cancel region.
setParentCancelRegion(bool Cancel=true)295   void setParentCancelRegion(bool Cancel = true) {
296     if (Stack.size() > 2)
297       Stack[Stack.size() - 2].CancelRegion =
298           Stack[Stack.size() - 2].CancelRegion || Cancel;
299   }
300   /// \brief Return true if current region has inner cancel construct.
isCancelRegion() const301   bool isCancelRegion() const {
302     return Stack.back().CancelRegion;
303   }
304 
305   /// \brief Set collapse value for the region.
setCollapseNumber(unsigned Val)306   void setCollapseNumber(unsigned Val) { Stack.back().CollapseNumber = Val; }
307   /// \brief Return collapse value for region.
getCollapseNumber() const308   unsigned getCollapseNumber() const {
309     return Stack.back().CollapseNumber;
310   }
311 
312   /// \brief Marks current target region as one with closely nested teams
313   /// region.
setParentTeamsRegionLoc(SourceLocation TeamsRegionLoc)314   void setParentTeamsRegionLoc(SourceLocation TeamsRegionLoc) {
315     if (Stack.size() > 2)
316       Stack[Stack.size() - 2].InnerTeamsRegionLoc = TeamsRegionLoc;
317   }
318   /// \brief Returns true, if current region has closely nested teams region.
hasInnerTeamsRegion() const319   bool hasInnerTeamsRegion() const {
320     return getInnerTeamsRegionLoc().isValid();
321   }
322   /// \brief Returns location of the nested teams region (if any).
getInnerTeamsRegionLoc() const323   SourceLocation getInnerTeamsRegionLoc() const {
324     if (Stack.size() > 1)
325       return Stack.back().InnerTeamsRegionLoc;
326     return SourceLocation();
327   }
328 
getCurScope() const329   Scope *getCurScope() const { return Stack.back().CurScope; }
getCurScope()330   Scope *getCurScope() { return Stack.back().CurScope; }
getConstructLoc()331   SourceLocation getConstructLoc() { return Stack.back().ConstructLoc; }
332 
getMapInfoForVar(VarDecl * VD)333   MapInfo getMapInfoForVar(VarDecl *VD) {
334     MapInfo VarMI = {0};
335     for (auto Cnt = Stack.size() - 1; Cnt > 0; --Cnt) {
336       if (Stack[Cnt].MappedDecls.count(VD)) {
337         VarMI = Stack[Cnt].MappedDecls[VD];
338         break;
339       }
340     }
341     return VarMI;
342   }
343 
addMapInfoForVar(VarDecl * VD,MapInfo MI)344   void addMapInfoForVar(VarDecl *VD, MapInfo MI) {
345     if (Stack.size() > 1) {
346       Stack.back().MappedDecls[VD] = MI;
347     }
348   }
349 
IsMappedInCurrentRegion(VarDecl * VD)350   MapInfo IsMappedInCurrentRegion(VarDecl *VD) {
351     assert(Stack.size() > 1 && "Target level is 0");
352     MapInfo VarMI = {0};
353     if (Stack.size() > 1 && Stack.back().MappedDecls.count(VD)) {
354       VarMI = Stack.back().MappedDecls[VD];
355     }
356     return VarMI;
357   }
358 };
isParallelOrTaskRegion(OpenMPDirectiveKind DKind)359 bool isParallelOrTaskRegion(OpenMPDirectiveKind DKind) {
360   return isOpenMPParallelDirective(DKind) || DKind == OMPD_task ||
361          isOpenMPTeamsDirective(DKind) || DKind == OMPD_unknown ||
362          isOpenMPTaskLoopDirective(DKind);
363 }
364 } // namespace
365 
getDSA(StackTy::reverse_iterator Iter,VarDecl * D)366 DSAStackTy::DSAVarData DSAStackTy::getDSA(StackTy::reverse_iterator Iter,
367                                           VarDecl *D) {
368   D = D->getCanonicalDecl();
369   DSAVarData DVar;
370   if (Iter == std::prev(Stack.rend())) {
371     // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced
372     // in a region but not in construct]
373     //  File-scope or namespace-scope variables referenced in called routines
374     //  in the region are shared unless they appear in a threadprivate
375     //  directive.
376     if (!D->isFunctionOrMethodVarDecl() && !isa<ParmVarDecl>(D))
377       DVar.CKind = OMPC_shared;
378 
379     // OpenMP [2.9.1.2, Data-sharing Attribute Rules for Variables Referenced
380     // in a region but not in construct]
381     //  Variables with static storage duration that are declared in called
382     //  routines in the region are shared.
383     if (D->hasGlobalStorage())
384       DVar.CKind = OMPC_shared;
385 
386     return DVar;
387   }
388 
389   DVar.DKind = Iter->Directive;
390   // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced
391   // in a Construct, C/C++, predetermined, p.1]
392   // Variables with automatic storage duration that are declared in a scope
393   // inside the construct are private.
394   if (isOpenMPLocal(D, Iter) && D->isLocalVarDecl() &&
395       (D->getStorageClass() == SC_Auto || D->getStorageClass() == SC_None)) {
396     DVar.CKind = OMPC_private;
397     return DVar;
398   }
399 
400   // Explicitly specified attributes and local variables with predetermined
401   // attributes.
402   if (Iter->SharingMap.count(D)) {
403     DVar.RefExpr = Iter->SharingMap[D].RefExpr;
404     DVar.CKind = Iter->SharingMap[D].Attributes;
405     DVar.ImplicitDSALoc = Iter->DefaultAttrLoc;
406     return DVar;
407   }
408 
409   // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced
410   // in a Construct, C/C++, implicitly determined, p.1]
411   //  In a parallel or task construct, the data-sharing attributes of these
412   //  variables are determined by the default clause, if present.
413   switch (Iter->DefaultAttr) {
414   case DSA_shared:
415     DVar.CKind = OMPC_shared;
416     DVar.ImplicitDSALoc = Iter->DefaultAttrLoc;
417     return DVar;
418   case DSA_none:
419     return DVar;
420   case DSA_unspecified:
421     // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced
422     // in a Construct, implicitly determined, p.2]
423     //  In a parallel construct, if no default clause is present, these
424     //  variables are shared.
425     DVar.ImplicitDSALoc = Iter->DefaultAttrLoc;
426     if (isOpenMPParallelDirective(DVar.DKind) ||
427         isOpenMPTeamsDirective(DVar.DKind)) {
428       DVar.CKind = OMPC_shared;
429       return DVar;
430     }
431 
432     // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced
433     // in a Construct, implicitly determined, p.4]
434     //  In a task construct, if no default clause is present, a variable that in
435     //  the enclosing context is determined to be shared by all implicit tasks
436     //  bound to the current team is shared.
437     if (DVar.DKind == OMPD_task) {
438       DSAVarData DVarTemp;
439       for (StackTy::reverse_iterator I = std::next(Iter), EE = Stack.rend();
440            I != EE; ++I) {
441         // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables
442         // Referenced
443         // in a Construct, implicitly determined, p.6]
444         //  In a task construct, if no default clause is present, a variable
445         //  whose data-sharing attribute is not determined by the rules above is
446         //  firstprivate.
447         DVarTemp = getDSA(I, D);
448         if (DVarTemp.CKind != OMPC_shared) {
449           DVar.RefExpr = nullptr;
450           DVar.DKind = OMPD_task;
451           DVar.CKind = OMPC_firstprivate;
452           return DVar;
453         }
454         if (isParallelOrTaskRegion(I->Directive))
455           break;
456       }
457       DVar.DKind = OMPD_task;
458       DVar.CKind =
459           (DVarTemp.CKind == OMPC_unknown) ? OMPC_firstprivate : OMPC_shared;
460       return DVar;
461     }
462   }
463   // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced
464   // in a Construct, implicitly determined, p.3]
465   //  For constructs other than task, if no default clause is present, these
466   //  variables inherit their data-sharing attributes from the enclosing
467   //  context.
468   return getDSA(std::next(Iter), D);
469 }
470 
addUniqueAligned(VarDecl * D,DeclRefExpr * NewDE)471 DeclRefExpr *DSAStackTy::addUniqueAligned(VarDecl *D, DeclRefExpr *NewDE) {
472   assert(Stack.size() > 1 && "Data sharing attributes stack is empty");
473   D = D->getCanonicalDecl();
474   auto It = Stack.back().AlignedMap.find(D);
475   if (It == Stack.back().AlignedMap.end()) {
476     assert(NewDE && "Unexpected nullptr expr to be added into aligned map");
477     Stack.back().AlignedMap[D] = NewDE;
478     return nullptr;
479   } else {
480     assert(It->second && "Unexpected nullptr expr in the aligned map");
481     return It->second;
482   }
483   return nullptr;
484 }
485 
addLoopControlVariable(VarDecl * D)486 void DSAStackTy::addLoopControlVariable(VarDecl *D) {
487   assert(Stack.size() > 1 && "Data-sharing attributes stack is empty");
488   D = D->getCanonicalDecl();
489   Stack.back().LCVSet.insert(D);
490 }
491 
isLoopControlVariable(VarDecl * D)492 bool DSAStackTy::isLoopControlVariable(VarDecl *D) {
493   assert(Stack.size() > 1 && "Data-sharing attributes stack is empty");
494   D = D->getCanonicalDecl();
495   return Stack.back().LCVSet.count(D) > 0;
496 }
497 
addDSA(VarDecl * D,DeclRefExpr * E,OpenMPClauseKind A)498 void DSAStackTy::addDSA(VarDecl *D, DeclRefExpr *E, OpenMPClauseKind A) {
499   D = D->getCanonicalDecl();
500   if (A == OMPC_threadprivate) {
501     Stack[0].SharingMap[D].Attributes = A;
502     Stack[0].SharingMap[D].RefExpr = E;
503   } else {
504     assert(Stack.size() > 1 && "Data-sharing attributes stack is empty");
505     Stack.back().SharingMap[D].Attributes = A;
506     Stack.back().SharingMap[D].RefExpr = E;
507   }
508 }
509 
isOpenMPLocal(VarDecl * D,StackTy::reverse_iterator Iter)510 bool DSAStackTy::isOpenMPLocal(VarDecl *D, StackTy::reverse_iterator Iter) {
511   D = D->getCanonicalDecl();
512   if (Stack.size() > 2) {
513     reverse_iterator I = Iter, E = std::prev(Stack.rend());
514     Scope *TopScope = nullptr;
515     while (I != E && !isParallelOrTaskRegion(I->Directive)) {
516       ++I;
517     }
518     if (I == E)
519       return false;
520     TopScope = I->CurScope ? I->CurScope->getParent() : nullptr;
521     Scope *CurScope = getCurScope();
522     while (CurScope != TopScope && !CurScope->isDeclScope(D)) {
523       CurScope = CurScope->getParent();
524     }
525     return CurScope != TopScope;
526   }
527   return false;
528 }
529 
530 /// \brief Build a variable declaration for OpenMP loop iteration variable.
buildVarDecl(Sema & SemaRef,SourceLocation Loc,QualType Type,StringRef Name,const AttrVec * Attrs=nullptr)531 static VarDecl *buildVarDecl(Sema &SemaRef, SourceLocation Loc, QualType Type,
532                              StringRef Name, const AttrVec *Attrs = nullptr) {
533   DeclContext *DC = SemaRef.CurContext;
534   IdentifierInfo *II = &SemaRef.PP.getIdentifierTable().get(Name);
535   TypeSourceInfo *TInfo = SemaRef.Context.getTrivialTypeSourceInfo(Type, Loc);
536   VarDecl *Decl =
537       VarDecl::Create(SemaRef.Context, DC, Loc, Loc, II, Type, TInfo, SC_None);
538   if (Attrs) {
539     for (specific_attr_iterator<AlignedAttr> I(Attrs->begin()), E(Attrs->end());
540          I != E; ++I)
541       Decl->addAttr(*I);
542   }
543   Decl->setImplicit();
544   return Decl;
545 }
546 
buildDeclRefExpr(Sema & S,VarDecl * D,QualType Ty,SourceLocation Loc,bool RefersToCapture=false)547 static DeclRefExpr *buildDeclRefExpr(Sema &S, VarDecl *D, QualType Ty,
548                                      SourceLocation Loc,
549                                      bool RefersToCapture = false) {
550   D->setReferenced();
551   D->markUsed(S.Context);
552   return DeclRefExpr::Create(S.getASTContext(), NestedNameSpecifierLoc(),
553                              SourceLocation(), D, RefersToCapture, Loc, Ty,
554                              VK_LValue);
555 }
556 
getTopDSA(VarDecl * D,bool FromParent)557 DSAStackTy::DSAVarData DSAStackTy::getTopDSA(VarDecl *D, bool FromParent) {
558   D = D->getCanonicalDecl();
559   DSAVarData DVar;
560 
561   // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced
562   // in a Construct, C/C++, predetermined, p.1]
563   //  Variables appearing in threadprivate directives are threadprivate.
564   if ((D->getTLSKind() != VarDecl::TLS_None &&
565        !(D->hasAttr<OMPThreadPrivateDeclAttr>() &&
566          SemaRef.getLangOpts().OpenMPUseTLS &&
567          SemaRef.getASTContext().getTargetInfo().isTLSSupported())) ||
568       (D->getStorageClass() == SC_Register && D->hasAttr<AsmLabelAttr>() &&
569        !D->isLocalVarDecl())) {
570     addDSA(D, buildDeclRefExpr(SemaRef, D, D->getType().getNonReferenceType(),
571                                D->getLocation()),
572            OMPC_threadprivate);
573   }
574   if (Stack[0].SharingMap.count(D)) {
575     DVar.RefExpr = Stack[0].SharingMap[D].RefExpr;
576     DVar.CKind = OMPC_threadprivate;
577     return DVar;
578   }
579 
580   // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced
581   // in a Construct, C/C++, predetermined, p.4]
582   //  Static data members are shared.
583   // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced
584   // in a Construct, C/C++, predetermined, p.7]
585   //  Variables with static storage duration that are declared in a scope
586   //  inside the construct are shared.
587   if (D->isStaticDataMember()) {
588     DSAVarData DVarTemp =
589         hasDSA(D, isOpenMPPrivate, MatchesAlways(), FromParent);
590     if (DVarTemp.CKind != OMPC_unknown && DVarTemp.RefExpr)
591       return DVar;
592 
593     DVar.CKind = OMPC_shared;
594     return DVar;
595   }
596 
597   QualType Type = D->getType().getNonReferenceType().getCanonicalType();
598   bool IsConstant = Type.isConstant(SemaRef.getASTContext());
599   Type = SemaRef.getASTContext().getBaseElementType(Type);
600   // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced
601   // in a Construct, C/C++, predetermined, p.6]
602   //  Variables with const qualified type having no mutable member are
603   //  shared.
604   CXXRecordDecl *RD =
605       SemaRef.getLangOpts().CPlusPlus ? Type->getAsCXXRecordDecl() : nullptr;
606   if (auto *CTSD = dyn_cast_or_null<ClassTemplateSpecializationDecl>(RD))
607     if (auto *CTD = CTSD->getSpecializedTemplate())
608       RD = CTD->getTemplatedDecl();
609   if (IsConstant &&
610       !(SemaRef.getLangOpts().CPlusPlus && RD && RD->hasMutableFields())) {
611     // Variables with const-qualified type having no mutable member may be
612     // listed in a firstprivate clause, even if they are static data members.
613     DSAVarData DVarTemp = hasDSA(D, MatchesAnyClause(OMPC_firstprivate),
614                                  MatchesAlways(), FromParent);
615     if (DVarTemp.CKind == OMPC_firstprivate && DVarTemp.RefExpr)
616       return DVar;
617 
618     DVar.CKind = OMPC_shared;
619     return DVar;
620   }
621 
622   // Explicitly specified attributes and local variables with predetermined
623   // attributes.
624   auto StartI = std::next(Stack.rbegin());
625   auto EndI = std::prev(Stack.rend());
626   if (FromParent && StartI != EndI) {
627     StartI = std::next(StartI);
628   }
629   auto I = std::prev(StartI);
630   if (I->SharingMap.count(D)) {
631     DVar.RefExpr = I->SharingMap[D].RefExpr;
632     DVar.CKind = I->SharingMap[D].Attributes;
633     DVar.ImplicitDSALoc = I->DefaultAttrLoc;
634   }
635 
636   return DVar;
637 }
638 
getImplicitDSA(VarDecl * D,bool FromParent)639 DSAStackTy::DSAVarData DSAStackTy::getImplicitDSA(VarDecl *D, bool FromParent) {
640   D = D->getCanonicalDecl();
641   auto StartI = Stack.rbegin();
642   auto EndI = std::prev(Stack.rend());
643   if (FromParent && StartI != EndI) {
644     StartI = std::next(StartI);
645   }
646   return getDSA(StartI, D);
647 }
648 
649 template <class ClausesPredicate, class DirectivesPredicate>
hasDSA(VarDecl * D,ClausesPredicate CPred,DirectivesPredicate DPred,bool FromParent)650 DSAStackTy::DSAVarData DSAStackTy::hasDSA(VarDecl *D, ClausesPredicate CPred,
651                                           DirectivesPredicate DPred,
652                                           bool FromParent) {
653   D = D->getCanonicalDecl();
654   auto StartI = std::next(Stack.rbegin());
655   auto EndI = std::prev(Stack.rend());
656   if (FromParent && StartI != EndI) {
657     StartI = std::next(StartI);
658   }
659   for (auto I = StartI, EE = EndI; I != EE; ++I) {
660     if (!DPred(I->Directive) && !isParallelOrTaskRegion(I->Directive))
661       continue;
662     DSAVarData DVar = getDSA(I, D);
663     if (CPred(DVar.CKind))
664       return DVar;
665   }
666   return DSAVarData();
667 }
668 
669 template <class ClausesPredicate, class DirectivesPredicate>
670 DSAStackTy::DSAVarData
hasInnermostDSA(VarDecl * D,ClausesPredicate CPred,DirectivesPredicate DPred,bool FromParent)671 DSAStackTy::hasInnermostDSA(VarDecl *D, ClausesPredicate CPred,
672                             DirectivesPredicate DPred, bool FromParent) {
673   D = D->getCanonicalDecl();
674   auto StartI = std::next(Stack.rbegin());
675   auto EndI = std::prev(Stack.rend());
676   if (FromParent && StartI != EndI) {
677     StartI = std::next(StartI);
678   }
679   for (auto I = StartI, EE = EndI; I != EE; ++I) {
680     if (!DPred(I->Directive))
681       break;
682     DSAVarData DVar = getDSA(I, D);
683     if (CPred(DVar.CKind))
684       return DVar;
685     return DSAVarData();
686   }
687   return DSAVarData();
688 }
689 
hasExplicitDSA(VarDecl * D,const llvm::function_ref<bool (OpenMPClauseKind)> & CPred,unsigned Level)690 bool DSAStackTy::hasExplicitDSA(
691     VarDecl *D, const llvm::function_ref<bool(OpenMPClauseKind)> &CPred,
692     unsigned Level) {
693   if (CPred(ClauseKindMode))
694     return true;
695   if (isClauseParsingMode())
696     ++Level;
697   D = D->getCanonicalDecl();
698   auto StartI = Stack.rbegin();
699   auto EndI = std::prev(Stack.rend());
700   if (std::distance(StartI, EndI) <= (int)Level)
701     return false;
702   std::advance(StartI, Level);
703   return (StartI->SharingMap.count(D) > 0) && StartI->SharingMap[D].RefExpr &&
704          CPred(StartI->SharingMap[D].Attributes);
705 }
706 
hasExplicitDirective(const llvm::function_ref<bool (OpenMPDirectiveKind)> & DPred,unsigned Level)707 bool DSAStackTy::hasExplicitDirective(
708     const llvm::function_ref<bool(OpenMPDirectiveKind)> &DPred,
709     unsigned Level) {
710   if (isClauseParsingMode())
711     ++Level;
712   auto StartI = Stack.rbegin();
713   auto EndI = std::prev(Stack.rend());
714   if (std::distance(StartI, EndI) <= (int)Level)
715     return false;
716   std::advance(StartI, Level);
717   return DPred(StartI->Directive);
718 }
719 
720 template <class NamedDirectivesPredicate>
hasDirective(NamedDirectivesPredicate DPred,bool FromParent)721 bool DSAStackTy::hasDirective(NamedDirectivesPredicate DPred, bool FromParent) {
722   auto StartI = std::next(Stack.rbegin());
723   auto EndI = std::prev(Stack.rend());
724   if (FromParent && StartI != EndI) {
725     StartI = std::next(StartI);
726   }
727   for (auto I = StartI, EE = EndI; I != EE; ++I) {
728     if (DPred(I->Directive, I->DirectiveName, I->ConstructLoc))
729       return true;
730   }
731   return false;
732 }
733 
getDirectiveForScope(const Scope * S) const734 OpenMPDirectiveKind DSAStackTy::getDirectiveForScope(const Scope *S) const {
735   for (auto I = Stack.rbegin(), EE = Stack.rend(); I != EE; ++I)
736     if (I->CurScope == S)
737       return I->Directive;
738   return OMPD_unknown;
739 }
740 
InitDataSharingAttributesStack()741 void Sema::InitDataSharingAttributesStack() {
742   VarDataSharingAttributesStack = new DSAStackTy(*this);
743 }
744 
745 #define DSAStack static_cast<DSAStackTy *>(VarDataSharingAttributesStack)
746 
IsOpenMPCapturedByRef(VarDecl * VD,const CapturedRegionScopeInfo * RSI)747 bool Sema::IsOpenMPCapturedByRef(VarDecl *VD,
748                                  const CapturedRegionScopeInfo *RSI) {
749   assert(LangOpts.OpenMP && "OpenMP is not allowed");
750 
751   auto &Ctx = getASTContext();
752   bool IsByRef = true;
753 
754   // Find the directive that is associated with the provided scope.
755   auto DKind = DSAStack->getDirectiveForScope(RSI->TheScope);
756   auto Ty = VD->getType();
757 
758   if (isOpenMPTargetDirective(DKind)) {
759     // This table summarizes how a given variable should be passed to the device
760     // given its type and the clauses where it appears. This table is based on
761     // the description in OpenMP 4.5 [2.10.4, target Construct] and
762     // OpenMP 4.5 [2.15.5, Data-mapping Attribute Rules and Clauses].
763     //
764     // =========================================================================
765     // | type |  defaultmap   | pvt | first | is_device_ptr |    map   | res.  |
766     // |      |(tofrom:scalar)|     |  pvt  |               |          |       |
767     // =========================================================================
768     // | scl  |               |     |       |       -       |          | bycopy|
769     // | scl  |               |  -  |   x   |       -       |     -    | bycopy|
770     // | scl  |               |  x  |   -   |       -       |     -    | null  |
771     // | scl  |       x       |     |       |       -       |          | byref |
772     // | scl  |       x       |  -  |   x   |       -       |     -    | bycopy|
773     // | scl  |       x       |  x  |   -   |       -       |     -    | null  |
774     // | scl  |               |  -  |   -   |       -       |     x    | byref |
775     // | scl  |       x       |  -  |   -   |       -       |     x    | byref |
776     //
777     // | agg  |      n.a.     |     |       |       -       |          | byref |
778     // | agg  |      n.a.     |  -  |   x   |       -       |     -    | byref |
779     // | agg  |      n.a.     |  x  |   -   |       -       |     -    | null  |
780     // | agg  |      n.a.     |  -  |   -   |       -       |     x    | byref |
781     // | agg  |      n.a.     |  -  |   -   |       -       |    x[]   | byref |
782     //
783     // | ptr  |      n.a.     |     |       |       -       |          | bycopy|
784     // | ptr  |      n.a.     |  -  |   x   |       -       |     -    | bycopy|
785     // | ptr  |      n.a.     |  x  |   -   |       -       |     -    | null  |
786     // | ptr  |      n.a.     |  -  |   -   |       -       |     x    | byref |
787     // | ptr  |      n.a.     |  -  |   -   |       -       |    x[]   | bycopy|
788     // | ptr  |      n.a.     |  -  |   -   |       x       |          | bycopy|
789     // | ptr  |      n.a.     |  -  |   -   |       x       |     x    | bycopy|
790     // | ptr  |      n.a.     |  -  |   -   |       x       |    x[]   | bycopy|
791     // =========================================================================
792     // Legend:
793     //  scl - scalar
794     //  ptr - pointer
795     //  agg - aggregate
796     //  x - applies
797     //  - - invalid in this combination
798     //  [] - mapped with an array section
799     //  byref - should be mapped by reference
800     //  byval - should be mapped by value
801     //  null - initialize a local variable to null on the device
802     //
803     // Observations:
804     //  - All scalar declarations that show up in a map clause have to be passed
805     //    by reference, because they may have been mapped in the enclosing data
806     //    environment.
807     //  - If the scalar value does not fit the size of uintptr, it has to be
808     //    passed by reference, regardless the result in the table above.
809     //  - For pointers mapped by value that have either an implicit map or an
810     //    array section, the runtime library may pass the NULL value to the
811     //    device instead of the value passed to it by the compiler.
812 
813     // FIXME: Right now, only implicit maps are implemented. Properly mapping
814     // values requires having the map, private, and firstprivate clauses SEMA
815     // and parsing in place, which we don't yet.
816 
817     if (Ty->isReferenceType())
818       Ty = Ty->castAs<ReferenceType>()->getPointeeType();
819     IsByRef = !Ty->isScalarType();
820   }
821 
822   // When passing data by value, we need to make sure it fits the uintptr size
823   // and alignment, because the runtime library only deals with uintptr types.
824   // If it does not fit the uintptr size, we need to pass the data by reference
825   // instead.
826   if (!IsByRef &&
827       (Ctx.getTypeSizeInChars(Ty) >
828            Ctx.getTypeSizeInChars(Ctx.getUIntPtrType()) ||
829        Ctx.getDeclAlign(VD) > Ctx.getTypeAlignInChars(Ctx.getUIntPtrType())))
830     IsByRef = true;
831 
832   return IsByRef;
833 }
834 
IsOpenMPCapturedVar(VarDecl * VD)835 bool Sema::IsOpenMPCapturedVar(VarDecl *VD) {
836   assert(LangOpts.OpenMP && "OpenMP is not allowed");
837   VD = VD->getCanonicalDecl();
838 
839   // If we are attempting to capture a global variable in a directive with
840   // 'target' we return true so that this global is also mapped to the device.
841   //
842   // FIXME: If the declaration is enclosed in a 'declare target' directive,
843   // then it should not be captured. Therefore, an extra check has to be
844   // inserted here once support for 'declare target' is added.
845   //
846   if (!VD->hasLocalStorage()) {
847     if (DSAStack->getCurrentDirective() == OMPD_target &&
848         !DSAStack->isClauseParsingMode()) {
849       return true;
850     }
851     if (DSAStack->getCurScope() &&
852         DSAStack->hasDirective(
853             [](OpenMPDirectiveKind K, const DeclarationNameInfo &DNI,
854                SourceLocation Loc) -> bool {
855               return isOpenMPTargetDirective(K);
856             },
857             false)) {
858       return true;
859     }
860   }
861 
862   if (DSAStack->getCurrentDirective() != OMPD_unknown &&
863       (!DSAStack->isClauseParsingMode() ||
864        DSAStack->getParentDirective() != OMPD_unknown)) {
865     if (DSAStack->isLoopControlVariable(VD) ||
866         (VD->hasLocalStorage() &&
867          isParallelOrTaskRegion(DSAStack->getCurrentDirective())) ||
868         DSAStack->isForceVarCapturing())
869       return true;
870     auto DVarPrivate = DSAStack->getTopDSA(VD, DSAStack->isClauseParsingMode());
871     if (DVarPrivate.CKind != OMPC_unknown && isOpenMPPrivate(DVarPrivate.CKind))
872       return true;
873     DVarPrivate = DSAStack->hasDSA(VD, isOpenMPPrivate, MatchesAlways(),
874                                    DSAStack->isClauseParsingMode());
875     return DVarPrivate.CKind != OMPC_unknown;
876   }
877   return false;
878 }
879 
isOpenMPPrivateVar(VarDecl * VD,unsigned Level)880 bool Sema::isOpenMPPrivateVar(VarDecl *VD, unsigned Level) {
881   assert(LangOpts.OpenMP && "OpenMP is not allowed");
882   return DSAStack->hasExplicitDSA(
883       VD, [](OpenMPClauseKind K) -> bool { return K == OMPC_private; }, Level);
884 }
885 
isOpenMPTargetCapturedVar(VarDecl * VD,unsigned Level)886 bool Sema::isOpenMPTargetCapturedVar(VarDecl *VD, unsigned Level) {
887   assert(LangOpts.OpenMP && "OpenMP is not allowed");
888   // Return true if the current level is no longer enclosed in a target region.
889 
890   return !VD->hasLocalStorage() &&
891          DSAStack->hasExplicitDirective(isOpenMPTargetDirective, Level);
892 }
893 
DestroyDataSharingAttributesStack()894 void Sema::DestroyDataSharingAttributesStack() { delete DSAStack; }
895 
StartOpenMPDSABlock(OpenMPDirectiveKind DKind,const DeclarationNameInfo & DirName,Scope * CurScope,SourceLocation Loc)896 void Sema::StartOpenMPDSABlock(OpenMPDirectiveKind DKind,
897                                const DeclarationNameInfo &DirName,
898                                Scope *CurScope, SourceLocation Loc) {
899   DSAStack->push(DKind, DirName, CurScope, Loc);
900   PushExpressionEvaluationContext(PotentiallyEvaluated);
901 }
902 
StartOpenMPClause(OpenMPClauseKind K)903 void Sema::StartOpenMPClause(OpenMPClauseKind K) {
904   DSAStack->setClauseParsingMode(K);
905 }
906 
EndOpenMPClause()907 void Sema::EndOpenMPClause() {
908   DSAStack->setClauseParsingMode(/*K=*/OMPC_unknown);
909 }
910 
EndOpenMPDSABlock(Stmt * CurDirective)911 void Sema::EndOpenMPDSABlock(Stmt *CurDirective) {
912   // OpenMP [2.14.3.5, Restrictions, C/C++, p.1]
913   //  A variable of class type (or array thereof) that appears in a lastprivate
914   //  clause requires an accessible, unambiguous default constructor for the
915   //  class type, unless the list item is also specified in a firstprivate
916   //  clause.
917   if (auto D = dyn_cast_or_null<OMPExecutableDirective>(CurDirective)) {
918     for (auto *C : D->clauses()) {
919       if (auto *Clause = dyn_cast<OMPLastprivateClause>(C)) {
920         SmallVector<Expr *, 8> PrivateCopies;
921         for (auto *DE : Clause->varlists()) {
922           if (DE->isValueDependent() || DE->isTypeDependent()) {
923             PrivateCopies.push_back(nullptr);
924             continue;
925           }
926           auto *VD = cast<VarDecl>(cast<DeclRefExpr>(DE)->getDecl());
927           QualType Type = VD->getType().getNonReferenceType();
928           auto DVar = DSAStack->getTopDSA(VD, false);
929           if (DVar.CKind == OMPC_lastprivate) {
930             // Generate helper private variable and initialize it with the
931             // default value. The address of the original variable is replaced
932             // by the address of the new private variable in CodeGen. This new
933             // variable is not added to IdResolver, so the code in the OpenMP
934             // region uses original variable for proper diagnostics.
935             auto *VDPrivate = buildVarDecl(
936                 *this, DE->getExprLoc(), Type.getUnqualifiedType(),
937                 VD->getName(), VD->hasAttrs() ? &VD->getAttrs() : nullptr);
938             ActOnUninitializedDecl(VDPrivate, /*TypeMayContainAuto=*/false);
939             if (VDPrivate->isInvalidDecl())
940               continue;
941             PrivateCopies.push_back(buildDeclRefExpr(
942                 *this, VDPrivate, DE->getType(), DE->getExprLoc()));
943           } else {
944             // The variable is also a firstprivate, so initialization sequence
945             // for private copy is generated already.
946             PrivateCopies.push_back(nullptr);
947           }
948         }
949         // Set initializers to private copies if no errors were found.
950         if (PrivateCopies.size() == Clause->varlist_size()) {
951           Clause->setPrivateCopies(PrivateCopies);
952         }
953       }
954     }
955   }
956 
957   DSAStack->pop();
958   DiscardCleanupsInEvaluationContext();
959   PopExpressionEvaluationContext();
960 }
961 
962 static bool FinishOpenMPLinearClause(OMPLinearClause &Clause, DeclRefExpr *IV,
963                                      Expr *NumIterations, Sema &SemaRef,
964                                      Scope *S);
965 
966 namespace {
967 
968 class VarDeclFilterCCC : public CorrectionCandidateCallback {
969 private:
970   Sema &SemaRef;
971 
972 public:
VarDeclFilterCCC(Sema & S)973   explicit VarDeclFilterCCC(Sema &S) : SemaRef(S) {}
ValidateCandidate(const TypoCorrection & Candidate)974   bool ValidateCandidate(const TypoCorrection &Candidate) override {
975     NamedDecl *ND = Candidate.getCorrectionDecl();
976     if (VarDecl *VD = dyn_cast_or_null<VarDecl>(ND)) {
977       return VD->hasGlobalStorage() &&
978              SemaRef.isDeclInScope(ND, SemaRef.getCurLexicalContext(),
979                                    SemaRef.getCurScope());
980     }
981     return false;
982   }
983 };
984 } // namespace
985 
ActOnOpenMPIdExpression(Scope * CurScope,CXXScopeSpec & ScopeSpec,const DeclarationNameInfo & Id)986 ExprResult Sema::ActOnOpenMPIdExpression(Scope *CurScope,
987                                          CXXScopeSpec &ScopeSpec,
988                                          const DeclarationNameInfo &Id) {
989   LookupResult Lookup(*this, Id, LookupOrdinaryName);
990   LookupParsedName(Lookup, CurScope, &ScopeSpec, true);
991 
992   if (Lookup.isAmbiguous())
993     return ExprError();
994 
995   VarDecl *VD;
996   if (!Lookup.isSingleResult()) {
997     if (TypoCorrection Corrected = CorrectTypo(
998             Id, LookupOrdinaryName, CurScope, nullptr,
999             llvm::make_unique<VarDeclFilterCCC>(*this), CTK_ErrorRecovery)) {
1000       diagnoseTypo(Corrected,
1001                    PDiag(Lookup.empty()
1002                              ? diag::err_undeclared_var_use_suggest
1003                              : diag::err_omp_expected_var_arg_suggest)
1004                        << Id.getName());
1005       VD = Corrected.getCorrectionDeclAs<VarDecl>();
1006     } else {
1007       Diag(Id.getLoc(), Lookup.empty() ? diag::err_undeclared_var_use
1008                                        : diag::err_omp_expected_var_arg)
1009           << Id.getName();
1010       return ExprError();
1011     }
1012   } else {
1013     if (!(VD = Lookup.getAsSingle<VarDecl>())) {
1014       Diag(Id.getLoc(), diag::err_omp_expected_var_arg) << Id.getName();
1015       Diag(Lookup.getFoundDecl()->getLocation(), diag::note_declared_at);
1016       return ExprError();
1017     }
1018   }
1019   Lookup.suppressDiagnostics();
1020 
1021   // OpenMP [2.9.2, Syntax, C/C++]
1022   //   Variables must be file-scope, namespace-scope, or static block-scope.
1023   if (!VD->hasGlobalStorage()) {
1024     Diag(Id.getLoc(), diag::err_omp_global_var_arg)
1025         << getOpenMPDirectiveName(OMPD_threadprivate) << !VD->isStaticLocal();
1026     bool IsDecl =
1027         VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly;
1028     Diag(VD->getLocation(),
1029          IsDecl ? diag::note_previous_decl : diag::note_defined_here)
1030         << VD;
1031     return ExprError();
1032   }
1033 
1034   VarDecl *CanonicalVD = VD->getCanonicalDecl();
1035   NamedDecl *ND = cast<NamedDecl>(CanonicalVD);
1036   // OpenMP [2.9.2, Restrictions, C/C++, p.2]
1037   //   A threadprivate directive for file-scope variables must appear outside
1038   //   any definition or declaration.
1039   if (CanonicalVD->getDeclContext()->isTranslationUnit() &&
1040       !getCurLexicalContext()->isTranslationUnit()) {
1041     Diag(Id.getLoc(), diag::err_omp_var_scope)
1042         << getOpenMPDirectiveName(OMPD_threadprivate) << VD;
1043     bool IsDecl =
1044         VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly;
1045     Diag(VD->getLocation(),
1046          IsDecl ? diag::note_previous_decl : diag::note_defined_here)
1047         << VD;
1048     return ExprError();
1049   }
1050   // OpenMP [2.9.2, Restrictions, C/C++, p.3]
1051   //   A threadprivate directive for static class member variables must appear
1052   //   in the class definition, in the same scope in which the member
1053   //   variables are declared.
1054   if (CanonicalVD->isStaticDataMember() &&
1055       !CanonicalVD->getDeclContext()->Equals(getCurLexicalContext())) {
1056     Diag(Id.getLoc(), diag::err_omp_var_scope)
1057         << getOpenMPDirectiveName(OMPD_threadprivate) << VD;
1058     bool IsDecl =
1059         VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly;
1060     Diag(VD->getLocation(),
1061          IsDecl ? diag::note_previous_decl : diag::note_defined_here)
1062         << VD;
1063     return ExprError();
1064   }
1065   // OpenMP [2.9.2, Restrictions, C/C++, p.4]
1066   //   A threadprivate directive for namespace-scope variables must appear
1067   //   outside any definition or declaration other than the namespace
1068   //   definition itself.
1069   if (CanonicalVD->getDeclContext()->isNamespace() &&
1070       (!getCurLexicalContext()->isFileContext() ||
1071        !getCurLexicalContext()->Encloses(CanonicalVD->getDeclContext()))) {
1072     Diag(Id.getLoc(), diag::err_omp_var_scope)
1073         << getOpenMPDirectiveName(OMPD_threadprivate) << VD;
1074     bool IsDecl =
1075         VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly;
1076     Diag(VD->getLocation(),
1077          IsDecl ? diag::note_previous_decl : diag::note_defined_here)
1078         << VD;
1079     return ExprError();
1080   }
1081   // OpenMP [2.9.2, Restrictions, C/C++, p.6]
1082   //   A threadprivate directive for static block-scope variables must appear
1083   //   in the scope of the variable and not in a nested scope.
1084   if (CanonicalVD->isStaticLocal() && CurScope &&
1085       !isDeclInScope(ND, getCurLexicalContext(), CurScope)) {
1086     Diag(Id.getLoc(), diag::err_omp_var_scope)
1087         << getOpenMPDirectiveName(OMPD_threadprivate) << VD;
1088     bool IsDecl =
1089         VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly;
1090     Diag(VD->getLocation(),
1091          IsDecl ? diag::note_previous_decl : diag::note_defined_here)
1092         << VD;
1093     return ExprError();
1094   }
1095 
1096   // OpenMP [2.9.2, Restrictions, C/C++, p.2-6]
1097   //   A threadprivate directive must lexically precede all references to any
1098   //   of the variables in its list.
1099   if (VD->isUsed() && !DSAStack->isThreadPrivate(VD)) {
1100     Diag(Id.getLoc(), diag::err_omp_var_used)
1101         << getOpenMPDirectiveName(OMPD_threadprivate) << VD;
1102     return ExprError();
1103   }
1104 
1105   QualType ExprType = VD->getType().getNonReferenceType();
1106   ExprResult DE = buildDeclRefExpr(*this, VD, ExprType, Id.getLoc());
1107   return DE;
1108 }
1109 
1110 Sema::DeclGroupPtrTy
ActOnOpenMPThreadprivateDirective(SourceLocation Loc,ArrayRef<Expr * > VarList)1111 Sema::ActOnOpenMPThreadprivateDirective(SourceLocation Loc,
1112                                         ArrayRef<Expr *> VarList) {
1113   if (OMPThreadPrivateDecl *D = CheckOMPThreadPrivateDecl(Loc, VarList)) {
1114     CurContext->addDecl(D);
1115     return DeclGroupPtrTy::make(DeclGroupRef(D));
1116   }
1117   return DeclGroupPtrTy();
1118 }
1119 
1120 namespace {
1121 class LocalVarRefChecker : public ConstStmtVisitor<LocalVarRefChecker, bool> {
1122   Sema &SemaRef;
1123 
1124 public:
VisitDeclRefExpr(const DeclRefExpr * E)1125   bool VisitDeclRefExpr(const DeclRefExpr *E) {
1126     if (auto VD = dyn_cast<VarDecl>(E->getDecl())) {
1127       if (VD->hasLocalStorage()) {
1128         SemaRef.Diag(E->getLocStart(),
1129                      diag::err_omp_local_var_in_threadprivate_init)
1130             << E->getSourceRange();
1131         SemaRef.Diag(VD->getLocation(), diag::note_defined_here)
1132             << VD << VD->getSourceRange();
1133         return true;
1134       }
1135     }
1136     return false;
1137   }
VisitStmt(const Stmt * S)1138   bool VisitStmt(const Stmt *S) {
1139     for (auto Child : S->children()) {
1140       if (Child && Visit(Child))
1141         return true;
1142     }
1143     return false;
1144   }
LocalVarRefChecker(Sema & SemaRef)1145   explicit LocalVarRefChecker(Sema &SemaRef) : SemaRef(SemaRef) {}
1146 };
1147 } // namespace
1148 
1149 OMPThreadPrivateDecl *
CheckOMPThreadPrivateDecl(SourceLocation Loc,ArrayRef<Expr * > VarList)1150 Sema::CheckOMPThreadPrivateDecl(SourceLocation Loc, ArrayRef<Expr *> VarList) {
1151   SmallVector<Expr *, 8> Vars;
1152   for (auto &RefExpr : VarList) {
1153     DeclRefExpr *DE = cast<DeclRefExpr>(RefExpr);
1154     VarDecl *VD = cast<VarDecl>(DE->getDecl());
1155     SourceLocation ILoc = DE->getExprLoc();
1156 
1157     QualType QType = VD->getType();
1158     if (QType->isDependentType() || QType->isInstantiationDependentType()) {
1159       // It will be analyzed later.
1160       Vars.push_back(DE);
1161       continue;
1162     }
1163 
1164     // OpenMP [2.9.2, Restrictions, C/C++, p.10]
1165     //   A threadprivate variable must not have an incomplete type.
1166     if (RequireCompleteType(ILoc, VD->getType(),
1167                             diag::err_omp_threadprivate_incomplete_type)) {
1168       continue;
1169     }
1170 
1171     // OpenMP [2.9.2, Restrictions, C/C++, p.10]
1172     //   A threadprivate variable must not have a reference type.
1173     if (VD->getType()->isReferenceType()) {
1174       Diag(ILoc, diag::err_omp_ref_type_arg)
1175           << getOpenMPDirectiveName(OMPD_threadprivate) << VD->getType();
1176       bool IsDecl =
1177           VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly;
1178       Diag(VD->getLocation(),
1179            IsDecl ? diag::note_previous_decl : diag::note_defined_here)
1180           << VD;
1181       continue;
1182     }
1183 
1184     // Check if this is a TLS variable. If TLS is not being supported, produce
1185     // the corresponding diagnostic.
1186     if ((VD->getTLSKind() != VarDecl::TLS_None &&
1187          !(VD->hasAttr<OMPThreadPrivateDeclAttr>() &&
1188            getLangOpts().OpenMPUseTLS &&
1189            getASTContext().getTargetInfo().isTLSSupported())) ||
1190         (VD->getStorageClass() == SC_Register && VD->hasAttr<AsmLabelAttr>() &&
1191          !VD->isLocalVarDecl())) {
1192       Diag(ILoc, diag::err_omp_var_thread_local)
1193           << VD << ((VD->getTLSKind() != VarDecl::TLS_None) ? 0 : 1);
1194       bool IsDecl =
1195           VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly;
1196       Diag(VD->getLocation(),
1197            IsDecl ? diag::note_previous_decl : diag::note_defined_here)
1198           << VD;
1199       continue;
1200     }
1201 
1202     // Check if initial value of threadprivate variable reference variable with
1203     // local storage (it is not supported by runtime).
1204     if (auto Init = VD->getAnyInitializer()) {
1205       LocalVarRefChecker Checker(*this);
1206       if (Checker.Visit(Init))
1207         continue;
1208     }
1209 
1210     Vars.push_back(RefExpr);
1211     DSAStack->addDSA(VD, DE, OMPC_threadprivate);
1212     VD->addAttr(OMPThreadPrivateDeclAttr::CreateImplicit(
1213         Context, SourceRange(Loc, Loc)));
1214     if (auto *ML = Context.getASTMutationListener())
1215       ML->DeclarationMarkedOpenMPThreadPrivate(VD);
1216   }
1217   OMPThreadPrivateDecl *D = nullptr;
1218   if (!Vars.empty()) {
1219     D = OMPThreadPrivateDecl::Create(Context, getCurLexicalContext(), Loc,
1220                                      Vars);
1221     D->setAccess(AS_public);
1222   }
1223   return D;
1224 }
1225 
ReportOriginalDSA(Sema & SemaRef,DSAStackTy * Stack,const VarDecl * VD,DSAStackTy::DSAVarData DVar,bool IsLoopIterVar=false)1226 static void ReportOriginalDSA(Sema &SemaRef, DSAStackTy *Stack,
1227                               const VarDecl *VD, DSAStackTy::DSAVarData DVar,
1228                               bool IsLoopIterVar = false) {
1229   if (DVar.RefExpr) {
1230     SemaRef.Diag(DVar.RefExpr->getExprLoc(), diag::note_omp_explicit_dsa)
1231         << getOpenMPClauseName(DVar.CKind);
1232     return;
1233   }
1234   enum {
1235     PDSA_StaticMemberShared,
1236     PDSA_StaticLocalVarShared,
1237     PDSA_LoopIterVarPrivate,
1238     PDSA_LoopIterVarLinear,
1239     PDSA_LoopIterVarLastprivate,
1240     PDSA_ConstVarShared,
1241     PDSA_GlobalVarShared,
1242     PDSA_TaskVarFirstprivate,
1243     PDSA_LocalVarPrivate,
1244     PDSA_Implicit
1245   } Reason = PDSA_Implicit;
1246   bool ReportHint = false;
1247   auto ReportLoc = VD->getLocation();
1248   if (IsLoopIterVar) {
1249     if (DVar.CKind == OMPC_private)
1250       Reason = PDSA_LoopIterVarPrivate;
1251     else if (DVar.CKind == OMPC_lastprivate)
1252       Reason = PDSA_LoopIterVarLastprivate;
1253     else
1254       Reason = PDSA_LoopIterVarLinear;
1255   } else if (DVar.DKind == OMPD_task && DVar.CKind == OMPC_firstprivate) {
1256     Reason = PDSA_TaskVarFirstprivate;
1257     ReportLoc = DVar.ImplicitDSALoc;
1258   } else if (VD->isStaticLocal())
1259     Reason = PDSA_StaticLocalVarShared;
1260   else if (VD->isStaticDataMember())
1261     Reason = PDSA_StaticMemberShared;
1262   else if (VD->isFileVarDecl())
1263     Reason = PDSA_GlobalVarShared;
1264   else if (VD->getType().isConstant(SemaRef.getASTContext()))
1265     Reason = PDSA_ConstVarShared;
1266   else if (VD->isLocalVarDecl() && DVar.CKind == OMPC_private) {
1267     ReportHint = true;
1268     Reason = PDSA_LocalVarPrivate;
1269   }
1270   if (Reason != PDSA_Implicit) {
1271     SemaRef.Diag(ReportLoc, diag::note_omp_predetermined_dsa)
1272         << Reason << ReportHint
1273         << getOpenMPDirectiveName(Stack->getCurrentDirective());
1274   } else if (DVar.ImplicitDSALoc.isValid()) {
1275     SemaRef.Diag(DVar.ImplicitDSALoc, diag::note_omp_implicit_dsa)
1276         << getOpenMPClauseName(DVar.CKind);
1277   }
1278 }
1279 
1280 namespace {
1281 class DSAAttrChecker : public StmtVisitor<DSAAttrChecker, void> {
1282   DSAStackTy *Stack;
1283   Sema &SemaRef;
1284   bool ErrorFound;
1285   CapturedStmt *CS;
1286   llvm::SmallVector<Expr *, 8> ImplicitFirstprivate;
1287   llvm::DenseMap<VarDecl *, Expr *> VarsWithInheritedDSA;
1288 
1289 public:
VisitDeclRefExpr(DeclRefExpr * E)1290   void VisitDeclRefExpr(DeclRefExpr *E) {
1291     if (auto *VD = dyn_cast<VarDecl>(E->getDecl())) {
1292       // Skip internally declared variables.
1293       if (VD->isLocalVarDecl() && !CS->capturesVariable(VD))
1294         return;
1295 
1296       auto DVar = Stack->getTopDSA(VD, false);
1297       // Check if the variable has explicit DSA set and stop analysis if it so.
1298       if (DVar.RefExpr) return;
1299 
1300       auto ELoc = E->getExprLoc();
1301       auto DKind = Stack->getCurrentDirective();
1302       // The default(none) clause requires that each variable that is referenced
1303       // in the construct, and does not have a predetermined data-sharing
1304       // attribute, must have its data-sharing attribute explicitly determined
1305       // by being listed in a data-sharing attribute clause.
1306       if (DVar.CKind == OMPC_unknown && Stack->getDefaultDSA() == DSA_none &&
1307           isParallelOrTaskRegion(DKind) &&
1308           VarsWithInheritedDSA.count(VD) == 0) {
1309         VarsWithInheritedDSA[VD] = E;
1310         return;
1311       }
1312 
1313       // OpenMP [2.9.3.6, Restrictions, p.2]
1314       //  A list item that appears in a reduction clause of the innermost
1315       //  enclosing worksharing or parallel construct may not be accessed in an
1316       //  explicit task.
1317       DVar = Stack->hasInnermostDSA(VD, MatchesAnyClause(OMPC_reduction),
1318                                     [](OpenMPDirectiveKind K) -> bool {
1319                                       return isOpenMPParallelDirective(K) ||
1320                                              isOpenMPWorksharingDirective(K) ||
1321                                              isOpenMPTeamsDirective(K);
1322                                     },
1323                                     false);
1324       if (DKind == OMPD_task && DVar.CKind == OMPC_reduction) {
1325         ErrorFound = true;
1326         SemaRef.Diag(ELoc, diag::err_omp_reduction_in_task);
1327         ReportOriginalDSA(SemaRef, Stack, VD, DVar);
1328         return;
1329       }
1330 
1331       // Define implicit data-sharing attributes for task.
1332       DVar = Stack->getImplicitDSA(VD, false);
1333       if (DKind == OMPD_task && DVar.CKind != OMPC_shared)
1334         ImplicitFirstprivate.push_back(E);
1335     }
1336   }
VisitOMPExecutableDirective(OMPExecutableDirective * S)1337   void VisitOMPExecutableDirective(OMPExecutableDirective *S) {
1338     for (auto *C : S->clauses()) {
1339       // Skip analysis of arguments of implicitly defined firstprivate clause
1340       // for task directives.
1341       if (C && (!isa<OMPFirstprivateClause>(C) || C->getLocStart().isValid()))
1342         for (auto *CC : C->children()) {
1343           if (CC)
1344             Visit(CC);
1345         }
1346     }
1347   }
VisitStmt(Stmt * S)1348   void VisitStmt(Stmt *S) {
1349     for (auto *C : S->children()) {
1350       if (C && !isa<OMPExecutableDirective>(C))
1351         Visit(C);
1352     }
1353   }
1354 
isErrorFound()1355   bool isErrorFound() { return ErrorFound; }
getImplicitFirstprivate()1356   ArrayRef<Expr *> getImplicitFirstprivate() { return ImplicitFirstprivate; }
getVarsWithInheritedDSA()1357   llvm::DenseMap<VarDecl *, Expr *> &getVarsWithInheritedDSA() {
1358     return VarsWithInheritedDSA;
1359   }
1360 
DSAAttrChecker(DSAStackTy * S,Sema & SemaRef,CapturedStmt * CS)1361   DSAAttrChecker(DSAStackTy *S, Sema &SemaRef, CapturedStmt *CS)
1362       : Stack(S), SemaRef(SemaRef), ErrorFound(false), CS(CS) {}
1363 };
1364 } // namespace
1365 
ActOnOpenMPRegionStart(OpenMPDirectiveKind DKind,Scope * CurScope)1366 void Sema::ActOnOpenMPRegionStart(OpenMPDirectiveKind DKind, Scope *CurScope) {
1367   switch (DKind) {
1368   case OMPD_parallel: {
1369     QualType KmpInt32Ty = Context.getIntTypeForBitwidth(32, 1);
1370     QualType KmpInt32PtrTy =
1371         Context.getPointerType(KmpInt32Ty).withConst().withRestrict();
1372     Sema::CapturedParamNameType Params[] = {
1373         std::make_pair(".global_tid.", KmpInt32PtrTy),
1374         std::make_pair(".bound_tid.", KmpInt32PtrTy),
1375         std::make_pair(StringRef(), QualType()) // __context with shared vars
1376     };
1377     ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
1378                              Params);
1379     break;
1380   }
1381   case OMPD_simd: {
1382     Sema::CapturedParamNameType Params[] = {
1383         std::make_pair(StringRef(), QualType()) // __context with shared vars
1384     };
1385     ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
1386                              Params);
1387     break;
1388   }
1389   case OMPD_for: {
1390     Sema::CapturedParamNameType Params[] = {
1391         std::make_pair(StringRef(), QualType()) // __context with shared vars
1392     };
1393     ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
1394                              Params);
1395     break;
1396   }
1397   case OMPD_for_simd: {
1398     Sema::CapturedParamNameType Params[] = {
1399         std::make_pair(StringRef(), QualType()) // __context with shared vars
1400     };
1401     ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
1402                              Params);
1403     break;
1404   }
1405   case OMPD_sections: {
1406     Sema::CapturedParamNameType Params[] = {
1407         std::make_pair(StringRef(), QualType()) // __context with shared vars
1408     };
1409     ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
1410                              Params);
1411     break;
1412   }
1413   case OMPD_section: {
1414     Sema::CapturedParamNameType Params[] = {
1415         std::make_pair(StringRef(), QualType()) // __context with shared vars
1416     };
1417     ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
1418                              Params);
1419     break;
1420   }
1421   case OMPD_single: {
1422     Sema::CapturedParamNameType Params[] = {
1423         std::make_pair(StringRef(), QualType()) // __context with shared vars
1424     };
1425     ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
1426                              Params);
1427     break;
1428   }
1429   case OMPD_master: {
1430     Sema::CapturedParamNameType Params[] = {
1431         std::make_pair(StringRef(), QualType()) // __context with shared vars
1432     };
1433     ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
1434                              Params);
1435     break;
1436   }
1437   case OMPD_critical: {
1438     Sema::CapturedParamNameType Params[] = {
1439         std::make_pair(StringRef(), QualType()) // __context with shared vars
1440     };
1441     ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
1442                              Params);
1443     break;
1444   }
1445   case OMPD_parallel_for: {
1446     QualType KmpInt32Ty = Context.getIntTypeForBitwidth(32, 1);
1447     QualType KmpInt32PtrTy =
1448         Context.getPointerType(KmpInt32Ty).withConst().withRestrict();
1449     Sema::CapturedParamNameType Params[] = {
1450         std::make_pair(".global_tid.", KmpInt32PtrTy),
1451         std::make_pair(".bound_tid.", KmpInt32PtrTy),
1452         std::make_pair(StringRef(), QualType()) // __context with shared vars
1453     };
1454     ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
1455                              Params);
1456     break;
1457   }
1458   case OMPD_parallel_for_simd: {
1459     QualType KmpInt32Ty = Context.getIntTypeForBitwidth(32, 1);
1460     QualType KmpInt32PtrTy =
1461         Context.getPointerType(KmpInt32Ty).withConst().withRestrict();
1462     Sema::CapturedParamNameType Params[] = {
1463         std::make_pair(".global_tid.", KmpInt32PtrTy),
1464         std::make_pair(".bound_tid.", KmpInt32PtrTy),
1465         std::make_pair(StringRef(), QualType()) // __context with shared vars
1466     };
1467     ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
1468                              Params);
1469     break;
1470   }
1471   case OMPD_parallel_sections: {
1472     QualType KmpInt32Ty = Context.getIntTypeForBitwidth(32, 1);
1473     QualType KmpInt32PtrTy =
1474         Context.getPointerType(KmpInt32Ty).withConst().withRestrict();
1475     Sema::CapturedParamNameType Params[] = {
1476         std::make_pair(".global_tid.", KmpInt32PtrTy),
1477         std::make_pair(".bound_tid.", KmpInt32PtrTy),
1478         std::make_pair(StringRef(), QualType()) // __context with shared vars
1479     };
1480     ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
1481                              Params);
1482     break;
1483   }
1484   case OMPD_task: {
1485     QualType KmpInt32Ty = Context.getIntTypeForBitwidth(32, 1);
1486     QualType Args[] = {Context.VoidPtrTy.withConst().withRestrict()};
1487     FunctionProtoType::ExtProtoInfo EPI;
1488     EPI.Variadic = true;
1489     QualType CopyFnType = Context.getFunctionType(Context.VoidTy, Args, EPI);
1490     Sema::CapturedParamNameType Params[] = {
1491         std::make_pair(".global_tid.", KmpInt32Ty),
1492         std::make_pair(".part_id.", KmpInt32Ty),
1493         std::make_pair(".privates.",
1494                        Context.VoidPtrTy.withConst().withRestrict()),
1495         std::make_pair(
1496             ".copy_fn.",
1497             Context.getPointerType(CopyFnType).withConst().withRestrict()),
1498         std::make_pair(StringRef(), QualType()) // __context with shared vars
1499     };
1500     ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
1501                              Params);
1502     // Mark this captured region as inlined, because we don't use outlined
1503     // function directly.
1504     getCurCapturedRegion()->TheCapturedDecl->addAttr(
1505         AlwaysInlineAttr::CreateImplicit(
1506             Context, AlwaysInlineAttr::Keyword_forceinline, SourceRange()));
1507     break;
1508   }
1509   case OMPD_ordered: {
1510     Sema::CapturedParamNameType Params[] = {
1511         std::make_pair(StringRef(), QualType()) // __context with shared vars
1512     };
1513     ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
1514                              Params);
1515     break;
1516   }
1517   case OMPD_atomic: {
1518     Sema::CapturedParamNameType Params[] = {
1519         std::make_pair(StringRef(), QualType()) // __context with shared vars
1520     };
1521     ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
1522                              Params);
1523     break;
1524   }
1525   case OMPD_target_data:
1526   case OMPD_target: {
1527     Sema::CapturedParamNameType Params[] = {
1528         std::make_pair(StringRef(), QualType()) // __context with shared vars
1529     };
1530     ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
1531                              Params);
1532     break;
1533   }
1534   case OMPD_teams: {
1535     QualType KmpInt32Ty = Context.getIntTypeForBitwidth(32, 1);
1536     QualType KmpInt32PtrTy =
1537         Context.getPointerType(KmpInt32Ty).withConst().withRestrict();
1538     Sema::CapturedParamNameType Params[] = {
1539         std::make_pair(".global_tid.", KmpInt32PtrTy),
1540         std::make_pair(".bound_tid.", KmpInt32PtrTy),
1541         std::make_pair(StringRef(), QualType()) // __context with shared vars
1542     };
1543     ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
1544                              Params);
1545     break;
1546   }
1547   case OMPD_taskgroup: {
1548     Sema::CapturedParamNameType Params[] = {
1549         std::make_pair(StringRef(), QualType()) // __context with shared vars
1550     };
1551     ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
1552                              Params);
1553     break;
1554   }
1555   case OMPD_taskloop: {
1556     Sema::CapturedParamNameType Params[] = {
1557         std::make_pair(StringRef(), QualType()) // __context with shared vars
1558     };
1559     ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
1560                              Params);
1561     break;
1562   }
1563   case OMPD_taskloop_simd: {
1564     Sema::CapturedParamNameType Params[] = {
1565         std::make_pair(StringRef(), QualType()) // __context with shared vars
1566     };
1567     ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
1568                              Params);
1569     break;
1570   }
1571   case OMPD_distribute: {
1572     Sema::CapturedParamNameType Params[] = {
1573         std::make_pair(StringRef(), QualType()) // __context with shared vars
1574     };
1575     ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP,
1576                              Params);
1577     break;
1578   }
1579   case OMPD_threadprivate:
1580   case OMPD_taskyield:
1581   case OMPD_barrier:
1582   case OMPD_taskwait:
1583   case OMPD_cancellation_point:
1584   case OMPD_cancel:
1585   case OMPD_flush:
1586     llvm_unreachable("OpenMP Directive is not allowed");
1587   case OMPD_unknown:
1588     llvm_unreachable("Unknown OpenMP directive");
1589   }
1590 }
1591 
ActOnOpenMPRegionEnd(StmtResult S,ArrayRef<OMPClause * > Clauses)1592 StmtResult Sema::ActOnOpenMPRegionEnd(StmtResult S,
1593                                       ArrayRef<OMPClause *> Clauses) {
1594   if (!S.isUsable()) {
1595     ActOnCapturedRegionError();
1596     return StmtError();
1597   }
1598   // This is required for proper codegen.
1599   for (auto *Clause : Clauses) {
1600     if (isOpenMPPrivate(Clause->getClauseKind()) ||
1601         Clause->getClauseKind() == OMPC_copyprivate ||
1602         (getLangOpts().OpenMPUseTLS &&
1603          getASTContext().getTargetInfo().isTLSSupported() &&
1604          Clause->getClauseKind() == OMPC_copyin)) {
1605       DSAStack->setForceVarCapturing(Clause->getClauseKind() == OMPC_copyin);
1606       // Mark all variables in private list clauses as used in inner region.
1607       for (auto *VarRef : Clause->children()) {
1608         if (auto *E = cast_or_null<Expr>(VarRef)) {
1609           MarkDeclarationsReferencedInExpr(E);
1610         }
1611       }
1612       DSAStack->setForceVarCapturing(/*V=*/false);
1613     } else if (isParallelOrTaskRegion(DSAStack->getCurrentDirective()) &&
1614                Clause->getClauseKind() == OMPC_schedule) {
1615       // Mark all variables in private list clauses as used in inner region.
1616       // Required for proper codegen of combined directives.
1617       // TODO: add processing for other clauses.
1618       if (auto *E = cast_or_null<Expr>(
1619               cast<OMPScheduleClause>(Clause)->getHelperChunkSize())) {
1620           MarkDeclarationsReferencedInExpr(E);
1621         }
1622     }
1623   }
1624   return ActOnCapturedRegionEnd(S.get());
1625 }
1626 
CheckNestingOfRegions(Sema & SemaRef,DSAStackTy * Stack,OpenMPDirectiveKind CurrentRegion,const DeclarationNameInfo & CurrentName,OpenMPDirectiveKind CancelRegion,SourceLocation StartLoc)1627 static bool CheckNestingOfRegions(Sema &SemaRef, DSAStackTy *Stack,
1628                                   OpenMPDirectiveKind CurrentRegion,
1629                                   const DeclarationNameInfo &CurrentName,
1630                                   OpenMPDirectiveKind CancelRegion,
1631                                   SourceLocation StartLoc) {
1632   // Allowed nesting of constructs
1633   // +------------------+-----------------+------------------------------------+
1634   // | Parent directive | Child directive | Closely (!), No-Closely(+), Both(*)|
1635   // +------------------+-----------------+------------------------------------+
1636   // | parallel         | parallel        | *                                  |
1637   // | parallel         | for             | *                                  |
1638   // | parallel         | for simd        | *                                  |
1639   // | parallel         | master          | *                                  |
1640   // | parallel         | critical        | *                                  |
1641   // | parallel         | simd            | *                                  |
1642   // | parallel         | sections        | *                                  |
1643   // | parallel         | section         | +                                  |
1644   // | parallel         | single          | *                                  |
1645   // | parallel         | parallel for    | *                                  |
1646   // | parallel         |parallel for simd| *                                  |
1647   // | parallel         |parallel sections| *                                  |
1648   // | parallel         | task            | *                                  |
1649   // | parallel         | taskyield       | *                                  |
1650   // | parallel         | barrier         | *                                  |
1651   // | parallel         | taskwait        | *                                  |
1652   // | parallel         | taskgroup       | *                                  |
1653   // | parallel         | flush           | *                                  |
1654   // | parallel         | ordered         | +                                  |
1655   // | parallel         | atomic          | *                                  |
1656   // | parallel         | target          | *                                  |
1657   // | parallel         | teams           | +                                  |
1658   // | parallel         | cancellation    |                                    |
1659   // |                  | point           | !                                  |
1660   // | parallel         | cancel          | !                                  |
1661   // | parallel         | taskloop        | *                                  |
1662   // | parallel         | taskloop simd   | *                                  |
1663   // | parallel         | distribute      |                                    |
1664   // +------------------+-----------------+------------------------------------+
1665   // | for              | parallel        | *                                  |
1666   // | for              | for             | +                                  |
1667   // | for              | for simd        | +                                  |
1668   // | for              | master          | +                                  |
1669   // | for              | critical        | *                                  |
1670   // | for              | simd            | *                                  |
1671   // | for              | sections        | +                                  |
1672   // | for              | section         | +                                  |
1673   // | for              | single          | +                                  |
1674   // | for              | parallel for    | *                                  |
1675   // | for              |parallel for simd| *                                  |
1676   // | for              |parallel sections| *                                  |
1677   // | for              | task            | *                                  |
1678   // | for              | taskyield       | *                                  |
1679   // | for              | barrier         | +                                  |
1680   // | for              | taskwait        | *                                  |
1681   // | for              | taskgroup       | *                                  |
1682   // | for              | flush           | *                                  |
1683   // | for              | ordered         | * (if construct is ordered)        |
1684   // | for              | atomic          | *                                  |
1685   // | for              | target          | *                                  |
1686   // | for              | teams           | +                                  |
1687   // | for              | cancellation    |                                    |
1688   // |                  | point           | !                                  |
1689   // | for              | cancel          | !                                  |
1690   // | for              | taskloop        | *                                  |
1691   // | for              | taskloop simd   | *                                  |
1692   // | for              | distribute      |                                    |
1693   // +------------------+-----------------+------------------------------------+
1694   // | master           | parallel        | *                                  |
1695   // | master           | for             | +                                  |
1696   // | master           | for simd        | +                                  |
1697   // | master           | master          | *                                  |
1698   // | master           | critical        | *                                  |
1699   // | master           | simd            | *                                  |
1700   // | master           | sections        | +                                  |
1701   // | master           | section         | +                                  |
1702   // | master           | single          | +                                  |
1703   // | master           | parallel for    | *                                  |
1704   // | master           |parallel for simd| *                                  |
1705   // | master           |parallel sections| *                                  |
1706   // | master           | task            | *                                  |
1707   // | master           | taskyield       | *                                  |
1708   // | master           | barrier         | +                                  |
1709   // | master           | taskwait        | *                                  |
1710   // | master           | taskgroup       | *                                  |
1711   // | master           | flush           | *                                  |
1712   // | master           | ordered         | +                                  |
1713   // | master           | atomic          | *                                  |
1714   // | master           | target          | *                                  |
1715   // | master           | teams           | +                                  |
1716   // | master           | cancellation    |                                    |
1717   // |                  | point           |                                    |
1718   // | master           | cancel          |                                    |
1719   // | master           | taskloop        | *                                  |
1720   // | master           | taskloop simd   | *                                  |
1721   // | master           | distribute      |                                    |
1722   // +------------------+-----------------+------------------------------------+
1723   // | critical         | parallel        | *                                  |
1724   // | critical         | for             | +                                  |
1725   // | critical         | for simd        | +                                  |
1726   // | critical         | master          | *                                  |
1727   // | critical         | critical        | * (should have different names)    |
1728   // | critical         | simd            | *                                  |
1729   // | critical         | sections        | +                                  |
1730   // | critical         | section         | +                                  |
1731   // | critical         | single          | +                                  |
1732   // | critical         | parallel for    | *                                  |
1733   // | critical         |parallel for simd| *                                  |
1734   // | critical         |parallel sections| *                                  |
1735   // | critical         | task            | *                                  |
1736   // | critical         | taskyield       | *                                  |
1737   // | critical         | barrier         | +                                  |
1738   // | critical         | taskwait        | *                                  |
1739   // | critical         | taskgroup       | *                                  |
1740   // | critical         | ordered         | +                                  |
1741   // | critical         | atomic          | *                                  |
1742   // | critical         | target          | *                                  |
1743   // | critical         | teams           | +                                  |
1744   // | critical         | cancellation    |                                    |
1745   // |                  | point           |                                    |
1746   // | critical         | cancel          |                                    |
1747   // | critical         | taskloop        | *                                  |
1748   // | critical         | taskloop simd   | *                                  |
1749   // | critical         | distribute      |                                    |
1750   // +------------------+-----------------+------------------------------------+
1751   // | simd             | parallel        |                                    |
1752   // | simd             | for             |                                    |
1753   // | simd             | for simd        |                                    |
1754   // | simd             | master          |                                    |
1755   // | simd             | critical        |                                    |
1756   // | simd             | simd            |                                    |
1757   // | simd             | sections        |                                    |
1758   // | simd             | section         |                                    |
1759   // | simd             | single          |                                    |
1760   // | simd             | parallel for    |                                    |
1761   // | simd             |parallel for simd|                                    |
1762   // | simd             |parallel sections|                                    |
1763   // | simd             | task            |                                    |
1764   // | simd             | taskyield       |                                    |
1765   // | simd             | barrier         |                                    |
1766   // | simd             | taskwait        |                                    |
1767   // | simd             | taskgroup       |                                    |
1768   // | simd             | flush           |                                    |
1769   // | simd             | ordered         | + (with simd clause)               |
1770   // | simd             | atomic          |                                    |
1771   // | simd             | target          |                                    |
1772   // | simd             | teams           |                                    |
1773   // | simd             | cancellation    |                                    |
1774   // |                  | point           |                                    |
1775   // | simd             | cancel          |                                    |
1776   // | simd             | taskloop        |                                    |
1777   // | simd             | taskloop simd   |                                    |
1778   // | simd             | distribute      |                                    |
1779   // +------------------+-----------------+------------------------------------+
1780   // | for simd         | parallel        |                                    |
1781   // | for simd         | for             |                                    |
1782   // | for simd         | for simd        |                                    |
1783   // | for simd         | master          |                                    |
1784   // | for simd         | critical        |                                    |
1785   // | for simd         | simd            |                                    |
1786   // | for simd         | sections        |                                    |
1787   // | for simd         | section         |                                    |
1788   // | for simd         | single          |                                    |
1789   // | for simd         | parallel for    |                                    |
1790   // | for simd         |parallel for simd|                                    |
1791   // | for simd         |parallel sections|                                    |
1792   // | for simd         | task            |                                    |
1793   // | for simd         | taskyield       |                                    |
1794   // | for simd         | barrier         |                                    |
1795   // | for simd         | taskwait        |                                    |
1796   // | for simd         | taskgroup       |                                    |
1797   // | for simd         | flush           |                                    |
1798   // | for simd         | ordered         | + (with simd clause)               |
1799   // | for simd         | atomic          |                                    |
1800   // | for simd         | target          |                                    |
1801   // | for simd         | teams           |                                    |
1802   // | for simd         | cancellation    |                                    |
1803   // |                  | point           |                                    |
1804   // | for simd         | cancel          |                                    |
1805   // | for simd         | taskloop        |                                    |
1806   // | for simd         | taskloop simd   |                                    |
1807   // | for simd         | distribute      |                                    |
1808   // +------------------+-----------------+------------------------------------+
1809   // | parallel for simd| parallel        |                                    |
1810   // | parallel for simd| for             |                                    |
1811   // | parallel for simd| for simd        |                                    |
1812   // | parallel for simd| master          |                                    |
1813   // | parallel for simd| critical        |                                    |
1814   // | parallel for simd| simd            |                                    |
1815   // | parallel for simd| sections        |                                    |
1816   // | parallel for simd| section         |                                    |
1817   // | parallel for simd| single          |                                    |
1818   // | parallel for simd| parallel for    |                                    |
1819   // | parallel for simd|parallel for simd|                                    |
1820   // | parallel for simd|parallel sections|                                    |
1821   // | parallel for simd| task            |                                    |
1822   // | parallel for simd| taskyield       |                                    |
1823   // | parallel for simd| barrier         |                                    |
1824   // | parallel for simd| taskwait        |                                    |
1825   // | parallel for simd| taskgroup       |                                    |
1826   // | parallel for simd| flush           |                                    |
1827   // | parallel for simd| ordered         | + (with simd clause)               |
1828   // | parallel for simd| atomic          |                                    |
1829   // | parallel for simd| target          |                                    |
1830   // | parallel for simd| teams           |                                    |
1831   // | parallel for simd| cancellation    |                                    |
1832   // |                  | point           |                                    |
1833   // | parallel for simd| cancel          |                                    |
1834   // | parallel for simd| taskloop        |                                    |
1835   // | parallel for simd| taskloop simd   |                                    |
1836   // | parallel for simd| distribute      |                                    |
1837   // +------------------+-----------------+------------------------------------+
1838   // | sections         | parallel        | *                                  |
1839   // | sections         | for             | +                                  |
1840   // | sections         | for simd        | +                                  |
1841   // | sections         | master          | +                                  |
1842   // | sections         | critical        | *                                  |
1843   // | sections         | simd            | *                                  |
1844   // | sections         | sections        | +                                  |
1845   // | sections         | section         | *                                  |
1846   // | sections         | single          | +                                  |
1847   // | sections         | parallel for    | *                                  |
1848   // | sections         |parallel for simd| *                                  |
1849   // | sections         |parallel sections| *                                  |
1850   // | sections         | task            | *                                  |
1851   // | sections         | taskyield       | *                                  |
1852   // | sections         | barrier         | +                                  |
1853   // | sections         | taskwait        | *                                  |
1854   // | sections         | taskgroup       | *                                  |
1855   // | sections         | flush           | *                                  |
1856   // | sections         | ordered         | +                                  |
1857   // | sections         | atomic          | *                                  |
1858   // | sections         | target          | *                                  |
1859   // | sections         | teams           | +                                  |
1860   // | sections         | cancellation    |                                    |
1861   // |                  | point           | !                                  |
1862   // | sections         | cancel          | !                                  |
1863   // | sections         | taskloop        | *                                  |
1864   // | sections         | taskloop simd   | *                                  |
1865   // | sections         | distribute      |                                    |
1866   // +------------------+-----------------+------------------------------------+
1867   // | section          | parallel        | *                                  |
1868   // | section          | for             | +                                  |
1869   // | section          | for simd        | +                                  |
1870   // | section          | master          | +                                  |
1871   // | section          | critical        | *                                  |
1872   // | section          | simd            | *                                  |
1873   // | section          | sections        | +                                  |
1874   // | section          | section         | +                                  |
1875   // | section          | single          | +                                  |
1876   // | section          | parallel for    | *                                  |
1877   // | section          |parallel for simd| *                                  |
1878   // | section          |parallel sections| *                                  |
1879   // | section          | task            | *                                  |
1880   // | section          | taskyield       | *                                  |
1881   // | section          | barrier         | +                                  |
1882   // | section          | taskwait        | *                                  |
1883   // | section          | taskgroup       | *                                  |
1884   // | section          | flush           | *                                  |
1885   // | section          | ordered         | +                                  |
1886   // | section          | atomic          | *                                  |
1887   // | section          | target          | *                                  |
1888   // | section          | teams           | +                                  |
1889   // | section          | cancellation    |                                    |
1890   // |                  | point           | !                                  |
1891   // | section          | cancel          | !                                  |
1892   // | section          | taskloop        | *                                  |
1893   // | section          | taskloop simd   | *                                  |
1894   // | section          | distribute      |                                    |
1895   // +------------------+-----------------+------------------------------------+
1896   // | single           | parallel        | *                                  |
1897   // | single           | for             | +                                  |
1898   // | single           | for simd        | +                                  |
1899   // | single           | master          | +                                  |
1900   // | single           | critical        | *                                  |
1901   // | single           | simd            | *                                  |
1902   // | single           | sections        | +                                  |
1903   // | single           | section         | +                                  |
1904   // | single           | single          | +                                  |
1905   // | single           | parallel for    | *                                  |
1906   // | single           |parallel for simd| *                                  |
1907   // | single           |parallel sections| *                                  |
1908   // | single           | task            | *                                  |
1909   // | single           | taskyield       | *                                  |
1910   // | single           | barrier         | +                                  |
1911   // | single           | taskwait        | *                                  |
1912   // | single           | taskgroup       | *                                  |
1913   // | single           | flush           | *                                  |
1914   // | single           | ordered         | +                                  |
1915   // | single           | atomic          | *                                  |
1916   // | single           | target          | *                                  |
1917   // | single           | teams           | +                                  |
1918   // | single           | cancellation    |                                    |
1919   // |                  | point           |                                    |
1920   // | single           | cancel          |                                    |
1921   // | single           | taskloop        | *                                  |
1922   // | single           | taskloop simd   | *                                  |
1923   // | single           | distribute      |                                    |
1924   // +------------------+-----------------+------------------------------------+
1925   // | parallel for     | parallel        | *                                  |
1926   // | parallel for     | for             | +                                  |
1927   // | parallel for     | for simd        | +                                  |
1928   // | parallel for     | master          | +                                  |
1929   // | parallel for     | critical        | *                                  |
1930   // | parallel for     | simd            | *                                  |
1931   // | parallel for     | sections        | +                                  |
1932   // | parallel for     | section         | +                                  |
1933   // | parallel for     | single          | +                                  |
1934   // | parallel for     | parallel for    | *                                  |
1935   // | parallel for     |parallel for simd| *                                  |
1936   // | parallel for     |parallel sections| *                                  |
1937   // | parallel for     | task            | *                                  |
1938   // | parallel for     | taskyield       | *                                  |
1939   // | parallel for     | barrier         | +                                  |
1940   // | parallel for     | taskwait        | *                                  |
1941   // | parallel for     | taskgroup       | *                                  |
1942   // | parallel for     | flush           | *                                  |
1943   // | parallel for     | ordered         | * (if construct is ordered)        |
1944   // | parallel for     | atomic          | *                                  |
1945   // | parallel for     | target          | *                                  |
1946   // | parallel for     | teams           | +                                  |
1947   // | parallel for     | cancellation    |                                    |
1948   // |                  | point           | !                                  |
1949   // | parallel for     | cancel          | !                                  |
1950   // | parallel for     | taskloop        | *                                  |
1951   // | parallel for     | taskloop simd   | *                                  |
1952   // | parallel for     | distribute      |                                    |
1953   // +------------------+-----------------+------------------------------------+
1954   // | parallel sections| parallel        | *                                  |
1955   // | parallel sections| for             | +                                  |
1956   // | parallel sections| for simd        | +                                  |
1957   // | parallel sections| master          | +                                  |
1958   // | parallel sections| critical        | +                                  |
1959   // | parallel sections| simd            | *                                  |
1960   // | parallel sections| sections        | +                                  |
1961   // | parallel sections| section         | *                                  |
1962   // | parallel sections| single          | +                                  |
1963   // | parallel sections| parallel for    | *                                  |
1964   // | parallel sections|parallel for simd| *                                  |
1965   // | parallel sections|parallel sections| *                                  |
1966   // | parallel sections| task            | *                                  |
1967   // | parallel sections| taskyield       | *                                  |
1968   // | parallel sections| barrier         | +                                  |
1969   // | parallel sections| taskwait        | *                                  |
1970   // | parallel sections| taskgroup       | *                                  |
1971   // | parallel sections| flush           | *                                  |
1972   // | parallel sections| ordered         | +                                  |
1973   // | parallel sections| atomic          | *                                  |
1974   // | parallel sections| target          | *                                  |
1975   // | parallel sections| teams           | +                                  |
1976   // | parallel sections| cancellation    |                                    |
1977   // |                  | point           | !                                  |
1978   // | parallel sections| cancel          | !                                  |
1979   // | parallel sections| taskloop        | *                                  |
1980   // | parallel sections| taskloop simd   | *                                  |
1981   // | parallel sections| distribute      |                                    |
1982   // +------------------+-----------------+------------------------------------+
1983   // | task             | parallel        | *                                  |
1984   // | task             | for             | +                                  |
1985   // | task             | for simd        | +                                  |
1986   // | task             | master          | +                                  |
1987   // | task             | critical        | *                                  |
1988   // | task             | simd            | *                                  |
1989   // | task             | sections        | +                                  |
1990   // | task             | section         | +                                  |
1991   // | task             | single          | +                                  |
1992   // | task             | parallel for    | *                                  |
1993   // | task             |parallel for simd| *                                  |
1994   // | task             |parallel sections| *                                  |
1995   // | task             | task            | *                                  |
1996   // | task             | taskyield       | *                                  |
1997   // | task             | barrier         | +                                  |
1998   // | task             | taskwait        | *                                  |
1999   // | task             | taskgroup       | *                                  |
2000   // | task             | flush           | *                                  |
2001   // | task             | ordered         | +                                  |
2002   // | task             | atomic          | *                                  |
2003   // | task             | target          | *                                  |
2004   // | task             | teams           | +                                  |
2005   // | task             | cancellation    |                                    |
2006   // |                  | point           | !                                  |
2007   // | task             | cancel          | !                                  |
2008   // | task             | taskloop        | *                                  |
2009   // | task             | taskloop simd   | *                                  |
2010   // | task             | distribute      |                                    |
2011   // +------------------+-----------------+------------------------------------+
2012   // | ordered          | parallel        | *                                  |
2013   // | ordered          | for             | +                                  |
2014   // | ordered          | for simd        | +                                  |
2015   // | ordered          | master          | *                                  |
2016   // | ordered          | critical        | *                                  |
2017   // | ordered          | simd            | *                                  |
2018   // | ordered          | sections        | +                                  |
2019   // | ordered          | section         | +                                  |
2020   // | ordered          | single          | +                                  |
2021   // | ordered          | parallel for    | *                                  |
2022   // | ordered          |parallel for simd| *                                  |
2023   // | ordered          |parallel sections| *                                  |
2024   // | ordered          | task            | *                                  |
2025   // | ordered          | taskyield       | *                                  |
2026   // | ordered          | barrier         | +                                  |
2027   // | ordered          | taskwait        | *                                  |
2028   // | ordered          | taskgroup       | *                                  |
2029   // | ordered          | flush           | *                                  |
2030   // | ordered          | ordered         | +                                  |
2031   // | ordered          | atomic          | *                                  |
2032   // | ordered          | target          | *                                  |
2033   // | ordered          | teams           | +                                  |
2034   // | ordered          | cancellation    |                                    |
2035   // |                  | point           |                                    |
2036   // | ordered          | cancel          |                                    |
2037   // | ordered          | taskloop        | *                                  |
2038   // | ordered          | taskloop simd   | *                                  |
2039   // | ordered          | distribute      |                                    |
2040   // +------------------+-----------------+------------------------------------+
2041   // | atomic           | parallel        |                                    |
2042   // | atomic           | for             |                                    |
2043   // | atomic           | for simd        |                                    |
2044   // | atomic           | master          |                                    |
2045   // | atomic           | critical        |                                    |
2046   // | atomic           | simd            |                                    |
2047   // | atomic           | sections        |                                    |
2048   // | atomic           | section         |                                    |
2049   // | atomic           | single          |                                    |
2050   // | atomic           | parallel for    |                                    |
2051   // | atomic           |parallel for simd|                                    |
2052   // | atomic           |parallel sections|                                    |
2053   // | atomic           | task            |                                    |
2054   // | atomic           | taskyield       |                                    |
2055   // | atomic           | barrier         |                                    |
2056   // | atomic           | taskwait        |                                    |
2057   // | atomic           | taskgroup       |                                    |
2058   // | atomic           | flush           |                                    |
2059   // | atomic           | ordered         |                                    |
2060   // | atomic           | atomic          |                                    |
2061   // | atomic           | target          |                                    |
2062   // | atomic           | teams           |                                    |
2063   // | atomic           | cancellation    |                                    |
2064   // |                  | point           |                                    |
2065   // | atomic           | cancel          |                                    |
2066   // | atomic           | taskloop        |                                    |
2067   // | atomic           | taskloop simd   |                                    |
2068   // | atomic           | distribute      |                                    |
2069   // +------------------+-----------------+------------------------------------+
2070   // | target           | parallel        | *                                  |
2071   // | target           | for             | *                                  |
2072   // | target           | for simd        | *                                  |
2073   // | target           | master          | *                                  |
2074   // | target           | critical        | *                                  |
2075   // | target           | simd            | *                                  |
2076   // | target           | sections        | *                                  |
2077   // | target           | section         | *                                  |
2078   // | target           | single          | *                                  |
2079   // | target           | parallel for    | *                                  |
2080   // | target           |parallel for simd| *                                  |
2081   // | target           |parallel sections| *                                  |
2082   // | target           | task            | *                                  |
2083   // | target           | taskyield       | *                                  |
2084   // | target           | barrier         | *                                  |
2085   // | target           | taskwait        | *                                  |
2086   // | target           | taskgroup       | *                                  |
2087   // | target           | flush           | *                                  |
2088   // | target           | ordered         | *                                  |
2089   // | target           | atomic          | *                                  |
2090   // | target           | target          | *                                  |
2091   // | target           | teams           | *                                  |
2092   // | target           | cancellation    |                                    |
2093   // |                  | point           |                                    |
2094   // | target           | cancel          |                                    |
2095   // | target           | taskloop        | *                                  |
2096   // | target           | taskloop simd   | *                                  |
2097   // | target           | distribute      |                                    |
2098   // +------------------+-----------------+------------------------------------+
2099   // | teams            | parallel        | *                                  |
2100   // | teams            | for             | +                                  |
2101   // | teams            | for simd        | +                                  |
2102   // | teams            | master          | +                                  |
2103   // | teams            | critical        | +                                  |
2104   // | teams            | simd            | +                                  |
2105   // | teams            | sections        | +                                  |
2106   // | teams            | section         | +                                  |
2107   // | teams            | single          | +                                  |
2108   // | teams            | parallel for    | *                                  |
2109   // | teams            |parallel for simd| *                                  |
2110   // | teams            |parallel sections| *                                  |
2111   // | teams            | task            | +                                  |
2112   // | teams            | taskyield       | +                                  |
2113   // | teams            | barrier         | +                                  |
2114   // | teams            | taskwait        | +                                  |
2115   // | teams            | taskgroup       | +                                  |
2116   // | teams            | flush           | +                                  |
2117   // | teams            | ordered         | +                                  |
2118   // | teams            | atomic          | +                                  |
2119   // | teams            | target          | +                                  |
2120   // | teams            | teams           | +                                  |
2121   // | teams            | cancellation    |                                    |
2122   // |                  | point           |                                    |
2123   // | teams            | cancel          |                                    |
2124   // | teams            | taskloop        | +                                  |
2125   // | teams            | taskloop simd   | +                                  |
2126   // | teams            | distribute      | !                                  |
2127   // +------------------+-----------------+------------------------------------+
2128   // | taskloop         | parallel        | *                                  |
2129   // | taskloop         | for             | +                                  |
2130   // | taskloop         | for simd        | +                                  |
2131   // | taskloop         | master          | +                                  |
2132   // | taskloop         | critical        | *                                  |
2133   // | taskloop         | simd            | *                                  |
2134   // | taskloop         | sections        | +                                  |
2135   // | taskloop         | section         | +                                  |
2136   // | taskloop         | single          | +                                  |
2137   // | taskloop         | parallel for    | *                                  |
2138   // | taskloop         |parallel for simd| *                                  |
2139   // | taskloop         |parallel sections| *                                  |
2140   // | taskloop         | task            | *                                  |
2141   // | taskloop         | taskyield       | *                                  |
2142   // | taskloop         | barrier         | +                                  |
2143   // | taskloop         | taskwait        | *                                  |
2144   // | taskloop         | taskgroup       | *                                  |
2145   // | taskloop         | flush           | *                                  |
2146   // | taskloop         | ordered         | +                                  |
2147   // | taskloop         | atomic          | *                                  |
2148   // | taskloop         | target          | *                                  |
2149   // | taskloop         | teams           | +                                  |
2150   // | taskloop         | cancellation    |                                    |
2151   // |                  | point           |                                    |
2152   // | taskloop         | cancel          |                                    |
2153   // | taskloop         | taskloop        | *                                  |
2154   // | taskloop         | distribute      |                                    |
2155   // +------------------+-----------------+------------------------------------+
2156   // | taskloop simd    | parallel        |                                    |
2157   // | taskloop simd    | for             |                                    |
2158   // | taskloop simd    | for simd        |                                    |
2159   // | taskloop simd    | master          |                                    |
2160   // | taskloop simd    | critical        |                                    |
2161   // | taskloop simd    | simd            |                                    |
2162   // | taskloop simd    | sections        |                                    |
2163   // | taskloop simd    | section         |                                    |
2164   // | taskloop simd    | single          |                                    |
2165   // | taskloop simd    | parallel for    |                                    |
2166   // | taskloop simd    |parallel for simd|                                    |
2167   // | taskloop simd    |parallel sections|                                    |
2168   // | taskloop simd    | task            |                                    |
2169   // | taskloop simd    | taskyield       |                                    |
2170   // | taskloop simd    | barrier         |                                    |
2171   // | taskloop simd    | taskwait        |                                    |
2172   // | taskloop simd    | taskgroup       |                                    |
2173   // | taskloop simd    | flush           |                                    |
2174   // | taskloop simd    | ordered         | + (with simd clause)               |
2175   // | taskloop simd    | atomic          |                                    |
2176   // | taskloop simd    | target          |                                    |
2177   // | taskloop simd    | teams           |                                    |
2178   // | taskloop simd    | cancellation    |                                    |
2179   // |                  | point           |                                    |
2180   // | taskloop simd    | cancel          |                                    |
2181   // | taskloop simd    | taskloop        |                                    |
2182   // | taskloop simd    | taskloop simd   |                                    |
2183   // | taskloop simd    | distribute      |                                    |
2184   // +------------------+-----------------+------------------------------------+
2185   // | distribute       | parallel        | *                                  |
2186   // | distribute       | for             | *                                  |
2187   // | distribute       | for simd        | *                                  |
2188   // | distribute       | master          | *                                  |
2189   // | distribute       | critical        | *                                  |
2190   // | distribute       | simd            | *                                  |
2191   // | distribute       | sections        | *                                  |
2192   // | distribute       | section         | *                                  |
2193   // | distribute       | single          | *                                  |
2194   // | distribute       | parallel for    | *                                  |
2195   // | distribute       |parallel for simd| *                                  |
2196   // | distribute       |parallel sections| *                                  |
2197   // | distribute       | task            | *                                  |
2198   // | distribute       | taskyield       | *                                  |
2199   // | distribute       | barrier         | *                                  |
2200   // | distribute       | taskwait        | *                                  |
2201   // | distribute       | taskgroup       | *                                  |
2202   // | distribute       | flush           | *                                  |
2203   // | distribute       | ordered         | +                                  |
2204   // | distribute       | atomic          | *                                  |
2205   // | distribute       | target          |                                    |
2206   // | distribute       | teams           |                                    |
2207   // | distribute       | cancellation    | +                                  |
2208   // |                  | point           |                                    |
2209   // | distribute       | cancel          | +                                  |
2210   // | distribute       | taskloop        | *                                  |
2211   // | distribute       | taskloop simd   | *                                  |
2212   // | distribute       | distribute      |                                    |
2213   // +------------------+-----------------+------------------------------------+
2214   if (Stack->getCurScope()) {
2215     auto ParentRegion = Stack->getParentDirective();
2216     bool NestingProhibited = false;
2217     bool CloseNesting = true;
2218     enum {
2219       NoRecommend,
2220       ShouldBeInParallelRegion,
2221       ShouldBeInOrderedRegion,
2222       ShouldBeInTargetRegion,
2223       ShouldBeInTeamsRegion
2224     } Recommend = NoRecommend;
2225     if (isOpenMPSimdDirective(ParentRegion) && CurrentRegion != OMPD_ordered) {
2226       // OpenMP [2.16, Nesting of Regions]
2227       // OpenMP constructs may not be nested inside a simd region.
2228       // OpenMP [2.8.1,simd Construct, Restrictions]
2229       // An ordered construct with the simd clause is the only OpenMP construct
2230       // that can appear in the simd region.
2231       SemaRef.Diag(StartLoc, diag::err_omp_prohibited_region_simd);
2232       return true;
2233     }
2234     if (ParentRegion == OMPD_atomic) {
2235       // OpenMP [2.16, Nesting of Regions]
2236       // OpenMP constructs may not be nested inside an atomic region.
2237       SemaRef.Diag(StartLoc, diag::err_omp_prohibited_region_atomic);
2238       return true;
2239     }
2240     if (CurrentRegion == OMPD_section) {
2241       // OpenMP [2.7.2, sections Construct, Restrictions]
2242       // Orphaned section directives are prohibited. That is, the section
2243       // directives must appear within the sections construct and must not be
2244       // encountered elsewhere in the sections region.
2245       if (ParentRegion != OMPD_sections &&
2246           ParentRegion != OMPD_parallel_sections) {
2247         SemaRef.Diag(StartLoc, diag::err_omp_orphaned_section_directive)
2248             << (ParentRegion != OMPD_unknown)
2249             << getOpenMPDirectiveName(ParentRegion);
2250         return true;
2251       }
2252       return false;
2253     }
2254     // Allow some constructs to be orphaned (they could be used in functions,
2255     // called from OpenMP regions with the required preconditions).
2256     if (ParentRegion == OMPD_unknown)
2257       return false;
2258     if (CurrentRegion == OMPD_cancellation_point ||
2259         CurrentRegion == OMPD_cancel) {
2260       // OpenMP [2.16, Nesting of Regions]
2261       // A cancellation point construct for which construct-type-clause is
2262       // taskgroup must be nested inside a task construct. A cancellation
2263       // point construct for which construct-type-clause is not taskgroup must
2264       // be closely nested inside an OpenMP construct that matches the type
2265       // specified in construct-type-clause.
2266       // A cancel construct for which construct-type-clause is taskgroup must be
2267       // nested inside a task construct. A cancel construct for which
2268       // construct-type-clause is not taskgroup must be closely nested inside an
2269       // OpenMP construct that matches the type specified in
2270       // construct-type-clause.
2271       NestingProhibited =
2272           !((CancelRegion == OMPD_parallel && ParentRegion == OMPD_parallel) ||
2273             (CancelRegion == OMPD_for &&
2274              (ParentRegion == OMPD_for || ParentRegion == OMPD_parallel_for)) ||
2275             (CancelRegion == OMPD_taskgroup && ParentRegion == OMPD_task) ||
2276             (CancelRegion == OMPD_sections &&
2277              (ParentRegion == OMPD_section || ParentRegion == OMPD_sections ||
2278               ParentRegion == OMPD_parallel_sections)));
2279     } else if (CurrentRegion == OMPD_master) {
2280       // OpenMP [2.16, Nesting of Regions]
2281       // A master region may not be closely nested inside a worksharing,
2282       // atomic, or explicit task region.
2283       NestingProhibited = isOpenMPWorksharingDirective(ParentRegion) ||
2284                           ParentRegion == OMPD_task ||
2285                           isOpenMPTaskLoopDirective(ParentRegion);
2286     } else if (CurrentRegion == OMPD_critical && CurrentName.getName()) {
2287       // OpenMP [2.16, Nesting of Regions]
2288       // A critical region may not be nested (closely or otherwise) inside a
2289       // critical region with the same name. Note that this restriction is not
2290       // sufficient to prevent deadlock.
2291       SourceLocation PreviousCriticalLoc;
2292       bool DeadLock =
2293           Stack->hasDirective([CurrentName, &PreviousCriticalLoc](
2294                                   OpenMPDirectiveKind K,
2295                                   const DeclarationNameInfo &DNI,
2296                                   SourceLocation Loc)
2297                                   ->bool {
2298                                 if (K == OMPD_critical &&
2299                                     DNI.getName() == CurrentName.getName()) {
2300                                   PreviousCriticalLoc = Loc;
2301                                   return true;
2302                                 } else
2303                                   return false;
2304                               },
2305                               false /* skip top directive */);
2306       if (DeadLock) {
2307         SemaRef.Diag(StartLoc,
2308                      diag::err_omp_prohibited_region_critical_same_name)
2309             << CurrentName.getName();
2310         if (PreviousCriticalLoc.isValid())
2311           SemaRef.Diag(PreviousCriticalLoc,
2312                        diag::note_omp_previous_critical_region);
2313         return true;
2314       }
2315     } else if (CurrentRegion == OMPD_barrier) {
2316       // OpenMP [2.16, Nesting of Regions]
2317       // A barrier region may not be closely nested inside a worksharing,
2318       // explicit task, critical, ordered, atomic, or master region.
2319       NestingProhibited =
2320           isOpenMPWorksharingDirective(ParentRegion) ||
2321           ParentRegion == OMPD_task || ParentRegion == OMPD_master ||
2322           ParentRegion == OMPD_critical || ParentRegion == OMPD_ordered ||
2323           isOpenMPTaskLoopDirective(ParentRegion);
2324     } else if (isOpenMPWorksharingDirective(CurrentRegion) &&
2325                !isOpenMPParallelDirective(CurrentRegion)) {
2326       // OpenMP [2.16, Nesting of Regions]
2327       // A worksharing region may not be closely nested inside a worksharing,
2328       // explicit task, critical, ordered, atomic, or master region.
2329       NestingProhibited =
2330           isOpenMPWorksharingDirective(ParentRegion) ||
2331           ParentRegion == OMPD_task || ParentRegion == OMPD_master ||
2332           ParentRegion == OMPD_critical || ParentRegion == OMPD_ordered ||
2333           isOpenMPTaskLoopDirective(ParentRegion);
2334       Recommend = ShouldBeInParallelRegion;
2335     } else if (CurrentRegion == OMPD_ordered) {
2336       // OpenMP [2.16, Nesting of Regions]
2337       // An ordered region may not be closely nested inside a critical,
2338       // atomic, or explicit task region.
2339       // An ordered region must be closely nested inside a loop region (or
2340       // parallel loop region) with an ordered clause.
2341       // OpenMP [2.8.1,simd Construct, Restrictions]
2342       // An ordered construct with the simd clause is the only OpenMP construct
2343       // that can appear in the simd region.
2344       NestingProhibited = ParentRegion == OMPD_critical ||
2345                           ParentRegion == OMPD_task ||
2346                           isOpenMPTaskLoopDirective(ParentRegion) ||
2347                           !(isOpenMPSimdDirective(ParentRegion) ||
2348                             Stack->isParentOrderedRegion());
2349       Recommend = ShouldBeInOrderedRegion;
2350     } else if (isOpenMPTeamsDirective(CurrentRegion)) {
2351       // OpenMP [2.16, Nesting of Regions]
2352       // If specified, a teams construct must be contained within a target
2353       // construct.
2354       NestingProhibited = ParentRegion != OMPD_target;
2355       Recommend = ShouldBeInTargetRegion;
2356       Stack->setParentTeamsRegionLoc(Stack->getConstructLoc());
2357     }
2358     if (!NestingProhibited && isOpenMPTeamsDirective(ParentRegion)) {
2359       // OpenMP [2.16, Nesting of Regions]
2360       // distribute, parallel, parallel sections, parallel workshare, and the
2361       // parallel loop and parallel loop SIMD constructs are the only OpenMP
2362       // constructs that can be closely nested in the teams region.
2363       NestingProhibited = !isOpenMPParallelDirective(CurrentRegion) &&
2364                           !isOpenMPDistributeDirective(CurrentRegion);
2365       Recommend = ShouldBeInParallelRegion;
2366     }
2367     if (!NestingProhibited && isOpenMPDistributeDirective(CurrentRegion)) {
2368       // OpenMP 4.5 [2.17 Nesting of Regions]
2369       // The region associated with the distribute construct must be strictly
2370       // nested inside a teams region
2371       NestingProhibited = !isOpenMPTeamsDirective(ParentRegion);
2372       Recommend = ShouldBeInTeamsRegion;
2373     }
2374     if (NestingProhibited) {
2375       SemaRef.Diag(StartLoc, diag::err_omp_prohibited_region)
2376           << CloseNesting << getOpenMPDirectiveName(ParentRegion) << Recommend
2377           << getOpenMPDirectiveName(CurrentRegion);
2378       return true;
2379     }
2380   }
2381   return false;
2382 }
2383 
checkIfClauses(Sema & S,OpenMPDirectiveKind Kind,ArrayRef<OMPClause * > Clauses,ArrayRef<OpenMPDirectiveKind> AllowedNameModifiers)2384 static bool checkIfClauses(Sema &S, OpenMPDirectiveKind Kind,
2385                            ArrayRef<OMPClause *> Clauses,
2386                            ArrayRef<OpenMPDirectiveKind> AllowedNameModifiers) {
2387   bool ErrorFound = false;
2388   unsigned NamedModifiersNumber = 0;
2389   SmallVector<const OMPIfClause *, OMPC_unknown + 1> FoundNameModifiers(
2390       OMPD_unknown + 1);
2391   SmallVector<SourceLocation, 4> NameModifierLoc;
2392   for (const auto *C : Clauses) {
2393     if (const auto *IC = dyn_cast_or_null<OMPIfClause>(C)) {
2394       // At most one if clause without a directive-name-modifier can appear on
2395       // the directive.
2396       OpenMPDirectiveKind CurNM = IC->getNameModifier();
2397       if (FoundNameModifiers[CurNM]) {
2398         S.Diag(C->getLocStart(), diag::err_omp_more_one_clause)
2399             << getOpenMPDirectiveName(Kind) << getOpenMPClauseName(OMPC_if)
2400             << (CurNM != OMPD_unknown) << getOpenMPDirectiveName(CurNM);
2401         ErrorFound = true;
2402       } else if (CurNM != OMPD_unknown) {
2403         NameModifierLoc.push_back(IC->getNameModifierLoc());
2404         ++NamedModifiersNumber;
2405       }
2406       FoundNameModifiers[CurNM] = IC;
2407       if (CurNM == OMPD_unknown)
2408         continue;
2409       // Check if the specified name modifier is allowed for the current
2410       // directive.
2411       // At most one if clause with the particular directive-name-modifier can
2412       // appear on the directive.
2413       bool MatchFound = false;
2414       for (auto NM : AllowedNameModifiers) {
2415         if (CurNM == NM) {
2416           MatchFound = true;
2417           break;
2418         }
2419       }
2420       if (!MatchFound) {
2421         S.Diag(IC->getNameModifierLoc(),
2422                diag::err_omp_wrong_if_directive_name_modifier)
2423             << getOpenMPDirectiveName(CurNM) << getOpenMPDirectiveName(Kind);
2424         ErrorFound = true;
2425       }
2426     }
2427   }
2428   // If any if clause on the directive includes a directive-name-modifier then
2429   // all if clauses on the directive must include a directive-name-modifier.
2430   if (FoundNameModifiers[OMPD_unknown] && NamedModifiersNumber > 0) {
2431     if (NamedModifiersNumber == AllowedNameModifiers.size()) {
2432       S.Diag(FoundNameModifiers[OMPD_unknown]->getLocStart(),
2433              diag::err_omp_no_more_if_clause);
2434     } else {
2435       std::string Values;
2436       std::string Sep(", ");
2437       unsigned AllowedCnt = 0;
2438       unsigned TotalAllowedNum =
2439           AllowedNameModifiers.size() - NamedModifiersNumber;
2440       for (unsigned Cnt = 0, End = AllowedNameModifiers.size(); Cnt < End;
2441            ++Cnt) {
2442         OpenMPDirectiveKind NM = AllowedNameModifiers[Cnt];
2443         if (!FoundNameModifiers[NM]) {
2444           Values += "'";
2445           Values += getOpenMPDirectiveName(NM);
2446           Values += "'";
2447           if (AllowedCnt + 2 == TotalAllowedNum)
2448             Values += " or ";
2449           else if (AllowedCnt + 1 != TotalAllowedNum)
2450             Values += Sep;
2451           ++AllowedCnt;
2452         }
2453       }
2454       S.Diag(FoundNameModifiers[OMPD_unknown]->getCondition()->getLocStart(),
2455              diag::err_omp_unnamed_if_clause)
2456           << (TotalAllowedNum > 1) << Values;
2457     }
2458     for (auto Loc : NameModifierLoc) {
2459       S.Diag(Loc, diag::note_omp_previous_named_if_clause);
2460     }
2461     ErrorFound = true;
2462   }
2463   return ErrorFound;
2464 }
2465 
ActOnOpenMPExecutableDirective(OpenMPDirectiveKind Kind,const DeclarationNameInfo & DirName,OpenMPDirectiveKind CancelRegion,ArrayRef<OMPClause * > Clauses,Stmt * AStmt,SourceLocation StartLoc,SourceLocation EndLoc)2466 StmtResult Sema::ActOnOpenMPExecutableDirective(
2467     OpenMPDirectiveKind Kind, const DeclarationNameInfo &DirName,
2468     OpenMPDirectiveKind CancelRegion, ArrayRef<OMPClause *> Clauses,
2469     Stmt *AStmt, SourceLocation StartLoc, SourceLocation EndLoc) {
2470   StmtResult Res = StmtError();
2471   if (CheckNestingOfRegions(*this, DSAStack, Kind, DirName, CancelRegion,
2472                             StartLoc))
2473     return StmtError();
2474 
2475   llvm::SmallVector<OMPClause *, 8> ClausesWithImplicit;
2476   llvm::DenseMap<VarDecl *, Expr *> VarsWithInheritedDSA;
2477   bool ErrorFound = false;
2478   ClausesWithImplicit.append(Clauses.begin(), Clauses.end());
2479   if (AStmt) {
2480     assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
2481 
2482     // Check default data sharing attributes for referenced variables.
2483     DSAAttrChecker DSAChecker(DSAStack, *this, cast<CapturedStmt>(AStmt));
2484     DSAChecker.Visit(cast<CapturedStmt>(AStmt)->getCapturedStmt());
2485     if (DSAChecker.isErrorFound())
2486       return StmtError();
2487     // Generate list of implicitly defined firstprivate variables.
2488     VarsWithInheritedDSA = DSAChecker.getVarsWithInheritedDSA();
2489 
2490     if (!DSAChecker.getImplicitFirstprivate().empty()) {
2491       if (OMPClause *Implicit = ActOnOpenMPFirstprivateClause(
2492               DSAChecker.getImplicitFirstprivate(), SourceLocation(),
2493               SourceLocation(), SourceLocation())) {
2494         ClausesWithImplicit.push_back(Implicit);
2495         ErrorFound = cast<OMPFirstprivateClause>(Implicit)->varlist_size() !=
2496                      DSAChecker.getImplicitFirstprivate().size();
2497       } else
2498         ErrorFound = true;
2499     }
2500   }
2501 
2502   llvm::SmallVector<OpenMPDirectiveKind, 4> AllowedNameModifiers;
2503   switch (Kind) {
2504   case OMPD_parallel:
2505     Res = ActOnOpenMPParallelDirective(ClausesWithImplicit, AStmt, StartLoc,
2506                                        EndLoc);
2507     AllowedNameModifiers.push_back(OMPD_parallel);
2508     break;
2509   case OMPD_simd:
2510     Res = ActOnOpenMPSimdDirective(ClausesWithImplicit, AStmt, StartLoc, EndLoc,
2511                                    VarsWithInheritedDSA);
2512     break;
2513   case OMPD_for:
2514     Res = ActOnOpenMPForDirective(ClausesWithImplicit, AStmt, StartLoc, EndLoc,
2515                                   VarsWithInheritedDSA);
2516     break;
2517   case OMPD_for_simd:
2518     Res = ActOnOpenMPForSimdDirective(ClausesWithImplicit, AStmt, StartLoc,
2519                                       EndLoc, VarsWithInheritedDSA);
2520     break;
2521   case OMPD_sections:
2522     Res = ActOnOpenMPSectionsDirective(ClausesWithImplicit, AStmt, StartLoc,
2523                                        EndLoc);
2524     break;
2525   case OMPD_section:
2526     assert(ClausesWithImplicit.empty() &&
2527            "No clauses are allowed for 'omp section' directive");
2528     Res = ActOnOpenMPSectionDirective(AStmt, StartLoc, EndLoc);
2529     break;
2530   case OMPD_single:
2531     Res = ActOnOpenMPSingleDirective(ClausesWithImplicit, AStmt, StartLoc,
2532                                      EndLoc);
2533     break;
2534   case OMPD_master:
2535     assert(ClausesWithImplicit.empty() &&
2536            "No clauses are allowed for 'omp master' directive");
2537     Res = ActOnOpenMPMasterDirective(AStmt, StartLoc, EndLoc);
2538     break;
2539   case OMPD_critical:
2540     Res = ActOnOpenMPCriticalDirective(DirName, ClausesWithImplicit, AStmt,
2541                                        StartLoc, EndLoc);
2542     break;
2543   case OMPD_parallel_for:
2544     Res = ActOnOpenMPParallelForDirective(ClausesWithImplicit, AStmt, StartLoc,
2545                                           EndLoc, VarsWithInheritedDSA);
2546     AllowedNameModifiers.push_back(OMPD_parallel);
2547     break;
2548   case OMPD_parallel_for_simd:
2549     Res = ActOnOpenMPParallelForSimdDirective(
2550         ClausesWithImplicit, AStmt, StartLoc, EndLoc, VarsWithInheritedDSA);
2551     AllowedNameModifiers.push_back(OMPD_parallel);
2552     break;
2553   case OMPD_parallel_sections:
2554     Res = ActOnOpenMPParallelSectionsDirective(ClausesWithImplicit, AStmt,
2555                                                StartLoc, EndLoc);
2556     AllowedNameModifiers.push_back(OMPD_parallel);
2557     break;
2558   case OMPD_task:
2559     Res =
2560         ActOnOpenMPTaskDirective(ClausesWithImplicit, AStmt, StartLoc, EndLoc);
2561     AllowedNameModifiers.push_back(OMPD_task);
2562     break;
2563   case OMPD_taskyield:
2564     assert(ClausesWithImplicit.empty() &&
2565            "No clauses are allowed for 'omp taskyield' directive");
2566     assert(AStmt == nullptr &&
2567            "No associated statement allowed for 'omp taskyield' directive");
2568     Res = ActOnOpenMPTaskyieldDirective(StartLoc, EndLoc);
2569     break;
2570   case OMPD_barrier:
2571     assert(ClausesWithImplicit.empty() &&
2572            "No clauses are allowed for 'omp barrier' directive");
2573     assert(AStmt == nullptr &&
2574            "No associated statement allowed for 'omp barrier' directive");
2575     Res = ActOnOpenMPBarrierDirective(StartLoc, EndLoc);
2576     break;
2577   case OMPD_taskwait:
2578     assert(ClausesWithImplicit.empty() &&
2579            "No clauses are allowed for 'omp taskwait' directive");
2580     assert(AStmt == nullptr &&
2581            "No associated statement allowed for 'omp taskwait' directive");
2582     Res = ActOnOpenMPTaskwaitDirective(StartLoc, EndLoc);
2583     break;
2584   case OMPD_taskgroup:
2585     assert(ClausesWithImplicit.empty() &&
2586            "No clauses are allowed for 'omp taskgroup' directive");
2587     Res = ActOnOpenMPTaskgroupDirective(AStmt, StartLoc, EndLoc);
2588     break;
2589   case OMPD_flush:
2590     assert(AStmt == nullptr &&
2591            "No associated statement allowed for 'omp flush' directive");
2592     Res = ActOnOpenMPFlushDirective(ClausesWithImplicit, StartLoc, EndLoc);
2593     break;
2594   case OMPD_ordered:
2595     Res = ActOnOpenMPOrderedDirective(ClausesWithImplicit, AStmt, StartLoc,
2596                                       EndLoc);
2597     break;
2598   case OMPD_atomic:
2599     Res = ActOnOpenMPAtomicDirective(ClausesWithImplicit, AStmt, StartLoc,
2600                                      EndLoc);
2601     break;
2602   case OMPD_teams:
2603     Res =
2604         ActOnOpenMPTeamsDirective(ClausesWithImplicit, AStmt, StartLoc, EndLoc);
2605     break;
2606   case OMPD_target:
2607     Res = ActOnOpenMPTargetDirective(ClausesWithImplicit, AStmt, StartLoc,
2608                                      EndLoc);
2609     AllowedNameModifiers.push_back(OMPD_target);
2610     break;
2611   case OMPD_cancellation_point:
2612     assert(ClausesWithImplicit.empty() &&
2613            "No clauses are allowed for 'omp cancellation point' directive");
2614     assert(AStmt == nullptr && "No associated statement allowed for 'omp "
2615                                "cancellation point' directive");
2616     Res = ActOnOpenMPCancellationPointDirective(StartLoc, EndLoc, CancelRegion);
2617     break;
2618   case OMPD_cancel:
2619     assert(AStmt == nullptr &&
2620            "No associated statement allowed for 'omp cancel' directive");
2621     Res = ActOnOpenMPCancelDirective(ClausesWithImplicit, StartLoc, EndLoc,
2622                                      CancelRegion);
2623     AllowedNameModifiers.push_back(OMPD_cancel);
2624     break;
2625   case OMPD_target_data:
2626     Res = ActOnOpenMPTargetDataDirective(ClausesWithImplicit, AStmt, StartLoc,
2627                                          EndLoc);
2628     AllowedNameModifiers.push_back(OMPD_target_data);
2629     break;
2630   case OMPD_taskloop:
2631     Res = ActOnOpenMPTaskLoopDirective(ClausesWithImplicit, AStmt, StartLoc,
2632                                        EndLoc, VarsWithInheritedDSA);
2633     AllowedNameModifiers.push_back(OMPD_taskloop);
2634     break;
2635   case OMPD_taskloop_simd:
2636     Res = ActOnOpenMPTaskLoopSimdDirective(ClausesWithImplicit, AStmt, StartLoc,
2637                                            EndLoc, VarsWithInheritedDSA);
2638     AllowedNameModifiers.push_back(OMPD_taskloop);
2639     break;
2640   case OMPD_distribute:
2641     Res = ActOnOpenMPDistributeDirective(ClausesWithImplicit, AStmt, StartLoc,
2642                                          EndLoc, VarsWithInheritedDSA);
2643     break;
2644   case OMPD_threadprivate:
2645     llvm_unreachable("OpenMP Directive is not allowed");
2646   case OMPD_unknown:
2647     llvm_unreachable("Unknown OpenMP directive");
2648   }
2649 
2650   for (auto P : VarsWithInheritedDSA) {
2651     Diag(P.second->getExprLoc(), diag::err_omp_no_dsa_for_variable)
2652         << P.first << P.second->getSourceRange();
2653   }
2654   ErrorFound = !VarsWithInheritedDSA.empty() || ErrorFound;
2655 
2656   if (!AllowedNameModifiers.empty())
2657     ErrorFound = checkIfClauses(*this, Kind, Clauses, AllowedNameModifiers) ||
2658                  ErrorFound;
2659 
2660   if (ErrorFound)
2661     return StmtError();
2662   return Res;
2663 }
2664 
ActOnOpenMPParallelDirective(ArrayRef<OMPClause * > Clauses,Stmt * AStmt,SourceLocation StartLoc,SourceLocation EndLoc)2665 StmtResult Sema::ActOnOpenMPParallelDirective(ArrayRef<OMPClause *> Clauses,
2666                                               Stmt *AStmt,
2667                                               SourceLocation StartLoc,
2668                                               SourceLocation EndLoc) {
2669   if (!AStmt)
2670     return StmtError();
2671 
2672   CapturedStmt *CS = cast<CapturedStmt>(AStmt);
2673   // 1.2.2 OpenMP Language Terminology
2674   // Structured block - An executable statement with a single entry at the
2675   // top and a single exit at the bottom.
2676   // The point of exit cannot be a branch out of the structured block.
2677   // longjmp() and throw() must not violate the entry/exit criteria.
2678   CS->getCapturedDecl()->setNothrow();
2679 
2680   getCurFunction()->setHasBranchProtectedScope();
2681 
2682   return OMPParallelDirective::Create(Context, StartLoc, EndLoc, Clauses, AStmt,
2683                                       DSAStack->isCancelRegion());
2684 }
2685 
2686 namespace {
2687 /// \brief Helper class for checking canonical form of the OpenMP loops and
2688 /// extracting iteration space of each loop in the loop nest, that will be used
2689 /// for IR generation.
2690 class OpenMPIterationSpaceChecker {
2691   /// \brief Reference to Sema.
2692   Sema &SemaRef;
2693   /// \brief A location for diagnostics (when there is no some better location).
2694   SourceLocation DefaultLoc;
2695   /// \brief A location for diagnostics (when increment is not compatible).
2696   SourceLocation ConditionLoc;
2697   /// \brief A source location for referring to loop init later.
2698   SourceRange InitSrcRange;
2699   /// \brief A source location for referring to condition later.
2700   SourceRange ConditionSrcRange;
2701   /// \brief A source location for referring to increment later.
2702   SourceRange IncrementSrcRange;
2703   /// \brief Loop variable.
2704   VarDecl *Var;
2705   /// \brief Reference to loop variable.
2706   DeclRefExpr *VarRef;
2707   /// \brief Lower bound (initializer for the var).
2708   Expr *LB;
2709   /// \brief Upper bound.
2710   Expr *UB;
2711   /// \brief Loop step (increment).
2712   Expr *Step;
2713   /// \brief This flag is true when condition is one of:
2714   ///   Var <  UB
2715   ///   Var <= UB
2716   ///   UB  >  Var
2717   ///   UB  >= Var
2718   bool TestIsLessOp;
2719   /// \brief This flag is true when condition is strict ( < or > ).
2720   bool TestIsStrictOp;
2721   /// \brief This flag is true when step is subtracted on each iteration.
2722   bool SubtractStep;
2723 
2724 public:
OpenMPIterationSpaceChecker(Sema & SemaRef,SourceLocation DefaultLoc)2725   OpenMPIterationSpaceChecker(Sema &SemaRef, SourceLocation DefaultLoc)
2726       : SemaRef(SemaRef), DefaultLoc(DefaultLoc), ConditionLoc(DefaultLoc),
2727         InitSrcRange(SourceRange()), ConditionSrcRange(SourceRange()),
2728         IncrementSrcRange(SourceRange()), Var(nullptr), VarRef(nullptr),
2729         LB(nullptr), UB(nullptr), Step(nullptr), TestIsLessOp(false),
2730         TestIsStrictOp(false), SubtractStep(false) {}
2731   /// \brief Check init-expr for canonical loop form and save loop counter
2732   /// variable - #Var and its initialization value - #LB.
2733   bool CheckInit(Stmt *S, bool EmitDiags = true);
2734   /// \brief Check test-expr for canonical form, save upper-bound (#UB), flags
2735   /// for less/greater and for strict/non-strict comparison.
2736   bool CheckCond(Expr *S);
2737   /// \brief Check incr-expr for canonical loop form and return true if it
2738   /// does not conform, otherwise save loop step (#Step).
2739   bool CheckInc(Expr *S);
2740   /// \brief Return the loop counter variable.
GetLoopVar() const2741   VarDecl *GetLoopVar() const { return Var; }
2742   /// \brief Return the reference expression to loop counter variable.
GetLoopVarRefExpr() const2743   DeclRefExpr *GetLoopVarRefExpr() const { return VarRef; }
2744   /// \brief Source range of the loop init.
GetInitSrcRange() const2745   SourceRange GetInitSrcRange() const { return InitSrcRange; }
2746   /// \brief Source range of the loop condition.
GetConditionSrcRange() const2747   SourceRange GetConditionSrcRange() const { return ConditionSrcRange; }
2748   /// \brief Source range of the loop increment.
GetIncrementSrcRange() const2749   SourceRange GetIncrementSrcRange() const { return IncrementSrcRange; }
2750   /// \brief True if the step should be subtracted.
ShouldSubtractStep() const2751   bool ShouldSubtractStep() const { return SubtractStep; }
2752   /// \brief Build the expression to calculate the number of iterations.
2753   Expr *BuildNumIterations(Scope *S, const bool LimitedType) const;
2754   /// \brief Build the precondition expression for the loops.
2755   Expr *BuildPreCond(Scope *S, Expr *Cond) const;
2756   /// \brief Build reference expression to the counter be used for codegen.
2757   Expr *BuildCounterVar() const;
2758   /// \brief Build reference expression to the private counter be used for
2759   /// codegen.
2760   Expr *BuildPrivateCounterVar() const;
2761   /// \brief Build initization of the counter be used for codegen.
2762   Expr *BuildCounterInit() const;
2763   /// \brief Build step of the counter be used for codegen.
2764   Expr *BuildCounterStep() const;
2765   /// \brief Return true if any expression is dependent.
2766   bool Dependent() const;
2767 
2768 private:
2769   /// \brief Check the right-hand side of an assignment in the increment
2770   /// expression.
2771   bool CheckIncRHS(Expr *RHS);
2772   /// \brief Helper to set loop counter variable and its initializer.
2773   bool SetVarAndLB(VarDecl *NewVar, DeclRefExpr *NewVarRefExpr, Expr *NewLB);
2774   /// \brief Helper to set upper bound.
2775   bool SetUB(Expr *NewUB, bool LessOp, bool StrictOp, SourceRange SR,
2776              SourceLocation SL);
2777   /// \brief Helper to set loop increment.
2778   bool SetStep(Expr *NewStep, bool Subtract);
2779 };
2780 
Dependent() const2781 bool OpenMPIterationSpaceChecker::Dependent() const {
2782   if (!Var) {
2783     assert(!LB && !UB && !Step);
2784     return false;
2785   }
2786   return Var->getType()->isDependentType() || (LB && LB->isValueDependent()) ||
2787          (UB && UB->isValueDependent()) || (Step && Step->isValueDependent());
2788 }
2789 
2790 template <typename T>
getExprAsWritten(T * E)2791 static T *getExprAsWritten(T *E) {
2792   if (auto *ExprTemp = dyn_cast<ExprWithCleanups>(E))
2793     E = ExprTemp->getSubExpr();
2794 
2795   if (auto *MTE = dyn_cast<MaterializeTemporaryExpr>(E))
2796     E = MTE->GetTemporaryExpr();
2797 
2798   while (auto *Binder = dyn_cast<CXXBindTemporaryExpr>(E))
2799     E = Binder->getSubExpr();
2800 
2801   if (auto *ICE = dyn_cast<ImplicitCastExpr>(E))
2802     E = ICE->getSubExprAsWritten();
2803   return E->IgnoreParens();
2804 }
2805 
SetVarAndLB(VarDecl * NewVar,DeclRefExpr * NewVarRefExpr,Expr * NewLB)2806 bool OpenMPIterationSpaceChecker::SetVarAndLB(VarDecl *NewVar,
2807                                               DeclRefExpr *NewVarRefExpr,
2808                                               Expr *NewLB) {
2809   // State consistency checking to ensure correct usage.
2810   assert(Var == nullptr && LB == nullptr && VarRef == nullptr &&
2811          UB == nullptr && Step == nullptr && !TestIsLessOp && !TestIsStrictOp);
2812   if (!NewVar || !NewLB)
2813     return true;
2814   Var = NewVar;
2815   VarRef = NewVarRefExpr;
2816   if (auto *CE = dyn_cast_or_null<CXXConstructExpr>(NewLB))
2817     if (const CXXConstructorDecl *Ctor = CE->getConstructor())
2818       if ((Ctor->isCopyOrMoveConstructor() ||
2819            Ctor->isConvertingConstructor(/*AllowExplicit=*/false)) &&
2820           CE->getNumArgs() > 0 && CE->getArg(0) != nullptr)
2821         NewLB = CE->getArg(0)->IgnoreParenImpCasts();
2822   LB = NewLB;
2823   return false;
2824 }
2825 
SetUB(Expr * NewUB,bool LessOp,bool StrictOp,SourceRange SR,SourceLocation SL)2826 bool OpenMPIterationSpaceChecker::SetUB(Expr *NewUB, bool LessOp, bool StrictOp,
2827                                         SourceRange SR, SourceLocation SL) {
2828   // State consistency checking to ensure correct usage.
2829   assert(Var != nullptr && LB != nullptr && UB == nullptr && Step == nullptr &&
2830          !TestIsLessOp && !TestIsStrictOp);
2831   if (!NewUB)
2832     return true;
2833   UB = NewUB;
2834   TestIsLessOp = LessOp;
2835   TestIsStrictOp = StrictOp;
2836   ConditionSrcRange = SR;
2837   ConditionLoc = SL;
2838   return false;
2839 }
2840 
SetStep(Expr * NewStep,bool Subtract)2841 bool OpenMPIterationSpaceChecker::SetStep(Expr *NewStep, bool Subtract) {
2842   // State consistency checking to ensure correct usage.
2843   assert(Var != nullptr && LB != nullptr && Step == nullptr);
2844   if (!NewStep)
2845     return true;
2846   if (!NewStep->isValueDependent()) {
2847     // Check that the step is integer expression.
2848     SourceLocation StepLoc = NewStep->getLocStart();
2849     ExprResult Val =
2850         SemaRef.PerformOpenMPImplicitIntegerConversion(StepLoc, NewStep);
2851     if (Val.isInvalid())
2852       return true;
2853     NewStep = Val.get();
2854 
2855     // OpenMP [2.6, Canonical Loop Form, Restrictions]
2856     //  If test-expr is of form var relational-op b and relational-op is < or
2857     //  <= then incr-expr must cause var to increase on each iteration of the
2858     //  loop. If test-expr is of form var relational-op b and relational-op is
2859     //  > or >= then incr-expr must cause var to decrease on each iteration of
2860     //  the loop.
2861     //  If test-expr is of form b relational-op var and relational-op is < or
2862     //  <= then incr-expr must cause var to decrease on each iteration of the
2863     //  loop. If test-expr is of form b relational-op var and relational-op is
2864     //  > or >= then incr-expr must cause var to increase on each iteration of
2865     //  the loop.
2866     llvm::APSInt Result;
2867     bool IsConstant = NewStep->isIntegerConstantExpr(Result, SemaRef.Context);
2868     bool IsUnsigned = !NewStep->getType()->hasSignedIntegerRepresentation();
2869     bool IsConstNeg =
2870         IsConstant && Result.isSigned() && (Subtract != Result.isNegative());
2871     bool IsConstPos =
2872         IsConstant && Result.isSigned() && (Subtract == Result.isNegative());
2873     bool IsConstZero = IsConstant && !Result.getBoolValue();
2874     if (UB && (IsConstZero ||
2875                (TestIsLessOp ? (IsConstNeg || (IsUnsigned && Subtract))
2876                              : (IsConstPos || (IsUnsigned && !Subtract))))) {
2877       SemaRef.Diag(NewStep->getExprLoc(),
2878                    diag::err_omp_loop_incr_not_compatible)
2879           << Var << TestIsLessOp << NewStep->getSourceRange();
2880       SemaRef.Diag(ConditionLoc,
2881                    diag::note_omp_loop_cond_requres_compatible_incr)
2882           << TestIsLessOp << ConditionSrcRange;
2883       return true;
2884     }
2885     if (TestIsLessOp == Subtract) {
2886       NewStep = SemaRef.CreateBuiltinUnaryOp(NewStep->getExprLoc(), UO_Minus,
2887                                              NewStep).get();
2888       Subtract = !Subtract;
2889     }
2890   }
2891 
2892   Step = NewStep;
2893   SubtractStep = Subtract;
2894   return false;
2895 }
2896 
CheckInit(Stmt * S,bool EmitDiags)2897 bool OpenMPIterationSpaceChecker::CheckInit(Stmt *S, bool EmitDiags) {
2898   // Check init-expr for canonical loop form and save loop counter
2899   // variable - #Var and its initialization value - #LB.
2900   // OpenMP [2.6] Canonical loop form. init-expr may be one of the following:
2901   //   var = lb
2902   //   integer-type var = lb
2903   //   random-access-iterator-type var = lb
2904   //   pointer-type var = lb
2905   //
2906   if (!S) {
2907     if (EmitDiags) {
2908       SemaRef.Diag(DefaultLoc, diag::err_omp_loop_not_canonical_init);
2909     }
2910     return true;
2911   }
2912   InitSrcRange = S->getSourceRange();
2913   if (Expr *E = dyn_cast<Expr>(S))
2914     S = E->IgnoreParens();
2915   if (auto BO = dyn_cast<BinaryOperator>(S)) {
2916     if (BO->getOpcode() == BO_Assign)
2917       if (auto DRE = dyn_cast<DeclRefExpr>(BO->getLHS()->IgnoreParens()))
2918         return SetVarAndLB(dyn_cast<VarDecl>(DRE->getDecl()), DRE,
2919                            BO->getRHS());
2920   } else if (auto DS = dyn_cast<DeclStmt>(S)) {
2921     if (DS->isSingleDecl()) {
2922       if (auto Var = dyn_cast_or_null<VarDecl>(DS->getSingleDecl())) {
2923         if (Var->hasInit() && !Var->getType()->isReferenceType()) {
2924           // Accept non-canonical init form here but emit ext. warning.
2925           if (Var->getInitStyle() != VarDecl::CInit && EmitDiags)
2926             SemaRef.Diag(S->getLocStart(),
2927                          diag::ext_omp_loop_not_canonical_init)
2928                 << S->getSourceRange();
2929           return SetVarAndLB(Var, nullptr, Var->getInit());
2930         }
2931       }
2932     }
2933   } else if (auto CE = dyn_cast<CXXOperatorCallExpr>(S))
2934     if (CE->getOperator() == OO_Equal)
2935       if (auto DRE = dyn_cast<DeclRefExpr>(CE->getArg(0)))
2936         return SetVarAndLB(dyn_cast<VarDecl>(DRE->getDecl()), DRE,
2937                            CE->getArg(1));
2938 
2939   if (EmitDiags) {
2940     SemaRef.Diag(S->getLocStart(), diag::err_omp_loop_not_canonical_init)
2941         << S->getSourceRange();
2942   }
2943   return true;
2944 }
2945 
2946 /// \brief Ignore parenthesizes, implicit casts, copy constructor and return the
2947 /// variable (which may be the loop variable) if possible.
GetInitVarDecl(const Expr * E)2948 static const VarDecl *GetInitVarDecl(const Expr *E) {
2949   if (!E)
2950     return nullptr;
2951   E = getExprAsWritten(E);
2952   if (auto *CE = dyn_cast_or_null<CXXConstructExpr>(E))
2953     if (const CXXConstructorDecl *Ctor = CE->getConstructor())
2954       if ((Ctor->isCopyOrMoveConstructor() ||
2955            Ctor->isConvertingConstructor(/*AllowExplicit=*/false)) &&
2956           CE->getNumArgs() > 0 && CE->getArg(0) != nullptr)
2957         E = CE->getArg(0)->IgnoreParenImpCasts();
2958   auto DRE = dyn_cast_or_null<DeclRefExpr>(E);
2959   if (!DRE)
2960     return nullptr;
2961   return dyn_cast<VarDecl>(DRE->getDecl());
2962 }
2963 
CheckCond(Expr * S)2964 bool OpenMPIterationSpaceChecker::CheckCond(Expr *S) {
2965   // Check test-expr for canonical form, save upper-bound UB, flags for
2966   // less/greater and for strict/non-strict comparison.
2967   // OpenMP [2.6] Canonical loop form. Test-expr may be one of the following:
2968   //   var relational-op b
2969   //   b relational-op var
2970   //
2971   if (!S) {
2972     SemaRef.Diag(DefaultLoc, diag::err_omp_loop_not_canonical_cond) << Var;
2973     return true;
2974   }
2975   S = getExprAsWritten(S);
2976   SourceLocation CondLoc = S->getLocStart();
2977   if (auto BO = dyn_cast<BinaryOperator>(S)) {
2978     if (BO->isRelationalOp()) {
2979       if (GetInitVarDecl(BO->getLHS()) == Var)
2980         return SetUB(BO->getRHS(),
2981                      (BO->getOpcode() == BO_LT || BO->getOpcode() == BO_LE),
2982                      (BO->getOpcode() == BO_LT || BO->getOpcode() == BO_GT),
2983                      BO->getSourceRange(), BO->getOperatorLoc());
2984       if (GetInitVarDecl(BO->getRHS()) == Var)
2985         return SetUB(BO->getLHS(),
2986                      (BO->getOpcode() == BO_GT || BO->getOpcode() == BO_GE),
2987                      (BO->getOpcode() == BO_LT || BO->getOpcode() == BO_GT),
2988                      BO->getSourceRange(), BO->getOperatorLoc());
2989     }
2990   } else if (auto CE = dyn_cast<CXXOperatorCallExpr>(S)) {
2991     if (CE->getNumArgs() == 2) {
2992       auto Op = CE->getOperator();
2993       switch (Op) {
2994       case OO_Greater:
2995       case OO_GreaterEqual:
2996       case OO_Less:
2997       case OO_LessEqual:
2998         if (GetInitVarDecl(CE->getArg(0)) == Var)
2999           return SetUB(CE->getArg(1), Op == OO_Less || Op == OO_LessEqual,
3000                        Op == OO_Less || Op == OO_Greater, CE->getSourceRange(),
3001                        CE->getOperatorLoc());
3002         if (GetInitVarDecl(CE->getArg(1)) == Var)
3003           return SetUB(CE->getArg(0), Op == OO_Greater || Op == OO_GreaterEqual,
3004                        Op == OO_Less || Op == OO_Greater, CE->getSourceRange(),
3005                        CE->getOperatorLoc());
3006         break;
3007       default:
3008         break;
3009       }
3010     }
3011   }
3012   SemaRef.Diag(CondLoc, diag::err_omp_loop_not_canonical_cond)
3013       << S->getSourceRange() << Var;
3014   return true;
3015 }
3016 
CheckIncRHS(Expr * RHS)3017 bool OpenMPIterationSpaceChecker::CheckIncRHS(Expr *RHS) {
3018   // RHS of canonical loop form increment can be:
3019   //   var + incr
3020   //   incr + var
3021   //   var - incr
3022   //
3023   RHS = RHS->IgnoreParenImpCasts();
3024   if (auto BO = dyn_cast<BinaryOperator>(RHS)) {
3025     if (BO->isAdditiveOp()) {
3026       bool IsAdd = BO->getOpcode() == BO_Add;
3027       if (GetInitVarDecl(BO->getLHS()) == Var)
3028         return SetStep(BO->getRHS(), !IsAdd);
3029       if (IsAdd && GetInitVarDecl(BO->getRHS()) == Var)
3030         return SetStep(BO->getLHS(), false);
3031     }
3032   } else if (auto CE = dyn_cast<CXXOperatorCallExpr>(RHS)) {
3033     bool IsAdd = CE->getOperator() == OO_Plus;
3034     if ((IsAdd || CE->getOperator() == OO_Minus) && CE->getNumArgs() == 2) {
3035       if (GetInitVarDecl(CE->getArg(0)) == Var)
3036         return SetStep(CE->getArg(1), !IsAdd);
3037       if (IsAdd && GetInitVarDecl(CE->getArg(1)) == Var)
3038         return SetStep(CE->getArg(0), false);
3039     }
3040   }
3041   SemaRef.Diag(RHS->getLocStart(), diag::err_omp_loop_not_canonical_incr)
3042       << RHS->getSourceRange() << Var;
3043   return true;
3044 }
3045 
CheckInc(Expr * S)3046 bool OpenMPIterationSpaceChecker::CheckInc(Expr *S) {
3047   // Check incr-expr for canonical loop form and return true if it
3048   // does not conform.
3049   // OpenMP [2.6] Canonical loop form. Test-expr may be one of the following:
3050   //   ++var
3051   //   var++
3052   //   --var
3053   //   var--
3054   //   var += incr
3055   //   var -= incr
3056   //   var = var + incr
3057   //   var = incr + var
3058   //   var = var - incr
3059   //
3060   if (!S) {
3061     SemaRef.Diag(DefaultLoc, diag::err_omp_loop_not_canonical_incr) << Var;
3062     return true;
3063   }
3064   IncrementSrcRange = S->getSourceRange();
3065   S = S->IgnoreParens();
3066   if (auto UO = dyn_cast<UnaryOperator>(S)) {
3067     if (UO->isIncrementDecrementOp() && GetInitVarDecl(UO->getSubExpr()) == Var)
3068       return SetStep(
3069           SemaRef.ActOnIntegerConstant(UO->getLocStart(),
3070                                        (UO->isDecrementOp() ? -1 : 1)).get(),
3071           false);
3072   } else if (auto BO = dyn_cast<BinaryOperator>(S)) {
3073     switch (BO->getOpcode()) {
3074     case BO_AddAssign:
3075     case BO_SubAssign:
3076       if (GetInitVarDecl(BO->getLHS()) == Var)
3077         return SetStep(BO->getRHS(), BO->getOpcode() == BO_SubAssign);
3078       break;
3079     case BO_Assign:
3080       if (GetInitVarDecl(BO->getLHS()) == Var)
3081         return CheckIncRHS(BO->getRHS());
3082       break;
3083     default:
3084       break;
3085     }
3086   } else if (auto CE = dyn_cast<CXXOperatorCallExpr>(S)) {
3087     switch (CE->getOperator()) {
3088     case OO_PlusPlus:
3089     case OO_MinusMinus:
3090       if (GetInitVarDecl(CE->getArg(0)) == Var)
3091         return SetStep(
3092             SemaRef.ActOnIntegerConstant(
3093                         CE->getLocStart(),
3094                         ((CE->getOperator() == OO_MinusMinus) ? -1 : 1)).get(),
3095             false);
3096       break;
3097     case OO_PlusEqual:
3098     case OO_MinusEqual:
3099       if (GetInitVarDecl(CE->getArg(0)) == Var)
3100         return SetStep(CE->getArg(1), CE->getOperator() == OO_MinusEqual);
3101       break;
3102     case OO_Equal:
3103       if (GetInitVarDecl(CE->getArg(0)) == Var)
3104         return CheckIncRHS(CE->getArg(1));
3105       break;
3106     default:
3107       break;
3108     }
3109   }
3110   SemaRef.Diag(S->getLocStart(), diag::err_omp_loop_not_canonical_incr)
3111       << S->getSourceRange() << Var;
3112   return true;
3113 }
3114 
3115 namespace {
3116 // Transform variables declared in GNU statement expressions to new ones to
3117 // avoid crash on codegen.
3118 class TransformToNewDefs : public TreeTransform<TransformToNewDefs> {
3119   typedef TreeTransform<TransformToNewDefs> BaseTransform;
3120 
3121 public:
TransformToNewDefs(Sema & SemaRef)3122   TransformToNewDefs(Sema &SemaRef) : BaseTransform(SemaRef) {}
3123 
TransformDefinition(SourceLocation Loc,Decl * D)3124   Decl *TransformDefinition(SourceLocation Loc, Decl *D) {
3125     if (auto *VD = cast<VarDecl>(D))
3126       if (!isa<ParmVarDecl>(D) && !isa<VarTemplateSpecializationDecl>(D) &&
3127           !isa<ImplicitParamDecl>(D)) {
3128         auto *NewVD = VarDecl::Create(
3129             SemaRef.Context, VD->getDeclContext(), VD->getLocStart(),
3130             VD->getLocation(), VD->getIdentifier(), VD->getType(),
3131             VD->getTypeSourceInfo(), VD->getStorageClass());
3132         NewVD->setTSCSpec(VD->getTSCSpec());
3133         NewVD->setInit(VD->getInit());
3134         NewVD->setInitStyle(VD->getInitStyle());
3135         NewVD->setExceptionVariable(VD->isExceptionVariable());
3136         NewVD->setNRVOVariable(VD->isNRVOVariable());
3137         NewVD->setCXXForRangeDecl(VD->isInExternCXXContext());
3138         NewVD->setConstexpr(VD->isConstexpr());
3139         NewVD->setInitCapture(VD->isInitCapture());
3140         NewVD->setPreviousDeclInSameBlockScope(
3141             VD->isPreviousDeclInSameBlockScope());
3142         VD->getDeclContext()->addHiddenDecl(NewVD);
3143         if (VD->hasAttrs())
3144           NewVD->setAttrs(VD->getAttrs());
3145         transformedLocalDecl(VD, NewVD);
3146         return NewVD;
3147       }
3148     return BaseTransform::TransformDefinition(Loc, D);
3149   }
3150 
TransformDeclRefExpr(DeclRefExpr * E)3151   ExprResult TransformDeclRefExpr(DeclRefExpr *E) {
3152     if (auto *NewD = TransformDecl(E->getExprLoc(), E->getDecl()))
3153       if (E->getDecl() != NewD) {
3154         NewD->setReferenced();
3155         NewD->markUsed(SemaRef.Context);
3156         return DeclRefExpr::Create(
3157             SemaRef.Context, E->getQualifierLoc(), E->getTemplateKeywordLoc(),
3158             cast<ValueDecl>(NewD), E->refersToEnclosingVariableOrCapture(),
3159             E->getNameInfo(), E->getType(), E->getValueKind());
3160       }
3161     return BaseTransform::TransformDeclRefExpr(E);
3162   }
3163 };
3164 }
3165 
3166 /// \brief Build the expression to calculate the number of iterations.
3167 Expr *
BuildNumIterations(Scope * S,const bool LimitedType) const3168 OpenMPIterationSpaceChecker::BuildNumIterations(Scope *S,
3169                                                 const bool LimitedType) const {
3170   TransformToNewDefs Transform(SemaRef);
3171   ExprResult Diff;
3172   auto VarType = Var->getType().getNonReferenceType();
3173   if (VarType->isIntegerType() || VarType->isPointerType() ||
3174       SemaRef.getLangOpts().CPlusPlus) {
3175     // Upper - Lower
3176     auto *UBExpr = TestIsLessOp ? UB : LB;
3177     auto *LBExpr = TestIsLessOp ? LB : UB;
3178     Expr *Upper = Transform.TransformExpr(UBExpr).get();
3179     Expr *Lower = Transform.TransformExpr(LBExpr).get();
3180     if (!Upper || !Lower)
3181       return nullptr;
3182     Upper = SemaRef.PerformImplicitConversion(Upper, UBExpr->getType(),
3183                                                     Sema::AA_Converting,
3184                                                     /*AllowExplicit=*/true)
3185                       .get();
3186     Lower = SemaRef.PerformImplicitConversion(Lower, LBExpr->getType(),
3187                                               Sema::AA_Converting,
3188                                               /*AllowExplicit=*/true)
3189                 .get();
3190     if (!Upper || !Lower)
3191       return nullptr;
3192 
3193     Diff = SemaRef.BuildBinOp(S, DefaultLoc, BO_Sub, Upper, Lower);
3194 
3195     if (!Diff.isUsable() && VarType->getAsCXXRecordDecl()) {
3196       // BuildBinOp already emitted error, this one is to point user to upper
3197       // and lower bound, and to tell what is passed to 'operator-'.
3198       SemaRef.Diag(Upper->getLocStart(), diag::err_omp_loop_diff_cxx)
3199           << Upper->getSourceRange() << Lower->getSourceRange();
3200       return nullptr;
3201     }
3202   }
3203 
3204   if (!Diff.isUsable())
3205     return nullptr;
3206 
3207   // Upper - Lower [- 1]
3208   if (TestIsStrictOp)
3209     Diff = SemaRef.BuildBinOp(
3210         S, DefaultLoc, BO_Sub, Diff.get(),
3211         SemaRef.ActOnIntegerConstant(SourceLocation(), 1).get());
3212   if (!Diff.isUsable())
3213     return nullptr;
3214 
3215   // Upper - Lower [- 1] + Step
3216   auto NewStep = Transform.TransformExpr(Step->IgnoreImplicit());
3217   if (NewStep.isInvalid())
3218     return nullptr;
3219   NewStep = SemaRef.PerformImplicitConversion(
3220       NewStep.get(), Step->IgnoreImplicit()->getType(), Sema::AA_Converting,
3221       /*AllowExplicit=*/true);
3222   if (NewStep.isInvalid())
3223     return nullptr;
3224   Diff = SemaRef.BuildBinOp(S, DefaultLoc, BO_Add, Diff.get(), NewStep.get());
3225   if (!Diff.isUsable())
3226     return nullptr;
3227 
3228   // Parentheses (for dumping/debugging purposes only).
3229   Diff = SemaRef.ActOnParenExpr(DefaultLoc, DefaultLoc, Diff.get());
3230   if (!Diff.isUsable())
3231     return nullptr;
3232 
3233   // (Upper - Lower [- 1] + Step) / Step
3234   NewStep = Transform.TransformExpr(Step->IgnoreImplicit());
3235   if (NewStep.isInvalid())
3236     return nullptr;
3237   NewStep = SemaRef.PerformImplicitConversion(
3238       NewStep.get(), Step->IgnoreImplicit()->getType(), Sema::AA_Converting,
3239       /*AllowExplicit=*/true);
3240   if (NewStep.isInvalid())
3241     return nullptr;
3242   Diff = SemaRef.BuildBinOp(S, DefaultLoc, BO_Div, Diff.get(), NewStep.get());
3243   if (!Diff.isUsable())
3244     return nullptr;
3245 
3246   // OpenMP runtime requires 32-bit or 64-bit loop variables.
3247   QualType Type = Diff.get()->getType();
3248   auto &C = SemaRef.Context;
3249   bool UseVarType = VarType->hasIntegerRepresentation() &&
3250                     C.getTypeSize(Type) > C.getTypeSize(VarType);
3251   if (!Type->isIntegerType() || UseVarType) {
3252     unsigned NewSize =
3253         UseVarType ? C.getTypeSize(VarType) : C.getTypeSize(Type);
3254     bool IsSigned = UseVarType ? VarType->hasSignedIntegerRepresentation()
3255                                : Type->hasSignedIntegerRepresentation();
3256     Type = C.getIntTypeForBitwidth(NewSize, IsSigned);
3257     Diff = SemaRef.PerformImplicitConversion(
3258         Diff.get(), Type, Sema::AA_Converting, /*AllowExplicit=*/true);
3259     if (!Diff.isUsable())
3260       return nullptr;
3261   }
3262   if (LimitedType) {
3263     unsigned NewSize = (C.getTypeSize(Type) > 32) ? 64 : 32;
3264     if (NewSize != C.getTypeSize(Type)) {
3265       if (NewSize < C.getTypeSize(Type)) {
3266         assert(NewSize == 64 && "incorrect loop var size");
3267         SemaRef.Diag(DefaultLoc, diag::warn_omp_loop_64_bit_var)
3268             << InitSrcRange << ConditionSrcRange;
3269       }
3270       QualType NewType = C.getIntTypeForBitwidth(
3271           NewSize, Type->hasSignedIntegerRepresentation() ||
3272                        C.getTypeSize(Type) < NewSize);
3273       Diff = SemaRef.PerformImplicitConversion(Diff.get(), NewType,
3274                                                Sema::AA_Converting, true);
3275       if (!Diff.isUsable())
3276         return nullptr;
3277     }
3278   }
3279 
3280   return Diff.get();
3281 }
3282 
BuildPreCond(Scope * S,Expr * Cond) const3283 Expr *OpenMPIterationSpaceChecker::BuildPreCond(Scope *S, Expr *Cond) const {
3284   // Try to build LB <op> UB, where <op> is <, >, <=, or >=.
3285   bool Suppress = SemaRef.getDiagnostics().getSuppressAllDiagnostics();
3286   SemaRef.getDiagnostics().setSuppressAllDiagnostics(/*Val=*/true);
3287   TransformToNewDefs Transform(SemaRef);
3288 
3289   auto NewLB = Transform.TransformExpr(LB);
3290   auto NewUB = Transform.TransformExpr(UB);
3291   if (NewLB.isInvalid() || NewUB.isInvalid())
3292     return Cond;
3293   NewLB = SemaRef.PerformImplicitConversion(NewLB.get(), LB->getType(),
3294                                             Sema::AA_Converting,
3295                                             /*AllowExplicit=*/true);
3296   NewUB = SemaRef.PerformImplicitConversion(NewUB.get(), UB->getType(),
3297                                             Sema::AA_Converting,
3298                                             /*AllowExplicit=*/true);
3299   if (NewLB.isInvalid() || NewUB.isInvalid())
3300     return Cond;
3301   auto CondExpr = SemaRef.BuildBinOp(
3302       S, DefaultLoc, TestIsLessOp ? (TestIsStrictOp ? BO_LT : BO_LE)
3303                                   : (TestIsStrictOp ? BO_GT : BO_GE),
3304       NewLB.get(), NewUB.get());
3305   if (CondExpr.isUsable()) {
3306     CondExpr = SemaRef.PerformImplicitConversion(
3307         CondExpr.get(), SemaRef.Context.BoolTy, /*Action=*/Sema::AA_Casting,
3308         /*AllowExplicit=*/true);
3309   }
3310   SemaRef.getDiagnostics().setSuppressAllDiagnostics(Suppress);
3311   // Otherwise use original loop conditon and evaluate it in runtime.
3312   return CondExpr.isUsable() ? CondExpr.get() : Cond;
3313 }
3314 
3315 /// \brief Build reference expression to the counter be used for codegen.
BuildCounterVar() const3316 Expr *OpenMPIterationSpaceChecker::BuildCounterVar() const {
3317   return buildDeclRefExpr(SemaRef, Var, Var->getType().getNonReferenceType(),
3318                           DefaultLoc);
3319 }
3320 
BuildPrivateCounterVar() const3321 Expr *OpenMPIterationSpaceChecker::BuildPrivateCounterVar() const {
3322   if (Var && !Var->isInvalidDecl()) {
3323     auto Type = Var->getType().getNonReferenceType();
3324     auto *PrivateVar =
3325         buildVarDecl(SemaRef, DefaultLoc, Type, Var->getName(),
3326                      Var->hasAttrs() ? &Var->getAttrs() : nullptr);
3327     if (PrivateVar->isInvalidDecl())
3328       return nullptr;
3329     return buildDeclRefExpr(SemaRef, PrivateVar, Type, DefaultLoc);
3330   }
3331   return nullptr;
3332 }
3333 
3334 /// \brief Build initization of the counter be used for codegen.
BuildCounterInit() const3335 Expr *OpenMPIterationSpaceChecker::BuildCounterInit() const { return LB; }
3336 
3337 /// \brief Build step of the counter be used for codegen.
BuildCounterStep() const3338 Expr *OpenMPIterationSpaceChecker::BuildCounterStep() const { return Step; }
3339 
3340 /// \brief Iteration space of a single for loop.
3341 struct LoopIterationSpace {
3342   /// \brief Condition of the loop.
3343   Expr *PreCond;
3344   /// \brief This expression calculates the number of iterations in the loop.
3345   /// It is always possible to calculate it before starting the loop.
3346   Expr *NumIterations;
3347   /// \brief The loop counter variable.
3348   Expr *CounterVar;
3349   /// \brief Private loop counter variable.
3350   Expr *PrivateCounterVar;
3351   /// \brief This is initializer for the initial value of #CounterVar.
3352   Expr *CounterInit;
3353   /// \brief This is step for the #CounterVar used to generate its update:
3354   /// #CounterVar = #CounterInit + #CounterStep * CurrentIteration.
3355   Expr *CounterStep;
3356   /// \brief Should step be subtracted?
3357   bool Subtract;
3358   /// \brief Source range of the loop init.
3359   SourceRange InitSrcRange;
3360   /// \brief Source range of the loop condition.
3361   SourceRange CondSrcRange;
3362   /// \brief Source range of the loop increment.
3363   SourceRange IncSrcRange;
3364 };
3365 
3366 } // namespace
3367 
ActOnOpenMPLoopInitialization(SourceLocation ForLoc,Stmt * Init)3368 void Sema::ActOnOpenMPLoopInitialization(SourceLocation ForLoc, Stmt *Init) {
3369   assert(getLangOpts().OpenMP && "OpenMP is not active.");
3370   assert(Init && "Expected loop in canonical form.");
3371   unsigned CollapseIteration = DSAStack->getCollapseNumber();
3372   if (CollapseIteration > 0 &&
3373       isOpenMPLoopDirective(DSAStack->getCurrentDirective())) {
3374     OpenMPIterationSpaceChecker ISC(*this, ForLoc);
3375     if (!ISC.CheckInit(Init, /*EmitDiags=*/false)) {
3376       DSAStack->addLoopControlVariable(ISC.GetLoopVar());
3377     }
3378     DSAStack->setCollapseNumber(CollapseIteration - 1);
3379   }
3380 }
3381 
3382 /// \brief Called on a for stmt to check and extract its iteration space
3383 /// for further processing (such as collapsing).
CheckOpenMPIterationSpace(OpenMPDirectiveKind DKind,Stmt * S,Sema & SemaRef,DSAStackTy & DSA,unsigned CurrentNestedLoopCount,unsigned NestedLoopCount,Expr * CollapseLoopCountExpr,Expr * OrderedLoopCountExpr,llvm::DenseMap<VarDecl *,Expr * > & VarsWithImplicitDSA,LoopIterationSpace & ResultIterSpace)3384 static bool CheckOpenMPIterationSpace(
3385     OpenMPDirectiveKind DKind, Stmt *S, Sema &SemaRef, DSAStackTy &DSA,
3386     unsigned CurrentNestedLoopCount, unsigned NestedLoopCount,
3387     Expr *CollapseLoopCountExpr, Expr *OrderedLoopCountExpr,
3388     llvm::DenseMap<VarDecl *, Expr *> &VarsWithImplicitDSA,
3389     LoopIterationSpace &ResultIterSpace) {
3390   // OpenMP [2.6, Canonical Loop Form]
3391   //   for (init-expr; test-expr; incr-expr) structured-block
3392   auto For = dyn_cast_or_null<ForStmt>(S);
3393   if (!For) {
3394     SemaRef.Diag(S->getLocStart(), diag::err_omp_not_for)
3395         << (CollapseLoopCountExpr != nullptr || OrderedLoopCountExpr != nullptr)
3396         << getOpenMPDirectiveName(DKind) << NestedLoopCount
3397         << (CurrentNestedLoopCount > 0) << CurrentNestedLoopCount;
3398     if (NestedLoopCount > 1) {
3399       if (CollapseLoopCountExpr && OrderedLoopCountExpr)
3400         SemaRef.Diag(DSA.getConstructLoc(),
3401                      diag::note_omp_collapse_ordered_expr)
3402             << 2 << CollapseLoopCountExpr->getSourceRange()
3403             << OrderedLoopCountExpr->getSourceRange();
3404       else if (CollapseLoopCountExpr)
3405         SemaRef.Diag(CollapseLoopCountExpr->getExprLoc(),
3406                      diag::note_omp_collapse_ordered_expr)
3407             << 0 << CollapseLoopCountExpr->getSourceRange();
3408       else
3409         SemaRef.Diag(OrderedLoopCountExpr->getExprLoc(),
3410                      diag::note_omp_collapse_ordered_expr)
3411             << 1 << OrderedLoopCountExpr->getSourceRange();
3412     }
3413     return true;
3414   }
3415   assert(For->getBody());
3416 
3417   OpenMPIterationSpaceChecker ISC(SemaRef, For->getForLoc());
3418 
3419   // Check init.
3420   auto Init = For->getInit();
3421   if (ISC.CheckInit(Init)) {
3422     return true;
3423   }
3424 
3425   bool HasErrors = false;
3426 
3427   // Check loop variable's type.
3428   auto Var = ISC.GetLoopVar();
3429 
3430   // OpenMP [2.6, Canonical Loop Form]
3431   // Var is one of the following:
3432   //   A variable of signed or unsigned integer type.
3433   //   For C++, a variable of a random access iterator type.
3434   //   For C, a variable of a pointer type.
3435   auto VarType = Var->getType().getNonReferenceType();
3436   if (!VarType->isDependentType() && !VarType->isIntegerType() &&
3437       !VarType->isPointerType() &&
3438       !(SemaRef.getLangOpts().CPlusPlus && VarType->isOverloadableType())) {
3439     SemaRef.Diag(Init->getLocStart(), diag::err_omp_loop_variable_type)
3440         << SemaRef.getLangOpts().CPlusPlus;
3441     HasErrors = true;
3442   }
3443 
3444   // OpenMP, 2.14.1.1 Data-sharing Attribute Rules for Variables Referenced in a
3445   // Construct
3446   // The loop iteration variable(s) in the associated for-loop(s) of a for or
3447   // parallel for construct is (are) private.
3448   // The loop iteration variable in the associated for-loop of a simd construct
3449   // with just one associated for-loop is linear with a constant-linear-step
3450   // that is the increment of the associated for-loop.
3451   // Exclude loop var from the list of variables with implicitly defined data
3452   // sharing attributes.
3453   VarsWithImplicitDSA.erase(Var);
3454 
3455   // OpenMP [2.14.1.1, Data-sharing Attribute Rules for Variables Referenced in
3456   // a Construct, C/C++].
3457   // The loop iteration variable in the associated for-loop of a simd construct
3458   // with just one associated for-loop may be listed in a linear clause with a
3459   // constant-linear-step that is the increment of the associated for-loop.
3460   // The loop iteration variable(s) in the associated for-loop(s) of a for or
3461   // parallel for construct may be listed in a private or lastprivate clause.
3462   DSAStackTy::DSAVarData DVar = DSA.getTopDSA(Var, false);
3463   auto LoopVarRefExpr = ISC.GetLoopVarRefExpr();
3464   // If LoopVarRefExpr is nullptr it means the corresponding loop variable is
3465   // declared in the loop and it is predetermined as a private.
3466   auto PredeterminedCKind =
3467       isOpenMPSimdDirective(DKind)
3468           ? ((NestedLoopCount == 1) ? OMPC_linear : OMPC_lastprivate)
3469           : OMPC_private;
3470   if (((isOpenMPSimdDirective(DKind) && DVar.CKind != OMPC_unknown &&
3471         DVar.CKind != OMPC_threadprivate && DVar.CKind != PredeterminedCKind) ||
3472        ((isOpenMPWorksharingDirective(DKind) || DKind == OMPD_taskloop ||
3473         isOpenMPDistributeDirective(DKind)) &&
3474         !isOpenMPSimdDirective(DKind) && DVar.CKind != OMPC_unknown &&
3475         DVar.CKind != OMPC_private && DVar.CKind != OMPC_lastprivate &&
3476         DVar.CKind != OMPC_threadprivate)) &&
3477       ((DVar.CKind != OMPC_private && DVar.CKind != OMPC_threadprivate) ||
3478        DVar.RefExpr != nullptr)) {
3479     SemaRef.Diag(Init->getLocStart(), diag::err_omp_loop_var_dsa)
3480         << getOpenMPClauseName(DVar.CKind) << getOpenMPDirectiveName(DKind)
3481         << getOpenMPClauseName(PredeterminedCKind);
3482     if (DVar.RefExpr == nullptr)
3483       DVar.CKind = PredeterminedCKind;
3484     ReportOriginalDSA(SemaRef, &DSA, Var, DVar, /*IsLoopIterVar=*/true);
3485     HasErrors = true;
3486   } else if (LoopVarRefExpr != nullptr) {
3487     // Make the loop iteration variable private (for worksharing constructs),
3488     // linear (for simd directives with the only one associated loop) or
3489     // lastprivate (for simd directives with several collapsed or ordered
3490     // loops).
3491     if (DVar.CKind == OMPC_unknown)
3492       DVar = DSA.hasDSA(Var, isOpenMPPrivate, MatchesAlways(),
3493                         /*FromParent=*/false);
3494     DSA.addDSA(Var, LoopVarRefExpr, PredeterminedCKind);
3495   }
3496 
3497   assert(isOpenMPLoopDirective(DKind) && "DSA for non-loop vars");
3498 
3499   // Check test-expr.
3500   HasErrors |= ISC.CheckCond(For->getCond());
3501 
3502   // Check incr-expr.
3503   HasErrors |= ISC.CheckInc(For->getInc());
3504 
3505   if (ISC.Dependent() || SemaRef.CurContext->isDependentContext() || HasErrors)
3506     return HasErrors;
3507 
3508   // Build the loop's iteration space representation.
3509   ResultIterSpace.PreCond = ISC.BuildPreCond(DSA.getCurScope(), For->getCond());
3510   ResultIterSpace.NumIterations = ISC.BuildNumIterations(
3511       DSA.getCurScope(), (isOpenMPWorksharingDirective(DKind) ||
3512                           isOpenMPTaskLoopDirective(DKind) ||
3513                           isOpenMPDistributeDirective(DKind)));
3514   ResultIterSpace.CounterVar = ISC.BuildCounterVar();
3515   ResultIterSpace.PrivateCounterVar = ISC.BuildPrivateCounterVar();
3516   ResultIterSpace.CounterInit = ISC.BuildCounterInit();
3517   ResultIterSpace.CounterStep = ISC.BuildCounterStep();
3518   ResultIterSpace.InitSrcRange = ISC.GetInitSrcRange();
3519   ResultIterSpace.CondSrcRange = ISC.GetConditionSrcRange();
3520   ResultIterSpace.IncSrcRange = ISC.GetIncrementSrcRange();
3521   ResultIterSpace.Subtract = ISC.ShouldSubtractStep();
3522 
3523   HasErrors |= (ResultIterSpace.PreCond == nullptr ||
3524                 ResultIterSpace.NumIterations == nullptr ||
3525                 ResultIterSpace.CounterVar == nullptr ||
3526                 ResultIterSpace.PrivateCounterVar == nullptr ||
3527                 ResultIterSpace.CounterInit == nullptr ||
3528                 ResultIterSpace.CounterStep == nullptr);
3529 
3530   return HasErrors;
3531 }
3532 
3533 /// \brief Build 'VarRef = Start.
BuildCounterInit(Sema & SemaRef,Scope * S,SourceLocation Loc,ExprResult VarRef,ExprResult Start)3534 static ExprResult BuildCounterInit(Sema &SemaRef, Scope *S, SourceLocation Loc,
3535                                    ExprResult VarRef, ExprResult Start) {
3536   TransformToNewDefs Transform(SemaRef);
3537   // Build 'VarRef = Start.
3538   auto NewStart = Transform.TransformExpr(Start.get()->IgnoreImplicit());
3539   if (NewStart.isInvalid())
3540     return ExprError();
3541   NewStart = SemaRef.PerformImplicitConversion(
3542       NewStart.get(), Start.get()->IgnoreImplicit()->getType(),
3543       Sema::AA_Converting,
3544       /*AllowExplicit=*/true);
3545   if (NewStart.isInvalid())
3546     return ExprError();
3547   NewStart = SemaRef.PerformImplicitConversion(
3548       NewStart.get(), VarRef.get()->getType(), Sema::AA_Converting,
3549       /*AllowExplicit=*/true);
3550   if (!NewStart.isUsable())
3551     return ExprError();
3552 
3553   auto Init =
3554       SemaRef.BuildBinOp(S, Loc, BO_Assign, VarRef.get(), NewStart.get());
3555   return Init;
3556 }
3557 
3558 /// \brief Build 'VarRef = Start + Iter * Step'.
BuildCounterUpdate(Sema & SemaRef,Scope * S,SourceLocation Loc,ExprResult VarRef,ExprResult Start,ExprResult Iter,ExprResult Step,bool Subtract)3559 static ExprResult BuildCounterUpdate(Sema &SemaRef, Scope *S,
3560                                      SourceLocation Loc, ExprResult VarRef,
3561                                      ExprResult Start, ExprResult Iter,
3562                                      ExprResult Step, bool Subtract) {
3563   // Add parentheses (for debugging purposes only).
3564   Iter = SemaRef.ActOnParenExpr(Loc, Loc, Iter.get());
3565   if (!VarRef.isUsable() || !Start.isUsable() || !Iter.isUsable() ||
3566       !Step.isUsable())
3567     return ExprError();
3568 
3569   TransformToNewDefs Transform(SemaRef);
3570   auto NewStep = Transform.TransformExpr(Step.get()->IgnoreImplicit());
3571   if (NewStep.isInvalid())
3572     return ExprError();
3573   NewStep = SemaRef.PerformImplicitConversion(
3574       NewStep.get(), Step.get()->IgnoreImplicit()->getType(),
3575       Sema::AA_Converting,
3576       /*AllowExplicit=*/true);
3577   if (NewStep.isInvalid())
3578     return ExprError();
3579   ExprResult Update =
3580       SemaRef.BuildBinOp(S, Loc, BO_Mul, Iter.get(), NewStep.get());
3581   if (!Update.isUsable())
3582     return ExprError();
3583 
3584   // Build 'VarRef = Start + Iter * Step'.
3585   auto NewStart = Transform.TransformExpr(Start.get()->IgnoreImplicit());
3586   if (NewStart.isInvalid())
3587     return ExprError();
3588   NewStart = SemaRef.PerformImplicitConversion(
3589       NewStart.get(), Start.get()->IgnoreImplicit()->getType(),
3590       Sema::AA_Converting,
3591       /*AllowExplicit=*/true);
3592   if (NewStart.isInvalid())
3593     return ExprError();
3594   Update = SemaRef.BuildBinOp(S, Loc, (Subtract ? BO_Sub : BO_Add),
3595                               NewStart.get(), Update.get());
3596   if (!Update.isUsable())
3597     return ExprError();
3598 
3599   Update = SemaRef.PerformImplicitConversion(
3600       Update.get(), VarRef.get()->getType(), Sema::AA_Converting, true);
3601   if (!Update.isUsable())
3602     return ExprError();
3603 
3604   Update = SemaRef.BuildBinOp(S, Loc, BO_Assign, VarRef.get(), Update.get());
3605   return Update;
3606 }
3607 
3608 /// \brief Convert integer expression \a E to make it have at least \a Bits
3609 /// bits.
WidenIterationCount(unsigned Bits,Expr * E,Sema & SemaRef)3610 static ExprResult WidenIterationCount(unsigned Bits, Expr *E,
3611                                       Sema &SemaRef) {
3612   if (E == nullptr)
3613     return ExprError();
3614   auto &C = SemaRef.Context;
3615   QualType OldType = E->getType();
3616   unsigned HasBits = C.getTypeSize(OldType);
3617   if (HasBits >= Bits)
3618     return ExprResult(E);
3619   // OK to convert to signed, because new type has more bits than old.
3620   QualType NewType = C.getIntTypeForBitwidth(Bits, /* Signed */ true);
3621   return SemaRef.PerformImplicitConversion(E, NewType, Sema::AA_Converting,
3622                                            true);
3623 }
3624 
3625 /// \brief Check if the given expression \a E is a constant integer that fits
3626 /// into \a Bits bits.
FitsInto(unsigned Bits,bool Signed,Expr * E,Sema & SemaRef)3627 static bool FitsInto(unsigned Bits, bool Signed, Expr *E, Sema &SemaRef) {
3628   if (E == nullptr)
3629     return false;
3630   llvm::APSInt Result;
3631   if (E->isIntegerConstantExpr(Result, SemaRef.Context))
3632     return Signed ? Result.isSignedIntN(Bits) : Result.isIntN(Bits);
3633   return false;
3634 }
3635 
3636 /// \brief Called on a for stmt to check itself and nested loops (if any).
3637 /// \return Returns 0 if one of the collapsed stmts is not canonical for loop,
3638 /// number of collapsed loops otherwise.
3639 static unsigned
CheckOpenMPLoop(OpenMPDirectiveKind DKind,Expr * CollapseLoopCountExpr,Expr * OrderedLoopCountExpr,Stmt * AStmt,Sema & SemaRef,DSAStackTy & DSA,llvm::DenseMap<VarDecl *,Expr * > & VarsWithImplicitDSA,OMPLoopDirective::HelperExprs & Built)3640 CheckOpenMPLoop(OpenMPDirectiveKind DKind, Expr *CollapseLoopCountExpr,
3641                 Expr *OrderedLoopCountExpr, Stmt *AStmt, Sema &SemaRef,
3642                 DSAStackTy &DSA,
3643                 llvm::DenseMap<VarDecl *, Expr *> &VarsWithImplicitDSA,
3644                 OMPLoopDirective::HelperExprs &Built) {
3645   unsigned NestedLoopCount = 1;
3646   if (CollapseLoopCountExpr) {
3647     // Found 'collapse' clause - calculate collapse number.
3648     llvm::APSInt Result;
3649     if (CollapseLoopCountExpr->EvaluateAsInt(Result, SemaRef.getASTContext()))
3650       NestedLoopCount = Result.getLimitedValue();
3651   }
3652   if (OrderedLoopCountExpr) {
3653     // Found 'ordered' clause - calculate collapse number.
3654     llvm::APSInt Result;
3655     if (OrderedLoopCountExpr->EvaluateAsInt(Result, SemaRef.getASTContext())) {
3656       if (Result.getLimitedValue() < NestedLoopCount) {
3657         SemaRef.Diag(OrderedLoopCountExpr->getExprLoc(),
3658                      diag::err_omp_wrong_ordered_loop_count)
3659             << OrderedLoopCountExpr->getSourceRange();
3660         SemaRef.Diag(CollapseLoopCountExpr->getExprLoc(),
3661                      diag::note_collapse_loop_count)
3662             << CollapseLoopCountExpr->getSourceRange();
3663       }
3664       NestedLoopCount = Result.getLimitedValue();
3665     }
3666   }
3667   // This is helper routine for loop directives (e.g., 'for', 'simd',
3668   // 'for simd', etc.).
3669   SmallVector<LoopIterationSpace, 4> IterSpaces;
3670   IterSpaces.resize(NestedLoopCount);
3671   Stmt *CurStmt = AStmt->IgnoreContainers(/* IgnoreCaptured */ true);
3672   for (unsigned Cnt = 0; Cnt < NestedLoopCount; ++Cnt) {
3673     if (CheckOpenMPIterationSpace(DKind, CurStmt, SemaRef, DSA, Cnt,
3674                                   NestedLoopCount, CollapseLoopCountExpr,
3675                                   OrderedLoopCountExpr, VarsWithImplicitDSA,
3676                                   IterSpaces[Cnt]))
3677       return 0;
3678     // Move on to the next nested for loop, or to the loop body.
3679     // OpenMP [2.8.1, simd construct, Restrictions]
3680     // All loops associated with the construct must be perfectly nested; that
3681     // is, there must be no intervening code nor any OpenMP directive between
3682     // any two loops.
3683     CurStmt = cast<ForStmt>(CurStmt)->getBody()->IgnoreContainers();
3684   }
3685 
3686   Built.clear(/* size */ NestedLoopCount);
3687 
3688   if (SemaRef.CurContext->isDependentContext())
3689     return NestedLoopCount;
3690 
3691   // An example of what is generated for the following code:
3692   //
3693   //   #pragma omp simd collapse(2) ordered(2)
3694   //   for (i = 0; i < NI; ++i)
3695   //     for (k = 0; k < NK; ++k)
3696   //       for (j = J0; j < NJ; j+=2) {
3697   //         <loop body>
3698   //       }
3699   //
3700   // We generate the code below.
3701   // Note: the loop body may be outlined in CodeGen.
3702   // Note: some counters may be C++ classes, operator- is used to find number of
3703   // iterations and operator+= to calculate counter value.
3704   // Note: decltype(NumIterations) must be integer type (in 'omp for', only i32
3705   // or i64 is currently supported).
3706   //
3707   //   #define NumIterations (NI * ((NJ - J0 - 1 + 2) / 2))
3708   //   for (int[32|64]_t IV = 0; IV < NumIterations; ++IV ) {
3709   //     .local.i = IV / ((NJ - J0 - 1 + 2) / 2);
3710   //     .local.j = J0 + (IV % ((NJ - J0 - 1 + 2) / 2)) * 2;
3711   //     // similar updates for vars in clauses (e.g. 'linear')
3712   //     <loop body (using local i and j)>
3713   //   }
3714   //   i = NI; // assign final values of counters
3715   //   j = NJ;
3716   //
3717 
3718   // Last iteration number is (I1 * I2 * ... In) - 1, where I1, I2 ... In are
3719   // the iteration counts of the collapsed for loops.
3720   // Precondition tests if there is at least one iteration (all conditions are
3721   // true).
3722   auto PreCond = ExprResult(IterSpaces[0].PreCond);
3723   auto N0 = IterSpaces[0].NumIterations;
3724   ExprResult LastIteration32 = WidenIterationCount(
3725       32 /* Bits */, SemaRef.PerformImplicitConversion(
3726                                 N0->IgnoreImpCasts(), N0->getType(),
3727                                 Sema::AA_Converting, /*AllowExplicit=*/true)
3728                          .get(),
3729       SemaRef);
3730   ExprResult LastIteration64 = WidenIterationCount(
3731       64 /* Bits */, SemaRef.PerformImplicitConversion(
3732                                 N0->IgnoreImpCasts(), N0->getType(),
3733                                 Sema::AA_Converting, /*AllowExplicit=*/true)
3734                          .get(),
3735       SemaRef);
3736 
3737   if (!LastIteration32.isUsable() || !LastIteration64.isUsable())
3738     return NestedLoopCount;
3739 
3740   auto &C = SemaRef.Context;
3741   bool AllCountsNeedLessThan32Bits = C.getTypeSize(N0->getType()) < 32;
3742 
3743   Scope *CurScope = DSA.getCurScope();
3744   for (unsigned Cnt = 1; Cnt < NestedLoopCount; ++Cnt) {
3745     if (PreCond.isUsable()) {
3746       PreCond = SemaRef.BuildBinOp(CurScope, SourceLocation(), BO_LAnd,
3747                                    PreCond.get(), IterSpaces[Cnt].PreCond);
3748     }
3749     auto N = IterSpaces[Cnt].NumIterations;
3750     AllCountsNeedLessThan32Bits &= C.getTypeSize(N->getType()) < 32;
3751     if (LastIteration32.isUsable())
3752       LastIteration32 = SemaRef.BuildBinOp(
3753           CurScope, SourceLocation(), BO_Mul, LastIteration32.get(),
3754           SemaRef.PerformImplicitConversion(N->IgnoreImpCasts(), N->getType(),
3755                                             Sema::AA_Converting,
3756                                             /*AllowExplicit=*/true)
3757               .get());
3758     if (LastIteration64.isUsable())
3759       LastIteration64 = SemaRef.BuildBinOp(
3760           CurScope, SourceLocation(), BO_Mul, LastIteration64.get(),
3761           SemaRef.PerformImplicitConversion(N->IgnoreImpCasts(), N->getType(),
3762                                             Sema::AA_Converting,
3763                                             /*AllowExplicit=*/true)
3764               .get());
3765   }
3766 
3767   // Choose either the 32-bit or 64-bit version.
3768   ExprResult LastIteration = LastIteration64;
3769   if (LastIteration32.isUsable() &&
3770       C.getTypeSize(LastIteration32.get()->getType()) == 32 &&
3771       (AllCountsNeedLessThan32Bits || NestedLoopCount == 1 ||
3772        FitsInto(
3773            32 /* Bits */,
3774            LastIteration32.get()->getType()->hasSignedIntegerRepresentation(),
3775            LastIteration64.get(), SemaRef)))
3776     LastIteration = LastIteration32;
3777 
3778   if (!LastIteration.isUsable())
3779     return 0;
3780 
3781   // Save the number of iterations.
3782   ExprResult NumIterations = LastIteration;
3783   {
3784     LastIteration = SemaRef.BuildBinOp(
3785         CurScope, SourceLocation(), BO_Sub, LastIteration.get(),
3786         SemaRef.ActOnIntegerConstant(SourceLocation(), 1).get());
3787     if (!LastIteration.isUsable())
3788       return 0;
3789   }
3790 
3791   // Calculate the last iteration number beforehand instead of doing this on
3792   // each iteration. Do not do this if the number of iterations may be kfold-ed.
3793   llvm::APSInt Result;
3794   bool IsConstant =
3795       LastIteration.get()->isIntegerConstantExpr(Result, SemaRef.Context);
3796   ExprResult CalcLastIteration;
3797   if (!IsConstant) {
3798     SourceLocation SaveLoc;
3799     VarDecl *SaveVar =
3800         buildVarDecl(SemaRef, SaveLoc, LastIteration.get()->getType(),
3801                      ".omp.last.iteration");
3802     ExprResult SaveRef = buildDeclRefExpr(
3803         SemaRef, SaveVar, LastIteration.get()->getType(), SaveLoc);
3804     CalcLastIteration = SemaRef.BuildBinOp(CurScope, SaveLoc, BO_Assign,
3805                                            SaveRef.get(), LastIteration.get());
3806     LastIteration = SaveRef;
3807 
3808     // Prepare SaveRef + 1.
3809     NumIterations = SemaRef.BuildBinOp(
3810         CurScope, SaveLoc, BO_Add, SaveRef.get(),
3811         SemaRef.ActOnIntegerConstant(SourceLocation(), 1).get());
3812     if (!NumIterations.isUsable())
3813       return 0;
3814   }
3815 
3816   SourceLocation InitLoc = IterSpaces[0].InitSrcRange.getBegin();
3817 
3818   QualType VType = LastIteration.get()->getType();
3819   // Build variables passed into runtime, nesessary for worksharing directives.
3820   ExprResult LB, UB, IL, ST, EUB;
3821   if (isOpenMPWorksharingDirective(DKind) || isOpenMPTaskLoopDirective(DKind) ||
3822       isOpenMPDistributeDirective(DKind)) {
3823     // Lower bound variable, initialized with zero.
3824     VarDecl *LBDecl = buildVarDecl(SemaRef, InitLoc, VType, ".omp.lb");
3825     LB = buildDeclRefExpr(SemaRef, LBDecl, VType, InitLoc);
3826     SemaRef.AddInitializerToDecl(
3827         LBDecl, SemaRef.ActOnIntegerConstant(InitLoc, 0).get(),
3828         /*DirectInit*/ false, /*TypeMayContainAuto*/ false);
3829 
3830     // Upper bound variable, initialized with last iteration number.
3831     VarDecl *UBDecl = buildVarDecl(SemaRef, InitLoc, VType, ".omp.ub");
3832     UB = buildDeclRefExpr(SemaRef, UBDecl, VType, InitLoc);
3833     SemaRef.AddInitializerToDecl(UBDecl, LastIteration.get(),
3834                                  /*DirectInit*/ false,
3835                                  /*TypeMayContainAuto*/ false);
3836 
3837     // A 32-bit variable-flag where runtime returns 1 for the last iteration.
3838     // This will be used to implement clause 'lastprivate'.
3839     QualType Int32Ty = SemaRef.Context.getIntTypeForBitwidth(32, true);
3840     VarDecl *ILDecl = buildVarDecl(SemaRef, InitLoc, Int32Ty, ".omp.is_last");
3841     IL = buildDeclRefExpr(SemaRef, ILDecl, Int32Ty, InitLoc);
3842     SemaRef.AddInitializerToDecl(
3843         ILDecl, SemaRef.ActOnIntegerConstant(InitLoc, 0).get(),
3844         /*DirectInit*/ false, /*TypeMayContainAuto*/ false);
3845 
3846     // Stride variable returned by runtime (we initialize it to 1 by default).
3847     VarDecl *STDecl = buildVarDecl(SemaRef, InitLoc, VType, ".omp.stride");
3848     ST = buildDeclRefExpr(SemaRef, STDecl, VType, InitLoc);
3849     SemaRef.AddInitializerToDecl(
3850         STDecl, SemaRef.ActOnIntegerConstant(InitLoc, 1).get(),
3851         /*DirectInit*/ false, /*TypeMayContainAuto*/ false);
3852 
3853     // Build expression: UB = min(UB, LastIteration)
3854     // It is nesessary for CodeGen of directives with static scheduling.
3855     ExprResult IsUBGreater = SemaRef.BuildBinOp(CurScope, InitLoc, BO_GT,
3856                                                 UB.get(), LastIteration.get());
3857     ExprResult CondOp = SemaRef.ActOnConditionalOp(
3858         InitLoc, InitLoc, IsUBGreater.get(), LastIteration.get(), UB.get());
3859     EUB = SemaRef.BuildBinOp(CurScope, InitLoc, BO_Assign, UB.get(),
3860                              CondOp.get());
3861     EUB = SemaRef.ActOnFinishFullExpr(EUB.get());
3862   }
3863 
3864   // Build the iteration variable and its initialization before loop.
3865   ExprResult IV;
3866   ExprResult Init;
3867   {
3868     VarDecl *IVDecl = buildVarDecl(SemaRef, InitLoc, VType, ".omp.iv");
3869     IV = buildDeclRefExpr(SemaRef, IVDecl, VType, InitLoc);
3870     Expr *RHS = (isOpenMPWorksharingDirective(DKind) ||
3871                  isOpenMPTaskLoopDirective(DKind) ||
3872                  isOpenMPDistributeDirective(DKind))
3873                     ? LB.get()
3874                     : SemaRef.ActOnIntegerConstant(SourceLocation(), 0).get();
3875     Init = SemaRef.BuildBinOp(CurScope, InitLoc, BO_Assign, IV.get(), RHS);
3876     Init = SemaRef.ActOnFinishFullExpr(Init.get());
3877   }
3878 
3879   // Loop condition (IV < NumIterations) or (IV <= UB) for worksharing loops.
3880   SourceLocation CondLoc;
3881   ExprResult Cond =
3882       (isOpenMPWorksharingDirective(DKind) ||
3883        isOpenMPTaskLoopDirective(DKind) || isOpenMPDistributeDirective(DKind))
3884           ? SemaRef.BuildBinOp(CurScope, CondLoc, BO_LE, IV.get(), UB.get())
3885           : SemaRef.BuildBinOp(CurScope, CondLoc, BO_LT, IV.get(),
3886                                NumIterations.get());
3887 
3888   // Loop increment (IV = IV + 1)
3889   SourceLocation IncLoc;
3890   ExprResult Inc =
3891       SemaRef.BuildBinOp(CurScope, IncLoc, BO_Add, IV.get(),
3892                          SemaRef.ActOnIntegerConstant(IncLoc, 1).get());
3893   if (!Inc.isUsable())
3894     return 0;
3895   Inc = SemaRef.BuildBinOp(CurScope, IncLoc, BO_Assign, IV.get(), Inc.get());
3896   Inc = SemaRef.ActOnFinishFullExpr(Inc.get());
3897   if (!Inc.isUsable())
3898     return 0;
3899 
3900   // Increments for worksharing loops (LB = LB + ST; UB = UB + ST).
3901   // Used for directives with static scheduling.
3902   ExprResult NextLB, NextUB;
3903   if (isOpenMPWorksharingDirective(DKind) || isOpenMPTaskLoopDirective(DKind) ||
3904       isOpenMPDistributeDirective(DKind)) {
3905     // LB + ST
3906     NextLB = SemaRef.BuildBinOp(CurScope, IncLoc, BO_Add, LB.get(), ST.get());
3907     if (!NextLB.isUsable())
3908       return 0;
3909     // LB = LB + ST
3910     NextLB =
3911         SemaRef.BuildBinOp(CurScope, IncLoc, BO_Assign, LB.get(), NextLB.get());
3912     NextLB = SemaRef.ActOnFinishFullExpr(NextLB.get());
3913     if (!NextLB.isUsable())
3914       return 0;
3915     // UB + ST
3916     NextUB = SemaRef.BuildBinOp(CurScope, IncLoc, BO_Add, UB.get(), ST.get());
3917     if (!NextUB.isUsable())
3918       return 0;
3919     // UB = UB + ST
3920     NextUB =
3921         SemaRef.BuildBinOp(CurScope, IncLoc, BO_Assign, UB.get(), NextUB.get());
3922     NextUB = SemaRef.ActOnFinishFullExpr(NextUB.get());
3923     if (!NextUB.isUsable())
3924       return 0;
3925   }
3926 
3927   // Build updates and final values of the loop counters.
3928   bool HasErrors = false;
3929   Built.Counters.resize(NestedLoopCount);
3930   Built.Inits.resize(NestedLoopCount);
3931   Built.Updates.resize(NestedLoopCount);
3932   Built.Finals.resize(NestedLoopCount);
3933   {
3934     ExprResult Div;
3935     // Go from inner nested loop to outer.
3936     for (int Cnt = NestedLoopCount - 1; Cnt >= 0; --Cnt) {
3937       LoopIterationSpace &IS = IterSpaces[Cnt];
3938       SourceLocation UpdLoc = IS.IncSrcRange.getBegin();
3939       // Build: Iter = (IV / Div) % IS.NumIters
3940       // where Div is product of previous iterations' IS.NumIters.
3941       ExprResult Iter;
3942       if (Div.isUsable()) {
3943         Iter =
3944             SemaRef.BuildBinOp(CurScope, UpdLoc, BO_Div, IV.get(), Div.get());
3945       } else {
3946         Iter = IV;
3947         assert((Cnt == (int)NestedLoopCount - 1) &&
3948                "unusable div expected on first iteration only");
3949       }
3950 
3951       if (Cnt != 0 && Iter.isUsable())
3952         Iter = SemaRef.BuildBinOp(CurScope, UpdLoc, BO_Rem, Iter.get(),
3953                                   IS.NumIterations);
3954       if (!Iter.isUsable()) {
3955         HasErrors = true;
3956         break;
3957       }
3958 
3959       // Build update: IS.CounterVar(Private) = IS.Start + Iter * IS.Step
3960       auto *CounterVar = buildDeclRefExpr(
3961           SemaRef, cast<VarDecl>(cast<DeclRefExpr>(IS.CounterVar)->getDecl()),
3962           IS.CounterVar->getType(), IS.CounterVar->getExprLoc(),
3963           /*RefersToCapture=*/true);
3964       ExprResult Init = BuildCounterInit(SemaRef, CurScope, UpdLoc, CounterVar,
3965                                          IS.CounterInit);
3966       if (!Init.isUsable()) {
3967         HasErrors = true;
3968         break;
3969       }
3970       ExprResult Update =
3971           BuildCounterUpdate(SemaRef, CurScope, UpdLoc, CounterVar,
3972                              IS.CounterInit, Iter, IS.CounterStep, IS.Subtract);
3973       if (!Update.isUsable()) {
3974         HasErrors = true;
3975         break;
3976       }
3977 
3978       // Build final: IS.CounterVar = IS.Start + IS.NumIters * IS.Step
3979       ExprResult Final = BuildCounterUpdate(
3980           SemaRef, CurScope, UpdLoc, CounterVar, IS.CounterInit,
3981           IS.NumIterations, IS.CounterStep, IS.Subtract);
3982       if (!Final.isUsable()) {
3983         HasErrors = true;
3984         break;
3985       }
3986 
3987       // Build Div for the next iteration: Div <- Div * IS.NumIters
3988       if (Cnt != 0) {
3989         if (Div.isUnset())
3990           Div = IS.NumIterations;
3991         else
3992           Div = SemaRef.BuildBinOp(CurScope, UpdLoc, BO_Mul, Div.get(),
3993                                    IS.NumIterations);
3994 
3995         // Add parentheses (for debugging purposes only).
3996         if (Div.isUsable())
3997           Div = SemaRef.ActOnParenExpr(UpdLoc, UpdLoc, Div.get());
3998         if (!Div.isUsable()) {
3999           HasErrors = true;
4000           break;
4001         }
4002       }
4003       if (!Update.isUsable() || !Final.isUsable()) {
4004         HasErrors = true;
4005         break;
4006       }
4007       // Save results
4008       Built.Counters[Cnt] = IS.CounterVar;
4009       Built.PrivateCounters[Cnt] = IS.PrivateCounterVar;
4010       Built.Inits[Cnt] = Init.get();
4011       Built.Updates[Cnt] = Update.get();
4012       Built.Finals[Cnt] = Final.get();
4013     }
4014   }
4015 
4016   if (HasErrors)
4017     return 0;
4018 
4019   // Save results
4020   Built.IterationVarRef = IV.get();
4021   Built.LastIteration = LastIteration.get();
4022   Built.NumIterations = NumIterations.get();
4023   Built.CalcLastIteration =
4024       SemaRef.ActOnFinishFullExpr(CalcLastIteration.get()).get();
4025   Built.PreCond = PreCond.get();
4026   Built.Cond = Cond.get();
4027   Built.Init = Init.get();
4028   Built.Inc = Inc.get();
4029   Built.LB = LB.get();
4030   Built.UB = UB.get();
4031   Built.IL = IL.get();
4032   Built.ST = ST.get();
4033   Built.EUB = EUB.get();
4034   Built.NLB = NextLB.get();
4035   Built.NUB = NextUB.get();
4036 
4037   return NestedLoopCount;
4038 }
4039 
getCollapseNumberExpr(ArrayRef<OMPClause * > Clauses)4040 static Expr *getCollapseNumberExpr(ArrayRef<OMPClause *> Clauses) {
4041   auto CollapseClauses =
4042       OMPExecutableDirective::getClausesOfKind<OMPCollapseClause>(Clauses);
4043   if (CollapseClauses.begin() != CollapseClauses.end())
4044     return (*CollapseClauses.begin())->getNumForLoops();
4045   return nullptr;
4046 }
4047 
getOrderedNumberExpr(ArrayRef<OMPClause * > Clauses)4048 static Expr *getOrderedNumberExpr(ArrayRef<OMPClause *> Clauses) {
4049   auto OrderedClauses =
4050       OMPExecutableDirective::getClausesOfKind<OMPOrderedClause>(Clauses);
4051   if (OrderedClauses.begin() != OrderedClauses.end())
4052     return (*OrderedClauses.begin())->getNumForLoops();
4053   return nullptr;
4054 }
4055 
checkSimdlenSafelenValues(Sema & S,const Expr * Simdlen,const Expr * Safelen)4056 static bool checkSimdlenSafelenValues(Sema &S, const Expr *Simdlen,
4057                                       const Expr *Safelen) {
4058   llvm::APSInt SimdlenRes, SafelenRes;
4059   if (Simdlen->isValueDependent() || Simdlen->isTypeDependent() ||
4060       Simdlen->isInstantiationDependent() ||
4061       Simdlen->containsUnexpandedParameterPack())
4062     return false;
4063   if (Safelen->isValueDependent() || Safelen->isTypeDependent() ||
4064       Safelen->isInstantiationDependent() ||
4065       Safelen->containsUnexpandedParameterPack())
4066     return false;
4067   Simdlen->EvaluateAsInt(SimdlenRes, S.Context);
4068   Safelen->EvaluateAsInt(SafelenRes, S.Context);
4069   // OpenMP 4.1 [2.8.1, simd Construct, Restrictions]
4070   // If both simdlen and safelen clauses are specified, the value of the simdlen
4071   // parameter must be less than or equal to the value of the safelen parameter.
4072   if (SimdlenRes > SafelenRes) {
4073     S.Diag(Simdlen->getExprLoc(), diag::err_omp_wrong_simdlen_safelen_values)
4074         << Simdlen->getSourceRange() << Safelen->getSourceRange();
4075     return true;
4076   }
4077   return false;
4078 }
4079 
ActOnOpenMPSimdDirective(ArrayRef<OMPClause * > Clauses,Stmt * AStmt,SourceLocation StartLoc,SourceLocation EndLoc,llvm::DenseMap<VarDecl *,Expr * > & VarsWithImplicitDSA)4080 StmtResult Sema::ActOnOpenMPSimdDirective(
4081     ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc,
4082     SourceLocation EndLoc,
4083     llvm::DenseMap<VarDecl *, Expr *> &VarsWithImplicitDSA) {
4084   if (!AStmt)
4085     return StmtError();
4086 
4087   assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
4088   OMPLoopDirective::HelperExprs B;
4089   // In presence of clause 'collapse' or 'ordered' with number of loops, it will
4090   // define the nested loops number.
4091   unsigned NestedLoopCount = CheckOpenMPLoop(
4092       OMPD_simd, getCollapseNumberExpr(Clauses), getOrderedNumberExpr(Clauses),
4093       AStmt, *this, *DSAStack, VarsWithImplicitDSA, B);
4094   if (NestedLoopCount == 0)
4095     return StmtError();
4096 
4097   assert((CurContext->isDependentContext() || B.builtAll()) &&
4098          "omp simd loop exprs were not built");
4099 
4100   if (!CurContext->isDependentContext()) {
4101     // Finalize the clauses that need pre-built expressions for CodeGen.
4102     for (auto C : Clauses) {
4103       if (auto LC = dyn_cast<OMPLinearClause>(C))
4104         if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef),
4105                                      B.NumIterations, *this, CurScope))
4106           return StmtError();
4107     }
4108   }
4109 
4110   // OpenMP 4.1 [2.8.1, simd Construct, Restrictions]
4111   // If both simdlen and safelen clauses are specified, the value of the simdlen
4112   // parameter must be less than or equal to the value of the safelen parameter.
4113   OMPSafelenClause *Safelen = nullptr;
4114   OMPSimdlenClause *Simdlen = nullptr;
4115   for (auto *Clause : Clauses) {
4116     if (Clause->getClauseKind() == OMPC_safelen)
4117       Safelen = cast<OMPSafelenClause>(Clause);
4118     else if (Clause->getClauseKind() == OMPC_simdlen)
4119       Simdlen = cast<OMPSimdlenClause>(Clause);
4120     if (Safelen && Simdlen)
4121       break;
4122   }
4123   if (Simdlen && Safelen &&
4124       checkSimdlenSafelenValues(*this, Simdlen->getSimdlen(),
4125                                 Safelen->getSafelen()))
4126     return StmtError();
4127 
4128   getCurFunction()->setHasBranchProtectedScope();
4129   return OMPSimdDirective::Create(Context, StartLoc, EndLoc, NestedLoopCount,
4130                                   Clauses, AStmt, B);
4131 }
4132 
ActOnOpenMPForDirective(ArrayRef<OMPClause * > Clauses,Stmt * AStmt,SourceLocation StartLoc,SourceLocation EndLoc,llvm::DenseMap<VarDecl *,Expr * > & VarsWithImplicitDSA)4133 StmtResult Sema::ActOnOpenMPForDirective(
4134     ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc,
4135     SourceLocation EndLoc,
4136     llvm::DenseMap<VarDecl *, Expr *> &VarsWithImplicitDSA) {
4137   if (!AStmt)
4138     return StmtError();
4139 
4140   assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
4141   OMPLoopDirective::HelperExprs B;
4142   // In presence of clause 'collapse' or 'ordered' with number of loops, it will
4143   // define the nested loops number.
4144   unsigned NestedLoopCount = CheckOpenMPLoop(
4145       OMPD_for, getCollapseNumberExpr(Clauses), getOrderedNumberExpr(Clauses),
4146       AStmt, *this, *DSAStack, VarsWithImplicitDSA, B);
4147   if (NestedLoopCount == 0)
4148     return StmtError();
4149 
4150   assert((CurContext->isDependentContext() || B.builtAll()) &&
4151          "omp for loop exprs were not built");
4152 
4153   if (!CurContext->isDependentContext()) {
4154     // Finalize the clauses that need pre-built expressions for CodeGen.
4155     for (auto C : Clauses) {
4156       if (auto LC = dyn_cast<OMPLinearClause>(C))
4157         if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef),
4158                                      B.NumIterations, *this, CurScope))
4159           return StmtError();
4160     }
4161   }
4162 
4163   getCurFunction()->setHasBranchProtectedScope();
4164   return OMPForDirective::Create(Context, StartLoc, EndLoc, NestedLoopCount,
4165                                  Clauses, AStmt, B, DSAStack->isCancelRegion());
4166 }
4167 
ActOnOpenMPForSimdDirective(ArrayRef<OMPClause * > Clauses,Stmt * AStmt,SourceLocation StartLoc,SourceLocation EndLoc,llvm::DenseMap<VarDecl *,Expr * > & VarsWithImplicitDSA)4168 StmtResult Sema::ActOnOpenMPForSimdDirective(
4169     ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc,
4170     SourceLocation EndLoc,
4171     llvm::DenseMap<VarDecl *, Expr *> &VarsWithImplicitDSA) {
4172   if (!AStmt)
4173     return StmtError();
4174 
4175   assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
4176   OMPLoopDirective::HelperExprs B;
4177   // In presence of clause 'collapse' or 'ordered' with number of loops, it will
4178   // define the nested loops number.
4179   unsigned NestedLoopCount =
4180       CheckOpenMPLoop(OMPD_for_simd, getCollapseNumberExpr(Clauses),
4181                       getOrderedNumberExpr(Clauses), AStmt, *this, *DSAStack,
4182                       VarsWithImplicitDSA, B);
4183   if (NestedLoopCount == 0)
4184     return StmtError();
4185 
4186   assert((CurContext->isDependentContext() || B.builtAll()) &&
4187          "omp for simd loop exprs were not built");
4188 
4189   if (!CurContext->isDependentContext()) {
4190     // Finalize the clauses that need pre-built expressions for CodeGen.
4191     for (auto C : Clauses) {
4192       if (auto LC = dyn_cast<OMPLinearClause>(C))
4193         if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef),
4194                                      B.NumIterations, *this, CurScope))
4195           return StmtError();
4196     }
4197   }
4198 
4199   // OpenMP 4.1 [2.8.1, simd Construct, Restrictions]
4200   // If both simdlen and safelen clauses are specified, the value of the simdlen
4201   // parameter must be less than or equal to the value of the safelen parameter.
4202   OMPSafelenClause *Safelen = nullptr;
4203   OMPSimdlenClause *Simdlen = nullptr;
4204   for (auto *Clause : Clauses) {
4205     if (Clause->getClauseKind() == OMPC_safelen)
4206       Safelen = cast<OMPSafelenClause>(Clause);
4207     else if (Clause->getClauseKind() == OMPC_simdlen)
4208       Simdlen = cast<OMPSimdlenClause>(Clause);
4209     if (Safelen && Simdlen)
4210       break;
4211   }
4212   if (Simdlen && Safelen &&
4213       checkSimdlenSafelenValues(*this, Simdlen->getSimdlen(),
4214                                 Safelen->getSafelen()))
4215     return StmtError();
4216 
4217   getCurFunction()->setHasBranchProtectedScope();
4218   return OMPForSimdDirective::Create(Context, StartLoc, EndLoc, NestedLoopCount,
4219                                      Clauses, AStmt, B);
4220 }
4221 
ActOnOpenMPSectionsDirective(ArrayRef<OMPClause * > Clauses,Stmt * AStmt,SourceLocation StartLoc,SourceLocation EndLoc)4222 StmtResult Sema::ActOnOpenMPSectionsDirective(ArrayRef<OMPClause *> Clauses,
4223                                               Stmt *AStmt,
4224                                               SourceLocation StartLoc,
4225                                               SourceLocation EndLoc) {
4226   if (!AStmt)
4227     return StmtError();
4228 
4229   assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
4230   auto BaseStmt = AStmt;
4231   while (CapturedStmt *CS = dyn_cast_or_null<CapturedStmt>(BaseStmt))
4232     BaseStmt = CS->getCapturedStmt();
4233   if (auto C = dyn_cast_or_null<CompoundStmt>(BaseStmt)) {
4234     auto S = C->children();
4235     if (S.begin() == S.end())
4236       return StmtError();
4237     // All associated statements must be '#pragma omp section' except for
4238     // the first one.
4239     for (Stmt *SectionStmt : llvm::make_range(std::next(S.begin()), S.end())) {
4240       if (!SectionStmt || !isa<OMPSectionDirective>(SectionStmt)) {
4241         if (SectionStmt)
4242           Diag(SectionStmt->getLocStart(),
4243                diag::err_omp_sections_substmt_not_section);
4244         return StmtError();
4245       }
4246       cast<OMPSectionDirective>(SectionStmt)
4247           ->setHasCancel(DSAStack->isCancelRegion());
4248     }
4249   } else {
4250     Diag(AStmt->getLocStart(), diag::err_omp_sections_not_compound_stmt);
4251     return StmtError();
4252   }
4253 
4254   getCurFunction()->setHasBranchProtectedScope();
4255 
4256   return OMPSectionsDirective::Create(Context, StartLoc, EndLoc, Clauses, AStmt,
4257                                       DSAStack->isCancelRegion());
4258 }
4259 
ActOnOpenMPSectionDirective(Stmt * AStmt,SourceLocation StartLoc,SourceLocation EndLoc)4260 StmtResult Sema::ActOnOpenMPSectionDirective(Stmt *AStmt,
4261                                              SourceLocation StartLoc,
4262                                              SourceLocation EndLoc) {
4263   if (!AStmt)
4264     return StmtError();
4265 
4266   assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
4267 
4268   getCurFunction()->setHasBranchProtectedScope();
4269   DSAStack->setParentCancelRegion(DSAStack->isCancelRegion());
4270 
4271   return OMPSectionDirective::Create(Context, StartLoc, EndLoc, AStmt,
4272                                      DSAStack->isCancelRegion());
4273 }
4274 
ActOnOpenMPSingleDirective(ArrayRef<OMPClause * > Clauses,Stmt * AStmt,SourceLocation StartLoc,SourceLocation EndLoc)4275 StmtResult Sema::ActOnOpenMPSingleDirective(ArrayRef<OMPClause *> Clauses,
4276                                             Stmt *AStmt,
4277                                             SourceLocation StartLoc,
4278                                             SourceLocation EndLoc) {
4279   if (!AStmt)
4280     return StmtError();
4281 
4282   assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
4283 
4284   getCurFunction()->setHasBranchProtectedScope();
4285 
4286   // OpenMP [2.7.3, single Construct, Restrictions]
4287   // The copyprivate clause must not be used with the nowait clause.
4288   OMPClause *Nowait = nullptr;
4289   OMPClause *Copyprivate = nullptr;
4290   for (auto *Clause : Clauses) {
4291     if (Clause->getClauseKind() == OMPC_nowait)
4292       Nowait = Clause;
4293     else if (Clause->getClauseKind() == OMPC_copyprivate)
4294       Copyprivate = Clause;
4295     if (Copyprivate && Nowait) {
4296       Diag(Copyprivate->getLocStart(),
4297            diag::err_omp_single_copyprivate_with_nowait);
4298       Diag(Nowait->getLocStart(), diag::note_omp_nowait_clause_here);
4299       return StmtError();
4300     }
4301   }
4302 
4303   return OMPSingleDirective::Create(Context, StartLoc, EndLoc, Clauses, AStmt);
4304 }
4305 
ActOnOpenMPMasterDirective(Stmt * AStmt,SourceLocation StartLoc,SourceLocation EndLoc)4306 StmtResult Sema::ActOnOpenMPMasterDirective(Stmt *AStmt,
4307                                             SourceLocation StartLoc,
4308                                             SourceLocation EndLoc) {
4309   if (!AStmt)
4310     return StmtError();
4311 
4312   assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
4313 
4314   getCurFunction()->setHasBranchProtectedScope();
4315 
4316   return OMPMasterDirective::Create(Context, StartLoc, EndLoc, AStmt);
4317 }
4318 
ActOnOpenMPCriticalDirective(const DeclarationNameInfo & DirName,ArrayRef<OMPClause * > Clauses,Stmt * AStmt,SourceLocation StartLoc,SourceLocation EndLoc)4319 StmtResult Sema::ActOnOpenMPCriticalDirective(
4320     const DeclarationNameInfo &DirName, ArrayRef<OMPClause *> Clauses,
4321     Stmt *AStmt, SourceLocation StartLoc, SourceLocation EndLoc) {
4322   if (!AStmt)
4323     return StmtError();
4324 
4325   assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
4326 
4327   bool ErrorFound = false;
4328   llvm::APSInt Hint;
4329   SourceLocation HintLoc;
4330   bool DependentHint = false;
4331   for (auto *C : Clauses) {
4332     if (C->getClauseKind() == OMPC_hint) {
4333       if (!DirName.getName()) {
4334         Diag(C->getLocStart(), diag::err_omp_hint_clause_no_name);
4335         ErrorFound = true;
4336       }
4337       Expr *E = cast<OMPHintClause>(C)->getHint();
4338       if (E->isTypeDependent() || E->isValueDependent() ||
4339           E->isInstantiationDependent())
4340         DependentHint = true;
4341       else {
4342         Hint = E->EvaluateKnownConstInt(Context);
4343         HintLoc = C->getLocStart();
4344       }
4345     }
4346   }
4347   if (ErrorFound)
4348     return StmtError();
4349   auto Pair = DSAStack->getCriticalWithHint(DirName);
4350   if (Pair.first && DirName.getName() && !DependentHint) {
4351     if (llvm::APSInt::compareValues(Hint, Pair.second) != 0) {
4352       Diag(StartLoc, diag::err_omp_critical_with_hint);
4353       if (HintLoc.isValid()) {
4354         Diag(HintLoc, diag::note_omp_critical_hint_here)
4355             << 0 << Hint.toString(/*Radix=*/10, /*Signed=*/false);
4356       } else
4357         Diag(StartLoc, diag::note_omp_critical_no_hint) << 0;
4358       if (auto *C = Pair.first->getSingleClause<OMPHintClause>()) {
4359         Diag(C->getLocStart(), diag::note_omp_critical_hint_here)
4360             << 1
4361             << C->getHint()->EvaluateKnownConstInt(Context).toString(
4362                    /*Radix=*/10, /*Signed=*/false);
4363       } else
4364         Diag(Pair.first->getLocStart(), diag::note_omp_critical_no_hint) << 1;
4365     }
4366   }
4367 
4368   getCurFunction()->setHasBranchProtectedScope();
4369 
4370   auto *Dir = OMPCriticalDirective::Create(Context, DirName, StartLoc, EndLoc,
4371                                            Clauses, AStmt);
4372   if (!Pair.first && DirName.getName() && !DependentHint)
4373     DSAStack->addCriticalWithHint(Dir, Hint);
4374   return Dir;
4375 }
4376 
ActOnOpenMPParallelForDirective(ArrayRef<OMPClause * > Clauses,Stmt * AStmt,SourceLocation StartLoc,SourceLocation EndLoc,llvm::DenseMap<VarDecl *,Expr * > & VarsWithImplicitDSA)4377 StmtResult Sema::ActOnOpenMPParallelForDirective(
4378     ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc,
4379     SourceLocation EndLoc,
4380     llvm::DenseMap<VarDecl *, Expr *> &VarsWithImplicitDSA) {
4381   if (!AStmt)
4382     return StmtError();
4383 
4384   CapturedStmt *CS = cast<CapturedStmt>(AStmt);
4385   // 1.2.2 OpenMP Language Terminology
4386   // Structured block - An executable statement with a single entry at the
4387   // top and a single exit at the bottom.
4388   // The point of exit cannot be a branch out of the structured block.
4389   // longjmp() and throw() must not violate the entry/exit criteria.
4390   CS->getCapturedDecl()->setNothrow();
4391 
4392   OMPLoopDirective::HelperExprs B;
4393   // In presence of clause 'collapse' or 'ordered' with number of loops, it will
4394   // define the nested loops number.
4395   unsigned NestedLoopCount =
4396       CheckOpenMPLoop(OMPD_parallel_for, getCollapseNumberExpr(Clauses),
4397                       getOrderedNumberExpr(Clauses), AStmt, *this, *DSAStack,
4398                       VarsWithImplicitDSA, B);
4399   if (NestedLoopCount == 0)
4400     return StmtError();
4401 
4402   assert((CurContext->isDependentContext() || B.builtAll()) &&
4403          "omp parallel for loop exprs were not built");
4404 
4405   if (!CurContext->isDependentContext()) {
4406     // Finalize the clauses that need pre-built expressions for CodeGen.
4407     for (auto C : Clauses) {
4408       if (auto LC = dyn_cast<OMPLinearClause>(C))
4409         if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef),
4410                                      B.NumIterations, *this, CurScope))
4411           return StmtError();
4412     }
4413   }
4414 
4415   getCurFunction()->setHasBranchProtectedScope();
4416   return OMPParallelForDirective::Create(Context, StartLoc, EndLoc,
4417                                          NestedLoopCount, Clauses, AStmt, B,
4418                                          DSAStack->isCancelRegion());
4419 }
4420 
ActOnOpenMPParallelForSimdDirective(ArrayRef<OMPClause * > Clauses,Stmt * AStmt,SourceLocation StartLoc,SourceLocation EndLoc,llvm::DenseMap<VarDecl *,Expr * > & VarsWithImplicitDSA)4421 StmtResult Sema::ActOnOpenMPParallelForSimdDirective(
4422     ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc,
4423     SourceLocation EndLoc,
4424     llvm::DenseMap<VarDecl *, Expr *> &VarsWithImplicitDSA) {
4425   if (!AStmt)
4426     return StmtError();
4427 
4428   CapturedStmt *CS = cast<CapturedStmt>(AStmt);
4429   // 1.2.2 OpenMP Language Terminology
4430   // Structured block - An executable statement with a single entry at the
4431   // top and a single exit at the bottom.
4432   // The point of exit cannot be a branch out of the structured block.
4433   // longjmp() and throw() must not violate the entry/exit criteria.
4434   CS->getCapturedDecl()->setNothrow();
4435 
4436   OMPLoopDirective::HelperExprs B;
4437   // In presence of clause 'collapse' or 'ordered' with number of loops, it will
4438   // define the nested loops number.
4439   unsigned NestedLoopCount =
4440       CheckOpenMPLoop(OMPD_parallel_for_simd, getCollapseNumberExpr(Clauses),
4441                       getOrderedNumberExpr(Clauses), AStmt, *this, *DSAStack,
4442                       VarsWithImplicitDSA, B);
4443   if (NestedLoopCount == 0)
4444     return StmtError();
4445 
4446   if (!CurContext->isDependentContext()) {
4447     // Finalize the clauses that need pre-built expressions for CodeGen.
4448     for (auto C : Clauses) {
4449       if (auto LC = dyn_cast<OMPLinearClause>(C))
4450         if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef),
4451                                      B.NumIterations, *this, CurScope))
4452           return StmtError();
4453     }
4454   }
4455 
4456   // OpenMP 4.1 [2.8.1, simd Construct, Restrictions]
4457   // If both simdlen and safelen clauses are specified, the value of the simdlen
4458   // parameter must be less than or equal to the value of the safelen parameter.
4459   OMPSafelenClause *Safelen = nullptr;
4460   OMPSimdlenClause *Simdlen = nullptr;
4461   for (auto *Clause : Clauses) {
4462     if (Clause->getClauseKind() == OMPC_safelen)
4463       Safelen = cast<OMPSafelenClause>(Clause);
4464     else if (Clause->getClauseKind() == OMPC_simdlen)
4465       Simdlen = cast<OMPSimdlenClause>(Clause);
4466     if (Safelen && Simdlen)
4467       break;
4468   }
4469   if (Simdlen && Safelen &&
4470       checkSimdlenSafelenValues(*this, Simdlen->getSimdlen(),
4471                                 Safelen->getSafelen()))
4472     return StmtError();
4473 
4474   getCurFunction()->setHasBranchProtectedScope();
4475   return OMPParallelForSimdDirective::Create(
4476       Context, StartLoc, EndLoc, NestedLoopCount, Clauses, AStmt, B);
4477 }
4478 
4479 StmtResult
ActOnOpenMPParallelSectionsDirective(ArrayRef<OMPClause * > Clauses,Stmt * AStmt,SourceLocation StartLoc,SourceLocation EndLoc)4480 Sema::ActOnOpenMPParallelSectionsDirective(ArrayRef<OMPClause *> Clauses,
4481                                            Stmt *AStmt, SourceLocation StartLoc,
4482                                            SourceLocation EndLoc) {
4483   if (!AStmt)
4484     return StmtError();
4485 
4486   assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
4487   auto BaseStmt = AStmt;
4488   while (CapturedStmt *CS = dyn_cast_or_null<CapturedStmt>(BaseStmt))
4489     BaseStmt = CS->getCapturedStmt();
4490   if (auto C = dyn_cast_or_null<CompoundStmt>(BaseStmt)) {
4491     auto S = C->children();
4492     if (S.begin() == S.end())
4493       return StmtError();
4494     // All associated statements must be '#pragma omp section' except for
4495     // the first one.
4496     for (Stmt *SectionStmt : llvm::make_range(std::next(S.begin()), S.end())) {
4497       if (!SectionStmt || !isa<OMPSectionDirective>(SectionStmt)) {
4498         if (SectionStmt)
4499           Diag(SectionStmt->getLocStart(),
4500                diag::err_omp_parallel_sections_substmt_not_section);
4501         return StmtError();
4502       }
4503       cast<OMPSectionDirective>(SectionStmt)
4504           ->setHasCancel(DSAStack->isCancelRegion());
4505     }
4506   } else {
4507     Diag(AStmt->getLocStart(),
4508          diag::err_omp_parallel_sections_not_compound_stmt);
4509     return StmtError();
4510   }
4511 
4512   getCurFunction()->setHasBranchProtectedScope();
4513 
4514   return OMPParallelSectionsDirective::Create(
4515       Context, StartLoc, EndLoc, Clauses, AStmt, DSAStack->isCancelRegion());
4516 }
4517 
ActOnOpenMPTaskDirective(ArrayRef<OMPClause * > Clauses,Stmt * AStmt,SourceLocation StartLoc,SourceLocation EndLoc)4518 StmtResult Sema::ActOnOpenMPTaskDirective(ArrayRef<OMPClause *> Clauses,
4519                                           Stmt *AStmt, SourceLocation StartLoc,
4520                                           SourceLocation EndLoc) {
4521   if (!AStmt)
4522     return StmtError();
4523 
4524   CapturedStmt *CS = cast<CapturedStmt>(AStmt);
4525   // 1.2.2 OpenMP Language Terminology
4526   // Structured block - An executable statement with a single entry at the
4527   // top and a single exit at the bottom.
4528   // The point of exit cannot be a branch out of the structured block.
4529   // longjmp() and throw() must not violate the entry/exit criteria.
4530   CS->getCapturedDecl()->setNothrow();
4531 
4532   getCurFunction()->setHasBranchProtectedScope();
4533 
4534   return OMPTaskDirective::Create(Context, StartLoc, EndLoc, Clauses, AStmt,
4535                                   DSAStack->isCancelRegion());
4536 }
4537 
ActOnOpenMPTaskyieldDirective(SourceLocation StartLoc,SourceLocation EndLoc)4538 StmtResult Sema::ActOnOpenMPTaskyieldDirective(SourceLocation StartLoc,
4539                                                SourceLocation EndLoc) {
4540   return OMPTaskyieldDirective::Create(Context, StartLoc, EndLoc);
4541 }
4542 
ActOnOpenMPBarrierDirective(SourceLocation StartLoc,SourceLocation EndLoc)4543 StmtResult Sema::ActOnOpenMPBarrierDirective(SourceLocation StartLoc,
4544                                              SourceLocation EndLoc) {
4545   return OMPBarrierDirective::Create(Context, StartLoc, EndLoc);
4546 }
4547 
ActOnOpenMPTaskwaitDirective(SourceLocation StartLoc,SourceLocation EndLoc)4548 StmtResult Sema::ActOnOpenMPTaskwaitDirective(SourceLocation StartLoc,
4549                                               SourceLocation EndLoc) {
4550   return OMPTaskwaitDirective::Create(Context, StartLoc, EndLoc);
4551 }
4552 
ActOnOpenMPTaskgroupDirective(Stmt * AStmt,SourceLocation StartLoc,SourceLocation EndLoc)4553 StmtResult Sema::ActOnOpenMPTaskgroupDirective(Stmt *AStmt,
4554                                                SourceLocation StartLoc,
4555                                                SourceLocation EndLoc) {
4556   if (!AStmt)
4557     return StmtError();
4558 
4559   assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
4560 
4561   getCurFunction()->setHasBranchProtectedScope();
4562 
4563   return OMPTaskgroupDirective::Create(Context, StartLoc, EndLoc, AStmt);
4564 }
4565 
ActOnOpenMPFlushDirective(ArrayRef<OMPClause * > Clauses,SourceLocation StartLoc,SourceLocation EndLoc)4566 StmtResult Sema::ActOnOpenMPFlushDirective(ArrayRef<OMPClause *> Clauses,
4567                                            SourceLocation StartLoc,
4568                                            SourceLocation EndLoc) {
4569   assert(Clauses.size() <= 1 && "Extra clauses in flush directive");
4570   return OMPFlushDirective::Create(Context, StartLoc, EndLoc, Clauses);
4571 }
4572 
ActOnOpenMPOrderedDirective(ArrayRef<OMPClause * > Clauses,Stmt * AStmt,SourceLocation StartLoc,SourceLocation EndLoc)4573 StmtResult Sema::ActOnOpenMPOrderedDirective(ArrayRef<OMPClause *> Clauses,
4574                                              Stmt *AStmt,
4575                                              SourceLocation StartLoc,
4576                                              SourceLocation EndLoc) {
4577   OMPClause *DependFound = nullptr;
4578   OMPClause *DependSourceClause = nullptr;
4579   bool ErrorFound = false;
4580   OMPThreadsClause *TC = nullptr;
4581   OMPSIMDClause *SC = nullptr;
4582   for (auto *C : Clauses) {
4583     if (auto *DC = dyn_cast<OMPDependClause>(C)) {
4584       DependFound = C;
4585       if (DC->getDependencyKind() == OMPC_DEPEND_source) {
4586         if (DependSourceClause) {
4587           Diag(C->getLocStart(), diag::err_omp_more_one_clause)
4588               << getOpenMPDirectiveName(OMPD_ordered)
4589               << getOpenMPClauseName(OMPC_depend) << 2;
4590           ErrorFound = true;
4591         } else
4592           DependSourceClause = C;
4593       }
4594     } else if (C->getClauseKind() == OMPC_threads)
4595       TC = cast<OMPThreadsClause>(C);
4596     else if (C->getClauseKind() == OMPC_simd)
4597       SC = cast<OMPSIMDClause>(C);
4598   }
4599   if (!ErrorFound && !SC &&
4600       isOpenMPSimdDirective(DSAStack->getParentDirective())) {
4601     // OpenMP [2.8.1,simd Construct, Restrictions]
4602     // An ordered construct with the simd clause is the only OpenMP construct
4603     // that can appear in the simd region.
4604     Diag(StartLoc, diag::err_omp_prohibited_region_simd);
4605     ErrorFound = true;
4606   } else if (DependFound && (TC || SC)) {
4607     Diag(DependFound->getLocStart(), diag::err_omp_depend_clause_thread_simd)
4608         << getOpenMPClauseName(TC ? TC->getClauseKind() : SC->getClauseKind());
4609     ErrorFound = true;
4610   } else if (DependFound && !DSAStack->getParentOrderedRegionParam()) {
4611     Diag(DependFound->getLocStart(),
4612          diag::err_omp_ordered_directive_without_param);
4613     ErrorFound = true;
4614   } else if (TC || Clauses.empty()) {
4615     if (auto *Param = DSAStack->getParentOrderedRegionParam()) {
4616       SourceLocation ErrLoc = TC ? TC->getLocStart() : StartLoc;
4617       Diag(ErrLoc, diag::err_omp_ordered_directive_with_param)
4618           << (TC != nullptr);
4619       Diag(Param->getLocStart(), diag::note_omp_ordered_param);
4620       ErrorFound = true;
4621     }
4622   }
4623   if ((!AStmt && !DependFound) || ErrorFound)
4624     return StmtError();
4625 
4626   if (AStmt) {
4627     assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
4628 
4629     getCurFunction()->setHasBranchProtectedScope();
4630   }
4631 
4632   return OMPOrderedDirective::Create(Context, StartLoc, EndLoc, Clauses, AStmt);
4633 }
4634 
4635 namespace {
4636 /// \brief Helper class for checking expression in 'omp atomic [update]'
4637 /// construct.
4638 class OpenMPAtomicUpdateChecker {
4639   /// \brief Error results for atomic update expressions.
4640   enum ExprAnalysisErrorCode {
4641     /// \brief A statement is not an expression statement.
4642     NotAnExpression,
4643     /// \brief Expression is not builtin binary or unary operation.
4644     NotABinaryOrUnaryExpression,
4645     /// \brief Unary operation is not post-/pre- increment/decrement operation.
4646     NotAnUnaryIncDecExpression,
4647     /// \brief An expression is not of scalar type.
4648     NotAScalarType,
4649     /// \brief A binary operation is not an assignment operation.
4650     NotAnAssignmentOp,
4651     /// \brief RHS part of the binary operation is not a binary expression.
4652     NotABinaryExpression,
4653     /// \brief RHS part is not additive/multiplicative/shift/biwise binary
4654     /// expression.
4655     NotABinaryOperator,
4656     /// \brief RHS binary operation does not have reference to the updated LHS
4657     /// part.
4658     NotAnUpdateExpression,
4659     /// \brief No errors is found.
4660     NoError
4661   };
4662   /// \brief Reference to Sema.
4663   Sema &SemaRef;
4664   /// \brief A location for note diagnostics (when error is found).
4665   SourceLocation NoteLoc;
4666   /// \brief 'x' lvalue part of the source atomic expression.
4667   Expr *X;
4668   /// \brief 'expr' rvalue part of the source atomic expression.
4669   Expr *E;
4670   /// \brief Helper expression of the form
4671   /// 'OpaqueValueExpr(x) binop OpaqueValueExpr(expr)' or
4672   /// 'OpaqueValueExpr(expr) binop OpaqueValueExpr(x)'.
4673   Expr *UpdateExpr;
4674   /// \brief Is 'x' a LHS in a RHS part of full update expression. It is
4675   /// important for non-associative operations.
4676   bool IsXLHSInRHSPart;
4677   BinaryOperatorKind Op;
4678   SourceLocation OpLoc;
4679   /// \brief true if the source expression is a postfix unary operation, false
4680   /// if it is a prefix unary operation.
4681   bool IsPostfixUpdate;
4682 
4683 public:
OpenMPAtomicUpdateChecker(Sema & SemaRef)4684   OpenMPAtomicUpdateChecker(Sema &SemaRef)
4685       : SemaRef(SemaRef), X(nullptr), E(nullptr), UpdateExpr(nullptr),
4686         IsXLHSInRHSPart(false), Op(BO_PtrMemD), IsPostfixUpdate(false) {}
4687   /// \brief Check specified statement that it is suitable for 'atomic update'
4688   /// constructs and extract 'x', 'expr' and Operation from the original
4689   /// expression. If DiagId and NoteId == 0, then only check is performed
4690   /// without error notification.
4691   /// \param DiagId Diagnostic which should be emitted if error is found.
4692   /// \param NoteId Diagnostic note for the main error message.
4693   /// \return true if statement is not an update expression, false otherwise.
4694   bool checkStatement(Stmt *S, unsigned DiagId = 0, unsigned NoteId = 0);
4695   /// \brief Return the 'x' lvalue part of the source atomic expression.
getX() const4696   Expr *getX() const { return X; }
4697   /// \brief Return the 'expr' rvalue part of the source atomic expression.
getExpr() const4698   Expr *getExpr() const { return E; }
4699   /// \brief Return the update expression used in calculation of the updated
4700   /// value. Always has form 'OpaqueValueExpr(x) binop OpaqueValueExpr(expr)' or
4701   /// 'OpaqueValueExpr(expr) binop OpaqueValueExpr(x)'.
getUpdateExpr() const4702   Expr *getUpdateExpr() const { return UpdateExpr; }
4703   /// \brief Return true if 'x' is LHS in RHS part of full update expression,
4704   /// false otherwise.
isXLHSInRHSPart() const4705   bool isXLHSInRHSPart() const { return IsXLHSInRHSPart; }
4706 
4707   /// \brief true if the source expression is a postfix unary operation, false
4708   /// if it is a prefix unary operation.
isPostfixUpdate() const4709   bool isPostfixUpdate() const { return IsPostfixUpdate; }
4710 
4711 private:
4712   bool checkBinaryOperation(BinaryOperator *AtomicBinOp, unsigned DiagId = 0,
4713                             unsigned NoteId = 0);
4714 };
4715 } // namespace
4716 
checkBinaryOperation(BinaryOperator * AtomicBinOp,unsigned DiagId,unsigned NoteId)4717 bool OpenMPAtomicUpdateChecker::checkBinaryOperation(
4718     BinaryOperator *AtomicBinOp, unsigned DiagId, unsigned NoteId) {
4719   ExprAnalysisErrorCode ErrorFound = NoError;
4720   SourceLocation ErrorLoc, NoteLoc;
4721   SourceRange ErrorRange, NoteRange;
4722   // Allowed constructs are:
4723   //  x = x binop expr;
4724   //  x = expr binop x;
4725   if (AtomicBinOp->getOpcode() == BO_Assign) {
4726     X = AtomicBinOp->getLHS();
4727     if (auto *AtomicInnerBinOp = dyn_cast<BinaryOperator>(
4728             AtomicBinOp->getRHS()->IgnoreParenImpCasts())) {
4729       if (AtomicInnerBinOp->isMultiplicativeOp() ||
4730           AtomicInnerBinOp->isAdditiveOp() || AtomicInnerBinOp->isShiftOp() ||
4731           AtomicInnerBinOp->isBitwiseOp()) {
4732         Op = AtomicInnerBinOp->getOpcode();
4733         OpLoc = AtomicInnerBinOp->getOperatorLoc();
4734         auto *LHS = AtomicInnerBinOp->getLHS();
4735         auto *RHS = AtomicInnerBinOp->getRHS();
4736         llvm::FoldingSetNodeID XId, LHSId, RHSId;
4737         X->IgnoreParenImpCasts()->Profile(XId, SemaRef.getASTContext(),
4738                                           /*Canonical=*/true);
4739         LHS->IgnoreParenImpCasts()->Profile(LHSId, SemaRef.getASTContext(),
4740                                             /*Canonical=*/true);
4741         RHS->IgnoreParenImpCasts()->Profile(RHSId, SemaRef.getASTContext(),
4742                                             /*Canonical=*/true);
4743         if (XId == LHSId) {
4744           E = RHS;
4745           IsXLHSInRHSPart = true;
4746         } else if (XId == RHSId) {
4747           E = LHS;
4748           IsXLHSInRHSPart = false;
4749         } else {
4750           ErrorLoc = AtomicInnerBinOp->getExprLoc();
4751           ErrorRange = AtomicInnerBinOp->getSourceRange();
4752           NoteLoc = X->getExprLoc();
4753           NoteRange = X->getSourceRange();
4754           ErrorFound = NotAnUpdateExpression;
4755         }
4756       } else {
4757         ErrorLoc = AtomicInnerBinOp->getExprLoc();
4758         ErrorRange = AtomicInnerBinOp->getSourceRange();
4759         NoteLoc = AtomicInnerBinOp->getOperatorLoc();
4760         NoteRange = SourceRange(NoteLoc, NoteLoc);
4761         ErrorFound = NotABinaryOperator;
4762       }
4763     } else {
4764       NoteLoc = ErrorLoc = AtomicBinOp->getRHS()->getExprLoc();
4765       NoteRange = ErrorRange = AtomicBinOp->getRHS()->getSourceRange();
4766       ErrorFound = NotABinaryExpression;
4767     }
4768   } else {
4769     ErrorLoc = AtomicBinOp->getExprLoc();
4770     ErrorRange = AtomicBinOp->getSourceRange();
4771     NoteLoc = AtomicBinOp->getOperatorLoc();
4772     NoteRange = SourceRange(NoteLoc, NoteLoc);
4773     ErrorFound = NotAnAssignmentOp;
4774   }
4775   if (ErrorFound != NoError && DiagId != 0 && NoteId != 0) {
4776     SemaRef.Diag(ErrorLoc, DiagId) << ErrorRange;
4777     SemaRef.Diag(NoteLoc, NoteId) << ErrorFound << NoteRange;
4778     return true;
4779   } else if (SemaRef.CurContext->isDependentContext())
4780     E = X = UpdateExpr = nullptr;
4781   return ErrorFound != NoError;
4782 }
4783 
checkStatement(Stmt * S,unsigned DiagId,unsigned NoteId)4784 bool OpenMPAtomicUpdateChecker::checkStatement(Stmt *S, unsigned DiagId,
4785                                                unsigned NoteId) {
4786   ExprAnalysisErrorCode ErrorFound = NoError;
4787   SourceLocation ErrorLoc, NoteLoc;
4788   SourceRange ErrorRange, NoteRange;
4789   // Allowed constructs are:
4790   //  x++;
4791   //  x--;
4792   //  ++x;
4793   //  --x;
4794   //  x binop= expr;
4795   //  x = x binop expr;
4796   //  x = expr binop x;
4797   if (auto *AtomicBody = dyn_cast<Expr>(S)) {
4798     AtomicBody = AtomicBody->IgnoreParenImpCasts();
4799     if (AtomicBody->getType()->isScalarType() ||
4800         AtomicBody->isInstantiationDependent()) {
4801       if (auto *AtomicCompAssignOp = dyn_cast<CompoundAssignOperator>(
4802               AtomicBody->IgnoreParenImpCasts())) {
4803         // Check for Compound Assignment Operation
4804         Op = BinaryOperator::getOpForCompoundAssignment(
4805             AtomicCompAssignOp->getOpcode());
4806         OpLoc = AtomicCompAssignOp->getOperatorLoc();
4807         E = AtomicCompAssignOp->getRHS();
4808         X = AtomicCompAssignOp->getLHS();
4809         IsXLHSInRHSPart = true;
4810       } else if (auto *AtomicBinOp = dyn_cast<BinaryOperator>(
4811                      AtomicBody->IgnoreParenImpCasts())) {
4812         // Check for Binary Operation
4813         if(checkBinaryOperation(AtomicBinOp, DiagId, NoteId))
4814           return true;
4815       } else if (auto *AtomicUnaryOp =
4816                  dyn_cast<UnaryOperator>(AtomicBody->IgnoreParenImpCasts())) {
4817         // Check for Unary Operation
4818         if (AtomicUnaryOp->isIncrementDecrementOp()) {
4819           IsPostfixUpdate = AtomicUnaryOp->isPostfix();
4820           Op = AtomicUnaryOp->isIncrementOp() ? BO_Add : BO_Sub;
4821           OpLoc = AtomicUnaryOp->getOperatorLoc();
4822           X = AtomicUnaryOp->getSubExpr();
4823           E = SemaRef.ActOnIntegerConstant(OpLoc, /*uint64_t Val=*/1).get();
4824           IsXLHSInRHSPart = true;
4825         } else {
4826           ErrorFound = NotAnUnaryIncDecExpression;
4827           ErrorLoc = AtomicUnaryOp->getExprLoc();
4828           ErrorRange = AtomicUnaryOp->getSourceRange();
4829           NoteLoc = AtomicUnaryOp->getOperatorLoc();
4830           NoteRange = SourceRange(NoteLoc, NoteLoc);
4831         }
4832       } else if (!AtomicBody->isInstantiationDependent()) {
4833         ErrorFound = NotABinaryOrUnaryExpression;
4834         NoteLoc = ErrorLoc = AtomicBody->getExprLoc();
4835         NoteRange = ErrorRange = AtomicBody->getSourceRange();
4836       }
4837     } else {
4838       ErrorFound = NotAScalarType;
4839       NoteLoc = ErrorLoc = AtomicBody->getLocStart();
4840       NoteRange = ErrorRange = SourceRange(NoteLoc, NoteLoc);
4841     }
4842   } else {
4843     ErrorFound = NotAnExpression;
4844     NoteLoc = ErrorLoc = S->getLocStart();
4845     NoteRange = ErrorRange = SourceRange(NoteLoc, NoteLoc);
4846   }
4847   if (ErrorFound != NoError && DiagId != 0 && NoteId != 0) {
4848     SemaRef.Diag(ErrorLoc, DiagId) << ErrorRange;
4849     SemaRef.Diag(NoteLoc, NoteId) << ErrorFound << NoteRange;
4850     return true;
4851   } else if (SemaRef.CurContext->isDependentContext())
4852     E = X = UpdateExpr = nullptr;
4853   if (ErrorFound == NoError && E && X) {
4854     // Build an update expression of form 'OpaqueValueExpr(x) binop
4855     // OpaqueValueExpr(expr)' or 'OpaqueValueExpr(expr) binop
4856     // OpaqueValueExpr(x)' and then cast it to the type of the 'x' expression.
4857     auto *OVEX = new (SemaRef.getASTContext())
4858         OpaqueValueExpr(X->getExprLoc(), X->getType(), VK_RValue);
4859     auto *OVEExpr = new (SemaRef.getASTContext())
4860         OpaqueValueExpr(E->getExprLoc(), E->getType(), VK_RValue);
4861     auto Update =
4862         SemaRef.CreateBuiltinBinOp(OpLoc, Op, IsXLHSInRHSPart ? OVEX : OVEExpr,
4863                                    IsXLHSInRHSPart ? OVEExpr : OVEX);
4864     if (Update.isInvalid())
4865       return true;
4866     Update = SemaRef.PerformImplicitConversion(Update.get(), X->getType(),
4867                                                Sema::AA_Casting);
4868     if (Update.isInvalid())
4869       return true;
4870     UpdateExpr = Update.get();
4871   }
4872   return ErrorFound != NoError;
4873 }
4874 
ActOnOpenMPAtomicDirective(ArrayRef<OMPClause * > Clauses,Stmt * AStmt,SourceLocation StartLoc,SourceLocation EndLoc)4875 StmtResult Sema::ActOnOpenMPAtomicDirective(ArrayRef<OMPClause *> Clauses,
4876                                             Stmt *AStmt,
4877                                             SourceLocation StartLoc,
4878                                             SourceLocation EndLoc) {
4879   if (!AStmt)
4880     return StmtError();
4881 
4882   auto CS = cast<CapturedStmt>(AStmt);
4883   // 1.2.2 OpenMP Language Terminology
4884   // Structured block - An executable statement with a single entry at the
4885   // top and a single exit at the bottom.
4886   // The point of exit cannot be a branch out of the structured block.
4887   // longjmp() and throw() must not violate the entry/exit criteria.
4888   OpenMPClauseKind AtomicKind = OMPC_unknown;
4889   SourceLocation AtomicKindLoc;
4890   for (auto *C : Clauses) {
4891     if (C->getClauseKind() == OMPC_read || C->getClauseKind() == OMPC_write ||
4892         C->getClauseKind() == OMPC_update ||
4893         C->getClauseKind() == OMPC_capture) {
4894       if (AtomicKind != OMPC_unknown) {
4895         Diag(C->getLocStart(), diag::err_omp_atomic_several_clauses)
4896             << SourceRange(C->getLocStart(), C->getLocEnd());
4897         Diag(AtomicKindLoc, diag::note_omp_atomic_previous_clause)
4898             << getOpenMPClauseName(AtomicKind);
4899       } else {
4900         AtomicKind = C->getClauseKind();
4901         AtomicKindLoc = C->getLocStart();
4902       }
4903     }
4904   }
4905 
4906   auto Body = CS->getCapturedStmt();
4907   if (auto *EWC = dyn_cast<ExprWithCleanups>(Body))
4908     Body = EWC->getSubExpr();
4909 
4910   Expr *X = nullptr;
4911   Expr *V = nullptr;
4912   Expr *E = nullptr;
4913   Expr *UE = nullptr;
4914   bool IsXLHSInRHSPart = false;
4915   bool IsPostfixUpdate = false;
4916   // OpenMP [2.12.6, atomic Construct]
4917   // In the next expressions:
4918   // * x and v (as applicable) are both l-value expressions with scalar type.
4919   // * During the execution of an atomic region, multiple syntactic
4920   // occurrences of x must designate the same storage location.
4921   // * Neither of v and expr (as applicable) may access the storage location
4922   // designated by x.
4923   // * Neither of x and expr (as applicable) may access the storage location
4924   // designated by v.
4925   // * expr is an expression with scalar type.
4926   // * binop is one of +, *, -, /, &, ^, |, <<, or >>.
4927   // * binop, binop=, ++, and -- are not overloaded operators.
4928   // * The expression x binop expr must be numerically equivalent to x binop
4929   // (expr). This requirement is satisfied if the operators in expr have
4930   // precedence greater than binop, or by using parentheses around expr or
4931   // subexpressions of expr.
4932   // * The expression expr binop x must be numerically equivalent to (expr)
4933   // binop x. This requirement is satisfied if the operators in expr have
4934   // precedence equal to or greater than binop, or by using parentheses around
4935   // expr or subexpressions of expr.
4936   // * For forms that allow multiple occurrences of x, the number of times
4937   // that x is evaluated is unspecified.
4938   if (AtomicKind == OMPC_read) {
4939     enum {
4940       NotAnExpression,
4941       NotAnAssignmentOp,
4942       NotAScalarType,
4943       NotAnLValue,
4944       NoError
4945     } ErrorFound = NoError;
4946     SourceLocation ErrorLoc, NoteLoc;
4947     SourceRange ErrorRange, NoteRange;
4948     // If clause is read:
4949     //  v = x;
4950     if (auto AtomicBody = dyn_cast<Expr>(Body)) {
4951       auto AtomicBinOp =
4952           dyn_cast<BinaryOperator>(AtomicBody->IgnoreParenImpCasts());
4953       if (AtomicBinOp && AtomicBinOp->getOpcode() == BO_Assign) {
4954         X = AtomicBinOp->getRHS()->IgnoreParenImpCasts();
4955         V = AtomicBinOp->getLHS()->IgnoreParenImpCasts();
4956         if ((X->isInstantiationDependent() || X->getType()->isScalarType()) &&
4957             (V->isInstantiationDependent() || V->getType()->isScalarType())) {
4958           if (!X->isLValue() || !V->isLValue()) {
4959             auto NotLValueExpr = X->isLValue() ? V : X;
4960             ErrorFound = NotAnLValue;
4961             ErrorLoc = AtomicBinOp->getExprLoc();
4962             ErrorRange = AtomicBinOp->getSourceRange();
4963             NoteLoc = NotLValueExpr->getExprLoc();
4964             NoteRange = NotLValueExpr->getSourceRange();
4965           }
4966         } else if (!X->isInstantiationDependent() ||
4967                    !V->isInstantiationDependent()) {
4968           auto NotScalarExpr =
4969               (X->isInstantiationDependent() || X->getType()->isScalarType())
4970                   ? V
4971                   : X;
4972           ErrorFound = NotAScalarType;
4973           ErrorLoc = AtomicBinOp->getExprLoc();
4974           ErrorRange = AtomicBinOp->getSourceRange();
4975           NoteLoc = NotScalarExpr->getExprLoc();
4976           NoteRange = NotScalarExpr->getSourceRange();
4977         }
4978       } else if (!AtomicBody->isInstantiationDependent()) {
4979         ErrorFound = NotAnAssignmentOp;
4980         ErrorLoc = AtomicBody->getExprLoc();
4981         ErrorRange = AtomicBody->getSourceRange();
4982         NoteLoc = AtomicBinOp ? AtomicBinOp->getOperatorLoc()
4983                               : AtomicBody->getExprLoc();
4984         NoteRange = AtomicBinOp ? AtomicBinOp->getSourceRange()
4985                                 : AtomicBody->getSourceRange();
4986       }
4987     } else {
4988       ErrorFound = NotAnExpression;
4989       NoteLoc = ErrorLoc = Body->getLocStart();
4990       NoteRange = ErrorRange = SourceRange(NoteLoc, NoteLoc);
4991     }
4992     if (ErrorFound != NoError) {
4993       Diag(ErrorLoc, diag::err_omp_atomic_read_not_expression_statement)
4994           << ErrorRange;
4995       Diag(NoteLoc, diag::note_omp_atomic_read_write) << ErrorFound
4996                                                       << NoteRange;
4997       return StmtError();
4998     } else if (CurContext->isDependentContext())
4999       V = X = nullptr;
5000   } else if (AtomicKind == OMPC_write) {
5001     enum {
5002       NotAnExpression,
5003       NotAnAssignmentOp,
5004       NotAScalarType,
5005       NotAnLValue,
5006       NoError
5007     } ErrorFound = NoError;
5008     SourceLocation ErrorLoc, NoteLoc;
5009     SourceRange ErrorRange, NoteRange;
5010     // If clause is write:
5011     //  x = expr;
5012     if (auto AtomicBody = dyn_cast<Expr>(Body)) {
5013       auto AtomicBinOp =
5014           dyn_cast<BinaryOperator>(AtomicBody->IgnoreParenImpCasts());
5015       if (AtomicBinOp && AtomicBinOp->getOpcode() == BO_Assign) {
5016         X = AtomicBinOp->getLHS();
5017         E = AtomicBinOp->getRHS();
5018         if ((X->isInstantiationDependent() || X->getType()->isScalarType()) &&
5019             (E->isInstantiationDependent() || E->getType()->isScalarType())) {
5020           if (!X->isLValue()) {
5021             ErrorFound = NotAnLValue;
5022             ErrorLoc = AtomicBinOp->getExprLoc();
5023             ErrorRange = AtomicBinOp->getSourceRange();
5024             NoteLoc = X->getExprLoc();
5025             NoteRange = X->getSourceRange();
5026           }
5027         } else if (!X->isInstantiationDependent() ||
5028                    !E->isInstantiationDependent()) {
5029           auto NotScalarExpr =
5030               (X->isInstantiationDependent() || X->getType()->isScalarType())
5031                   ? E
5032                   : X;
5033           ErrorFound = NotAScalarType;
5034           ErrorLoc = AtomicBinOp->getExprLoc();
5035           ErrorRange = AtomicBinOp->getSourceRange();
5036           NoteLoc = NotScalarExpr->getExprLoc();
5037           NoteRange = NotScalarExpr->getSourceRange();
5038         }
5039       } else if (!AtomicBody->isInstantiationDependent()) {
5040         ErrorFound = NotAnAssignmentOp;
5041         ErrorLoc = AtomicBody->getExprLoc();
5042         ErrorRange = AtomicBody->getSourceRange();
5043         NoteLoc = AtomicBinOp ? AtomicBinOp->getOperatorLoc()
5044                               : AtomicBody->getExprLoc();
5045         NoteRange = AtomicBinOp ? AtomicBinOp->getSourceRange()
5046                                 : AtomicBody->getSourceRange();
5047       }
5048     } else {
5049       ErrorFound = NotAnExpression;
5050       NoteLoc = ErrorLoc = Body->getLocStart();
5051       NoteRange = ErrorRange = SourceRange(NoteLoc, NoteLoc);
5052     }
5053     if (ErrorFound != NoError) {
5054       Diag(ErrorLoc, diag::err_omp_atomic_write_not_expression_statement)
5055           << ErrorRange;
5056       Diag(NoteLoc, diag::note_omp_atomic_read_write) << ErrorFound
5057                                                       << NoteRange;
5058       return StmtError();
5059     } else if (CurContext->isDependentContext())
5060       E = X = nullptr;
5061   } else if (AtomicKind == OMPC_update || AtomicKind == OMPC_unknown) {
5062     // If clause is update:
5063     //  x++;
5064     //  x--;
5065     //  ++x;
5066     //  --x;
5067     //  x binop= expr;
5068     //  x = x binop expr;
5069     //  x = expr binop x;
5070     OpenMPAtomicUpdateChecker Checker(*this);
5071     if (Checker.checkStatement(
5072             Body, (AtomicKind == OMPC_update)
5073                       ? diag::err_omp_atomic_update_not_expression_statement
5074                       : diag::err_omp_atomic_not_expression_statement,
5075             diag::note_omp_atomic_update))
5076       return StmtError();
5077     if (!CurContext->isDependentContext()) {
5078       E = Checker.getExpr();
5079       X = Checker.getX();
5080       UE = Checker.getUpdateExpr();
5081       IsXLHSInRHSPart = Checker.isXLHSInRHSPart();
5082     }
5083   } else if (AtomicKind == OMPC_capture) {
5084     enum {
5085       NotAnAssignmentOp,
5086       NotACompoundStatement,
5087       NotTwoSubstatements,
5088       NotASpecificExpression,
5089       NoError
5090     } ErrorFound = NoError;
5091     SourceLocation ErrorLoc, NoteLoc;
5092     SourceRange ErrorRange, NoteRange;
5093     if (auto *AtomicBody = dyn_cast<Expr>(Body)) {
5094       // If clause is a capture:
5095       //  v = x++;
5096       //  v = x--;
5097       //  v = ++x;
5098       //  v = --x;
5099       //  v = x binop= expr;
5100       //  v = x = x binop expr;
5101       //  v = x = expr binop x;
5102       auto *AtomicBinOp =
5103           dyn_cast<BinaryOperator>(AtomicBody->IgnoreParenImpCasts());
5104       if (AtomicBinOp && AtomicBinOp->getOpcode() == BO_Assign) {
5105         V = AtomicBinOp->getLHS();
5106         Body = AtomicBinOp->getRHS()->IgnoreParenImpCasts();
5107         OpenMPAtomicUpdateChecker Checker(*this);
5108         if (Checker.checkStatement(
5109                 Body, diag::err_omp_atomic_capture_not_expression_statement,
5110                 diag::note_omp_atomic_update))
5111           return StmtError();
5112         E = Checker.getExpr();
5113         X = Checker.getX();
5114         UE = Checker.getUpdateExpr();
5115         IsXLHSInRHSPart = Checker.isXLHSInRHSPart();
5116         IsPostfixUpdate = Checker.isPostfixUpdate();
5117       } else if (!AtomicBody->isInstantiationDependent()) {
5118         ErrorLoc = AtomicBody->getExprLoc();
5119         ErrorRange = AtomicBody->getSourceRange();
5120         NoteLoc = AtomicBinOp ? AtomicBinOp->getOperatorLoc()
5121                               : AtomicBody->getExprLoc();
5122         NoteRange = AtomicBinOp ? AtomicBinOp->getSourceRange()
5123                                 : AtomicBody->getSourceRange();
5124         ErrorFound = NotAnAssignmentOp;
5125       }
5126       if (ErrorFound != NoError) {
5127         Diag(ErrorLoc, diag::err_omp_atomic_capture_not_expression_statement)
5128             << ErrorRange;
5129         Diag(NoteLoc, diag::note_omp_atomic_capture) << ErrorFound << NoteRange;
5130         return StmtError();
5131       } else if (CurContext->isDependentContext()) {
5132         UE = V = E = X = nullptr;
5133       }
5134     } else {
5135       // If clause is a capture:
5136       //  { v = x; x = expr; }
5137       //  { v = x; x++; }
5138       //  { v = x; x--; }
5139       //  { v = x; ++x; }
5140       //  { v = x; --x; }
5141       //  { v = x; x binop= expr; }
5142       //  { v = x; x = x binop expr; }
5143       //  { v = x; x = expr binop x; }
5144       //  { x++; v = x; }
5145       //  { x--; v = x; }
5146       //  { ++x; v = x; }
5147       //  { --x; v = x; }
5148       //  { x binop= expr; v = x; }
5149       //  { x = x binop expr; v = x; }
5150       //  { x = expr binop x; v = x; }
5151       if (auto *CS = dyn_cast<CompoundStmt>(Body)) {
5152         // Check that this is { expr1; expr2; }
5153         if (CS->size() == 2) {
5154           auto *First = CS->body_front();
5155           auto *Second = CS->body_back();
5156           if (auto *EWC = dyn_cast<ExprWithCleanups>(First))
5157             First = EWC->getSubExpr()->IgnoreParenImpCasts();
5158           if (auto *EWC = dyn_cast<ExprWithCleanups>(Second))
5159             Second = EWC->getSubExpr()->IgnoreParenImpCasts();
5160           // Need to find what subexpression is 'v' and what is 'x'.
5161           OpenMPAtomicUpdateChecker Checker(*this);
5162           bool IsUpdateExprFound = !Checker.checkStatement(Second);
5163           BinaryOperator *BinOp = nullptr;
5164           if (IsUpdateExprFound) {
5165             BinOp = dyn_cast<BinaryOperator>(First);
5166             IsUpdateExprFound = BinOp && BinOp->getOpcode() == BO_Assign;
5167           }
5168           if (IsUpdateExprFound && !CurContext->isDependentContext()) {
5169             //  { v = x; x++; }
5170             //  { v = x; x--; }
5171             //  { v = x; ++x; }
5172             //  { v = x; --x; }
5173             //  { v = x; x binop= expr; }
5174             //  { v = x; x = x binop expr; }
5175             //  { v = x; x = expr binop x; }
5176             // Check that the first expression has form v = x.
5177             auto *PossibleX = BinOp->getRHS()->IgnoreParenImpCasts();
5178             llvm::FoldingSetNodeID XId, PossibleXId;
5179             Checker.getX()->Profile(XId, Context, /*Canonical=*/true);
5180             PossibleX->Profile(PossibleXId, Context, /*Canonical=*/true);
5181             IsUpdateExprFound = XId == PossibleXId;
5182             if (IsUpdateExprFound) {
5183               V = BinOp->getLHS();
5184               X = Checker.getX();
5185               E = Checker.getExpr();
5186               UE = Checker.getUpdateExpr();
5187               IsXLHSInRHSPart = Checker.isXLHSInRHSPart();
5188               IsPostfixUpdate = true;
5189             }
5190           }
5191           if (!IsUpdateExprFound) {
5192             IsUpdateExprFound = !Checker.checkStatement(First);
5193             BinOp = nullptr;
5194             if (IsUpdateExprFound) {
5195               BinOp = dyn_cast<BinaryOperator>(Second);
5196               IsUpdateExprFound = BinOp && BinOp->getOpcode() == BO_Assign;
5197             }
5198             if (IsUpdateExprFound && !CurContext->isDependentContext()) {
5199               //  { x++; v = x; }
5200               //  { x--; v = x; }
5201               //  { ++x; v = x; }
5202               //  { --x; v = x; }
5203               //  { x binop= expr; v = x; }
5204               //  { x = x binop expr; v = x; }
5205               //  { x = expr binop x; v = x; }
5206               // Check that the second expression has form v = x.
5207               auto *PossibleX = BinOp->getRHS()->IgnoreParenImpCasts();
5208               llvm::FoldingSetNodeID XId, PossibleXId;
5209               Checker.getX()->Profile(XId, Context, /*Canonical=*/true);
5210               PossibleX->Profile(PossibleXId, Context, /*Canonical=*/true);
5211               IsUpdateExprFound = XId == PossibleXId;
5212               if (IsUpdateExprFound) {
5213                 V = BinOp->getLHS();
5214                 X = Checker.getX();
5215                 E = Checker.getExpr();
5216                 UE = Checker.getUpdateExpr();
5217                 IsXLHSInRHSPart = Checker.isXLHSInRHSPart();
5218                 IsPostfixUpdate = false;
5219               }
5220             }
5221           }
5222           if (!IsUpdateExprFound) {
5223             //  { v = x; x = expr; }
5224             auto *FirstExpr = dyn_cast<Expr>(First);
5225             auto *SecondExpr = dyn_cast<Expr>(Second);
5226             if (!FirstExpr || !SecondExpr ||
5227                 !(FirstExpr->isInstantiationDependent() ||
5228                   SecondExpr->isInstantiationDependent())) {
5229               auto *FirstBinOp = dyn_cast<BinaryOperator>(First);
5230               if (!FirstBinOp || FirstBinOp->getOpcode() != BO_Assign) {
5231                 ErrorFound = NotAnAssignmentOp;
5232                 NoteLoc = ErrorLoc = FirstBinOp ? FirstBinOp->getOperatorLoc()
5233                                                 : First->getLocStart();
5234                 NoteRange = ErrorRange = FirstBinOp
5235                                              ? FirstBinOp->getSourceRange()
5236                                              : SourceRange(ErrorLoc, ErrorLoc);
5237               } else {
5238                 auto *SecondBinOp = dyn_cast<BinaryOperator>(Second);
5239                 if (!SecondBinOp || SecondBinOp->getOpcode() != BO_Assign) {
5240                   ErrorFound = NotAnAssignmentOp;
5241                   NoteLoc = ErrorLoc = SecondBinOp
5242                                            ? SecondBinOp->getOperatorLoc()
5243                                            : Second->getLocStart();
5244                   NoteRange = ErrorRange =
5245                       SecondBinOp ? SecondBinOp->getSourceRange()
5246                                   : SourceRange(ErrorLoc, ErrorLoc);
5247                 } else {
5248                   auto *PossibleXRHSInFirst =
5249                       FirstBinOp->getRHS()->IgnoreParenImpCasts();
5250                   auto *PossibleXLHSInSecond =
5251                       SecondBinOp->getLHS()->IgnoreParenImpCasts();
5252                   llvm::FoldingSetNodeID X1Id, X2Id;
5253                   PossibleXRHSInFirst->Profile(X1Id, Context,
5254                                                /*Canonical=*/true);
5255                   PossibleXLHSInSecond->Profile(X2Id, Context,
5256                                                 /*Canonical=*/true);
5257                   IsUpdateExprFound = X1Id == X2Id;
5258                   if (IsUpdateExprFound) {
5259                     V = FirstBinOp->getLHS();
5260                     X = SecondBinOp->getLHS();
5261                     E = SecondBinOp->getRHS();
5262                     UE = nullptr;
5263                     IsXLHSInRHSPart = false;
5264                     IsPostfixUpdate = true;
5265                   } else {
5266                     ErrorFound = NotASpecificExpression;
5267                     ErrorLoc = FirstBinOp->getExprLoc();
5268                     ErrorRange = FirstBinOp->getSourceRange();
5269                     NoteLoc = SecondBinOp->getLHS()->getExprLoc();
5270                     NoteRange = SecondBinOp->getRHS()->getSourceRange();
5271                   }
5272                 }
5273               }
5274             }
5275           }
5276         } else {
5277           NoteLoc = ErrorLoc = Body->getLocStart();
5278           NoteRange = ErrorRange =
5279               SourceRange(Body->getLocStart(), Body->getLocStart());
5280           ErrorFound = NotTwoSubstatements;
5281         }
5282       } else {
5283         NoteLoc = ErrorLoc = Body->getLocStart();
5284         NoteRange = ErrorRange =
5285             SourceRange(Body->getLocStart(), Body->getLocStart());
5286         ErrorFound = NotACompoundStatement;
5287       }
5288       if (ErrorFound != NoError) {
5289         Diag(ErrorLoc, diag::err_omp_atomic_capture_not_compound_statement)
5290             << ErrorRange;
5291         Diag(NoteLoc, diag::note_omp_atomic_capture) << ErrorFound << NoteRange;
5292         return StmtError();
5293       } else if (CurContext->isDependentContext()) {
5294         UE = V = E = X = nullptr;
5295       }
5296     }
5297   }
5298 
5299   getCurFunction()->setHasBranchProtectedScope();
5300 
5301   return OMPAtomicDirective::Create(Context, StartLoc, EndLoc, Clauses, AStmt,
5302                                     X, V, E, UE, IsXLHSInRHSPart,
5303                                     IsPostfixUpdate);
5304 }
5305 
ActOnOpenMPTargetDirective(ArrayRef<OMPClause * > Clauses,Stmt * AStmt,SourceLocation StartLoc,SourceLocation EndLoc)5306 StmtResult Sema::ActOnOpenMPTargetDirective(ArrayRef<OMPClause *> Clauses,
5307                                             Stmt *AStmt,
5308                                             SourceLocation StartLoc,
5309                                             SourceLocation EndLoc) {
5310   if (!AStmt)
5311     return StmtError();
5312 
5313   CapturedStmt *CS = cast<CapturedStmt>(AStmt);
5314   // 1.2.2 OpenMP Language Terminology
5315   // Structured block - An executable statement with a single entry at the
5316   // top and a single exit at the bottom.
5317   // The point of exit cannot be a branch out of the structured block.
5318   // longjmp() and throw() must not violate the entry/exit criteria.
5319   CS->getCapturedDecl()->setNothrow();
5320 
5321   // OpenMP [2.16, Nesting of Regions]
5322   // If specified, a teams construct must be contained within a target
5323   // construct. That target construct must contain no statements or directives
5324   // outside of the teams construct.
5325   if (DSAStack->hasInnerTeamsRegion()) {
5326     auto S = AStmt->IgnoreContainers(/*IgnoreCaptured*/ true);
5327     bool OMPTeamsFound = true;
5328     if (auto *CS = dyn_cast<CompoundStmt>(S)) {
5329       auto I = CS->body_begin();
5330       while (I != CS->body_end()) {
5331         auto OED = dyn_cast<OMPExecutableDirective>(*I);
5332         if (!OED || !isOpenMPTeamsDirective(OED->getDirectiveKind())) {
5333           OMPTeamsFound = false;
5334           break;
5335         }
5336         ++I;
5337       }
5338       assert(I != CS->body_end() && "Not found statement");
5339       S = *I;
5340     }
5341     if (!OMPTeamsFound) {
5342       Diag(StartLoc, diag::err_omp_target_contains_not_only_teams);
5343       Diag(DSAStack->getInnerTeamsRegionLoc(),
5344            diag::note_omp_nested_teams_construct_here);
5345       Diag(S->getLocStart(), diag::note_omp_nested_statement_here)
5346           << isa<OMPExecutableDirective>(S);
5347       return StmtError();
5348     }
5349   }
5350 
5351   getCurFunction()->setHasBranchProtectedScope();
5352 
5353   return OMPTargetDirective::Create(Context, StartLoc, EndLoc, Clauses, AStmt);
5354 }
5355 
ActOnOpenMPTargetDataDirective(ArrayRef<OMPClause * > Clauses,Stmt * AStmt,SourceLocation StartLoc,SourceLocation EndLoc)5356 StmtResult Sema::ActOnOpenMPTargetDataDirective(ArrayRef<OMPClause *> Clauses,
5357                                                 Stmt *AStmt,
5358                                                 SourceLocation StartLoc,
5359                                                 SourceLocation EndLoc) {
5360   if (!AStmt)
5361     return StmtError();
5362 
5363   assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
5364 
5365   getCurFunction()->setHasBranchProtectedScope();
5366 
5367   return OMPTargetDataDirective::Create(Context, StartLoc, EndLoc, Clauses,
5368                                         AStmt);
5369 }
5370 
ActOnOpenMPTeamsDirective(ArrayRef<OMPClause * > Clauses,Stmt * AStmt,SourceLocation StartLoc,SourceLocation EndLoc)5371 StmtResult Sema::ActOnOpenMPTeamsDirective(ArrayRef<OMPClause *> Clauses,
5372                                            Stmt *AStmt, SourceLocation StartLoc,
5373                                            SourceLocation EndLoc) {
5374   if (!AStmt)
5375     return StmtError();
5376 
5377   CapturedStmt *CS = cast<CapturedStmt>(AStmt);
5378   // 1.2.2 OpenMP Language Terminology
5379   // Structured block - An executable statement with a single entry at the
5380   // top and a single exit at the bottom.
5381   // The point of exit cannot be a branch out of the structured block.
5382   // longjmp() and throw() must not violate the entry/exit criteria.
5383   CS->getCapturedDecl()->setNothrow();
5384 
5385   getCurFunction()->setHasBranchProtectedScope();
5386 
5387   return OMPTeamsDirective::Create(Context, StartLoc, EndLoc, Clauses, AStmt);
5388 }
5389 
5390 StmtResult
ActOnOpenMPCancellationPointDirective(SourceLocation StartLoc,SourceLocation EndLoc,OpenMPDirectiveKind CancelRegion)5391 Sema::ActOnOpenMPCancellationPointDirective(SourceLocation StartLoc,
5392                                             SourceLocation EndLoc,
5393                                             OpenMPDirectiveKind CancelRegion) {
5394   if (CancelRegion != OMPD_parallel && CancelRegion != OMPD_for &&
5395       CancelRegion != OMPD_sections && CancelRegion != OMPD_taskgroup) {
5396     Diag(StartLoc, diag::err_omp_wrong_cancel_region)
5397         << getOpenMPDirectiveName(CancelRegion);
5398     return StmtError();
5399   }
5400   if (DSAStack->isParentNowaitRegion()) {
5401     Diag(StartLoc, diag::err_omp_parent_cancel_region_nowait) << 0;
5402     return StmtError();
5403   }
5404   if (DSAStack->isParentOrderedRegion()) {
5405     Diag(StartLoc, diag::err_omp_parent_cancel_region_ordered) << 0;
5406     return StmtError();
5407   }
5408   return OMPCancellationPointDirective::Create(Context, StartLoc, EndLoc,
5409                                                CancelRegion);
5410 }
5411 
ActOnOpenMPCancelDirective(ArrayRef<OMPClause * > Clauses,SourceLocation StartLoc,SourceLocation EndLoc,OpenMPDirectiveKind CancelRegion)5412 StmtResult Sema::ActOnOpenMPCancelDirective(ArrayRef<OMPClause *> Clauses,
5413                                             SourceLocation StartLoc,
5414                                             SourceLocation EndLoc,
5415                                             OpenMPDirectiveKind CancelRegion) {
5416   if (CancelRegion != OMPD_parallel && CancelRegion != OMPD_for &&
5417       CancelRegion != OMPD_sections && CancelRegion != OMPD_taskgroup) {
5418     Diag(StartLoc, diag::err_omp_wrong_cancel_region)
5419         << getOpenMPDirectiveName(CancelRegion);
5420     return StmtError();
5421   }
5422   if (DSAStack->isParentNowaitRegion()) {
5423     Diag(StartLoc, diag::err_omp_parent_cancel_region_nowait) << 1;
5424     return StmtError();
5425   }
5426   if (DSAStack->isParentOrderedRegion()) {
5427     Diag(StartLoc, diag::err_omp_parent_cancel_region_ordered) << 1;
5428     return StmtError();
5429   }
5430   DSAStack->setParentCancelRegion(/*Cancel=*/true);
5431   return OMPCancelDirective::Create(Context, StartLoc, EndLoc, Clauses,
5432                                     CancelRegion);
5433 }
5434 
checkGrainsizeNumTasksClauses(Sema & S,ArrayRef<OMPClause * > Clauses)5435 static bool checkGrainsizeNumTasksClauses(Sema &S,
5436                                           ArrayRef<OMPClause *> Clauses) {
5437   OMPClause *PrevClause = nullptr;
5438   bool ErrorFound = false;
5439   for (auto *C : Clauses) {
5440     if (C->getClauseKind() == OMPC_grainsize ||
5441         C->getClauseKind() == OMPC_num_tasks) {
5442       if (!PrevClause)
5443         PrevClause = C;
5444       else if (PrevClause->getClauseKind() != C->getClauseKind()) {
5445         S.Diag(C->getLocStart(),
5446                diag::err_omp_grainsize_num_tasks_mutually_exclusive)
5447             << getOpenMPClauseName(C->getClauseKind())
5448             << getOpenMPClauseName(PrevClause->getClauseKind());
5449         S.Diag(PrevClause->getLocStart(),
5450                diag::note_omp_previous_grainsize_num_tasks)
5451             << getOpenMPClauseName(PrevClause->getClauseKind());
5452         ErrorFound = true;
5453       }
5454     }
5455   }
5456   return ErrorFound;
5457 }
5458 
ActOnOpenMPTaskLoopDirective(ArrayRef<OMPClause * > Clauses,Stmt * AStmt,SourceLocation StartLoc,SourceLocation EndLoc,llvm::DenseMap<VarDecl *,Expr * > & VarsWithImplicitDSA)5459 StmtResult Sema::ActOnOpenMPTaskLoopDirective(
5460     ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc,
5461     SourceLocation EndLoc,
5462     llvm::DenseMap<VarDecl *, Expr *> &VarsWithImplicitDSA) {
5463   if (!AStmt)
5464     return StmtError();
5465 
5466   assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
5467   OMPLoopDirective::HelperExprs B;
5468   // In presence of clause 'collapse' or 'ordered' with number of loops, it will
5469   // define the nested loops number.
5470   unsigned NestedLoopCount =
5471       CheckOpenMPLoop(OMPD_taskloop, getCollapseNumberExpr(Clauses),
5472                       /*OrderedLoopCountExpr=*/nullptr, AStmt, *this, *DSAStack,
5473                       VarsWithImplicitDSA, B);
5474   if (NestedLoopCount == 0)
5475     return StmtError();
5476 
5477   assert((CurContext->isDependentContext() || B.builtAll()) &&
5478          "omp for loop exprs were not built");
5479 
5480   // OpenMP, [2.9.2 taskloop Construct, Restrictions]
5481   // The grainsize clause and num_tasks clause are mutually exclusive and may
5482   // not appear on the same taskloop directive.
5483   if (checkGrainsizeNumTasksClauses(*this, Clauses))
5484     return StmtError();
5485 
5486   getCurFunction()->setHasBranchProtectedScope();
5487   return OMPTaskLoopDirective::Create(Context, StartLoc, EndLoc,
5488                                       NestedLoopCount, Clauses, AStmt, B);
5489 }
5490 
ActOnOpenMPTaskLoopSimdDirective(ArrayRef<OMPClause * > Clauses,Stmt * AStmt,SourceLocation StartLoc,SourceLocation EndLoc,llvm::DenseMap<VarDecl *,Expr * > & VarsWithImplicitDSA)5491 StmtResult Sema::ActOnOpenMPTaskLoopSimdDirective(
5492     ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc,
5493     SourceLocation EndLoc,
5494     llvm::DenseMap<VarDecl *, Expr *> &VarsWithImplicitDSA) {
5495   if (!AStmt)
5496     return StmtError();
5497 
5498   assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
5499   OMPLoopDirective::HelperExprs B;
5500   // In presence of clause 'collapse' or 'ordered' with number of loops, it will
5501   // define the nested loops number.
5502   unsigned NestedLoopCount =
5503       CheckOpenMPLoop(OMPD_taskloop_simd, getCollapseNumberExpr(Clauses),
5504                       /*OrderedLoopCountExpr=*/nullptr, AStmt, *this, *DSAStack,
5505                       VarsWithImplicitDSA, B);
5506   if (NestedLoopCount == 0)
5507     return StmtError();
5508 
5509   assert((CurContext->isDependentContext() || B.builtAll()) &&
5510          "omp for loop exprs were not built");
5511 
5512   // OpenMP, [2.9.2 taskloop Construct, Restrictions]
5513   // The grainsize clause and num_tasks clause are mutually exclusive and may
5514   // not appear on the same taskloop directive.
5515   if (checkGrainsizeNumTasksClauses(*this, Clauses))
5516     return StmtError();
5517 
5518   getCurFunction()->setHasBranchProtectedScope();
5519   return OMPTaskLoopSimdDirective::Create(Context, StartLoc, EndLoc,
5520                                           NestedLoopCount, Clauses, AStmt, B);
5521 }
5522 
ActOnOpenMPDistributeDirective(ArrayRef<OMPClause * > Clauses,Stmt * AStmt,SourceLocation StartLoc,SourceLocation EndLoc,llvm::DenseMap<VarDecl *,Expr * > & VarsWithImplicitDSA)5523 StmtResult Sema::ActOnOpenMPDistributeDirective(
5524     ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc,
5525     SourceLocation EndLoc,
5526     llvm::DenseMap<VarDecl *, Expr *> &VarsWithImplicitDSA) {
5527   if (!AStmt)
5528     return StmtError();
5529 
5530   assert(isa<CapturedStmt>(AStmt) && "Captured statement expected");
5531   OMPLoopDirective::HelperExprs B;
5532   // In presence of clause 'collapse' with number of loops, it will
5533   // define the nested loops number.
5534   unsigned NestedLoopCount =
5535       CheckOpenMPLoop(OMPD_distribute, getCollapseNumberExpr(Clauses),
5536                       nullptr /*ordered not a clause on distribute*/, AStmt,
5537                       *this, *DSAStack, VarsWithImplicitDSA, B);
5538   if (NestedLoopCount == 0)
5539     return StmtError();
5540 
5541   assert((CurContext->isDependentContext() || B.builtAll()) &&
5542          "omp for loop exprs were not built");
5543 
5544   getCurFunction()->setHasBranchProtectedScope();
5545   return OMPDistributeDirective::Create(Context, StartLoc, EndLoc,
5546                                         NestedLoopCount, Clauses, AStmt, B);
5547 }
5548 
ActOnOpenMPSingleExprClause(OpenMPClauseKind Kind,Expr * Expr,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc)5549 OMPClause *Sema::ActOnOpenMPSingleExprClause(OpenMPClauseKind Kind, Expr *Expr,
5550                                              SourceLocation StartLoc,
5551                                              SourceLocation LParenLoc,
5552                                              SourceLocation EndLoc) {
5553   OMPClause *Res = nullptr;
5554   switch (Kind) {
5555   case OMPC_final:
5556     Res = ActOnOpenMPFinalClause(Expr, StartLoc, LParenLoc, EndLoc);
5557     break;
5558   case OMPC_num_threads:
5559     Res = ActOnOpenMPNumThreadsClause(Expr, StartLoc, LParenLoc, EndLoc);
5560     break;
5561   case OMPC_safelen:
5562     Res = ActOnOpenMPSafelenClause(Expr, StartLoc, LParenLoc, EndLoc);
5563     break;
5564   case OMPC_simdlen:
5565     Res = ActOnOpenMPSimdlenClause(Expr, StartLoc, LParenLoc, EndLoc);
5566     break;
5567   case OMPC_collapse:
5568     Res = ActOnOpenMPCollapseClause(Expr, StartLoc, LParenLoc, EndLoc);
5569     break;
5570   case OMPC_ordered:
5571     Res = ActOnOpenMPOrderedClause(StartLoc, EndLoc, LParenLoc, Expr);
5572     break;
5573   case OMPC_device:
5574     Res = ActOnOpenMPDeviceClause(Expr, StartLoc, LParenLoc, EndLoc);
5575     break;
5576   case OMPC_num_teams:
5577     Res = ActOnOpenMPNumTeamsClause(Expr, StartLoc, LParenLoc, EndLoc);
5578     break;
5579   case OMPC_thread_limit:
5580     Res = ActOnOpenMPThreadLimitClause(Expr, StartLoc, LParenLoc, EndLoc);
5581     break;
5582   case OMPC_priority:
5583     Res = ActOnOpenMPPriorityClause(Expr, StartLoc, LParenLoc, EndLoc);
5584     break;
5585   case OMPC_grainsize:
5586     Res = ActOnOpenMPGrainsizeClause(Expr, StartLoc, LParenLoc, EndLoc);
5587     break;
5588   case OMPC_num_tasks:
5589     Res = ActOnOpenMPNumTasksClause(Expr, StartLoc, LParenLoc, EndLoc);
5590     break;
5591   case OMPC_hint:
5592     Res = ActOnOpenMPHintClause(Expr, StartLoc, LParenLoc, EndLoc);
5593     break;
5594   case OMPC_if:
5595   case OMPC_default:
5596   case OMPC_proc_bind:
5597   case OMPC_schedule:
5598   case OMPC_private:
5599   case OMPC_firstprivate:
5600   case OMPC_lastprivate:
5601   case OMPC_shared:
5602   case OMPC_reduction:
5603   case OMPC_linear:
5604   case OMPC_aligned:
5605   case OMPC_copyin:
5606   case OMPC_copyprivate:
5607   case OMPC_nowait:
5608   case OMPC_untied:
5609   case OMPC_mergeable:
5610   case OMPC_threadprivate:
5611   case OMPC_flush:
5612   case OMPC_read:
5613   case OMPC_write:
5614   case OMPC_update:
5615   case OMPC_capture:
5616   case OMPC_seq_cst:
5617   case OMPC_depend:
5618   case OMPC_threads:
5619   case OMPC_simd:
5620   case OMPC_map:
5621   case OMPC_nogroup:
5622   case OMPC_unknown:
5623     llvm_unreachable("Clause is not allowed.");
5624   }
5625   return Res;
5626 }
5627 
ActOnOpenMPIfClause(OpenMPDirectiveKind NameModifier,Expr * Condition,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation NameModifierLoc,SourceLocation ColonLoc,SourceLocation EndLoc)5628 OMPClause *Sema::ActOnOpenMPIfClause(OpenMPDirectiveKind NameModifier,
5629                                      Expr *Condition, SourceLocation StartLoc,
5630                                      SourceLocation LParenLoc,
5631                                      SourceLocation NameModifierLoc,
5632                                      SourceLocation ColonLoc,
5633                                      SourceLocation EndLoc) {
5634   Expr *ValExpr = Condition;
5635   if (!Condition->isValueDependent() && !Condition->isTypeDependent() &&
5636       !Condition->isInstantiationDependent() &&
5637       !Condition->containsUnexpandedParameterPack()) {
5638     ExprResult Val = ActOnBooleanCondition(DSAStack->getCurScope(),
5639                                            Condition->getExprLoc(), Condition);
5640     if (Val.isInvalid())
5641       return nullptr;
5642 
5643     ValExpr = Val.get();
5644   }
5645 
5646   return new (Context) OMPIfClause(NameModifier, ValExpr, StartLoc, LParenLoc,
5647                                    NameModifierLoc, ColonLoc, EndLoc);
5648 }
5649 
ActOnOpenMPFinalClause(Expr * Condition,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc)5650 OMPClause *Sema::ActOnOpenMPFinalClause(Expr *Condition,
5651                                         SourceLocation StartLoc,
5652                                         SourceLocation LParenLoc,
5653                                         SourceLocation EndLoc) {
5654   Expr *ValExpr = Condition;
5655   if (!Condition->isValueDependent() && !Condition->isTypeDependent() &&
5656       !Condition->isInstantiationDependent() &&
5657       !Condition->containsUnexpandedParameterPack()) {
5658     ExprResult Val = ActOnBooleanCondition(DSAStack->getCurScope(),
5659                                            Condition->getExprLoc(), Condition);
5660     if (Val.isInvalid())
5661       return nullptr;
5662 
5663     ValExpr = Val.get();
5664   }
5665 
5666   return new (Context) OMPFinalClause(ValExpr, StartLoc, LParenLoc, EndLoc);
5667 }
PerformOpenMPImplicitIntegerConversion(SourceLocation Loc,Expr * Op)5668 ExprResult Sema::PerformOpenMPImplicitIntegerConversion(SourceLocation Loc,
5669                                                         Expr *Op) {
5670   if (!Op)
5671     return ExprError();
5672 
5673   class IntConvertDiagnoser : public ICEConvertDiagnoser {
5674   public:
5675     IntConvertDiagnoser()
5676         : ICEConvertDiagnoser(/*AllowScopedEnumerations*/ false, false, true) {}
5677     SemaDiagnosticBuilder diagnoseNotInt(Sema &S, SourceLocation Loc,
5678                                          QualType T) override {
5679       return S.Diag(Loc, diag::err_omp_not_integral) << T;
5680     }
5681     SemaDiagnosticBuilder diagnoseIncomplete(Sema &S, SourceLocation Loc,
5682                                              QualType T) override {
5683       return S.Diag(Loc, diag::err_omp_incomplete_type) << T;
5684     }
5685     SemaDiagnosticBuilder diagnoseExplicitConv(Sema &S, SourceLocation Loc,
5686                                                QualType T,
5687                                                QualType ConvTy) override {
5688       return S.Diag(Loc, diag::err_omp_explicit_conversion) << T << ConvTy;
5689     }
5690     SemaDiagnosticBuilder noteExplicitConv(Sema &S, CXXConversionDecl *Conv,
5691                                            QualType ConvTy) override {
5692       return S.Diag(Conv->getLocation(), diag::note_omp_conversion_here)
5693              << ConvTy->isEnumeralType() << ConvTy;
5694     }
5695     SemaDiagnosticBuilder diagnoseAmbiguous(Sema &S, SourceLocation Loc,
5696                                             QualType T) override {
5697       return S.Diag(Loc, diag::err_omp_ambiguous_conversion) << T;
5698     }
5699     SemaDiagnosticBuilder noteAmbiguous(Sema &S, CXXConversionDecl *Conv,
5700                                         QualType ConvTy) override {
5701       return S.Diag(Conv->getLocation(), diag::note_omp_conversion_here)
5702              << ConvTy->isEnumeralType() << ConvTy;
5703     }
5704     SemaDiagnosticBuilder diagnoseConversion(Sema &, SourceLocation, QualType,
5705                                              QualType) override {
5706       llvm_unreachable("conversion functions are permitted");
5707     }
5708   } ConvertDiagnoser;
5709   return PerformContextualImplicitConversion(Loc, Op, ConvertDiagnoser);
5710 }
5711 
IsNonNegativeIntegerValue(Expr * & ValExpr,Sema & SemaRef,OpenMPClauseKind CKind,bool StrictlyPositive)5712 static bool IsNonNegativeIntegerValue(Expr *&ValExpr, Sema &SemaRef,
5713                                       OpenMPClauseKind CKind,
5714                                       bool StrictlyPositive) {
5715   if (!ValExpr->isTypeDependent() && !ValExpr->isValueDependent() &&
5716       !ValExpr->isInstantiationDependent()) {
5717     SourceLocation Loc = ValExpr->getExprLoc();
5718     ExprResult Value =
5719         SemaRef.PerformOpenMPImplicitIntegerConversion(Loc, ValExpr);
5720     if (Value.isInvalid())
5721       return false;
5722 
5723     ValExpr = Value.get();
5724     // The expression must evaluate to a non-negative integer value.
5725     llvm::APSInt Result;
5726     if (ValExpr->isIntegerConstantExpr(Result, SemaRef.Context) &&
5727         Result.isSigned() &&
5728         !((!StrictlyPositive && Result.isNonNegative()) ||
5729           (StrictlyPositive && Result.isStrictlyPositive()))) {
5730       SemaRef.Diag(Loc, diag::err_omp_negative_expression_in_clause)
5731           << getOpenMPClauseName(CKind) << (StrictlyPositive ? 1 : 0)
5732           << ValExpr->getSourceRange();
5733       return false;
5734     }
5735   }
5736   return true;
5737 }
5738 
ActOnOpenMPNumThreadsClause(Expr * NumThreads,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc)5739 OMPClause *Sema::ActOnOpenMPNumThreadsClause(Expr *NumThreads,
5740                                              SourceLocation StartLoc,
5741                                              SourceLocation LParenLoc,
5742                                              SourceLocation EndLoc) {
5743   Expr *ValExpr = NumThreads;
5744 
5745   // OpenMP [2.5, Restrictions]
5746   //  The num_threads expression must evaluate to a positive integer value.
5747   if (!IsNonNegativeIntegerValue(ValExpr, *this, OMPC_num_threads,
5748                                  /*StrictlyPositive=*/true))
5749     return nullptr;
5750 
5751   return new (Context)
5752       OMPNumThreadsClause(ValExpr, StartLoc, LParenLoc, EndLoc);
5753 }
5754 
VerifyPositiveIntegerConstantInClause(Expr * E,OpenMPClauseKind CKind)5755 ExprResult Sema::VerifyPositiveIntegerConstantInClause(Expr *E,
5756                                                        OpenMPClauseKind CKind) {
5757   if (!E)
5758     return ExprError();
5759   if (E->isValueDependent() || E->isTypeDependent() ||
5760       E->isInstantiationDependent() || E->containsUnexpandedParameterPack())
5761     return E;
5762   llvm::APSInt Result;
5763   ExprResult ICE = VerifyIntegerConstantExpression(E, &Result);
5764   if (ICE.isInvalid())
5765     return ExprError();
5766   if (!Result.isStrictlyPositive()) {
5767     Diag(E->getExprLoc(), diag::err_omp_negative_expression_in_clause)
5768         << getOpenMPClauseName(CKind) << 1 << E->getSourceRange();
5769     return ExprError();
5770   }
5771   if (CKind == OMPC_aligned && !Result.isPowerOf2()) {
5772     Diag(E->getExprLoc(), diag::warn_omp_alignment_not_power_of_two)
5773         << E->getSourceRange();
5774     return ExprError();
5775   }
5776   if (CKind == OMPC_collapse)
5777     DSAStack->setCollapseNumber(Result.getExtValue());
5778   else if (CKind == OMPC_ordered)
5779     DSAStack->setCollapseNumber(Result.getExtValue());
5780   return ICE;
5781 }
5782 
ActOnOpenMPSafelenClause(Expr * Len,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc)5783 OMPClause *Sema::ActOnOpenMPSafelenClause(Expr *Len, SourceLocation StartLoc,
5784                                           SourceLocation LParenLoc,
5785                                           SourceLocation EndLoc) {
5786   // OpenMP [2.8.1, simd construct, Description]
5787   // The parameter of the safelen clause must be a constant
5788   // positive integer expression.
5789   ExprResult Safelen = VerifyPositiveIntegerConstantInClause(Len, OMPC_safelen);
5790   if (Safelen.isInvalid())
5791     return nullptr;
5792   return new (Context)
5793       OMPSafelenClause(Safelen.get(), StartLoc, LParenLoc, EndLoc);
5794 }
5795 
ActOnOpenMPSimdlenClause(Expr * Len,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc)5796 OMPClause *Sema::ActOnOpenMPSimdlenClause(Expr *Len, SourceLocation StartLoc,
5797                                           SourceLocation LParenLoc,
5798                                           SourceLocation EndLoc) {
5799   // OpenMP [2.8.1, simd construct, Description]
5800   // The parameter of the simdlen clause must be a constant
5801   // positive integer expression.
5802   ExprResult Simdlen = VerifyPositiveIntegerConstantInClause(Len, OMPC_simdlen);
5803   if (Simdlen.isInvalid())
5804     return nullptr;
5805   return new (Context)
5806       OMPSimdlenClause(Simdlen.get(), StartLoc, LParenLoc, EndLoc);
5807 }
5808 
ActOnOpenMPCollapseClause(Expr * NumForLoops,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc)5809 OMPClause *Sema::ActOnOpenMPCollapseClause(Expr *NumForLoops,
5810                                            SourceLocation StartLoc,
5811                                            SourceLocation LParenLoc,
5812                                            SourceLocation EndLoc) {
5813   // OpenMP [2.7.1, loop construct, Description]
5814   // OpenMP [2.8.1, simd construct, Description]
5815   // OpenMP [2.9.6, distribute construct, Description]
5816   // The parameter of the collapse clause must be a constant
5817   // positive integer expression.
5818   ExprResult NumForLoopsResult =
5819       VerifyPositiveIntegerConstantInClause(NumForLoops, OMPC_collapse);
5820   if (NumForLoopsResult.isInvalid())
5821     return nullptr;
5822   return new (Context)
5823       OMPCollapseClause(NumForLoopsResult.get(), StartLoc, LParenLoc, EndLoc);
5824 }
5825 
ActOnOpenMPOrderedClause(SourceLocation StartLoc,SourceLocation EndLoc,SourceLocation LParenLoc,Expr * NumForLoops)5826 OMPClause *Sema::ActOnOpenMPOrderedClause(SourceLocation StartLoc,
5827                                           SourceLocation EndLoc,
5828                                           SourceLocation LParenLoc,
5829                                           Expr *NumForLoops) {
5830   // OpenMP [2.7.1, loop construct, Description]
5831   // OpenMP [2.8.1, simd construct, Description]
5832   // OpenMP [2.9.6, distribute construct, Description]
5833   // The parameter of the ordered clause must be a constant
5834   // positive integer expression if any.
5835   if (NumForLoops && LParenLoc.isValid()) {
5836     ExprResult NumForLoopsResult =
5837         VerifyPositiveIntegerConstantInClause(NumForLoops, OMPC_ordered);
5838     if (NumForLoopsResult.isInvalid())
5839       return nullptr;
5840     NumForLoops = NumForLoopsResult.get();
5841   } else
5842     NumForLoops = nullptr;
5843   DSAStack->setOrderedRegion(/*IsOrdered=*/true, NumForLoops);
5844   return new (Context)
5845       OMPOrderedClause(NumForLoops, StartLoc, LParenLoc, EndLoc);
5846 }
5847 
ActOnOpenMPSimpleClause(OpenMPClauseKind Kind,unsigned Argument,SourceLocation ArgumentLoc,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc)5848 OMPClause *Sema::ActOnOpenMPSimpleClause(
5849     OpenMPClauseKind Kind, unsigned Argument, SourceLocation ArgumentLoc,
5850     SourceLocation StartLoc, SourceLocation LParenLoc, SourceLocation EndLoc) {
5851   OMPClause *Res = nullptr;
5852   switch (Kind) {
5853   case OMPC_default:
5854     Res =
5855         ActOnOpenMPDefaultClause(static_cast<OpenMPDefaultClauseKind>(Argument),
5856                                  ArgumentLoc, StartLoc, LParenLoc, EndLoc);
5857     break;
5858   case OMPC_proc_bind:
5859     Res = ActOnOpenMPProcBindClause(
5860         static_cast<OpenMPProcBindClauseKind>(Argument), ArgumentLoc, StartLoc,
5861         LParenLoc, EndLoc);
5862     break;
5863   case OMPC_if:
5864   case OMPC_final:
5865   case OMPC_num_threads:
5866   case OMPC_safelen:
5867   case OMPC_simdlen:
5868   case OMPC_collapse:
5869   case OMPC_schedule:
5870   case OMPC_private:
5871   case OMPC_firstprivate:
5872   case OMPC_lastprivate:
5873   case OMPC_shared:
5874   case OMPC_reduction:
5875   case OMPC_linear:
5876   case OMPC_aligned:
5877   case OMPC_copyin:
5878   case OMPC_copyprivate:
5879   case OMPC_ordered:
5880   case OMPC_nowait:
5881   case OMPC_untied:
5882   case OMPC_mergeable:
5883   case OMPC_threadprivate:
5884   case OMPC_flush:
5885   case OMPC_read:
5886   case OMPC_write:
5887   case OMPC_update:
5888   case OMPC_capture:
5889   case OMPC_seq_cst:
5890   case OMPC_depend:
5891   case OMPC_device:
5892   case OMPC_threads:
5893   case OMPC_simd:
5894   case OMPC_map:
5895   case OMPC_num_teams:
5896   case OMPC_thread_limit:
5897   case OMPC_priority:
5898   case OMPC_grainsize:
5899   case OMPC_nogroup:
5900   case OMPC_num_tasks:
5901   case OMPC_hint:
5902   case OMPC_unknown:
5903     llvm_unreachable("Clause is not allowed.");
5904   }
5905   return Res;
5906 }
5907 
ActOnOpenMPDefaultClause(OpenMPDefaultClauseKind Kind,SourceLocation KindKwLoc,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc)5908 OMPClause *Sema::ActOnOpenMPDefaultClause(OpenMPDefaultClauseKind Kind,
5909                                           SourceLocation KindKwLoc,
5910                                           SourceLocation StartLoc,
5911                                           SourceLocation LParenLoc,
5912                                           SourceLocation EndLoc) {
5913   if (Kind == OMPC_DEFAULT_unknown) {
5914     std::string Values;
5915     static_assert(OMPC_DEFAULT_unknown > 0,
5916                   "OMPC_DEFAULT_unknown not greater than 0");
5917     std::string Sep(", ");
5918     for (unsigned i = 0; i < OMPC_DEFAULT_unknown; ++i) {
5919       Values += "'";
5920       Values += getOpenMPSimpleClauseTypeName(OMPC_default, i);
5921       Values += "'";
5922       switch (i) {
5923       case OMPC_DEFAULT_unknown - 2:
5924         Values += " or ";
5925         break;
5926       case OMPC_DEFAULT_unknown - 1:
5927         break;
5928       default:
5929         Values += Sep;
5930         break;
5931       }
5932     }
5933     Diag(KindKwLoc, diag::err_omp_unexpected_clause_value)
5934         << Values << getOpenMPClauseName(OMPC_default);
5935     return nullptr;
5936   }
5937   switch (Kind) {
5938   case OMPC_DEFAULT_none:
5939     DSAStack->setDefaultDSANone(KindKwLoc);
5940     break;
5941   case OMPC_DEFAULT_shared:
5942     DSAStack->setDefaultDSAShared(KindKwLoc);
5943     break;
5944   case OMPC_DEFAULT_unknown:
5945     llvm_unreachable("Clause kind is not allowed.");
5946     break;
5947   }
5948   return new (Context)
5949       OMPDefaultClause(Kind, KindKwLoc, StartLoc, LParenLoc, EndLoc);
5950 }
5951 
ActOnOpenMPProcBindClause(OpenMPProcBindClauseKind Kind,SourceLocation KindKwLoc,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc)5952 OMPClause *Sema::ActOnOpenMPProcBindClause(OpenMPProcBindClauseKind Kind,
5953                                            SourceLocation KindKwLoc,
5954                                            SourceLocation StartLoc,
5955                                            SourceLocation LParenLoc,
5956                                            SourceLocation EndLoc) {
5957   if (Kind == OMPC_PROC_BIND_unknown) {
5958     std::string Values;
5959     std::string Sep(", ");
5960     for (unsigned i = 0; i < OMPC_PROC_BIND_unknown; ++i) {
5961       Values += "'";
5962       Values += getOpenMPSimpleClauseTypeName(OMPC_proc_bind, i);
5963       Values += "'";
5964       switch (i) {
5965       case OMPC_PROC_BIND_unknown - 2:
5966         Values += " or ";
5967         break;
5968       case OMPC_PROC_BIND_unknown - 1:
5969         break;
5970       default:
5971         Values += Sep;
5972         break;
5973       }
5974     }
5975     Diag(KindKwLoc, diag::err_omp_unexpected_clause_value)
5976         << Values << getOpenMPClauseName(OMPC_proc_bind);
5977     return nullptr;
5978   }
5979   return new (Context)
5980       OMPProcBindClause(Kind, KindKwLoc, StartLoc, LParenLoc, EndLoc);
5981 }
5982 
ActOnOpenMPSingleExprWithArgClause(OpenMPClauseKind Kind,unsigned Argument,Expr * Expr,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation ArgumentLoc,SourceLocation DelimLoc,SourceLocation EndLoc)5983 OMPClause *Sema::ActOnOpenMPSingleExprWithArgClause(
5984     OpenMPClauseKind Kind, unsigned Argument, Expr *Expr,
5985     SourceLocation StartLoc, SourceLocation LParenLoc,
5986     SourceLocation ArgumentLoc, SourceLocation DelimLoc,
5987     SourceLocation EndLoc) {
5988   OMPClause *Res = nullptr;
5989   switch (Kind) {
5990   case OMPC_schedule:
5991     Res = ActOnOpenMPScheduleClause(
5992         static_cast<OpenMPScheduleClauseKind>(Argument), Expr, StartLoc,
5993         LParenLoc, ArgumentLoc, DelimLoc, EndLoc);
5994     break;
5995   case OMPC_if:
5996     Res =
5997         ActOnOpenMPIfClause(static_cast<OpenMPDirectiveKind>(Argument), Expr,
5998                             StartLoc, LParenLoc, ArgumentLoc, DelimLoc, EndLoc);
5999     break;
6000   case OMPC_final:
6001   case OMPC_num_threads:
6002   case OMPC_safelen:
6003   case OMPC_simdlen:
6004   case OMPC_collapse:
6005   case OMPC_default:
6006   case OMPC_proc_bind:
6007   case OMPC_private:
6008   case OMPC_firstprivate:
6009   case OMPC_lastprivate:
6010   case OMPC_shared:
6011   case OMPC_reduction:
6012   case OMPC_linear:
6013   case OMPC_aligned:
6014   case OMPC_copyin:
6015   case OMPC_copyprivate:
6016   case OMPC_ordered:
6017   case OMPC_nowait:
6018   case OMPC_untied:
6019   case OMPC_mergeable:
6020   case OMPC_threadprivate:
6021   case OMPC_flush:
6022   case OMPC_read:
6023   case OMPC_write:
6024   case OMPC_update:
6025   case OMPC_capture:
6026   case OMPC_seq_cst:
6027   case OMPC_depend:
6028   case OMPC_device:
6029   case OMPC_threads:
6030   case OMPC_simd:
6031   case OMPC_map:
6032   case OMPC_num_teams:
6033   case OMPC_thread_limit:
6034   case OMPC_priority:
6035   case OMPC_grainsize:
6036   case OMPC_nogroup:
6037   case OMPC_num_tasks:
6038   case OMPC_hint:
6039   case OMPC_unknown:
6040     llvm_unreachable("Clause is not allowed.");
6041   }
6042   return Res;
6043 }
6044 
ActOnOpenMPScheduleClause(OpenMPScheduleClauseKind Kind,Expr * ChunkSize,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation KindLoc,SourceLocation CommaLoc,SourceLocation EndLoc)6045 OMPClause *Sema::ActOnOpenMPScheduleClause(
6046     OpenMPScheduleClauseKind Kind, Expr *ChunkSize, SourceLocation StartLoc,
6047     SourceLocation LParenLoc, SourceLocation KindLoc, SourceLocation CommaLoc,
6048     SourceLocation EndLoc) {
6049   if (Kind == OMPC_SCHEDULE_unknown) {
6050     std::string Values;
6051     std::string Sep(", ");
6052     for (unsigned i = 0; i < OMPC_SCHEDULE_unknown; ++i) {
6053       Values += "'";
6054       Values += getOpenMPSimpleClauseTypeName(OMPC_schedule, i);
6055       Values += "'";
6056       switch (i) {
6057       case OMPC_SCHEDULE_unknown - 2:
6058         Values += " or ";
6059         break;
6060       case OMPC_SCHEDULE_unknown - 1:
6061         break;
6062       default:
6063         Values += Sep;
6064         break;
6065       }
6066     }
6067     Diag(KindLoc, diag::err_omp_unexpected_clause_value)
6068         << Values << getOpenMPClauseName(OMPC_schedule);
6069     return nullptr;
6070   }
6071   Expr *ValExpr = ChunkSize;
6072   Expr *HelperValExpr = nullptr;
6073   if (ChunkSize) {
6074     if (!ChunkSize->isValueDependent() && !ChunkSize->isTypeDependent() &&
6075         !ChunkSize->isInstantiationDependent() &&
6076         !ChunkSize->containsUnexpandedParameterPack()) {
6077       SourceLocation ChunkSizeLoc = ChunkSize->getLocStart();
6078       ExprResult Val =
6079           PerformOpenMPImplicitIntegerConversion(ChunkSizeLoc, ChunkSize);
6080       if (Val.isInvalid())
6081         return nullptr;
6082 
6083       ValExpr = Val.get();
6084 
6085       // OpenMP [2.7.1, Restrictions]
6086       //  chunk_size must be a loop invariant integer expression with a positive
6087       //  value.
6088       llvm::APSInt Result;
6089       if (ValExpr->isIntegerConstantExpr(Result, Context)) {
6090         if (Result.isSigned() && !Result.isStrictlyPositive()) {
6091           Diag(ChunkSizeLoc, diag::err_omp_negative_expression_in_clause)
6092               << "schedule" << 1 << ChunkSize->getSourceRange();
6093           return nullptr;
6094         }
6095       } else if (isParallelOrTaskRegion(DSAStack->getCurrentDirective())) {
6096         auto *ImpVar = buildVarDecl(*this, ChunkSize->getExprLoc(),
6097                                     ChunkSize->getType(), ".chunk.");
6098         auto *ImpVarRef = buildDeclRefExpr(*this, ImpVar, ChunkSize->getType(),
6099                                            ChunkSize->getExprLoc(),
6100                                            /*RefersToCapture=*/true);
6101         HelperValExpr = ImpVarRef;
6102       }
6103     }
6104   }
6105 
6106   return new (Context) OMPScheduleClause(StartLoc, LParenLoc, KindLoc, CommaLoc,
6107                                          EndLoc, Kind, ValExpr, HelperValExpr);
6108 }
6109 
ActOnOpenMPClause(OpenMPClauseKind Kind,SourceLocation StartLoc,SourceLocation EndLoc)6110 OMPClause *Sema::ActOnOpenMPClause(OpenMPClauseKind Kind,
6111                                    SourceLocation StartLoc,
6112                                    SourceLocation EndLoc) {
6113   OMPClause *Res = nullptr;
6114   switch (Kind) {
6115   case OMPC_ordered:
6116     Res = ActOnOpenMPOrderedClause(StartLoc, EndLoc);
6117     break;
6118   case OMPC_nowait:
6119     Res = ActOnOpenMPNowaitClause(StartLoc, EndLoc);
6120     break;
6121   case OMPC_untied:
6122     Res = ActOnOpenMPUntiedClause(StartLoc, EndLoc);
6123     break;
6124   case OMPC_mergeable:
6125     Res = ActOnOpenMPMergeableClause(StartLoc, EndLoc);
6126     break;
6127   case OMPC_read:
6128     Res = ActOnOpenMPReadClause(StartLoc, EndLoc);
6129     break;
6130   case OMPC_write:
6131     Res = ActOnOpenMPWriteClause(StartLoc, EndLoc);
6132     break;
6133   case OMPC_update:
6134     Res = ActOnOpenMPUpdateClause(StartLoc, EndLoc);
6135     break;
6136   case OMPC_capture:
6137     Res = ActOnOpenMPCaptureClause(StartLoc, EndLoc);
6138     break;
6139   case OMPC_seq_cst:
6140     Res = ActOnOpenMPSeqCstClause(StartLoc, EndLoc);
6141     break;
6142   case OMPC_threads:
6143     Res = ActOnOpenMPThreadsClause(StartLoc, EndLoc);
6144     break;
6145   case OMPC_simd:
6146     Res = ActOnOpenMPSIMDClause(StartLoc, EndLoc);
6147     break;
6148   case OMPC_nogroup:
6149     Res = ActOnOpenMPNogroupClause(StartLoc, EndLoc);
6150     break;
6151   case OMPC_if:
6152   case OMPC_final:
6153   case OMPC_num_threads:
6154   case OMPC_safelen:
6155   case OMPC_simdlen:
6156   case OMPC_collapse:
6157   case OMPC_schedule:
6158   case OMPC_private:
6159   case OMPC_firstprivate:
6160   case OMPC_lastprivate:
6161   case OMPC_shared:
6162   case OMPC_reduction:
6163   case OMPC_linear:
6164   case OMPC_aligned:
6165   case OMPC_copyin:
6166   case OMPC_copyprivate:
6167   case OMPC_default:
6168   case OMPC_proc_bind:
6169   case OMPC_threadprivate:
6170   case OMPC_flush:
6171   case OMPC_depend:
6172   case OMPC_device:
6173   case OMPC_map:
6174   case OMPC_num_teams:
6175   case OMPC_thread_limit:
6176   case OMPC_priority:
6177   case OMPC_grainsize:
6178   case OMPC_num_tasks:
6179   case OMPC_hint:
6180   case OMPC_unknown:
6181     llvm_unreachable("Clause is not allowed.");
6182   }
6183   return Res;
6184 }
6185 
ActOnOpenMPNowaitClause(SourceLocation StartLoc,SourceLocation EndLoc)6186 OMPClause *Sema::ActOnOpenMPNowaitClause(SourceLocation StartLoc,
6187                                          SourceLocation EndLoc) {
6188   DSAStack->setNowaitRegion();
6189   return new (Context) OMPNowaitClause(StartLoc, EndLoc);
6190 }
6191 
ActOnOpenMPUntiedClause(SourceLocation StartLoc,SourceLocation EndLoc)6192 OMPClause *Sema::ActOnOpenMPUntiedClause(SourceLocation StartLoc,
6193                                          SourceLocation EndLoc) {
6194   return new (Context) OMPUntiedClause(StartLoc, EndLoc);
6195 }
6196 
ActOnOpenMPMergeableClause(SourceLocation StartLoc,SourceLocation EndLoc)6197 OMPClause *Sema::ActOnOpenMPMergeableClause(SourceLocation StartLoc,
6198                                             SourceLocation EndLoc) {
6199   return new (Context) OMPMergeableClause(StartLoc, EndLoc);
6200 }
6201 
ActOnOpenMPReadClause(SourceLocation StartLoc,SourceLocation EndLoc)6202 OMPClause *Sema::ActOnOpenMPReadClause(SourceLocation StartLoc,
6203                                        SourceLocation EndLoc) {
6204   return new (Context) OMPReadClause(StartLoc, EndLoc);
6205 }
6206 
ActOnOpenMPWriteClause(SourceLocation StartLoc,SourceLocation EndLoc)6207 OMPClause *Sema::ActOnOpenMPWriteClause(SourceLocation StartLoc,
6208                                         SourceLocation EndLoc) {
6209   return new (Context) OMPWriteClause(StartLoc, EndLoc);
6210 }
6211 
ActOnOpenMPUpdateClause(SourceLocation StartLoc,SourceLocation EndLoc)6212 OMPClause *Sema::ActOnOpenMPUpdateClause(SourceLocation StartLoc,
6213                                          SourceLocation EndLoc) {
6214   return new (Context) OMPUpdateClause(StartLoc, EndLoc);
6215 }
6216 
ActOnOpenMPCaptureClause(SourceLocation StartLoc,SourceLocation EndLoc)6217 OMPClause *Sema::ActOnOpenMPCaptureClause(SourceLocation StartLoc,
6218                                           SourceLocation EndLoc) {
6219   return new (Context) OMPCaptureClause(StartLoc, EndLoc);
6220 }
6221 
ActOnOpenMPSeqCstClause(SourceLocation StartLoc,SourceLocation EndLoc)6222 OMPClause *Sema::ActOnOpenMPSeqCstClause(SourceLocation StartLoc,
6223                                          SourceLocation EndLoc) {
6224   return new (Context) OMPSeqCstClause(StartLoc, EndLoc);
6225 }
6226 
ActOnOpenMPThreadsClause(SourceLocation StartLoc,SourceLocation EndLoc)6227 OMPClause *Sema::ActOnOpenMPThreadsClause(SourceLocation StartLoc,
6228                                           SourceLocation EndLoc) {
6229   return new (Context) OMPThreadsClause(StartLoc, EndLoc);
6230 }
6231 
ActOnOpenMPSIMDClause(SourceLocation StartLoc,SourceLocation EndLoc)6232 OMPClause *Sema::ActOnOpenMPSIMDClause(SourceLocation StartLoc,
6233                                        SourceLocation EndLoc) {
6234   return new (Context) OMPSIMDClause(StartLoc, EndLoc);
6235 }
6236 
ActOnOpenMPNogroupClause(SourceLocation StartLoc,SourceLocation EndLoc)6237 OMPClause *Sema::ActOnOpenMPNogroupClause(SourceLocation StartLoc,
6238                                           SourceLocation EndLoc) {
6239   return new (Context) OMPNogroupClause(StartLoc, EndLoc);
6240 }
6241 
ActOnOpenMPVarListClause(OpenMPClauseKind Kind,ArrayRef<Expr * > VarList,Expr * TailExpr,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation ColonLoc,SourceLocation EndLoc,CXXScopeSpec & ReductionIdScopeSpec,const DeclarationNameInfo & ReductionId,OpenMPDependClauseKind DepKind,OpenMPLinearClauseKind LinKind,OpenMPMapClauseKind MapTypeModifier,OpenMPMapClauseKind MapType,SourceLocation DepLinMapLoc)6242 OMPClause *Sema::ActOnOpenMPVarListClause(
6243     OpenMPClauseKind Kind, ArrayRef<Expr *> VarList, Expr *TailExpr,
6244     SourceLocation StartLoc, SourceLocation LParenLoc, SourceLocation ColonLoc,
6245     SourceLocation EndLoc, CXXScopeSpec &ReductionIdScopeSpec,
6246     const DeclarationNameInfo &ReductionId, OpenMPDependClauseKind DepKind,
6247     OpenMPLinearClauseKind LinKind, OpenMPMapClauseKind MapTypeModifier,
6248     OpenMPMapClauseKind MapType, SourceLocation DepLinMapLoc) {
6249   OMPClause *Res = nullptr;
6250   switch (Kind) {
6251   case OMPC_private:
6252     Res = ActOnOpenMPPrivateClause(VarList, StartLoc, LParenLoc, EndLoc);
6253     break;
6254   case OMPC_firstprivate:
6255     Res = ActOnOpenMPFirstprivateClause(VarList, StartLoc, LParenLoc, EndLoc);
6256     break;
6257   case OMPC_lastprivate:
6258     Res = ActOnOpenMPLastprivateClause(VarList, StartLoc, LParenLoc, EndLoc);
6259     break;
6260   case OMPC_shared:
6261     Res = ActOnOpenMPSharedClause(VarList, StartLoc, LParenLoc, EndLoc);
6262     break;
6263   case OMPC_reduction:
6264     Res = ActOnOpenMPReductionClause(VarList, StartLoc, LParenLoc, ColonLoc,
6265                                      EndLoc, ReductionIdScopeSpec, ReductionId);
6266     break;
6267   case OMPC_linear:
6268     Res = ActOnOpenMPLinearClause(VarList, TailExpr, StartLoc, LParenLoc,
6269                                   LinKind, DepLinMapLoc, ColonLoc, EndLoc);
6270     break;
6271   case OMPC_aligned:
6272     Res = ActOnOpenMPAlignedClause(VarList, TailExpr, StartLoc, LParenLoc,
6273                                    ColonLoc, EndLoc);
6274     break;
6275   case OMPC_copyin:
6276     Res = ActOnOpenMPCopyinClause(VarList, StartLoc, LParenLoc, EndLoc);
6277     break;
6278   case OMPC_copyprivate:
6279     Res = ActOnOpenMPCopyprivateClause(VarList, StartLoc, LParenLoc, EndLoc);
6280     break;
6281   case OMPC_flush:
6282     Res = ActOnOpenMPFlushClause(VarList, StartLoc, LParenLoc, EndLoc);
6283     break;
6284   case OMPC_depend:
6285     Res = ActOnOpenMPDependClause(DepKind, DepLinMapLoc, ColonLoc, VarList,
6286                                   StartLoc, LParenLoc, EndLoc);
6287     break;
6288   case OMPC_map:
6289     Res = ActOnOpenMPMapClause(MapTypeModifier, MapType, DepLinMapLoc, ColonLoc,
6290                                VarList, StartLoc, LParenLoc, EndLoc);
6291     break;
6292   case OMPC_if:
6293   case OMPC_final:
6294   case OMPC_num_threads:
6295   case OMPC_safelen:
6296   case OMPC_simdlen:
6297   case OMPC_collapse:
6298   case OMPC_default:
6299   case OMPC_proc_bind:
6300   case OMPC_schedule:
6301   case OMPC_ordered:
6302   case OMPC_nowait:
6303   case OMPC_untied:
6304   case OMPC_mergeable:
6305   case OMPC_threadprivate:
6306   case OMPC_read:
6307   case OMPC_write:
6308   case OMPC_update:
6309   case OMPC_capture:
6310   case OMPC_seq_cst:
6311   case OMPC_device:
6312   case OMPC_threads:
6313   case OMPC_simd:
6314   case OMPC_num_teams:
6315   case OMPC_thread_limit:
6316   case OMPC_priority:
6317   case OMPC_grainsize:
6318   case OMPC_nogroup:
6319   case OMPC_num_tasks:
6320   case OMPC_hint:
6321   case OMPC_unknown:
6322     llvm_unreachable("Clause is not allowed.");
6323   }
6324   return Res;
6325 }
6326 
ActOnOpenMPPrivateClause(ArrayRef<Expr * > VarList,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc)6327 OMPClause *Sema::ActOnOpenMPPrivateClause(ArrayRef<Expr *> VarList,
6328                                           SourceLocation StartLoc,
6329                                           SourceLocation LParenLoc,
6330                                           SourceLocation EndLoc) {
6331   SmallVector<Expr *, 8> Vars;
6332   SmallVector<Expr *, 8> PrivateCopies;
6333   for (auto &RefExpr : VarList) {
6334     assert(RefExpr && "NULL expr in OpenMP private clause.");
6335     if (isa<DependentScopeDeclRefExpr>(RefExpr)) {
6336       // It will be analyzed later.
6337       Vars.push_back(RefExpr);
6338       PrivateCopies.push_back(nullptr);
6339       continue;
6340     }
6341 
6342     SourceLocation ELoc = RefExpr->getExprLoc();
6343     // OpenMP [2.1, C/C++]
6344     //  A list item is a variable name.
6345     // OpenMP  [2.9.3.3, Restrictions, p.1]
6346     //  A variable that is part of another variable (as an array or
6347     //  structure element) cannot appear in a private clause.
6348     DeclRefExpr *DE = dyn_cast_or_null<DeclRefExpr>(RefExpr);
6349     if (!DE || !isa<VarDecl>(DE->getDecl())) {
6350       Diag(ELoc, diag::err_omp_expected_var_name) << RefExpr->getSourceRange();
6351       continue;
6352     }
6353     Decl *D = DE->getDecl();
6354     VarDecl *VD = cast<VarDecl>(D);
6355 
6356     QualType Type = VD->getType();
6357     if (Type->isDependentType() || Type->isInstantiationDependentType()) {
6358       // It will be analyzed later.
6359       Vars.push_back(DE);
6360       PrivateCopies.push_back(nullptr);
6361       continue;
6362     }
6363 
6364     // OpenMP [2.9.3.3, Restrictions, C/C++, p.3]
6365     //  A variable that appears in a private clause must not have an incomplete
6366     //  type or a reference type.
6367     if (RequireCompleteType(ELoc, Type,
6368                             diag::err_omp_private_incomplete_type)) {
6369       continue;
6370     }
6371     Type = Type.getNonReferenceType();
6372 
6373     // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced
6374     // in a Construct]
6375     //  Variables with the predetermined data-sharing attributes may not be
6376     //  listed in data-sharing attributes clauses, except for the cases
6377     //  listed below. For these exceptions only, listing a predetermined
6378     //  variable in a data-sharing attribute clause is allowed and overrides
6379     //  the variable's predetermined data-sharing attributes.
6380     DSAStackTy::DSAVarData DVar = DSAStack->getTopDSA(VD, false);
6381     if (DVar.CKind != OMPC_unknown && DVar.CKind != OMPC_private) {
6382       Diag(ELoc, diag::err_omp_wrong_dsa) << getOpenMPClauseName(DVar.CKind)
6383                                           << getOpenMPClauseName(OMPC_private);
6384       ReportOriginalDSA(*this, DSAStack, VD, DVar);
6385       continue;
6386     }
6387 
6388     // Variably modified types are not supported for tasks.
6389     if (!Type->isAnyPointerType() && Type->isVariablyModifiedType() &&
6390         DSAStack->getCurrentDirective() == OMPD_task) {
6391       Diag(ELoc, diag::err_omp_variably_modified_type_not_supported)
6392           << getOpenMPClauseName(OMPC_private) << Type
6393           << getOpenMPDirectiveName(DSAStack->getCurrentDirective());
6394       bool IsDecl =
6395           VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly;
6396       Diag(VD->getLocation(),
6397            IsDecl ? diag::note_previous_decl : diag::note_defined_here)
6398           << VD;
6399       continue;
6400     }
6401 
6402     // OpenMP [2.9.3.3, Restrictions, C/C++, p.1]
6403     //  A variable of class type (or array thereof) that appears in a private
6404     //  clause requires an accessible, unambiguous default constructor for the
6405     //  class type.
6406     // Generate helper private variable and initialize it with the default
6407     // value. The address of the original variable is replaced by the address of
6408     // the new private variable in CodeGen. This new variable is not added to
6409     // IdResolver, so the code in the OpenMP region uses original variable for
6410     // proper diagnostics.
6411     Type = Type.getUnqualifiedType();
6412     auto VDPrivate = buildVarDecl(*this, DE->getExprLoc(), Type, VD->getName(),
6413                                   VD->hasAttrs() ? &VD->getAttrs() : nullptr);
6414     ActOnUninitializedDecl(VDPrivate, /*TypeMayContainAuto=*/false);
6415     if (VDPrivate->isInvalidDecl())
6416       continue;
6417     auto VDPrivateRefExpr = buildDeclRefExpr(
6418         *this, VDPrivate, DE->getType().getUnqualifiedType(), DE->getExprLoc());
6419 
6420     DSAStack->addDSA(VD, DE, OMPC_private);
6421     Vars.push_back(DE);
6422     PrivateCopies.push_back(VDPrivateRefExpr);
6423   }
6424 
6425   if (Vars.empty())
6426     return nullptr;
6427 
6428   return OMPPrivateClause::Create(Context, StartLoc, LParenLoc, EndLoc, Vars,
6429                                   PrivateCopies);
6430 }
6431 
6432 namespace {
6433 class DiagsUninitializedSeveretyRAII {
6434 private:
6435   DiagnosticsEngine &Diags;
6436   SourceLocation SavedLoc;
6437   bool IsIgnored;
6438 
6439 public:
DiagsUninitializedSeveretyRAII(DiagnosticsEngine & Diags,SourceLocation Loc,bool IsIgnored)6440   DiagsUninitializedSeveretyRAII(DiagnosticsEngine &Diags, SourceLocation Loc,
6441                                  bool IsIgnored)
6442       : Diags(Diags), SavedLoc(Loc), IsIgnored(IsIgnored) {
6443     if (!IsIgnored) {
6444       Diags.setSeverity(/*Diag*/ diag::warn_uninit_self_reference_in_init,
6445                         /*Map*/ diag::Severity::Ignored, Loc);
6446     }
6447   }
~DiagsUninitializedSeveretyRAII()6448   ~DiagsUninitializedSeveretyRAII() {
6449     if (!IsIgnored)
6450       Diags.popMappings(SavedLoc);
6451   }
6452 };
6453 }
6454 
ActOnOpenMPFirstprivateClause(ArrayRef<Expr * > VarList,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc)6455 OMPClause *Sema::ActOnOpenMPFirstprivateClause(ArrayRef<Expr *> VarList,
6456                                                SourceLocation StartLoc,
6457                                                SourceLocation LParenLoc,
6458                                                SourceLocation EndLoc) {
6459   SmallVector<Expr *, 8> Vars;
6460   SmallVector<Expr *, 8> PrivateCopies;
6461   SmallVector<Expr *, 8> Inits;
6462   bool IsImplicitClause =
6463       StartLoc.isInvalid() && LParenLoc.isInvalid() && EndLoc.isInvalid();
6464   auto ImplicitClauseLoc = DSAStack->getConstructLoc();
6465 
6466   for (auto &RefExpr : VarList) {
6467     assert(RefExpr && "NULL expr in OpenMP firstprivate clause.");
6468     if (isa<DependentScopeDeclRefExpr>(RefExpr)) {
6469       // It will be analyzed later.
6470       Vars.push_back(RefExpr);
6471       PrivateCopies.push_back(nullptr);
6472       Inits.push_back(nullptr);
6473       continue;
6474     }
6475 
6476     SourceLocation ELoc =
6477         IsImplicitClause ? ImplicitClauseLoc : RefExpr->getExprLoc();
6478     // OpenMP [2.1, C/C++]
6479     //  A list item is a variable name.
6480     // OpenMP  [2.9.3.3, Restrictions, p.1]
6481     //  A variable that is part of another variable (as an array or
6482     //  structure element) cannot appear in a private clause.
6483     DeclRefExpr *DE = dyn_cast_or_null<DeclRefExpr>(RefExpr);
6484     if (!DE || !isa<VarDecl>(DE->getDecl())) {
6485       Diag(ELoc, diag::err_omp_expected_var_name) << RefExpr->getSourceRange();
6486       continue;
6487     }
6488     Decl *D = DE->getDecl();
6489     VarDecl *VD = cast<VarDecl>(D);
6490 
6491     QualType Type = VD->getType();
6492     if (Type->isDependentType() || Type->isInstantiationDependentType()) {
6493       // It will be analyzed later.
6494       Vars.push_back(DE);
6495       PrivateCopies.push_back(nullptr);
6496       Inits.push_back(nullptr);
6497       continue;
6498     }
6499 
6500     // OpenMP [2.9.3.3, Restrictions, C/C++, p.3]
6501     //  A variable that appears in a private clause must not have an incomplete
6502     //  type or a reference type.
6503     if (RequireCompleteType(ELoc, Type,
6504                             diag::err_omp_firstprivate_incomplete_type)) {
6505       continue;
6506     }
6507     Type = Type.getNonReferenceType();
6508 
6509     // OpenMP [2.9.3.4, Restrictions, C/C++, p.1]
6510     //  A variable of class type (or array thereof) that appears in a private
6511     //  clause requires an accessible, unambiguous copy constructor for the
6512     //  class type.
6513     auto ElemType = Context.getBaseElementType(Type).getNonReferenceType();
6514 
6515     // If an implicit firstprivate variable found it was checked already.
6516     if (!IsImplicitClause) {
6517       DSAStackTy::DSAVarData DVar = DSAStack->getTopDSA(VD, false);
6518       bool IsConstant = ElemType.isConstant(Context);
6519       // OpenMP [2.4.13, Data-sharing Attribute Clauses]
6520       //  A list item that specifies a given variable may not appear in more
6521       // than one clause on the same directive, except that a variable may be
6522       //  specified in both firstprivate and lastprivate clauses.
6523       if (DVar.CKind != OMPC_unknown && DVar.CKind != OMPC_firstprivate &&
6524           DVar.CKind != OMPC_lastprivate && DVar.RefExpr) {
6525         Diag(ELoc, diag::err_omp_wrong_dsa)
6526             << getOpenMPClauseName(DVar.CKind)
6527             << getOpenMPClauseName(OMPC_firstprivate);
6528         ReportOriginalDSA(*this, DSAStack, VD, DVar);
6529         continue;
6530       }
6531 
6532       // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced
6533       // in a Construct]
6534       //  Variables with the predetermined data-sharing attributes may not be
6535       //  listed in data-sharing attributes clauses, except for the cases
6536       //  listed below. For these exceptions only, listing a predetermined
6537       //  variable in a data-sharing attribute clause is allowed and overrides
6538       //  the variable's predetermined data-sharing attributes.
6539       // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced
6540       // in a Construct, C/C++, p.2]
6541       //  Variables with const-qualified type having no mutable member may be
6542       //  listed in a firstprivate clause, even if they are static data members.
6543       if (!(IsConstant || VD->isStaticDataMember()) && !DVar.RefExpr &&
6544           DVar.CKind != OMPC_unknown && DVar.CKind != OMPC_shared) {
6545         Diag(ELoc, diag::err_omp_wrong_dsa)
6546             << getOpenMPClauseName(DVar.CKind)
6547             << getOpenMPClauseName(OMPC_firstprivate);
6548         ReportOriginalDSA(*this, DSAStack, VD, DVar);
6549         continue;
6550       }
6551 
6552       OpenMPDirectiveKind CurrDir = DSAStack->getCurrentDirective();
6553       // OpenMP [2.9.3.4, Restrictions, p.2]
6554       //  A list item that is private within a parallel region must not appear
6555       //  in a firstprivate clause on a worksharing construct if any of the
6556       //  worksharing regions arising from the worksharing construct ever bind
6557       //  to any of the parallel regions arising from the parallel construct.
6558       if (isOpenMPWorksharingDirective(CurrDir) &&
6559           !isOpenMPParallelDirective(CurrDir)) {
6560         DVar = DSAStack->getImplicitDSA(VD, true);
6561         if (DVar.CKind != OMPC_shared &&
6562             (isOpenMPParallelDirective(DVar.DKind) ||
6563              DVar.DKind == OMPD_unknown)) {
6564           Diag(ELoc, diag::err_omp_required_access)
6565               << getOpenMPClauseName(OMPC_firstprivate)
6566               << getOpenMPClauseName(OMPC_shared);
6567           ReportOriginalDSA(*this, DSAStack, VD, DVar);
6568           continue;
6569         }
6570       }
6571       // OpenMP [2.9.3.4, Restrictions, p.3]
6572       //  A list item that appears in a reduction clause of a parallel construct
6573       //  must not appear in a firstprivate clause on a worksharing or task
6574       //  construct if any of the worksharing or task regions arising from the
6575       //  worksharing or task construct ever bind to any of the parallel regions
6576       //  arising from the parallel construct.
6577       // OpenMP [2.9.3.4, Restrictions, p.4]
6578       //  A list item that appears in a reduction clause in worksharing
6579       //  construct must not appear in a firstprivate clause in a task construct
6580       //  encountered during execution of any of the worksharing regions arising
6581       //  from the worksharing construct.
6582       if (CurrDir == OMPD_task) {
6583         DVar =
6584             DSAStack->hasInnermostDSA(VD, MatchesAnyClause(OMPC_reduction),
6585                                       [](OpenMPDirectiveKind K) -> bool {
6586                                         return isOpenMPParallelDirective(K) ||
6587                                                isOpenMPWorksharingDirective(K);
6588                                       },
6589                                       false);
6590         if (DVar.CKind == OMPC_reduction &&
6591             (isOpenMPParallelDirective(DVar.DKind) ||
6592              isOpenMPWorksharingDirective(DVar.DKind))) {
6593           Diag(ELoc, diag::err_omp_parallel_reduction_in_task_firstprivate)
6594               << getOpenMPDirectiveName(DVar.DKind);
6595           ReportOriginalDSA(*this, DSAStack, VD, DVar);
6596           continue;
6597         }
6598       }
6599 
6600       // OpenMP 4.5 [2.15.3.4, Restrictions, p.3]
6601       // A list item that is private within a teams region must not appear in a
6602       // firstprivate clause on a distribute construct if any of the distribute
6603       // regions arising from the distribute construct ever bind to any of the
6604       // teams regions arising from the teams construct.
6605       // OpenMP 4.5 [2.15.3.4, Restrictions, p.3]
6606       // A list item that appears in a reduction clause of a teams construct
6607       // must not appear in a firstprivate clause on a distribute construct if
6608       // any of the distribute regions arising from the distribute construct
6609       // ever bind to any of the teams regions arising from the teams construct.
6610       // OpenMP 4.5 [2.10.8, Distribute Construct, p.3]
6611       // A list item may appear in a firstprivate or lastprivate clause but not
6612       // both.
6613       if (CurrDir == OMPD_distribute) {
6614         DVar = DSAStack->hasInnermostDSA(VD, MatchesAnyClause(OMPC_private),
6615                                          [](OpenMPDirectiveKind K) -> bool {
6616                                            return isOpenMPTeamsDirective(K);
6617                                          },
6618                                          false);
6619         if (DVar.CKind == OMPC_private && isOpenMPTeamsDirective(DVar.DKind)) {
6620           Diag(ELoc, diag::err_omp_firstprivate_distribute_private_teams);
6621           ReportOriginalDSA(*this, DSAStack, VD, DVar);
6622           continue;
6623         }
6624         DVar = DSAStack->hasInnermostDSA(VD, MatchesAnyClause(OMPC_reduction),
6625                                          [](OpenMPDirectiveKind K) -> bool {
6626                                            return isOpenMPTeamsDirective(K);
6627                                          },
6628                                          false);
6629         if (DVar.CKind == OMPC_reduction &&
6630             isOpenMPTeamsDirective(DVar.DKind)) {
6631           Diag(ELoc, diag::err_omp_firstprivate_distribute_in_teams_reduction);
6632           ReportOriginalDSA(*this, DSAStack, VD, DVar);
6633           continue;
6634         }
6635         DVar = DSAStack->getTopDSA(VD, false);
6636         if (DVar.CKind == OMPC_lastprivate) {
6637           Diag(ELoc, diag::err_omp_firstprivate_and_lastprivate_in_distribute);
6638           ReportOriginalDSA(*this, DSAStack, VD, DVar);
6639           continue;
6640         }
6641       }
6642     }
6643 
6644     // Variably modified types are not supported for tasks.
6645     if (!Type->isAnyPointerType() && Type->isVariablyModifiedType() &&
6646         DSAStack->getCurrentDirective() == OMPD_task) {
6647       Diag(ELoc, diag::err_omp_variably_modified_type_not_supported)
6648           << getOpenMPClauseName(OMPC_firstprivate) << Type
6649           << getOpenMPDirectiveName(DSAStack->getCurrentDirective());
6650       bool IsDecl =
6651           VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly;
6652       Diag(VD->getLocation(),
6653            IsDecl ? diag::note_previous_decl : diag::note_defined_here)
6654           << VD;
6655       continue;
6656     }
6657 
6658     Type = Type.getUnqualifiedType();
6659     auto VDPrivate = buildVarDecl(*this, ELoc, Type, VD->getName(),
6660                                   VD->hasAttrs() ? &VD->getAttrs() : nullptr);
6661     // Generate helper private variable and initialize it with the value of the
6662     // original variable. The address of the original variable is replaced by
6663     // the address of the new private variable in the CodeGen. This new variable
6664     // is not added to IdResolver, so the code in the OpenMP region uses
6665     // original variable for proper diagnostics and variable capturing.
6666     Expr *VDInitRefExpr = nullptr;
6667     // For arrays generate initializer for single element and replace it by the
6668     // original array element in CodeGen.
6669     if (Type->isArrayType()) {
6670       auto VDInit =
6671           buildVarDecl(*this, DE->getExprLoc(), ElemType, VD->getName());
6672       VDInitRefExpr = buildDeclRefExpr(*this, VDInit, ElemType, ELoc);
6673       auto Init = DefaultLvalueConversion(VDInitRefExpr).get();
6674       ElemType = ElemType.getUnqualifiedType();
6675       auto *VDInitTemp = buildVarDecl(*this, DE->getLocStart(), ElemType,
6676                                       ".firstprivate.temp");
6677       InitializedEntity Entity =
6678           InitializedEntity::InitializeVariable(VDInitTemp);
6679       InitializationKind Kind = InitializationKind::CreateCopy(ELoc, ELoc);
6680 
6681       InitializationSequence InitSeq(*this, Entity, Kind, Init);
6682       ExprResult Result = InitSeq.Perform(*this, Entity, Kind, Init);
6683       if (Result.isInvalid())
6684         VDPrivate->setInvalidDecl();
6685       else
6686         VDPrivate->setInit(Result.getAs<Expr>());
6687       // Remove temp variable declaration.
6688       Context.Deallocate(VDInitTemp);
6689     } else {
6690       auto *VDInit =
6691           buildVarDecl(*this, DE->getLocStart(), Type, ".firstprivate.temp");
6692       VDInitRefExpr =
6693           buildDeclRefExpr(*this, VDInit, DE->getType(), DE->getExprLoc());
6694       AddInitializerToDecl(VDPrivate,
6695                            DefaultLvalueConversion(VDInitRefExpr).get(),
6696                            /*DirectInit=*/false, /*TypeMayContainAuto=*/false);
6697     }
6698     if (VDPrivate->isInvalidDecl()) {
6699       if (IsImplicitClause) {
6700         Diag(DE->getExprLoc(),
6701              diag::note_omp_task_predetermined_firstprivate_here);
6702       }
6703       continue;
6704     }
6705     CurContext->addDecl(VDPrivate);
6706     auto VDPrivateRefExpr = buildDeclRefExpr(
6707         *this, VDPrivate, DE->getType().getUnqualifiedType(), DE->getExprLoc());
6708     DSAStack->addDSA(VD, DE, OMPC_firstprivate);
6709     Vars.push_back(DE);
6710     PrivateCopies.push_back(VDPrivateRefExpr);
6711     Inits.push_back(VDInitRefExpr);
6712   }
6713 
6714   if (Vars.empty())
6715     return nullptr;
6716 
6717   return OMPFirstprivateClause::Create(Context, StartLoc, LParenLoc, EndLoc,
6718                                        Vars, PrivateCopies, Inits);
6719 }
6720 
ActOnOpenMPLastprivateClause(ArrayRef<Expr * > VarList,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc)6721 OMPClause *Sema::ActOnOpenMPLastprivateClause(ArrayRef<Expr *> VarList,
6722                                               SourceLocation StartLoc,
6723                                               SourceLocation LParenLoc,
6724                                               SourceLocation EndLoc) {
6725   SmallVector<Expr *, 8> Vars;
6726   SmallVector<Expr *, 8> SrcExprs;
6727   SmallVector<Expr *, 8> DstExprs;
6728   SmallVector<Expr *, 8> AssignmentOps;
6729   for (auto &RefExpr : VarList) {
6730     assert(RefExpr && "NULL expr in OpenMP lastprivate clause.");
6731     if (isa<DependentScopeDeclRefExpr>(RefExpr)) {
6732       // It will be analyzed later.
6733       Vars.push_back(RefExpr);
6734       SrcExprs.push_back(nullptr);
6735       DstExprs.push_back(nullptr);
6736       AssignmentOps.push_back(nullptr);
6737       continue;
6738     }
6739 
6740     SourceLocation ELoc = RefExpr->getExprLoc();
6741     // OpenMP [2.1, C/C++]
6742     //  A list item is a variable name.
6743     // OpenMP  [2.14.3.5, Restrictions, p.1]
6744     //  A variable that is part of another variable (as an array or structure
6745     //  element) cannot appear in a lastprivate clause.
6746     DeclRefExpr *DE = dyn_cast_or_null<DeclRefExpr>(RefExpr);
6747     if (!DE || !isa<VarDecl>(DE->getDecl())) {
6748       Diag(ELoc, diag::err_omp_expected_var_name) << RefExpr->getSourceRange();
6749       continue;
6750     }
6751     Decl *D = DE->getDecl();
6752     VarDecl *VD = cast<VarDecl>(D);
6753 
6754     QualType Type = VD->getType();
6755     if (Type->isDependentType() || Type->isInstantiationDependentType()) {
6756       // It will be analyzed later.
6757       Vars.push_back(DE);
6758       SrcExprs.push_back(nullptr);
6759       DstExprs.push_back(nullptr);
6760       AssignmentOps.push_back(nullptr);
6761       continue;
6762     }
6763 
6764     // OpenMP [2.14.3.5, Restrictions, C/C++, p.2]
6765     //  A variable that appears in a lastprivate clause must not have an
6766     //  incomplete type or a reference type.
6767     if (RequireCompleteType(ELoc, Type,
6768                             diag::err_omp_lastprivate_incomplete_type)) {
6769       continue;
6770     }
6771     Type = Type.getNonReferenceType();
6772 
6773     // OpenMP [2.14.1.1, Data-sharing Attribute Rules for Variables Referenced
6774     // in a Construct]
6775     //  Variables with the predetermined data-sharing attributes may not be
6776     //  listed in data-sharing attributes clauses, except for the cases
6777     //  listed below.
6778     DSAStackTy::DSAVarData DVar = DSAStack->getTopDSA(VD, false);
6779     if (DVar.CKind != OMPC_unknown && DVar.CKind != OMPC_lastprivate &&
6780         DVar.CKind != OMPC_firstprivate &&
6781         (DVar.CKind != OMPC_private || DVar.RefExpr != nullptr)) {
6782       Diag(ELoc, diag::err_omp_wrong_dsa)
6783           << getOpenMPClauseName(DVar.CKind)
6784           << getOpenMPClauseName(OMPC_lastprivate);
6785       ReportOriginalDSA(*this, DSAStack, VD, DVar);
6786       continue;
6787     }
6788 
6789     OpenMPDirectiveKind CurrDir = DSAStack->getCurrentDirective();
6790     // OpenMP [2.14.3.5, Restrictions, p.2]
6791     // A list item that is private within a parallel region, or that appears in
6792     // the reduction clause of a parallel construct, must not appear in a
6793     // lastprivate clause on a worksharing construct if any of the corresponding
6794     // worksharing regions ever binds to any of the corresponding parallel
6795     // regions.
6796     DSAStackTy::DSAVarData TopDVar = DVar;
6797     if (isOpenMPWorksharingDirective(CurrDir) &&
6798         !isOpenMPParallelDirective(CurrDir)) {
6799       DVar = DSAStack->getImplicitDSA(VD, true);
6800       if (DVar.CKind != OMPC_shared) {
6801         Diag(ELoc, diag::err_omp_required_access)
6802             << getOpenMPClauseName(OMPC_lastprivate)
6803             << getOpenMPClauseName(OMPC_shared);
6804         ReportOriginalDSA(*this, DSAStack, VD, DVar);
6805         continue;
6806       }
6807     }
6808     // OpenMP [2.14.3.5, Restrictions, C++, p.1,2]
6809     //  A variable of class type (or array thereof) that appears in a
6810     //  lastprivate clause requires an accessible, unambiguous default
6811     //  constructor for the class type, unless the list item is also specified
6812     //  in a firstprivate clause.
6813     //  A variable of class type (or array thereof) that appears in a
6814     //  lastprivate clause requires an accessible, unambiguous copy assignment
6815     //  operator for the class type.
6816     Type = Context.getBaseElementType(Type).getNonReferenceType();
6817     auto *SrcVD = buildVarDecl(*this, DE->getLocStart(),
6818                                Type.getUnqualifiedType(), ".lastprivate.src",
6819                                VD->hasAttrs() ? &VD->getAttrs() : nullptr);
6820     auto *PseudoSrcExpr = buildDeclRefExpr(
6821         *this, SrcVD, Type.getUnqualifiedType(), DE->getExprLoc());
6822     auto *DstVD =
6823         buildVarDecl(*this, DE->getLocStart(), Type, ".lastprivate.dst",
6824                      VD->hasAttrs() ? &VD->getAttrs() : nullptr);
6825     auto *PseudoDstExpr =
6826         buildDeclRefExpr(*this, DstVD, Type, DE->getExprLoc());
6827     // For arrays generate assignment operation for single element and replace
6828     // it by the original array element in CodeGen.
6829     auto AssignmentOp = BuildBinOp(/*S=*/nullptr, DE->getExprLoc(), BO_Assign,
6830                                    PseudoDstExpr, PseudoSrcExpr);
6831     if (AssignmentOp.isInvalid())
6832       continue;
6833     AssignmentOp = ActOnFinishFullExpr(AssignmentOp.get(), DE->getExprLoc(),
6834                                        /*DiscardedValue=*/true);
6835     if (AssignmentOp.isInvalid())
6836       continue;
6837 
6838     // OpenMP 4.5 [2.10.8, Distribute Construct, p.3]
6839     // A list item may appear in a firstprivate or lastprivate clause but not
6840     // both.
6841     if (CurrDir == OMPD_distribute) {
6842       DSAStackTy::DSAVarData DVar = DSAStack->getTopDSA(VD, false);
6843       if (DVar.CKind == OMPC_firstprivate) {
6844         Diag(ELoc, diag::err_omp_firstprivate_and_lastprivate_in_distribute);
6845         ReportOriginalDSA(*this, DSAStack, VD, DVar);
6846         continue;
6847       }
6848     }
6849 
6850     if (TopDVar.CKind != OMPC_firstprivate)
6851       DSAStack->addDSA(VD, DE, OMPC_lastprivate);
6852     Vars.push_back(DE);
6853     SrcExprs.push_back(PseudoSrcExpr);
6854     DstExprs.push_back(PseudoDstExpr);
6855     AssignmentOps.push_back(AssignmentOp.get());
6856   }
6857 
6858   if (Vars.empty())
6859     return nullptr;
6860 
6861   return OMPLastprivateClause::Create(Context, StartLoc, LParenLoc, EndLoc,
6862                                       Vars, SrcExprs, DstExprs, AssignmentOps);
6863 }
6864 
ActOnOpenMPSharedClause(ArrayRef<Expr * > VarList,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc)6865 OMPClause *Sema::ActOnOpenMPSharedClause(ArrayRef<Expr *> VarList,
6866                                          SourceLocation StartLoc,
6867                                          SourceLocation LParenLoc,
6868                                          SourceLocation EndLoc) {
6869   SmallVector<Expr *, 8> Vars;
6870   for (auto &RefExpr : VarList) {
6871     assert(RefExpr && "NULL expr in OpenMP shared clause.");
6872     if (isa<DependentScopeDeclRefExpr>(RefExpr)) {
6873       // It will be analyzed later.
6874       Vars.push_back(RefExpr);
6875       continue;
6876     }
6877 
6878     SourceLocation ELoc = RefExpr->getExprLoc();
6879     // OpenMP [2.1, C/C++]
6880     //  A list item is a variable name.
6881     // OpenMP  [2.14.3.2, Restrictions, p.1]
6882     //  A variable that is part of another variable (as an array or structure
6883     //  element) cannot appear in a shared unless it is a static data member
6884     //  of a C++ class.
6885     DeclRefExpr *DE = dyn_cast<DeclRefExpr>(RefExpr);
6886     if (!DE || !isa<VarDecl>(DE->getDecl())) {
6887       Diag(ELoc, diag::err_omp_expected_var_name) << RefExpr->getSourceRange();
6888       continue;
6889     }
6890     Decl *D = DE->getDecl();
6891     VarDecl *VD = cast<VarDecl>(D);
6892 
6893     QualType Type = VD->getType();
6894     if (Type->isDependentType() || Type->isInstantiationDependentType()) {
6895       // It will be analyzed later.
6896       Vars.push_back(DE);
6897       continue;
6898     }
6899 
6900     // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced
6901     // in a Construct]
6902     //  Variables with the predetermined data-sharing attributes may not be
6903     //  listed in data-sharing attributes clauses, except for the cases
6904     //  listed below. For these exceptions only, listing a predetermined
6905     //  variable in a data-sharing attribute clause is allowed and overrides
6906     //  the variable's predetermined data-sharing attributes.
6907     DSAStackTy::DSAVarData DVar = DSAStack->getTopDSA(VD, false);
6908     if (DVar.CKind != OMPC_unknown && DVar.CKind != OMPC_shared &&
6909         DVar.RefExpr) {
6910       Diag(ELoc, diag::err_omp_wrong_dsa) << getOpenMPClauseName(DVar.CKind)
6911                                           << getOpenMPClauseName(OMPC_shared);
6912       ReportOriginalDSA(*this, DSAStack, VD, DVar);
6913       continue;
6914     }
6915 
6916     DSAStack->addDSA(VD, DE, OMPC_shared);
6917     Vars.push_back(DE);
6918   }
6919 
6920   if (Vars.empty())
6921     return nullptr;
6922 
6923   return OMPSharedClause::Create(Context, StartLoc, LParenLoc, EndLoc, Vars);
6924 }
6925 
6926 namespace {
6927 class DSARefChecker : public StmtVisitor<DSARefChecker, bool> {
6928   DSAStackTy *Stack;
6929 
6930 public:
VisitDeclRefExpr(DeclRefExpr * E)6931   bool VisitDeclRefExpr(DeclRefExpr *E) {
6932     if (VarDecl *VD = dyn_cast<VarDecl>(E->getDecl())) {
6933       DSAStackTy::DSAVarData DVar = Stack->getTopDSA(VD, false);
6934       if (DVar.CKind == OMPC_shared && !DVar.RefExpr)
6935         return false;
6936       if (DVar.CKind != OMPC_unknown)
6937         return true;
6938       DSAStackTy::DSAVarData DVarPrivate =
6939           Stack->hasDSA(VD, isOpenMPPrivate, MatchesAlways(), false);
6940       if (DVarPrivate.CKind != OMPC_unknown)
6941         return true;
6942       return false;
6943     }
6944     return false;
6945   }
VisitStmt(Stmt * S)6946   bool VisitStmt(Stmt *S) {
6947     for (auto Child : S->children()) {
6948       if (Child && Visit(Child))
6949         return true;
6950     }
6951     return false;
6952   }
DSARefChecker(DSAStackTy * S)6953   explicit DSARefChecker(DSAStackTy *S) : Stack(S) {}
6954 };
6955 } // namespace
6956 
ActOnOpenMPReductionClause(ArrayRef<Expr * > VarList,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation ColonLoc,SourceLocation EndLoc,CXXScopeSpec & ReductionIdScopeSpec,const DeclarationNameInfo & ReductionId)6957 OMPClause *Sema::ActOnOpenMPReductionClause(
6958     ArrayRef<Expr *> VarList, SourceLocation StartLoc, SourceLocation LParenLoc,
6959     SourceLocation ColonLoc, SourceLocation EndLoc,
6960     CXXScopeSpec &ReductionIdScopeSpec,
6961     const DeclarationNameInfo &ReductionId) {
6962   // TODO: Allow scope specification search when 'declare reduction' is
6963   // supported.
6964   assert(ReductionIdScopeSpec.isEmpty() &&
6965          "No support for scoped reduction identifiers yet.");
6966 
6967   auto DN = ReductionId.getName();
6968   auto OOK = DN.getCXXOverloadedOperator();
6969   BinaryOperatorKind BOK = BO_Comma;
6970 
6971   // OpenMP [2.14.3.6, reduction clause]
6972   // C
6973   // reduction-identifier is either an identifier or one of the following
6974   // operators: +, -, *,  &, |, ^, && and ||
6975   // C++
6976   // reduction-identifier is either an id-expression or one of the following
6977   // operators: +, -, *, &, |, ^, && and ||
6978   // FIXME: Only 'min' and 'max' identifiers are supported for now.
6979   switch (OOK) {
6980   case OO_Plus:
6981   case OO_Minus:
6982     BOK = BO_Add;
6983     break;
6984   case OO_Star:
6985     BOK = BO_Mul;
6986     break;
6987   case OO_Amp:
6988     BOK = BO_And;
6989     break;
6990   case OO_Pipe:
6991     BOK = BO_Or;
6992     break;
6993   case OO_Caret:
6994     BOK = BO_Xor;
6995     break;
6996   case OO_AmpAmp:
6997     BOK = BO_LAnd;
6998     break;
6999   case OO_PipePipe:
7000     BOK = BO_LOr;
7001     break;
7002   case OO_New:
7003   case OO_Delete:
7004   case OO_Array_New:
7005   case OO_Array_Delete:
7006   case OO_Slash:
7007   case OO_Percent:
7008   case OO_Tilde:
7009   case OO_Exclaim:
7010   case OO_Equal:
7011   case OO_Less:
7012   case OO_Greater:
7013   case OO_LessEqual:
7014   case OO_GreaterEqual:
7015   case OO_PlusEqual:
7016   case OO_MinusEqual:
7017   case OO_StarEqual:
7018   case OO_SlashEqual:
7019   case OO_PercentEqual:
7020   case OO_CaretEqual:
7021   case OO_AmpEqual:
7022   case OO_PipeEqual:
7023   case OO_LessLess:
7024   case OO_GreaterGreater:
7025   case OO_LessLessEqual:
7026   case OO_GreaterGreaterEqual:
7027   case OO_EqualEqual:
7028   case OO_ExclaimEqual:
7029   case OO_PlusPlus:
7030   case OO_MinusMinus:
7031   case OO_Comma:
7032   case OO_ArrowStar:
7033   case OO_Arrow:
7034   case OO_Call:
7035   case OO_Subscript:
7036   case OO_Conditional:
7037   case OO_Coawait:
7038   case NUM_OVERLOADED_OPERATORS:
7039     llvm_unreachable("Unexpected reduction identifier");
7040   case OO_None:
7041     if (auto II = DN.getAsIdentifierInfo()) {
7042       if (II->isStr("max"))
7043         BOK = BO_GT;
7044       else if (II->isStr("min"))
7045         BOK = BO_LT;
7046     }
7047     break;
7048   }
7049   SourceRange ReductionIdRange;
7050   if (ReductionIdScopeSpec.isValid()) {
7051     ReductionIdRange.setBegin(ReductionIdScopeSpec.getBeginLoc());
7052   }
7053   ReductionIdRange.setEnd(ReductionId.getEndLoc());
7054   if (BOK == BO_Comma) {
7055     // Not allowed reduction identifier is found.
7056     Diag(ReductionId.getLocStart(), diag::err_omp_unknown_reduction_identifier)
7057         << ReductionIdRange;
7058     return nullptr;
7059   }
7060 
7061   SmallVector<Expr *, 8> Vars;
7062   SmallVector<Expr *, 8> Privates;
7063   SmallVector<Expr *, 8> LHSs;
7064   SmallVector<Expr *, 8> RHSs;
7065   SmallVector<Expr *, 8> ReductionOps;
7066   for (auto RefExpr : VarList) {
7067     assert(RefExpr && "nullptr expr in OpenMP reduction clause.");
7068     if (isa<DependentScopeDeclRefExpr>(RefExpr)) {
7069       // It will be analyzed later.
7070       Vars.push_back(RefExpr);
7071       Privates.push_back(nullptr);
7072       LHSs.push_back(nullptr);
7073       RHSs.push_back(nullptr);
7074       ReductionOps.push_back(nullptr);
7075       continue;
7076     }
7077 
7078     if (RefExpr->isTypeDependent() || RefExpr->isValueDependent() ||
7079         RefExpr->isInstantiationDependent() ||
7080         RefExpr->containsUnexpandedParameterPack()) {
7081       // It will be analyzed later.
7082       Vars.push_back(RefExpr);
7083       Privates.push_back(nullptr);
7084       LHSs.push_back(nullptr);
7085       RHSs.push_back(nullptr);
7086       ReductionOps.push_back(nullptr);
7087       continue;
7088     }
7089 
7090     auto ELoc = RefExpr->getExprLoc();
7091     auto ERange = RefExpr->getSourceRange();
7092     // OpenMP [2.1, C/C++]
7093     //  A list item is a variable or array section, subject to the restrictions
7094     //  specified in Section 2.4 on page 42 and in each of the sections
7095     // describing clauses and directives for which a list appears.
7096     // OpenMP  [2.14.3.3, Restrictions, p.1]
7097     //  A variable that is part of another variable (as an array or
7098     //  structure element) cannot appear in a private clause.
7099     auto *DE = dyn_cast<DeclRefExpr>(RefExpr);
7100     auto *ASE = dyn_cast<ArraySubscriptExpr>(RefExpr);
7101     auto *OASE = dyn_cast<OMPArraySectionExpr>(RefExpr);
7102     if (!ASE && !OASE && (!DE || !isa<VarDecl>(DE->getDecl()))) {
7103       Diag(ELoc, diag::err_omp_expected_var_name_or_array_item) << ERange;
7104       continue;
7105     }
7106     QualType Type;
7107     VarDecl *VD = nullptr;
7108     if (DE) {
7109       auto D = DE->getDecl();
7110       VD = cast<VarDecl>(D);
7111       Type = VD->getType();
7112     } else if (ASE) {
7113       Type = ASE->getType();
7114       auto *Base = ASE->getBase()->IgnoreParenImpCasts();
7115       while (auto *TempASE = dyn_cast<ArraySubscriptExpr>(Base))
7116         Base = TempASE->getBase()->IgnoreParenImpCasts();
7117       DE = dyn_cast<DeclRefExpr>(Base);
7118       if (DE)
7119         VD = dyn_cast<VarDecl>(DE->getDecl());
7120       if (!VD) {
7121         Diag(Base->getExprLoc(), diag::err_omp_expected_base_var_name)
7122             << 0 << Base->getSourceRange();
7123         continue;
7124       }
7125     } else if (OASE) {
7126       auto BaseType = OMPArraySectionExpr::getBaseOriginalType(OASE->getBase());
7127       if (auto *ATy = BaseType->getAsArrayTypeUnsafe())
7128         Type = ATy->getElementType();
7129       else
7130         Type = BaseType->getPointeeType();
7131       auto *Base = OASE->getBase()->IgnoreParenImpCasts();
7132       while (auto *TempOASE = dyn_cast<OMPArraySectionExpr>(Base))
7133         Base = TempOASE->getBase()->IgnoreParenImpCasts();
7134       while (auto *TempASE = dyn_cast<ArraySubscriptExpr>(Base))
7135         Base = TempASE->getBase()->IgnoreParenImpCasts();
7136       DE = dyn_cast<DeclRefExpr>(Base);
7137       if (DE)
7138         VD = dyn_cast<VarDecl>(DE->getDecl());
7139       if (!VD) {
7140         Diag(Base->getExprLoc(), diag::err_omp_expected_base_var_name)
7141             << 1 << Base->getSourceRange();
7142         continue;
7143       }
7144     }
7145 
7146     // OpenMP [2.9.3.3, Restrictions, C/C++, p.3]
7147     //  A variable that appears in a private clause must not have an incomplete
7148     //  type or a reference type.
7149     if (RequireCompleteType(ELoc, Type,
7150                             diag::err_omp_reduction_incomplete_type))
7151       continue;
7152     // OpenMP [2.14.3.6, reduction clause, Restrictions]
7153     // Arrays may not appear in a reduction clause.
7154     if (Type.getNonReferenceType()->isArrayType()) {
7155       Diag(ELoc, diag::err_omp_reduction_type_array) << Type << ERange;
7156       if (!ASE && !OASE) {
7157         bool IsDecl = VD->isThisDeclarationADefinition(Context) ==
7158                       VarDecl::DeclarationOnly;
7159         Diag(VD->getLocation(),
7160              IsDecl ? diag::note_previous_decl : diag::note_defined_here)
7161             << VD;
7162       }
7163       continue;
7164     }
7165     // OpenMP [2.14.3.6, reduction clause, Restrictions]
7166     // A list item that appears in a reduction clause must not be
7167     // const-qualified.
7168     if (Type.getNonReferenceType().isConstant(Context)) {
7169       Diag(ELoc, diag::err_omp_const_reduction_list_item)
7170           << getOpenMPClauseName(OMPC_reduction) << Type << ERange;
7171       if (!ASE && !OASE) {
7172         bool IsDecl = VD->isThisDeclarationADefinition(Context) ==
7173                       VarDecl::DeclarationOnly;
7174         Diag(VD->getLocation(),
7175              IsDecl ? diag::note_previous_decl : diag::note_defined_here)
7176             << VD;
7177       }
7178       continue;
7179     }
7180     // OpenMP [2.9.3.6, Restrictions, C/C++, p.4]
7181     //  If a list-item is a reference type then it must bind to the same object
7182     //  for all threads of the team.
7183     if (!ASE && !OASE) {
7184       VarDecl *VDDef = VD->getDefinition();
7185       if (Type->isReferenceType() && VDDef) {
7186         DSARefChecker Check(DSAStack);
7187         if (Check.Visit(VDDef->getInit())) {
7188           Diag(ELoc, diag::err_omp_reduction_ref_type_arg) << ERange;
7189           Diag(VDDef->getLocation(), diag::note_defined_here) << VDDef;
7190           continue;
7191         }
7192       }
7193     }
7194     // OpenMP [2.14.3.6, reduction clause, Restrictions]
7195     // The type of a list item that appears in a reduction clause must be valid
7196     // for the reduction-identifier. For a max or min reduction in C, the type
7197     // of the list item must be an allowed arithmetic data type: char, int,
7198     // float, double, or _Bool, possibly modified with long, short, signed, or
7199     // unsigned. For a max or min reduction in C++, the type of the list item
7200     // must be an allowed arithmetic data type: char, wchar_t, int, float,
7201     // double, or bool, possibly modified with long, short, signed, or unsigned.
7202     if ((BOK == BO_GT || BOK == BO_LT) &&
7203         !(Type->isScalarType() ||
7204           (getLangOpts().CPlusPlus && Type->isArithmeticType()))) {
7205       Diag(ELoc, diag::err_omp_clause_not_arithmetic_type_arg)
7206           << getLangOpts().CPlusPlus;
7207       if (!ASE && !OASE) {
7208         bool IsDecl = VD->isThisDeclarationADefinition(Context) ==
7209                       VarDecl::DeclarationOnly;
7210         Diag(VD->getLocation(),
7211              IsDecl ? diag::note_previous_decl : diag::note_defined_here)
7212             << VD;
7213       }
7214       continue;
7215     }
7216     if ((BOK == BO_OrAssign || BOK == BO_AndAssign || BOK == BO_XorAssign) &&
7217         !getLangOpts().CPlusPlus && Type->isFloatingType()) {
7218       Diag(ELoc, diag::err_omp_clause_floating_type_arg);
7219       if (!ASE && !OASE) {
7220         bool IsDecl = VD->isThisDeclarationADefinition(Context) ==
7221                       VarDecl::DeclarationOnly;
7222         Diag(VD->getLocation(),
7223              IsDecl ? diag::note_previous_decl : diag::note_defined_here)
7224             << VD;
7225       }
7226       continue;
7227     }
7228     // OpenMP [2.14.1.1, Data-sharing Attribute Rules for Variables Referenced
7229     // in a Construct]
7230     //  Variables with the predetermined data-sharing attributes may not be
7231     //  listed in data-sharing attributes clauses, except for the cases
7232     //  listed below. For these exceptions only, listing a predetermined
7233     //  variable in a data-sharing attribute clause is allowed and overrides
7234     //  the variable's predetermined data-sharing attributes.
7235     // OpenMP [2.14.3.6, Restrictions, p.3]
7236     //  Any number of reduction clauses can be specified on the directive,
7237     //  but a list item can appear only once in the reduction clauses for that
7238     //  directive.
7239     DSAStackTy::DSAVarData DVar;
7240     DVar = DSAStack->getTopDSA(VD, false);
7241     if (DVar.CKind == OMPC_reduction) {
7242       Diag(ELoc, diag::err_omp_once_referenced)
7243           << getOpenMPClauseName(OMPC_reduction);
7244       if (DVar.RefExpr) {
7245         Diag(DVar.RefExpr->getExprLoc(), diag::note_omp_referenced);
7246       }
7247     } else if (DVar.CKind != OMPC_unknown) {
7248       Diag(ELoc, diag::err_omp_wrong_dsa)
7249           << getOpenMPClauseName(DVar.CKind)
7250           << getOpenMPClauseName(OMPC_reduction);
7251       ReportOriginalDSA(*this, DSAStack, VD, DVar);
7252       continue;
7253     }
7254 
7255     // OpenMP [2.14.3.6, Restrictions, p.1]
7256     //  A list item that appears in a reduction clause of a worksharing
7257     //  construct must be shared in the parallel regions to which any of the
7258     //  worksharing regions arising from the worksharing construct bind.
7259     OpenMPDirectiveKind CurrDir = DSAStack->getCurrentDirective();
7260     if (isOpenMPWorksharingDirective(CurrDir) &&
7261         !isOpenMPParallelDirective(CurrDir)) {
7262       DVar = DSAStack->getImplicitDSA(VD, true);
7263       if (DVar.CKind != OMPC_shared) {
7264         Diag(ELoc, diag::err_omp_required_access)
7265             << getOpenMPClauseName(OMPC_reduction)
7266             << getOpenMPClauseName(OMPC_shared);
7267         ReportOriginalDSA(*this, DSAStack, VD, DVar);
7268         continue;
7269       }
7270     }
7271 
7272     Type = Type.getNonLValueExprType(Context).getUnqualifiedType();
7273     auto *LHSVD = buildVarDecl(*this, ELoc, Type, ".reduction.lhs",
7274                                VD->hasAttrs() ? &VD->getAttrs() : nullptr);
7275     auto *RHSVD = buildVarDecl(*this, ELoc, Type, VD->getName(),
7276                                VD->hasAttrs() ? &VD->getAttrs() : nullptr);
7277     auto PrivateTy = Type;
7278     if (OASE) {
7279       // For array sections only:
7280       // Create pseudo array type for private copy. The size for this array will
7281       // be generated during codegen.
7282       // For array subscripts or single variables Private Ty is the same as Type
7283       // (type of the variable or single array element).
7284       PrivateTy = Context.getVariableArrayType(
7285           Type, new (Context) OpaqueValueExpr(SourceLocation(),
7286                                               Context.getSizeType(), VK_RValue),
7287           ArrayType::Normal, /*IndexTypeQuals=*/0, SourceRange());
7288     }
7289     // Private copy.
7290     auto *PrivateVD = buildVarDecl(*this, ELoc, PrivateTy, VD->getName(),
7291                                    VD->hasAttrs() ? &VD->getAttrs() : nullptr);
7292     // Add initializer for private variable.
7293     Expr *Init = nullptr;
7294     switch (BOK) {
7295     case BO_Add:
7296     case BO_Xor:
7297     case BO_Or:
7298     case BO_LOr:
7299       // '+', '-', '^', '|', '||' reduction ops - initializer is '0'.
7300       if (Type->isScalarType() || Type->isAnyComplexType()) {
7301         Init = ActOnIntegerConstant(ELoc, /*Val=*/0).get();
7302       }
7303       break;
7304     case BO_Mul:
7305     case BO_LAnd:
7306       if (Type->isScalarType() || Type->isAnyComplexType()) {
7307         // '*' and '&&' reduction ops - initializer is '1'.
7308         Init = ActOnIntegerConstant(ELoc, /*Val=*/1).get();
7309       }
7310       break;
7311     case BO_And: {
7312       // '&' reduction op - initializer is '~0'.
7313       QualType OrigType = Type;
7314       if (auto *ComplexTy = OrigType->getAs<ComplexType>()) {
7315         Type = ComplexTy->getElementType();
7316       }
7317       if (Type->isRealFloatingType()) {
7318         llvm::APFloat InitValue =
7319             llvm::APFloat::getAllOnesValue(Context.getTypeSize(Type),
7320                                            /*isIEEE=*/true);
7321         Init = FloatingLiteral::Create(Context, InitValue, /*isexact=*/true,
7322                                        Type, ELoc);
7323       } else if (Type->isScalarType()) {
7324         auto Size = Context.getTypeSize(Type);
7325         QualType IntTy = Context.getIntTypeForBitwidth(Size, /*Signed=*/0);
7326         llvm::APInt InitValue = llvm::APInt::getAllOnesValue(Size);
7327         Init = IntegerLiteral::Create(Context, InitValue, IntTy, ELoc);
7328       }
7329       if (Init && OrigType->isAnyComplexType()) {
7330         // Init = 0xFFFF + 0xFFFFi;
7331         auto *Im = new (Context) ImaginaryLiteral(Init, OrigType);
7332         Init = CreateBuiltinBinOp(ELoc, BO_Add, Init, Im).get();
7333       }
7334       Type = OrigType;
7335       break;
7336     }
7337     case BO_LT:
7338     case BO_GT: {
7339       // 'min' reduction op - initializer is 'Largest representable number in
7340       // the reduction list item type'.
7341       // 'max' reduction op - initializer is 'Least representable number in
7342       // the reduction list item type'.
7343       if (Type->isIntegerType() || Type->isPointerType()) {
7344         bool IsSigned = Type->hasSignedIntegerRepresentation();
7345         auto Size = Context.getTypeSize(Type);
7346         QualType IntTy =
7347             Context.getIntTypeForBitwidth(Size, /*Signed=*/IsSigned);
7348         llvm::APInt InitValue =
7349             (BOK != BO_LT)
7350                 ? IsSigned ? llvm::APInt::getSignedMinValue(Size)
7351                            : llvm::APInt::getMinValue(Size)
7352                 : IsSigned ? llvm::APInt::getSignedMaxValue(Size)
7353                            : llvm::APInt::getMaxValue(Size);
7354         Init = IntegerLiteral::Create(Context, InitValue, IntTy, ELoc);
7355         if (Type->isPointerType()) {
7356           // Cast to pointer type.
7357           auto CastExpr = BuildCStyleCastExpr(
7358               SourceLocation(), Context.getTrivialTypeSourceInfo(Type, ELoc),
7359               SourceLocation(), Init);
7360           if (CastExpr.isInvalid())
7361             continue;
7362           Init = CastExpr.get();
7363         }
7364       } else if (Type->isRealFloatingType()) {
7365         llvm::APFloat InitValue = llvm::APFloat::getLargest(
7366             Context.getFloatTypeSemantics(Type), BOK != BO_LT);
7367         Init = FloatingLiteral::Create(Context, InitValue, /*isexact=*/true,
7368                                        Type, ELoc);
7369       }
7370       break;
7371     }
7372     case BO_PtrMemD:
7373     case BO_PtrMemI:
7374     case BO_MulAssign:
7375     case BO_Div:
7376     case BO_Rem:
7377     case BO_Sub:
7378     case BO_Shl:
7379     case BO_Shr:
7380     case BO_LE:
7381     case BO_GE:
7382     case BO_EQ:
7383     case BO_NE:
7384     case BO_AndAssign:
7385     case BO_XorAssign:
7386     case BO_OrAssign:
7387     case BO_Assign:
7388     case BO_AddAssign:
7389     case BO_SubAssign:
7390     case BO_DivAssign:
7391     case BO_RemAssign:
7392     case BO_ShlAssign:
7393     case BO_ShrAssign:
7394     case BO_Comma:
7395       llvm_unreachable("Unexpected reduction operation");
7396     }
7397     if (Init) {
7398       AddInitializerToDecl(RHSVD, Init, /*DirectInit=*/false,
7399                            /*TypeMayContainAuto=*/false);
7400     } else
7401       ActOnUninitializedDecl(RHSVD, /*TypeMayContainAuto=*/false);
7402     if (!RHSVD->hasInit()) {
7403       Diag(ELoc, diag::err_omp_reduction_id_not_compatible) << Type
7404                                                             << ReductionIdRange;
7405       if (VD) {
7406         bool IsDecl = VD->isThisDeclarationADefinition(Context) ==
7407                       VarDecl::DeclarationOnly;
7408         Diag(VD->getLocation(),
7409              IsDecl ? diag::note_previous_decl : diag::note_defined_here)
7410             << VD;
7411       }
7412       continue;
7413     }
7414     // Store initializer for single element in private copy. Will be used during
7415     // codegen.
7416     PrivateVD->setInit(RHSVD->getInit());
7417     PrivateVD->setInitStyle(RHSVD->getInitStyle());
7418     auto *LHSDRE = buildDeclRefExpr(*this, LHSVD, Type, ELoc);
7419     auto *RHSDRE = buildDeclRefExpr(*this, RHSVD, Type, ELoc);
7420     auto *PrivateDRE = buildDeclRefExpr(*this, PrivateVD, PrivateTy, ELoc);
7421     ExprResult ReductionOp =
7422         BuildBinOp(DSAStack->getCurScope(), ReductionId.getLocStart(), BOK,
7423                    LHSDRE, RHSDRE);
7424     if (ReductionOp.isUsable()) {
7425       if (BOK != BO_LT && BOK != BO_GT) {
7426         ReductionOp =
7427             BuildBinOp(DSAStack->getCurScope(), ReductionId.getLocStart(),
7428                        BO_Assign, LHSDRE, ReductionOp.get());
7429       } else {
7430         auto *ConditionalOp = new (Context) ConditionalOperator(
7431             ReductionOp.get(), SourceLocation(), LHSDRE, SourceLocation(),
7432             RHSDRE, Type, VK_LValue, OK_Ordinary);
7433         ReductionOp =
7434             BuildBinOp(DSAStack->getCurScope(), ReductionId.getLocStart(),
7435                        BO_Assign, LHSDRE, ConditionalOp);
7436       }
7437       ReductionOp = ActOnFinishFullExpr(ReductionOp.get());
7438     }
7439     if (ReductionOp.isInvalid())
7440       continue;
7441 
7442     DSAStack->addDSA(VD, DE, OMPC_reduction);
7443     Vars.push_back(RefExpr);
7444     Privates.push_back(PrivateDRE);
7445     LHSs.push_back(LHSDRE);
7446     RHSs.push_back(RHSDRE);
7447     ReductionOps.push_back(ReductionOp.get());
7448   }
7449 
7450   if (Vars.empty())
7451     return nullptr;
7452 
7453   return OMPReductionClause::Create(
7454       Context, StartLoc, LParenLoc, ColonLoc, EndLoc, Vars,
7455       ReductionIdScopeSpec.getWithLocInContext(Context), ReductionId, Privates,
7456       LHSs, RHSs, ReductionOps);
7457 }
7458 
ActOnOpenMPLinearClause(ArrayRef<Expr * > VarList,Expr * Step,SourceLocation StartLoc,SourceLocation LParenLoc,OpenMPLinearClauseKind LinKind,SourceLocation LinLoc,SourceLocation ColonLoc,SourceLocation EndLoc)7459 OMPClause *Sema::ActOnOpenMPLinearClause(
7460     ArrayRef<Expr *> VarList, Expr *Step, SourceLocation StartLoc,
7461     SourceLocation LParenLoc, OpenMPLinearClauseKind LinKind,
7462     SourceLocation LinLoc, SourceLocation ColonLoc, SourceLocation EndLoc) {
7463   SmallVector<Expr *, 8> Vars;
7464   SmallVector<Expr *, 8> Privates;
7465   SmallVector<Expr *, 8> Inits;
7466   if ((!LangOpts.CPlusPlus && LinKind != OMPC_LINEAR_val) ||
7467       LinKind == OMPC_LINEAR_unknown) {
7468     Diag(LinLoc, diag::err_omp_wrong_linear_modifier) << LangOpts.CPlusPlus;
7469     LinKind = OMPC_LINEAR_val;
7470   }
7471   for (auto &RefExpr : VarList) {
7472     assert(RefExpr && "NULL expr in OpenMP linear clause.");
7473     if (isa<DependentScopeDeclRefExpr>(RefExpr)) {
7474       // It will be analyzed later.
7475       Vars.push_back(RefExpr);
7476       Privates.push_back(nullptr);
7477       Inits.push_back(nullptr);
7478       continue;
7479     }
7480 
7481     // OpenMP [2.14.3.7, linear clause]
7482     // A list item that appears in a linear clause is subject to the private
7483     // clause semantics described in Section 2.14.3.3 on page 159 except as
7484     // noted. In addition, the value of the new list item on each iteration
7485     // of the associated loop(s) corresponds to the value of the original
7486     // list item before entering the construct plus the logical number of
7487     // the iteration times linear-step.
7488 
7489     SourceLocation ELoc = RefExpr->getExprLoc();
7490     // OpenMP [2.1, C/C++]
7491     //  A list item is a variable name.
7492     // OpenMP  [2.14.3.3, Restrictions, p.1]
7493     //  A variable that is part of another variable (as an array or
7494     //  structure element) cannot appear in a private clause.
7495     DeclRefExpr *DE = dyn_cast<DeclRefExpr>(RefExpr);
7496     if (!DE || !isa<VarDecl>(DE->getDecl())) {
7497       Diag(ELoc, diag::err_omp_expected_var_name) << RefExpr->getSourceRange();
7498       continue;
7499     }
7500 
7501     VarDecl *VD = cast<VarDecl>(DE->getDecl());
7502 
7503     // OpenMP [2.14.3.7, linear clause]
7504     //  A list-item cannot appear in more than one linear clause.
7505     //  A list-item that appears in a linear clause cannot appear in any
7506     //  other data-sharing attribute clause.
7507     DSAStackTy::DSAVarData DVar = DSAStack->getTopDSA(VD, false);
7508     if (DVar.RefExpr) {
7509       Diag(ELoc, diag::err_omp_wrong_dsa) << getOpenMPClauseName(DVar.CKind)
7510                                           << getOpenMPClauseName(OMPC_linear);
7511       ReportOriginalDSA(*this, DSAStack, VD, DVar);
7512       continue;
7513     }
7514 
7515     QualType QType = VD->getType();
7516     if (QType->isDependentType() || QType->isInstantiationDependentType()) {
7517       // It will be analyzed later.
7518       Vars.push_back(DE);
7519       Privates.push_back(nullptr);
7520       Inits.push_back(nullptr);
7521       continue;
7522     }
7523 
7524     // A variable must not have an incomplete type or a reference type.
7525     if (RequireCompleteType(ELoc, QType,
7526                             diag::err_omp_linear_incomplete_type)) {
7527       continue;
7528     }
7529     if ((LinKind == OMPC_LINEAR_uval || LinKind == OMPC_LINEAR_ref) &&
7530         !QType->isReferenceType()) {
7531       Diag(ELoc, diag::err_omp_wrong_linear_modifier_non_reference)
7532           << QType << getOpenMPSimpleClauseTypeName(OMPC_linear, LinKind);
7533       continue;
7534     }
7535     QType = QType.getNonReferenceType();
7536 
7537     // A list item must not be const-qualified.
7538     if (QType.isConstant(Context)) {
7539       Diag(ELoc, diag::err_omp_const_variable)
7540           << getOpenMPClauseName(OMPC_linear);
7541       bool IsDecl =
7542           VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly;
7543       Diag(VD->getLocation(),
7544            IsDecl ? diag::note_previous_decl : diag::note_defined_here)
7545           << VD;
7546       continue;
7547     }
7548 
7549     // A list item must be of integral or pointer type.
7550     QType = QType.getUnqualifiedType().getCanonicalType();
7551     const Type *Ty = QType.getTypePtrOrNull();
7552     if (!Ty || (!Ty->isDependentType() && !Ty->isIntegralType(Context) &&
7553                 !Ty->isPointerType())) {
7554       Diag(ELoc, diag::err_omp_linear_expected_int_or_ptr) << QType;
7555       bool IsDecl =
7556           VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly;
7557       Diag(VD->getLocation(),
7558            IsDecl ? diag::note_previous_decl : diag::note_defined_here)
7559           << VD;
7560       continue;
7561     }
7562 
7563     // Build private copy of original var.
7564     auto *Private = buildVarDecl(*this, ELoc, QType, VD->getName(),
7565                                  VD->hasAttrs() ? &VD->getAttrs() : nullptr);
7566     auto *PrivateRef = buildDeclRefExpr(
7567         *this, Private, DE->getType().getUnqualifiedType(), DE->getExprLoc());
7568     // Build var to save initial value.
7569     VarDecl *Init = buildVarDecl(*this, ELoc, QType, ".linear.start");
7570     Expr *InitExpr;
7571     if (LinKind == OMPC_LINEAR_uval)
7572       InitExpr = VD->getInit();
7573     else
7574       InitExpr = DE;
7575     AddInitializerToDecl(Init, DefaultLvalueConversion(InitExpr).get(),
7576                          /*DirectInit*/ false, /*TypeMayContainAuto*/ false);
7577     auto InitRef = buildDeclRefExpr(
7578         *this, Init, DE->getType().getUnqualifiedType(), DE->getExprLoc());
7579     DSAStack->addDSA(VD, DE, OMPC_linear);
7580     Vars.push_back(DE);
7581     Privates.push_back(PrivateRef);
7582     Inits.push_back(InitRef);
7583   }
7584 
7585   if (Vars.empty())
7586     return nullptr;
7587 
7588   Expr *StepExpr = Step;
7589   Expr *CalcStepExpr = nullptr;
7590   if (Step && !Step->isValueDependent() && !Step->isTypeDependent() &&
7591       !Step->isInstantiationDependent() &&
7592       !Step->containsUnexpandedParameterPack()) {
7593     SourceLocation StepLoc = Step->getLocStart();
7594     ExprResult Val = PerformOpenMPImplicitIntegerConversion(StepLoc, Step);
7595     if (Val.isInvalid())
7596       return nullptr;
7597     StepExpr = Val.get();
7598 
7599     // Build var to save the step value.
7600     VarDecl *SaveVar =
7601         buildVarDecl(*this, StepLoc, StepExpr->getType(), ".linear.step");
7602     ExprResult SaveRef =
7603         buildDeclRefExpr(*this, SaveVar, StepExpr->getType(), StepLoc);
7604     ExprResult CalcStep =
7605         BuildBinOp(CurScope, StepLoc, BO_Assign, SaveRef.get(), StepExpr);
7606     CalcStep = ActOnFinishFullExpr(CalcStep.get());
7607 
7608     // Warn about zero linear step (it would be probably better specified as
7609     // making corresponding variables 'const').
7610     llvm::APSInt Result;
7611     bool IsConstant = StepExpr->isIntegerConstantExpr(Result, Context);
7612     if (IsConstant && !Result.isNegative() && !Result.isStrictlyPositive())
7613       Diag(StepLoc, diag::warn_omp_linear_step_zero) << Vars[0]
7614                                                      << (Vars.size() > 1);
7615     if (!IsConstant && CalcStep.isUsable()) {
7616       // Calculate the step beforehand instead of doing this on each iteration.
7617       // (This is not used if the number of iterations may be kfold-ed).
7618       CalcStepExpr = CalcStep.get();
7619     }
7620   }
7621 
7622   return OMPLinearClause::Create(Context, StartLoc, LParenLoc, LinKind, LinLoc,
7623                                  ColonLoc, EndLoc, Vars, Privates, Inits,
7624                                  StepExpr, CalcStepExpr);
7625 }
7626 
FinishOpenMPLinearClause(OMPLinearClause & Clause,DeclRefExpr * IV,Expr * NumIterations,Sema & SemaRef,Scope * S)7627 static bool FinishOpenMPLinearClause(OMPLinearClause &Clause, DeclRefExpr *IV,
7628                                      Expr *NumIterations, Sema &SemaRef,
7629                                      Scope *S) {
7630   // Walk the vars and build update/final expressions for the CodeGen.
7631   SmallVector<Expr *, 8> Updates;
7632   SmallVector<Expr *, 8> Finals;
7633   Expr *Step = Clause.getStep();
7634   Expr *CalcStep = Clause.getCalcStep();
7635   // OpenMP [2.14.3.7, linear clause]
7636   // If linear-step is not specified it is assumed to be 1.
7637   if (Step == nullptr)
7638     Step = SemaRef.ActOnIntegerConstant(SourceLocation(), 1).get();
7639   else if (CalcStep)
7640     Step = cast<BinaryOperator>(CalcStep)->getLHS();
7641   bool HasErrors = false;
7642   auto CurInit = Clause.inits().begin();
7643   auto CurPrivate = Clause.privates().begin();
7644   auto LinKind = Clause.getModifier();
7645   for (auto &RefExpr : Clause.varlists()) {
7646     Expr *InitExpr = *CurInit;
7647 
7648     // Build privatized reference to the current linear var.
7649     auto DE = cast<DeclRefExpr>(RefExpr);
7650     Expr *CapturedRef;
7651     if (LinKind == OMPC_LINEAR_uval)
7652       CapturedRef = cast<VarDecl>(DE->getDecl())->getInit();
7653     else
7654       CapturedRef =
7655           buildDeclRefExpr(SemaRef, cast<VarDecl>(DE->getDecl()),
7656                            DE->getType().getUnqualifiedType(), DE->getExprLoc(),
7657                            /*RefersToCapture=*/true);
7658 
7659     // Build update: Var = InitExpr + IV * Step
7660     ExprResult Update =
7661         BuildCounterUpdate(SemaRef, S, RefExpr->getExprLoc(), *CurPrivate,
7662                            InitExpr, IV, Step, /* Subtract */ false);
7663     Update = SemaRef.ActOnFinishFullExpr(Update.get(), DE->getLocStart(),
7664                                          /*DiscardedValue=*/true);
7665 
7666     // Build final: Var = InitExpr + NumIterations * Step
7667     ExprResult Final =
7668         BuildCounterUpdate(SemaRef, S, RefExpr->getExprLoc(), CapturedRef,
7669                            InitExpr, NumIterations, Step, /* Subtract */ false);
7670     Final = SemaRef.ActOnFinishFullExpr(Final.get(), DE->getLocStart(),
7671                                         /*DiscardedValue=*/true);
7672     if (!Update.isUsable() || !Final.isUsable()) {
7673       Updates.push_back(nullptr);
7674       Finals.push_back(nullptr);
7675       HasErrors = true;
7676     } else {
7677       Updates.push_back(Update.get());
7678       Finals.push_back(Final.get());
7679     }
7680     ++CurInit, ++CurPrivate;
7681   }
7682   Clause.setUpdates(Updates);
7683   Clause.setFinals(Finals);
7684   return HasErrors;
7685 }
7686 
ActOnOpenMPAlignedClause(ArrayRef<Expr * > VarList,Expr * Alignment,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation ColonLoc,SourceLocation EndLoc)7687 OMPClause *Sema::ActOnOpenMPAlignedClause(
7688     ArrayRef<Expr *> VarList, Expr *Alignment, SourceLocation StartLoc,
7689     SourceLocation LParenLoc, SourceLocation ColonLoc, SourceLocation EndLoc) {
7690 
7691   SmallVector<Expr *, 8> Vars;
7692   for (auto &RefExpr : VarList) {
7693     assert(RefExpr && "NULL expr in OpenMP aligned clause.");
7694     if (isa<DependentScopeDeclRefExpr>(RefExpr)) {
7695       // It will be analyzed later.
7696       Vars.push_back(RefExpr);
7697       continue;
7698     }
7699 
7700     SourceLocation ELoc = RefExpr->getExprLoc();
7701     // OpenMP [2.1, C/C++]
7702     //  A list item is a variable name.
7703     DeclRefExpr *DE = dyn_cast<DeclRefExpr>(RefExpr);
7704     if (!DE || !isa<VarDecl>(DE->getDecl())) {
7705       Diag(ELoc, diag::err_omp_expected_var_name) << RefExpr->getSourceRange();
7706       continue;
7707     }
7708 
7709     VarDecl *VD = cast<VarDecl>(DE->getDecl());
7710 
7711     // OpenMP  [2.8.1, simd construct, Restrictions]
7712     // The type of list items appearing in the aligned clause must be
7713     // array, pointer, reference to array, or reference to pointer.
7714     QualType QType = VD->getType();
7715     QType = QType.getNonReferenceType().getUnqualifiedType().getCanonicalType();
7716     const Type *Ty = QType.getTypePtrOrNull();
7717     if (!Ty || (!Ty->isDependentType() && !Ty->isArrayType() &&
7718                 !Ty->isPointerType())) {
7719       Diag(ELoc, diag::err_omp_aligned_expected_array_or_ptr)
7720           << QType << getLangOpts().CPlusPlus << RefExpr->getSourceRange();
7721       bool IsDecl =
7722           VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly;
7723       Diag(VD->getLocation(),
7724            IsDecl ? diag::note_previous_decl : diag::note_defined_here)
7725           << VD;
7726       continue;
7727     }
7728 
7729     // OpenMP  [2.8.1, simd construct, Restrictions]
7730     // A list-item cannot appear in more than one aligned clause.
7731     if (DeclRefExpr *PrevRef = DSAStack->addUniqueAligned(VD, DE)) {
7732       Diag(ELoc, diag::err_omp_aligned_twice) << RefExpr->getSourceRange();
7733       Diag(PrevRef->getExprLoc(), diag::note_omp_explicit_dsa)
7734           << getOpenMPClauseName(OMPC_aligned);
7735       continue;
7736     }
7737 
7738     Vars.push_back(DE);
7739   }
7740 
7741   // OpenMP [2.8.1, simd construct, Description]
7742   // The parameter of the aligned clause, alignment, must be a constant
7743   // positive integer expression.
7744   // If no optional parameter is specified, implementation-defined default
7745   // alignments for SIMD instructions on the target platforms are assumed.
7746   if (Alignment != nullptr) {
7747     ExprResult AlignResult =
7748         VerifyPositiveIntegerConstantInClause(Alignment, OMPC_aligned);
7749     if (AlignResult.isInvalid())
7750       return nullptr;
7751     Alignment = AlignResult.get();
7752   }
7753   if (Vars.empty())
7754     return nullptr;
7755 
7756   return OMPAlignedClause::Create(Context, StartLoc, LParenLoc, ColonLoc,
7757                                   EndLoc, Vars, Alignment);
7758 }
7759 
ActOnOpenMPCopyinClause(ArrayRef<Expr * > VarList,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc)7760 OMPClause *Sema::ActOnOpenMPCopyinClause(ArrayRef<Expr *> VarList,
7761                                          SourceLocation StartLoc,
7762                                          SourceLocation LParenLoc,
7763                                          SourceLocation EndLoc) {
7764   SmallVector<Expr *, 8> Vars;
7765   SmallVector<Expr *, 8> SrcExprs;
7766   SmallVector<Expr *, 8> DstExprs;
7767   SmallVector<Expr *, 8> AssignmentOps;
7768   for (auto &RefExpr : VarList) {
7769     assert(RefExpr && "NULL expr in OpenMP copyin clause.");
7770     if (isa<DependentScopeDeclRefExpr>(RefExpr)) {
7771       // It will be analyzed later.
7772       Vars.push_back(RefExpr);
7773       SrcExprs.push_back(nullptr);
7774       DstExprs.push_back(nullptr);
7775       AssignmentOps.push_back(nullptr);
7776       continue;
7777     }
7778 
7779     SourceLocation ELoc = RefExpr->getExprLoc();
7780     // OpenMP [2.1, C/C++]
7781     //  A list item is a variable name.
7782     // OpenMP  [2.14.4.1, Restrictions, p.1]
7783     //  A list item that appears in a copyin clause must be threadprivate.
7784     DeclRefExpr *DE = dyn_cast<DeclRefExpr>(RefExpr);
7785     if (!DE || !isa<VarDecl>(DE->getDecl())) {
7786       Diag(ELoc, diag::err_omp_expected_var_name) << RefExpr->getSourceRange();
7787       continue;
7788     }
7789 
7790     Decl *D = DE->getDecl();
7791     VarDecl *VD = cast<VarDecl>(D);
7792 
7793     QualType Type = VD->getType();
7794     if (Type->isDependentType() || Type->isInstantiationDependentType()) {
7795       // It will be analyzed later.
7796       Vars.push_back(DE);
7797       SrcExprs.push_back(nullptr);
7798       DstExprs.push_back(nullptr);
7799       AssignmentOps.push_back(nullptr);
7800       continue;
7801     }
7802 
7803     // OpenMP [2.14.4.1, Restrictions, C/C++, p.1]
7804     //  A list item that appears in a copyin clause must be threadprivate.
7805     if (!DSAStack->isThreadPrivate(VD)) {
7806       Diag(ELoc, diag::err_omp_required_access)
7807           << getOpenMPClauseName(OMPC_copyin)
7808           << getOpenMPDirectiveName(OMPD_threadprivate);
7809       continue;
7810     }
7811 
7812     // OpenMP [2.14.4.1, Restrictions, C/C++, p.2]
7813     //  A variable of class type (or array thereof) that appears in a
7814     //  copyin clause requires an accessible, unambiguous copy assignment
7815     //  operator for the class type.
7816     auto ElemType = Context.getBaseElementType(Type).getNonReferenceType();
7817     auto *SrcVD =
7818         buildVarDecl(*this, DE->getLocStart(), ElemType.getUnqualifiedType(),
7819                      ".copyin.src", VD->hasAttrs() ? &VD->getAttrs() : nullptr);
7820     auto *PseudoSrcExpr = buildDeclRefExpr(
7821         *this, SrcVD, ElemType.getUnqualifiedType(), DE->getExprLoc());
7822     auto *DstVD =
7823         buildVarDecl(*this, DE->getLocStart(), ElemType, ".copyin.dst",
7824                      VD->hasAttrs() ? &VD->getAttrs() : nullptr);
7825     auto *PseudoDstExpr =
7826         buildDeclRefExpr(*this, DstVD, ElemType, DE->getExprLoc());
7827     // For arrays generate assignment operation for single element and replace
7828     // it by the original array element in CodeGen.
7829     auto AssignmentOp = BuildBinOp(/*S=*/nullptr, DE->getExprLoc(), BO_Assign,
7830                                    PseudoDstExpr, PseudoSrcExpr);
7831     if (AssignmentOp.isInvalid())
7832       continue;
7833     AssignmentOp = ActOnFinishFullExpr(AssignmentOp.get(), DE->getExprLoc(),
7834                                        /*DiscardedValue=*/true);
7835     if (AssignmentOp.isInvalid())
7836       continue;
7837 
7838     DSAStack->addDSA(VD, DE, OMPC_copyin);
7839     Vars.push_back(DE);
7840     SrcExprs.push_back(PseudoSrcExpr);
7841     DstExprs.push_back(PseudoDstExpr);
7842     AssignmentOps.push_back(AssignmentOp.get());
7843   }
7844 
7845   if (Vars.empty())
7846     return nullptr;
7847 
7848   return OMPCopyinClause::Create(Context, StartLoc, LParenLoc, EndLoc, Vars,
7849                                  SrcExprs, DstExprs, AssignmentOps);
7850 }
7851 
ActOnOpenMPCopyprivateClause(ArrayRef<Expr * > VarList,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc)7852 OMPClause *Sema::ActOnOpenMPCopyprivateClause(ArrayRef<Expr *> VarList,
7853                                               SourceLocation StartLoc,
7854                                               SourceLocation LParenLoc,
7855                                               SourceLocation EndLoc) {
7856   SmallVector<Expr *, 8> Vars;
7857   SmallVector<Expr *, 8> SrcExprs;
7858   SmallVector<Expr *, 8> DstExprs;
7859   SmallVector<Expr *, 8> AssignmentOps;
7860   for (auto &RefExpr : VarList) {
7861     assert(RefExpr && "NULL expr in OpenMP copyprivate clause.");
7862     if (isa<DependentScopeDeclRefExpr>(RefExpr)) {
7863       // It will be analyzed later.
7864       Vars.push_back(RefExpr);
7865       SrcExprs.push_back(nullptr);
7866       DstExprs.push_back(nullptr);
7867       AssignmentOps.push_back(nullptr);
7868       continue;
7869     }
7870 
7871     SourceLocation ELoc = RefExpr->getExprLoc();
7872     // OpenMP [2.1, C/C++]
7873     //  A list item is a variable name.
7874     // OpenMP  [2.14.4.1, Restrictions, p.1]
7875     //  A list item that appears in a copyin clause must be threadprivate.
7876     DeclRefExpr *DE = dyn_cast<DeclRefExpr>(RefExpr);
7877     if (!DE || !isa<VarDecl>(DE->getDecl())) {
7878       Diag(ELoc, diag::err_omp_expected_var_name) << RefExpr->getSourceRange();
7879       continue;
7880     }
7881 
7882     Decl *D = DE->getDecl();
7883     VarDecl *VD = cast<VarDecl>(D);
7884 
7885     QualType Type = VD->getType();
7886     if (Type->isDependentType() || Type->isInstantiationDependentType()) {
7887       // It will be analyzed later.
7888       Vars.push_back(DE);
7889       SrcExprs.push_back(nullptr);
7890       DstExprs.push_back(nullptr);
7891       AssignmentOps.push_back(nullptr);
7892       continue;
7893     }
7894 
7895     // OpenMP [2.14.4.2, Restrictions, p.2]
7896     //  A list item that appears in a copyprivate clause may not appear in a
7897     //  private or firstprivate clause on the single construct.
7898     if (!DSAStack->isThreadPrivate(VD)) {
7899       auto DVar = DSAStack->getTopDSA(VD, false);
7900       if (DVar.CKind != OMPC_unknown && DVar.CKind != OMPC_copyprivate &&
7901           DVar.RefExpr) {
7902         Diag(ELoc, diag::err_omp_wrong_dsa)
7903             << getOpenMPClauseName(DVar.CKind)
7904             << getOpenMPClauseName(OMPC_copyprivate);
7905         ReportOriginalDSA(*this, DSAStack, VD, DVar);
7906         continue;
7907       }
7908 
7909       // OpenMP [2.11.4.2, Restrictions, p.1]
7910       //  All list items that appear in a copyprivate clause must be either
7911       //  threadprivate or private in the enclosing context.
7912       if (DVar.CKind == OMPC_unknown) {
7913         DVar = DSAStack->getImplicitDSA(VD, false);
7914         if (DVar.CKind == OMPC_shared) {
7915           Diag(ELoc, diag::err_omp_required_access)
7916               << getOpenMPClauseName(OMPC_copyprivate)
7917               << "threadprivate or private in the enclosing context";
7918           ReportOriginalDSA(*this, DSAStack, VD, DVar);
7919           continue;
7920         }
7921       }
7922     }
7923 
7924     // Variably modified types are not supported.
7925     if (!Type->isAnyPointerType() && Type->isVariablyModifiedType()) {
7926       Diag(ELoc, diag::err_omp_variably_modified_type_not_supported)
7927           << getOpenMPClauseName(OMPC_copyprivate) << Type
7928           << getOpenMPDirectiveName(DSAStack->getCurrentDirective());
7929       bool IsDecl =
7930           VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly;
7931       Diag(VD->getLocation(),
7932            IsDecl ? diag::note_previous_decl : diag::note_defined_here)
7933           << VD;
7934       continue;
7935     }
7936 
7937     // OpenMP [2.14.4.1, Restrictions, C/C++, p.2]
7938     //  A variable of class type (or array thereof) that appears in a
7939     //  copyin clause requires an accessible, unambiguous copy assignment
7940     //  operator for the class type.
7941     Type = Context.getBaseElementType(Type.getNonReferenceType())
7942                .getUnqualifiedType();
7943     auto *SrcVD =
7944         buildVarDecl(*this, DE->getLocStart(), Type, ".copyprivate.src",
7945                      VD->hasAttrs() ? &VD->getAttrs() : nullptr);
7946     auto *PseudoSrcExpr =
7947         buildDeclRefExpr(*this, SrcVD, Type, DE->getExprLoc());
7948     auto *DstVD =
7949         buildVarDecl(*this, DE->getLocStart(), Type, ".copyprivate.dst",
7950                      VD->hasAttrs() ? &VD->getAttrs() : nullptr);
7951     auto *PseudoDstExpr =
7952         buildDeclRefExpr(*this, DstVD, Type, DE->getExprLoc());
7953     auto AssignmentOp = BuildBinOp(/*S=*/nullptr, DE->getExprLoc(), BO_Assign,
7954                                    PseudoDstExpr, PseudoSrcExpr);
7955     if (AssignmentOp.isInvalid())
7956       continue;
7957     AssignmentOp = ActOnFinishFullExpr(AssignmentOp.get(), DE->getExprLoc(),
7958                                        /*DiscardedValue=*/true);
7959     if (AssignmentOp.isInvalid())
7960       continue;
7961 
7962     // No need to mark vars as copyprivate, they are already threadprivate or
7963     // implicitly private.
7964     Vars.push_back(DE);
7965     SrcExprs.push_back(PseudoSrcExpr);
7966     DstExprs.push_back(PseudoDstExpr);
7967     AssignmentOps.push_back(AssignmentOp.get());
7968   }
7969 
7970   if (Vars.empty())
7971     return nullptr;
7972 
7973   return OMPCopyprivateClause::Create(Context, StartLoc, LParenLoc, EndLoc,
7974                                       Vars, SrcExprs, DstExprs, AssignmentOps);
7975 }
7976 
ActOnOpenMPFlushClause(ArrayRef<Expr * > VarList,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc)7977 OMPClause *Sema::ActOnOpenMPFlushClause(ArrayRef<Expr *> VarList,
7978                                         SourceLocation StartLoc,
7979                                         SourceLocation LParenLoc,
7980                                         SourceLocation EndLoc) {
7981   if (VarList.empty())
7982     return nullptr;
7983 
7984   return OMPFlushClause::Create(Context, StartLoc, LParenLoc, EndLoc, VarList);
7985 }
7986 
7987 OMPClause *
ActOnOpenMPDependClause(OpenMPDependClauseKind DepKind,SourceLocation DepLoc,SourceLocation ColonLoc,ArrayRef<Expr * > VarList,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc)7988 Sema::ActOnOpenMPDependClause(OpenMPDependClauseKind DepKind,
7989                               SourceLocation DepLoc, SourceLocation ColonLoc,
7990                               ArrayRef<Expr *> VarList, SourceLocation StartLoc,
7991                               SourceLocation LParenLoc, SourceLocation EndLoc) {
7992   if (DSAStack->getCurrentDirective() == OMPD_ordered &&
7993       DepKind != OMPC_DEPEND_source) {
7994     std::string Values = "'";
7995     Values += getOpenMPSimpleClauseTypeName(OMPC_depend, OMPC_DEPEND_source);
7996     Values += "'";
7997     Diag(DepLoc, diag::err_omp_unexpected_clause_value)
7998         << Values << getOpenMPClauseName(OMPC_depend);
7999     return nullptr;
8000   }
8001   if (DSAStack->getCurrentDirective() != OMPD_ordered &&
8002       (DepKind == OMPC_DEPEND_unknown || DepKind == OMPC_DEPEND_source)) {
8003     std::string Values;
8004     std::string Sep(", ");
8005     for (unsigned i = 0; i < OMPC_DEPEND_unknown; ++i) {
8006       if (i == OMPC_DEPEND_source)
8007         continue;
8008       Values += "'";
8009       Values += getOpenMPSimpleClauseTypeName(OMPC_depend, i);
8010       Values += "'";
8011       switch (i) {
8012       case OMPC_DEPEND_unknown - 3:
8013         Values += " or ";
8014         break;
8015       case OMPC_DEPEND_unknown - 2:
8016         break;
8017       default:
8018         Values += Sep;
8019         break;
8020       }
8021     }
8022     Diag(DepLoc, diag::err_omp_unexpected_clause_value)
8023         << Values << getOpenMPClauseName(OMPC_depend);
8024     return nullptr;
8025   }
8026   SmallVector<Expr *, 8> Vars;
8027   for (auto &RefExpr : VarList) {
8028     assert(RefExpr && "NULL expr in OpenMP shared clause.");
8029     if (isa<DependentScopeDeclRefExpr>(RefExpr)) {
8030       // It will be analyzed later.
8031       Vars.push_back(RefExpr);
8032       continue;
8033     }
8034 
8035     SourceLocation ELoc = RefExpr->getExprLoc();
8036     // OpenMP  [2.11.1.1, Restrictions, p.3]
8037     //  A variable that is part of another variable (such as a field of a
8038     //  structure) but is not an array element or an array section cannot appear
8039     //  in a depend clause.
8040     auto *SimpleExpr = RefExpr->IgnoreParenCasts();
8041     auto *DE = dyn_cast<DeclRefExpr>(SimpleExpr);
8042     auto *ASE = dyn_cast<ArraySubscriptExpr>(SimpleExpr);
8043     auto *OASE = dyn_cast<OMPArraySectionExpr>(SimpleExpr);
8044     if (!RefExpr->IgnoreParenImpCasts()->isLValue() ||
8045         (!ASE && !DE && !OASE) || (DE && !isa<VarDecl>(DE->getDecl())) ||
8046         (ASE && !ASE->getBase()->getType()->isAnyPointerType() &&
8047          !ASE->getBase()->getType()->isArrayType())) {
8048       Diag(ELoc, diag::err_omp_expected_var_name_or_array_item)
8049           << RefExpr->getSourceRange();
8050       continue;
8051     }
8052 
8053     Vars.push_back(RefExpr->IgnoreParenImpCasts());
8054   }
8055 
8056   if (DepKind != OMPC_DEPEND_source && Vars.empty())
8057     return nullptr;
8058 
8059   return OMPDependClause::Create(Context, StartLoc, LParenLoc, EndLoc, DepKind,
8060                                  DepLoc, ColonLoc, Vars);
8061 }
8062 
ActOnOpenMPDeviceClause(Expr * Device,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc)8063 OMPClause *Sema::ActOnOpenMPDeviceClause(Expr *Device, SourceLocation StartLoc,
8064                                          SourceLocation LParenLoc,
8065                                          SourceLocation EndLoc) {
8066   Expr *ValExpr = Device;
8067 
8068   // OpenMP [2.9.1, Restrictions]
8069   // The device expression must evaluate to a non-negative integer value.
8070   if (!IsNonNegativeIntegerValue(ValExpr, *this, OMPC_device,
8071                                  /*StrictlyPositive=*/false))
8072     return nullptr;
8073 
8074   return new (Context) OMPDeviceClause(ValExpr, StartLoc, LParenLoc, EndLoc);
8075 }
8076 
IsCXXRecordForMappable(Sema & SemaRef,SourceLocation Loc,DSAStackTy * Stack,CXXRecordDecl * RD)8077 static bool IsCXXRecordForMappable(Sema &SemaRef, SourceLocation Loc,
8078                                    DSAStackTy *Stack, CXXRecordDecl *RD) {
8079   if (!RD || RD->isInvalidDecl())
8080     return true;
8081 
8082   if (auto *CTSD = dyn_cast<ClassTemplateSpecializationDecl>(RD))
8083     if (auto *CTD = CTSD->getSpecializedTemplate())
8084       RD = CTD->getTemplatedDecl();
8085   auto QTy = SemaRef.Context.getRecordType(RD);
8086   if (RD->isDynamicClass()) {
8087     SemaRef.Diag(Loc, diag::err_omp_not_mappable_type) << QTy;
8088     SemaRef.Diag(RD->getLocation(), diag::note_omp_polymorphic_in_target);
8089     return false;
8090   }
8091   auto *DC = RD;
8092   bool IsCorrect = true;
8093   for (auto *I : DC->decls()) {
8094     if (I) {
8095       if (auto *MD = dyn_cast<CXXMethodDecl>(I)) {
8096         if (MD->isStatic()) {
8097           SemaRef.Diag(Loc, diag::err_omp_not_mappable_type) << QTy;
8098           SemaRef.Diag(MD->getLocation(),
8099                        diag::note_omp_static_member_in_target);
8100           IsCorrect = false;
8101         }
8102       } else if (auto *VD = dyn_cast<VarDecl>(I)) {
8103         if (VD->isStaticDataMember()) {
8104           SemaRef.Diag(Loc, diag::err_omp_not_mappable_type) << QTy;
8105           SemaRef.Diag(VD->getLocation(),
8106                        diag::note_omp_static_member_in_target);
8107           IsCorrect = false;
8108         }
8109       }
8110     }
8111   }
8112 
8113   for (auto &I : RD->bases()) {
8114     if (!IsCXXRecordForMappable(SemaRef, I.getLocStart(), Stack,
8115                                 I.getType()->getAsCXXRecordDecl()))
8116       IsCorrect = false;
8117   }
8118   return IsCorrect;
8119 }
8120 
CheckTypeMappable(SourceLocation SL,SourceRange SR,Sema & SemaRef,DSAStackTy * Stack,QualType QTy)8121 static bool CheckTypeMappable(SourceLocation SL, SourceRange SR, Sema &SemaRef,
8122                               DSAStackTy *Stack, QualType QTy) {
8123   NamedDecl *ND;
8124   if (QTy->isIncompleteType(&ND)) {
8125     SemaRef.Diag(SL, diag::err_incomplete_type) << QTy << SR;
8126     return false;
8127   } else if (CXXRecordDecl *RD = dyn_cast_or_null<CXXRecordDecl>(ND)) {
8128     if (!RD->isInvalidDecl() &&
8129         !IsCXXRecordForMappable(SemaRef, SL, Stack, RD))
8130       return false;
8131   }
8132   return true;
8133 }
8134 
ActOnOpenMPMapClause(OpenMPMapClauseKind MapTypeModifier,OpenMPMapClauseKind MapType,SourceLocation MapLoc,SourceLocation ColonLoc,ArrayRef<Expr * > VarList,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc)8135 OMPClause *Sema::ActOnOpenMPMapClause(
8136     OpenMPMapClauseKind MapTypeModifier, OpenMPMapClauseKind MapType,
8137     SourceLocation MapLoc, SourceLocation ColonLoc, ArrayRef<Expr *> VarList,
8138     SourceLocation StartLoc, SourceLocation LParenLoc, SourceLocation EndLoc) {
8139   SmallVector<Expr *, 4> Vars;
8140 
8141   for (auto &RE : VarList) {
8142     assert(RE && "Null expr in omp map");
8143     if (isa<DependentScopeDeclRefExpr>(RE)) {
8144       // It will be analyzed later.
8145       Vars.push_back(RE);
8146       continue;
8147     }
8148     SourceLocation ELoc = RE->getExprLoc();
8149 
8150     // OpenMP [2.14.5, Restrictions]
8151     //  A variable that is part of another variable (such as field of a
8152     //  structure) but is not an array element or an array section cannot appear
8153     //  in a map clause.
8154     auto *VE = RE->IgnoreParenLValueCasts();
8155 
8156     if (VE->isValueDependent() || VE->isTypeDependent() ||
8157         VE->isInstantiationDependent() ||
8158         VE->containsUnexpandedParameterPack()) {
8159       // It will be analyzed later.
8160       Vars.push_back(RE);
8161       continue;
8162     }
8163 
8164     auto *SimpleExpr = RE->IgnoreParenCasts();
8165     auto *DE = dyn_cast<DeclRefExpr>(SimpleExpr);
8166     auto *ASE = dyn_cast<ArraySubscriptExpr>(SimpleExpr);
8167     auto *OASE = dyn_cast<OMPArraySectionExpr>(SimpleExpr);
8168 
8169     if (!RE->IgnoreParenImpCasts()->isLValue() ||
8170         (!OASE && !ASE && !DE) ||
8171         (DE && !isa<VarDecl>(DE->getDecl())) ||
8172         (ASE && !ASE->getBase()->getType()->isAnyPointerType() &&
8173          !ASE->getBase()->getType()->isArrayType())) {
8174       Diag(ELoc, diag::err_omp_expected_var_name_or_array_item)
8175         << RE->getSourceRange();
8176       continue;
8177     }
8178 
8179     Decl *D = nullptr;
8180     if (DE) {
8181       D = DE->getDecl();
8182     } else if (ASE) {
8183       auto *B = ASE->getBase()->IgnoreParenCasts();
8184       D = dyn_cast<DeclRefExpr>(B)->getDecl();
8185     } else if (OASE) {
8186       auto *B = OASE->getBase();
8187       D = dyn_cast<DeclRefExpr>(B)->getDecl();
8188     }
8189     assert(D && "Null decl on map clause.");
8190     auto *VD = cast<VarDecl>(D);
8191 
8192     // OpenMP [2.14.5, Restrictions, p.8]
8193     // threadprivate variables cannot appear in a map clause.
8194     if (DSAStack->isThreadPrivate(VD)) {
8195       auto DVar = DSAStack->getTopDSA(VD, false);
8196       Diag(ELoc, diag::err_omp_threadprivate_in_map);
8197       ReportOriginalDSA(*this, DSAStack, VD, DVar);
8198       continue;
8199     }
8200 
8201     // OpenMP [2.14.5, Restrictions, p.2]
8202     //  At most one list item can be an array item derived from a given variable
8203     //  in map clauses of the same construct.
8204     // OpenMP [2.14.5, Restrictions, p.3]
8205     //  List items of map clauses in the same construct must not share original
8206     //  storage.
8207     // OpenMP [2.14.5, Restrictions, C/C++, p.2]
8208     //  A variable for which the type is pointer, reference to array, or
8209     //  reference to pointer and an array section derived from that variable
8210     //  must not appear as list items of map clauses of the same construct.
8211     DSAStackTy::MapInfo MI = DSAStack->IsMappedInCurrentRegion(VD);
8212     if (MI.RefExpr) {
8213       Diag(ELoc, diag::err_omp_map_shared_storage) << ELoc;
8214       Diag(MI.RefExpr->getExprLoc(), diag::note_used_here)
8215           << MI.RefExpr->getSourceRange();
8216       continue;
8217     }
8218 
8219     // OpenMP [2.14.5, Restrictions, C/C++, p.3,4]
8220     //  A variable for which the type is pointer, reference to array, or
8221     //  reference to pointer must not appear as a list item if the enclosing
8222     //  device data environment already contains an array section derived from
8223     //  that variable.
8224     //  An array section derived from a variable for which the type is pointer,
8225     //  reference to array, or reference to pointer must not appear as a list
8226     //  item if the enclosing device data environment already contains that
8227     //  variable.
8228     QualType Type = VD->getType();
8229     MI = DSAStack->getMapInfoForVar(VD);
8230     if (MI.RefExpr && (isa<DeclRefExpr>(MI.RefExpr->IgnoreParenLValueCasts()) !=
8231                        isa<DeclRefExpr>(VE)) &&
8232         (Type->isPointerType() || Type->isReferenceType())) {
8233       Diag(ELoc, diag::err_omp_map_shared_storage) << ELoc;
8234       Diag(MI.RefExpr->getExprLoc(), diag::note_used_here)
8235           << MI.RefExpr->getSourceRange();
8236       continue;
8237     }
8238 
8239     // OpenMP [2.14.5, Restrictions, C/C++, p.7]
8240     //  A list item must have a mappable type.
8241     if (!CheckTypeMappable(VE->getExprLoc(), VE->getSourceRange(), *this,
8242                            DSAStack, Type))
8243       continue;
8244 
8245     Vars.push_back(RE);
8246     MI.RefExpr = RE;
8247     DSAStack->addMapInfoForVar(VD, MI);
8248   }
8249   if (Vars.empty())
8250     return nullptr;
8251 
8252   return OMPMapClause::Create(Context, StartLoc, LParenLoc, EndLoc, Vars,
8253                               MapTypeModifier, MapType, MapLoc);
8254 }
8255 
ActOnOpenMPNumTeamsClause(Expr * NumTeams,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc)8256 OMPClause *Sema::ActOnOpenMPNumTeamsClause(Expr *NumTeams,
8257                                            SourceLocation StartLoc,
8258                                            SourceLocation LParenLoc,
8259                                            SourceLocation EndLoc) {
8260   Expr *ValExpr = NumTeams;
8261 
8262   // OpenMP [teams Constrcut, Restrictions]
8263   // The num_teams expression must evaluate to a positive integer value.
8264   if (!IsNonNegativeIntegerValue(ValExpr, *this, OMPC_num_teams,
8265                                  /*StrictlyPositive=*/true))
8266     return nullptr;
8267 
8268   return new (Context) OMPNumTeamsClause(ValExpr, StartLoc, LParenLoc, EndLoc);
8269 }
8270 
ActOnOpenMPThreadLimitClause(Expr * ThreadLimit,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc)8271 OMPClause *Sema::ActOnOpenMPThreadLimitClause(Expr *ThreadLimit,
8272                                               SourceLocation StartLoc,
8273                                               SourceLocation LParenLoc,
8274                                               SourceLocation EndLoc) {
8275   Expr *ValExpr = ThreadLimit;
8276 
8277   // OpenMP [teams Constrcut, Restrictions]
8278   // The thread_limit expression must evaluate to a positive integer value.
8279   if (!IsNonNegativeIntegerValue(ValExpr, *this, OMPC_thread_limit,
8280                                  /*StrictlyPositive=*/true))
8281     return nullptr;
8282 
8283   return new (Context) OMPThreadLimitClause(ValExpr, StartLoc, LParenLoc,
8284                                             EndLoc);
8285 }
8286 
ActOnOpenMPPriorityClause(Expr * Priority,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc)8287 OMPClause *Sema::ActOnOpenMPPriorityClause(Expr *Priority,
8288                                            SourceLocation StartLoc,
8289                                            SourceLocation LParenLoc,
8290                                            SourceLocation EndLoc) {
8291   Expr *ValExpr = Priority;
8292 
8293   // OpenMP [2.9.1, task Constrcut]
8294   // The priority-value is a non-negative numerical scalar expression.
8295   if (!IsNonNegativeIntegerValue(ValExpr, *this, OMPC_priority,
8296                                  /*StrictlyPositive=*/false))
8297     return nullptr;
8298 
8299   return new (Context) OMPPriorityClause(ValExpr, StartLoc, LParenLoc, EndLoc);
8300 }
8301 
ActOnOpenMPGrainsizeClause(Expr * Grainsize,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc)8302 OMPClause *Sema::ActOnOpenMPGrainsizeClause(Expr *Grainsize,
8303                                             SourceLocation StartLoc,
8304                                             SourceLocation LParenLoc,
8305                                             SourceLocation EndLoc) {
8306   Expr *ValExpr = Grainsize;
8307 
8308   // OpenMP [2.9.2, taskloop Constrcut]
8309   // The parameter of the grainsize clause must be a positive integer
8310   // expression.
8311   if (!IsNonNegativeIntegerValue(ValExpr, *this, OMPC_grainsize,
8312                                  /*StrictlyPositive=*/true))
8313     return nullptr;
8314 
8315   return new (Context) OMPGrainsizeClause(ValExpr, StartLoc, LParenLoc, EndLoc);
8316 }
8317 
ActOnOpenMPNumTasksClause(Expr * NumTasks,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc)8318 OMPClause *Sema::ActOnOpenMPNumTasksClause(Expr *NumTasks,
8319                                            SourceLocation StartLoc,
8320                                            SourceLocation LParenLoc,
8321                                            SourceLocation EndLoc) {
8322   Expr *ValExpr = NumTasks;
8323 
8324   // OpenMP [2.9.2, taskloop Constrcut]
8325   // The parameter of the num_tasks clause must be a positive integer
8326   // expression.
8327   if (!IsNonNegativeIntegerValue(ValExpr, *this, OMPC_num_tasks,
8328                                  /*StrictlyPositive=*/true))
8329     return nullptr;
8330 
8331   return new (Context) OMPNumTasksClause(ValExpr, StartLoc, LParenLoc, EndLoc);
8332 }
8333 
ActOnOpenMPHintClause(Expr * Hint,SourceLocation StartLoc,SourceLocation LParenLoc,SourceLocation EndLoc)8334 OMPClause *Sema::ActOnOpenMPHintClause(Expr *Hint, SourceLocation StartLoc,
8335                                        SourceLocation LParenLoc,
8336                                        SourceLocation EndLoc) {
8337   // OpenMP [2.13.2, critical construct, Description]
8338   // ... where hint-expression is an integer constant expression that evaluates
8339   // to a valid lock hint.
8340   ExprResult HintExpr = VerifyPositiveIntegerConstantInClause(Hint, OMPC_hint);
8341   if (HintExpr.isInvalid())
8342     return nullptr;
8343   return new (Context)
8344       OMPHintClause(HintExpr.get(), StartLoc, LParenLoc, EndLoc);
8345 }
8346 
8347