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/CXXInheritance.h" 19 #include "clang/AST/Decl.h" 20 #include "clang/AST/DeclCXX.h" 21 #include "clang/AST/DeclOpenMP.h" 22 #include "clang/AST/StmtCXX.h" 23 #include "clang/AST/StmtOpenMP.h" 24 #include "clang/AST/StmtVisitor.h" 25 #include "clang/Basic/OpenMPKinds.h" 26 #include "clang/Sema/Initialization.h" 27 #include "clang/Sema/Lookup.h" 28 #include "clang/Sema/Scope.h" 29 #include "clang/Sema/ScopeInfo.h" 30 #include "clang/Sema/SemaInternal.h" 31 #include "llvm/ADT/PointerEmbeddedInt.h" 32 using namespace clang; 33 34 //===----------------------------------------------------------------------===// 35 // Stack of data-sharing attributes for variables 36 //===----------------------------------------------------------------------===// 37 38 static Expr *CheckMapClauseExpressionBase( 39 Sema &SemaRef, Expr *E, 40 OMPClauseMappableExprCommon::MappableExprComponentList &CurComponents, 41 OpenMPClauseKind CKind); 42 43 namespace { 44 /// \brief Default data sharing attributes, which can be applied to directive. 45 enum DefaultDataSharingAttributes { 46 DSA_unspecified = 0, /// \brief Data sharing attribute not specified. 47 DSA_none = 1 << 0, /// \brief Default data sharing attribute 'none'. 48 DSA_shared = 1 << 1, /// \brief Default data sharing attribute 'shared'. 49 }; 50 51 /// Attributes of the defaultmap clause. 52 enum DefaultMapAttributes { 53 DMA_unspecified, /// Default mapping is not specified. 54 DMA_tofrom_scalar, /// Default mapping is 'tofrom:scalar'. 55 }; 56 57 /// \brief Stack for tracking declarations used in OpenMP directives and 58 /// clauses and their data-sharing attributes. 59 class DSAStackTy final { 60 public: 61 struct DSAVarData final { 62 OpenMPDirectiveKind DKind = OMPD_unknown; 63 OpenMPClauseKind CKind = OMPC_unknown; 64 Expr *RefExpr = nullptr; 65 DeclRefExpr *PrivateCopy = nullptr; 66 SourceLocation ImplicitDSALoc; 67 DSAVarData() = default; 68 DSAVarData(OpenMPDirectiveKind DKind, OpenMPClauseKind CKind, Expr *RefExpr, 69 DeclRefExpr *PrivateCopy, SourceLocation ImplicitDSALoc) 70 : DKind(DKind), CKind(CKind), RefExpr(RefExpr), 71 PrivateCopy(PrivateCopy), ImplicitDSALoc(ImplicitDSALoc) {} 72 }; 73 typedef llvm::SmallVector<std::pair<Expr *, OverloadedOperatorKind>, 4> 74 OperatorOffsetTy; 75 76 private: 77 struct DSAInfo final { 78 OpenMPClauseKind Attributes = OMPC_unknown; 79 /// Pointer to a reference expression and a flag which shows that the 80 /// variable is marked as lastprivate(true) or not (false). 81 llvm::PointerIntPair<Expr *, 1, bool> RefExpr; 82 DeclRefExpr *PrivateCopy = nullptr; 83 }; 84 typedef llvm::DenseMap<ValueDecl *, DSAInfo> DeclSAMapTy; 85 typedef llvm::DenseMap<ValueDecl *, Expr *> AlignedMapTy; 86 typedef std::pair<unsigned, VarDecl *> LCDeclInfo; 87 typedef llvm::DenseMap<ValueDecl *, LCDeclInfo> LoopControlVariablesMapTy; 88 /// Struct that associates a component with the clause kind where they are 89 /// found. 90 struct MappedExprComponentTy { 91 OMPClauseMappableExprCommon::MappableExprComponentLists Components; 92 OpenMPClauseKind Kind = OMPC_unknown; 93 }; 94 typedef llvm::DenseMap<ValueDecl *, MappedExprComponentTy> 95 MappedExprComponentsTy; 96 typedef llvm::StringMap<std::pair<OMPCriticalDirective *, llvm::APSInt>> 97 CriticalsWithHintsTy; 98 typedef llvm::DenseMap<OMPDependClause *, OperatorOffsetTy> 99 DoacrossDependMapTy; 100 struct ReductionData { 101 typedef llvm::PointerEmbeddedInt<BinaryOperatorKind, 16> BOKPtrType; 102 SourceRange ReductionRange; 103 llvm::PointerUnion<const Expr *, BOKPtrType> ReductionOp; 104 ReductionData() = default; 105 void set(BinaryOperatorKind BO, SourceRange RR) { 106 ReductionRange = RR; 107 ReductionOp = BO; 108 } 109 void set(const Expr *RefExpr, SourceRange RR) { 110 ReductionRange = RR; 111 ReductionOp = RefExpr; 112 } 113 }; 114 typedef llvm::DenseMap<ValueDecl *, ReductionData> DeclReductionMapTy; 115 116 struct SharingMapTy final { 117 DeclSAMapTy SharingMap; 118 DeclReductionMapTy ReductionMap; 119 AlignedMapTy AlignedMap; 120 MappedExprComponentsTy MappedExprComponents; 121 LoopControlVariablesMapTy LCVMap; 122 DefaultDataSharingAttributes DefaultAttr = DSA_unspecified; 123 SourceLocation DefaultAttrLoc; 124 DefaultMapAttributes DefaultMapAttr = DMA_unspecified; 125 SourceLocation DefaultMapAttrLoc; 126 OpenMPDirectiveKind Directive = OMPD_unknown; 127 DeclarationNameInfo DirectiveName; 128 Scope *CurScope = nullptr; 129 SourceLocation ConstructLoc; 130 /// Set of 'depend' clauses with 'sink|source' dependence kind. Required to 131 /// get the data (loop counters etc.) about enclosing loop-based construct. 132 /// This data is required during codegen. 133 DoacrossDependMapTy DoacrossDepends; 134 /// \brief first argument (Expr *) contains optional argument of the 135 /// 'ordered' clause, the second one is true if the regions has 'ordered' 136 /// clause, false otherwise. 137 llvm::PointerIntPair<Expr *, 1, bool> OrderedRegion; 138 bool NowaitRegion = false; 139 bool CancelRegion = false; 140 unsigned AssociatedLoops = 1; 141 SourceLocation InnerTeamsRegionLoc; 142 /// Reference to the taskgroup task_reduction reference expression. 143 Expr *TaskgroupReductionRef = nullptr; 144 SharingMapTy(OpenMPDirectiveKind DKind, DeclarationNameInfo Name, 145 Scope *CurScope, SourceLocation Loc) 146 : Directive(DKind), DirectiveName(Name), CurScope(CurScope), 147 ConstructLoc(Loc) {} 148 SharingMapTy() = default; 149 }; 150 151 typedef SmallVector<SharingMapTy, 4> StackTy; 152 153 /// \brief Stack of used declaration and their data-sharing attributes. 154 DeclSAMapTy Threadprivates; 155 const FunctionScopeInfo *CurrentNonCapturingFunctionScope = nullptr; 156 SmallVector<std::pair<StackTy, const FunctionScopeInfo *>, 4> Stack; 157 /// \brief true, if check for DSA must be from parent directive, false, if 158 /// from current directive. 159 OpenMPClauseKind ClauseKindMode = OMPC_unknown; 160 Sema &SemaRef; 161 bool ForceCapturing = false; 162 CriticalsWithHintsTy Criticals; 163 164 typedef SmallVector<SharingMapTy, 8>::reverse_iterator reverse_iterator; 165 166 DSAVarData getDSA(StackTy::reverse_iterator &Iter, ValueDecl *D); 167 168 /// \brief Checks if the variable is a local for OpenMP region. 169 bool isOpenMPLocal(VarDecl *D, StackTy::reverse_iterator Iter); 170 171 bool isStackEmpty() const { 172 return Stack.empty() || 173 Stack.back().second != CurrentNonCapturingFunctionScope || 174 Stack.back().first.empty(); 175 } 176 177 public: 178 explicit DSAStackTy(Sema &S) : SemaRef(S) {} 179 180 bool isClauseParsingMode() const { return ClauseKindMode != OMPC_unknown; } 181 void setClauseParsingMode(OpenMPClauseKind K) { ClauseKindMode = K; } 182 183 bool isForceVarCapturing() const { return ForceCapturing; } 184 void setForceVarCapturing(bool V) { ForceCapturing = V; } 185 186 void push(OpenMPDirectiveKind DKind, const DeclarationNameInfo &DirName, 187 Scope *CurScope, SourceLocation Loc) { 188 if (Stack.empty() || 189 Stack.back().second != CurrentNonCapturingFunctionScope) 190 Stack.emplace_back(StackTy(), CurrentNonCapturingFunctionScope); 191 Stack.back().first.emplace_back(DKind, DirName, CurScope, Loc); 192 Stack.back().first.back().DefaultAttrLoc = Loc; 193 } 194 195 void pop() { 196 assert(!Stack.back().first.empty() && 197 "Data-sharing attributes stack is empty!"); 198 Stack.back().first.pop_back(); 199 } 200 201 /// Start new OpenMP region stack in new non-capturing function. 202 void pushFunction() { 203 const FunctionScopeInfo *CurFnScope = SemaRef.getCurFunction(); 204 assert(!isa<CapturingScopeInfo>(CurFnScope)); 205 CurrentNonCapturingFunctionScope = CurFnScope; 206 } 207 /// Pop region stack for non-capturing function. 208 void popFunction(const FunctionScopeInfo *OldFSI) { 209 if (!Stack.empty() && Stack.back().second == OldFSI) { 210 assert(Stack.back().first.empty()); 211 Stack.pop_back(); 212 } 213 CurrentNonCapturingFunctionScope = nullptr; 214 for (const FunctionScopeInfo *FSI : llvm::reverse(SemaRef.FunctionScopes)) { 215 if (!isa<CapturingScopeInfo>(FSI)) { 216 CurrentNonCapturingFunctionScope = FSI; 217 break; 218 } 219 } 220 } 221 222 void addCriticalWithHint(OMPCriticalDirective *D, llvm::APSInt Hint) { 223 Criticals[D->getDirectiveName().getAsString()] = std::make_pair(D, Hint); 224 } 225 const std::pair<OMPCriticalDirective *, llvm::APSInt> 226 getCriticalWithHint(const DeclarationNameInfo &Name) const { 227 auto I = Criticals.find(Name.getAsString()); 228 if (I != Criticals.end()) 229 return I->second; 230 return std::make_pair(nullptr, llvm::APSInt()); 231 } 232 /// \brief If 'aligned' declaration for given variable \a D was not seen yet, 233 /// add it and return NULL; otherwise return previous occurrence's expression 234 /// for diagnostics. 235 Expr *addUniqueAligned(ValueDecl *D, Expr *NewDE); 236 237 /// \brief Register specified variable as loop control variable. 238 void addLoopControlVariable(ValueDecl *D, VarDecl *Capture); 239 /// \brief Check if the specified variable is a loop control variable for 240 /// current region. 241 /// \return The index of the loop control variable in the list of associated 242 /// for-loops (from outer to inner). 243 LCDeclInfo isLoopControlVariable(ValueDecl *D); 244 /// \brief Check if the specified variable is a loop control variable for 245 /// parent region. 246 /// \return The index of the loop control variable in the list of associated 247 /// for-loops (from outer to inner). 248 LCDeclInfo isParentLoopControlVariable(ValueDecl *D); 249 /// \brief Get the loop control variable for the I-th loop (or nullptr) in 250 /// parent directive. 251 ValueDecl *getParentLoopControlVariable(unsigned I); 252 253 /// \brief Adds explicit data sharing attribute to the specified declaration. 254 void addDSA(ValueDecl *D, Expr *E, OpenMPClauseKind A, 255 DeclRefExpr *PrivateCopy = nullptr); 256 257 /// Adds additional information for the reduction items with the reduction id 258 /// represented as an operator. 259 void addTaskgroupReductionData(ValueDecl *D, SourceRange SR, 260 BinaryOperatorKind BOK); 261 /// Adds additional information for the reduction items with the reduction id 262 /// represented as reduction identifier. 263 void addTaskgroupReductionData(ValueDecl *D, SourceRange SR, 264 const Expr *ReductionRef); 265 /// Returns the location and reduction operation from the innermost parent 266 /// region for the given \p D. 267 DSAVarData getTopMostTaskgroupReductionData(ValueDecl *D, SourceRange &SR, 268 BinaryOperatorKind &BOK, 269 Expr *&TaskgroupDescriptor); 270 /// Returns the location and reduction operation from the innermost parent 271 /// region for the given \p D. 272 DSAVarData getTopMostTaskgroupReductionData(ValueDecl *D, SourceRange &SR, 273 const Expr *&ReductionRef, 274 Expr *&TaskgroupDescriptor); 275 /// Return reduction reference expression for the current taskgroup. 276 Expr *getTaskgroupReductionRef() const { 277 assert(Stack.back().first.back().Directive == OMPD_taskgroup && 278 "taskgroup reference expression requested for non taskgroup " 279 "directive."); 280 return Stack.back().first.back().TaskgroupReductionRef; 281 } 282 /// Checks if the given \p VD declaration is actually a taskgroup reduction 283 /// descriptor variable at the \p Level of OpenMP regions. 284 bool isTaskgroupReductionRef(ValueDecl *VD, unsigned Level) const { 285 return Stack.back().first[Level].TaskgroupReductionRef && 286 cast<DeclRefExpr>(Stack.back().first[Level].TaskgroupReductionRef) 287 ->getDecl() == VD; 288 } 289 290 /// \brief Returns data sharing attributes from top of the stack for the 291 /// specified declaration. 292 DSAVarData getTopDSA(ValueDecl *D, bool FromParent); 293 /// \brief Returns data-sharing attributes for the specified declaration. 294 DSAVarData getImplicitDSA(ValueDecl *D, bool FromParent); 295 /// \brief Checks if the specified variables has data-sharing attributes which 296 /// match specified \a CPred predicate in any directive which matches \a DPred 297 /// predicate. 298 DSAVarData hasDSA(ValueDecl *D, 299 const llvm::function_ref<bool(OpenMPClauseKind)> &CPred, 300 const llvm::function_ref<bool(OpenMPDirectiveKind)> &DPred, 301 bool FromParent); 302 /// \brief Checks if the specified variables has data-sharing attributes which 303 /// match specified \a CPred predicate in any innermost directive which 304 /// matches \a DPred predicate. 305 DSAVarData 306 hasInnermostDSA(ValueDecl *D, 307 const llvm::function_ref<bool(OpenMPClauseKind)> &CPred, 308 const llvm::function_ref<bool(OpenMPDirectiveKind)> &DPred, 309 bool FromParent); 310 /// \brief Checks if the specified variables has explicit data-sharing 311 /// attributes which match specified \a CPred predicate at the specified 312 /// OpenMP region. 313 bool hasExplicitDSA(ValueDecl *D, 314 const llvm::function_ref<bool(OpenMPClauseKind)> &CPred, 315 unsigned Level, bool NotLastprivate = false); 316 317 /// \brief Returns true if the directive at level \Level matches in the 318 /// specified \a DPred predicate. 319 bool hasExplicitDirective( 320 const llvm::function_ref<bool(OpenMPDirectiveKind)> &DPred, 321 unsigned Level); 322 323 /// \brief Finds a directive which matches specified \a DPred predicate. 324 bool hasDirective(const llvm::function_ref<bool(OpenMPDirectiveKind, 325 const DeclarationNameInfo &, 326 SourceLocation)> &DPred, 327 bool FromParent); 328 329 /// \brief Returns currently analyzed directive. 330 OpenMPDirectiveKind getCurrentDirective() const { 331 return isStackEmpty() ? OMPD_unknown : Stack.back().first.back().Directive; 332 } 333 /// \brief Returns parent directive. 334 OpenMPDirectiveKind getParentDirective() const { 335 if (isStackEmpty() || Stack.back().first.size() == 1) 336 return OMPD_unknown; 337 return std::next(Stack.back().first.rbegin())->Directive; 338 } 339 340 /// \brief Set default data sharing attribute to none. 341 void setDefaultDSANone(SourceLocation Loc) { 342 assert(!isStackEmpty()); 343 Stack.back().first.back().DefaultAttr = DSA_none; 344 Stack.back().first.back().DefaultAttrLoc = Loc; 345 } 346 /// \brief Set default data sharing attribute to shared. 347 void setDefaultDSAShared(SourceLocation Loc) { 348 assert(!isStackEmpty()); 349 Stack.back().first.back().DefaultAttr = DSA_shared; 350 Stack.back().first.back().DefaultAttrLoc = Loc; 351 } 352 /// Set default data mapping attribute to 'tofrom:scalar'. 353 void setDefaultDMAToFromScalar(SourceLocation Loc) { 354 assert(!isStackEmpty()); 355 Stack.back().first.back().DefaultMapAttr = DMA_tofrom_scalar; 356 Stack.back().first.back().DefaultMapAttrLoc = Loc; 357 } 358 359 DefaultDataSharingAttributes getDefaultDSA() const { 360 return isStackEmpty() ? DSA_unspecified 361 : Stack.back().first.back().DefaultAttr; 362 } 363 SourceLocation getDefaultDSALocation() const { 364 return isStackEmpty() ? SourceLocation() 365 : Stack.back().first.back().DefaultAttrLoc; 366 } 367 DefaultMapAttributes getDefaultDMA() const { 368 return isStackEmpty() ? DMA_unspecified 369 : Stack.back().first.back().DefaultMapAttr; 370 } 371 DefaultMapAttributes getDefaultDMAAtLevel(unsigned Level) const { 372 return Stack.back().first[Level].DefaultMapAttr; 373 } 374 SourceLocation getDefaultDMALocation() const { 375 return isStackEmpty() ? SourceLocation() 376 : Stack.back().first.back().DefaultMapAttrLoc; 377 } 378 379 /// \brief Checks if the specified variable is a threadprivate. 380 bool isThreadPrivate(VarDecl *D) { 381 DSAVarData DVar = getTopDSA(D, false); 382 return isOpenMPThreadPrivate(DVar.CKind); 383 } 384 385 /// \brief Marks current region as ordered (it has an 'ordered' clause). 386 void setOrderedRegion(bool IsOrdered, Expr *Param) { 387 assert(!isStackEmpty()); 388 Stack.back().first.back().OrderedRegion.setInt(IsOrdered); 389 Stack.back().first.back().OrderedRegion.setPointer(Param); 390 } 391 /// \brief Returns true, if parent region is ordered (has associated 392 /// 'ordered' clause), false - otherwise. 393 bool isParentOrderedRegion() const { 394 if (isStackEmpty() || Stack.back().first.size() == 1) 395 return false; 396 return std::next(Stack.back().first.rbegin())->OrderedRegion.getInt(); 397 } 398 /// \brief Returns optional parameter for the ordered region. 399 Expr *getParentOrderedRegionParam() const { 400 if (isStackEmpty() || Stack.back().first.size() == 1) 401 return nullptr; 402 return std::next(Stack.back().first.rbegin())->OrderedRegion.getPointer(); 403 } 404 /// \brief Marks current region as nowait (it has a 'nowait' clause). 405 void setNowaitRegion(bool IsNowait = true) { 406 assert(!isStackEmpty()); 407 Stack.back().first.back().NowaitRegion = IsNowait; 408 } 409 /// \brief Returns true, if parent region is nowait (has associated 410 /// 'nowait' clause), false - otherwise. 411 bool isParentNowaitRegion() const { 412 if (isStackEmpty() || Stack.back().first.size() == 1) 413 return false; 414 return std::next(Stack.back().first.rbegin())->NowaitRegion; 415 } 416 /// \brief Marks parent region as cancel region. 417 void setParentCancelRegion(bool Cancel = true) { 418 if (!isStackEmpty() && Stack.back().first.size() > 1) { 419 auto &StackElemRef = *std::next(Stack.back().first.rbegin()); 420 StackElemRef.CancelRegion |= StackElemRef.CancelRegion || Cancel; 421 } 422 } 423 /// \brief Return true if current region has inner cancel construct. 424 bool isCancelRegion() const { 425 return isStackEmpty() ? false : Stack.back().first.back().CancelRegion; 426 } 427 428 /// \brief Set collapse value for the region. 429 void setAssociatedLoops(unsigned Val) { 430 assert(!isStackEmpty()); 431 Stack.back().first.back().AssociatedLoops = Val; 432 } 433 /// \brief Return collapse value for region. 434 unsigned getAssociatedLoops() const { 435 return isStackEmpty() ? 0 : Stack.back().first.back().AssociatedLoops; 436 } 437 438 /// \brief Marks current target region as one with closely nested teams 439 /// region. 440 void setParentTeamsRegionLoc(SourceLocation TeamsRegionLoc) { 441 if (!isStackEmpty() && Stack.back().first.size() > 1) { 442 std::next(Stack.back().first.rbegin())->InnerTeamsRegionLoc = 443 TeamsRegionLoc; 444 } 445 } 446 /// \brief Returns true, if current region has closely nested teams region. 447 bool hasInnerTeamsRegion() const { 448 return getInnerTeamsRegionLoc().isValid(); 449 } 450 /// \brief Returns location of the nested teams region (if any). 451 SourceLocation getInnerTeamsRegionLoc() const { 452 return isStackEmpty() ? SourceLocation() 453 : Stack.back().first.back().InnerTeamsRegionLoc; 454 } 455 456 Scope *getCurScope() const { 457 return isStackEmpty() ? nullptr : Stack.back().first.back().CurScope; 458 } 459 Scope *getCurScope() { 460 return isStackEmpty() ? nullptr : Stack.back().first.back().CurScope; 461 } 462 SourceLocation getConstructLoc() { 463 return isStackEmpty() ? SourceLocation() 464 : Stack.back().first.back().ConstructLoc; 465 } 466 467 /// Do the check specified in \a Check to all component lists and return true 468 /// if any issue is found. 469 bool checkMappableExprComponentListsForDecl( 470 ValueDecl *VD, bool CurrentRegionOnly, 471 const llvm::function_ref< 472 bool(OMPClauseMappableExprCommon::MappableExprComponentListRef, 473 OpenMPClauseKind)> &Check) { 474 if (isStackEmpty()) 475 return false; 476 auto SI = Stack.back().first.rbegin(); 477 auto SE = Stack.back().first.rend(); 478 479 if (SI == SE) 480 return false; 481 482 if (CurrentRegionOnly) { 483 SE = std::next(SI); 484 } else { 485 ++SI; 486 } 487 488 for (; SI != SE; ++SI) { 489 auto MI = SI->MappedExprComponents.find(VD); 490 if (MI != SI->MappedExprComponents.end()) 491 for (auto &L : MI->second.Components) 492 if (Check(L, MI->second.Kind)) 493 return true; 494 } 495 return false; 496 } 497 498 /// Do the check specified in \a Check to all component lists at a given level 499 /// and return true if any issue is found. 500 bool checkMappableExprComponentListsForDeclAtLevel( 501 ValueDecl *VD, unsigned Level, 502 const llvm::function_ref< 503 bool(OMPClauseMappableExprCommon::MappableExprComponentListRef, 504 OpenMPClauseKind)> &Check) { 505 if (isStackEmpty()) 506 return false; 507 508 auto StartI = Stack.back().first.begin(); 509 auto EndI = Stack.back().first.end(); 510 if (std::distance(StartI, EndI) <= (int)Level) 511 return false; 512 std::advance(StartI, Level); 513 514 auto MI = StartI->MappedExprComponents.find(VD); 515 if (MI != StartI->MappedExprComponents.end()) 516 for (auto &L : MI->second.Components) 517 if (Check(L, MI->second.Kind)) 518 return true; 519 return false; 520 } 521 522 /// Create a new mappable expression component list associated with a given 523 /// declaration and initialize it with the provided list of components. 524 void addMappableExpressionComponents( 525 ValueDecl *VD, 526 OMPClauseMappableExprCommon::MappableExprComponentListRef Components, 527 OpenMPClauseKind WhereFoundClauseKind) { 528 assert(!isStackEmpty() && 529 "Not expecting to retrieve components from a empty stack!"); 530 auto &MEC = Stack.back().first.back().MappedExprComponents[VD]; 531 // Create new entry and append the new components there. 532 MEC.Components.resize(MEC.Components.size() + 1); 533 MEC.Components.back().append(Components.begin(), Components.end()); 534 MEC.Kind = WhereFoundClauseKind; 535 } 536 537 unsigned getNestingLevel() const { 538 assert(!isStackEmpty()); 539 return Stack.back().first.size() - 1; 540 } 541 void addDoacrossDependClause(OMPDependClause *C, OperatorOffsetTy &OpsOffs) { 542 assert(!isStackEmpty() && Stack.back().first.size() > 1); 543 auto &StackElem = *std::next(Stack.back().first.rbegin()); 544 assert(isOpenMPWorksharingDirective(StackElem.Directive)); 545 StackElem.DoacrossDepends.insert({C, OpsOffs}); 546 } 547 llvm::iterator_range<DoacrossDependMapTy::const_iterator> 548 getDoacrossDependClauses() const { 549 assert(!isStackEmpty()); 550 auto &StackElem = Stack.back().first.back(); 551 if (isOpenMPWorksharingDirective(StackElem.Directive)) { 552 auto &Ref = StackElem.DoacrossDepends; 553 return llvm::make_range(Ref.begin(), Ref.end()); 554 } 555 return llvm::make_range(StackElem.DoacrossDepends.end(), 556 StackElem.DoacrossDepends.end()); 557 } 558 }; 559 bool isParallelOrTaskRegion(OpenMPDirectiveKind DKind) { 560 return isOpenMPParallelDirective(DKind) || isOpenMPTaskingDirective(DKind) || 561 isOpenMPTeamsDirective(DKind) || DKind == OMPD_unknown; 562 } 563 } // namespace 564 565 static Expr *getExprAsWritten(Expr *E) { 566 if (auto *ExprTemp = dyn_cast<ExprWithCleanups>(E)) 567 E = ExprTemp->getSubExpr(); 568 569 if (auto *MTE = dyn_cast<MaterializeTemporaryExpr>(E)) 570 E = MTE->GetTemporaryExpr(); 571 572 while (auto *Binder = dyn_cast<CXXBindTemporaryExpr>(E)) 573 E = Binder->getSubExpr(); 574 575 if (auto *ICE = dyn_cast<ImplicitCastExpr>(E)) 576 E = ICE->getSubExprAsWritten(); 577 return E->IgnoreParens(); 578 } 579 580 static ValueDecl *getCanonicalDecl(ValueDecl *D) { 581 if (auto *CED = dyn_cast<OMPCapturedExprDecl>(D)) 582 if (auto *ME = dyn_cast<MemberExpr>(getExprAsWritten(CED->getInit()))) 583 D = ME->getMemberDecl(); 584 auto *VD = dyn_cast<VarDecl>(D); 585 auto *FD = dyn_cast<FieldDecl>(D); 586 if (VD != nullptr) { 587 VD = VD->getCanonicalDecl(); 588 D = VD; 589 } else { 590 assert(FD); 591 FD = FD->getCanonicalDecl(); 592 D = FD; 593 } 594 return D; 595 } 596 597 DSAStackTy::DSAVarData DSAStackTy::getDSA(StackTy::reverse_iterator &Iter, 598 ValueDecl *D) { 599 D = getCanonicalDecl(D); 600 auto *VD = dyn_cast<VarDecl>(D); 601 auto *FD = dyn_cast<FieldDecl>(D); 602 DSAVarData DVar; 603 if (isStackEmpty() || Iter == Stack.back().first.rend()) { 604 // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced 605 // in a region but not in construct] 606 // File-scope or namespace-scope variables referenced in called routines 607 // in the region are shared unless they appear in a threadprivate 608 // directive. 609 if (VD && !VD->isFunctionOrMethodVarDecl() && !isa<ParmVarDecl>(D)) 610 DVar.CKind = OMPC_shared; 611 612 // OpenMP [2.9.1.2, Data-sharing Attribute Rules for Variables Referenced 613 // in a region but not in construct] 614 // Variables with static storage duration that are declared in called 615 // routines in the region are shared. 616 if (VD && VD->hasGlobalStorage()) 617 DVar.CKind = OMPC_shared; 618 619 // Non-static data members are shared by default. 620 if (FD) 621 DVar.CKind = OMPC_shared; 622 623 return DVar; 624 } 625 626 // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced 627 // in a Construct, C/C++, predetermined, p.1] 628 // Variables with automatic storage duration that are declared in a scope 629 // inside the construct are private. 630 if (VD && isOpenMPLocal(VD, Iter) && VD->isLocalVarDecl() && 631 (VD->getStorageClass() == SC_Auto || VD->getStorageClass() == SC_None)) { 632 DVar.CKind = OMPC_private; 633 return DVar; 634 } 635 636 DVar.DKind = Iter->Directive; 637 // Explicitly specified attributes and local variables with predetermined 638 // attributes. 639 if (Iter->SharingMap.count(D)) { 640 DVar.RefExpr = Iter->SharingMap[D].RefExpr.getPointer(); 641 DVar.PrivateCopy = Iter->SharingMap[D].PrivateCopy; 642 DVar.CKind = Iter->SharingMap[D].Attributes; 643 DVar.ImplicitDSALoc = Iter->DefaultAttrLoc; 644 return DVar; 645 } 646 647 // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced 648 // in a Construct, C/C++, implicitly determined, p.1] 649 // In a parallel or task construct, the data-sharing attributes of these 650 // variables are determined by the default clause, if present. 651 switch (Iter->DefaultAttr) { 652 case DSA_shared: 653 DVar.CKind = OMPC_shared; 654 DVar.ImplicitDSALoc = Iter->DefaultAttrLoc; 655 return DVar; 656 case DSA_none: 657 return DVar; 658 case DSA_unspecified: 659 // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced 660 // in a Construct, implicitly determined, p.2] 661 // In a parallel construct, if no default clause is present, these 662 // variables are shared. 663 DVar.ImplicitDSALoc = Iter->DefaultAttrLoc; 664 if (isOpenMPParallelDirective(DVar.DKind) || 665 isOpenMPTeamsDirective(DVar.DKind)) { 666 DVar.CKind = OMPC_shared; 667 return DVar; 668 } 669 670 // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced 671 // in a Construct, implicitly determined, p.4] 672 // In a task construct, if no default clause is present, a variable that in 673 // the enclosing context is determined to be shared by all implicit tasks 674 // bound to the current team is shared. 675 if (isOpenMPTaskingDirective(DVar.DKind)) { 676 DSAVarData DVarTemp; 677 auto I = Iter, E = Stack.back().first.rend(); 678 do { 679 ++I; 680 // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables 681 // Referenced in a Construct, implicitly determined, p.6] 682 // In a task construct, if no default clause is present, a variable 683 // whose data-sharing attribute is not determined by the rules above is 684 // firstprivate. 685 DVarTemp = getDSA(I, D); 686 if (DVarTemp.CKind != OMPC_shared) { 687 DVar.RefExpr = nullptr; 688 DVar.CKind = OMPC_firstprivate; 689 return DVar; 690 } 691 } while (I != E && !isParallelOrTaskRegion(I->Directive)); 692 DVar.CKind = 693 (DVarTemp.CKind == OMPC_unknown) ? OMPC_firstprivate : OMPC_shared; 694 return DVar; 695 } 696 } 697 // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced 698 // in a Construct, implicitly determined, p.3] 699 // For constructs other than task, if no default clause is present, these 700 // variables inherit their data-sharing attributes from the enclosing 701 // context. 702 return getDSA(++Iter, D); 703 } 704 705 Expr *DSAStackTy::addUniqueAligned(ValueDecl *D, Expr *NewDE) { 706 assert(!isStackEmpty() && "Data sharing attributes stack is empty"); 707 D = getCanonicalDecl(D); 708 auto &StackElem = Stack.back().first.back(); 709 auto It = StackElem.AlignedMap.find(D); 710 if (It == StackElem.AlignedMap.end()) { 711 assert(NewDE && "Unexpected nullptr expr to be added into aligned map"); 712 StackElem.AlignedMap[D] = NewDE; 713 return nullptr; 714 } else { 715 assert(It->second && "Unexpected nullptr expr in the aligned map"); 716 return It->second; 717 } 718 return nullptr; 719 } 720 721 void DSAStackTy::addLoopControlVariable(ValueDecl *D, VarDecl *Capture) { 722 assert(!isStackEmpty() && "Data-sharing attributes stack is empty"); 723 D = getCanonicalDecl(D); 724 auto &StackElem = Stack.back().first.back(); 725 StackElem.LCVMap.insert( 726 {D, LCDeclInfo(StackElem.LCVMap.size() + 1, Capture)}); 727 } 728 729 DSAStackTy::LCDeclInfo DSAStackTy::isLoopControlVariable(ValueDecl *D) { 730 assert(!isStackEmpty() && "Data-sharing attributes stack is empty"); 731 D = getCanonicalDecl(D); 732 auto &StackElem = Stack.back().first.back(); 733 auto It = StackElem.LCVMap.find(D); 734 if (It != StackElem.LCVMap.end()) 735 return It->second; 736 return {0, nullptr}; 737 } 738 739 DSAStackTy::LCDeclInfo DSAStackTy::isParentLoopControlVariable(ValueDecl *D) { 740 assert(!isStackEmpty() && Stack.back().first.size() > 1 && 741 "Data-sharing attributes stack is empty"); 742 D = getCanonicalDecl(D); 743 auto &StackElem = *std::next(Stack.back().first.rbegin()); 744 auto It = StackElem.LCVMap.find(D); 745 if (It != StackElem.LCVMap.end()) 746 return It->second; 747 return {0, nullptr}; 748 } 749 750 ValueDecl *DSAStackTy::getParentLoopControlVariable(unsigned I) { 751 assert(!isStackEmpty() && Stack.back().first.size() > 1 && 752 "Data-sharing attributes stack is empty"); 753 auto &StackElem = *std::next(Stack.back().first.rbegin()); 754 if (StackElem.LCVMap.size() < I) 755 return nullptr; 756 for (auto &Pair : StackElem.LCVMap) 757 if (Pair.second.first == I) 758 return Pair.first; 759 return nullptr; 760 } 761 762 void DSAStackTy::addDSA(ValueDecl *D, Expr *E, OpenMPClauseKind A, 763 DeclRefExpr *PrivateCopy) { 764 D = getCanonicalDecl(D); 765 if (A == OMPC_threadprivate) { 766 auto &Data = Threadprivates[D]; 767 Data.Attributes = A; 768 Data.RefExpr.setPointer(E); 769 Data.PrivateCopy = nullptr; 770 } else { 771 assert(!isStackEmpty() && "Data-sharing attributes stack is empty"); 772 auto &Data = Stack.back().first.back().SharingMap[D]; 773 assert(Data.Attributes == OMPC_unknown || (A == Data.Attributes) || 774 (A == OMPC_firstprivate && Data.Attributes == OMPC_lastprivate) || 775 (A == OMPC_lastprivate && Data.Attributes == OMPC_firstprivate) || 776 (isLoopControlVariable(D).first && A == OMPC_private)); 777 if (A == OMPC_lastprivate && Data.Attributes == OMPC_firstprivate) { 778 Data.RefExpr.setInt(/*IntVal=*/true); 779 return; 780 } 781 const bool IsLastprivate = 782 A == OMPC_lastprivate || Data.Attributes == OMPC_lastprivate; 783 Data.Attributes = A; 784 Data.RefExpr.setPointerAndInt(E, IsLastprivate); 785 Data.PrivateCopy = PrivateCopy; 786 if (PrivateCopy) { 787 auto &Data = Stack.back().first.back().SharingMap[PrivateCopy->getDecl()]; 788 Data.Attributes = A; 789 Data.RefExpr.setPointerAndInt(PrivateCopy, IsLastprivate); 790 Data.PrivateCopy = nullptr; 791 } 792 } 793 } 794 795 /// \brief Build a variable declaration for OpenMP loop iteration variable. 796 static VarDecl *buildVarDecl(Sema &SemaRef, SourceLocation Loc, QualType Type, 797 StringRef Name, const AttrVec *Attrs = nullptr) { 798 DeclContext *DC = SemaRef.CurContext; 799 IdentifierInfo *II = &SemaRef.PP.getIdentifierTable().get(Name); 800 TypeSourceInfo *TInfo = SemaRef.Context.getTrivialTypeSourceInfo(Type, Loc); 801 VarDecl *Decl = 802 VarDecl::Create(SemaRef.Context, DC, Loc, Loc, II, Type, TInfo, SC_None); 803 if (Attrs) { 804 for (specific_attr_iterator<AlignedAttr> I(Attrs->begin()), E(Attrs->end()); 805 I != E; ++I) 806 Decl->addAttr(*I); 807 } 808 Decl->setImplicit(); 809 return Decl; 810 } 811 812 static DeclRefExpr *buildDeclRefExpr(Sema &S, VarDecl *D, QualType Ty, 813 SourceLocation Loc, 814 bool RefersToCapture = false) { 815 D->setReferenced(); 816 D->markUsed(S.Context); 817 return DeclRefExpr::Create(S.getASTContext(), NestedNameSpecifierLoc(), 818 SourceLocation(), D, RefersToCapture, Loc, Ty, 819 VK_LValue); 820 } 821 822 void DSAStackTy::addTaskgroupReductionData(ValueDecl *D, SourceRange SR, 823 BinaryOperatorKind BOK) { 824 D = getCanonicalDecl(D); 825 assert(!isStackEmpty() && "Data-sharing attributes stack is empty"); 826 assert( 827 Stack.back().first.back().SharingMap[D].Attributes == OMPC_reduction && 828 "Additional reduction info may be specified only for reduction items."); 829 auto &ReductionData = Stack.back().first.back().ReductionMap[D]; 830 assert(ReductionData.ReductionRange.isInvalid() && 831 Stack.back().first.back().Directive == OMPD_taskgroup && 832 "Additional reduction info may be specified only once for reduction " 833 "items."); 834 ReductionData.set(BOK, SR); 835 Expr *&TaskgroupReductionRef = 836 Stack.back().first.back().TaskgroupReductionRef; 837 if (!TaskgroupReductionRef) { 838 auto *VD = buildVarDecl(SemaRef, SourceLocation(), 839 SemaRef.Context.VoidPtrTy, ".task_red."); 840 TaskgroupReductionRef = buildDeclRefExpr( 841 SemaRef, VD, SemaRef.Context.VoidPtrTy, SourceLocation()); 842 } 843 } 844 845 void DSAStackTy::addTaskgroupReductionData(ValueDecl *D, SourceRange SR, 846 const Expr *ReductionRef) { 847 D = getCanonicalDecl(D); 848 assert(!isStackEmpty() && "Data-sharing attributes stack is empty"); 849 assert( 850 Stack.back().first.back().SharingMap[D].Attributes == OMPC_reduction && 851 "Additional reduction info may be specified only for reduction items."); 852 auto &ReductionData = Stack.back().first.back().ReductionMap[D]; 853 assert(ReductionData.ReductionRange.isInvalid() && 854 Stack.back().first.back().Directive == OMPD_taskgroup && 855 "Additional reduction info may be specified only once for reduction " 856 "items."); 857 ReductionData.set(ReductionRef, SR); 858 Expr *&TaskgroupReductionRef = 859 Stack.back().first.back().TaskgroupReductionRef; 860 if (!TaskgroupReductionRef) { 861 auto *VD = buildVarDecl(SemaRef, SourceLocation(), 862 SemaRef.Context.VoidPtrTy, ".task_red."); 863 TaskgroupReductionRef = buildDeclRefExpr( 864 SemaRef, VD, SemaRef.Context.VoidPtrTy, SourceLocation()); 865 } 866 } 867 868 DSAStackTy::DSAVarData 869 DSAStackTy::getTopMostTaskgroupReductionData(ValueDecl *D, SourceRange &SR, 870 BinaryOperatorKind &BOK, 871 Expr *&TaskgroupDescriptor) { 872 D = getCanonicalDecl(D); 873 assert(!isStackEmpty() && "Data-sharing attributes stack is empty."); 874 if (Stack.back().first.empty()) 875 return DSAVarData(); 876 for (auto I = std::next(Stack.back().first.rbegin(), 1), 877 E = Stack.back().first.rend(); 878 I != E; std::advance(I, 1)) { 879 auto &Data = I->SharingMap[D]; 880 if (Data.Attributes != OMPC_reduction || I->Directive != OMPD_taskgroup) 881 continue; 882 auto &ReductionData = I->ReductionMap[D]; 883 if (!ReductionData.ReductionOp || 884 ReductionData.ReductionOp.is<const Expr *>()) 885 return DSAVarData(); 886 SR = ReductionData.ReductionRange; 887 BOK = ReductionData.ReductionOp.get<ReductionData::BOKPtrType>(); 888 assert(I->TaskgroupReductionRef && "taskgroup reduction reference " 889 "expression for the descriptor is not " 890 "set."); 891 TaskgroupDescriptor = I->TaskgroupReductionRef; 892 return DSAVarData(OMPD_taskgroup, OMPC_reduction, Data.RefExpr.getPointer(), 893 Data.PrivateCopy, I->DefaultAttrLoc); 894 } 895 return DSAVarData(); 896 } 897 898 DSAStackTy::DSAVarData 899 DSAStackTy::getTopMostTaskgroupReductionData(ValueDecl *D, SourceRange &SR, 900 const Expr *&ReductionRef, 901 Expr *&TaskgroupDescriptor) { 902 D = getCanonicalDecl(D); 903 assert(!isStackEmpty() && "Data-sharing attributes stack is empty."); 904 if (Stack.back().first.empty()) 905 return DSAVarData(); 906 for (auto I = std::next(Stack.back().first.rbegin(), 1), 907 E = Stack.back().first.rend(); 908 I != E; std::advance(I, 1)) { 909 auto &Data = I->SharingMap[D]; 910 if (Data.Attributes != OMPC_reduction || I->Directive != OMPD_taskgroup) 911 continue; 912 auto &ReductionData = I->ReductionMap[D]; 913 if (!ReductionData.ReductionOp || 914 !ReductionData.ReductionOp.is<const Expr *>()) 915 return DSAVarData(); 916 SR = ReductionData.ReductionRange; 917 ReductionRef = ReductionData.ReductionOp.get<const Expr *>(); 918 assert(I->TaskgroupReductionRef && "taskgroup reduction reference " 919 "expression for the descriptor is not " 920 "set."); 921 TaskgroupDescriptor = I->TaskgroupReductionRef; 922 return DSAVarData(OMPD_taskgroup, OMPC_reduction, Data.RefExpr.getPointer(), 923 Data.PrivateCopy, I->DefaultAttrLoc); 924 } 925 return DSAVarData(); 926 } 927 928 bool DSAStackTy::isOpenMPLocal(VarDecl *D, StackTy::reverse_iterator Iter) { 929 D = D->getCanonicalDecl(); 930 if (!isStackEmpty() && Stack.back().first.size() > 1) { 931 reverse_iterator I = Iter, E = Stack.back().first.rend(); 932 Scope *TopScope = nullptr; 933 while (I != E && !isParallelOrTaskRegion(I->Directive)) 934 ++I; 935 if (I == E) 936 return false; 937 TopScope = I->CurScope ? I->CurScope->getParent() : nullptr; 938 Scope *CurScope = getCurScope(); 939 while (CurScope != TopScope && !CurScope->isDeclScope(D)) 940 CurScope = CurScope->getParent(); 941 return CurScope != TopScope; 942 } 943 return false; 944 } 945 946 DSAStackTy::DSAVarData DSAStackTy::getTopDSA(ValueDecl *D, bool FromParent) { 947 D = getCanonicalDecl(D); 948 DSAVarData DVar; 949 950 // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced 951 // in a Construct, C/C++, predetermined, p.1] 952 // Variables appearing in threadprivate directives are threadprivate. 953 auto *VD = dyn_cast<VarDecl>(D); 954 if ((VD && VD->getTLSKind() != VarDecl::TLS_None && 955 !(VD->hasAttr<OMPThreadPrivateDeclAttr>() && 956 SemaRef.getLangOpts().OpenMPUseTLS && 957 SemaRef.getASTContext().getTargetInfo().isTLSSupported())) || 958 (VD && VD->getStorageClass() == SC_Register && 959 VD->hasAttr<AsmLabelAttr>() && !VD->isLocalVarDecl())) { 960 addDSA(D, buildDeclRefExpr(SemaRef, VD, D->getType().getNonReferenceType(), 961 D->getLocation()), 962 OMPC_threadprivate); 963 } 964 auto TI = Threadprivates.find(D); 965 if (TI != Threadprivates.end()) { 966 DVar.RefExpr = TI->getSecond().RefExpr.getPointer(); 967 DVar.CKind = OMPC_threadprivate; 968 return DVar; 969 } 970 971 if (isStackEmpty()) 972 // Not in OpenMP execution region and top scope was already checked. 973 return DVar; 974 975 // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced 976 // in a Construct, C/C++, predetermined, p.4] 977 // Static data members are shared. 978 // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced 979 // in a Construct, C/C++, predetermined, p.7] 980 // Variables with static storage duration that are declared in a scope 981 // inside the construct are shared. 982 auto &&MatchesAlways = [](OpenMPDirectiveKind) -> bool { return true; }; 983 if (VD && VD->isStaticDataMember()) { 984 DSAVarData DVarTemp = hasDSA(D, isOpenMPPrivate, MatchesAlways, FromParent); 985 if (DVarTemp.CKind != OMPC_unknown && DVarTemp.RefExpr) 986 return DVar; 987 988 DVar.CKind = OMPC_shared; 989 return DVar; 990 } 991 992 QualType Type = D->getType().getNonReferenceType().getCanonicalType(); 993 bool IsConstant = Type.isConstant(SemaRef.getASTContext()); 994 Type = SemaRef.getASTContext().getBaseElementType(Type); 995 // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced 996 // in a Construct, C/C++, predetermined, p.6] 997 // Variables with const qualified type having no mutable member are 998 // shared. 999 CXXRecordDecl *RD = 1000 SemaRef.getLangOpts().CPlusPlus ? Type->getAsCXXRecordDecl() : nullptr; 1001 if (auto *CTSD = dyn_cast_or_null<ClassTemplateSpecializationDecl>(RD)) 1002 if (auto *CTD = CTSD->getSpecializedTemplate()) 1003 RD = CTD->getTemplatedDecl(); 1004 if (IsConstant && 1005 !(SemaRef.getLangOpts().CPlusPlus && RD && RD->hasDefinition() && 1006 RD->hasMutableFields())) { 1007 // Variables with const-qualified type having no mutable member may be 1008 // listed in a firstprivate clause, even if they are static data members. 1009 DSAVarData DVarTemp = hasDSA( 1010 D, [](OpenMPClauseKind C) -> bool { return C == OMPC_firstprivate; }, 1011 MatchesAlways, FromParent); 1012 if (DVarTemp.CKind == OMPC_firstprivate && DVarTemp.RefExpr) 1013 return DVar; 1014 1015 DVar.CKind = OMPC_shared; 1016 return DVar; 1017 } 1018 1019 // Explicitly specified attributes and local variables with predetermined 1020 // attributes. 1021 auto I = Stack.back().first.rbegin(); 1022 auto EndI = Stack.back().first.rend(); 1023 if (FromParent && I != EndI) 1024 std::advance(I, 1); 1025 if (I->SharingMap.count(D)) { 1026 DVar.RefExpr = I->SharingMap[D].RefExpr.getPointer(); 1027 DVar.PrivateCopy = I->SharingMap[D].PrivateCopy; 1028 DVar.CKind = I->SharingMap[D].Attributes; 1029 DVar.ImplicitDSALoc = I->DefaultAttrLoc; 1030 DVar.DKind = I->Directive; 1031 } 1032 1033 return DVar; 1034 } 1035 1036 DSAStackTy::DSAVarData DSAStackTy::getImplicitDSA(ValueDecl *D, 1037 bool FromParent) { 1038 if (isStackEmpty()) { 1039 StackTy::reverse_iterator I; 1040 return getDSA(I, D); 1041 } 1042 D = getCanonicalDecl(D); 1043 auto StartI = Stack.back().first.rbegin(); 1044 auto EndI = Stack.back().first.rend(); 1045 if (FromParent && StartI != EndI) 1046 std::advance(StartI, 1); 1047 return getDSA(StartI, D); 1048 } 1049 1050 DSAStackTy::DSAVarData 1051 DSAStackTy::hasDSA(ValueDecl *D, 1052 const llvm::function_ref<bool(OpenMPClauseKind)> &CPred, 1053 const llvm::function_ref<bool(OpenMPDirectiveKind)> &DPred, 1054 bool FromParent) { 1055 if (isStackEmpty()) 1056 return {}; 1057 D = getCanonicalDecl(D); 1058 auto I = Stack.back().first.rbegin(); 1059 auto EndI = Stack.back().first.rend(); 1060 if (FromParent && I != EndI) 1061 std::advance(I, 1); 1062 for (; I != EndI; std::advance(I, 1)) { 1063 if (!DPred(I->Directive) && !isParallelOrTaskRegion(I->Directive)) 1064 continue; 1065 auto NewI = I; 1066 DSAVarData DVar = getDSA(NewI, D); 1067 if (I == NewI && CPred(DVar.CKind)) 1068 return DVar; 1069 } 1070 return {}; 1071 } 1072 1073 DSAStackTy::DSAVarData DSAStackTy::hasInnermostDSA( 1074 ValueDecl *D, const llvm::function_ref<bool(OpenMPClauseKind)> &CPred, 1075 const llvm::function_ref<bool(OpenMPDirectiveKind)> &DPred, 1076 bool FromParent) { 1077 if (isStackEmpty()) 1078 return {}; 1079 D = getCanonicalDecl(D); 1080 auto StartI = Stack.back().first.rbegin(); 1081 auto EndI = Stack.back().first.rend(); 1082 if (FromParent && StartI != EndI) 1083 std::advance(StartI, 1); 1084 if (StartI == EndI || !DPred(StartI->Directive)) 1085 return {}; 1086 auto NewI = StartI; 1087 DSAVarData DVar = getDSA(NewI, D); 1088 return (NewI == StartI && CPred(DVar.CKind)) ? DVar : DSAVarData(); 1089 } 1090 1091 bool DSAStackTy::hasExplicitDSA( 1092 ValueDecl *D, const llvm::function_ref<bool(OpenMPClauseKind)> &CPred, 1093 unsigned Level, bool NotLastprivate) { 1094 if (CPred(ClauseKindMode)) 1095 return true; 1096 if (isStackEmpty()) 1097 return false; 1098 D = getCanonicalDecl(D); 1099 auto StartI = Stack.back().first.begin(); 1100 auto EndI = Stack.back().first.end(); 1101 if (std::distance(StartI, EndI) <= (int)Level) 1102 return false; 1103 std::advance(StartI, Level); 1104 return (StartI->SharingMap.count(D) > 0) && 1105 StartI->SharingMap[D].RefExpr.getPointer() && 1106 CPred(StartI->SharingMap[D].Attributes) && 1107 (!NotLastprivate || !StartI->SharingMap[D].RefExpr.getInt()); 1108 } 1109 1110 bool DSAStackTy::hasExplicitDirective( 1111 const llvm::function_ref<bool(OpenMPDirectiveKind)> &DPred, 1112 unsigned Level) { 1113 if (isStackEmpty()) 1114 return false; 1115 auto StartI = Stack.back().first.begin(); 1116 auto EndI = Stack.back().first.end(); 1117 if (std::distance(StartI, EndI) <= (int)Level) 1118 return false; 1119 std::advance(StartI, Level); 1120 return DPred(StartI->Directive); 1121 } 1122 1123 bool DSAStackTy::hasDirective( 1124 const llvm::function_ref<bool(OpenMPDirectiveKind, 1125 const DeclarationNameInfo &, SourceLocation)> 1126 &DPred, 1127 bool FromParent) { 1128 // We look only in the enclosing region. 1129 if (isStackEmpty()) 1130 return false; 1131 auto StartI = std::next(Stack.back().first.rbegin()); 1132 auto EndI = Stack.back().first.rend(); 1133 if (FromParent && StartI != EndI) 1134 StartI = std::next(StartI); 1135 for (auto I = StartI, EE = EndI; I != EE; ++I) { 1136 if (DPred(I->Directive, I->DirectiveName, I->ConstructLoc)) 1137 return true; 1138 } 1139 return false; 1140 } 1141 1142 void Sema::InitDataSharingAttributesStack() { 1143 VarDataSharingAttributesStack = new DSAStackTy(*this); 1144 } 1145 1146 #define DSAStack static_cast<DSAStackTy *>(VarDataSharingAttributesStack) 1147 1148 void Sema::pushOpenMPFunctionRegion() { 1149 DSAStack->pushFunction(); 1150 } 1151 1152 void Sema::popOpenMPFunctionRegion(const FunctionScopeInfo *OldFSI) { 1153 DSAStack->popFunction(OldFSI); 1154 } 1155 1156 bool Sema::IsOpenMPCapturedByRef(ValueDecl *D, unsigned Level) { 1157 assert(LangOpts.OpenMP && "OpenMP is not allowed"); 1158 1159 auto &Ctx = getASTContext(); 1160 bool IsByRef = true; 1161 1162 // Find the directive that is associated with the provided scope. 1163 D = cast<ValueDecl>(D->getCanonicalDecl()); 1164 auto Ty = D->getType(); 1165 1166 if (DSAStack->hasExplicitDirective(isOpenMPTargetExecutionDirective, Level)) { 1167 // This table summarizes how a given variable should be passed to the device 1168 // given its type and the clauses where it appears. This table is based on 1169 // the description in OpenMP 4.5 [2.10.4, target Construct] and 1170 // OpenMP 4.5 [2.15.5, Data-mapping Attribute Rules and Clauses]. 1171 // 1172 // ========================================================================= 1173 // | type | defaultmap | pvt | first | is_device_ptr | map | res. | 1174 // | |(tofrom:scalar)| | pvt | | | | 1175 // ========================================================================= 1176 // | scl | | | | - | | bycopy| 1177 // | scl | | - | x | - | - | bycopy| 1178 // | scl | | x | - | - | - | null | 1179 // | scl | x | | | - | | byref | 1180 // | scl | x | - | x | - | - | bycopy| 1181 // | scl | x | x | - | - | - | null | 1182 // | scl | | - | - | - | x | byref | 1183 // | scl | x | - | - | - | x | byref | 1184 // 1185 // | agg | n.a. | | | - | | byref | 1186 // | agg | n.a. | - | x | - | - | byref | 1187 // | agg | n.a. | x | - | - | - | null | 1188 // | agg | n.a. | - | - | - | x | byref | 1189 // | agg | n.a. | - | - | - | x[] | byref | 1190 // 1191 // | ptr | n.a. | | | - | | bycopy| 1192 // | ptr | n.a. | - | x | - | - | bycopy| 1193 // | ptr | n.a. | x | - | - | - | null | 1194 // | ptr | n.a. | - | - | - | x | byref | 1195 // | ptr | n.a. | - | - | - | x[] | bycopy| 1196 // | ptr | n.a. | - | - | x | | bycopy| 1197 // | ptr | n.a. | - | - | x | x | bycopy| 1198 // | ptr | n.a. | - | - | x | x[] | bycopy| 1199 // ========================================================================= 1200 // Legend: 1201 // scl - scalar 1202 // ptr - pointer 1203 // agg - aggregate 1204 // x - applies 1205 // - - invalid in this combination 1206 // [] - mapped with an array section 1207 // byref - should be mapped by reference 1208 // byval - should be mapped by value 1209 // null - initialize a local variable to null on the device 1210 // 1211 // Observations: 1212 // - All scalar declarations that show up in a map clause have to be passed 1213 // by reference, because they may have been mapped in the enclosing data 1214 // environment. 1215 // - If the scalar value does not fit the size of uintptr, it has to be 1216 // passed by reference, regardless the result in the table above. 1217 // - For pointers mapped by value that have either an implicit map or an 1218 // array section, the runtime library may pass the NULL value to the 1219 // device instead of the value passed to it by the compiler. 1220 1221 if (Ty->isReferenceType()) 1222 Ty = Ty->castAs<ReferenceType>()->getPointeeType(); 1223 1224 // Locate map clauses and see if the variable being captured is referred to 1225 // in any of those clauses. Here we only care about variables, not fields, 1226 // because fields are part of aggregates. 1227 bool IsVariableUsedInMapClause = false; 1228 bool IsVariableAssociatedWithSection = false; 1229 1230 DSAStack->checkMappableExprComponentListsForDeclAtLevel( 1231 D, Level, [&](OMPClauseMappableExprCommon::MappableExprComponentListRef 1232 MapExprComponents, 1233 OpenMPClauseKind WhereFoundClauseKind) { 1234 // Only the map clause information influences how a variable is 1235 // captured. E.g. is_device_ptr does not require changing the default 1236 // behavior. 1237 if (WhereFoundClauseKind != OMPC_map) 1238 return false; 1239 1240 auto EI = MapExprComponents.rbegin(); 1241 auto EE = MapExprComponents.rend(); 1242 1243 assert(EI != EE && "Invalid map expression!"); 1244 1245 if (isa<DeclRefExpr>(EI->getAssociatedExpression())) 1246 IsVariableUsedInMapClause |= EI->getAssociatedDeclaration() == D; 1247 1248 ++EI; 1249 if (EI == EE) 1250 return false; 1251 1252 if (isa<ArraySubscriptExpr>(EI->getAssociatedExpression()) || 1253 isa<OMPArraySectionExpr>(EI->getAssociatedExpression()) || 1254 isa<MemberExpr>(EI->getAssociatedExpression())) { 1255 IsVariableAssociatedWithSection = true; 1256 // There is nothing more we need to know about this variable. 1257 return true; 1258 } 1259 1260 // Keep looking for more map info. 1261 return false; 1262 }); 1263 1264 if (IsVariableUsedInMapClause) { 1265 // If variable is identified in a map clause it is always captured by 1266 // reference except if it is a pointer that is dereferenced somehow. 1267 IsByRef = !(Ty->isPointerType() && IsVariableAssociatedWithSection); 1268 } else { 1269 // By default, all the data that has a scalar type is mapped by copy. 1270 IsByRef = !Ty->isScalarType() || 1271 DSAStack->getDefaultDMAAtLevel(Level) == DMA_tofrom_scalar; 1272 } 1273 } 1274 1275 if (IsByRef && Ty.getNonReferenceType()->isScalarType()) { 1276 IsByRef = !DSAStack->hasExplicitDSA( 1277 D, [](OpenMPClauseKind K) -> bool { return K == OMPC_firstprivate; }, 1278 Level, /*NotLastprivate=*/true); 1279 } 1280 1281 // When passing data by copy, we need to make sure it fits the uintptr size 1282 // and alignment, because the runtime library only deals with uintptr types. 1283 // If it does not fit the uintptr size, we need to pass the data by reference 1284 // instead. 1285 if (!IsByRef && 1286 (Ctx.getTypeSizeInChars(Ty) > 1287 Ctx.getTypeSizeInChars(Ctx.getUIntPtrType()) || 1288 Ctx.getDeclAlign(D) > Ctx.getTypeAlignInChars(Ctx.getUIntPtrType()))) { 1289 IsByRef = true; 1290 } 1291 1292 return IsByRef; 1293 } 1294 1295 unsigned Sema::getOpenMPNestingLevel() const { 1296 assert(getLangOpts().OpenMP); 1297 return DSAStack->getNestingLevel(); 1298 } 1299 1300 VarDecl *Sema::IsOpenMPCapturedDecl(ValueDecl *D) { 1301 assert(LangOpts.OpenMP && "OpenMP is not allowed"); 1302 D = getCanonicalDecl(D); 1303 1304 // If we are attempting to capture a global variable in a directive with 1305 // 'target' we return true so that this global is also mapped to the device. 1306 // 1307 // FIXME: If the declaration is enclosed in a 'declare target' directive, 1308 // then it should not be captured. Therefore, an extra check has to be 1309 // inserted here once support for 'declare target' is added. 1310 // 1311 auto *VD = dyn_cast<VarDecl>(D); 1312 if (VD && !VD->hasLocalStorage()) { 1313 if (isOpenMPTargetExecutionDirective(DSAStack->getCurrentDirective()) && 1314 !DSAStack->isClauseParsingMode()) 1315 return VD; 1316 if (DSAStack->hasDirective( 1317 [](OpenMPDirectiveKind K, const DeclarationNameInfo &, 1318 SourceLocation) -> bool { 1319 return isOpenMPTargetExecutionDirective(K); 1320 }, 1321 false)) 1322 return VD; 1323 } 1324 1325 if (DSAStack->getCurrentDirective() != OMPD_unknown && 1326 (!DSAStack->isClauseParsingMode() || 1327 DSAStack->getParentDirective() != OMPD_unknown)) { 1328 auto &&Info = DSAStack->isLoopControlVariable(D); 1329 if (Info.first || 1330 (VD && VD->hasLocalStorage() && 1331 isParallelOrTaskRegion(DSAStack->getCurrentDirective())) || 1332 (VD && DSAStack->isForceVarCapturing())) 1333 return VD ? VD : Info.second; 1334 auto DVarPrivate = DSAStack->getTopDSA(D, DSAStack->isClauseParsingMode()); 1335 if (DVarPrivate.CKind != OMPC_unknown && isOpenMPPrivate(DVarPrivate.CKind)) 1336 return VD ? VD : cast<VarDecl>(DVarPrivate.PrivateCopy->getDecl()); 1337 DVarPrivate = DSAStack->hasDSA( 1338 D, isOpenMPPrivate, [](OpenMPDirectiveKind) -> bool { return true; }, 1339 DSAStack->isClauseParsingMode()); 1340 if (DVarPrivate.CKind != OMPC_unknown) 1341 return VD ? VD : cast<VarDecl>(DVarPrivate.PrivateCopy->getDecl()); 1342 } 1343 return nullptr; 1344 } 1345 1346 bool Sema::isOpenMPPrivateDecl(ValueDecl *D, unsigned Level) { 1347 assert(LangOpts.OpenMP && "OpenMP is not allowed"); 1348 return DSAStack->hasExplicitDSA( 1349 D, [](OpenMPClauseKind K) -> bool { return K == OMPC_private; }, 1350 Level) || 1351 // Consider taskgroup reduction descriptor variable a private to avoid 1352 // possible capture in the region. 1353 (DSAStack->hasExplicitDirective( 1354 [](OpenMPDirectiveKind K) { return K == OMPD_taskgroup; }, 1355 Level) && 1356 DSAStack->isTaskgroupReductionRef(D, Level)); 1357 } 1358 1359 void Sema::setOpenMPCaptureKind(FieldDecl *FD, ValueDecl *D, unsigned Level) { 1360 assert(LangOpts.OpenMP && "OpenMP is not allowed"); 1361 D = getCanonicalDecl(D); 1362 OpenMPClauseKind OMPC = OMPC_unknown; 1363 for (unsigned I = DSAStack->getNestingLevel() + 1; I > Level; --I) { 1364 const unsigned NewLevel = I - 1; 1365 if (DSAStack->hasExplicitDSA(D, 1366 [&OMPC](const OpenMPClauseKind K) { 1367 if (isOpenMPPrivate(K)) { 1368 OMPC = K; 1369 return true; 1370 } 1371 return false; 1372 }, 1373 NewLevel)) 1374 break; 1375 if (DSAStack->checkMappableExprComponentListsForDeclAtLevel( 1376 D, NewLevel, 1377 [](OMPClauseMappableExprCommon::MappableExprComponentListRef, 1378 OpenMPClauseKind) { return true; })) { 1379 OMPC = OMPC_map; 1380 break; 1381 } 1382 if (DSAStack->hasExplicitDirective(isOpenMPTargetExecutionDirective, 1383 NewLevel)) { 1384 OMPC = OMPC_firstprivate; 1385 break; 1386 } 1387 } 1388 if (OMPC != OMPC_unknown) 1389 FD->addAttr(OMPCaptureKindAttr::CreateImplicit(Context, OMPC)); 1390 } 1391 1392 bool Sema::isOpenMPTargetCapturedDecl(ValueDecl *D, unsigned Level) { 1393 assert(LangOpts.OpenMP && "OpenMP is not allowed"); 1394 // Return true if the current level is no longer enclosed in a target region. 1395 1396 auto *VD = dyn_cast<VarDecl>(D); 1397 return VD && !VD->hasLocalStorage() && 1398 DSAStack->hasExplicitDirective(isOpenMPTargetExecutionDirective, 1399 Level); 1400 } 1401 1402 void Sema::DestroyDataSharingAttributesStack() { delete DSAStack; } 1403 1404 void Sema::StartOpenMPDSABlock(OpenMPDirectiveKind DKind, 1405 const DeclarationNameInfo &DirName, 1406 Scope *CurScope, SourceLocation Loc) { 1407 DSAStack->push(DKind, DirName, CurScope, Loc); 1408 PushExpressionEvaluationContext( 1409 ExpressionEvaluationContext::PotentiallyEvaluated); 1410 } 1411 1412 void Sema::StartOpenMPClause(OpenMPClauseKind K) { 1413 DSAStack->setClauseParsingMode(K); 1414 } 1415 1416 void Sema::EndOpenMPClause() { 1417 DSAStack->setClauseParsingMode(/*K=*/OMPC_unknown); 1418 } 1419 1420 void Sema::EndOpenMPDSABlock(Stmt *CurDirective) { 1421 // OpenMP [2.14.3.5, Restrictions, C/C++, p.1] 1422 // A variable of class type (or array thereof) that appears in a lastprivate 1423 // clause requires an accessible, unambiguous default constructor for the 1424 // class type, unless the list item is also specified in a firstprivate 1425 // clause. 1426 if (auto *D = dyn_cast_or_null<OMPExecutableDirective>(CurDirective)) { 1427 for (auto *C : D->clauses()) { 1428 if (auto *Clause = dyn_cast<OMPLastprivateClause>(C)) { 1429 SmallVector<Expr *, 8> PrivateCopies; 1430 for (auto *DE : Clause->varlists()) { 1431 if (DE->isValueDependent() || DE->isTypeDependent()) { 1432 PrivateCopies.push_back(nullptr); 1433 continue; 1434 } 1435 auto *DRE = cast<DeclRefExpr>(DE->IgnoreParens()); 1436 VarDecl *VD = cast<VarDecl>(DRE->getDecl()); 1437 QualType Type = VD->getType().getNonReferenceType(); 1438 auto DVar = DSAStack->getTopDSA(VD, false); 1439 if (DVar.CKind == OMPC_lastprivate) { 1440 // Generate helper private variable and initialize it with the 1441 // default value. The address of the original variable is replaced 1442 // by the address of the new private variable in CodeGen. This new 1443 // variable is not added to IdResolver, so the code in the OpenMP 1444 // region uses original variable for proper diagnostics. 1445 auto *VDPrivate = buildVarDecl( 1446 *this, DE->getExprLoc(), Type.getUnqualifiedType(), 1447 VD->getName(), VD->hasAttrs() ? &VD->getAttrs() : nullptr); 1448 ActOnUninitializedDecl(VDPrivate); 1449 if (VDPrivate->isInvalidDecl()) 1450 continue; 1451 PrivateCopies.push_back(buildDeclRefExpr( 1452 *this, VDPrivate, DE->getType(), DE->getExprLoc())); 1453 } else { 1454 // The variable is also a firstprivate, so initialization sequence 1455 // for private copy is generated already. 1456 PrivateCopies.push_back(nullptr); 1457 } 1458 } 1459 // Set initializers to private copies if no errors were found. 1460 if (PrivateCopies.size() == Clause->varlist_size()) 1461 Clause->setPrivateCopies(PrivateCopies); 1462 } 1463 } 1464 } 1465 1466 DSAStack->pop(); 1467 DiscardCleanupsInEvaluationContext(); 1468 PopExpressionEvaluationContext(); 1469 } 1470 1471 static bool FinishOpenMPLinearClause(OMPLinearClause &Clause, DeclRefExpr *IV, 1472 Expr *NumIterations, Sema &SemaRef, 1473 Scope *S, DSAStackTy *Stack); 1474 1475 namespace { 1476 1477 class VarDeclFilterCCC : public CorrectionCandidateCallback { 1478 private: 1479 Sema &SemaRef; 1480 1481 public: 1482 explicit VarDeclFilterCCC(Sema &S) : SemaRef(S) {} 1483 bool ValidateCandidate(const TypoCorrection &Candidate) override { 1484 NamedDecl *ND = Candidate.getCorrectionDecl(); 1485 if (auto *VD = dyn_cast_or_null<VarDecl>(ND)) { 1486 return VD->hasGlobalStorage() && 1487 SemaRef.isDeclInScope(ND, SemaRef.getCurLexicalContext(), 1488 SemaRef.getCurScope()); 1489 } 1490 return false; 1491 } 1492 }; 1493 1494 class VarOrFuncDeclFilterCCC : public CorrectionCandidateCallback { 1495 private: 1496 Sema &SemaRef; 1497 1498 public: 1499 explicit VarOrFuncDeclFilterCCC(Sema &S) : SemaRef(S) {} 1500 bool ValidateCandidate(const TypoCorrection &Candidate) override { 1501 NamedDecl *ND = Candidate.getCorrectionDecl(); 1502 if (isa<VarDecl>(ND) || isa<FunctionDecl>(ND)) { 1503 return SemaRef.isDeclInScope(ND, SemaRef.getCurLexicalContext(), 1504 SemaRef.getCurScope()); 1505 } 1506 return false; 1507 } 1508 }; 1509 1510 } // namespace 1511 1512 ExprResult Sema::ActOnOpenMPIdExpression(Scope *CurScope, 1513 CXXScopeSpec &ScopeSpec, 1514 const DeclarationNameInfo &Id) { 1515 LookupResult Lookup(*this, Id, LookupOrdinaryName); 1516 LookupParsedName(Lookup, CurScope, &ScopeSpec, true); 1517 1518 if (Lookup.isAmbiguous()) 1519 return ExprError(); 1520 1521 VarDecl *VD; 1522 if (!Lookup.isSingleResult()) { 1523 if (TypoCorrection Corrected = CorrectTypo( 1524 Id, LookupOrdinaryName, CurScope, nullptr, 1525 llvm::make_unique<VarDeclFilterCCC>(*this), CTK_ErrorRecovery)) { 1526 diagnoseTypo(Corrected, 1527 PDiag(Lookup.empty() 1528 ? diag::err_undeclared_var_use_suggest 1529 : diag::err_omp_expected_var_arg_suggest) 1530 << Id.getName()); 1531 VD = Corrected.getCorrectionDeclAs<VarDecl>(); 1532 } else { 1533 Diag(Id.getLoc(), Lookup.empty() ? diag::err_undeclared_var_use 1534 : diag::err_omp_expected_var_arg) 1535 << Id.getName(); 1536 return ExprError(); 1537 } 1538 } else { 1539 if (!(VD = Lookup.getAsSingle<VarDecl>())) { 1540 Diag(Id.getLoc(), diag::err_omp_expected_var_arg) << Id.getName(); 1541 Diag(Lookup.getFoundDecl()->getLocation(), diag::note_declared_at); 1542 return ExprError(); 1543 } 1544 } 1545 Lookup.suppressDiagnostics(); 1546 1547 // OpenMP [2.9.2, Syntax, C/C++] 1548 // Variables must be file-scope, namespace-scope, or static block-scope. 1549 if (!VD->hasGlobalStorage()) { 1550 Diag(Id.getLoc(), diag::err_omp_global_var_arg) 1551 << getOpenMPDirectiveName(OMPD_threadprivate) << !VD->isStaticLocal(); 1552 bool IsDecl = 1553 VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly; 1554 Diag(VD->getLocation(), 1555 IsDecl ? diag::note_previous_decl : diag::note_defined_here) 1556 << VD; 1557 return ExprError(); 1558 } 1559 1560 VarDecl *CanonicalVD = VD->getCanonicalDecl(); 1561 NamedDecl *ND = cast<NamedDecl>(CanonicalVD); 1562 // OpenMP [2.9.2, Restrictions, C/C++, p.2] 1563 // A threadprivate directive for file-scope variables must appear outside 1564 // any definition or declaration. 1565 if (CanonicalVD->getDeclContext()->isTranslationUnit() && 1566 !getCurLexicalContext()->isTranslationUnit()) { 1567 Diag(Id.getLoc(), diag::err_omp_var_scope) 1568 << getOpenMPDirectiveName(OMPD_threadprivate) << VD; 1569 bool IsDecl = 1570 VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly; 1571 Diag(VD->getLocation(), 1572 IsDecl ? diag::note_previous_decl : diag::note_defined_here) 1573 << VD; 1574 return ExprError(); 1575 } 1576 // OpenMP [2.9.2, Restrictions, C/C++, p.3] 1577 // A threadprivate directive for static class member variables must appear 1578 // in the class definition, in the same scope in which the member 1579 // variables are declared. 1580 if (CanonicalVD->isStaticDataMember() && 1581 !CanonicalVD->getDeclContext()->Equals(getCurLexicalContext())) { 1582 Diag(Id.getLoc(), diag::err_omp_var_scope) 1583 << getOpenMPDirectiveName(OMPD_threadprivate) << VD; 1584 bool IsDecl = 1585 VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly; 1586 Diag(VD->getLocation(), 1587 IsDecl ? diag::note_previous_decl : diag::note_defined_here) 1588 << VD; 1589 return ExprError(); 1590 } 1591 // OpenMP [2.9.2, Restrictions, C/C++, p.4] 1592 // A threadprivate directive for namespace-scope variables must appear 1593 // outside any definition or declaration other than the namespace 1594 // definition itself. 1595 if (CanonicalVD->getDeclContext()->isNamespace() && 1596 (!getCurLexicalContext()->isFileContext() || 1597 !getCurLexicalContext()->Encloses(CanonicalVD->getDeclContext()))) { 1598 Diag(Id.getLoc(), diag::err_omp_var_scope) 1599 << getOpenMPDirectiveName(OMPD_threadprivate) << VD; 1600 bool IsDecl = 1601 VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly; 1602 Diag(VD->getLocation(), 1603 IsDecl ? diag::note_previous_decl : diag::note_defined_here) 1604 << VD; 1605 return ExprError(); 1606 } 1607 // OpenMP [2.9.2, Restrictions, C/C++, p.6] 1608 // A threadprivate directive for static block-scope variables must appear 1609 // in the scope of the variable and not in a nested scope. 1610 if (CanonicalVD->isStaticLocal() && CurScope && 1611 !isDeclInScope(ND, getCurLexicalContext(), CurScope)) { 1612 Diag(Id.getLoc(), diag::err_omp_var_scope) 1613 << getOpenMPDirectiveName(OMPD_threadprivate) << VD; 1614 bool IsDecl = 1615 VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly; 1616 Diag(VD->getLocation(), 1617 IsDecl ? diag::note_previous_decl : diag::note_defined_here) 1618 << VD; 1619 return ExprError(); 1620 } 1621 1622 // OpenMP [2.9.2, Restrictions, C/C++, p.2-6] 1623 // A threadprivate directive must lexically precede all references to any 1624 // of the variables in its list. 1625 if (VD->isUsed() && !DSAStack->isThreadPrivate(VD)) { 1626 Diag(Id.getLoc(), diag::err_omp_var_used) 1627 << getOpenMPDirectiveName(OMPD_threadprivate) << VD; 1628 return ExprError(); 1629 } 1630 1631 QualType ExprType = VD->getType().getNonReferenceType(); 1632 return DeclRefExpr::Create(Context, NestedNameSpecifierLoc(), 1633 SourceLocation(), VD, 1634 /*RefersToEnclosingVariableOrCapture=*/false, 1635 Id.getLoc(), ExprType, VK_LValue); 1636 } 1637 1638 Sema::DeclGroupPtrTy 1639 Sema::ActOnOpenMPThreadprivateDirective(SourceLocation Loc, 1640 ArrayRef<Expr *> VarList) { 1641 if (OMPThreadPrivateDecl *D = CheckOMPThreadPrivateDecl(Loc, VarList)) { 1642 CurContext->addDecl(D); 1643 return DeclGroupPtrTy::make(DeclGroupRef(D)); 1644 } 1645 return nullptr; 1646 } 1647 1648 namespace { 1649 class LocalVarRefChecker : public ConstStmtVisitor<LocalVarRefChecker, bool> { 1650 Sema &SemaRef; 1651 1652 public: 1653 bool VisitDeclRefExpr(const DeclRefExpr *E) { 1654 if (auto *VD = dyn_cast<VarDecl>(E->getDecl())) { 1655 if (VD->hasLocalStorage()) { 1656 SemaRef.Diag(E->getLocStart(), 1657 diag::err_omp_local_var_in_threadprivate_init) 1658 << E->getSourceRange(); 1659 SemaRef.Diag(VD->getLocation(), diag::note_defined_here) 1660 << VD << VD->getSourceRange(); 1661 return true; 1662 } 1663 } 1664 return false; 1665 } 1666 bool VisitStmt(const Stmt *S) { 1667 for (auto Child : S->children()) { 1668 if (Child && Visit(Child)) 1669 return true; 1670 } 1671 return false; 1672 } 1673 explicit LocalVarRefChecker(Sema &SemaRef) : SemaRef(SemaRef) {} 1674 }; 1675 } // namespace 1676 1677 OMPThreadPrivateDecl * 1678 Sema::CheckOMPThreadPrivateDecl(SourceLocation Loc, ArrayRef<Expr *> VarList) { 1679 SmallVector<Expr *, 8> Vars; 1680 for (auto &RefExpr : VarList) { 1681 DeclRefExpr *DE = cast<DeclRefExpr>(RefExpr); 1682 VarDecl *VD = cast<VarDecl>(DE->getDecl()); 1683 SourceLocation ILoc = DE->getExprLoc(); 1684 1685 // Mark variable as used. 1686 VD->setReferenced(); 1687 VD->markUsed(Context); 1688 1689 QualType QType = VD->getType(); 1690 if (QType->isDependentType() || QType->isInstantiationDependentType()) { 1691 // It will be analyzed later. 1692 Vars.push_back(DE); 1693 continue; 1694 } 1695 1696 // OpenMP [2.9.2, Restrictions, C/C++, p.10] 1697 // A threadprivate variable must not have an incomplete type. 1698 if (RequireCompleteType(ILoc, VD->getType(), 1699 diag::err_omp_threadprivate_incomplete_type)) { 1700 continue; 1701 } 1702 1703 // OpenMP [2.9.2, Restrictions, C/C++, p.10] 1704 // A threadprivate variable must not have a reference type. 1705 if (VD->getType()->isReferenceType()) { 1706 Diag(ILoc, diag::err_omp_ref_type_arg) 1707 << getOpenMPDirectiveName(OMPD_threadprivate) << VD->getType(); 1708 bool IsDecl = 1709 VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly; 1710 Diag(VD->getLocation(), 1711 IsDecl ? diag::note_previous_decl : diag::note_defined_here) 1712 << VD; 1713 continue; 1714 } 1715 1716 // Check if this is a TLS variable. If TLS is not being supported, produce 1717 // the corresponding diagnostic. 1718 if ((VD->getTLSKind() != VarDecl::TLS_None && 1719 !(VD->hasAttr<OMPThreadPrivateDeclAttr>() && 1720 getLangOpts().OpenMPUseTLS && 1721 getASTContext().getTargetInfo().isTLSSupported())) || 1722 (VD->getStorageClass() == SC_Register && VD->hasAttr<AsmLabelAttr>() && 1723 !VD->isLocalVarDecl())) { 1724 Diag(ILoc, diag::err_omp_var_thread_local) 1725 << VD << ((VD->getTLSKind() != VarDecl::TLS_None) ? 0 : 1); 1726 bool IsDecl = 1727 VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly; 1728 Diag(VD->getLocation(), 1729 IsDecl ? diag::note_previous_decl : diag::note_defined_here) 1730 << VD; 1731 continue; 1732 } 1733 1734 // Check if initial value of threadprivate variable reference variable with 1735 // local storage (it is not supported by runtime). 1736 if (auto Init = VD->getAnyInitializer()) { 1737 LocalVarRefChecker Checker(*this); 1738 if (Checker.Visit(Init)) 1739 continue; 1740 } 1741 1742 Vars.push_back(RefExpr); 1743 DSAStack->addDSA(VD, DE, OMPC_threadprivate); 1744 VD->addAttr(OMPThreadPrivateDeclAttr::CreateImplicit( 1745 Context, SourceRange(Loc, Loc))); 1746 if (auto *ML = Context.getASTMutationListener()) 1747 ML->DeclarationMarkedOpenMPThreadPrivate(VD); 1748 } 1749 OMPThreadPrivateDecl *D = nullptr; 1750 if (!Vars.empty()) { 1751 D = OMPThreadPrivateDecl::Create(Context, getCurLexicalContext(), Loc, 1752 Vars); 1753 D->setAccess(AS_public); 1754 } 1755 return D; 1756 } 1757 1758 static void ReportOriginalDSA(Sema &SemaRef, DSAStackTy *Stack, 1759 const ValueDecl *D, DSAStackTy::DSAVarData DVar, 1760 bool IsLoopIterVar = false) { 1761 if (DVar.RefExpr) { 1762 SemaRef.Diag(DVar.RefExpr->getExprLoc(), diag::note_omp_explicit_dsa) 1763 << getOpenMPClauseName(DVar.CKind); 1764 return; 1765 } 1766 enum { 1767 PDSA_StaticMemberShared, 1768 PDSA_StaticLocalVarShared, 1769 PDSA_LoopIterVarPrivate, 1770 PDSA_LoopIterVarLinear, 1771 PDSA_LoopIterVarLastprivate, 1772 PDSA_ConstVarShared, 1773 PDSA_GlobalVarShared, 1774 PDSA_TaskVarFirstprivate, 1775 PDSA_LocalVarPrivate, 1776 PDSA_Implicit 1777 } Reason = PDSA_Implicit; 1778 bool ReportHint = false; 1779 auto ReportLoc = D->getLocation(); 1780 auto *VD = dyn_cast<VarDecl>(D); 1781 if (IsLoopIterVar) { 1782 if (DVar.CKind == OMPC_private) 1783 Reason = PDSA_LoopIterVarPrivate; 1784 else if (DVar.CKind == OMPC_lastprivate) 1785 Reason = PDSA_LoopIterVarLastprivate; 1786 else 1787 Reason = PDSA_LoopIterVarLinear; 1788 } else if (isOpenMPTaskingDirective(DVar.DKind) && 1789 DVar.CKind == OMPC_firstprivate) { 1790 Reason = PDSA_TaskVarFirstprivate; 1791 ReportLoc = DVar.ImplicitDSALoc; 1792 } else if (VD && VD->isStaticLocal()) 1793 Reason = PDSA_StaticLocalVarShared; 1794 else if (VD && VD->isStaticDataMember()) 1795 Reason = PDSA_StaticMemberShared; 1796 else if (VD && VD->isFileVarDecl()) 1797 Reason = PDSA_GlobalVarShared; 1798 else if (D->getType().isConstant(SemaRef.getASTContext())) 1799 Reason = PDSA_ConstVarShared; 1800 else if (VD && VD->isLocalVarDecl() && DVar.CKind == OMPC_private) { 1801 ReportHint = true; 1802 Reason = PDSA_LocalVarPrivate; 1803 } 1804 if (Reason != PDSA_Implicit) { 1805 SemaRef.Diag(ReportLoc, diag::note_omp_predetermined_dsa) 1806 << Reason << ReportHint 1807 << getOpenMPDirectiveName(Stack->getCurrentDirective()); 1808 } else if (DVar.ImplicitDSALoc.isValid()) { 1809 SemaRef.Diag(DVar.ImplicitDSALoc, diag::note_omp_implicit_dsa) 1810 << getOpenMPClauseName(DVar.CKind); 1811 } 1812 } 1813 1814 namespace { 1815 class DSAAttrChecker : public StmtVisitor<DSAAttrChecker, void> { 1816 DSAStackTy *Stack; 1817 Sema &SemaRef; 1818 bool ErrorFound; 1819 CapturedStmt *CS; 1820 llvm::SmallVector<Expr *, 8> ImplicitFirstprivate; 1821 llvm::SmallVector<Expr *, 8> ImplicitMap; 1822 llvm::DenseMap<ValueDecl *, Expr *> VarsWithInheritedDSA; 1823 llvm::DenseSet<ValueDecl *> ImplicitDeclarations; 1824 1825 public: 1826 void VisitDeclRefExpr(DeclRefExpr *E) { 1827 if (E->isTypeDependent() || E->isValueDependent() || 1828 E->containsUnexpandedParameterPack() || E->isInstantiationDependent()) 1829 return; 1830 if (auto *VD = dyn_cast<VarDecl>(E->getDecl())) { 1831 VD = VD->getCanonicalDecl(); 1832 // Skip internally declared variables. 1833 if (VD->hasLocalStorage() && !CS->capturesVariable(VD)) 1834 return; 1835 1836 auto DVar = Stack->getTopDSA(VD, false); 1837 // Check if the variable has explicit DSA set and stop analysis if it so. 1838 if (DVar.RefExpr || !ImplicitDeclarations.insert(VD).second) 1839 return; 1840 1841 // Skip internally declared static variables. 1842 if (VD->hasGlobalStorage() && !CS->capturesVariable(VD)) 1843 return; 1844 1845 auto ELoc = E->getExprLoc(); 1846 auto DKind = Stack->getCurrentDirective(); 1847 // The default(none) clause requires that each variable that is referenced 1848 // in the construct, and does not have a predetermined data-sharing 1849 // attribute, must have its data-sharing attribute explicitly determined 1850 // by being listed in a data-sharing attribute clause. 1851 if (DVar.CKind == OMPC_unknown && Stack->getDefaultDSA() == DSA_none && 1852 isParallelOrTaskRegion(DKind) && 1853 VarsWithInheritedDSA.count(VD) == 0) { 1854 VarsWithInheritedDSA[VD] = E; 1855 return; 1856 } 1857 1858 if (isOpenMPTargetExecutionDirective(DKind) && 1859 !Stack->isLoopControlVariable(VD).first) { 1860 if (!Stack->checkMappableExprComponentListsForDecl( 1861 VD, /*CurrentRegionOnly=*/true, 1862 [](OMPClauseMappableExprCommon::MappableExprComponentListRef 1863 StackComponents, 1864 OpenMPClauseKind) { 1865 // Variable is used if it has been marked as an array, array 1866 // section or the variable iself. 1867 return StackComponents.size() == 1 || 1868 std::all_of( 1869 std::next(StackComponents.rbegin()), 1870 StackComponents.rend(), 1871 [](const OMPClauseMappableExprCommon:: 1872 MappableComponent &MC) { 1873 return MC.getAssociatedDeclaration() == 1874 nullptr && 1875 (isa<OMPArraySectionExpr>( 1876 MC.getAssociatedExpression()) || 1877 isa<ArraySubscriptExpr>( 1878 MC.getAssociatedExpression())); 1879 }); 1880 })) { 1881 bool IsFirstprivate = false; 1882 // By default lambdas are captured as firstprivates. 1883 if (const auto *RD = 1884 VD->getType().getNonReferenceType()->getAsCXXRecordDecl()) 1885 IsFirstprivate = RD->isLambda(); 1886 IsFirstprivate = 1887 IsFirstprivate || 1888 (VD->getType().getNonReferenceType()->isScalarType() && 1889 Stack->getDefaultDMA() != DMA_tofrom_scalar); 1890 if (IsFirstprivate) 1891 ImplicitFirstprivate.emplace_back(E); 1892 else 1893 ImplicitMap.emplace_back(E); 1894 return; 1895 } 1896 } 1897 1898 // OpenMP [2.9.3.6, Restrictions, p.2] 1899 // A list item that appears in a reduction clause of the innermost 1900 // enclosing worksharing or parallel construct may not be accessed in an 1901 // explicit task. 1902 DVar = Stack->hasInnermostDSA( 1903 VD, [](OpenMPClauseKind C) -> bool { return C == OMPC_reduction; }, 1904 [](OpenMPDirectiveKind K) -> bool { 1905 return isOpenMPParallelDirective(K) || 1906 isOpenMPWorksharingDirective(K) || isOpenMPTeamsDirective(K); 1907 }, 1908 /*FromParent=*/true); 1909 if (isOpenMPTaskingDirective(DKind) && DVar.CKind == OMPC_reduction) { 1910 ErrorFound = true; 1911 SemaRef.Diag(ELoc, diag::err_omp_reduction_in_task); 1912 ReportOriginalDSA(SemaRef, Stack, VD, DVar); 1913 return; 1914 } 1915 1916 // Define implicit data-sharing attributes for task. 1917 DVar = Stack->getImplicitDSA(VD, false); 1918 if (isOpenMPTaskingDirective(DKind) && DVar.CKind != OMPC_shared && 1919 !Stack->isLoopControlVariable(VD).first) 1920 ImplicitFirstprivate.push_back(E); 1921 } 1922 } 1923 void VisitMemberExpr(MemberExpr *E) { 1924 if (E->isTypeDependent() || E->isValueDependent() || 1925 E->containsUnexpandedParameterPack() || E->isInstantiationDependent()) 1926 return; 1927 auto *FD = dyn_cast<FieldDecl>(E->getMemberDecl()); 1928 if (!FD) 1929 return; 1930 OpenMPDirectiveKind DKind = Stack->getCurrentDirective(); 1931 if (isa<CXXThisExpr>(E->getBase()->IgnoreParens())) { 1932 auto DVar = Stack->getTopDSA(FD, false); 1933 // Check if the variable has explicit DSA set and stop analysis if it 1934 // so. 1935 if (DVar.RefExpr || !ImplicitDeclarations.insert(FD).second) 1936 return; 1937 1938 if (isOpenMPTargetExecutionDirective(DKind) && 1939 !Stack->isLoopControlVariable(FD).first && 1940 !Stack->checkMappableExprComponentListsForDecl( 1941 FD, /*CurrentRegionOnly=*/true, 1942 [](OMPClauseMappableExprCommon::MappableExprComponentListRef 1943 StackComponents, 1944 OpenMPClauseKind) { 1945 return isa<CXXThisExpr>( 1946 cast<MemberExpr>( 1947 StackComponents.back().getAssociatedExpression()) 1948 ->getBase() 1949 ->IgnoreParens()); 1950 })) { 1951 // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, C/C++, p.3] 1952 // A bit-field cannot appear in a map clause. 1953 // 1954 if (FD->isBitField()) { 1955 SemaRef.Diag(E->getMemberLoc(), 1956 diag::err_omp_bit_fields_forbidden_in_clause) 1957 << E->getSourceRange() << getOpenMPClauseName(OMPC_map); 1958 return; 1959 } 1960 ImplicitMap.emplace_back(E); 1961 return; 1962 } 1963 1964 auto ELoc = E->getExprLoc(); 1965 // OpenMP [2.9.3.6, Restrictions, p.2] 1966 // A list item that appears in a reduction clause of the innermost 1967 // enclosing worksharing or parallel construct may not be accessed in 1968 // an explicit task. 1969 DVar = Stack->hasInnermostDSA( 1970 FD, [](OpenMPClauseKind C) -> bool { return C == OMPC_reduction; }, 1971 [](OpenMPDirectiveKind K) -> bool { 1972 return isOpenMPParallelDirective(K) || 1973 isOpenMPWorksharingDirective(K) || isOpenMPTeamsDirective(K); 1974 }, 1975 /*FromParent=*/true); 1976 if (isOpenMPTaskingDirective(DKind) && DVar.CKind == OMPC_reduction) { 1977 ErrorFound = true; 1978 SemaRef.Diag(ELoc, diag::err_omp_reduction_in_task); 1979 ReportOriginalDSA(SemaRef, Stack, FD, DVar); 1980 return; 1981 } 1982 1983 // Define implicit data-sharing attributes for task. 1984 DVar = Stack->getImplicitDSA(FD, false); 1985 if (isOpenMPTaskingDirective(DKind) && DVar.CKind != OMPC_shared && 1986 !Stack->isLoopControlVariable(FD).first) 1987 ImplicitFirstprivate.push_back(E); 1988 return; 1989 } 1990 if (isOpenMPTargetExecutionDirective(DKind) && !FD->isBitField()) { 1991 OMPClauseMappableExprCommon::MappableExprComponentList CurComponents; 1992 CheckMapClauseExpressionBase(SemaRef, E, CurComponents, OMPC_map); 1993 auto *VD = cast<ValueDecl>( 1994 CurComponents.back().getAssociatedDeclaration()->getCanonicalDecl()); 1995 if (!Stack->checkMappableExprComponentListsForDecl( 1996 VD, /*CurrentRegionOnly=*/true, 1997 [&CurComponents]( 1998 OMPClauseMappableExprCommon::MappableExprComponentListRef 1999 StackComponents, 2000 OpenMPClauseKind) { 2001 auto CCI = CurComponents.rbegin(); 2002 auto CCE = CurComponents.rend(); 2003 for (const auto &SC : llvm::reverse(StackComponents)) { 2004 // Do both expressions have the same kind? 2005 if (CCI->getAssociatedExpression()->getStmtClass() != 2006 SC.getAssociatedExpression()->getStmtClass()) 2007 if (!(isa<OMPArraySectionExpr>( 2008 SC.getAssociatedExpression()) && 2009 isa<ArraySubscriptExpr>( 2010 CCI->getAssociatedExpression()))) 2011 return false; 2012 2013 Decl *CCD = CCI->getAssociatedDeclaration(); 2014 Decl *SCD = SC.getAssociatedDeclaration(); 2015 CCD = CCD ? CCD->getCanonicalDecl() : nullptr; 2016 SCD = SCD ? SCD->getCanonicalDecl() : nullptr; 2017 if (SCD != CCD) 2018 return false; 2019 std::advance(CCI, 1); 2020 if (CCI == CCE) 2021 break; 2022 } 2023 return true; 2024 })) { 2025 Visit(E->getBase()); 2026 } 2027 } else 2028 Visit(E->getBase()); 2029 } 2030 void VisitOMPExecutableDirective(OMPExecutableDirective *S) { 2031 for (auto *C : S->clauses()) { 2032 // Skip analysis of arguments of implicitly defined firstprivate clause 2033 // for task|target directives. 2034 // Skip analysis of arguments of implicitly defined map clause for target 2035 // directives. 2036 if (C && !((isa<OMPFirstprivateClause>(C) || isa<OMPMapClause>(C)) && 2037 C->isImplicit())) { 2038 for (auto *CC : C->children()) { 2039 if (CC) 2040 Visit(CC); 2041 } 2042 } 2043 } 2044 } 2045 void VisitStmt(Stmt *S) { 2046 for (auto *C : S->children()) { 2047 if (C && !isa<OMPExecutableDirective>(C)) 2048 Visit(C); 2049 } 2050 } 2051 2052 bool isErrorFound() { return ErrorFound; } 2053 ArrayRef<Expr *> getImplicitFirstprivate() const { 2054 return ImplicitFirstprivate; 2055 } 2056 ArrayRef<Expr *> getImplicitMap() const { return ImplicitMap; } 2057 llvm::DenseMap<ValueDecl *, Expr *> &getVarsWithInheritedDSA() { 2058 return VarsWithInheritedDSA; 2059 } 2060 2061 DSAAttrChecker(DSAStackTy *S, Sema &SemaRef, CapturedStmt *CS) 2062 : Stack(S), SemaRef(SemaRef), ErrorFound(false), CS(CS) {} 2063 }; 2064 } // namespace 2065 2066 void Sema::ActOnOpenMPRegionStart(OpenMPDirectiveKind DKind, Scope *CurScope) { 2067 switch (DKind) { 2068 case OMPD_parallel: 2069 case OMPD_parallel_for: 2070 case OMPD_parallel_for_simd: 2071 case OMPD_parallel_sections: 2072 case OMPD_teams: 2073 case OMPD_teams_distribute: { 2074 QualType KmpInt32Ty = Context.getIntTypeForBitwidth(32, 1); 2075 QualType KmpInt32PtrTy = 2076 Context.getPointerType(KmpInt32Ty).withConst().withRestrict(); 2077 Sema::CapturedParamNameType Params[] = { 2078 std::make_pair(".global_tid.", KmpInt32PtrTy), 2079 std::make_pair(".bound_tid.", KmpInt32PtrTy), 2080 std::make_pair(StringRef(), QualType()) // __context with shared vars 2081 }; 2082 ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP, 2083 Params); 2084 break; 2085 } 2086 case OMPD_target_teams: 2087 case OMPD_target_parallel: { 2088 Sema::CapturedParamNameType ParamsTarget[] = { 2089 std::make_pair(StringRef(), QualType()) // __context with shared vars 2090 }; 2091 // Start a captured region for 'target' with no implicit parameters. 2092 ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP, 2093 ParamsTarget); 2094 QualType KmpInt32Ty = Context.getIntTypeForBitwidth(32, 1); 2095 QualType KmpInt32PtrTy = 2096 Context.getPointerType(KmpInt32Ty).withConst().withRestrict(); 2097 Sema::CapturedParamNameType ParamsTeamsOrParallel[] = { 2098 std::make_pair(".global_tid.", KmpInt32PtrTy), 2099 std::make_pair(".bound_tid.", KmpInt32PtrTy), 2100 std::make_pair(StringRef(), QualType()) // __context with shared vars 2101 }; 2102 // Start a captured region for 'teams' or 'parallel'. Both regions have 2103 // the same implicit parameters. 2104 ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP, 2105 ParamsTeamsOrParallel); 2106 break; 2107 } 2108 case OMPD_simd: 2109 case OMPD_for: 2110 case OMPD_for_simd: 2111 case OMPD_sections: 2112 case OMPD_section: 2113 case OMPD_single: 2114 case OMPD_master: 2115 case OMPD_critical: 2116 case OMPD_taskgroup: 2117 case OMPD_distribute: 2118 case OMPD_ordered: 2119 case OMPD_atomic: 2120 case OMPD_target_data: 2121 case OMPD_target: 2122 case OMPD_target_parallel_for: 2123 case OMPD_target_parallel_for_simd: 2124 case OMPD_target_simd: { 2125 Sema::CapturedParamNameType Params[] = { 2126 std::make_pair(StringRef(), QualType()) // __context with shared vars 2127 }; 2128 ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP, 2129 Params); 2130 break; 2131 } 2132 case OMPD_task: { 2133 QualType KmpInt32Ty = Context.getIntTypeForBitwidth(32, 1); 2134 QualType Args[] = {Context.VoidPtrTy.withConst().withRestrict()}; 2135 FunctionProtoType::ExtProtoInfo EPI; 2136 EPI.Variadic = true; 2137 QualType CopyFnType = Context.getFunctionType(Context.VoidTy, Args, EPI); 2138 Sema::CapturedParamNameType Params[] = { 2139 std::make_pair(".global_tid.", KmpInt32Ty), 2140 std::make_pair(".part_id.", Context.getPointerType(KmpInt32Ty)), 2141 std::make_pair(".privates.", Context.VoidPtrTy.withConst()), 2142 std::make_pair(".copy_fn.", 2143 Context.getPointerType(CopyFnType).withConst()), 2144 std::make_pair(".task_t.", Context.VoidPtrTy.withConst()), 2145 std::make_pair(StringRef(), QualType()) // __context with shared vars 2146 }; 2147 ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP, 2148 Params); 2149 // Mark this captured region as inlined, because we don't use outlined 2150 // function directly. 2151 getCurCapturedRegion()->TheCapturedDecl->addAttr( 2152 AlwaysInlineAttr::CreateImplicit( 2153 Context, AlwaysInlineAttr::Keyword_forceinline, SourceRange())); 2154 break; 2155 } 2156 case OMPD_taskloop: 2157 case OMPD_taskloop_simd: { 2158 QualType KmpInt32Ty = 2159 Context.getIntTypeForBitwidth(/*DestWidth=*/32, /*Signed=*/1); 2160 QualType KmpUInt64Ty = 2161 Context.getIntTypeForBitwidth(/*DestWidth=*/64, /*Signed=*/0); 2162 QualType KmpInt64Ty = 2163 Context.getIntTypeForBitwidth(/*DestWidth=*/64, /*Signed=*/1); 2164 QualType Args[] = {Context.VoidPtrTy.withConst().withRestrict()}; 2165 FunctionProtoType::ExtProtoInfo EPI; 2166 EPI.Variadic = true; 2167 QualType CopyFnType = Context.getFunctionType(Context.VoidTy, Args, EPI); 2168 Sema::CapturedParamNameType Params[] = { 2169 std::make_pair(".global_tid.", KmpInt32Ty), 2170 std::make_pair(".part_id.", Context.getPointerType(KmpInt32Ty)), 2171 std::make_pair(".privates.", 2172 Context.VoidPtrTy.withConst().withRestrict()), 2173 std::make_pair( 2174 ".copy_fn.", 2175 Context.getPointerType(CopyFnType).withConst().withRestrict()), 2176 std::make_pair(".task_t.", Context.VoidPtrTy.withConst()), 2177 std::make_pair(".lb.", KmpUInt64Ty), 2178 std::make_pair(".ub.", KmpUInt64Ty), std::make_pair(".st.", KmpInt64Ty), 2179 std::make_pair(".liter.", KmpInt32Ty), 2180 std::make_pair(".reductions.", 2181 Context.VoidPtrTy.withConst().withRestrict()), 2182 std::make_pair(StringRef(), QualType()) // __context with shared vars 2183 }; 2184 ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP, 2185 Params); 2186 // Mark this captured region as inlined, because we don't use outlined 2187 // function directly. 2188 getCurCapturedRegion()->TheCapturedDecl->addAttr( 2189 AlwaysInlineAttr::CreateImplicit( 2190 Context, AlwaysInlineAttr::Keyword_forceinline, SourceRange())); 2191 break; 2192 } 2193 case OMPD_distribute_parallel_for_simd: 2194 case OMPD_distribute_simd: 2195 case OMPD_distribute_parallel_for: 2196 case OMPD_teams_distribute_simd: 2197 case OMPD_teams_distribute_parallel_for_simd: 2198 case OMPD_teams_distribute_parallel_for: 2199 case OMPD_target_teams_distribute: 2200 case OMPD_target_teams_distribute_parallel_for: 2201 case OMPD_target_teams_distribute_parallel_for_simd: 2202 case OMPD_target_teams_distribute_simd: { 2203 QualType KmpInt32Ty = Context.getIntTypeForBitwidth(32, 1); 2204 QualType KmpInt32PtrTy = 2205 Context.getPointerType(KmpInt32Ty).withConst().withRestrict(); 2206 Sema::CapturedParamNameType Params[] = { 2207 std::make_pair(".global_tid.", KmpInt32PtrTy), 2208 std::make_pair(".bound_tid.", KmpInt32PtrTy), 2209 std::make_pair(".previous.lb.", Context.getSizeType()), 2210 std::make_pair(".previous.ub.", Context.getSizeType()), 2211 std::make_pair(StringRef(), QualType()) // __context with shared vars 2212 }; 2213 ActOnCapturedRegionStart(DSAStack->getConstructLoc(), CurScope, CR_OpenMP, 2214 Params); 2215 break; 2216 } 2217 case OMPD_threadprivate: 2218 case OMPD_taskyield: 2219 case OMPD_barrier: 2220 case OMPD_taskwait: 2221 case OMPD_cancellation_point: 2222 case OMPD_cancel: 2223 case OMPD_flush: 2224 case OMPD_target_enter_data: 2225 case OMPD_target_exit_data: 2226 case OMPD_declare_reduction: 2227 case OMPD_declare_simd: 2228 case OMPD_declare_target: 2229 case OMPD_end_declare_target: 2230 case OMPD_target_update: 2231 llvm_unreachable("OpenMP Directive is not allowed"); 2232 case OMPD_unknown: 2233 llvm_unreachable("Unknown OpenMP directive"); 2234 } 2235 } 2236 2237 int Sema::getOpenMPCaptureLevels(OpenMPDirectiveKind DKind) { 2238 SmallVector<OpenMPDirectiveKind, 4> CaptureRegions; 2239 getOpenMPCaptureRegions(CaptureRegions, DKind); 2240 return CaptureRegions.size(); 2241 } 2242 2243 static OMPCapturedExprDecl *buildCaptureDecl(Sema &S, IdentifierInfo *Id, 2244 Expr *CaptureExpr, bool WithInit, 2245 bool AsExpression) { 2246 assert(CaptureExpr); 2247 ASTContext &C = S.getASTContext(); 2248 Expr *Init = AsExpression ? CaptureExpr : CaptureExpr->IgnoreImpCasts(); 2249 QualType Ty = Init->getType(); 2250 if (CaptureExpr->getObjectKind() == OK_Ordinary && CaptureExpr->isGLValue()) { 2251 if (S.getLangOpts().CPlusPlus) 2252 Ty = C.getLValueReferenceType(Ty); 2253 else { 2254 Ty = C.getPointerType(Ty); 2255 ExprResult Res = 2256 S.CreateBuiltinUnaryOp(CaptureExpr->getExprLoc(), UO_AddrOf, Init); 2257 if (!Res.isUsable()) 2258 return nullptr; 2259 Init = Res.get(); 2260 } 2261 WithInit = true; 2262 } 2263 auto *CED = OMPCapturedExprDecl::Create(C, S.CurContext, Id, Ty, 2264 CaptureExpr->getLocStart()); 2265 if (!WithInit) 2266 CED->addAttr(OMPCaptureNoInitAttr::CreateImplicit(C, SourceRange())); 2267 S.CurContext->addHiddenDecl(CED); 2268 S.AddInitializerToDecl(CED, Init, /*DirectInit=*/false); 2269 return CED; 2270 } 2271 2272 static DeclRefExpr *buildCapture(Sema &S, ValueDecl *D, Expr *CaptureExpr, 2273 bool WithInit) { 2274 OMPCapturedExprDecl *CD; 2275 if (auto *VD = S.IsOpenMPCapturedDecl(D)) 2276 CD = cast<OMPCapturedExprDecl>(VD); 2277 else 2278 CD = buildCaptureDecl(S, D->getIdentifier(), CaptureExpr, WithInit, 2279 /*AsExpression=*/false); 2280 return buildDeclRefExpr(S, CD, CD->getType().getNonReferenceType(), 2281 CaptureExpr->getExprLoc()); 2282 } 2283 2284 static ExprResult buildCapture(Sema &S, Expr *CaptureExpr, DeclRefExpr *&Ref) { 2285 if (!Ref) { 2286 auto *CD = 2287 buildCaptureDecl(S, &S.getASTContext().Idents.get(".capture_expr."), 2288 CaptureExpr, /*WithInit=*/true, /*AsExpression=*/true); 2289 Ref = buildDeclRefExpr(S, CD, CD->getType().getNonReferenceType(), 2290 CaptureExpr->getExprLoc()); 2291 } 2292 ExprResult Res = Ref; 2293 if (!S.getLangOpts().CPlusPlus && 2294 CaptureExpr->getObjectKind() == OK_Ordinary && CaptureExpr->isGLValue() && 2295 Ref->getType()->isPointerType()) 2296 Res = S.CreateBuiltinUnaryOp(CaptureExpr->getExprLoc(), UO_Deref, Ref); 2297 if (!Res.isUsable()) 2298 return ExprError(); 2299 return CaptureExpr->isGLValue() ? Res : S.DefaultLvalueConversion(Res.get()); 2300 } 2301 2302 namespace { 2303 // OpenMP directives parsed in this section are represented as a 2304 // CapturedStatement with an associated statement. If a syntax error 2305 // is detected during the parsing of the associated statement, the 2306 // compiler must abort processing and close the CapturedStatement. 2307 // 2308 // Combined directives such as 'target parallel' have more than one 2309 // nested CapturedStatements. This RAII ensures that we unwind out 2310 // of all the nested CapturedStatements when an error is found. 2311 class CaptureRegionUnwinderRAII { 2312 private: 2313 Sema &S; 2314 bool &ErrorFound; 2315 OpenMPDirectiveKind DKind; 2316 2317 public: 2318 CaptureRegionUnwinderRAII(Sema &S, bool &ErrorFound, 2319 OpenMPDirectiveKind DKind) 2320 : S(S), ErrorFound(ErrorFound), DKind(DKind) {} 2321 ~CaptureRegionUnwinderRAII() { 2322 if (ErrorFound) { 2323 int ThisCaptureLevel = S.getOpenMPCaptureLevels(DKind); 2324 while (--ThisCaptureLevel >= 0) 2325 S.ActOnCapturedRegionError(); 2326 } 2327 } 2328 }; 2329 } // namespace 2330 2331 StmtResult Sema::ActOnOpenMPRegionEnd(StmtResult S, 2332 ArrayRef<OMPClause *> Clauses) { 2333 bool ErrorFound = false; 2334 CaptureRegionUnwinderRAII CaptureRegionUnwinder( 2335 *this, ErrorFound, DSAStack->getCurrentDirective()); 2336 if (!S.isUsable()) { 2337 ErrorFound = true; 2338 return StmtError(); 2339 } 2340 2341 OMPOrderedClause *OC = nullptr; 2342 OMPScheduleClause *SC = nullptr; 2343 SmallVector<OMPLinearClause *, 4> LCs; 2344 SmallVector<OMPClauseWithPreInit *, 8> PICs; 2345 // This is required for proper codegen. 2346 for (auto *Clause : Clauses) { 2347 if (isOpenMPTaskingDirective(DSAStack->getCurrentDirective()) && 2348 Clause->getClauseKind() == OMPC_in_reduction) { 2349 // Capture taskgroup task_reduction descriptors inside the tasking regions 2350 // with the corresponding in_reduction items. 2351 auto *IRC = cast<OMPInReductionClause>(Clause); 2352 for (auto *E : IRC->taskgroup_descriptors()) 2353 if (E) 2354 MarkDeclarationsReferencedInExpr(E); 2355 } 2356 if (isOpenMPPrivate(Clause->getClauseKind()) || 2357 Clause->getClauseKind() == OMPC_copyprivate || 2358 (getLangOpts().OpenMPUseTLS && 2359 getASTContext().getTargetInfo().isTLSSupported() && 2360 Clause->getClauseKind() == OMPC_copyin)) { 2361 DSAStack->setForceVarCapturing(Clause->getClauseKind() == OMPC_copyin); 2362 // Mark all variables in private list clauses as used in inner region. 2363 for (auto *VarRef : Clause->children()) { 2364 if (auto *E = cast_or_null<Expr>(VarRef)) { 2365 MarkDeclarationsReferencedInExpr(E); 2366 } 2367 } 2368 DSAStack->setForceVarCapturing(/*V=*/false); 2369 } else if (isParallelOrTaskRegion(DSAStack->getCurrentDirective())) { 2370 if (auto *C = OMPClauseWithPreInit::get(Clause)) 2371 PICs.push_back(C); 2372 if (auto *C = OMPClauseWithPostUpdate::get(Clause)) { 2373 if (auto *E = C->getPostUpdateExpr()) 2374 MarkDeclarationsReferencedInExpr(E); 2375 } 2376 } 2377 if (Clause->getClauseKind() == OMPC_schedule) 2378 SC = cast<OMPScheduleClause>(Clause); 2379 else if (Clause->getClauseKind() == OMPC_ordered) 2380 OC = cast<OMPOrderedClause>(Clause); 2381 else if (Clause->getClauseKind() == OMPC_linear) 2382 LCs.push_back(cast<OMPLinearClause>(Clause)); 2383 } 2384 // OpenMP, 2.7.1 Loop Construct, Restrictions 2385 // The nonmonotonic modifier cannot be specified if an ordered clause is 2386 // specified. 2387 if (SC && 2388 (SC->getFirstScheduleModifier() == OMPC_SCHEDULE_MODIFIER_nonmonotonic || 2389 SC->getSecondScheduleModifier() == 2390 OMPC_SCHEDULE_MODIFIER_nonmonotonic) && 2391 OC) { 2392 Diag(SC->getFirstScheduleModifier() == OMPC_SCHEDULE_MODIFIER_nonmonotonic 2393 ? SC->getFirstScheduleModifierLoc() 2394 : SC->getSecondScheduleModifierLoc(), 2395 diag::err_omp_schedule_nonmonotonic_ordered) 2396 << SourceRange(OC->getLocStart(), OC->getLocEnd()); 2397 ErrorFound = true; 2398 } 2399 if (!LCs.empty() && OC && OC->getNumForLoops()) { 2400 for (auto *C : LCs) { 2401 Diag(C->getLocStart(), diag::err_omp_linear_ordered) 2402 << SourceRange(OC->getLocStart(), OC->getLocEnd()); 2403 } 2404 ErrorFound = true; 2405 } 2406 if (isOpenMPWorksharingDirective(DSAStack->getCurrentDirective()) && 2407 isOpenMPSimdDirective(DSAStack->getCurrentDirective()) && OC && 2408 OC->getNumForLoops()) { 2409 Diag(OC->getLocStart(), diag::err_omp_ordered_simd) 2410 << getOpenMPDirectiveName(DSAStack->getCurrentDirective()); 2411 ErrorFound = true; 2412 } 2413 if (ErrorFound) { 2414 return StmtError(); 2415 } 2416 StmtResult SR = S; 2417 SmallVector<OpenMPDirectiveKind, 4> CaptureRegions; 2418 getOpenMPCaptureRegions(CaptureRegions, DSAStack->getCurrentDirective()); 2419 for (auto ThisCaptureRegion : llvm::reverse(CaptureRegions)) { 2420 // Mark all variables in private list clauses as used in inner region. 2421 // Required for proper codegen of combined directives. 2422 // TODO: add processing for other clauses. 2423 if (isParallelOrTaskRegion(DSAStack->getCurrentDirective())) { 2424 for (auto *C : PICs) { 2425 OpenMPDirectiveKind CaptureRegion = C->getCaptureRegion(); 2426 // Find the particular capture region for the clause if the 2427 // directive is a combined one with multiple capture regions. 2428 // If the directive is not a combined one, the capture region 2429 // associated with the clause is OMPD_unknown and is generated 2430 // only once. 2431 if (CaptureRegion == ThisCaptureRegion || 2432 CaptureRegion == OMPD_unknown) { 2433 if (auto *DS = cast_or_null<DeclStmt>(C->getPreInitStmt())) { 2434 for (auto *D : DS->decls()) 2435 MarkVariableReferenced(D->getLocation(), cast<VarDecl>(D)); 2436 } 2437 } 2438 } 2439 } 2440 SR = ActOnCapturedRegionEnd(SR.get()); 2441 } 2442 return SR; 2443 } 2444 2445 static bool checkCancelRegion(Sema &SemaRef, OpenMPDirectiveKind CurrentRegion, 2446 OpenMPDirectiveKind CancelRegion, 2447 SourceLocation StartLoc) { 2448 // CancelRegion is only needed for cancel and cancellation_point. 2449 if (CurrentRegion != OMPD_cancel && CurrentRegion != OMPD_cancellation_point) 2450 return false; 2451 2452 if (CancelRegion == OMPD_parallel || CancelRegion == OMPD_for || 2453 CancelRegion == OMPD_sections || CancelRegion == OMPD_taskgroup) 2454 return false; 2455 2456 SemaRef.Diag(StartLoc, diag::err_omp_wrong_cancel_region) 2457 << getOpenMPDirectiveName(CancelRegion); 2458 return true; 2459 } 2460 2461 static bool checkNestingOfRegions(Sema &SemaRef, DSAStackTy *Stack, 2462 OpenMPDirectiveKind CurrentRegion, 2463 const DeclarationNameInfo &CurrentName, 2464 OpenMPDirectiveKind CancelRegion, 2465 SourceLocation StartLoc) { 2466 if (Stack->getCurScope()) { 2467 auto ParentRegion = Stack->getParentDirective(); 2468 auto OffendingRegion = ParentRegion; 2469 bool NestingProhibited = false; 2470 bool CloseNesting = true; 2471 bool OrphanSeen = false; 2472 enum { 2473 NoRecommend, 2474 ShouldBeInParallelRegion, 2475 ShouldBeInOrderedRegion, 2476 ShouldBeInTargetRegion, 2477 ShouldBeInTeamsRegion 2478 } Recommend = NoRecommend; 2479 if (isOpenMPSimdDirective(ParentRegion) && CurrentRegion != OMPD_ordered) { 2480 // OpenMP [2.16, Nesting of Regions] 2481 // OpenMP constructs may not be nested inside a simd region. 2482 // OpenMP [2.8.1,simd Construct, Restrictions] 2483 // An ordered construct with the simd clause is the only OpenMP 2484 // construct that can appear in the simd region. 2485 // Allowing a SIMD construct nested in another SIMD construct is an 2486 // extension. The OpenMP 4.5 spec does not allow it. Issue a warning 2487 // message. 2488 SemaRef.Diag(StartLoc, (CurrentRegion != OMPD_simd) 2489 ? diag::err_omp_prohibited_region_simd 2490 : diag::warn_omp_nesting_simd); 2491 return CurrentRegion != OMPD_simd; 2492 } 2493 if (ParentRegion == OMPD_atomic) { 2494 // OpenMP [2.16, Nesting of Regions] 2495 // OpenMP constructs may not be nested inside an atomic region. 2496 SemaRef.Diag(StartLoc, diag::err_omp_prohibited_region_atomic); 2497 return true; 2498 } 2499 if (CurrentRegion == OMPD_section) { 2500 // OpenMP [2.7.2, sections Construct, Restrictions] 2501 // Orphaned section directives are prohibited. That is, the section 2502 // directives must appear within the sections construct and must not be 2503 // encountered elsewhere in the sections region. 2504 if (ParentRegion != OMPD_sections && 2505 ParentRegion != OMPD_parallel_sections) { 2506 SemaRef.Diag(StartLoc, diag::err_omp_orphaned_section_directive) 2507 << (ParentRegion != OMPD_unknown) 2508 << getOpenMPDirectiveName(ParentRegion); 2509 return true; 2510 } 2511 return false; 2512 } 2513 // Allow some constructs (except teams) to be orphaned (they could be 2514 // used in functions, called from OpenMP regions with the required 2515 // preconditions). 2516 if (ParentRegion == OMPD_unknown && 2517 !isOpenMPNestingTeamsDirective(CurrentRegion)) 2518 return false; 2519 if (CurrentRegion == OMPD_cancellation_point || 2520 CurrentRegion == OMPD_cancel) { 2521 // OpenMP [2.16, Nesting of Regions] 2522 // A cancellation point construct for which construct-type-clause is 2523 // taskgroup must be nested inside a task construct. A cancellation 2524 // point construct for which construct-type-clause is not taskgroup must 2525 // be closely nested inside an OpenMP construct that matches the type 2526 // specified in construct-type-clause. 2527 // A cancel construct for which construct-type-clause is taskgroup must be 2528 // nested inside a task construct. A cancel construct for which 2529 // construct-type-clause is not taskgroup must be closely nested inside an 2530 // OpenMP construct that matches the type specified in 2531 // construct-type-clause. 2532 NestingProhibited = 2533 !((CancelRegion == OMPD_parallel && 2534 (ParentRegion == OMPD_parallel || 2535 ParentRegion == OMPD_target_parallel)) || 2536 (CancelRegion == OMPD_for && 2537 (ParentRegion == OMPD_for || ParentRegion == OMPD_parallel_for || 2538 ParentRegion == OMPD_target_parallel_for)) || 2539 (CancelRegion == OMPD_taskgroup && ParentRegion == OMPD_task) || 2540 (CancelRegion == OMPD_sections && 2541 (ParentRegion == OMPD_section || ParentRegion == OMPD_sections || 2542 ParentRegion == OMPD_parallel_sections))); 2543 } else if (CurrentRegion == OMPD_master) { 2544 // OpenMP [2.16, Nesting of Regions] 2545 // A master region may not be closely nested inside a worksharing, 2546 // atomic, or explicit task region. 2547 NestingProhibited = isOpenMPWorksharingDirective(ParentRegion) || 2548 isOpenMPTaskingDirective(ParentRegion); 2549 } else if (CurrentRegion == OMPD_critical && CurrentName.getName()) { 2550 // OpenMP [2.16, Nesting of Regions] 2551 // A critical region may not be nested (closely or otherwise) inside a 2552 // critical region with the same name. Note that this restriction is not 2553 // sufficient to prevent deadlock. 2554 SourceLocation PreviousCriticalLoc; 2555 bool DeadLock = Stack->hasDirective( 2556 [CurrentName, &PreviousCriticalLoc](OpenMPDirectiveKind K, 2557 const DeclarationNameInfo &DNI, 2558 SourceLocation Loc) -> bool { 2559 if (K == OMPD_critical && DNI.getName() == CurrentName.getName()) { 2560 PreviousCriticalLoc = Loc; 2561 return true; 2562 } else 2563 return false; 2564 }, 2565 false /* skip top directive */); 2566 if (DeadLock) { 2567 SemaRef.Diag(StartLoc, 2568 diag::err_omp_prohibited_region_critical_same_name) 2569 << CurrentName.getName(); 2570 if (PreviousCriticalLoc.isValid()) 2571 SemaRef.Diag(PreviousCriticalLoc, 2572 diag::note_omp_previous_critical_region); 2573 return true; 2574 } 2575 } else if (CurrentRegion == OMPD_barrier) { 2576 // OpenMP [2.16, Nesting of Regions] 2577 // A barrier region may not be closely nested inside a worksharing, 2578 // explicit task, critical, ordered, atomic, or master region. 2579 NestingProhibited = isOpenMPWorksharingDirective(ParentRegion) || 2580 isOpenMPTaskingDirective(ParentRegion) || 2581 ParentRegion == OMPD_master || 2582 ParentRegion == OMPD_critical || 2583 ParentRegion == OMPD_ordered; 2584 } else if (isOpenMPWorksharingDirective(CurrentRegion) && 2585 !isOpenMPParallelDirective(CurrentRegion) && 2586 !isOpenMPTeamsDirective(CurrentRegion)) { 2587 // OpenMP [2.16, Nesting of Regions] 2588 // A worksharing region may not be closely nested inside a worksharing, 2589 // explicit task, critical, ordered, atomic, or master region. 2590 NestingProhibited = isOpenMPWorksharingDirective(ParentRegion) || 2591 isOpenMPTaskingDirective(ParentRegion) || 2592 ParentRegion == OMPD_master || 2593 ParentRegion == OMPD_critical || 2594 ParentRegion == OMPD_ordered; 2595 Recommend = ShouldBeInParallelRegion; 2596 } else if (CurrentRegion == OMPD_ordered) { 2597 // OpenMP [2.16, Nesting of Regions] 2598 // An ordered region may not be closely nested inside a critical, 2599 // atomic, or explicit task region. 2600 // An ordered region must be closely nested inside a loop region (or 2601 // parallel loop region) with an ordered clause. 2602 // OpenMP [2.8.1,simd Construct, Restrictions] 2603 // An ordered construct with the simd clause is the only OpenMP construct 2604 // that can appear in the simd region. 2605 NestingProhibited = ParentRegion == OMPD_critical || 2606 isOpenMPTaskingDirective(ParentRegion) || 2607 !(isOpenMPSimdDirective(ParentRegion) || 2608 Stack->isParentOrderedRegion()); 2609 Recommend = ShouldBeInOrderedRegion; 2610 } else if (isOpenMPNestingTeamsDirective(CurrentRegion)) { 2611 // OpenMP [2.16, Nesting of Regions] 2612 // If specified, a teams construct must be contained within a target 2613 // construct. 2614 NestingProhibited = ParentRegion != OMPD_target; 2615 OrphanSeen = ParentRegion == OMPD_unknown; 2616 Recommend = ShouldBeInTargetRegion; 2617 Stack->setParentTeamsRegionLoc(Stack->getConstructLoc()); 2618 } 2619 if (!NestingProhibited && 2620 !isOpenMPTargetExecutionDirective(CurrentRegion) && 2621 !isOpenMPTargetDataManagementDirective(CurrentRegion) && 2622 (ParentRegion == OMPD_teams || ParentRegion == OMPD_target_teams)) { 2623 // OpenMP [2.16, Nesting of Regions] 2624 // distribute, parallel, parallel sections, parallel workshare, and the 2625 // parallel loop and parallel loop SIMD constructs are the only OpenMP 2626 // constructs that can be closely nested in the teams region. 2627 NestingProhibited = !isOpenMPParallelDirective(CurrentRegion) && 2628 !isOpenMPDistributeDirective(CurrentRegion); 2629 Recommend = ShouldBeInParallelRegion; 2630 } 2631 if (!NestingProhibited && 2632 isOpenMPNestingDistributeDirective(CurrentRegion)) { 2633 // OpenMP 4.5 [2.17 Nesting of Regions] 2634 // The region associated with the distribute construct must be strictly 2635 // nested inside a teams region 2636 NestingProhibited = 2637 (ParentRegion != OMPD_teams && ParentRegion != OMPD_target_teams); 2638 Recommend = ShouldBeInTeamsRegion; 2639 } 2640 if (!NestingProhibited && 2641 (isOpenMPTargetExecutionDirective(CurrentRegion) || 2642 isOpenMPTargetDataManagementDirective(CurrentRegion))) { 2643 // OpenMP 4.5 [2.17 Nesting of Regions] 2644 // If a target, target update, target data, target enter data, or 2645 // target exit data construct is encountered during execution of a 2646 // target region, the behavior is unspecified. 2647 NestingProhibited = Stack->hasDirective( 2648 [&OffendingRegion](OpenMPDirectiveKind K, const DeclarationNameInfo &, 2649 SourceLocation) -> bool { 2650 if (isOpenMPTargetExecutionDirective(K)) { 2651 OffendingRegion = K; 2652 return true; 2653 } else 2654 return false; 2655 }, 2656 false /* don't skip top directive */); 2657 CloseNesting = false; 2658 } 2659 if (NestingProhibited) { 2660 if (OrphanSeen) { 2661 SemaRef.Diag(StartLoc, diag::err_omp_orphaned_device_directive) 2662 << getOpenMPDirectiveName(CurrentRegion) << Recommend; 2663 } else { 2664 SemaRef.Diag(StartLoc, diag::err_omp_prohibited_region) 2665 << CloseNesting << getOpenMPDirectiveName(OffendingRegion) 2666 << Recommend << getOpenMPDirectiveName(CurrentRegion); 2667 } 2668 return true; 2669 } 2670 } 2671 return false; 2672 } 2673 2674 static bool checkIfClauses(Sema &S, OpenMPDirectiveKind Kind, 2675 ArrayRef<OMPClause *> Clauses, 2676 ArrayRef<OpenMPDirectiveKind> AllowedNameModifiers) { 2677 bool ErrorFound = false; 2678 unsigned NamedModifiersNumber = 0; 2679 SmallVector<const OMPIfClause *, OMPC_unknown + 1> FoundNameModifiers( 2680 OMPD_unknown + 1); 2681 SmallVector<SourceLocation, 4> NameModifierLoc; 2682 for (const auto *C : Clauses) { 2683 if (const auto *IC = dyn_cast_or_null<OMPIfClause>(C)) { 2684 // At most one if clause without a directive-name-modifier can appear on 2685 // the directive. 2686 OpenMPDirectiveKind CurNM = IC->getNameModifier(); 2687 if (FoundNameModifiers[CurNM]) { 2688 S.Diag(C->getLocStart(), diag::err_omp_more_one_clause) 2689 << getOpenMPDirectiveName(Kind) << getOpenMPClauseName(OMPC_if) 2690 << (CurNM != OMPD_unknown) << getOpenMPDirectiveName(CurNM); 2691 ErrorFound = true; 2692 } else if (CurNM != OMPD_unknown) { 2693 NameModifierLoc.push_back(IC->getNameModifierLoc()); 2694 ++NamedModifiersNumber; 2695 } 2696 FoundNameModifiers[CurNM] = IC; 2697 if (CurNM == OMPD_unknown) 2698 continue; 2699 // Check if the specified name modifier is allowed for the current 2700 // directive. 2701 // At most one if clause with the particular directive-name-modifier can 2702 // appear on the directive. 2703 bool MatchFound = false; 2704 for (auto NM : AllowedNameModifiers) { 2705 if (CurNM == NM) { 2706 MatchFound = true; 2707 break; 2708 } 2709 } 2710 if (!MatchFound) { 2711 S.Diag(IC->getNameModifierLoc(), 2712 diag::err_omp_wrong_if_directive_name_modifier) 2713 << getOpenMPDirectiveName(CurNM) << getOpenMPDirectiveName(Kind); 2714 ErrorFound = true; 2715 } 2716 } 2717 } 2718 // If any if clause on the directive includes a directive-name-modifier then 2719 // all if clauses on the directive must include a directive-name-modifier. 2720 if (FoundNameModifiers[OMPD_unknown] && NamedModifiersNumber > 0) { 2721 if (NamedModifiersNumber == AllowedNameModifiers.size()) { 2722 S.Diag(FoundNameModifiers[OMPD_unknown]->getLocStart(), 2723 diag::err_omp_no_more_if_clause); 2724 } else { 2725 std::string Values; 2726 std::string Sep(", "); 2727 unsigned AllowedCnt = 0; 2728 unsigned TotalAllowedNum = 2729 AllowedNameModifiers.size() - NamedModifiersNumber; 2730 for (unsigned Cnt = 0, End = AllowedNameModifiers.size(); Cnt < End; 2731 ++Cnt) { 2732 OpenMPDirectiveKind NM = AllowedNameModifiers[Cnt]; 2733 if (!FoundNameModifiers[NM]) { 2734 Values += "'"; 2735 Values += getOpenMPDirectiveName(NM); 2736 Values += "'"; 2737 if (AllowedCnt + 2 == TotalAllowedNum) 2738 Values += " or "; 2739 else if (AllowedCnt + 1 != TotalAllowedNum) 2740 Values += Sep; 2741 ++AllowedCnt; 2742 } 2743 } 2744 S.Diag(FoundNameModifiers[OMPD_unknown]->getCondition()->getLocStart(), 2745 diag::err_omp_unnamed_if_clause) 2746 << (TotalAllowedNum > 1) << Values; 2747 } 2748 for (auto Loc : NameModifierLoc) { 2749 S.Diag(Loc, diag::note_omp_previous_named_if_clause); 2750 } 2751 ErrorFound = true; 2752 } 2753 return ErrorFound; 2754 } 2755 2756 StmtResult Sema::ActOnOpenMPExecutableDirective( 2757 OpenMPDirectiveKind Kind, const DeclarationNameInfo &DirName, 2758 OpenMPDirectiveKind CancelRegion, ArrayRef<OMPClause *> Clauses, 2759 Stmt *AStmt, SourceLocation StartLoc, SourceLocation EndLoc) { 2760 StmtResult Res = StmtError(); 2761 // First check CancelRegion which is then used in checkNestingOfRegions. 2762 if (checkCancelRegion(*this, Kind, CancelRegion, StartLoc) || 2763 checkNestingOfRegions(*this, DSAStack, Kind, DirName, CancelRegion, 2764 StartLoc)) 2765 return StmtError(); 2766 2767 llvm::SmallVector<OMPClause *, 8> ClausesWithImplicit; 2768 llvm::DenseMap<ValueDecl *, Expr *> VarsWithInheritedDSA; 2769 bool ErrorFound = false; 2770 ClausesWithImplicit.append(Clauses.begin(), Clauses.end()); 2771 if (AStmt && !CurContext->isDependentContext()) { 2772 assert(isa<CapturedStmt>(AStmt) && "Captured statement expected"); 2773 2774 // Check default data sharing attributes for referenced variables. 2775 DSAAttrChecker DSAChecker(DSAStack, *this, cast<CapturedStmt>(AStmt)); 2776 int ThisCaptureLevel = getOpenMPCaptureLevels(Kind); 2777 Stmt *S = AStmt; 2778 while (--ThisCaptureLevel >= 0) 2779 S = cast<CapturedStmt>(S)->getCapturedStmt(); 2780 DSAChecker.Visit(S); 2781 if (DSAChecker.isErrorFound()) 2782 return StmtError(); 2783 // Generate list of implicitly defined firstprivate variables. 2784 VarsWithInheritedDSA = DSAChecker.getVarsWithInheritedDSA(); 2785 2786 SmallVector<Expr *, 4> ImplicitFirstprivates( 2787 DSAChecker.getImplicitFirstprivate().begin(), 2788 DSAChecker.getImplicitFirstprivate().end()); 2789 SmallVector<Expr *, 4> ImplicitMaps(DSAChecker.getImplicitMap().begin(), 2790 DSAChecker.getImplicitMap().end()); 2791 // Mark taskgroup task_reduction descriptors as implicitly firstprivate. 2792 for (auto *C : Clauses) { 2793 if (auto *IRC = dyn_cast<OMPInReductionClause>(C)) { 2794 for (auto *E : IRC->taskgroup_descriptors()) 2795 if (E) 2796 ImplicitFirstprivates.emplace_back(E); 2797 } 2798 } 2799 if (!ImplicitFirstprivates.empty()) { 2800 if (OMPClause *Implicit = ActOnOpenMPFirstprivateClause( 2801 ImplicitFirstprivates, SourceLocation(), SourceLocation(), 2802 SourceLocation())) { 2803 ClausesWithImplicit.push_back(Implicit); 2804 ErrorFound = cast<OMPFirstprivateClause>(Implicit)->varlist_size() != 2805 ImplicitFirstprivates.size(); 2806 } else 2807 ErrorFound = true; 2808 } 2809 if (!ImplicitMaps.empty()) { 2810 if (OMPClause *Implicit = ActOnOpenMPMapClause( 2811 OMPC_MAP_unknown, OMPC_MAP_tofrom, /*IsMapTypeImplicit=*/true, 2812 SourceLocation(), SourceLocation(), ImplicitMaps, 2813 SourceLocation(), SourceLocation(), SourceLocation())) { 2814 ClausesWithImplicit.emplace_back(Implicit); 2815 ErrorFound |= 2816 cast<OMPMapClause>(Implicit)->varlist_size() != ImplicitMaps.size(); 2817 } else 2818 ErrorFound = true; 2819 } 2820 } 2821 2822 llvm::SmallVector<OpenMPDirectiveKind, 4> AllowedNameModifiers; 2823 switch (Kind) { 2824 case OMPD_parallel: 2825 Res = ActOnOpenMPParallelDirective(ClausesWithImplicit, AStmt, StartLoc, 2826 EndLoc); 2827 AllowedNameModifiers.push_back(OMPD_parallel); 2828 break; 2829 case OMPD_simd: 2830 Res = ActOnOpenMPSimdDirective(ClausesWithImplicit, AStmt, StartLoc, EndLoc, 2831 VarsWithInheritedDSA); 2832 break; 2833 case OMPD_for: 2834 Res = ActOnOpenMPForDirective(ClausesWithImplicit, AStmt, StartLoc, EndLoc, 2835 VarsWithInheritedDSA); 2836 break; 2837 case OMPD_for_simd: 2838 Res = ActOnOpenMPForSimdDirective(ClausesWithImplicit, AStmt, StartLoc, 2839 EndLoc, VarsWithInheritedDSA); 2840 break; 2841 case OMPD_sections: 2842 Res = ActOnOpenMPSectionsDirective(ClausesWithImplicit, AStmt, StartLoc, 2843 EndLoc); 2844 break; 2845 case OMPD_section: 2846 assert(ClausesWithImplicit.empty() && 2847 "No clauses are allowed for 'omp section' directive"); 2848 Res = ActOnOpenMPSectionDirective(AStmt, StartLoc, EndLoc); 2849 break; 2850 case OMPD_single: 2851 Res = ActOnOpenMPSingleDirective(ClausesWithImplicit, AStmt, StartLoc, 2852 EndLoc); 2853 break; 2854 case OMPD_master: 2855 assert(ClausesWithImplicit.empty() && 2856 "No clauses are allowed for 'omp master' directive"); 2857 Res = ActOnOpenMPMasterDirective(AStmt, StartLoc, EndLoc); 2858 break; 2859 case OMPD_critical: 2860 Res = ActOnOpenMPCriticalDirective(DirName, ClausesWithImplicit, AStmt, 2861 StartLoc, EndLoc); 2862 break; 2863 case OMPD_parallel_for: 2864 Res = ActOnOpenMPParallelForDirective(ClausesWithImplicit, AStmt, StartLoc, 2865 EndLoc, VarsWithInheritedDSA); 2866 AllowedNameModifiers.push_back(OMPD_parallel); 2867 break; 2868 case OMPD_parallel_for_simd: 2869 Res = ActOnOpenMPParallelForSimdDirective( 2870 ClausesWithImplicit, AStmt, StartLoc, EndLoc, VarsWithInheritedDSA); 2871 AllowedNameModifiers.push_back(OMPD_parallel); 2872 break; 2873 case OMPD_parallel_sections: 2874 Res = ActOnOpenMPParallelSectionsDirective(ClausesWithImplicit, AStmt, 2875 StartLoc, EndLoc); 2876 AllowedNameModifiers.push_back(OMPD_parallel); 2877 break; 2878 case OMPD_task: 2879 Res = 2880 ActOnOpenMPTaskDirective(ClausesWithImplicit, AStmt, StartLoc, EndLoc); 2881 AllowedNameModifiers.push_back(OMPD_task); 2882 break; 2883 case OMPD_taskyield: 2884 assert(ClausesWithImplicit.empty() && 2885 "No clauses are allowed for 'omp taskyield' directive"); 2886 assert(AStmt == nullptr && 2887 "No associated statement allowed for 'omp taskyield' directive"); 2888 Res = ActOnOpenMPTaskyieldDirective(StartLoc, EndLoc); 2889 break; 2890 case OMPD_barrier: 2891 assert(ClausesWithImplicit.empty() && 2892 "No clauses are allowed for 'omp barrier' directive"); 2893 assert(AStmt == nullptr && 2894 "No associated statement allowed for 'omp barrier' directive"); 2895 Res = ActOnOpenMPBarrierDirective(StartLoc, EndLoc); 2896 break; 2897 case OMPD_taskwait: 2898 assert(ClausesWithImplicit.empty() && 2899 "No clauses are allowed for 'omp taskwait' directive"); 2900 assert(AStmt == nullptr && 2901 "No associated statement allowed for 'omp taskwait' directive"); 2902 Res = ActOnOpenMPTaskwaitDirective(StartLoc, EndLoc); 2903 break; 2904 case OMPD_taskgroup: 2905 Res = ActOnOpenMPTaskgroupDirective(ClausesWithImplicit, AStmt, StartLoc, 2906 EndLoc); 2907 break; 2908 case OMPD_flush: 2909 assert(AStmt == nullptr && 2910 "No associated statement allowed for 'omp flush' directive"); 2911 Res = ActOnOpenMPFlushDirective(ClausesWithImplicit, StartLoc, EndLoc); 2912 break; 2913 case OMPD_ordered: 2914 Res = ActOnOpenMPOrderedDirective(ClausesWithImplicit, AStmt, StartLoc, 2915 EndLoc); 2916 break; 2917 case OMPD_atomic: 2918 Res = ActOnOpenMPAtomicDirective(ClausesWithImplicit, AStmt, StartLoc, 2919 EndLoc); 2920 break; 2921 case OMPD_teams: 2922 Res = 2923 ActOnOpenMPTeamsDirective(ClausesWithImplicit, AStmt, StartLoc, EndLoc); 2924 break; 2925 case OMPD_target: 2926 Res = ActOnOpenMPTargetDirective(ClausesWithImplicit, AStmt, StartLoc, 2927 EndLoc); 2928 AllowedNameModifiers.push_back(OMPD_target); 2929 break; 2930 case OMPD_target_parallel: 2931 Res = ActOnOpenMPTargetParallelDirective(ClausesWithImplicit, AStmt, 2932 StartLoc, EndLoc); 2933 AllowedNameModifiers.push_back(OMPD_target); 2934 AllowedNameModifiers.push_back(OMPD_parallel); 2935 break; 2936 case OMPD_target_parallel_for: 2937 Res = ActOnOpenMPTargetParallelForDirective( 2938 ClausesWithImplicit, AStmt, StartLoc, EndLoc, VarsWithInheritedDSA); 2939 AllowedNameModifiers.push_back(OMPD_target); 2940 AllowedNameModifiers.push_back(OMPD_parallel); 2941 break; 2942 case OMPD_cancellation_point: 2943 assert(ClausesWithImplicit.empty() && 2944 "No clauses are allowed for 'omp cancellation point' directive"); 2945 assert(AStmt == nullptr && "No associated statement allowed for 'omp " 2946 "cancellation point' directive"); 2947 Res = ActOnOpenMPCancellationPointDirective(StartLoc, EndLoc, CancelRegion); 2948 break; 2949 case OMPD_cancel: 2950 assert(AStmt == nullptr && 2951 "No associated statement allowed for 'omp cancel' directive"); 2952 Res = ActOnOpenMPCancelDirective(ClausesWithImplicit, StartLoc, EndLoc, 2953 CancelRegion); 2954 AllowedNameModifiers.push_back(OMPD_cancel); 2955 break; 2956 case OMPD_target_data: 2957 Res = ActOnOpenMPTargetDataDirective(ClausesWithImplicit, AStmt, StartLoc, 2958 EndLoc); 2959 AllowedNameModifiers.push_back(OMPD_target_data); 2960 break; 2961 case OMPD_target_enter_data: 2962 Res = ActOnOpenMPTargetEnterDataDirective(ClausesWithImplicit, StartLoc, 2963 EndLoc); 2964 AllowedNameModifiers.push_back(OMPD_target_enter_data); 2965 break; 2966 case OMPD_target_exit_data: 2967 Res = ActOnOpenMPTargetExitDataDirective(ClausesWithImplicit, StartLoc, 2968 EndLoc); 2969 AllowedNameModifiers.push_back(OMPD_target_exit_data); 2970 break; 2971 case OMPD_taskloop: 2972 Res = ActOnOpenMPTaskLoopDirective(ClausesWithImplicit, AStmt, StartLoc, 2973 EndLoc, VarsWithInheritedDSA); 2974 AllowedNameModifiers.push_back(OMPD_taskloop); 2975 break; 2976 case OMPD_taskloop_simd: 2977 Res = ActOnOpenMPTaskLoopSimdDirective(ClausesWithImplicit, AStmt, StartLoc, 2978 EndLoc, VarsWithInheritedDSA); 2979 AllowedNameModifiers.push_back(OMPD_taskloop); 2980 break; 2981 case OMPD_distribute: 2982 Res = ActOnOpenMPDistributeDirective(ClausesWithImplicit, AStmt, StartLoc, 2983 EndLoc, VarsWithInheritedDSA); 2984 break; 2985 case OMPD_target_update: 2986 assert(!AStmt && "Statement is not allowed for target update"); 2987 Res = 2988 ActOnOpenMPTargetUpdateDirective(ClausesWithImplicit, StartLoc, EndLoc); 2989 AllowedNameModifiers.push_back(OMPD_target_update); 2990 break; 2991 case OMPD_distribute_parallel_for: 2992 Res = ActOnOpenMPDistributeParallelForDirective( 2993 ClausesWithImplicit, AStmt, StartLoc, EndLoc, VarsWithInheritedDSA); 2994 AllowedNameModifiers.push_back(OMPD_parallel); 2995 break; 2996 case OMPD_distribute_parallel_for_simd: 2997 Res = ActOnOpenMPDistributeParallelForSimdDirective( 2998 ClausesWithImplicit, AStmt, StartLoc, EndLoc, VarsWithInheritedDSA); 2999 AllowedNameModifiers.push_back(OMPD_parallel); 3000 break; 3001 case OMPD_distribute_simd: 3002 Res = ActOnOpenMPDistributeSimdDirective( 3003 ClausesWithImplicit, AStmt, StartLoc, EndLoc, VarsWithInheritedDSA); 3004 break; 3005 case OMPD_target_parallel_for_simd: 3006 Res = ActOnOpenMPTargetParallelForSimdDirective( 3007 ClausesWithImplicit, AStmt, StartLoc, EndLoc, VarsWithInheritedDSA); 3008 AllowedNameModifiers.push_back(OMPD_target); 3009 AllowedNameModifiers.push_back(OMPD_parallel); 3010 break; 3011 case OMPD_target_simd: 3012 Res = ActOnOpenMPTargetSimdDirective(ClausesWithImplicit, AStmt, StartLoc, 3013 EndLoc, VarsWithInheritedDSA); 3014 AllowedNameModifiers.push_back(OMPD_target); 3015 break; 3016 case OMPD_teams_distribute: 3017 Res = ActOnOpenMPTeamsDistributeDirective( 3018 ClausesWithImplicit, AStmt, StartLoc, EndLoc, VarsWithInheritedDSA); 3019 break; 3020 case OMPD_teams_distribute_simd: 3021 Res = ActOnOpenMPTeamsDistributeSimdDirective( 3022 ClausesWithImplicit, AStmt, StartLoc, EndLoc, VarsWithInheritedDSA); 3023 break; 3024 case OMPD_teams_distribute_parallel_for_simd: 3025 Res = ActOnOpenMPTeamsDistributeParallelForSimdDirective( 3026 ClausesWithImplicit, AStmt, StartLoc, EndLoc, VarsWithInheritedDSA); 3027 AllowedNameModifiers.push_back(OMPD_parallel); 3028 break; 3029 case OMPD_teams_distribute_parallel_for: 3030 Res = ActOnOpenMPTeamsDistributeParallelForDirective( 3031 ClausesWithImplicit, AStmt, StartLoc, EndLoc, VarsWithInheritedDSA); 3032 AllowedNameModifiers.push_back(OMPD_parallel); 3033 break; 3034 case OMPD_target_teams: 3035 Res = ActOnOpenMPTargetTeamsDirective(ClausesWithImplicit, AStmt, StartLoc, 3036 EndLoc); 3037 AllowedNameModifiers.push_back(OMPD_target); 3038 break; 3039 case OMPD_target_teams_distribute: 3040 Res = ActOnOpenMPTargetTeamsDistributeDirective( 3041 ClausesWithImplicit, AStmt, StartLoc, EndLoc, VarsWithInheritedDSA); 3042 AllowedNameModifiers.push_back(OMPD_target); 3043 break; 3044 case OMPD_target_teams_distribute_parallel_for: 3045 Res = ActOnOpenMPTargetTeamsDistributeParallelForDirective( 3046 ClausesWithImplicit, AStmt, StartLoc, EndLoc, VarsWithInheritedDSA); 3047 AllowedNameModifiers.push_back(OMPD_target); 3048 AllowedNameModifiers.push_back(OMPD_parallel); 3049 break; 3050 case OMPD_target_teams_distribute_parallel_for_simd: 3051 Res = ActOnOpenMPTargetTeamsDistributeParallelForSimdDirective( 3052 ClausesWithImplicit, AStmt, StartLoc, EndLoc, VarsWithInheritedDSA); 3053 AllowedNameModifiers.push_back(OMPD_target); 3054 AllowedNameModifiers.push_back(OMPD_parallel); 3055 break; 3056 case OMPD_target_teams_distribute_simd: 3057 Res = ActOnOpenMPTargetTeamsDistributeSimdDirective( 3058 ClausesWithImplicit, AStmt, StartLoc, EndLoc, VarsWithInheritedDSA); 3059 AllowedNameModifiers.push_back(OMPD_target); 3060 break; 3061 case OMPD_declare_target: 3062 case OMPD_end_declare_target: 3063 case OMPD_threadprivate: 3064 case OMPD_declare_reduction: 3065 case OMPD_declare_simd: 3066 llvm_unreachable("OpenMP Directive is not allowed"); 3067 case OMPD_unknown: 3068 llvm_unreachable("Unknown OpenMP directive"); 3069 } 3070 3071 for (auto P : VarsWithInheritedDSA) { 3072 Diag(P.second->getExprLoc(), diag::err_omp_no_dsa_for_variable) 3073 << P.first << P.second->getSourceRange(); 3074 } 3075 ErrorFound = !VarsWithInheritedDSA.empty() || ErrorFound; 3076 3077 if (!AllowedNameModifiers.empty()) 3078 ErrorFound = checkIfClauses(*this, Kind, Clauses, AllowedNameModifiers) || 3079 ErrorFound; 3080 3081 if (ErrorFound) 3082 return StmtError(); 3083 return Res; 3084 } 3085 3086 Sema::DeclGroupPtrTy Sema::ActOnOpenMPDeclareSimdDirective( 3087 DeclGroupPtrTy DG, OMPDeclareSimdDeclAttr::BranchStateTy BS, Expr *Simdlen, 3088 ArrayRef<Expr *> Uniforms, ArrayRef<Expr *> Aligneds, 3089 ArrayRef<Expr *> Alignments, ArrayRef<Expr *> Linears, 3090 ArrayRef<unsigned> LinModifiers, ArrayRef<Expr *> Steps, SourceRange SR) { 3091 assert(Aligneds.size() == Alignments.size()); 3092 assert(Linears.size() == LinModifiers.size()); 3093 assert(Linears.size() == Steps.size()); 3094 if (!DG || DG.get().isNull()) 3095 return DeclGroupPtrTy(); 3096 3097 if (!DG.get().isSingleDecl()) { 3098 Diag(SR.getBegin(), diag::err_omp_single_decl_in_declare_simd); 3099 return DG; 3100 } 3101 auto *ADecl = DG.get().getSingleDecl(); 3102 if (auto *FTD = dyn_cast<FunctionTemplateDecl>(ADecl)) 3103 ADecl = FTD->getTemplatedDecl(); 3104 3105 auto *FD = dyn_cast<FunctionDecl>(ADecl); 3106 if (!FD) { 3107 Diag(ADecl->getLocation(), diag::err_omp_function_expected); 3108 return DeclGroupPtrTy(); 3109 } 3110 3111 // OpenMP [2.8.2, declare simd construct, Description] 3112 // The parameter of the simdlen clause must be a constant positive integer 3113 // expression. 3114 ExprResult SL; 3115 if (Simdlen) 3116 SL = VerifyPositiveIntegerConstantInClause(Simdlen, OMPC_simdlen); 3117 // OpenMP [2.8.2, declare simd construct, Description] 3118 // The special this pointer can be used as if was one of the arguments to the 3119 // function in any of the linear, aligned, or uniform clauses. 3120 // The uniform clause declares one or more arguments to have an invariant 3121 // value for all concurrent invocations of the function in the execution of a 3122 // single SIMD loop. 3123 llvm::DenseMap<Decl *, Expr *> UniformedArgs; 3124 Expr *UniformedLinearThis = nullptr; 3125 for (auto *E : Uniforms) { 3126 E = E->IgnoreParenImpCasts(); 3127 if (auto *DRE = dyn_cast<DeclRefExpr>(E)) 3128 if (auto *PVD = dyn_cast<ParmVarDecl>(DRE->getDecl())) 3129 if (FD->getNumParams() > PVD->getFunctionScopeIndex() && 3130 FD->getParamDecl(PVD->getFunctionScopeIndex()) 3131 ->getCanonicalDecl() == PVD->getCanonicalDecl()) { 3132 UniformedArgs.insert(std::make_pair(PVD->getCanonicalDecl(), E)); 3133 continue; 3134 } 3135 if (isa<CXXThisExpr>(E)) { 3136 UniformedLinearThis = E; 3137 continue; 3138 } 3139 Diag(E->getExprLoc(), diag::err_omp_param_or_this_in_clause) 3140 << FD->getDeclName() << (isa<CXXMethodDecl>(ADecl) ? 1 : 0); 3141 } 3142 // OpenMP [2.8.2, declare simd construct, Description] 3143 // The aligned clause declares that the object to which each list item points 3144 // is aligned to the number of bytes expressed in the optional parameter of 3145 // the aligned clause. 3146 // The special this pointer can be used as if was one of the arguments to the 3147 // function in any of the linear, aligned, or uniform clauses. 3148 // The type of list items appearing in the aligned clause must be array, 3149 // pointer, reference to array, or reference to pointer. 3150 llvm::DenseMap<Decl *, Expr *> AlignedArgs; 3151 Expr *AlignedThis = nullptr; 3152 for (auto *E : Aligneds) { 3153 E = E->IgnoreParenImpCasts(); 3154 if (auto *DRE = dyn_cast<DeclRefExpr>(E)) 3155 if (auto *PVD = dyn_cast<ParmVarDecl>(DRE->getDecl())) { 3156 auto *CanonPVD = PVD->getCanonicalDecl(); 3157 if (FD->getNumParams() > PVD->getFunctionScopeIndex() && 3158 FD->getParamDecl(PVD->getFunctionScopeIndex()) 3159 ->getCanonicalDecl() == CanonPVD) { 3160 // OpenMP [2.8.1, simd construct, Restrictions] 3161 // A list-item cannot appear in more than one aligned clause. 3162 if (AlignedArgs.count(CanonPVD) > 0) { 3163 Diag(E->getExprLoc(), diag::err_omp_aligned_twice) 3164 << 1 << E->getSourceRange(); 3165 Diag(AlignedArgs[CanonPVD]->getExprLoc(), 3166 diag::note_omp_explicit_dsa) 3167 << getOpenMPClauseName(OMPC_aligned); 3168 continue; 3169 } 3170 AlignedArgs[CanonPVD] = E; 3171 QualType QTy = PVD->getType() 3172 .getNonReferenceType() 3173 .getUnqualifiedType() 3174 .getCanonicalType(); 3175 const Type *Ty = QTy.getTypePtrOrNull(); 3176 if (!Ty || (!Ty->isArrayType() && !Ty->isPointerType())) { 3177 Diag(E->getExprLoc(), diag::err_omp_aligned_expected_array_or_ptr) 3178 << QTy << getLangOpts().CPlusPlus << E->getSourceRange(); 3179 Diag(PVD->getLocation(), diag::note_previous_decl) << PVD; 3180 } 3181 continue; 3182 } 3183 } 3184 if (isa<CXXThisExpr>(E)) { 3185 if (AlignedThis) { 3186 Diag(E->getExprLoc(), diag::err_omp_aligned_twice) 3187 << 2 << E->getSourceRange(); 3188 Diag(AlignedThis->getExprLoc(), diag::note_omp_explicit_dsa) 3189 << getOpenMPClauseName(OMPC_aligned); 3190 } 3191 AlignedThis = E; 3192 continue; 3193 } 3194 Diag(E->getExprLoc(), diag::err_omp_param_or_this_in_clause) 3195 << FD->getDeclName() << (isa<CXXMethodDecl>(ADecl) ? 1 : 0); 3196 } 3197 // The optional parameter of the aligned clause, alignment, must be a constant 3198 // positive integer expression. If no optional parameter is specified, 3199 // implementation-defined default alignments for SIMD instructions on the 3200 // target platforms are assumed. 3201 SmallVector<Expr *, 4> NewAligns; 3202 for (auto *E : Alignments) { 3203 ExprResult Align; 3204 if (E) 3205 Align = VerifyPositiveIntegerConstantInClause(E, OMPC_aligned); 3206 NewAligns.push_back(Align.get()); 3207 } 3208 // OpenMP [2.8.2, declare simd construct, Description] 3209 // The linear clause declares one or more list items to be private to a SIMD 3210 // lane and to have a linear relationship with respect to the iteration space 3211 // of a loop. 3212 // The special this pointer can be used as if was one of the arguments to the 3213 // function in any of the linear, aligned, or uniform clauses. 3214 // When a linear-step expression is specified in a linear clause it must be 3215 // either a constant integer expression or an integer-typed parameter that is 3216 // specified in a uniform clause on the directive. 3217 llvm::DenseMap<Decl *, Expr *> LinearArgs; 3218 const bool IsUniformedThis = UniformedLinearThis != nullptr; 3219 auto MI = LinModifiers.begin(); 3220 for (auto *E : Linears) { 3221 auto LinKind = static_cast<OpenMPLinearClauseKind>(*MI); 3222 ++MI; 3223 E = E->IgnoreParenImpCasts(); 3224 if (auto *DRE = dyn_cast<DeclRefExpr>(E)) 3225 if (auto *PVD = dyn_cast<ParmVarDecl>(DRE->getDecl())) { 3226 auto *CanonPVD = PVD->getCanonicalDecl(); 3227 if (FD->getNumParams() > PVD->getFunctionScopeIndex() && 3228 FD->getParamDecl(PVD->getFunctionScopeIndex()) 3229 ->getCanonicalDecl() == CanonPVD) { 3230 // OpenMP [2.15.3.7, linear Clause, Restrictions] 3231 // A list-item cannot appear in more than one linear clause. 3232 if (LinearArgs.count(CanonPVD) > 0) { 3233 Diag(E->getExprLoc(), diag::err_omp_wrong_dsa) 3234 << getOpenMPClauseName(OMPC_linear) 3235 << getOpenMPClauseName(OMPC_linear) << E->getSourceRange(); 3236 Diag(LinearArgs[CanonPVD]->getExprLoc(), 3237 diag::note_omp_explicit_dsa) 3238 << getOpenMPClauseName(OMPC_linear); 3239 continue; 3240 } 3241 // Each argument can appear in at most one uniform or linear clause. 3242 if (UniformedArgs.count(CanonPVD) > 0) { 3243 Diag(E->getExprLoc(), diag::err_omp_wrong_dsa) 3244 << getOpenMPClauseName(OMPC_linear) 3245 << getOpenMPClauseName(OMPC_uniform) << E->getSourceRange(); 3246 Diag(UniformedArgs[CanonPVD]->getExprLoc(), 3247 diag::note_omp_explicit_dsa) 3248 << getOpenMPClauseName(OMPC_uniform); 3249 continue; 3250 } 3251 LinearArgs[CanonPVD] = E; 3252 if (E->isValueDependent() || E->isTypeDependent() || 3253 E->isInstantiationDependent() || 3254 E->containsUnexpandedParameterPack()) 3255 continue; 3256 (void)CheckOpenMPLinearDecl(CanonPVD, E->getExprLoc(), LinKind, 3257 PVD->getOriginalType()); 3258 continue; 3259 } 3260 } 3261 if (isa<CXXThisExpr>(E)) { 3262 if (UniformedLinearThis) { 3263 Diag(E->getExprLoc(), diag::err_omp_wrong_dsa) 3264 << getOpenMPClauseName(OMPC_linear) 3265 << getOpenMPClauseName(IsUniformedThis ? OMPC_uniform : OMPC_linear) 3266 << E->getSourceRange(); 3267 Diag(UniformedLinearThis->getExprLoc(), diag::note_omp_explicit_dsa) 3268 << getOpenMPClauseName(IsUniformedThis ? OMPC_uniform 3269 : OMPC_linear); 3270 continue; 3271 } 3272 UniformedLinearThis = E; 3273 if (E->isValueDependent() || E->isTypeDependent() || 3274 E->isInstantiationDependent() || E->containsUnexpandedParameterPack()) 3275 continue; 3276 (void)CheckOpenMPLinearDecl(/*D=*/nullptr, E->getExprLoc(), LinKind, 3277 E->getType()); 3278 continue; 3279 } 3280 Diag(E->getExprLoc(), diag::err_omp_param_or_this_in_clause) 3281 << FD->getDeclName() << (isa<CXXMethodDecl>(ADecl) ? 1 : 0); 3282 } 3283 Expr *Step = nullptr; 3284 Expr *NewStep = nullptr; 3285 SmallVector<Expr *, 4> NewSteps; 3286 for (auto *E : Steps) { 3287 // Skip the same step expression, it was checked already. 3288 if (Step == E || !E) { 3289 NewSteps.push_back(E ? NewStep : nullptr); 3290 continue; 3291 } 3292 Step = E; 3293 if (auto *DRE = dyn_cast<DeclRefExpr>(Step)) 3294 if (auto *PVD = dyn_cast<ParmVarDecl>(DRE->getDecl())) { 3295 auto *CanonPVD = PVD->getCanonicalDecl(); 3296 if (UniformedArgs.count(CanonPVD) == 0) { 3297 Diag(Step->getExprLoc(), diag::err_omp_expected_uniform_param) 3298 << Step->getSourceRange(); 3299 } else if (E->isValueDependent() || E->isTypeDependent() || 3300 E->isInstantiationDependent() || 3301 E->containsUnexpandedParameterPack() || 3302 CanonPVD->getType()->hasIntegerRepresentation()) 3303 NewSteps.push_back(Step); 3304 else { 3305 Diag(Step->getExprLoc(), diag::err_omp_expected_int_param) 3306 << Step->getSourceRange(); 3307 } 3308 continue; 3309 } 3310 NewStep = Step; 3311 if (Step && !Step->isValueDependent() && !Step->isTypeDependent() && 3312 !Step->isInstantiationDependent() && 3313 !Step->containsUnexpandedParameterPack()) { 3314 NewStep = PerformOpenMPImplicitIntegerConversion(Step->getExprLoc(), Step) 3315 .get(); 3316 if (NewStep) 3317 NewStep = VerifyIntegerConstantExpression(NewStep).get(); 3318 } 3319 NewSteps.push_back(NewStep); 3320 } 3321 auto *NewAttr = OMPDeclareSimdDeclAttr::CreateImplicit( 3322 Context, BS, SL.get(), const_cast<Expr **>(Uniforms.data()), 3323 Uniforms.size(), const_cast<Expr **>(Aligneds.data()), Aligneds.size(), 3324 const_cast<Expr **>(NewAligns.data()), NewAligns.size(), 3325 const_cast<Expr **>(Linears.data()), Linears.size(), 3326 const_cast<unsigned *>(LinModifiers.data()), LinModifiers.size(), 3327 NewSteps.data(), NewSteps.size(), SR); 3328 ADecl->addAttr(NewAttr); 3329 return ConvertDeclToDeclGroup(ADecl); 3330 } 3331 3332 StmtResult Sema::ActOnOpenMPParallelDirective(ArrayRef<OMPClause *> Clauses, 3333 Stmt *AStmt, 3334 SourceLocation StartLoc, 3335 SourceLocation EndLoc) { 3336 if (!AStmt) 3337 return StmtError(); 3338 3339 CapturedStmt *CS = cast<CapturedStmt>(AStmt); 3340 // 1.2.2 OpenMP Language Terminology 3341 // Structured block - An executable statement with a single entry at the 3342 // top and a single exit at the bottom. 3343 // The point of exit cannot be a branch out of the structured block. 3344 // longjmp() and throw() must not violate the entry/exit criteria. 3345 CS->getCapturedDecl()->setNothrow(); 3346 3347 getCurFunction()->setHasBranchProtectedScope(); 3348 3349 return OMPParallelDirective::Create(Context, StartLoc, EndLoc, Clauses, AStmt, 3350 DSAStack->isCancelRegion()); 3351 } 3352 3353 namespace { 3354 /// \brief Helper class for checking canonical form of the OpenMP loops and 3355 /// extracting iteration space of each loop in the loop nest, that will be used 3356 /// for IR generation. 3357 class OpenMPIterationSpaceChecker { 3358 /// \brief Reference to Sema. 3359 Sema &SemaRef; 3360 /// \brief A location for diagnostics (when there is no some better location). 3361 SourceLocation DefaultLoc; 3362 /// \brief A location for diagnostics (when increment is not compatible). 3363 SourceLocation ConditionLoc; 3364 /// \brief A source location for referring to loop init later. 3365 SourceRange InitSrcRange; 3366 /// \brief A source location for referring to condition later. 3367 SourceRange ConditionSrcRange; 3368 /// \brief A source location for referring to increment later. 3369 SourceRange IncrementSrcRange; 3370 /// \brief Loop variable. 3371 ValueDecl *LCDecl = nullptr; 3372 /// \brief Reference to loop variable. 3373 Expr *LCRef = nullptr; 3374 /// \brief Lower bound (initializer for the var). 3375 Expr *LB = nullptr; 3376 /// \brief Upper bound. 3377 Expr *UB = nullptr; 3378 /// \brief Loop step (increment). 3379 Expr *Step = nullptr; 3380 /// \brief This flag is true when condition is one of: 3381 /// Var < UB 3382 /// Var <= UB 3383 /// UB > Var 3384 /// UB >= Var 3385 bool TestIsLessOp = false; 3386 /// \brief This flag is true when condition is strict ( < or > ). 3387 bool TestIsStrictOp = false; 3388 /// \brief This flag is true when step is subtracted on each iteration. 3389 bool SubtractStep = false; 3390 3391 public: 3392 OpenMPIterationSpaceChecker(Sema &SemaRef, SourceLocation DefaultLoc) 3393 : SemaRef(SemaRef), DefaultLoc(DefaultLoc), ConditionLoc(DefaultLoc) {} 3394 /// \brief Check init-expr for canonical loop form and save loop counter 3395 /// variable - #Var and its initialization value - #LB. 3396 bool CheckInit(Stmt *S, bool EmitDiags = true); 3397 /// \brief Check test-expr for canonical form, save upper-bound (#UB), flags 3398 /// for less/greater and for strict/non-strict comparison. 3399 bool CheckCond(Expr *S); 3400 /// \brief Check incr-expr for canonical loop form and return true if it 3401 /// does not conform, otherwise save loop step (#Step). 3402 bool CheckInc(Expr *S); 3403 /// \brief Return the loop counter variable. 3404 ValueDecl *GetLoopDecl() const { return LCDecl; } 3405 /// \brief Return the reference expression to loop counter variable. 3406 Expr *GetLoopDeclRefExpr() const { return LCRef; } 3407 /// \brief Source range of the loop init. 3408 SourceRange GetInitSrcRange() const { return InitSrcRange; } 3409 /// \brief Source range of the loop condition. 3410 SourceRange GetConditionSrcRange() const { return ConditionSrcRange; } 3411 /// \brief Source range of the loop increment. 3412 SourceRange GetIncrementSrcRange() const { return IncrementSrcRange; } 3413 /// \brief True if the step should be subtracted. 3414 bool ShouldSubtractStep() const { return SubtractStep; } 3415 /// \brief Build the expression to calculate the number of iterations. 3416 Expr * 3417 BuildNumIterations(Scope *S, const bool LimitedType, 3418 llvm::MapVector<Expr *, DeclRefExpr *> &Captures) const; 3419 /// \brief Build the precondition expression for the loops. 3420 Expr *BuildPreCond(Scope *S, Expr *Cond, 3421 llvm::MapVector<Expr *, DeclRefExpr *> &Captures) const; 3422 /// \brief Build reference expression to the counter be used for codegen. 3423 DeclRefExpr *BuildCounterVar(llvm::MapVector<Expr *, DeclRefExpr *> &Captures, 3424 DSAStackTy &DSA) const; 3425 /// \brief Build reference expression to the private counter be used for 3426 /// codegen. 3427 Expr *BuildPrivateCounterVar() const; 3428 /// \brief Build initialization of the counter be used for codegen. 3429 Expr *BuildCounterInit() const; 3430 /// \brief Build step of the counter be used for codegen. 3431 Expr *BuildCounterStep() const; 3432 /// \brief Return true if any expression is dependent. 3433 bool Dependent() const; 3434 3435 private: 3436 /// \brief Check the right-hand side of an assignment in the increment 3437 /// expression. 3438 bool CheckIncRHS(Expr *RHS); 3439 /// \brief Helper to set loop counter variable and its initializer. 3440 bool SetLCDeclAndLB(ValueDecl *NewLCDecl, Expr *NewDeclRefExpr, Expr *NewLB); 3441 /// \brief Helper to set upper bound. 3442 bool SetUB(Expr *NewUB, bool LessOp, bool StrictOp, SourceRange SR, 3443 SourceLocation SL); 3444 /// \brief Helper to set loop increment. 3445 bool SetStep(Expr *NewStep, bool Subtract); 3446 }; 3447 3448 bool OpenMPIterationSpaceChecker::Dependent() const { 3449 if (!LCDecl) { 3450 assert(!LB && !UB && !Step); 3451 return false; 3452 } 3453 return LCDecl->getType()->isDependentType() || 3454 (LB && LB->isValueDependent()) || (UB && UB->isValueDependent()) || 3455 (Step && Step->isValueDependent()); 3456 } 3457 3458 bool OpenMPIterationSpaceChecker::SetLCDeclAndLB(ValueDecl *NewLCDecl, 3459 Expr *NewLCRefExpr, 3460 Expr *NewLB) { 3461 // State consistency checking to ensure correct usage. 3462 assert(LCDecl == nullptr && LB == nullptr && LCRef == nullptr && 3463 UB == nullptr && Step == nullptr && !TestIsLessOp && !TestIsStrictOp); 3464 if (!NewLCDecl || !NewLB) 3465 return true; 3466 LCDecl = getCanonicalDecl(NewLCDecl); 3467 LCRef = NewLCRefExpr; 3468 if (auto *CE = dyn_cast_or_null<CXXConstructExpr>(NewLB)) 3469 if (const CXXConstructorDecl *Ctor = CE->getConstructor()) 3470 if ((Ctor->isCopyOrMoveConstructor() || 3471 Ctor->isConvertingConstructor(/*AllowExplicit=*/false)) && 3472 CE->getNumArgs() > 0 && CE->getArg(0) != nullptr) 3473 NewLB = CE->getArg(0)->IgnoreParenImpCasts(); 3474 LB = NewLB; 3475 return false; 3476 } 3477 3478 bool OpenMPIterationSpaceChecker::SetUB(Expr *NewUB, bool LessOp, bool StrictOp, 3479 SourceRange SR, SourceLocation SL) { 3480 // State consistency checking to ensure correct usage. 3481 assert(LCDecl != nullptr && LB != nullptr && UB == nullptr && 3482 Step == nullptr && !TestIsLessOp && !TestIsStrictOp); 3483 if (!NewUB) 3484 return true; 3485 UB = NewUB; 3486 TestIsLessOp = LessOp; 3487 TestIsStrictOp = StrictOp; 3488 ConditionSrcRange = SR; 3489 ConditionLoc = SL; 3490 return false; 3491 } 3492 3493 bool OpenMPIterationSpaceChecker::SetStep(Expr *NewStep, bool Subtract) { 3494 // State consistency checking to ensure correct usage. 3495 assert(LCDecl != nullptr && LB != nullptr && Step == nullptr); 3496 if (!NewStep) 3497 return true; 3498 if (!NewStep->isValueDependent()) { 3499 // Check that the step is integer expression. 3500 SourceLocation StepLoc = NewStep->getLocStart(); 3501 ExprResult Val = SemaRef.PerformOpenMPImplicitIntegerConversion( 3502 StepLoc, getExprAsWritten(NewStep)); 3503 if (Val.isInvalid()) 3504 return true; 3505 NewStep = Val.get(); 3506 3507 // OpenMP [2.6, Canonical Loop Form, Restrictions] 3508 // If test-expr is of form var relational-op b and relational-op is < or 3509 // <= then incr-expr must cause var to increase on each iteration of the 3510 // loop. If test-expr is of form var relational-op b and relational-op is 3511 // > or >= then incr-expr must cause var to decrease on each iteration of 3512 // the loop. 3513 // If test-expr is of form b relational-op var and relational-op is < or 3514 // <= then incr-expr must cause var to decrease on each iteration of the 3515 // loop. If test-expr is of form b relational-op var and relational-op is 3516 // > or >= then incr-expr must cause var to increase on each iteration of 3517 // the loop. 3518 llvm::APSInt Result; 3519 bool IsConstant = NewStep->isIntegerConstantExpr(Result, SemaRef.Context); 3520 bool IsUnsigned = !NewStep->getType()->hasSignedIntegerRepresentation(); 3521 bool IsConstNeg = 3522 IsConstant && Result.isSigned() && (Subtract != Result.isNegative()); 3523 bool IsConstPos = 3524 IsConstant && Result.isSigned() && (Subtract == Result.isNegative()); 3525 bool IsConstZero = IsConstant && !Result.getBoolValue(); 3526 if (UB && (IsConstZero || 3527 (TestIsLessOp ? (IsConstNeg || (IsUnsigned && Subtract)) 3528 : (IsConstPos || (IsUnsigned && !Subtract))))) { 3529 SemaRef.Diag(NewStep->getExprLoc(), 3530 diag::err_omp_loop_incr_not_compatible) 3531 << LCDecl << TestIsLessOp << NewStep->getSourceRange(); 3532 SemaRef.Diag(ConditionLoc, 3533 diag::note_omp_loop_cond_requres_compatible_incr) 3534 << TestIsLessOp << ConditionSrcRange; 3535 return true; 3536 } 3537 if (TestIsLessOp == Subtract) { 3538 NewStep = 3539 SemaRef.CreateBuiltinUnaryOp(NewStep->getExprLoc(), UO_Minus, NewStep) 3540 .get(); 3541 Subtract = !Subtract; 3542 } 3543 } 3544 3545 Step = NewStep; 3546 SubtractStep = Subtract; 3547 return false; 3548 } 3549 3550 bool OpenMPIterationSpaceChecker::CheckInit(Stmt *S, bool EmitDiags) { 3551 // Check init-expr for canonical loop form and save loop counter 3552 // variable - #Var and its initialization value - #LB. 3553 // OpenMP [2.6] Canonical loop form. init-expr may be one of the following: 3554 // var = lb 3555 // integer-type var = lb 3556 // random-access-iterator-type var = lb 3557 // pointer-type var = lb 3558 // 3559 if (!S) { 3560 if (EmitDiags) { 3561 SemaRef.Diag(DefaultLoc, diag::err_omp_loop_not_canonical_init); 3562 } 3563 return true; 3564 } 3565 if (auto *ExprTemp = dyn_cast<ExprWithCleanups>(S)) 3566 if (!ExprTemp->cleanupsHaveSideEffects()) 3567 S = ExprTemp->getSubExpr(); 3568 3569 InitSrcRange = S->getSourceRange(); 3570 if (Expr *E = dyn_cast<Expr>(S)) 3571 S = E->IgnoreParens(); 3572 if (auto *BO = dyn_cast<BinaryOperator>(S)) { 3573 if (BO->getOpcode() == BO_Assign) { 3574 auto *LHS = BO->getLHS()->IgnoreParens(); 3575 if (auto *DRE = dyn_cast<DeclRefExpr>(LHS)) { 3576 if (auto *CED = dyn_cast<OMPCapturedExprDecl>(DRE->getDecl())) 3577 if (auto *ME = dyn_cast<MemberExpr>(getExprAsWritten(CED->getInit()))) 3578 return SetLCDeclAndLB(ME->getMemberDecl(), ME, BO->getRHS()); 3579 return SetLCDeclAndLB(DRE->getDecl(), DRE, BO->getRHS()); 3580 } 3581 if (auto *ME = dyn_cast<MemberExpr>(LHS)) { 3582 if (ME->isArrow() && 3583 isa<CXXThisExpr>(ME->getBase()->IgnoreParenImpCasts())) 3584 return SetLCDeclAndLB(ME->getMemberDecl(), ME, BO->getRHS()); 3585 } 3586 } 3587 } else if (auto *DS = dyn_cast<DeclStmt>(S)) { 3588 if (DS->isSingleDecl()) { 3589 if (auto *Var = dyn_cast_or_null<VarDecl>(DS->getSingleDecl())) { 3590 if (Var->hasInit() && !Var->getType()->isReferenceType()) { 3591 // Accept non-canonical init form here but emit ext. warning. 3592 if (Var->getInitStyle() != VarDecl::CInit && EmitDiags) 3593 SemaRef.Diag(S->getLocStart(), 3594 diag::ext_omp_loop_not_canonical_init) 3595 << S->getSourceRange(); 3596 return SetLCDeclAndLB(Var, nullptr, Var->getInit()); 3597 } 3598 } 3599 } 3600 } else if (auto *CE = dyn_cast<CXXOperatorCallExpr>(S)) { 3601 if (CE->getOperator() == OO_Equal) { 3602 auto *LHS = CE->getArg(0); 3603 if (auto *DRE = dyn_cast<DeclRefExpr>(LHS)) { 3604 if (auto *CED = dyn_cast<OMPCapturedExprDecl>(DRE->getDecl())) 3605 if (auto *ME = dyn_cast<MemberExpr>(getExprAsWritten(CED->getInit()))) 3606 return SetLCDeclAndLB(ME->getMemberDecl(), ME, BO->getRHS()); 3607 return SetLCDeclAndLB(DRE->getDecl(), DRE, CE->getArg(1)); 3608 } 3609 if (auto *ME = dyn_cast<MemberExpr>(LHS)) { 3610 if (ME->isArrow() && 3611 isa<CXXThisExpr>(ME->getBase()->IgnoreParenImpCasts())) 3612 return SetLCDeclAndLB(ME->getMemberDecl(), ME, BO->getRHS()); 3613 } 3614 } 3615 } 3616 3617 if (Dependent() || SemaRef.CurContext->isDependentContext()) 3618 return false; 3619 if (EmitDiags) { 3620 SemaRef.Diag(S->getLocStart(), diag::err_omp_loop_not_canonical_init) 3621 << S->getSourceRange(); 3622 } 3623 return true; 3624 } 3625 3626 /// \brief Ignore parenthesizes, implicit casts, copy constructor and return the 3627 /// variable (which may be the loop variable) if possible. 3628 static const ValueDecl *GetInitLCDecl(Expr *E) { 3629 if (!E) 3630 return nullptr; 3631 E = getExprAsWritten(E); 3632 if (auto *CE = dyn_cast_or_null<CXXConstructExpr>(E)) 3633 if (const CXXConstructorDecl *Ctor = CE->getConstructor()) 3634 if ((Ctor->isCopyOrMoveConstructor() || 3635 Ctor->isConvertingConstructor(/*AllowExplicit=*/false)) && 3636 CE->getNumArgs() > 0 && CE->getArg(0) != nullptr) 3637 E = CE->getArg(0)->IgnoreParenImpCasts(); 3638 if (auto *DRE = dyn_cast_or_null<DeclRefExpr>(E)) { 3639 if (auto *VD = dyn_cast<VarDecl>(DRE->getDecl())) 3640 return getCanonicalDecl(VD); 3641 } 3642 if (auto *ME = dyn_cast_or_null<MemberExpr>(E)) 3643 if (ME->isArrow() && isa<CXXThisExpr>(ME->getBase()->IgnoreParenImpCasts())) 3644 return getCanonicalDecl(ME->getMemberDecl()); 3645 return nullptr; 3646 } 3647 3648 bool OpenMPIterationSpaceChecker::CheckCond(Expr *S) { 3649 // Check test-expr for canonical form, save upper-bound UB, flags for 3650 // less/greater and for strict/non-strict comparison. 3651 // OpenMP [2.6] Canonical loop form. Test-expr may be one of the following: 3652 // var relational-op b 3653 // b relational-op var 3654 // 3655 if (!S) { 3656 SemaRef.Diag(DefaultLoc, diag::err_omp_loop_not_canonical_cond) << LCDecl; 3657 return true; 3658 } 3659 S = getExprAsWritten(S); 3660 SourceLocation CondLoc = S->getLocStart(); 3661 if (auto *BO = dyn_cast<BinaryOperator>(S)) { 3662 if (BO->isRelationalOp()) { 3663 if (GetInitLCDecl(BO->getLHS()) == LCDecl) 3664 return SetUB(BO->getRHS(), 3665 (BO->getOpcode() == BO_LT || BO->getOpcode() == BO_LE), 3666 (BO->getOpcode() == BO_LT || BO->getOpcode() == BO_GT), 3667 BO->getSourceRange(), BO->getOperatorLoc()); 3668 if (GetInitLCDecl(BO->getRHS()) == LCDecl) 3669 return SetUB(BO->getLHS(), 3670 (BO->getOpcode() == BO_GT || BO->getOpcode() == BO_GE), 3671 (BO->getOpcode() == BO_LT || BO->getOpcode() == BO_GT), 3672 BO->getSourceRange(), BO->getOperatorLoc()); 3673 } 3674 } else if (auto *CE = dyn_cast<CXXOperatorCallExpr>(S)) { 3675 if (CE->getNumArgs() == 2) { 3676 auto Op = CE->getOperator(); 3677 switch (Op) { 3678 case OO_Greater: 3679 case OO_GreaterEqual: 3680 case OO_Less: 3681 case OO_LessEqual: 3682 if (GetInitLCDecl(CE->getArg(0)) == LCDecl) 3683 return SetUB(CE->getArg(1), Op == OO_Less || Op == OO_LessEqual, 3684 Op == OO_Less || Op == OO_Greater, CE->getSourceRange(), 3685 CE->getOperatorLoc()); 3686 if (GetInitLCDecl(CE->getArg(1)) == LCDecl) 3687 return SetUB(CE->getArg(0), Op == OO_Greater || Op == OO_GreaterEqual, 3688 Op == OO_Less || Op == OO_Greater, CE->getSourceRange(), 3689 CE->getOperatorLoc()); 3690 break; 3691 default: 3692 break; 3693 } 3694 } 3695 } 3696 if (Dependent() || SemaRef.CurContext->isDependentContext()) 3697 return false; 3698 SemaRef.Diag(CondLoc, diag::err_omp_loop_not_canonical_cond) 3699 << S->getSourceRange() << LCDecl; 3700 return true; 3701 } 3702 3703 bool OpenMPIterationSpaceChecker::CheckIncRHS(Expr *RHS) { 3704 // RHS of canonical loop form increment can be: 3705 // var + incr 3706 // incr + var 3707 // var - incr 3708 // 3709 RHS = RHS->IgnoreParenImpCasts(); 3710 if (auto *BO = dyn_cast<BinaryOperator>(RHS)) { 3711 if (BO->isAdditiveOp()) { 3712 bool IsAdd = BO->getOpcode() == BO_Add; 3713 if (GetInitLCDecl(BO->getLHS()) == LCDecl) 3714 return SetStep(BO->getRHS(), !IsAdd); 3715 if (IsAdd && GetInitLCDecl(BO->getRHS()) == LCDecl) 3716 return SetStep(BO->getLHS(), false); 3717 } 3718 } else if (auto *CE = dyn_cast<CXXOperatorCallExpr>(RHS)) { 3719 bool IsAdd = CE->getOperator() == OO_Plus; 3720 if ((IsAdd || CE->getOperator() == OO_Minus) && CE->getNumArgs() == 2) { 3721 if (GetInitLCDecl(CE->getArg(0)) == LCDecl) 3722 return SetStep(CE->getArg(1), !IsAdd); 3723 if (IsAdd && GetInitLCDecl(CE->getArg(1)) == LCDecl) 3724 return SetStep(CE->getArg(0), false); 3725 } 3726 } 3727 if (Dependent() || SemaRef.CurContext->isDependentContext()) 3728 return false; 3729 SemaRef.Diag(RHS->getLocStart(), diag::err_omp_loop_not_canonical_incr) 3730 << RHS->getSourceRange() << LCDecl; 3731 return true; 3732 } 3733 3734 bool OpenMPIterationSpaceChecker::CheckInc(Expr *S) { 3735 // Check incr-expr for canonical loop form and return true if it 3736 // does not conform. 3737 // OpenMP [2.6] Canonical loop form. Test-expr may be one of the following: 3738 // ++var 3739 // var++ 3740 // --var 3741 // var-- 3742 // var += incr 3743 // var -= incr 3744 // var = var + incr 3745 // var = incr + var 3746 // var = var - incr 3747 // 3748 if (!S) { 3749 SemaRef.Diag(DefaultLoc, diag::err_omp_loop_not_canonical_incr) << LCDecl; 3750 return true; 3751 } 3752 if (auto *ExprTemp = dyn_cast<ExprWithCleanups>(S)) 3753 if (!ExprTemp->cleanupsHaveSideEffects()) 3754 S = ExprTemp->getSubExpr(); 3755 3756 IncrementSrcRange = S->getSourceRange(); 3757 S = S->IgnoreParens(); 3758 if (auto *UO = dyn_cast<UnaryOperator>(S)) { 3759 if (UO->isIncrementDecrementOp() && 3760 GetInitLCDecl(UO->getSubExpr()) == LCDecl) 3761 return SetStep(SemaRef 3762 .ActOnIntegerConstant(UO->getLocStart(), 3763 (UO->isDecrementOp() ? -1 : 1)) 3764 .get(), 3765 false); 3766 } else if (auto *BO = dyn_cast<BinaryOperator>(S)) { 3767 switch (BO->getOpcode()) { 3768 case BO_AddAssign: 3769 case BO_SubAssign: 3770 if (GetInitLCDecl(BO->getLHS()) == LCDecl) 3771 return SetStep(BO->getRHS(), BO->getOpcode() == BO_SubAssign); 3772 break; 3773 case BO_Assign: 3774 if (GetInitLCDecl(BO->getLHS()) == LCDecl) 3775 return CheckIncRHS(BO->getRHS()); 3776 break; 3777 default: 3778 break; 3779 } 3780 } else if (auto *CE = dyn_cast<CXXOperatorCallExpr>(S)) { 3781 switch (CE->getOperator()) { 3782 case OO_PlusPlus: 3783 case OO_MinusMinus: 3784 if (GetInitLCDecl(CE->getArg(0)) == LCDecl) 3785 return SetStep(SemaRef 3786 .ActOnIntegerConstant( 3787 CE->getLocStart(), 3788 ((CE->getOperator() == OO_MinusMinus) ? -1 : 1)) 3789 .get(), 3790 false); 3791 break; 3792 case OO_PlusEqual: 3793 case OO_MinusEqual: 3794 if (GetInitLCDecl(CE->getArg(0)) == LCDecl) 3795 return SetStep(CE->getArg(1), CE->getOperator() == OO_MinusEqual); 3796 break; 3797 case OO_Equal: 3798 if (GetInitLCDecl(CE->getArg(0)) == LCDecl) 3799 return CheckIncRHS(CE->getArg(1)); 3800 break; 3801 default: 3802 break; 3803 } 3804 } 3805 if (Dependent() || SemaRef.CurContext->isDependentContext()) 3806 return false; 3807 SemaRef.Diag(S->getLocStart(), diag::err_omp_loop_not_canonical_incr) 3808 << S->getSourceRange() << LCDecl; 3809 return true; 3810 } 3811 3812 static ExprResult 3813 tryBuildCapture(Sema &SemaRef, Expr *Capture, 3814 llvm::MapVector<Expr *, DeclRefExpr *> &Captures) { 3815 if (SemaRef.CurContext->isDependentContext()) 3816 return ExprResult(Capture); 3817 if (Capture->isEvaluatable(SemaRef.Context, Expr::SE_AllowSideEffects)) 3818 return SemaRef.PerformImplicitConversion( 3819 Capture->IgnoreImpCasts(), Capture->getType(), Sema::AA_Converting, 3820 /*AllowExplicit=*/true); 3821 auto I = Captures.find(Capture); 3822 if (I != Captures.end()) 3823 return buildCapture(SemaRef, Capture, I->second); 3824 DeclRefExpr *Ref = nullptr; 3825 ExprResult Res = buildCapture(SemaRef, Capture, Ref); 3826 Captures[Capture] = Ref; 3827 return Res; 3828 } 3829 3830 /// \brief Build the expression to calculate the number of iterations. 3831 Expr *OpenMPIterationSpaceChecker::BuildNumIterations( 3832 Scope *S, const bool LimitedType, 3833 llvm::MapVector<Expr *, DeclRefExpr *> &Captures) const { 3834 ExprResult Diff; 3835 auto VarType = LCDecl->getType().getNonReferenceType(); 3836 if (VarType->isIntegerType() || VarType->isPointerType() || 3837 SemaRef.getLangOpts().CPlusPlus) { 3838 // Upper - Lower 3839 auto *UBExpr = TestIsLessOp ? UB : LB; 3840 auto *LBExpr = TestIsLessOp ? LB : UB; 3841 Expr *Upper = tryBuildCapture(SemaRef, UBExpr, Captures).get(); 3842 Expr *Lower = tryBuildCapture(SemaRef, LBExpr, Captures).get(); 3843 if (!Upper || !Lower) 3844 return nullptr; 3845 3846 Diff = SemaRef.BuildBinOp(S, DefaultLoc, BO_Sub, Upper, Lower); 3847 3848 if (!Diff.isUsable() && VarType->getAsCXXRecordDecl()) { 3849 // BuildBinOp already emitted error, this one is to point user to upper 3850 // and lower bound, and to tell what is passed to 'operator-'. 3851 SemaRef.Diag(Upper->getLocStart(), diag::err_omp_loop_diff_cxx) 3852 << Upper->getSourceRange() << Lower->getSourceRange(); 3853 return nullptr; 3854 } 3855 } 3856 3857 if (!Diff.isUsable()) 3858 return nullptr; 3859 3860 // Upper - Lower [- 1] 3861 if (TestIsStrictOp) 3862 Diff = SemaRef.BuildBinOp( 3863 S, DefaultLoc, BO_Sub, Diff.get(), 3864 SemaRef.ActOnIntegerConstant(SourceLocation(), 1).get()); 3865 if (!Diff.isUsable()) 3866 return nullptr; 3867 3868 // Upper - Lower [- 1] + Step 3869 auto NewStep = tryBuildCapture(SemaRef, Step, Captures); 3870 if (!NewStep.isUsable()) 3871 return nullptr; 3872 Diff = SemaRef.BuildBinOp(S, DefaultLoc, BO_Add, Diff.get(), NewStep.get()); 3873 if (!Diff.isUsable()) 3874 return nullptr; 3875 3876 // Parentheses (for dumping/debugging purposes only). 3877 Diff = SemaRef.ActOnParenExpr(DefaultLoc, DefaultLoc, Diff.get()); 3878 if (!Diff.isUsable()) 3879 return nullptr; 3880 3881 // (Upper - Lower [- 1] + Step) / Step 3882 Diff = SemaRef.BuildBinOp(S, DefaultLoc, BO_Div, Diff.get(), NewStep.get()); 3883 if (!Diff.isUsable()) 3884 return nullptr; 3885 3886 // OpenMP runtime requires 32-bit or 64-bit loop variables. 3887 QualType Type = Diff.get()->getType(); 3888 auto &C = SemaRef.Context; 3889 bool UseVarType = VarType->hasIntegerRepresentation() && 3890 C.getTypeSize(Type) > C.getTypeSize(VarType); 3891 if (!Type->isIntegerType() || UseVarType) { 3892 unsigned NewSize = 3893 UseVarType ? C.getTypeSize(VarType) : C.getTypeSize(Type); 3894 bool IsSigned = UseVarType ? VarType->hasSignedIntegerRepresentation() 3895 : Type->hasSignedIntegerRepresentation(); 3896 Type = C.getIntTypeForBitwidth(NewSize, IsSigned); 3897 if (!SemaRef.Context.hasSameType(Diff.get()->getType(), Type)) { 3898 Diff = SemaRef.PerformImplicitConversion( 3899 Diff.get(), Type, Sema::AA_Converting, /*AllowExplicit=*/true); 3900 if (!Diff.isUsable()) 3901 return nullptr; 3902 } 3903 } 3904 if (LimitedType) { 3905 unsigned NewSize = (C.getTypeSize(Type) > 32) ? 64 : 32; 3906 if (NewSize != C.getTypeSize(Type)) { 3907 if (NewSize < C.getTypeSize(Type)) { 3908 assert(NewSize == 64 && "incorrect loop var size"); 3909 SemaRef.Diag(DefaultLoc, diag::warn_omp_loop_64_bit_var) 3910 << InitSrcRange << ConditionSrcRange; 3911 } 3912 QualType NewType = C.getIntTypeForBitwidth( 3913 NewSize, Type->hasSignedIntegerRepresentation() || 3914 C.getTypeSize(Type) < NewSize); 3915 if (!SemaRef.Context.hasSameType(Diff.get()->getType(), NewType)) { 3916 Diff = SemaRef.PerformImplicitConversion(Diff.get(), NewType, 3917 Sema::AA_Converting, true); 3918 if (!Diff.isUsable()) 3919 return nullptr; 3920 } 3921 } 3922 } 3923 3924 return Diff.get(); 3925 } 3926 3927 Expr *OpenMPIterationSpaceChecker::BuildPreCond( 3928 Scope *S, Expr *Cond, 3929 llvm::MapVector<Expr *, DeclRefExpr *> &Captures) const { 3930 // Try to build LB <op> UB, where <op> is <, >, <=, or >=. 3931 bool Suppress = SemaRef.getDiagnostics().getSuppressAllDiagnostics(); 3932 SemaRef.getDiagnostics().setSuppressAllDiagnostics(/*Val=*/true); 3933 3934 auto NewLB = tryBuildCapture(SemaRef, LB, Captures); 3935 auto NewUB = tryBuildCapture(SemaRef, UB, Captures); 3936 if (!NewLB.isUsable() || !NewUB.isUsable()) 3937 return nullptr; 3938 3939 auto CondExpr = SemaRef.BuildBinOp( 3940 S, DefaultLoc, TestIsLessOp ? (TestIsStrictOp ? BO_LT : BO_LE) 3941 : (TestIsStrictOp ? BO_GT : BO_GE), 3942 NewLB.get(), NewUB.get()); 3943 if (CondExpr.isUsable()) { 3944 if (!SemaRef.Context.hasSameUnqualifiedType(CondExpr.get()->getType(), 3945 SemaRef.Context.BoolTy)) 3946 CondExpr = SemaRef.PerformImplicitConversion( 3947 CondExpr.get(), SemaRef.Context.BoolTy, /*Action=*/Sema::AA_Casting, 3948 /*AllowExplicit=*/true); 3949 } 3950 SemaRef.getDiagnostics().setSuppressAllDiagnostics(Suppress); 3951 // Otherwise use original loop conditon and evaluate it in runtime. 3952 return CondExpr.isUsable() ? CondExpr.get() : Cond; 3953 } 3954 3955 /// \brief Build reference expression to the counter be used for codegen. 3956 DeclRefExpr *OpenMPIterationSpaceChecker::BuildCounterVar( 3957 llvm::MapVector<Expr *, DeclRefExpr *> &Captures, DSAStackTy &DSA) const { 3958 auto *VD = dyn_cast<VarDecl>(LCDecl); 3959 if (!VD) { 3960 VD = SemaRef.IsOpenMPCapturedDecl(LCDecl); 3961 auto *Ref = buildDeclRefExpr( 3962 SemaRef, VD, VD->getType().getNonReferenceType(), DefaultLoc); 3963 DSAStackTy::DSAVarData Data = DSA.getTopDSA(LCDecl, /*FromParent=*/false); 3964 // If the loop control decl is explicitly marked as private, do not mark it 3965 // as captured again. 3966 if (!isOpenMPPrivate(Data.CKind) || !Data.RefExpr) 3967 Captures.insert(std::make_pair(LCRef, Ref)); 3968 return Ref; 3969 } 3970 return buildDeclRefExpr(SemaRef, VD, VD->getType().getNonReferenceType(), 3971 DefaultLoc); 3972 } 3973 3974 Expr *OpenMPIterationSpaceChecker::BuildPrivateCounterVar() const { 3975 if (LCDecl && !LCDecl->isInvalidDecl()) { 3976 auto Type = LCDecl->getType().getNonReferenceType(); 3977 auto *PrivateVar = 3978 buildVarDecl(SemaRef, DefaultLoc, Type, LCDecl->getName(), 3979 LCDecl->hasAttrs() ? &LCDecl->getAttrs() : nullptr); 3980 if (PrivateVar->isInvalidDecl()) 3981 return nullptr; 3982 return buildDeclRefExpr(SemaRef, PrivateVar, Type, DefaultLoc); 3983 } 3984 return nullptr; 3985 } 3986 3987 /// \brief Build initialization of the counter to be used for codegen. 3988 Expr *OpenMPIterationSpaceChecker::BuildCounterInit() const { return LB; } 3989 3990 /// \brief Build step of the counter be used for codegen. 3991 Expr *OpenMPIterationSpaceChecker::BuildCounterStep() const { return Step; } 3992 3993 /// \brief Iteration space of a single for loop. 3994 struct LoopIterationSpace final { 3995 /// \brief Condition of the loop. 3996 Expr *PreCond = nullptr; 3997 /// \brief This expression calculates the number of iterations in the loop. 3998 /// It is always possible to calculate it before starting the loop. 3999 Expr *NumIterations = nullptr; 4000 /// \brief The loop counter variable. 4001 Expr *CounterVar = nullptr; 4002 /// \brief Private loop counter variable. 4003 Expr *PrivateCounterVar = nullptr; 4004 /// \brief This is initializer for the initial value of #CounterVar. 4005 Expr *CounterInit = nullptr; 4006 /// \brief This is step for the #CounterVar used to generate its update: 4007 /// #CounterVar = #CounterInit + #CounterStep * CurrentIteration. 4008 Expr *CounterStep = nullptr; 4009 /// \brief Should step be subtracted? 4010 bool Subtract = false; 4011 /// \brief Source range of the loop init. 4012 SourceRange InitSrcRange; 4013 /// \brief Source range of the loop condition. 4014 SourceRange CondSrcRange; 4015 /// \brief Source range of the loop increment. 4016 SourceRange IncSrcRange; 4017 }; 4018 4019 } // namespace 4020 4021 void Sema::ActOnOpenMPLoopInitialization(SourceLocation ForLoc, Stmt *Init) { 4022 assert(getLangOpts().OpenMP && "OpenMP is not active."); 4023 assert(Init && "Expected loop in canonical form."); 4024 unsigned AssociatedLoops = DSAStack->getAssociatedLoops(); 4025 if (AssociatedLoops > 0 && 4026 isOpenMPLoopDirective(DSAStack->getCurrentDirective())) { 4027 OpenMPIterationSpaceChecker ISC(*this, ForLoc); 4028 if (!ISC.CheckInit(Init, /*EmitDiags=*/false)) { 4029 if (auto *D = ISC.GetLoopDecl()) { 4030 auto *VD = dyn_cast<VarDecl>(D); 4031 if (!VD) { 4032 if (auto *Private = IsOpenMPCapturedDecl(D)) 4033 VD = Private; 4034 else { 4035 auto *Ref = buildCapture(*this, D, ISC.GetLoopDeclRefExpr(), 4036 /*WithInit=*/false); 4037 VD = cast<VarDecl>(Ref->getDecl()); 4038 } 4039 } 4040 DSAStack->addLoopControlVariable(D, VD); 4041 } 4042 } 4043 DSAStack->setAssociatedLoops(AssociatedLoops - 1); 4044 } 4045 } 4046 4047 /// \brief Called on a for stmt to check and extract its iteration space 4048 /// for further processing (such as collapsing). 4049 static bool CheckOpenMPIterationSpace( 4050 OpenMPDirectiveKind DKind, Stmt *S, Sema &SemaRef, DSAStackTy &DSA, 4051 unsigned CurrentNestedLoopCount, unsigned NestedLoopCount, 4052 Expr *CollapseLoopCountExpr, Expr *OrderedLoopCountExpr, 4053 llvm::DenseMap<ValueDecl *, Expr *> &VarsWithImplicitDSA, 4054 LoopIterationSpace &ResultIterSpace, 4055 llvm::MapVector<Expr *, DeclRefExpr *> &Captures) { 4056 // OpenMP [2.6, Canonical Loop Form] 4057 // for (init-expr; test-expr; incr-expr) structured-block 4058 auto *For = dyn_cast_or_null<ForStmt>(S); 4059 if (!For) { 4060 SemaRef.Diag(S->getLocStart(), diag::err_omp_not_for) 4061 << (CollapseLoopCountExpr != nullptr || OrderedLoopCountExpr != nullptr) 4062 << getOpenMPDirectiveName(DKind) << NestedLoopCount 4063 << (CurrentNestedLoopCount > 0) << CurrentNestedLoopCount; 4064 if (NestedLoopCount > 1) { 4065 if (CollapseLoopCountExpr && OrderedLoopCountExpr) 4066 SemaRef.Diag(DSA.getConstructLoc(), 4067 diag::note_omp_collapse_ordered_expr) 4068 << 2 << CollapseLoopCountExpr->getSourceRange() 4069 << OrderedLoopCountExpr->getSourceRange(); 4070 else if (CollapseLoopCountExpr) 4071 SemaRef.Diag(CollapseLoopCountExpr->getExprLoc(), 4072 diag::note_omp_collapse_ordered_expr) 4073 << 0 << CollapseLoopCountExpr->getSourceRange(); 4074 else 4075 SemaRef.Diag(OrderedLoopCountExpr->getExprLoc(), 4076 diag::note_omp_collapse_ordered_expr) 4077 << 1 << OrderedLoopCountExpr->getSourceRange(); 4078 } 4079 return true; 4080 } 4081 assert(For->getBody()); 4082 4083 OpenMPIterationSpaceChecker ISC(SemaRef, For->getForLoc()); 4084 4085 // Check init. 4086 auto Init = For->getInit(); 4087 if (ISC.CheckInit(Init)) 4088 return true; 4089 4090 bool HasErrors = false; 4091 4092 // Check loop variable's type. 4093 if (auto *LCDecl = ISC.GetLoopDecl()) { 4094 auto *LoopDeclRefExpr = ISC.GetLoopDeclRefExpr(); 4095 4096 // OpenMP [2.6, Canonical Loop Form] 4097 // Var is one of the following: 4098 // A variable of signed or unsigned integer type. 4099 // For C++, a variable of a random access iterator type. 4100 // For C, a variable of a pointer type. 4101 auto VarType = LCDecl->getType().getNonReferenceType(); 4102 if (!VarType->isDependentType() && !VarType->isIntegerType() && 4103 !VarType->isPointerType() && 4104 !(SemaRef.getLangOpts().CPlusPlus && VarType->isOverloadableType())) { 4105 SemaRef.Diag(Init->getLocStart(), diag::err_omp_loop_variable_type) 4106 << SemaRef.getLangOpts().CPlusPlus; 4107 HasErrors = true; 4108 } 4109 4110 // OpenMP, 2.14.1.1 Data-sharing Attribute Rules for Variables Referenced in 4111 // a Construct 4112 // The loop iteration variable(s) in the associated for-loop(s) of a for or 4113 // parallel for construct is (are) private. 4114 // The loop iteration variable in the associated for-loop of a simd 4115 // construct with just one associated for-loop is linear with a 4116 // constant-linear-step that is the increment of the associated for-loop. 4117 // Exclude loop var from the list of variables with implicitly defined data 4118 // sharing attributes. 4119 VarsWithImplicitDSA.erase(LCDecl); 4120 4121 // OpenMP [2.14.1.1, Data-sharing Attribute Rules for Variables Referenced 4122 // in a Construct, C/C++]. 4123 // The loop iteration variable in the associated for-loop of a simd 4124 // construct with just one associated for-loop may be listed in a linear 4125 // clause with a constant-linear-step that is the increment of the 4126 // associated for-loop. 4127 // The loop iteration variable(s) in the associated for-loop(s) of a for or 4128 // parallel for construct may be listed in a private or lastprivate clause. 4129 DSAStackTy::DSAVarData DVar = DSA.getTopDSA(LCDecl, false); 4130 // If LoopVarRefExpr is nullptr it means the corresponding loop variable is 4131 // declared in the loop and it is predetermined as a private. 4132 auto PredeterminedCKind = 4133 isOpenMPSimdDirective(DKind) 4134 ? ((NestedLoopCount == 1) ? OMPC_linear : OMPC_lastprivate) 4135 : OMPC_private; 4136 if (((isOpenMPSimdDirective(DKind) && DVar.CKind != OMPC_unknown && 4137 DVar.CKind != PredeterminedCKind) || 4138 ((isOpenMPWorksharingDirective(DKind) || DKind == OMPD_taskloop || 4139 isOpenMPDistributeDirective(DKind)) && 4140 !isOpenMPSimdDirective(DKind) && DVar.CKind != OMPC_unknown && 4141 DVar.CKind != OMPC_private && DVar.CKind != OMPC_lastprivate)) && 4142 (DVar.CKind != OMPC_private || DVar.RefExpr != nullptr)) { 4143 SemaRef.Diag(Init->getLocStart(), diag::err_omp_loop_var_dsa) 4144 << getOpenMPClauseName(DVar.CKind) << getOpenMPDirectiveName(DKind) 4145 << getOpenMPClauseName(PredeterminedCKind); 4146 if (DVar.RefExpr == nullptr) 4147 DVar.CKind = PredeterminedCKind; 4148 ReportOriginalDSA(SemaRef, &DSA, LCDecl, DVar, /*IsLoopIterVar=*/true); 4149 HasErrors = true; 4150 } else if (LoopDeclRefExpr != nullptr) { 4151 // Make the loop iteration variable private (for worksharing constructs), 4152 // linear (for simd directives with the only one associated loop) or 4153 // lastprivate (for simd directives with several collapsed or ordered 4154 // loops). 4155 if (DVar.CKind == OMPC_unknown) 4156 DVar = DSA.hasDSA(LCDecl, isOpenMPPrivate, 4157 [](OpenMPDirectiveKind) -> bool { return true; }, 4158 /*FromParent=*/false); 4159 DSA.addDSA(LCDecl, LoopDeclRefExpr, PredeterminedCKind); 4160 } 4161 4162 assert(isOpenMPLoopDirective(DKind) && "DSA for non-loop vars"); 4163 4164 // Check test-expr. 4165 HasErrors |= ISC.CheckCond(For->getCond()); 4166 4167 // Check incr-expr. 4168 HasErrors |= ISC.CheckInc(For->getInc()); 4169 } 4170 4171 if (ISC.Dependent() || SemaRef.CurContext->isDependentContext() || HasErrors) 4172 return HasErrors; 4173 4174 // Build the loop's iteration space representation. 4175 ResultIterSpace.PreCond = 4176 ISC.BuildPreCond(DSA.getCurScope(), For->getCond(), Captures); 4177 ResultIterSpace.NumIterations = ISC.BuildNumIterations( 4178 DSA.getCurScope(), 4179 (isOpenMPWorksharingDirective(DKind) || 4180 isOpenMPTaskLoopDirective(DKind) || isOpenMPDistributeDirective(DKind)), 4181 Captures); 4182 ResultIterSpace.CounterVar = ISC.BuildCounterVar(Captures, DSA); 4183 ResultIterSpace.PrivateCounterVar = ISC.BuildPrivateCounterVar(); 4184 ResultIterSpace.CounterInit = ISC.BuildCounterInit(); 4185 ResultIterSpace.CounterStep = ISC.BuildCounterStep(); 4186 ResultIterSpace.InitSrcRange = ISC.GetInitSrcRange(); 4187 ResultIterSpace.CondSrcRange = ISC.GetConditionSrcRange(); 4188 ResultIterSpace.IncSrcRange = ISC.GetIncrementSrcRange(); 4189 ResultIterSpace.Subtract = ISC.ShouldSubtractStep(); 4190 4191 HasErrors |= (ResultIterSpace.PreCond == nullptr || 4192 ResultIterSpace.NumIterations == nullptr || 4193 ResultIterSpace.CounterVar == nullptr || 4194 ResultIterSpace.PrivateCounterVar == nullptr || 4195 ResultIterSpace.CounterInit == nullptr || 4196 ResultIterSpace.CounterStep == nullptr); 4197 4198 return HasErrors; 4199 } 4200 4201 /// \brief Build 'VarRef = Start. 4202 static ExprResult 4203 BuildCounterInit(Sema &SemaRef, Scope *S, SourceLocation Loc, ExprResult VarRef, 4204 ExprResult Start, 4205 llvm::MapVector<Expr *, DeclRefExpr *> &Captures) { 4206 // Build 'VarRef = Start. 4207 auto NewStart = tryBuildCapture(SemaRef, Start.get(), Captures); 4208 if (!NewStart.isUsable()) 4209 return ExprError(); 4210 if (!SemaRef.Context.hasSameType(NewStart.get()->getType(), 4211 VarRef.get()->getType())) { 4212 NewStart = SemaRef.PerformImplicitConversion( 4213 NewStart.get(), VarRef.get()->getType(), Sema::AA_Converting, 4214 /*AllowExplicit=*/true); 4215 if (!NewStart.isUsable()) 4216 return ExprError(); 4217 } 4218 4219 auto Init = 4220 SemaRef.BuildBinOp(S, Loc, BO_Assign, VarRef.get(), NewStart.get()); 4221 return Init; 4222 } 4223 4224 /// \brief Build 'VarRef = Start + Iter * Step'. 4225 static ExprResult 4226 BuildCounterUpdate(Sema &SemaRef, Scope *S, SourceLocation Loc, 4227 ExprResult VarRef, ExprResult Start, ExprResult Iter, 4228 ExprResult Step, bool Subtract, 4229 llvm::MapVector<Expr *, DeclRefExpr *> *Captures = nullptr) { 4230 // Add parentheses (for debugging purposes only). 4231 Iter = SemaRef.ActOnParenExpr(Loc, Loc, Iter.get()); 4232 if (!VarRef.isUsable() || !Start.isUsable() || !Iter.isUsable() || 4233 !Step.isUsable()) 4234 return ExprError(); 4235 4236 ExprResult NewStep = Step; 4237 if (Captures) 4238 NewStep = tryBuildCapture(SemaRef, Step.get(), *Captures); 4239 if (NewStep.isInvalid()) 4240 return ExprError(); 4241 ExprResult Update = 4242 SemaRef.BuildBinOp(S, Loc, BO_Mul, Iter.get(), NewStep.get()); 4243 if (!Update.isUsable()) 4244 return ExprError(); 4245 4246 // Try to build 'VarRef = Start, VarRef (+|-)= Iter * Step' or 4247 // 'VarRef = Start (+|-) Iter * Step'. 4248 ExprResult NewStart = Start; 4249 if (Captures) 4250 NewStart = tryBuildCapture(SemaRef, Start.get(), *Captures); 4251 if (NewStart.isInvalid()) 4252 return ExprError(); 4253 4254 // First attempt: try to build 'VarRef = Start, VarRef += Iter * Step'. 4255 ExprResult SavedUpdate = Update; 4256 ExprResult UpdateVal; 4257 if (VarRef.get()->getType()->isOverloadableType() || 4258 NewStart.get()->getType()->isOverloadableType() || 4259 Update.get()->getType()->isOverloadableType()) { 4260 bool Suppress = SemaRef.getDiagnostics().getSuppressAllDiagnostics(); 4261 SemaRef.getDiagnostics().setSuppressAllDiagnostics(/*Val=*/true); 4262 Update = 4263 SemaRef.BuildBinOp(S, Loc, BO_Assign, VarRef.get(), NewStart.get()); 4264 if (Update.isUsable()) { 4265 UpdateVal = 4266 SemaRef.BuildBinOp(S, Loc, Subtract ? BO_SubAssign : BO_AddAssign, 4267 VarRef.get(), SavedUpdate.get()); 4268 if (UpdateVal.isUsable()) { 4269 Update = SemaRef.CreateBuiltinBinOp(Loc, BO_Comma, Update.get(), 4270 UpdateVal.get()); 4271 } 4272 } 4273 SemaRef.getDiagnostics().setSuppressAllDiagnostics(Suppress); 4274 } 4275 4276 // Second attempt: try to build 'VarRef = Start (+|-) Iter * Step'. 4277 if (!Update.isUsable() || !UpdateVal.isUsable()) { 4278 Update = SemaRef.BuildBinOp(S, Loc, Subtract ? BO_Sub : BO_Add, 4279 NewStart.get(), SavedUpdate.get()); 4280 if (!Update.isUsable()) 4281 return ExprError(); 4282 4283 if (!SemaRef.Context.hasSameType(Update.get()->getType(), 4284 VarRef.get()->getType())) { 4285 Update = SemaRef.PerformImplicitConversion( 4286 Update.get(), VarRef.get()->getType(), Sema::AA_Converting, true); 4287 if (!Update.isUsable()) 4288 return ExprError(); 4289 } 4290 4291 Update = SemaRef.BuildBinOp(S, Loc, BO_Assign, VarRef.get(), Update.get()); 4292 } 4293 return Update; 4294 } 4295 4296 /// \brief Convert integer expression \a E to make it have at least \a Bits 4297 /// bits. 4298 static ExprResult WidenIterationCount(unsigned Bits, Expr *E, Sema &SemaRef) { 4299 if (E == nullptr) 4300 return ExprError(); 4301 auto &C = SemaRef.Context; 4302 QualType OldType = E->getType(); 4303 unsigned HasBits = C.getTypeSize(OldType); 4304 if (HasBits >= Bits) 4305 return ExprResult(E); 4306 // OK to convert to signed, because new type has more bits than old. 4307 QualType NewType = C.getIntTypeForBitwidth(Bits, /* Signed */ true); 4308 return SemaRef.PerformImplicitConversion(E, NewType, Sema::AA_Converting, 4309 true); 4310 } 4311 4312 /// \brief Check if the given expression \a E is a constant integer that fits 4313 /// into \a Bits bits. 4314 static bool FitsInto(unsigned Bits, bool Signed, Expr *E, Sema &SemaRef) { 4315 if (E == nullptr) 4316 return false; 4317 llvm::APSInt Result; 4318 if (E->isIntegerConstantExpr(Result, SemaRef.Context)) 4319 return Signed ? Result.isSignedIntN(Bits) : Result.isIntN(Bits); 4320 return false; 4321 } 4322 4323 /// Build preinits statement for the given declarations. 4324 static Stmt *buildPreInits(ASTContext &Context, 4325 SmallVectorImpl<Decl *> &PreInits) { 4326 if (!PreInits.empty()) { 4327 return new (Context) DeclStmt( 4328 DeclGroupRef::Create(Context, PreInits.begin(), PreInits.size()), 4329 SourceLocation(), SourceLocation()); 4330 } 4331 return nullptr; 4332 } 4333 4334 /// Build preinits statement for the given declarations. 4335 static Stmt *buildPreInits(ASTContext &Context, 4336 llvm::MapVector<Expr *, DeclRefExpr *> &Captures) { 4337 if (!Captures.empty()) { 4338 SmallVector<Decl *, 16> PreInits; 4339 for (auto &Pair : Captures) 4340 PreInits.push_back(Pair.second->getDecl()); 4341 return buildPreInits(Context, PreInits); 4342 } 4343 return nullptr; 4344 } 4345 4346 /// Build postupdate expression for the given list of postupdates expressions. 4347 static Expr *buildPostUpdate(Sema &S, ArrayRef<Expr *> PostUpdates) { 4348 Expr *PostUpdate = nullptr; 4349 if (!PostUpdates.empty()) { 4350 for (auto *E : PostUpdates) { 4351 Expr *ConvE = S.BuildCStyleCastExpr( 4352 E->getExprLoc(), 4353 S.Context.getTrivialTypeSourceInfo(S.Context.VoidTy), 4354 E->getExprLoc(), E) 4355 .get(); 4356 PostUpdate = PostUpdate 4357 ? S.CreateBuiltinBinOp(ConvE->getExprLoc(), BO_Comma, 4358 PostUpdate, ConvE) 4359 .get() 4360 : ConvE; 4361 } 4362 } 4363 return PostUpdate; 4364 } 4365 4366 /// \brief Called on a for stmt to check itself and nested loops (if any). 4367 /// \return Returns 0 if one of the collapsed stmts is not canonical for loop, 4368 /// number of collapsed loops otherwise. 4369 static unsigned 4370 CheckOpenMPLoop(OpenMPDirectiveKind DKind, Expr *CollapseLoopCountExpr, 4371 Expr *OrderedLoopCountExpr, Stmt *AStmt, Sema &SemaRef, 4372 DSAStackTy &DSA, 4373 llvm::DenseMap<ValueDecl *, Expr *> &VarsWithImplicitDSA, 4374 OMPLoopDirective::HelperExprs &Built) { 4375 unsigned NestedLoopCount = 1; 4376 if (CollapseLoopCountExpr) { 4377 // Found 'collapse' clause - calculate collapse number. 4378 llvm::APSInt Result; 4379 if (CollapseLoopCountExpr->EvaluateAsInt(Result, SemaRef.getASTContext())) 4380 NestedLoopCount = Result.getLimitedValue(); 4381 } 4382 if (OrderedLoopCountExpr) { 4383 // Found 'ordered' clause - calculate collapse number. 4384 llvm::APSInt Result; 4385 if (OrderedLoopCountExpr->EvaluateAsInt(Result, SemaRef.getASTContext())) { 4386 if (Result.getLimitedValue() < NestedLoopCount) { 4387 SemaRef.Diag(OrderedLoopCountExpr->getExprLoc(), 4388 diag::err_omp_wrong_ordered_loop_count) 4389 << OrderedLoopCountExpr->getSourceRange(); 4390 SemaRef.Diag(CollapseLoopCountExpr->getExprLoc(), 4391 diag::note_collapse_loop_count) 4392 << CollapseLoopCountExpr->getSourceRange(); 4393 } 4394 NestedLoopCount = Result.getLimitedValue(); 4395 } 4396 } 4397 // This is helper routine for loop directives (e.g., 'for', 'simd', 4398 // 'for simd', etc.). 4399 llvm::MapVector<Expr *, DeclRefExpr *> Captures; 4400 SmallVector<LoopIterationSpace, 4> IterSpaces; 4401 IterSpaces.resize(NestedLoopCount); 4402 Stmt *CurStmt = AStmt->IgnoreContainers(/* IgnoreCaptured */ true); 4403 for (unsigned Cnt = 0; Cnt < NestedLoopCount; ++Cnt) { 4404 if (CheckOpenMPIterationSpace(DKind, CurStmt, SemaRef, DSA, Cnt, 4405 NestedLoopCount, CollapseLoopCountExpr, 4406 OrderedLoopCountExpr, VarsWithImplicitDSA, 4407 IterSpaces[Cnt], Captures)) 4408 return 0; 4409 // Move on to the next nested for loop, or to the loop body. 4410 // OpenMP [2.8.1, simd construct, Restrictions] 4411 // All loops associated with the construct must be perfectly nested; that 4412 // is, there must be no intervening code nor any OpenMP directive between 4413 // any two loops. 4414 CurStmt = cast<ForStmt>(CurStmt)->getBody()->IgnoreContainers(); 4415 } 4416 4417 Built.clear(/* size */ NestedLoopCount); 4418 4419 if (SemaRef.CurContext->isDependentContext()) 4420 return NestedLoopCount; 4421 4422 // An example of what is generated for the following code: 4423 // 4424 // #pragma omp simd collapse(2) ordered(2) 4425 // for (i = 0; i < NI; ++i) 4426 // for (k = 0; k < NK; ++k) 4427 // for (j = J0; j < NJ; j+=2) { 4428 // <loop body> 4429 // } 4430 // 4431 // We generate the code below. 4432 // Note: the loop body may be outlined in CodeGen. 4433 // Note: some counters may be C++ classes, operator- is used to find number of 4434 // iterations and operator+= to calculate counter value. 4435 // Note: decltype(NumIterations) must be integer type (in 'omp for', only i32 4436 // or i64 is currently supported). 4437 // 4438 // #define NumIterations (NI * ((NJ - J0 - 1 + 2) / 2)) 4439 // for (int[32|64]_t IV = 0; IV < NumIterations; ++IV ) { 4440 // .local.i = IV / ((NJ - J0 - 1 + 2) / 2); 4441 // .local.j = J0 + (IV % ((NJ - J0 - 1 + 2) / 2)) * 2; 4442 // // similar updates for vars in clauses (e.g. 'linear') 4443 // <loop body (using local i and j)> 4444 // } 4445 // i = NI; // assign final values of counters 4446 // j = NJ; 4447 // 4448 4449 // Last iteration number is (I1 * I2 * ... In) - 1, where I1, I2 ... In are 4450 // the iteration counts of the collapsed for loops. 4451 // Precondition tests if there is at least one iteration (all conditions are 4452 // true). 4453 auto PreCond = ExprResult(IterSpaces[0].PreCond); 4454 auto N0 = IterSpaces[0].NumIterations; 4455 ExprResult LastIteration32 = WidenIterationCount( 4456 32 /* Bits */, SemaRef 4457 .PerformImplicitConversion( 4458 N0->IgnoreImpCasts(), N0->getType(), 4459 Sema::AA_Converting, /*AllowExplicit=*/true) 4460 .get(), 4461 SemaRef); 4462 ExprResult LastIteration64 = WidenIterationCount( 4463 64 /* Bits */, SemaRef 4464 .PerformImplicitConversion( 4465 N0->IgnoreImpCasts(), N0->getType(), 4466 Sema::AA_Converting, /*AllowExplicit=*/true) 4467 .get(), 4468 SemaRef); 4469 4470 if (!LastIteration32.isUsable() || !LastIteration64.isUsable()) 4471 return NestedLoopCount; 4472 4473 auto &C = SemaRef.Context; 4474 bool AllCountsNeedLessThan32Bits = C.getTypeSize(N0->getType()) < 32; 4475 4476 Scope *CurScope = DSA.getCurScope(); 4477 for (unsigned Cnt = 1; Cnt < NestedLoopCount; ++Cnt) { 4478 if (PreCond.isUsable()) { 4479 PreCond = 4480 SemaRef.BuildBinOp(CurScope, PreCond.get()->getExprLoc(), BO_LAnd, 4481 PreCond.get(), IterSpaces[Cnt].PreCond); 4482 } 4483 auto N = IterSpaces[Cnt].NumIterations; 4484 SourceLocation Loc = N->getExprLoc(); 4485 AllCountsNeedLessThan32Bits &= C.getTypeSize(N->getType()) < 32; 4486 if (LastIteration32.isUsable()) 4487 LastIteration32 = SemaRef.BuildBinOp( 4488 CurScope, Loc, BO_Mul, LastIteration32.get(), 4489 SemaRef 4490 .PerformImplicitConversion(N->IgnoreImpCasts(), N->getType(), 4491 Sema::AA_Converting, 4492 /*AllowExplicit=*/true) 4493 .get()); 4494 if (LastIteration64.isUsable()) 4495 LastIteration64 = SemaRef.BuildBinOp( 4496 CurScope, Loc, BO_Mul, LastIteration64.get(), 4497 SemaRef 4498 .PerformImplicitConversion(N->IgnoreImpCasts(), N->getType(), 4499 Sema::AA_Converting, 4500 /*AllowExplicit=*/true) 4501 .get()); 4502 } 4503 4504 // Choose either the 32-bit or 64-bit version. 4505 ExprResult LastIteration = LastIteration64; 4506 if (LastIteration32.isUsable() && 4507 C.getTypeSize(LastIteration32.get()->getType()) == 32 && 4508 (AllCountsNeedLessThan32Bits || NestedLoopCount == 1 || 4509 FitsInto( 4510 32 /* Bits */, 4511 LastIteration32.get()->getType()->hasSignedIntegerRepresentation(), 4512 LastIteration64.get(), SemaRef))) 4513 LastIteration = LastIteration32; 4514 QualType VType = LastIteration.get()->getType(); 4515 QualType RealVType = VType; 4516 QualType StrideVType = VType; 4517 if (isOpenMPTaskLoopDirective(DKind)) { 4518 VType = 4519 SemaRef.Context.getIntTypeForBitwidth(/*DestWidth=*/64, /*Signed=*/0); 4520 StrideVType = 4521 SemaRef.Context.getIntTypeForBitwidth(/*DestWidth=*/64, /*Signed=*/1); 4522 } 4523 4524 if (!LastIteration.isUsable()) 4525 return 0; 4526 4527 // Save the number of iterations. 4528 ExprResult NumIterations = LastIteration; 4529 { 4530 LastIteration = SemaRef.BuildBinOp( 4531 CurScope, LastIteration.get()->getExprLoc(), BO_Sub, 4532 LastIteration.get(), 4533 SemaRef.ActOnIntegerConstant(SourceLocation(), 1).get()); 4534 if (!LastIteration.isUsable()) 4535 return 0; 4536 } 4537 4538 // Calculate the last iteration number beforehand instead of doing this on 4539 // each iteration. Do not do this if the number of iterations may be kfold-ed. 4540 llvm::APSInt Result; 4541 bool IsConstant = 4542 LastIteration.get()->isIntegerConstantExpr(Result, SemaRef.Context); 4543 ExprResult CalcLastIteration; 4544 if (!IsConstant) { 4545 ExprResult SaveRef = 4546 tryBuildCapture(SemaRef, LastIteration.get(), Captures); 4547 LastIteration = SaveRef; 4548 4549 // Prepare SaveRef + 1. 4550 NumIterations = SemaRef.BuildBinOp( 4551 CurScope, SaveRef.get()->getExprLoc(), BO_Add, SaveRef.get(), 4552 SemaRef.ActOnIntegerConstant(SourceLocation(), 1).get()); 4553 if (!NumIterations.isUsable()) 4554 return 0; 4555 } 4556 4557 SourceLocation InitLoc = IterSpaces[0].InitSrcRange.getBegin(); 4558 4559 // Build variables passed into runtime, necessary for worksharing directives. 4560 ExprResult LB, UB, IL, ST, EUB, CombLB, CombUB, PrevLB, PrevUB, CombEUB; 4561 if (isOpenMPWorksharingDirective(DKind) || isOpenMPTaskLoopDirective(DKind) || 4562 isOpenMPDistributeDirective(DKind)) { 4563 // Lower bound variable, initialized with zero. 4564 VarDecl *LBDecl = buildVarDecl(SemaRef, InitLoc, VType, ".omp.lb"); 4565 LB = buildDeclRefExpr(SemaRef, LBDecl, VType, InitLoc); 4566 SemaRef.AddInitializerToDecl(LBDecl, 4567 SemaRef.ActOnIntegerConstant(InitLoc, 0).get(), 4568 /*DirectInit*/ false); 4569 4570 // Upper bound variable, initialized with last iteration number. 4571 VarDecl *UBDecl = buildVarDecl(SemaRef, InitLoc, VType, ".omp.ub"); 4572 UB = buildDeclRefExpr(SemaRef, UBDecl, VType, InitLoc); 4573 SemaRef.AddInitializerToDecl(UBDecl, LastIteration.get(), 4574 /*DirectInit*/ false); 4575 4576 // A 32-bit variable-flag where runtime returns 1 for the last iteration. 4577 // This will be used to implement clause 'lastprivate'. 4578 QualType Int32Ty = SemaRef.Context.getIntTypeForBitwidth(32, true); 4579 VarDecl *ILDecl = buildVarDecl(SemaRef, InitLoc, Int32Ty, ".omp.is_last"); 4580 IL = buildDeclRefExpr(SemaRef, ILDecl, Int32Ty, InitLoc); 4581 SemaRef.AddInitializerToDecl(ILDecl, 4582 SemaRef.ActOnIntegerConstant(InitLoc, 0).get(), 4583 /*DirectInit*/ false); 4584 4585 // Stride variable returned by runtime (we initialize it to 1 by default). 4586 VarDecl *STDecl = 4587 buildVarDecl(SemaRef, InitLoc, StrideVType, ".omp.stride"); 4588 ST = buildDeclRefExpr(SemaRef, STDecl, StrideVType, InitLoc); 4589 SemaRef.AddInitializerToDecl(STDecl, 4590 SemaRef.ActOnIntegerConstant(InitLoc, 1).get(), 4591 /*DirectInit*/ false); 4592 4593 // Build expression: UB = min(UB, LastIteration) 4594 // It is necessary for CodeGen of directives with static scheduling. 4595 ExprResult IsUBGreater = SemaRef.BuildBinOp(CurScope, InitLoc, BO_GT, 4596 UB.get(), LastIteration.get()); 4597 ExprResult CondOp = SemaRef.ActOnConditionalOp( 4598 InitLoc, InitLoc, IsUBGreater.get(), LastIteration.get(), UB.get()); 4599 EUB = SemaRef.BuildBinOp(CurScope, InitLoc, BO_Assign, UB.get(), 4600 CondOp.get()); 4601 EUB = SemaRef.ActOnFinishFullExpr(EUB.get()); 4602 4603 // If we have a combined directive that combines 'distribute', 'for' or 4604 // 'simd' we need to be able to access the bounds of the schedule of the 4605 // enclosing region. E.g. in 'distribute parallel for' the bounds obtained 4606 // by scheduling 'distribute' have to be passed to the schedule of 'for'. 4607 if (isOpenMPLoopBoundSharingDirective(DKind)) { 4608 4609 // Lower bound variable, initialized with zero. 4610 VarDecl *CombLBDecl = 4611 buildVarDecl(SemaRef, InitLoc, VType, ".omp.comb.lb"); 4612 CombLB = buildDeclRefExpr(SemaRef, CombLBDecl, VType, InitLoc); 4613 SemaRef.AddInitializerToDecl( 4614 CombLBDecl, SemaRef.ActOnIntegerConstant(InitLoc, 0).get(), 4615 /*DirectInit*/ false); 4616 4617 // Upper bound variable, initialized with last iteration number. 4618 VarDecl *CombUBDecl = 4619 buildVarDecl(SemaRef, InitLoc, VType, ".omp.comb.ub"); 4620 CombUB = buildDeclRefExpr(SemaRef, CombUBDecl, VType, InitLoc); 4621 SemaRef.AddInitializerToDecl(CombUBDecl, LastIteration.get(), 4622 /*DirectInit*/ false); 4623 4624 ExprResult CombIsUBGreater = SemaRef.BuildBinOp( 4625 CurScope, InitLoc, BO_GT, CombUB.get(), LastIteration.get()); 4626 ExprResult CombCondOp = 4627 SemaRef.ActOnConditionalOp(InitLoc, InitLoc, CombIsUBGreater.get(), 4628 LastIteration.get(), CombUB.get()); 4629 CombEUB = SemaRef.BuildBinOp(CurScope, InitLoc, BO_Assign, CombUB.get(), 4630 CombCondOp.get()); 4631 CombEUB = SemaRef.ActOnFinishFullExpr(CombEUB.get()); 4632 4633 auto *CD = cast<CapturedStmt>(AStmt)->getCapturedDecl(); 4634 // We expect to have at least 2 more parameters than the 'parallel' 4635 // directive does - the lower and upper bounds of the previous schedule. 4636 assert(CD->getNumParams() >= 4 && 4637 "Unexpected number of parameters in loop combined directive"); 4638 4639 // Set the proper type for the bounds given what we learned from the 4640 // enclosed loops. 4641 auto *PrevLBDecl = CD->getParam(/*PrevLB=*/2); 4642 auto *PrevUBDecl = CD->getParam(/*PrevUB=*/3); 4643 4644 // Previous lower and upper bounds are obtained from the region 4645 // parameters. 4646 PrevLB = 4647 buildDeclRefExpr(SemaRef, PrevLBDecl, PrevLBDecl->getType(), InitLoc); 4648 PrevUB = 4649 buildDeclRefExpr(SemaRef, PrevUBDecl, PrevUBDecl->getType(), InitLoc); 4650 } 4651 } 4652 4653 // Build the iteration variable and its initialization before loop. 4654 ExprResult IV; 4655 ExprResult Init, CombInit; 4656 { 4657 VarDecl *IVDecl = buildVarDecl(SemaRef, InitLoc, RealVType, ".omp.iv"); 4658 IV = buildDeclRefExpr(SemaRef, IVDecl, RealVType, InitLoc); 4659 Expr *RHS = 4660 (isOpenMPWorksharingDirective(DKind) || 4661 isOpenMPTaskLoopDirective(DKind) || isOpenMPDistributeDirective(DKind)) 4662 ? LB.get() 4663 : SemaRef.ActOnIntegerConstant(SourceLocation(), 0).get(); 4664 Init = SemaRef.BuildBinOp(CurScope, InitLoc, BO_Assign, IV.get(), RHS); 4665 Init = SemaRef.ActOnFinishFullExpr(Init.get()); 4666 4667 if (isOpenMPLoopBoundSharingDirective(DKind)) { 4668 Expr *CombRHS = 4669 (isOpenMPWorksharingDirective(DKind) || 4670 isOpenMPTaskLoopDirective(DKind) || 4671 isOpenMPDistributeDirective(DKind)) 4672 ? CombLB.get() 4673 : SemaRef.ActOnIntegerConstant(SourceLocation(), 0).get(); 4674 CombInit = 4675 SemaRef.BuildBinOp(CurScope, InitLoc, BO_Assign, IV.get(), CombRHS); 4676 CombInit = SemaRef.ActOnFinishFullExpr(CombInit.get()); 4677 } 4678 } 4679 4680 // Loop condition (IV < NumIterations) or (IV <= UB) for worksharing loops. 4681 SourceLocation CondLoc; 4682 ExprResult Cond = 4683 (isOpenMPWorksharingDirective(DKind) || 4684 isOpenMPTaskLoopDirective(DKind) || isOpenMPDistributeDirective(DKind)) 4685 ? SemaRef.BuildBinOp(CurScope, CondLoc, BO_LE, IV.get(), UB.get()) 4686 : SemaRef.BuildBinOp(CurScope, CondLoc, BO_LT, IV.get(), 4687 NumIterations.get()); 4688 ExprResult CombCond; 4689 if (isOpenMPLoopBoundSharingDirective(DKind)) { 4690 CombCond = 4691 SemaRef.BuildBinOp(CurScope, CondLoc, BO_LE, IV.get(), CombUB.get()); 4692 } 4693 // Loop increment (IV = IV + 1) 4694 SourceLocation IncLoc; 4695 ExprResult Inc = 4696 SemaRef.BuildBinOp(CurScope, IncLoc, BO_Add, IV.get(), 4697 SemaRef.ActOnIntegerConstant(IncLoc, 1).get()); 4698 if (!Inc.isUsable()) 4699 return 0; 4700 Inc = SemaRef.BuildBinOp(CurScope, IncLoc, BO_Assign, IV.get(), Inc.get()); 4701 Inc = SemaRef.ActOnFinishFullExpr(Inc.get()); 4702 if (!Inc.isUsable()) 4703 return 0; 4704 4705 // Increments for worksharing loops (LB = LB + ST; UB = UB + ST). 4706 // Used for directives with static scheduling. 4707 // In combined construct, add combined version that use CombLB and CombUB 4708 // base variables for the update 4709 ExprResult NextLB, NextUB, CombNextLB, CombNextUB; 4710 if (isOpenMPWorksharingDirective(DKind) || isOpenMPTaskLoopDirective(DKind) || 4711 isOpenMPDistributeDirective(DKind)) { 4712 // LB + ST 4713 NextLB = SemaRef.BuildBinOp(CurScope, IncLoc, BO_Add, LB.get(), ST.get()); 4714 if (!NextLB.isUsable()) 4715 return 0; 4716 // LB = LB + ST 4717 NextLB = 4718 SemaRef.BuildBinOp(CurScope, IncLoc, BO_Assign, LB.get(), NextLB.get()); 4719 NextLB = SemaRef.ActOnFinishFullExpr(NextLB.get()); 4720 if (!NextLB.isUsable()) 4721 return 0; 4722 // UB + ST 4723 NextUB = SemaRef.BuildBinOp(CurScope, IncLoc, BO_Add, UB.get(), ST.get()); 4724 if (!NextUB.isUsable()) 4725 return 0; 4726 // UB = UB + ST 4727 NextUB = 4728 SemaRef.BuildBinOp(CurScope, IncLoc, BO_Assign, UB.get(), NextUB.get()); 4729 NextUB = SemaRef.ActOnFinishFullExpr(NextUB.get()); 4730 if (!NextUB.isUsable()) 4731 return 0; 4732 if (isOpenMPLoopBoundSharingDirective(DKind)) { 4733 CombNextLB = 4734 SemaRef.BuildBinOp(CurScope, IncLoc, BO_Add, CombLB.get(), ST.get()); 4735 if (!NextLB.isUsable()) 4736 return 0; 4737 // LB = LB + ST 4738 CombNextLB = SemaRef.BuildBinOp(CurScope, IncLoc, BO_Assign, CombLB.get(), 4739 CombNextLB.get()); 4740 CombNextLB = SemaRef.ActOnFinishFullExpr(CombNextLB.get()); 4741 if (!CombNextLB.isUsable()) 4742 return 0; 4743 // UB + ST 4744 CombNextUB = 4745 SemaRef.BuildBinOp(CurScope, IncLoc, BO_Add, CombUB.get(), ST.get()); 4746 if (!CombNextUB.isUsable()) 4747 return 0; 4748 // UB = UB + ST 4749 CombNextUB = SemaRef.BuildBinOp(CurScope, IncLoc, BO_Assign, CombUB.get(), 4750 CombNextUB.get()); 4751 CombNextUB = SemaRef.ActOnFinishFullExpr(CombNextUB.get()); 4752 if (!CombNextUB.isUsable()) 4753 return 0; 4754 } 4755 } 4756 4757 // Create increment expression for distribute loop when combined in a same 4758 // directive with for as IV = IV + ST; ensure upper bound expression based 4759 // on PrevUB instead of NumIterations - used to implement 'for' when found 4760 // in combination with 'distribute', like in 'distribute parallel for' 4761 SourceLocation DistIncLoc; 4762 ExprResult DistCond, DistInc, PrevEUB; 4763 if (isOpenMPLoopBoundSharingDirective(DKind)) { 4764 DistCond = SemaRef.BuildBinOp(CurScope, CondLoc, BO_LE, IV.get(), UB.get()); 4765 assert(DistCond.isUsable() && "distribute cond expr was not built"); 4766 4767 DistInc = 4768 SemaRef.BuildBinOp(CurScope, DistIncLoc, BO_Add, IV.get(), ST.get()); 4769 assert(DistInc.isUsable() && "distribute inc expr was not built"); 4770 DistInc = SemaRef.BuildBinOp(CurScope, DistIncLoc, BO_Assign, IV.get(), 4771 DistInc.get()); 4772 DistInc = SemaRef.ActOnFinishFullExpr(DistInc.get()); 4773 assert(DistInc.isUsable() && "distribute inc expr was not built"); 4774 4775 // Build expression: UB = min(UB, prevUB) for #for in composite or combined 4776 // construct 4777 SourceLocation DistEUBLoc; 4778 ExprResult IsUBGreater = 4779 SemaRef.BuildBinOp(CurScope, DistEUBLoc, BO_GT, UB.get(), PrevUB.get()); 4780 ExprResult CondOp = SemaRef.ActOnConditionalOp( 4781 DistEUBLoc, DistEUBLoc, IsUBGreater.get(), PrevUB.get(), UB.get()); 4782 PrevEUB = SemaRef.BuildBinOp(CurScope, DistIncLoc, BO_Assign, UB.get(), 4783 CondOp.get()); 4784 PrevEUB = SemaRef.ActOnFinishFullExpr(PrevEUB.get()); 4785 } 4786 4787 // Build updates and final values of the loop counters. 4788 bool HasErrors = false; 4789 Built.Counters.resize(NestedLoopCount); 4790 Built.Inits.resize(NestedLoopCount); 4791 Built.Updates.resize(NestedLoopCount); 4792 Built.Finals.resize(NestedLoopCount); 4793 SmallVector<Expr *, 4> LoopMultipliers; 4794 { 4795 ExprResult Div; 4796 // Go from inner nested loop to outer. 4797 for (int Cnt = NestedLoopCount - 1; Cnt >= 0; --Cnt) { 4798 LoopIterationSpace &IS = IterSpaces[Cnt]; 4799 SourceLocation UpdLoc = IS.IncSrcRange.getBegin(); 4800 // Build: Iter = (IV / Div) % IS.NumIters 4801 // where Div is product of previous iterations' IS.NumIters. 4802 ExprResult Iter; 4803 if (Div.isUsable()) { 4804 Iter = 4805 SemaRef.BuildBinOp(CurScope, UpdLoc, BO_Div, IV.get(), Div.get()); 4806 } else { 4807 Iter = IV; 4808 assert((Cnt == (int)NestedLoopCount - 1) && 4809 "unusable div expected on first iteration only"); 4810 } 4811 4812 if (Cnt != 0 && Iter.isUsable()) 4813 Iter = SemaRef.BuildBinOp(CurScope, UpdLoc, BO_Rem, Iter.get(), 4814 IS.NumIterations); 4815 if (!Iter.isUsable()) { 4816 HasErrors = true; 4817 break; 4818 } 4819 4820 // Build update: IS.CounterVar(Private) = IS.Start + Iter * IS.Step 4821 auto *VD = cast<VarDecl>(cast<DeclRefExpr>(IS.CounterVar)->getDecl()); 4822 auto *CounterVar = buildDeclRefExpr(SemaRef, VD, IS.CounterVar->getType(), 4823 IS.CounterVar->getExprLoc(), 4824 /*RefersToCapture=*/true); 4825 ExprResult Init = BuildCounterInit(SemaRef, CurScope, UpdLoc, CounterVar, 4826 IS.CounterInit, Captures); 4827 if (!Init.isUsable()) { 4828 HasErrors = true; 4829 break; 4830 } 4831 ExprResult Update = BuildCounterUpdate( 4832 SemaRef, CurScope, UpdLoc, CounterVar, IS.CounterInit, Iter, 4833 IS.CounterStep, IS.Subtract, &Captures); 4834 if (!Update.isUsable()) { 4835 HasErrors = true; 4836 break; 4837 } 4838 4839 // Build final: IS.CounterVar = IS.Start + IS.NumIters * IS.Step 4840 ExprResult Final = BuildCounterUpdate( 4841 SemaRef, CurScope, UpdLoc, CounterVar, IS.CounterInit, 4842 IS.NumIterations, IS.CounterStep, IS.Subtract, &Captures); 4843 if (!Final.isUsable()) { 4844 HasErrors = true; 4845 break; 4846 } 4847 4848 // Build Div for the next iteration: Div <- Div * IS.NumIters 4849 if (Cnt != 0) { 4850 if (Div.isUnset()) 4851 Div = IS.NumIterations; 4852 else 4853 Div = SemaRef.BuildBinOp(CurScope, UpdLoc, BO_Mul, Div.get(), 4854 IS.NumIterations); 4855 4856 // Add parentheses (for debugging purposes only). 4857 if (Div.isUsable()) 4858 Div = tryBuildCapture(SemaRef, Div.get(), Captures); 4859 if (!Div.isUsable()) { 4860 HasErrors = true; 4861 break; 4862 } 4863 LoopMultipliers.push_back(Div.get()); 4864 } 4865 if (!Update.isUsable() || !Final.isUsable()) { 4866 HasErrors = true; 4867 break; 4868 } 4869 // Save results 4870 Built.Counters[Cnt] = IS.CounterVar; 4871 Built.PrivateCounters[Cnt] = IS.PrivateCounterVar; 4872 Built.Inits[Cnt] = Init.get(); 4873 Built.Updates[Cnt] = Update.get(); 4874 Built.Finals[Cnt] = Final.get(); 4875 } 4876 } 4877 4878 if (HasErrors) 4879 return 0; 4880 4881 // Save results 4882 Built.IterationVarRef = IV.get(); 4883 Built.LastIteration = LastIteration.get(); 4884 Built.NumIterations = NumIterations.get(); 4885 Built.CalcLastIteration = 4886 SemaRef.ActOnFinishFullExpr(CalcLastIteration.get()).get(); 4887 Built.PreCond = PreCond.get(); 4888 Built.PreInits = buildPreInits(C, Captures); 4889 Built.Cond = Cond.get(); 4890 Built.Init = Init.get(); 4891 Built.Inc = Inc.get(); 4892 Built.LB = LB.get(); 4893 Built.UB = UB.get(); 4894 Built.IL = IL.get(); 4895 Built.ST = ST.get(); 4896 Built.EUB = EUB.get(); 4897 Built.NLB = NextLB.get(); 4898 Built.NUB = NextUB.get(); 4899 Built.PrevLB = PrevLB.get(); 4900 Built.PrevUB = PrevUB.get(); 4901 Built.DistInc = DistInc.get(); 4902 Built.PrevEUB = PrevEUB.get(); 4903 Built.DistCombinedFields.LB = CombLB.get(); 4904 Built.DistCombinedFields.UB = CombUB.get(); 4905 Built.DistCombinedFields.EUB = CombEUB.get(); 4906 Built.DistCombinedFields.Init = CombInit.get(); 4907 Built.DistCombinedFields.Cond = CombCond.get(); 4908 Built.DistCombinedFields.NLB = CombNextLB.get(); 4909 Built.DistCombinedFields.NUB = CombNextUB.get(); 4910 4911 Expr *CounterVal = SemaRef.DefaultLvalueConversion(IV.get()).get(); 4912 // Fill data for doacross depend clauses. 4913 for (auto Pair : DSA.getDoacrossDependClauses()) { 4914 if (Pair.first->getDependencyKind() == OMPC_DEPEND_source) 4915 Pair.first->setCounterValue(CounterVal); 4916 else { 4917 if (NestedLoopCount != Pair.second.size() || 4918 NestedLoopCount != LoopMultipliers.size() + 1) { 4919 // Erroneous case - clause has some problems. 4920 Pair.first->setCounterValue(CounterVal); 4921 continue; 4922 } 4923 assert(Pair.first->getDependencyKind() == OMPC_DEPEND_sink); 4924 auto I = Pair.second.rbegin(); 4925 auto IS = IterSpaces.rbegin(); 4926 auto ILM = LoopMultipliers.rbegin(); 4927 Expr *UpCounterVal = CounterVal; 4928 Expr *Multiplier = nullptr; 4929 for (int Cnt = NestedLoopCount - 1; Cnt >= 0; --Cnt) { 4930 if (I->first) { 4931 assert(IS->CounterStep); 4932 Expr *NormalizedOffset = 4933 SemaRef 4934 .BuildBinOp(CurScope, I->first->getExprLoc(), BO_Div, 4935 I->first, IS->CounterStep) 4936 .get(); 4937 if (Multiplier) { 4938 NormalizedOffset = 4939 SemaRef 4940 .BuildBinOp(CurScope, I->first->getExprLoc(), BO_Mul, 4941 NormalizedOffset, Multiplier) 4942 .get(); 4943 } 4944 assert(I->second == OO_Plus || I->second == OO_Minus); 4945 BinaryOperatorKind BOK = (I->second == OO_Plus) ? BO_Add : BO_Sub; 4946 UpCounterVal = SemaRef 4947 .BuildBinOp(CurScope, I->first->getExprLoc(), BOK, 4948 UpCounterVal, NormalizedOffset) 4949 .get(); 4950 } 4951 Multiplier = *ILM; 4952 ++I; 4953 ++IS; 4954 ++ILM; 4955 } 4956 Pair.first->setCounterValue(UpCounterVal); 4957 } 4958 } 4959 4960 return NestedLoopCount; 4961 } 4962 4963 static Expr *getCollapseNumberExpr(ArrayRef<OMPClause *> Clauses) { 4964 auto CollapseClauses = 4965 OMPExecutableDirective::getClausesOfKind<OMPCollapseClause>(Clauses); 4966 if (CollapseClauses.begin() != CollapseClauses.end()) 4967 return (*CollapseClauses.begin())->getNumForLoops(); 4968 return nullptr; 4969 } 4970 4971 static Expr *getOrderedNumberExpr(ArrayRef<OMPClause *> Clauses) { 4972 auto OrderedClauses = 4973 OMPExecutableDirective::getClausesOfKind<OMPOrderedClause>(Clauses); 4974 if (OrderedClauses.begin() != OrderedClauses.end()) 4975 return (*OrderedClauses.begin())->getNumForLoops(); 4976 return nullptr; 4977 } 4978 4979 static bool checkSimdlenSafelenSpecified(Sema &S, 4980 const ArrayRef<OMPClause *> Clauses) { 4981 OMPSafelenClause *Safelen = nullptr; 4982 OMPSimdlenClause *Simdlen = nullptr; 4983 4984 for (auto *Clause : Clauses) { 4985 if (Clause->getClauseKind() == OMPC_safelen) 4986 Safelen = cast<OMPSafelenClause>(Clause); 4987 else if (Clause->getClauseKind() == OMPC_simdlen) 4988 Simdlen = cast<OMPSimdlenClause>(Clause); 4989 if (Safelen && Simdlen) 4990 break; 4991 } 4992 4993 if (Simdlen && Safelen) { 4994 llvm::APSInt SimdlenRes, SafelenRes; 4995 auto SimdlenLength = Simdlen->getSimdlen(); 4996 auto SafelenLength = Safelen->getSafelen(); 4997 if (SimdlenLength->isValueDependent() || SimdlenLength->isTypeDependent() || 4998 SimdlenLength->isInstantiationDependent() || 4999 SimdlenLength->containsUnexpandedParameterPack()) 5000 return false; 5001 if (SafelenLength->isValueDependent() || SafelenLength->isTypeDependent() || 5002 SafelenLength->isInstantiationDependent() || 5003 SafelenLength->containsUnexpandedParameterPack()) 5004 return false; 5005 SimdlenLength->EvaluateAsInt(SimdlenRes, S.Context); 5006 SafelenLength->EvaluateAsInt(SafelenRes, S.Context); 5007 // OpenMP 4.5 [2.8.1, simd Construct, Restrictions] 5008 // If both simdlen and safelen clauses are specified, the value of the 5009 // simdlen parameter must be less than or equal to the value of the safelen 5010 // parameter. 5011 if (SimdlenRes > SafelenRes) { 5012 S.Diag(SimdlenLength->getExprLoc(), 5013 diag::err_omp_wrong_simdlen_safelen_values) 5014 << SimdlenLength->getSourceRange() << SafelenLength->getSourceRange(); 5015 return true; 5016 } 5017 } 5018 return false; 5019 } 5020 5021 StmtResult Sema::ActOnOpenMPSimdDirective( 5022 ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc, 5023 SourceLocation EndLoc, 5024 llvm::DenseMap<ValueDecl *, Expr *> &VarsWithImplicitDSA) { 5025 if (!AStmt) 5026 return StmtError(); 5027 5028 assert(isa<CapturedStmt>(AStmt) && "Captured statement expected"); 5029 OMPLoopDirective::HelperExprs B; 5030 // In presence of clause 'collapse' or 'ordered' with number of loops, it will 5031 // define the nested loops number. 5032 unsigned NestedLoopCount = CheckOpenMPLoop( 5033 OMPD_simd, getCollapseNumberExpr(Clauses), getOrderedNumberExpr(Clauses), 5034 AStmt, *this, *DSAStack, VarsWithImplicitDSA, B); 5035 if (NestedLoopCount == 0) 5036 return StmtError(); 5037 5038 assert((CurContext->isDependentContext() || B.builtAll()) && 5039 "omp simd loop exprs were not built"); 5040 5041 if (!CurContext->isDependentContext()) { 5042 // Finalize the clauses that need pre-built expressions for CodeGen. 5043 for (auto C : Clauses) { 5044 if (auto *LC = dyn_cast<OMPLinearClause>(C)) 5045 if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef), 5046 B.NumIterations, *this, CurScope, 5047 DSAStack)) 5048 return StmtError(); 5049 } 5050 } 5051 5052 if (checkSimdlenSafelenSpecified(*this, Clauses)) 5053 return StmtError(); 5054 5055 getCurFunction()->setHasBranchProtectedScope(); 5056 return OMPSimdDirective::Create(Context, StartLoc, EndLoc, NestedLoopCount, 5057 Clauses, AStmt, B); 5058 } 5059 5060 StmtResult Sema::ActOnOpenMPForDirective( 5061 ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc, 5062 SourceLocation EndLoc, 5063 llvm::DenseMap<ValueDecl *, Expr *> &VarsWithImplicitDSA) { 5064 if (!AStmt) 5065 return StmtError(); 5066 5067 assert(isa<CapturedStmt>(AStmt) && "Captured statement expected"); 5068 OMPLoopDirective::HelperExprs B; 5069 // In presence of clause 'collapse' or 'ordered' with number of loops, it will 5070 // define the nested loops number. 5071 unsigned NestedLoopCount = CheckOpenMPLoop( 5072 OMPD_for, getCollapseNumberExpr(Clauses), getOrderedNumberExpr(Clauses), 5073 AStmt, *this, *DSAStack, VarsWithImplicitDSA, B); 5074 if (NestedLoopCount == 0) 5075 return StmtError(); 5076 5077 assert((CurContext->isDependentContext() || B.builtAll()) && 5078 "omp for loop exprs were not built"); 5079 5080 if (!CurContext->isDependentContext()) { 5081 // Finalize the clauses that need pre-built expressions for CodeGen. 5082 for (auto C : Clauses) { 5083 if (auto *LC = dyn_cast<OMPLinearClause>(C)) 5084 if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef), 5085 B.NumIterations, *this, CurScope, 5086 DSAStack)) 5087 return StmtError(); 5088 } 5089 } 5090 5091 getCurFunction()->setHasBranchProtectedScope(); 5092 return OMPForDirective::Create(Context, StartLoc, EndLoc, NestedLoopCount, 5093 Clauses, AStmt, B, DSAStack->isCancelRegion()); 5094 } 5095 5096 StmtResult Sema::ActOnOpenMPForSimdDirective( 5097 ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc, 5098 SourceLocation EndLoc, 5099 llvm::DenseMap<ValueDecl *, Expr *> &VarsWithImplicitDSA) { 5100 if (!AStmt) 5101 return StmtError(); 5102 5103 assert(isa<CapturedStmt>(AStmt) && "Captured statement expected"); 5104 OMPLoopDirective::HelperExprs B; 5105 // In presence of clause 'collapse' or 'ordered' with number of loops, it will 5106 // define the nested loops number. 5107 unsigned NestedLoopCount = 5108 CheckOpenMPLoop(OMPD_for_simd, getCollapseNumberExpr(Clauses), 5109 getOrderedNumberExpr(Clauses), AStmt, *this, *DSAStack, 5110 VarsWithImplicitDSA, B); 5111 if (NestedLoopCount == 0) 5112 return StmtError(); 5113 5114 assert((CurContext->isDependentContext() || B.builtAll()) && 5115 "omp for simd loop exprs were not built"); 5116 5117 if (!CurContext->isDependentContext()) { 5118 // Finalize the clauses that need pre-built expressions for CodeGen. 5119 for (auto C : Clauses) { 5120 if (auto *LC = dyn_cast<OMPLinearClause>(C)) 5121 if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef), 5122 B.NumIterations, *this, CurScope, 5123 DSAStack)) 5124 return StmtError(); 5125 } 5126 } 5127 5128 if (checkSimdlenSafelenSpecified(*this, Clauses)) 5129 return StmtError(); 5130 5131 getCurFunction()->setHasBranchProtectedScope(); 5132 return OMPForSimdDirective::Create(Context, StartLoc, EndLoc, NestedLoopCount, 5133 Clauses, AStmt, B); 5134 } 5135 5136 StmtResult Sema::ActOnOpenMPSectionsDirective(ArrayRef<OMPClause *> Clauses, 5137 Stmt *AStmt, 5138 SourceLocation StartLoc, 5139 SourceLocation EndLoc) { 5140 if (!AStmt) 5141 return StmtError(); 5142 5143 assert(isa<CapturedStmt>(AStmt) && "Captured statement expected"); 5144 auto BaseStmt = AStmt; 5145 while (auto *CS = dyn_cast_or_null<CapturedStmt>(BaseStmt)) 5146 BaseStmt = CS->getCapturedStmt(); 5147 if (auto *C = dyn_cast_or_null<CompoundStmt>(BaseStmt)) { 5148 auto S = C->children(); 5149 if (S.begin() == S.end()) 5150 return StmtError(); 5151 // All associated statements must be '#pragma omp section' except for 5152 // the first one. 5153 for (Stmt *SectionStmt : llvm::make_range(std::next(S.begin()), S.end())) { 5154 if (!SectionStmt || !isa<OMPSectionDirective>(SectionStmt)) { 5155 if (SectionStmt) 5156 Diag(SectionStmt->getLocStart(), 5157 diag::err_omp_sections_substmt_not_section); 5158 return StmtError(); 5159 } 5160 cast<OMPSectionDirective>(SectionStmt) 5161 ->setHasCancel(DSAStack->isCancelRegion()); 5162 } 5163 } else { 5164 Diag(AStmt->getLocStart(), diag::err_omp_sections_not_compound_stmt); 5165 return StmtError(); 5166 } 5167 5168 getCurFunction()->setHasBranchProtectedScope(); 5169 5170 return OMPSectionsDirective::Create(Context, StartLoc, EndLoc, Clauses, AStmt, 5171 DSAStack->isCancelRegion()); 5172 } 5173 5174 StmtResult Sema::ActOnOpenMPSectionDirective(Stmt *AStmt, 5175 SourceLocation StartLoc, 5176 SourceLocation EndLoc) { 5177 if (!AStmt) 5178 return StmtError(); 5179 5180 assert(isa<CapturedStmt>(AStmt) && "Captured statement expected"); 5181 5182 getCurFunction()->setHasBranchProtectedScope(); 5183 DSAStack->setParentCancelRegion(DSAStack->isCancelRegion()); 5184 5185 return OMPSectionDirective::Create(Context, StartLoc, EndLoc, AStmt, 5186 DSAStack->isCancelRegion()); 5187 } 5188 5189 StmtResult Sema::ActOnOpenMPSingleDirective(ArrayRef<OMPClause *> Clauses, 5190 Stmt *AStmt, 5191 SourceLocation StartLoc, 5192 SourceLocation EndLoc) { 5193 if (!AStmt) 5194 return StmtError(); 5195 5196 assert(isa<CapturedStmt>(AStmt) && "Captured statement expected"); 5197 5198 getCurFunction()->setHasBranchProtectedScope(); 5199 5200 // OpenMP [2.7.3, single Construct, Restrictions] 5201 // The copyprivate clause must not be used with the nowait clause. 5202 OMPClause *Nowait = nullptr; 5203 OMPClause *Copyprivate = nullptr; 5204 for (auto *Clause : Clauses) { 5205 if (Clause->getClauseKind() == OMPC_nowait) 5206 Nowait = Clause; 5207 else if (Clause->getClauseKind() == OMPC_copyprivate) 5208 Copyprivate = Clause; 5209 if (Copyprivate && Nowait) { 5210 Diag(Copyprivate->getLocStart(), 5211 diag::err_omp_single_copyprivate_with_nowait); 5212 Diag(Nowait->getLocStart(), diag::note_omp_nowait_clause_here); 5213 return StmtError(); 5214 } 5215 } 5216 5217 return OMPSingleDirective::Create(Context, StartLoc, EndLoc, Clauses, AStmt); 5218 } 5219 5220 StmtResult Sema::ActOnOpenMPMasterDirective(Stmt *AStmt, 5221 SourceLocation StartLoc, 5222 SourceLocation EndLoc) { 5223 if (!AStmt) 5224 return StmtError(); 5225 5226 assert(isa<CapturedStmt>(AStmt) && "Captured statement expected"); 5227 5228 getCurFunction()->setHasBranchProtectedScope(); 5229 5230 return OMPMasterDirective::Create(Context, StartLoc, EndLoc, AStmt); 5231 } 5232 5233 StmtResult Sema::ActOnOpenMPCriticalDirective( 5234 const DeclarationNameInfo &DirName, ArrayRef<OMPClause *> Clauses, 5235 Stmt *AStmt, SourceLocation StartLoc, SourceLocation EndLoc) { 5236 if (!AStmt) 5237 return StmtError(); 5238 5239 assert(isa<CapturedStmt>(AStmt) && "Captured statement expected"); 5240 5241 bool ErrorFound = false; 5242 llvm::APSInt Hint; 5243 SourceLocation HintLoc; 5244 bool DependentHint = false; 5245 for (auto *C : Clauses) { 5246 if (C->getClauseKind() == OMPC_hint) { 5247 if (!DirName.getName()) { 5248 Diag(C->getLocStart(), diag::err_omp_hint_clause_no_name); 5249 ErrorFound = true; 5250 } 5251 Expr *E = cast<OMPHintClause>(C)->getHint(); 5252 if (E->isTypeDependent() || E->isValueDependent() || 5253 E->isInstantiationDependent()) 5254 DependentHint = true; 5255 else { 5256 Hint = E->EvaluateKnownConstInt(Context); 5257 HintLoc = C->getLocStart(); 5258 } 5259 } 5260 } 5261 if (ErrorFound) 5262 return StmtError(); 5263 auto Pair = DSAStack->getCriticalWithHint(DirName); 5264 if (Pair.first && DirName.getName() && !DependentHint) { 5265 if (llvm::APSInt::compareValues(Hint, Pair.second) != 0) { 5266 Diag(StartLoc, diag::err_omp_critical_with_hint); 5267 if (HintLoc.isValid()) { 5268 Diag(HintLoc, diag::note_omp_critical_hint_here) 5269 << 0 << Hint.toString(/*Radix=*/10, /*Signed=*/false); 5270 } else 5271 Diag(StartLoc, diag::note_omp_critical_no_hint) << 0; 5272 if (auto *C = Pair.first->getSingleClause<OMPHintClause>()) { 5273 Diag(C->getLocStart(), diag::note_omp_critical_hint_here) 5274 << 1 5275 << C->getHint()->EvaluateKnownConstInt(Context).toString( 5276 /*Radix=*/10, /*Signed=*/false); 5277 } else 5278 Diag(Pair.first->getLocStart(), diag::note_omp_critical_no_hint) << 1; 5279 } 5280 } 5281 5282 getCurFunction()->setHasBranchProtectedScope(); 5283 5284 auto *Dir = OMPCriticalDirective::Create(Context, DirName, StartLoc, EndLoc, 5285 Clauses, AStmt); 5286 if (!Pair.first && DirName.getName() && !DependentHint) 5287 DSAStack->addCriticalWithHint(Dir, Hint); 5288 return Dir; 5289 } 5290 5291 StmtResult Sema::ActOnOpenMPParallelForDirective( 5292 ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc, 5293 SourceLocation EndLoc, 5294 llvm::DenseMap<ValueDecl *, Expr *> &VarsWithImplicitDSA) { 5295 if (!AStmt) 5296 return StmtError(); 5297 5298 CapturedStmt *CS = cast<CapturedStmt>(AStmt); 5299 // 1.2.2 OpenMP Language Terminology 5300 // Structured block - An executable statement with a single entry at the 5301 // top and a single exit at the bottom. 5302 // The point of exit cannot be a branch out of the structured block. 5303 // longjmp() and throw() must not violate the entry/exit criteria. 5304 CS->getCapturedDecl()->setNothrow(); 5305 5306 OMPLoopDirective::HelperExprs B; 5307 // In presence of clause 'collapse' or 'ordered' with number of loops, it will 5308 // define the nested loops number. 5309 unsigned NestedLoopCount = 5310 CheckOpenMPLoop(OMPD_parallel_for, getCollapseNumberExpr(Clauses), 5311 getOrderedNumberExpr(Clauses), AStmt, *this, *DSAStack, 5312 VarsWithImplicitDSA, B); 5313 if (NestedLoopCount == 0) 5314 return StmtError(); 5315 5316 assert((CurContext->isDependentContext() || B.builtAll()) && 5317 "omp parallel for loop exprs were not built"); 5318 5319 if (!CurContext->isDependentContext()) { 5320 // Finalize the clauses that need pre-built expressions for CodeGen. 5321 for (auto C : Clauses) { 5322 if (auto *LC = dyn_cast<OMPLinearClause>(C)) 5323 if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef), 5324 B.NumIterations, *this, CurScope, 5325 DSAStack)) 5326 return StmtError(); 5327 } 5328 } 5329 5330 getCurFunction()->setHasBranchProtectedScope(); 5331 return OMPParallelForDirective::Create(Context, StartLoc, EndLoc, 5332 NestedLoopCount, Clauses, AStmt, B, 5333 DSAStack->isCancelRegion()); 5334 } 5335 5336 StmtResult Sema::ActOnOpenMPParallelForSimdDirective( 5337 ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc, 5338 SourceLocation EndLoc, 5339 llvm::DenseMap<ValueDecl *, Expr *> &VarsWithImplicitDSA) { 5340 if (!AStmt) 5341 return StmtError(); 5342 5343 CapturedStmt *CS = cast<CapturedStmt>(AStmt); 5344 // 1.2.2 OpenMP Language Terminology 5345 // Structured block - An executable statement with a single entry at the 5346 // top and a single exit at the bottom. 5347 // The point of exit cannot be a branch out of the structured block. 5348 // longjmp() and throw() must not violate the entry/exit criteria. 5349 CS->getCapturedDecl()->setNothrow(); 5350 5351 OMPLoopDirective::HelperExprs B; 5352 // In presence of clause 'collapse' or 'ordered' with number of loops, it will 5353 // define the nested loops number. 5354 unsigned NestedLoopCount = 5355 CheckOpenMPLoop(OMPD_parallel_for_simd, getCollapseNumberExpr(Clauses), 5356 getOrderedNumberExpr(Clauses), AStmt, *this, *DSAStack, 5357 VarsWithImplicitDSA, B); 5358 if (NestedLoopCount == 0) 5359 return StmtError(); 5360 5361 if (!CurContext->isDependentContext()) { 5362 // Finalize the clauses that need pre-built expressions for CodeGen. 5363 for (auto C : Clauses) { 5364 if (auto *LC = dyn_cast<OMPLinearClause>(C)) 5365 if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef), 5366 B.NumIterations, *this, CurScope, 5367 DSAStack)) 5368 return StmtError(); 5369 } 5370 } 5371 5372 if (checkSimdlenSafelenSpecified(*this, Clauses)) 5373 return StmtError(); 5374 5375 getCurFunction()->setHasBranchProtectedScope(); 5376 return OMPParallelForSimdDirective::Create( 5377 Context, StartLoc, EndLoc, NestedLoopCount, Clauses, AStmt, B); 5378 } 5379 5380 StmtResult 5381 Sema::ActOnOpenMPParallelSectionsDirective(ArrayRef<OMPClause *> Clauses, 5382 Stmt *AStmt, SourceLocation StartLoc, 5383 SourceLocation EndLoc) { 5384 if (!AStmt) 5385 return StmtError(); 5386 5387 assert(isa<CapturedStmt>(AStmt) && "Captured statement expected"); 5388 auto BaseStmt = AStmt; 5389 while (auto *CS = dyn_cast_or_null<CapturedStmt>(BaseStmt)) 5390 BaseStmt = CS->getCapturedStmt(); 5391 if (auto *C = dyn_cast_or_null<CompoundStmt>(BaseStmt)) { 5392 auto S = C->children(); 5393 if (S.begin() == S.end()) 5394 return StmtError(); 5395 // All associated statements must be '#pragma omp section' except for 5396 // the first one. 5397 for (Stmt *SectionStmt : llvm::make_range(std::next(S.begin()), S.end())) { 5398 if (!SectionStmt || !isa<OMPSectionDirective>(SectionStmt)) { 5399 if (SectionStmt) 5400 Diag(SectionStmt->getLocStart(), 5401 diag::err_omp_parallel_sections_substmt_not_section); 5402 return StmtError(); 5403 } 5404 cast<OMPSectionDirective>(SectionStmt) 5405 ->setHasCancel(DSAStack->isCancelRegion()); 5406 } 5407 } else { 5408 Diag(AStmt->getLocStart(), 5409 diag::err_omp_parallel_sections_not_compound_stmt); 5410 return StmtError(); 5411 } 5412 5413 getCurFunction()->setHasBranchProtectedScope(); 5414 5415 return OMPParallelSectionsDirective::Create( 5416 Context, StartLoc, EndLoc, Clauses, AStmt, DSAStack->isCancelRegion()); 5417 } 5418 5419 StmtResult Sema::ActOnOpenMPTaskDirective(ArrayRef<OMPClause *> Clauses, 5420 Stmt *AStmt, SourceLocation StartLoc, 5421 SourceLocation EndLoc) { 5422 if (!AStmt) 5423 return StmtError(); 5424 5425 auto *CS = cast<CapturedStmt>(AStmt); 5426 // 1.2.2 OpenMP Language Terminology 5427 // Structured block - An executable statement with a single entry at the 5428 // top and a single exit at the bottom. 5429 // The point of exit cannot be a branch out of the structured block. 5430 // longjmp() and throw() must not violate the entry/exit criteria. 5431 CS->getCapturedDecl()->setNothrow(); 5432 5433 getCurFunction()->setHasBranchProtectedScope(); 5434 5435 return OMPTaskDirective::Create(Context, StartLoc, EndLoc, Clauses, AStmt, 5436 DSAStack->isCancelRegion()); 5437 } 5438 5439 StmtResult Sema::ActOnOpenMPTaskyieldDirective(SourceLocation StartLoc, 5440 SourceLocation EndLoc) { 5441 return OMPTaskyieldDirective::Create(Context, StartLoc, EndLoc); 5442 } 5443 5444 StmtResult Sema::ActOnOpenMPBarrierDirective(SourceLocation StartLoc, 5445 SourceLocation EndLoc) { 5446 return OMPBarrierDirective::Create(Context, StartLoc, EndLoc); 5447 } 5448 5449 StmtResult Sema::ActOnOpenMPTaskwaitDirective(SourceLocation StartLoc, 5450 SourceLocation EndLoc) { 5451 return OMPTaskwaitDirective::Create(Context, StartLoc, EndLoc); 5452 } 5453 5454 StmtResult Sema::ActOnOpenMPTaskgroupDirective(ArrayRef<OMPClause *> Clauses, 5455 Stmt *AStmt, 5456 SourceLocation StartLoc, 5457 SourceLocation EndLoc) { 5458 if (!AStmt) 5459 return StmtError(); 5460 5461 assert(isa<CapturedStmt>(AStmt) && "Captured statement expected"); 5462 5463 getCurFunction()->setHasBranchProtectedScope(); 5464 5465 return OMPTaskgroupDirective::Create(Context, StartLoc, EndLoc, Clauses, 5466 AStmt, 5467 DSAStack->getTaskgroupReductionRef()); 5468 } 5469 5470 StmtResult Sema::ActOnOpenMPFlushDirective(ArrayRef<OMPClause *> Clauses, 5471 SourceLocation StartLoc, 5472 SourceLocation EndLoc) { 5473 assert(Clauses.size() <= 1 && "Extra clauses in flush directive"); 5474 return OMPFlushDirective::Create(Context, StartLoc, EndLoc, Clauses); 5475 } 5476 5477 StmtResult Sema::ActOnOpenMPOrderedDirective(ArrayRef<OMPClause *> Clauses, 5478 Stmt *AStmt, 5479 SourceLocation StartLoc, 5480 SourceLocation EndLoc) { 5481 OMPClause *DependFound = nullptr; 5482 OMPClause *DependSourceClause = nullptr; 5483 OMPClause *DependSinkClause = nullptr; 5484 bool ErrorFound = false; 5485 OMPThreadsClause *TC = nullptr; 5486 OMPSIMDClause *SC = nullptr; 5487 for (auto *C : Clauses) { 5488 if (auto *DC = dyn_cast<OMPDependClause>(C)) { 5489 DependFound = C; 5490 if (DC->getDependencyKind() == OMPC_DEPEND_source) { 5491 if (DependSourceClause) { 5492 Diag(C->getLocStart(), diag::err_omp_more_one_clause) 5493 << getOpenMPDirectiveName(OMPD_ordered) 5494 << getOpenMPClauseName(OMPC_depend) << 2; 5495 ErrorFound = true; 5496 } else 5497 DependSourceClause = C; 5498 if (DependSinkClause) { 5499 Diag(C->getLocStart(), diag::err_omp_depend_sink_source_not_allowed) 5500 << 0; 5501 ErrorFound = true; 5502 } 5503 } else if (DC->getDependencyKind() == OMPC_DEPEND_sink) { 5504 if (DependSourceClause) { 5505 Diag(C->getLocStart(), diag::err_omp_depend_sink_source_not_allowed) 5506 << 1; 5507 ErrorFound = true; 5508 } 5509 DependSinkClause = C; 5510 } 5511 } else if (C->getClauseKind() == OMPC_threads) 5512 TC = cast<OMPThreadsClause>(C); 5513 else if (C->getClauseKind() == OMPC_simd) 5514 SC = cast<OMPSIMDClause>(C); 5515 } 5516 if (!ErrorFound && !SC && 5517 isOpenMPSimdDirective(DSAStack->getParentDirective())) { 5518 // OpenMP [2.8.1,simd Construct, Restrictions] 5519 // An ordered construct with the simd clause is the only OpenMP construct 5520 // that can appear in the simd region. 5521 Diag(StartLoc, diag::err_omp_prohibited_region_simd); 5522 ErrorFound = true; 5523 } else if (DependFound && (TC || SC)) { 5524 Diag(DependFound->getLocStart(), diag::err_omp_depend_clause_thread_simd) 5525 << getOpenMPClauseName(TC ? TC->getClauseKind() : SC->getClauseKind()); 5526 ErrorFound = true; 5527 } else if (DependFound && !DSAStack->getParentOrderedRegionParam()) { 5528 Diag(DependFound->getLocStart(), 5529 diag::err_omp_ordered_directive_without_param); 5530 ErrorFound = true; 5531 } else if (TC || Clauses.empty()) { 5532 if (auto *Param = DSAStack->getParentOrderedRegionParam()) { 5533 SourceLocation ErrLoc = TC ? TC->getLocStart() : StartLoc; 5534 Diag(ErrLoc, diag::err_omp_ordered_directive_with_param) 5535 << (TC != nullptr); 5536 Diag(Param->getLocStart(), diag::note_omp_ordered_param); 5537 ErrorFound = true; 5538 } 5539 } 5540 if ((!AStmt && !DependFound) || ErrorFound) 5541 return StmtError(); 5542 5543 if (AStmt) { 5544 assert(isa<CapturedStmt>(AStmt) && "Captured statement expected"); 5545 5546 getCurFunction()->setHasBranchProtectedScope(); 5547 } 5548 5549 return OMPOrderedDirective::Create(Context, StartLoc, EndLoc, Clauses, AStmt); 5550 } 5551 5552 namespace { 5553 /// \brief Helper class for checking expression in 'omp atomic [update]' 5554 /// construct. 5555 class OpenMPAtomicUpdateChecker { 5556 /// \brief Error results for atomic update expressions. 5557 enum ExprAnalysisErrorCode { 5558 /// \brief A statement is not an expression statement. 5559 NotAnExpression, 5560 /// \brief Expression is not builtin binary or unary operation. 5561 NotABinaryOrUnaryExpression, 5562 /// \brief Unary operation is not post-/pre- increment/decrement operation. 5563 NotAnUnaryIncDecExpression, 5564 /// \brief An expression is not of scalar type. 5565 NotAScalarType, 5566 /// \brief A binary operation is not an assignment operation. 5567 NotAnAssignmentOp, 5568 /// \brief RHS part of the binary operation is not a binary expression. 5569 NotABinaryExpression, 5570 /// \brief RHS part is not additive/multiplicative/shift/biwise binary 5571 /// expression. 5572 NotABinaryOperator, 5573 /// \brief RHS binary operation does not have reference to the updated LHS 5574 /// part. 5575 NotAnUpdateExpression, 5576 /// \brief No errors is found. 5577 NoError 5578 }; 5579 /// \brief Reference to Sema. 5580 Sema &SemaRef; 5581 /// \brief A location for note diagnostics (when error is found). 5582 SourceLocation NoteLoc; 5583 /// \brief 'x' lvalue part of the source atomic expression. 5584 Expr *X; 5585 /// \brief 'expr' rvalue part of the source atomic expression. 5586 Expr *E; 5587 /// \brief Helper expression of the form 5588 /// 'OpaqueValueExpr(x) binop OpaqueValueExpr(expr)' or 5589 /// 'OpaqueValueExpr(expr) binop OpaqueValueExpr(x)'. 5590 Expr *UpdateExpr; 5591 /// \brief Is 'x' a LHS in a RHS part of full update expression. It is 5592 /// important for non-associative operations. 5593 bool IsXLHSInRHSPart; 5594 BinaryOperatorKind Op; 5595 SourceLocation OpLoc; 5596 /// \brief true if the source expression is a postfix unary operation, false 5597 /// if it is a prefix unary operation. 5598 bool IsPostfixUpdate; 5599 5600 public: 5601 OpenMPAtomicUpdateChecker(Sema &SemaRef) 5602 : SemaRef(SemaRef), X(nullptr), E(nullptr), UpdateExpr(nullptr), 5603 IsXLHSInRHSPart(false), Op(BO_PtrMemD), IsPostfixUpdate(false) {} 5604 /// \brief Check specified statement that it is suitable for 'atomic update' 5605 /// constructs and extract 'x', 'expr' and Operation from the original 5606 /// expression. If DiagId and NoteId == 0, then only check is performed 5607 /// without error notification. 5608 /// \param DiagId Diagnostic which should be emitted if error is found. 5609 /// \param NoteId Diagnostic note for the main error message. 5610 /// \return true if statement is not an update expression, false otherwise. 5611 bool checkStatement(Stmt *S, unsigned DiagId = 0, unsigned NoteId = 0); 5612 /// \brief Return the 'x' lvalue part of the source atomic expression. 5613 Expr *getX() const { return X; } 5614 /// \brief Return the 'expr' rvalue part of the source atomic expression. 5615 Expr *getExpr() const { return E; } 5616 /// \brief Return the update expression used in calculation of the updated 5617 /// value. Always has form 'OpaqueValueExpr(x) binop OpaqueValueExpr(expr)' or 5618 /// 'OpaqueValueExpr(expr) binop OpaqueValueExpr(x)'. 5619 Expr *getUpdateExpr() const { return UpdateExpr; } 5620 /// \brief Return true if 'x' is LHS in RHS part of full update expression, 5621 /// false otherwise. 5622 bool isXLHSInRHSPart() const { return IsXLHSInRHSPart; } 5623 5624 /// \brief true if the source expression is a postfix unary operation, false 5625 /// if it is a prefix unary operation. 5626 bool isPostfixUpdate() const { return IsPostfixUpdate; } 5627 5628 private: 5629 bool checkBinaryOperation(BinaryOperator *AtomicBinOp, unsigned DiagId = 0, 5630 unsigned NoteId = 0); 5631 }; 5632 } // namespace 5633 5634 bool OpenMPAtomicUpdateChecker::checkBinaryOperation( 5635 BinaryOperator *AtomicBinOp, unsigned DiagId, unsigned NoteId) { 5636 ExprAnalysisErrorCode ErrorFound = NoError; 5637 SourceLocation ErrorLoc, NoteLoc; 5638 SourceRange ErrorRange, NoteRange; 5639 // Allowed constructs are: 5640 // x = x binop expr; 5641 // x = expr binop x; 5642 if (AtomicBinOp->getOpcode() == BO_Assign) { 5643 X = AtomicBinOp->getLHS(); 5644 if (auto *AtomicInnerBinOp = dyn_cast<BinaryOperator>( 5645 AtomicBinOp->getRHS()->IgnoreParenImpCasts())) { 5646 if (AtomicInnerBinOp->isMultiplicativeOp() || 5647 AtomicInnerBinOp->isAdditiveOp() || AtomicInnerBinOp->isShiftOp() || 5648 AtomicInnerBinOp->isBitwiseOp()) { 5649 Op = AtomicInnerBinOp->getOpcode(); 5650 OpLoc = AtomicInnerBinOp->getOperatorLoc(); 5651 auto *LHS = AtomicInnerBinOp->getLHS(); 5652 auto *RHS = AtomicInnerBinOp->getRHS(); 5653 llvm::FoldingSetNodeID XId, LHSId, RHSId; 5654 X->IgnoreParenImpCasts()->Profile(XId, SemaRef.getASTContext(), 5655 /*Canonical=*/true); 5656 LHS->IgnoreParenImpCasts()->Profile(LHSId, SemaRef.getASTContext(), 5657 /*Canonical=*/true); 5658 RHS->IgnoreParenImpCasts()->Profile(RHSId, SemaRef.getASTContext(), 5659 /*Canonical=*/true); 5660 if (XId == LHSId) { 5661 E = RHS; 5662 IsXLHSInRHSPart = true; 5663 } else if (XId == RHSId) { 5664 E = LHS; 5665 IsXLHSInRHSPart = false; 5666 } else { 5667 ErrorLoc = AtomicInnerBinOp->getExprLoc(); 5668 ErrorRange = AtomicInnerBinOp->getSourceRange(); 5669 NoteLoc = X->getExprLoc(); 5670 NoteRange = X->getSourceRange(); 5671 ErrorFound = NotAnUpdateExpression; 5672 } 5673 } else { 5674 ErrorLoc = AtomicInnerBinOp->getExprLoc(); 5675 ErrorRange = AtomicInnerBinOp->getSourceRange(); 5676 NoteLoc = AtomicInnerBinOp->getOperatorLoc(); 5677 NoteRange = SourceRange(NoteLoc, NoteLoc); 5678 ErrorFound = NotABinaryOperator; 5679 } 5680 } else { 5681 NoteLoc = ErrorLoc = AtomicBinOp->getRHS()->getExprLoc(); 5682 NoteRange = ErrorRange = AtomicBinOp->getRHS()->getSourceRange(); 5683 ErrorFound = NotABinaryExpression; 5684 } 5685 } else { 5686 ErrorLoc = AtomicBinOp->getExprLoc(); 5687 ErrorRange = AtomicBinOp->getSourceRange(); 5688 NoteLoc = AtomicBinOp->getOperatorLoc(); 5689 NoteRange = SourceRange(NoteLoc, NoteLoc); 5690 ErrorFound = NotAnAssignmentOp; 5691 } 5692 if (ErrorFound != NoError && DiagId != 0 && NoteId != 0) { 5693 SemaRef.Diag(ErrorLoc, DiagId) << ErrorRange; 5694 SemaRef.Diag(NoteLoc, NoteId) << ErrorFound << NoteRange; 5695 return true; 5696 } else if (SemaRef.CurContext->isDependentContext()) 5697 E = X = UpdateExpr = nullptr; 5698 return ErrorFound != NoError; 5699 } 5700 5701 bool OpenMPAtomicUpdateChecker::checkStatement(Stmt *S, unsigned DiagId, 5702 unsigned NoteId) { 5703 ExprAnalysisErrorCode ErrorFound = NoError; 5704 SourceLocation ErrorLoc, NoteLoc; 5705 SourceRange ErrorRange, NoteRange; 5706 // Allowed constructs are: 5707 // x++; 5708 // x--; 5709 // ++x; 5710 // --x; 5711 // x binop= expr; 5712 // x = x binop expr; 5713 // x = expr binop x; 5714 if (auto *AtomicBody = dyn_cast<Expr>(S)) { 5715 AtomicBody = AtomicBody->IgnoreParenImpCasts(); 5716 if (AtomicBody->getType()->isScalarType() || 5717 AtomicBody->isInstantiationDependent()) { 5718 if (auto *AtomicCompAssignOp = dyn_cast<CompoundAssignOperator>( 5719 AtomicBody->IgnoreParenImpCasts())) { 5720 // Check for Compound Assignment Operation 5721 Op = BinaryOperator::getOpForCompoundAssignment( 5722 AtomicCompAssignOp->getOpcode()); 5723 OpLoc = AtomicCompAssignOp->getOperatorLoc(); 5724 E = AtomicCompAssignOp->getRHS(); 5725 X = AtomicCompAssignOp->getLHS()->IgnoreParens(); 5726 IsXLHSInRHSPart = true; 5727 } else if (auto *AtomicBinOp = dyn_cast<BinaryOperator>( 5728 AtomicBody->IgnoreParenImpCasts())) { 5729 // Check for Binary Operation 5730 if (checkBinaryOperation(AtomicBinOp, DiagId, NoteId)) 5731 return true; 5732 } else if (auto *AtomicUnaryOp = dyn_cast<UnaryOperator>( 5733 AtomicBody->IgnoreParenImpCasts())) { 5734 // Check for Unary Operation 5735 if (AtomicUnaryOp->isIncrementDecrementOp()) { 5736 IsPostfixUpdate = AtomicUnaryOp->isPostfix(); 5737 Op = AtomicUnaryOp->isIncrementOp() ? BO_Add : BO_Sub; 5738 OpLoc = AtomicUnaryOp->getOperatorLoc(); 5739 X = AtomicUnaryOp->getSubExpr()->IgnoreParens(); 5740 E = SemaRef.ActOnIntegerConstant(OpLoc, /*uint64_t Val=*/1).get(); 5741 IsXLHSInRHSPart = true; 5742 } else { 5743 ErrorFound = NotAnUnaryIncDecExpression; 5744 ErrorLoc = AtomicUnaryOp->getExprLoc(); 5745 ErrorRange = AtomicUnaryOp->getSourceRange(); 5746 NoteLoc = AtomicUnaryOp->getOperatorLoc(); 5747 NoteRange = SourceRange(NoteLoc, NoteLoc); 5748 } 5749 } else if (!AtomicBody->isInstantiationDependent()) { 5750 ErrorFound = NotABinaryOrUnaryExpression; 5751 NoteLoc = ErrorLoc = AtomicBody->getExprLoc(); 5752 NoteRange = ErrorRange = AtomicBody->getSourceRange(); 5753 } 5754 } else { 5755 ErrorFound = NotAScalarType; 5756 NoteLoc = ErrorLoc = AtomicBody->getLocStart(); 5757 NoteRange = ErrorRange = SourceRange(NoteLoc, NoteLoc); 5758 } 5759 } else { 5760 ErrorFound = NotAnExpression; 5761 NoteLoc = ErrorLoc = S->getLocStart(); 5762 NoteRange = ErrorRange = SourceRange(NoteLoc, NoteLoc); 5763 } 5764 if (ErrorFound != NoError && DiagId != 0 && NoteId != 0) { 5765 SemaRef.Diag(ErrorLoc, DiagId) << ErrorRange; 5766 SemaRef.Diag(NoteLoc, NoteId) << ErrorFound << NoteRange; 5767 return true; 5768 } else if (SemaRef.CurContext->isDependentContext()) 5769 E = X = UpdateExpr = nullptr; 5770 if (ErrorFound == NoError && E && X) { 5771 // Build an update expression of form 'OpaqueValueExpr(x) binop 5772 // OpaqueValueExpr(expr)' or 'OpaqueValueExpr(expr) binop 5773 // OpaqueValueExpr(x)' and then cast it to the type of the 'x' expression. 5774 auto *OVEX = new (SemaRef.getASTContext()) 5775 OpaqueValueExpr(X->getExprLoc(), X->getType(), VK_RValue); 5776 auto *OVEExpr = new (SemaRef.getASTContext()) 5777 OpaqueValueExpr(E->getExprLoc(), E->getType(), VK_RValue); 5778 auto Update = 5779 SemaRef.CreateBuiltinBinOp(OpLoc, Op, IsXLHSInRHSPart ? OVEX : OVEExpr, 5780 IsXLHSInRHSPart ? OVEExpr : OVEX); 5781 if (Update.isInvalid()) 5782 return true; 5783 Update = SemaRef.PerformImplicitConversion(Update.get(), X->getType(), 5784 Sema::AA_Casting); 5785 if (Update.isInvalid()) 5786 return true; 5787 UpdateExpr = Update.get(); 5788 } 5789 return ErrorFound != NoError; 5790 } 5791 5792 StmtResult Sema::ActOnOpenMPAtomicDirective(ArrayRef<OMPClause *> Clauses, 5793 Stmt *AStmt, 5794 SourceLocation StartLoc, 5795 SourceLocation EndLoc) { 5796 if (!AStmt) 5797 return StmtError(); 5798 5799 auto *CS = cast<CapturedStmt>(AStmt); 5800 // 1.2.2 OpenMP Language Terminology 5801 // Structured block - An executable statement with a single entry at the 5802 // top and a single exit at the bottom. 5803 // The point of exit cannot be a branch out of the structured block. 5804 // longjmp() and throw() must not violate the entry/exit criteria. 5805 OpenMPClauseKind AtomicKind = OMPC_unknown; 5806 SourceLocation AtomicKindLoc; 5807 for (auto *C : Clauses) { 5808 if (C->getClauseKind() == OMPC_read || C->getClauseKind() == OMPC_write || 5809 C->getClauseKind() == OMPC_update || 5810 C->getClauseKind() == OMPC_capture) { 5811 if (AtomicKind != OMPC_unknown) { 5812 Diag(C->getLocStart(), diag::err_omp_atomic_several_clauses) 5813 << SourceRange(C->getLocStart(), C->getLocEnd()); 5814 Diag(AtomicKindLoc, diag::note_omp_atomic_previous_clause) 5815 << getOpenMPClauseName(AtomicKind); 5816 } else { 5817 AtomicKind = C->getClauseKind(); 5818 AtomicKindLoc = C->getLocStart(); 5819 } 5820 } 5821 } 5822 5823 auto Body = CS->getCapturedStmt(); 5824 if (auto *EWC = dyn_cast<ExprWithCleanups>(Body)) 5825 Body = EWC->getSubExpr(); 5826 5827 Expr *X = nullptr; 5828 Expr *V = nullptr; 5829 Expr *E = nullptr; 5830 Expr *UE = nullptr; 5831 bool IsXLHSInRHSPart = false; 5832 bool IsPostfixUpdate = false; 5833 // OpenMP [2.12.6, atomic Construct] 5834 // In the next expressions: 5835 // * x and v (as applicable) are both l-value expressions with scalar type. 5836 // * During the execution of an atomic region, multiple syntactic 5837 // occurrences of x must designate the same storage location. 5838 // * Neither of v and expr (as applicable) may access the storage location 5839 // designated by x. 5840 // * Neither of x and expr (as applicable) may access the storage location 5841 // designated by v. 5842 // * expr is an expression with scalar type. 5843 // * binop is one of +, *, -, /, &, ^, |, <<, or >>. 5844 // * binop, binop=, ++, and -- are not overloaded operators. 5845 // * The expression x binop expr must be numerically equivalent to x binop 5846 // (expr). This requirement is satisfied if the operators in expr have 5847 // precedence greater than binop, or by using parentheses around expr or 5848 // subexpressions of expr. 5849 // * The expression expr binop x must be numerically equivalent to (expr) 5850 // binop x. This requirement is satisfied if the operators in expr have 5851 // precedence equal to or greater than binop, or by using parentheses around 5852 // expr or subexpressions of expr. 5853 // * For forms that allow multiple occurrences of x, the number of times 5854 // that x is evaluated is unspecified. 5855 if (AtomicKind == OMPC_read) { 5856 enum { 5857 NotAnExpression, 5858 NotAnAssignmentOp, 5859 NotAScalarType, 5860 NotAnLValue, 5861 NoError 5862 } ErrorFound = NoError; 5863 SourceLocation ErrorLoc, NoteLoc; 5864 SourceRange ErrorRange, NoteRange; 5865 // If clause is read: 5866 // v = x; 5867 if (auto *AtomicBody = dyn_cast<Expr>(Body)) { 5868 auto *AtomicBinOp = 5869 dyn_cast<BinaryOperator>(AtomicBody->IgnoreParenImpCasts()); 5870 if (AtomicBinOp && AtomicBinOp->getOpcode() == BO_Assign) { 5871 X = AtomicBinOp->getRHS()->IgnoreParenImpCasts(); 5872 V = AtomicBinOp->getLHS()->IgnoreParenImpCasts(); 5873 if ((X->isInstantiationDependent() || X->getType()->isScalarType()) && 5874 (V->isInstantiationDependent() || V->getType()->isScalarType())) { 5875 if (!X->isLValue() || !V->isLValue()) { 5876 auto NotLValueExpr = X->isLValue() ? V : X; 5877 ErrorFound = NotAnLValue; 5878 ErrorLoc = AtomicBinOp->getExprLoc(); 5879 ErrorRange = AtomicBinOp->getSourceRange(); 5880 NoteLoc = NotLValueExpr->getExprLoc(); 5881 NoteRange = NotLValueExpr->getSourceRange(); 5882 } 5883 } else if (!X->isInstantiationDependent() || 5884 !V->isInstantiationDependent()) { 5885 auto NotScalarExpr = 5886 (X->isInstantiationDependent() || X->getType()->isScalarType()) 5887 ? V 5888 : X; 5889 ErrorFound = NotAScalarType; 5890 ErrorLoc = AtomicBinOp->getExprLoc(); 5891 ErrorRange = AtomicBinOp->getSourceRange(); 5892 NoteLoc = NotScalarExpr->getExprLoc(); 5893 NoteRange = NotScalarExpr->getSourceRange(); 5894 } 5895 } else if (!AtomicBody->isInstantiationDependent()) { 5896 ErrorFound = NotAnAssignmentOp; 5897 ErrorLoc = AtomicBody->getExprLoc(); 5898 ErrorRange = AtomicBody->getSourceRange(); 5899 NoteLoc = AtomicBinOp ? AtomicBinOp->getOperatorLoc() 5900 : AtomicBody->getExprLoc(); 5901 NoteRange = AtomicBinOp ? AtomicBinOp->getSourceRange() 5902 : AtomicBody->getSourceRange(); 5903 } 5904 } else { 5905 ErrorFound = NotAnExpression; 5906 NoteLoc = ErrorLoc = Body->getLocStart(); 5907 NoteRange = ErrorRange = SourceRange(NoteLoc, NoteLoc); 5908 } 5909 if (ErrorFound != NoError) { 5910 Diag(ErrorLoc, diag::err_omp_atomic_read_not_expression_statement) 5911 << ErrorRange; 5912 Diag(NoteLoc, diag::note_omp_atomic_read_write) << ErrorFound 5913 << NoteRange; 5914 return StmtError(); 5915 } else if (CurContext->isDependentContext()) 5916 V = X = nullptr; 5917 } else if (AtomicKind == OMPC_write) { 5918 enum { 5919 NotAnExpression, 5920 NotAnAssignmentOp, 5921 NotAScalarType, 5922 NotAnLValue, 5923 NoError 5924 } ErrorFound = NoError; 5925 SourceLocation ErrorLoc, NoteLoc; 5926 SourceRange ErrorRange, NoteRange; 5927 // If clause is write: 5928 // x = expr; 5929 if (auto *AtomicBody = dyn_cast<Expr>(Body)) { 5930 auto *AtomicBinOp = 5931 dyn_cast<BinaryOperator>(AtomicBody->IgnoreParenImpCasts()); 5932 if (AtomicBinOp && AtomicBinOp->getOpcode() == BO_Assign) { 5933 X = AtomicBinOp->getLHS(); 5934 E = AtomicBinOp->getRHS(); 5935 if ((X->isInstantiationDependent() || X->getType()->isScalarType()) && 5936 (E->isInstantiationDependent() || E->getType()->isScalarType())) { 5937 if (!X->isLValue()) { 5938 ErrorFound = NotAnLValue; 5939 ErrorLoc = AtomicBinOp->getExprLoc(); 5940 ErrorRange = AtomicBinOp->getSourceRange(); 5941 NoteLoc = X->getExprLoc(); 5942 NoteRange = X->getSourceRange(); 5943 } 5944 } else if (!X->isInstantiationDependent() || 5945 !E->isInstantiationDependent()) { 5946 auto NotScalarExpr = 5947 (X->isInstantiationDependent() || X->getType()->isScalarType()) 5948 ? E 5949 : X; 5950 ErrorFound = NotAScalarType; 5951 ErrorLoc = AtomicBinOp->getExprLoc(); 5952 ErrorRange = AtomicBinOp->getSourceRange(); 5953 NoteLoc = NotScalarExpr->getExprLoc(); 5954 NoteRange = NotScalarExpr->getSourceRange(); 5955 } 5956 } else if (!AtomicBody->isInstantiationDependent()) { 5957 ErrorFound = NotAnAssignmentOp; 5958 ErrorLoc = AtomicBody->getExprLoc(); 5959 ErrorRange = AtomicBody->getSourceRange(); 5960 NoteLoc = AtomicBinOp ? AtomicBinOp->getOperatorLoc() 5961 : AtomicBody->getExprLoc(); 5962 NoteRange = AtomicBinOp ? AtomicBinOp->getSourceRange() 5963 : AtomicBody->getSourceRange(); 5964 } 5965 } else { 5966 ErrorFound = NotAnExpression; 5967 NoteLoc = ErrorLoc = Body->getLocStart(); 5968 NoteRange = ErrorRange = SourceRange(NoteLoc, NoteLoc); 5969 } 5970 if (ErrorFound != NoError) { 5971 Diag(ErrorLoc, diag::err_omp_atomic_write_not_expression_statement) 5972 << ErrorRange; 5973 Diag(NoteLoc, diag::note_omp_atomic_read_write) << ErrorFound 5974 << NoteRange; 5975 return StmtError(); 5976 } else if (CurContext->isDependentContext()) 5977 E = X = nullptr; 5978 } else if (AtomicKind == OMPC_update || AtomicKind == OMPC_unknown) { 5979 // If clause is update: 5980 // x++; 5981 // x--; 5982 // ++x; 5983 // --x; 5984 // x binop= expr; 5985 // x = x binop expr; 5986 // x = expr binop x; 5987 OpenMPAtomicUpdateChecker Checker(*this); 5988 if (Checker.checkStatement( 5989 Body, (AtomicKind == OMPC_update) 5990 ? diag::err_omp_atomic_update_not_expression_statement 5991 : diag::err_omp_atomic_not_expression_statement, 5992 diag::note_omp_atomic_update)) 5993 return StmtError(); 5994 if (!CurContext->isDependentContext()) { 5995 E = Checker.getExpr(); 5996 X = Checker.getX(); 5997 UE = Checker.getUpdateExpr(); 5998 IsXLHSInRHSPart = Checker.isXLHSInRHSPart(); 5999 } 6000 } else if (AtomicKind == OMPC_capture) { 6001 enum { 6002 NotAnAssignmentOp, 6003 NotACompoundStatement, 6004 NotTwoSubstatements, 6005 NotASpecificExpression, 6006 NoError 6007 } ErrorFound = NoError; 6008 SourceLocation ErrorLoc, NoteLoc; 6009 SourceRange ErrorRange, NoteRange; 6010 if (auto *AtomicBody = dyn_cast<Expr>(Body)) { 6011 // If clause is a capture: 6012 // v = x++; 6013 // v = x--; 6014 // v = ++x; 6015 // v = --x; 6016 // v = x binop= expr; 6017 // v = x = x binop expr; 6018 // v = x = expr binop x; 6019 auto *AtomicBinOp = 6020 dyn_cast<BinaryOperator>(AtomicBody->IgnoreParenImpCasts()); 6021 if (AtomicBinOp && AtomicBinOp->getOpcode() == BO_Assign) { 6022 V = AtomicBinOp->getLHS(); 6023 Body = AtomicBinOp->getRHS()->IgnoreParenImpCasts(); 6024 OpenMPAtomicUpdateChecker Checker(*this); 6025 if (Checker.checkStatement( 6026 Body, diag::err_omp_atomic_capture_not_expression_statement, 6027 diag::note_omp_atomic_update)) 6028 return StmtError(); 6029 E = Checker.getExpr(); 6030 X = Checker.getX(); 6031 UE = Checker.getUpdateExpr(); 6032 IsXLHSInRHSPart = Checker.isXLHSInRHSPart(); 6033 IsPostfixUpdate = Checker.isPostfixUpdate(); 6034 } else if (!AtomicBody->isInstantiationDependent()) { 6035 ErrorLoc = AtomicBody->getExprLoc(); 6036 ErrorRange = AtomicBody->getSourceRange(); 6037 NoteLoc = AtomicBinOp ? AtomicBinOp->getOperatorLoc() 6038 : AtomicBody->getExprLoc(); 6039 NoteRange = AtomicBinOp ? AtomicBinOp->getSourceRange() 6040 : AtomicBody->getSourceRange(); 6041 ErrorFound = NotAnAssignmentOp; 6042 } 6043 if (ErrorFound != NoError) { 6044 Diag(ErrorLoc, diag::err_omp_atomic_capture_not_expression_statement) 6045 << ErrorRange; 6046 Diag(NoteLoc, diag::note_omp_atomic_capture) << ErrorFound << NoteRange; 6047 return StmtError(); 6048 } else if (CurContext->isDependentContext()) { 6049 UE = V = E = X = nullptr; 6050 } 6051 } else { 6052 // If clause is a capture: 6053 // { v = x; x = expr; } 6054 // { v = x; x++; } 6055 // { v = x; x--; } 6056 // { v = x; ++x; } 6057 // { v = x; --x; } 6058 // { v = x; x binop= expr; } 6059 // { v = x; x = x binop expr; } 6060 // { v = x; x = expr binop x; } 6061 // { x++; v = x; } 6062 // { x--; v = x; } 6063 // { ++x; v = x; } 6064 // { --x; v = x; } 6065 // { x binop= expr; v = x; } 6066 // { x = x binop expr; v = x; } 6067 // { x = expr binop x; v = x; } 6068 if (auto *CS = dyn_cast<CompoundStmt>(Body)) { 6069 // Check that this is { expr1; expr2; } 6070 if (CS->size() == 2) { 6071 auto *First = CS->body_front(); 6072 auto *Second = CS->body_back(); 6073 if (auto *EWC = dyn_cast<ExprWithCleanups>(First)) 6074 First = EWC->getSubExpr()->IgnoreParenImpCasts(); 6075 if (auto *EWC = dyn_cast<ExprWithCleanups>(Second)) 6076 Second = EWC->getSubExpr()->IgnoreParenImpCasts(); 6077 // Need to find what subexpression is 'v' and what is 'x'. 6078 OpenMPAtomicUpdateChecker Checker(*this); 6079 bool IsUpdateExprFound = !Checker.checkStatement(Second); 6080 BinaryOperator *BinOp = nullptr; 6081 if (IsUpdateExprFound) { 6082 BinOp = dyn_cast<BinaryOperator>(First); 6083 IsUpdateExprFound = BinOp && BinOp->getOpcode() == BO_Assign; 6084 } 6085 if (IsUpdateExprFound && !CurContext->isDependentContext()) { 6086 // { v = x; x++; } 6087 // { v = x; x--; } 6088 // { v = x; ++x; } 6089 // { v = x; --x; } 6090 // { v = x; x binop= expr; } 6091 // { v = x; x = x binop expr; } 6092 // { v = x; x = expr binop x; } 6093 // Check that the first expression has form v = x. 6094 auto *PossibleX = BinOp->getRHS()->IgnoreParenImpCasts(); 6095 llvm::FoldingSetNodeID XId, PossibleXId; 6096 Checker.getX()->Profile(XId, Context, /*Canonical=*/true); 6097 PossibleX->Profile(PossibleXId, Context, /*Canonical=*/true); 6098 IsUpdateExprFound = XId == PossibleXId; 6099 if (IsUpdateExprFound) { 6100 V = BinOp->getLHS(); 6101 X = Checker.getX(); 6102 E = Checker.getExpr(); 6103 UE = Checker.getUpdateExpr(); 6104 IsXLHSInRHSPart = Checker.isXLHSInRHSPart(); 6105 IsPostfixUpdate = true; 6106 } 6107 } 6108 if (!IsUpdateExprFound) { 6109 IsUpdateExprFound = !Checker.checkStatement(First); 6110 BinOp = nullptr; 6111 if (IsUpdateExprFound) { 6112 BinOp = dyn_cast<BinaryOperator>(Second); 6113 IsUpdateExprFound = BinOp && BinOp->getOpcode() == BO_Assign; 6114 } 6115 if (IsUpdateExprFound && !CurContext->isDependentContext()) { 6116 // { x++; v = x; } 6117 // { x--; v = x; } 6118 // { ++x; v = x; } 6119 // { --x; v = x; } 6120 // { x binop= expr; v = x; } 6121 // { x = x binop expr; v = x; } 6122 // { x = expr binop x; v = x; } 6123 // Check that the second expression has form v = x. 6124 auto *PossibleX = BinOp->getRHS()->IgnoreParenImpCasts(); 6125 llvm::FoldingSetNodeID XId, PossibleXId; 6126 Checker.getX()->Profile(XId, Context, /*Canonical=*/true); 6127 PossibleX->Profile(PossibleXId, Context, /*Canonical=*/true); 6128 IsUpdateExprFound = XId == PossibleXId; 6129 if (IsUpdateExprFound) { 6130 V = BinOp->getLHS(); 6131 X = Checker.getX(); 6132 E = Checker.getExpr(); 6133 UE = Checker.getUpdateExpr(); 6134 IsXLHSInRHSPart = Checker.isXLHSInRHSPart(); 6135 IsPostfixUpdate = false; 6136 } 6137 } 6138 } 6139 if (!IsUpdateExprFound) { 6140 // { v = x; x = expr; } 6141 auto *FirstExpr = dyn_cast<Expr>(First); 6142 auto *SecondExpr = dyn_cast<Expr>(Second); 6143 if (!FirstExpr || !SecondExpr || 6144 !(FirstExpr->isInstantiationDependent() || 6145 SecondExpr->isInstantiationDependent())) { 6146 auto *FirstBinOp = dyn_cast<BinaryOperator>(First); 6147 if (!FirstBinOp || FirstBinOp->getOpcode() != BO_Assign) { 6148 ErrorFound = NotAnAssignmentOp; 6149 NoteLoc = ErrorLoc = FirstBinOp ? FirstBinOp->getOperatorLoc() 6150 : First->getLocStart(); 6151 NoteRange = ErrorRange = FirstBinOp 6152 ? FirstBinOp->getSourceRange() 6153 : SourceRange(ErrorLoc, ErrorLoc); 6154 } else { 6155 auto *SecondBinOp = dyn_cast<BinaryOperator>(Second); 6156 if (!SecondBinOp || SecondBinOp->getOpcode() != BO_Assign) { 6157 ErrorFound = NotAnAssignmentOp; 6158 NoteLoc = ErrorLoc = SecondBinOp 6159 ? SecondBinOp->getOperatorLoc() 6160 : Second->getLocStart(); 6161 NoteRange = ErrorRange = 6162 SecondBinOp ? SecondBinOp->getSourceRange() 6163 : SourceRange(ErrorLoc, ErrorLoc); 6164 } else { 6165 auto *PossibleXRHSInFirst = 6166 FirstBinOp->getRHS()->IgnoreParenImpCasts(); 6167 auto *PossibleXLHSInSecond = 6168 SecondBinOp->getLHS()->IgnoreParenImpCasts(); 6169 llvm::FoldingSetNodeID X1Id, X2Id; 6170 PossibleXRHSInFirst->Profile(X1Id, Context, 6171 /*Canonical=*/true); 6172 PossibleXLHSInSecond->Profile(X2Id, Context, 6173 /*Canonical=*/true); 6174 IsUpdateExprFound = X1Id == X2Id; 6175 if (IsUpdateExprFound) { 6176 V = FirstBinOp->getLHS(); 6177 X = SecondBinOp->getLHS(); 6178 E = SecondBinOp->getRHS(); 6179 UE = nullptr; 6180 IsXLHSInRHSPart = false; 6181 IsPostfixUpdate = true; 6182 } else { 6183 ErrorFound = NotASpecificExpression; 6184 ErrorLoc = FirstBinOp->getExprLoc(); 6185 ErrorRange = FirstBinOp->getSourceRange(); 6186 NoteLoc = SecondBinOp->getLHS()->getExprLoc(); 6187 NoteRange = SecondBinOp->getRHS()->getSourceRange(); 6188 } 6189 } 6190 } 6191 } 6192 } 6193 } else { 6194 NoteLoc = ErrorLoc = Body->getLocStart(); 6195 NoteRange = ErrorRange = 6196 SourceRange(Body->getLocStart(), Body->getLocStart()); 6197 ErrorFound = NotTwoSubstatements; 6198 } 6199 } else { 6200 NoteLoc = ErrorLoc = Body->getLocStart(); 6201 NoteRange = ErrorRange = 6202 SourceRange(Body->getLocStart(), Body->getLocStart()); 6203 ErrorFound = NotACompoundStatement; 6204 } 6205 if (ErrorFound != NoError) { 6206 Diag(ErrorLoc, diag::err_omp_atomic_capture_not_compound_statement) 6207 << ErrorRange; 6208 Diag(NoteLoc, diag::note_omp_atomic_capture) << ErrorFound << NoteRange; 6209 return StmtError(); 6210 } else if (CurContext->isDependentContext()) { 6211 UE = V = E = X = nullptr; 6212 } 6213 } 6214 } 6215 6216 getCurFunction()->setHasBranchProtectedScope(); 6217 6218 return OMPAtomicDirective::Create(Context, StartLoc, EndLoc, Clauses, AStmt, 6219 X, V, E, UE, IsXLHSInRHSPart, 6220 IsPostfixUpdate); 6221 } 6222 6223 StmtResult Sema::ActOnOpenMPTargetDirective(ArrayRef<OMPClause *> Clauses, 6224 Stmt *AStmt, 6225 SourceLocation StartLoc, 6226 SourceLocation EndLoc) { 6227 if (!AStmt) 6228 return StmtError(); 6229 6230 CapturedStmt *CS = cast<CapturedStmt>(AStmt); 6231 // 1.2.2 OpenMP Language Terminology 6232 // Structured block - An executable statement with a single entry at the 6233 // top and a single exit at the bottom. 6234 // The point of exit cannot be a branch out of the structured block. 6235 // longjmp() and throw() must not violate the entry/exit criteria. 6236 CS->getCapturedDecl()->setNothrow(); 6237 6238 // OpenMP [2.16, Nesting of Regions] 6239 // If specified, a teams construct must be contained within a target 6240 // construct. That target construct must contain no statements or directives 6241 // outside of the teams construct. 6242 if (DSAStack->hasInnerTeamsRegion()) { 6243 auto S = AStmt->IgnoreContainers(/*IgnoreCaptured*/ true); 6244 bool OMPTeamsFound = true; 6245 if (auto *CS = dyn_cast<CompoundStmt>(S)) { 6246 auto I = CS->body_begin(); 6247 while (I != CS->body_end()) { 6248 auto *OED = dyn_cast<OMPExecutableDirective>(*I); 6249 if (!OED || !isOpenMPTeamsDirective(OED->getDirectiveKind())) { 6250 OMPTeamsFound = false; 6251 break; 6252 } 6253 ++I; 6254 } 6255 assert(I != CS->body_end() && "Not found statement"); 6256 S = *I; 6257 } else { 6258 auto *OED = dyn_cast<OMPExecutableDirective>(S); 6259 OMPTeamsFound = OED && isOpenMPTeamsDirective(OED->getDirectiveKind()); 6260 } 6261 if (!OMPTeamsFound) { 6262 Diag(StartLoc, diag::err_omp_target_contains_not_only_teams); 6263 Diag(DSAStack->getInnerTeamsRegionLoc(), 6264 diag::note_omp_nested_teams_construct_here); 6265 Diag(S->getLocStart(), diag::note_omp_nested_statement_here) 6266 << isa<OMPExecutableDirective>(S); 6267 return StmtError(); 6268 } 6269 } 6270 6271 getCurFunction()->setHasBranchProtectedScope(); 6272 6273 return OMPTargetDirective::Create(Context, StartLoc, EndLoc, Clauses, AStmt); 6274 } 6275 6276 StmtResult 6277 Sema::ActOnOpenMPTargetParallelDirective(ArrayRef<OMPClause *> Clauses, 6278 Stmt *AStmt, SourceLocation StartLoc, 6279 SourceLocation EndLoc) { 6280 if (!AStmt) 6281 return StmtError(); 6282 6283 CapturedStmt *CS = cast<CapturedStmt>(AStmt); 6284 // 1.2.2 OpenMP Language Terminology 6285 // Structured block - An executable statement with a single entry at the 6286 // top and a single exit at the bottom. 6287 // The point of exit cannot be a branch out of the structured block. 6288 // longjmp() and throw() must not violate the entry/exit criteria. 6289 CS->getCapturedDecl()->setNothrow(); 6290 6291 getCurFunction()->setHasBranchProtectedScope(); 6292 6293 return OMPTargetParallelDirective::Create(Context, StartLoc, EndLoc, Clauses, 6294 AStmt); 6295 } 6296 6297 StmtResult Sema::ActOnOpenMPTargetParallelForDirective( 6298 ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc, 6299 SourceLocation EndLoc, 6300 llvm::DenseMap<ValueDecl *, Expr *> &VarsWithImplicitDSA) { 6301 if (!AStmt) 6302 return StmtError(); 6303 6304 CapturedStmt *CS = cast<CapturedStmt>(AStmt); 6305 // 1.2.2 OpenMP Language Terminology 6306 // Structured block - An executable statement with a single entry at the 6307 // top and a single exit at the bottom. 6308 // The point of exit cannot be a branch out of the structured block. 6309 // longjmp() and throw() must not violate the entry/exit criteria. 6310 CS->getCapturedDecl()->setNothrow(); 6311 6312 OMPLoopDirective::HelperExprs B; 6313 // In presence of clause 'collapse' or 'ordered' with number of loops, it will 6314 // define the nested loops number. 6315 unsigned NestedLoopCount = 6316 CheckOpenMPLoop(OMPD_target_parallel_for, getCollapseNumberExpr(Clauses), 6317 getOrderedNumberExpr(Clauses), AStmt, *this, *DSAStack, 6318 VarsWithImplicitDSA, B); 6319 if (NestedLoopCount == 0) 6320 return StmtError(); 6321 6322 assert((CurContext->isDependentContext() || B.builtAll()) && 6323 "omp target parallel for loop exprs were not built"); 6324 6325 if (!CurContext->isDependentContext()) { 6326 // Finalize the clauses that need pre-built expressions for CodeGen. 6327 for (auto C : Clauses) { 6328 if (auto *LC = dyn_cast<OMPLinearClause>(C)) 6329 if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef), 6330 B.NumIterations, *this, CurScope, 6331 DSAStack)) 6332 return StmtError(); 6333 } 6334 } 6335 6336 getCurFunction()->setHasBranchProtectedScope(); 6337 return OMPTargetParallelForDirective::Create(Context, StartLoc, EndLoc, 6338 NestedLoopCount, Clauses, AStmt, 6339 B, DSAStack->isCancelRegion()); 6340 } 6341 6342 /// Check for existence of a map clause in the list of clauses. 6343 static bool hasClauses(ArrayRef<OMPClause *> Clauses, 6344 const OpenMPClauseKind K) { 6345 return llvm::any_of( 6346 Clauses, [K](const OMPClause *C) { return C->getClauseKind() == K; }); 6347 } 6348 6349 template <typename... Params> 6350 static bool hasClauses(ArrayRef<OMPClause *> Clauses, const OpenMPClauseKind K, 6351 const Params... ClauseTypes) { 6352 return hasClauses(Clauses, K) || hasClauses(Clauses, ClauseTypes...); 6353 } 6354 6355 StmtResult Sema::ActOnOpenMPTargetDataDirective(ArrayRef<OMPClause *> Clauses, 6356 Stmt *AStmt, 6357 SourceLocation StartLoc, 6358 SourceLocation EndLoc) { 6359 if (!AStmt) 6360 return StmtError(); 6361 6362 assert(isa<CapturedStmt>(AStmt) && "Captured statement expected"); 6363 6364 // OpenMP [2.10.1, Restrictions, p. 97] 6365 // At least one map clause must appear on the directive. 6366 if (!hasClauses(Clauses, OMPC_map, OMPC_use_device_ptr)) { 6367 Diag(StartLoc, diag::err_omp_no_clause_for_directive) 6368 << "'map' or 'use_device_ptr'" 6369 << getOpenMPDirectiveName(OMPD_target_data); 6370 return StmtError(); 6371 } 6372 6373 getCurFunction()->setHasBranchProtectedScope(); 6374 6375 return OMPTargetDataDirective::Create(Context, StartLoc, EndLoc, Clauses, 6376 AStmt); 6377 } 6378 6379 StmtResult 6380 Sema::ActOnOpenMPTargetEnterDataDirective(ArrayRef<OMPClause *> Clauses, 6381 SourceLocation StartLoc, 6382 SourceLocation EndLoc) { 6383 // OpenMP [2.10.2, Restrictions, p. 99] 6384 // At least one map clause must appear on the directive. 6385 if (!hasClauses(Clauses, OMPC_map)) { 6386 Diag(StartLoc, diag::err_omp_no_clause_for_directive) 6387 << "'map'" << getOpenMPDirectiveName(OMPD_target_enter_data); 6388 return StmtError(); 6389 } 6390 6391 return OMPTargetEnterDataDirective::Create(Context, StartLoc, EndLoc, 6392 Clauses); 6393 } 6394 6395 StmtResult 6396 Sema::ActOnOpenMPTargetExitDataDirective(ArrayRef<OMPClause *> Clauses, 6397 SourceLocation StartLoc, 6398 SourceLocation EndLoc) { 6399 // OpenMP [2.10.3, Restrictions, p. 102] 6400 // At least one map clause must appear on the directive. 6401 if (!hasClauses(Clauses, OMPC_map)) { 6402 Diag(StartLoc, diag::err_omp_no_clause_for_directive) 6403 << "'map'" << getOpenMPDirectiveName(OMPD_target_exit_data); 6404 return StmtError(); 6405 } 6406 6407 return OMPTargetExitDataDirective::Create(Context, StartLoc, EndLoc, Clauses); 6408 } 6409 6410 StmtResult Sema::ActOnOpenMPTargetUpdateDirective(ArrayRef<OMPClause *> Clauses, 6411 SourceLocation StartLoc, 6412 SourceLocation EndLoc) { 6413 if (!hasClauses(Clauses, OMPC_to, OMPC_from)) { 6414 Diag(StartLoc, diag::err_omp_at_least_one_motion_clause_required); 6415 return StmtError(); 6416 } 6417 return OMPTargetUpdateDirective::Create(Context, StartLoc, EndLoc, Clauses); 6418 } 6419 6420 StmtResult Sema::ActOnOpenMPTeamsDirective(ArrayRef<OMPClause *> Clauses, 6421 Stmt *AStmt, SourceLocation StartLoc, 6422 SourceLocation EndLoc) { 6423 if (!AStmt) 6424 return StmtError(); 6425 6426 CapturedStmt *CS = cast<CapturedStmt>(AStmt); 6427 // 1.2.2 OpenMP Language Terminology 6428 // Structured block - An executable statement with a single entry at the 6429 // top and a single exit at the bottom. 6430 // The point of exit cannot be a branch out of the structured block. 6431 // longjmp() and throw() must not violate the entry/exit criteria. 6432 CS->getCapturedDecl()->setNothrow(); 6433 6434 getCurFunction()->setHasBranchProtectedScope(); 6435 6436 return OMPTeamsDirective::Create(Context, StartLoc, EndLoc, Clauses, AStmt); 6437 } 6438 6439 StmtResult 6440 Sema::ActOnOpenMPCancellationPointDirective(SourceLocation StartLoc, 6441 SourceLocation EndLoc, 6442 OpenMPDirectiveKind CancelRegion) { 6443 if (DSAStack->isParentNowaitRegion()) { 6444 Diag(StartLoc, diag::err_omp_parent_cancel_region_nowait) << 0; 6445 return StmtError(); 6446 } 6447 if (DSAStack->isParentOrderedRegion()) { 6448 Diag(StartLoc, diag::err_omp_parent_cancel_region_ordered) << 0; 6449 return StmtError(); 6450 } 6451 return OMPCancellationPointDirective::Create(Context, StartLoc, EndLoc, 6452 CancelRegion); 6453 } 6454 6455 StmtResult Sema::ActOnOpenMPCancelDirective(ArrayRef<OMPClause *> Clauses, 6456 SourceLocation StartLoc, 6457 SourceLocation EndLoc, 6458 OpenMPDirectiveKind CancelRegion) { 6459 if (DSAStack->isParentNowaitRegion()) { 6460 Diag(StartLoc, diag::err_omp_parent_cancel_region_nowait) << 1; 6461 return StmtError(); 6462 } 6463 if (DSAStack->isParentOrderedRegion()) { 6464 Diag(StartLoc, diag::err_omp_parent_cancel_region_ordered) << 1; 6465 return StmtError(); 6466 } 6467 DSAStack->setParentCancelRegion(/*Cancel=*/true); 6468 return OMPCancelDirective::Create(Context, StartLoc, EndLoc, Clauses, 6469 CancelRegion); 6470 } 6471 6472 static bool checkGrainsizeNumTasksClauses(Sema &S, 6473 ArrayRef<OMPClause *> Clauses) { 6474 OMPClause *PrevClause = nullptr; 6475 bool ErrorFound = false; 6476 for (auto *C : Clauses) { 6477 if (C->getClauseKind() == OMPC_grainsize || 6478 C->getClauseKind() == OMPC_num_tasks) { 6479 if (!PrevClause) 6480 PrevClause = C; 6481 else if (PrevClause->getClauseKind() != C->getClauseKind()) { 6482 S.Diag(C->getLocStart(), 6483 diag::err_omp_grainsize_num_tasks_mutually_exclusive) 6484 << getOpenMPClauseName(C->getClauseKind()) 6485 << getOpenMPClauseName(PrevClause->getClauseKind()); 6486 S.Diag(PrevClause->getLocStart(), 6487 diag::note_omp_previous_grainsize_num_tasks) 6488 << getOpenMPClauseName(PrevClause->getClauseKind()); 6489 ErrorFound = true; 6490 } 6491 } 6492 } 6493 return ErrorFound; 6494 } 6495 6496 static bool checkReductionClauseWithNogroup(Sema &S, 6497 ArrayRef<OMPClause *> Clauses) { 6498 OMPClause *ReductionClause = nullptr; 6499 OMPClause *NogroupClause = nullptr; 6500 for (auto *C : Clauses) { 6501 if (C->getClauseKind() == OMPC_reduction) { 6502 ReductionClause = C; 6503 if (NogroupClause) 6504 break; 6505 continue; 6506 } 6507 if (C->getClauseKind() == OMPC_nogroup) { 6508 NogroupClause = C; 6509 if (ReductionClause) 6510 break; 6511 continue; 6512 } 6513 } 6514 if (ReductionClause && NogroupClause) { 6515 S.Diag(ReductionClause->getLocStart(), diag::err_omp_reduction_with_nogroup) 6516 << SourceRange(NogroupClause->getLocStart(), 6517 NogroupClause->getLocEnd()); 6518 return true; 6519 } 6520 return false; 6521 } 6522 6523 StmtResult Sema::ActOnOpenMPTaskLoopDirective( 6524 ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc, 6525 SourceLocation EndLoc, 6526 llvm::DenseMap<ValueDecl *, Expr *> &VarsWithImplicitDSA) { 6527 if (!AStmt) 6528 return StmtError(); 6529 6530 assert(isa<CapturedStmt>(AStmt) && "Captured statement expected"); 6531 OMPLoopDirective::HelperExprs B; 6532 // In presence of clause 'collapse' or 'ordered' with number of loops, it will 6533 // define the nested loops number. 6534 unsigned NestedLoopCount = 6535 CheckOpenMPLoop(OMPD_taskloop, getCollapseNumberExpr(Clauses), 6536 /*OrderedLoopCountExpr=*/nullptr, AStmt, *this, *DSAStack, 6537 VarsWithImplicitDSA, B); 6538 if (NestedLoopCount == 0) 6539 return StmtError(); 6540 6541 assert((CurContext->isDependentContext() || B.builtAll()) && 6542 "omp for loop exprs were not built"); 6543 6544 // OpenMP, [2.9.2 taskloop Construct, Restrictions] 6545 // The grainsize clause and num_tasks clause are mutually exclusive and may 6546 // not appear on the same taskloop directive. 6547 if (checkGrainsizeNumTasksClauses(*this, Clauses)) 6548 return StmtError(); 6549 // OpenMP, [2.9.2 taskloop Construct, Restrictions] 6550 // If a reduction clause is present on the taskloop directive, the nogroup 6551 // clause must not be specified. 6552 if (checkReductionClauseWithNogroup(*this, Clauses)) 6553 return StmtError(); 6554 6555 getCurFunction()->setHasBranchProtectedScope(); 6556 return OMPTaskLoopDirective::Create(Context, StartLoc, EndLoc, 6557 NestedLoopCount, Clauses, AStmt, B); 6558 } 6559 6560 StmtResult Sema::ActOnOpenMPTaskLoopSimdDirective( 6561 ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc, 6562 SourceLocation EndLoc, 6563 llvm::DenseMap<ValueDecl *, Expr *> &VarsWithImplicitDSA) { 6564 if (!AStmt) 6565 return StmtError(); 6566 6567 assert(isa<CapturedStmt>(AStmt) && "Captured statement expected"); 6568 OMPLoopDirective::HelperExprs B; 6569 // In presence of clause 'collapse' or 'ordered' with number of loops, it will 6570 // define the nested loops number. 6571 unsigned NestedLoopCount = 6572 CheckOpenMPLoop(OMPD_taskloop_simd, getCollapseNumberExpr(Clauses), 6573 /*OrderedLoopCountExpr=*/nullptr, AStmt, *this, *DSAStack, 6574 VarsWithImplicitDSA, B); 6575 if (NestedLoopCount == 0) 6576 return StmtError(); 6577 6578 assert((CurContext->isDependentContext() || B.builtAll()) && 6579 "omp for loop exprs were not built"); 6580 6581 if (!CurContext->isDependentContext()) { 6582 // Finalize the clauses that need pre-built expressions for CodeGen. 6583 for (auto C : Clauses) { 6584 if (auto *LC = dyn_cast<OMPLinearClause>(C)) 6585 if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef), 6586 B.NumIterations, *this, CurScope, 6587 DSAStack)) 6588 return StmtError(); 6589 } 6590 } 6591 6592 // OpenMP, [2.9.2 taskloop Construct, Restrictions] 6593 // The grainsize clause and num_tasks clause are mutually exclusive and may 6594 // not appear on the same taskloop directive. 6595 if (checkGrainsizeNumTasksClauses(*this, Clauses)) 6596 return StmtError(); 6597 // OpenMP, [2.9.2 taskloop Construct, Restrictions] 6598 // If a reduction clause is present on the taskloop directive, the nogroup 6599 // clause must not be specified. 6600 if (checkReductionClauseWithNogroup(*this, Clauses)) 6601 return StmtError(); 6602 6603 getCurFunction()->setHasBranchProtectedScope(); 6604 return OMPTaskLoopSimdDirective::Create(Context, StartLoc, EndLoc, 6605 NestedLoopCount, Clauses, AStmt, B); 6606 } 6607 6608 StmtResult Sema::ActOnOpenMPDistributeDirective( 6609 ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc, 6610 SourceLocation EndLoc, 6611 llvm::DenseMap<ValueDecl *, Expr *> &VarsWithImplicitDSA) { 6612 if (!AStmt) 6613 return StmtError(); 6614 6615 assert(isa<CapturedStmt>(AStmt) && "Captured statement expected"); 6616 OMPLoopDirective::HelperExprs B; 6617 // In presence of clause 'collapse' with number of loops, it will 6618 // define the nested loops number. 6619 unsigned NestedLoopCount = 6620 CheckOpenMPLoop(OMPD_distribute, getCollapseNumberExpr(Clauses), 6621 nullptr /*ordered not a clause on distribute*/, AStmt, 6622 *this, *DSAStack, VarsWithImplicitDSA, B); 6623 if (NestedLoopCount == 0) 6624 return StmtError(); 6625 6626 assert((CurContext->isDependentContext() || B.builtAll()) && 6627 "omp for loop exprs were not built"); 6628 6629 getCurFunction()->setHasBranchProtectedScope(); 6630 return OMPDistributeDirective::Create(Context, StartLoc, EndLoc, 6631 NestedLoopCount, Clauses, AStmt, B); 6632 } 6633 6634 StmtResult Sema::ActOnOpenMPDistributeParallelForDirective( 6635 ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc, 6636 SourceLocation EndLoc, 6637 llvm::DenseMap<ValueDecl *, Expr *> &VarsWithImplicitDSA) { 6638 if (!AStmt) 6639 return StmtError(); 6640 6641 CapturedStmt *CS = cast<CapturedStmt>(AStmt); 6642 // 1.2.2 OpenMP Language Terminology 6643 // Structured block - An executable statement with a single entry at the 6644 // top and a single exit at the bottom. 6645 // The point of exit cannot be a branch out of the structured block. 6646 // longjmp() and throw() must not violate the entry/exit criteria. 6647 CS->getCapturedDecl()->setNothrow(); 6648 6649 OMPLoopDirective::HelperExprs B; 6650 // In presence of clause 'collapse' with number of loops, it will 6651 // define the nested loops number. 6652 unsigned NestedLoopCount = CheckOpenMPLoop( 6653 OMPD_distribute_parallel_for, getCollapseNumberExpr(Clauses), 6654 nullptr /*ordered not a clause on distribute*/, AStmt, *this, *DSAStack, 6655 VarsWithImplicitDSA, B); 6656 if (NestedLoopCount == 0) 6657 return StmtError(); 6658 6659 assert((CurContext->isDependentContext() || B.builtAll()) && 6660 "omp for loop exprs were not built"); 6661 6662 getCurFunction()->setHasBranchProtectedScope(); 6663 return OMPDistributeParallelForDirective::Create( 6664 Context, StartLoc, EndLoc, NestedLoopCount, Clauses, AStmt, B); 6665 } 6666 6667 StmtResult Sema::ActOnOpenMPDistributeParallelForSimdDirective( 6668 ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc, 6669 SourceLocation EndLoc, 6670 llvm::DenseMap<ValueDecl *, Expr *> &VarsWithImplicitDSA) { 6671 if (!AStmt) 6672 return StmtError(); 6673 6674 CapturedStmt *CS = cast<CapturedStmt>(AStmt); 6675 // 1.2.2 OpenMP Language Terminology 6676 // Structured block - An executable statement with a single entry at the 6677 // top and a single exit at the bottom. 6678 // The point of exit cannot be a branch out of the structured block. 6679 // longjmp() and throw() must not violate the entry/exit criteria. 6680 CS->getCapturedDecl()->setNothrow(); 6681 6682 OMPLoopDirective::HelperExprs B; 6683 // In presence of clause 'collapse' with number of loops, it will 6684 // define the nested loops number. 6685 unsigned NestedLoopCount = CheckOpenMPLoop( 6686 OMPD_distribute_parallel_for_simd, getCollapseNumberExpr(Clauses), 6687 nullptr /*ordered not a clause on distribute*/, AStmt, *this, *DSAStack, 6688 VarsWithImplicitDSA, B); 6689 if (NestedLoopCount == 0) 6690 return StmtError(); 6691 6692 assert((CurContext->isDependentContext() || B.builtAll()) && 6693 "omp for loop exprs were not built"); 6694 6695 if (checkSimdlenSafelenSpecified(*this, Clauses)) 6696 return StmtError(); 6697 6698 getCurFunction()->setHasBranchProtectedScope(); 6699 return OMPDistributeParallelForSimdDirective::Create( 6700 Context, StartLoc, EndLoc, NestedLoopCount, Clauses, AStmt, B); 6701 } 6702 6703 StmtResult Sema::ActOnOpenMPDistributeSimdDirective( 6704 ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc, 6705 SourceLocation EndLoc, 6706 llvm::DenseMap<ValueDecl *, Expr *> &VarsWithImplicitDSA) { 6707 if (!AStmt) 6708 return StmtError(); 6709 6710 CapturedStmt *CS = cast<CapturedStmt>(AStmt); 6711 // 1.2.2 OpenMP Language Terminology 6712 // Structured block - An executable statement with a single entry at the 6713 // top and a single exit at the bottom. 6714 // The point of exit cannot be a branch out of the structured block. 6715 // longjmp() and throw() must not violate the entry/exit criteria. 6716 CS->getCapturedDecl()->setNothrow(); 6717 6718 OMPLoopDirective::HelperExprs B; 6719 // In presence of clause 'collapse' with number of loops, it will 6720 // define the nested loops number. 6721 unsigned NestedLoopCount = 6722 CheckOpenMPLoop(OMPD_distribute_simd, getCollapseNumberExpr(Clauses), 6723 nullptr /*ordered not a clause on distribute*/, AStmt, 6724 *this, *DSAStack, VarsWithImplicitDSA, B); 6725 if (NestedLoopCount == 0) 6726 return StmtError(); 6727 6728 assert((CurContext->isDependentContext() || B.builtAll()) && 6729 "omp for loop exprs were not built"); 6730 6731 if (checkSimdlenSafelenSpecified(*this, Clauses)) 6732 return StmtError(); 6733 6734 getCurFunction()->setHasBranchProtectedScope(); 6735 return OMPDistributeSimdDirective::Create(Context, StartLoc, EndLoc, 6736 NestedLoopCount, Clauses, AStmt, B); 6737 } 6738 6739 StmtResult Sema::ActOnOpenMPTargetParallelForSimdDirective( 6740 ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc, 6741 SourceLocation EndLoc, 6742 llvm::DenseMap<ValueDecl *, Expr *> &VarsWithImplicitDSA) { 6743 if (!AStmt) 6744 return StmtError(); 6745 6746 CapturedStmt *CS = cast<CapturedStmt>(AStmt); 6747 // 1.2.2 OpenMP Language Terminology 6748 // Structured block - An executable statement with a single entry at the 6749 // top and a single exit at the bottom. 6750 // The point of exit cannot be a branch out of the structured block. 6751 // longjmp() and throw() must not violate the entry/exit criteria. 6752 CS->getCapturedDecl()->setNothrow(); 6753 6754 OMPLoopDirective::HelperExprs B; 6755 // In presence of clause 'collapse' or 'ordered' with number of loops, it will 6756 // define the nested loops number. 6757 unsigned NestedLoopCount = CheckOpenMPLoop( 6758 OMPD_target_parallel_for_simd, getCollapseNumberExpr(Clauses), 6759 getOrderedNumberExpr(Clauses), AStmt, *this, *DSAStack, 6760 VarsWithImplicitDSA, B); 6761 if (NestedLoopCount == 0) 6762 return StmtError(); 6763 6764 assert((CurContext->isDependentContext() || B.builtAll()) && 6765 "omp target parallel for simd loop exprs were not built"); 6766 6767 if (!CurContext->isDependentContext()) { 6768 // Finalize the clauses that need pre-built expressions for CodeGen. 6769 for (auto C : Clauses) { 6770 if (auto *LC = dyn_cast<OMPLinearClause>(C)) 6771 if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef), 6772 B.NumIterations, *this, CurScope, 6773 DSAStack)) 6774 return StmtError(); 6775 } 6776 } 6777 if (checkSimdlenSafelenSpecified(*this, Clauses)) 6778 return StmtError(); 6779 6780 getCurFunction()->setHasBranchProtectedScope(); 6781 return OMPTargetParallelForSimdDirective::Create( 6782 Context, StartLoc, EndLoc, NestedLoopCount, Clauses, AStmt, B); 6783 } 6784 6785 StmtResult Sema::ActOnOpenMPTargetSimdDirective( 6786 ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc, 6787 SourceLocation EndLoc, 6788 llvm::DenseMap<ValueDecl *, Expr *> &VarsWithImplicitDSA) { 6789 if (!AStmt) 6790 return StmtError(); 6791 6792 CapturedStmt *CS = cast<CapturedStmt>(AStmt); 6793 // 1.2.2 OpenMP Language Terminology 6794 // Structured block - An executable statement with a single entry at the 6795 // top and a single exit at the bottom. 6796 // The point of exit cannot be a branch out of the structured block. 6797 // longjmp() and throw() must not violate the entry/exit criteria. 6798 CS->getCapturedDecl()->setNothrow(); 6799 6800 OMPLoopDirective::HelperExprs B; 6801 // In presence of clause 'collapse' with number of loops, it will define the 6802 // nested loops number. 6803 unsigned NestedLoopCount = 6804 CheckOpenMPLoop(OMPD_target_simd, getCollapseNumberExpr(Clauses), 6805 getOrderedNumberExpr(Clauses), AStmt, *this, *DSAStack, 6806 VarsWithImplicitDSA, B); 6807 if (NestedLoopCount == 0) 6808 return StmtError(); 6809 6810 assert((CurContext->isDependentContext() || B.builtAll()) && 6811 "omp target simd loop exprs were not built"); 6812 6813 if (!CurContext->isDependentContext()) { 6814 // Finalize the clauses that need pre-built expressions for CodeGen. 6815 for (auto C : Clauses) { 6816 if (auto *LC = dyn_cast<OMPLinearClause>(C)) 6817 if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef), 6818 B.NumIterations, *this, CurScope, 6819 DSAStack)) 6820 return StmtError(); 6821 } 6822 } 6823 6824 if (checkSimdlenSafelenSpecified(*this, Clauses)) 6825 return StmtError(); 6826 6827 getCurFunction()->setHasBranchProtectedScope(); 6828 return OMPTargetSimdDirective::Create(Context, StartLoc, EndLoc, 6829 NestedLoopCount, Clauses, AStmt, B); 6830 } 6831 6832 StmtResult Sema::ActOnOpenMPTeamsDistributeDirective( 6833 ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc, 6834 SourceLocation EndLoc, 6835 llvm::DenseMap<ValueDecl *, Expr *> &VarsWithImplicitDSA) { 6836 if (!AStmt) 6837 return StmtError(); 6838 6839 CapturedStmt *CS = cast<CapturedStmt>(AStmt); 6840 // 1.2.2 OpenMP Language Terminology 6841 // Structured block - An executable statement with a single entry at the 6842 // top and a single exit at the bottom. 6843 // The point of exit cannot be a branch out of the structured block. 6844 // longjmp() and throw() must not violate the entry/exit criteria. 6845 CS->getCapturedDecl()->setNothrow(); 6846 6847 OMPLoopDirective::HelperExprs B; 6848 // In presence of clause 'collapse' with number of loops, it will 6849 // define the nested loops number. 6850 unsigned NestedLoopCount = 6851 CheckOpenMPLoop(OMPD_teams_distribute, getCollapseNumberExpr(Clauses), 6852 nullptr /*ordered not a clause on distribute*/, AStmt, 6853 *this, *DSAStack, VarsWithImplicitDSA, B); 6854 if (NestedLoopCount == 0) 6855 return StmtError(); 6856 6857 assert((CurContext->isDependentContext() || B.builtAll()) && 6858 "omp teams distribute loop exprs were not built"); 6859 6860 getCurFunction()->setHasBranchProtectedScope(); 6861 return OMPTeamsDistributeDirective::Create( 6862 Context, StartLoc, EndLoc, NestedLoopCount, Clauses, AStmt, B); 6863 } 6864 6865 StmtResult Sema::ActOnOpenMPTeamsDistributeSimdDirective( 6866 ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc, 6867 SourceLocation EndLoc, 6868 llvm::DenseMap<ValueDecl *, Expr *> &VarsWithImplicitDSA) { 6869 if (!AStmt) 6870 return StmtError(); 6871 6872 CapturedStmt *CS = cast<CapturedStmt>(AStmt); 6873 // 1.2.2 OpenMP Language Terminology 6874 // Structured block - An executable statement with a single entry at the 6875 // top and a single exit at the bottom. 6876 // The point of exit cannot be a branch out of the structured block. 6877 // longjmp() and throw() must not violate the entry/exit criteria. 6878 CS->getCapturedDecl()->setNothrow(); 6879 6880 OMPLoopDirective::HelperExprs B; 6881 // In presence of clause 'collapse' with number of loops, it will 6882 // define the nested loops number. 6883 unsigned NestedLoopCount = CheckOpenMPLoop( 6884 OMPD_teams_distribute_simd, getCollapseNumberExpr(Clauses), 6885 nullptr /*ordered not a clause on distribute*/, AStmt, *this, *DSAStack, 6886 VarsWithImplicitDSA, B); 6887 6888 if (NestedLoopCount == 0) 6889 return StmtError(); 6890 6891 assert((CurContext->isDependentContext() || B.builtAll()) && 6892 "omp teams distribute simd loop exprs were not built"); 6893 6894 if (!CurContext->isDependentContext()) { 6895 // Finalize the clauses that need pre-built expressions for CodeGen. 6896 for (auto C : Clauses) { 6897 if (auto *LC = dyn_cast<OMPLinearClause>(C)) 6898 if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef), 6899 B.NumIterations, *this, CurScope, 6900 DSAStack)) 6901 return StmtError(); 6902 } 6903 } 6904 6905 if (checkSimdlenSafelenSpecified(*this, Clauses)) 6906 return StmtError(); 6907 6908 getCurFunction()->setHasBranchProtectedScope(); 6909 return OMPTeamsDistributeSimdDirective::Create( 6910 Context, StartLoc, EndLoc, NestedLoopCount, Clauses, AStmt, B); 6911 } 6912 6913 StmtResult Sema::ActOnOpenMPTeamsDistributeParallelForSimdDirective( 6914 ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc, 6915 SourceLocation EndLoc, 6916 llvm::DenseMap<ValueDecl *, Expr *> &VarsWithImplicitDSA) { 6917 if (!AStmt) 6918 return StmtError(); 6919 6920 CapturedStmt *CS = cast<CapturedStmt>(AStmt); 6921 // 1.2.2 OpenMP Language Terminology 6922 // Structured block - An executable statement with a single entry at the 6923 // top and a single exit at the bottom. 6924 // The point of exit cannot be a branch out of the structured block. 6925 // longjmp() and throw() must not violate the entry/exit criteria. 6926 CS->getCapturedDecl()->setNothrow(); 6927 6928 OMPLoopDirective::HelperExprs B; 6929 // In presence of clause 'collapse' with number of loops, it will 6930 // define the nested loops number. 6931 auto NestedLoopCount = CheckOpenMPLoop( 6932 OMPD_teams_distribute_parallel_for_simd, getCollapseNumberExpr(Clauses), 6933 nullptr /*ordered not a clause on distribute*/, AStmt, *this, *DSAStack, 6934 VarsWithImplicitDSA, B); 6935 6936 if (NestedLoopCount == 0) 6937 return StmtError(); 6938 6939 assert((CurContext->isDependentContext() || B.builtAll()) && 6940 "omp for loop exprs were not built"); 6941 6942 if (!CurContext->isDependentContext()) { 6943 // Finalize the clauses that need pre-built expressions for CodeGen. 6944 for (auto C : Clauses) { 6945 if (auto *LC = dyn_cast<OMPLinearClause>(C)) 6946 if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef), 6947 B.NumIterations, *this, CurScope, 6948 DSAStack)) 6949 return StmtError(); 6950 } 6951 } 6952 6953 if (checkSimdlenSafelenSpecified(*this, Clauses)) 6954 return StmtError(); 6955 6956 getCurFunction()->setHasBranchProtectedScope(); 6957 return OMPTeamsDistributeParallelForSimdDirective::Create( 6958 Context, StartLoc, EndLoc, NestedLoopCount, Clauses, AStmt, B); 6959 } 6960 6961 StmtResult Sema::ActOnOpenMPTeamsDistributeParallelForDirective( 6962 ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc, 6963 SourceLocation EndLoc, 6964 llvm::DenseMap<ValueDecl *, Expr *> &VarsWithImplicitDSA) { 6965 if (!AStmt) 6966 return StmtError(); 6967 6968 CapturedStmt *CS = cast<CapturedStmt>(AStmt); 6969 // 1.2.2 OpenMP Language Terminology 6970 // Structured block - An executable statement with a single entry at the 6971 // top and a single exit at the bottom. 6972 // The point of exit cannot be a branch out of the structured block. 6973 // longjmp() and throw() must not violate the entry/exit criteria. 6974 CS->getCapturedDecl()->setNothrow(); 6975 6976 OMPLoopDirective::HelperExprs B; 6977 // In presence of clause 'collapse' with number of loops, it will 6978 // define the nested loops number. 6979 unsigned NestedLoopCount = CheckOpenMPLoop( 6980 OMPD_teams_distribute_parallel_for, getCollapseNumberExpr(Clauses), 6981 nullptr /*ordered not a clause on distribute*/, AStmt, *this, *DSAStack, 6982 VarsWithImplicitDSA, B); 6983 6984 if (NestedLoopCount == 0) 6985 return StmtError(); 6986 6987 assert((CurContext->isDependentContext() || B.builtAll()) && 6988 "omp for loop exprs were not built"); 6989 6990 if (!CurContext->isDependentContext()) { 6991 // Finalize the clauses that need pre-built expressions for CodeGen. 6992 for (auto C : Clauses) { 6993 if (auto *LC = dyn_cast<OMPLinearClause>(C)) 6994 if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef), 6995 B.NumIterations, *this, CurScope, 6996 DSAStack)) 6997 return StmtError(); 6998 } 6999 } 7000 7001 getCurFunction()->setHasBranchProtectedScope(); 7002 return OMPTeamsDistributeParallelForDirective::Create( 7003 Context, StartLoc, EndLoc, NestedLoopCount, Clauses, AStmt, B); 7004 } 7005 7006 StmtResult Sema::ActOnOpenMPTargetTeamsDirective(ArrayRef<OMPClause *> Clauses, 7007 Stmt *AStmt, 7008 SourceLocation StartLoc, 7009 SourceLocation EndLoc) { 7010 if (!AStmt) 7011 return StmtError(); 7012 7013 CapturedStmt *CS = cast<CapturedStmt>(AStmt); 7014 // 1.2.2 OpenMP Language Terminology 7015 // Structured block - An executable statement with a single entry at the 7016 // top and a single exit at the bottom. 7017 // The point of exit cannot be a branch out of the structured block. 7018 // longjmp() and throw() must not violate the entry/exit criteria. 7019 CS->getCapturedDecl()->setNothrow(); 7020 7021 getCurFunction()->setHasBranchProtectedScope(); 7022 7023 return OMPTargetTeamsDirective::Create(Context, StartLoc, EndLoc, Clauses, 7024 AStmt); 7025 } 7026 7027 StmtResult Sema::ActOnOpenMPTargetTeamsDistributeDirective( 7028 ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc, 7029 SourceLocation EndLoc, 7030 llvm::DenseMap<ValueDecl *, Expr *> &VarsWithImplicitDSA) { 7031 if (!AStmt) 7032 return StmtError(); 7033 7034 CapturedStmt *CS = cast<CapturedStmt>(AStmt); 7035 // 1.2.2 OpenMP Language Terminology 7036 // Structured block - An executable statement with a single entry at the 7037 // top and a single exit at the bottom. 7038 // The point of exit cannot be a branch out of the structured block. 7039 // longjmp() and throw() must not violate the entry/exit criteria. 7040 CS->getCapturedDecl()->setNothrow(); 7041 7042 OMPLoopDirective::HelperExprs B; 7043 // In presence of clause 'collapse' with number of loops, it will 7044 // define the nested loops number. 7045 auto NestedLoopCount = CheckOpenMPLoop( 7046 OMPD_target_teams_distribute, 7047 getCollapseNumberExpr(Clauses), 7048 nullptr /*ordered not a clause on distribute*/, AStmt, *this, *DSAStack, 7049 VarsWithImplicitDSA, B); 7050 if (NestedLoopCount == 0) 7051 return StmtError(); 7052 7053 assert((CurContext->isDependentContext() || B.builtAll()) && 7054 "omp target teams distribute loop exprs were not built"); 7055 7056 getCurFunction()->setHasBranchProtectedScope(); 7057 return OMPTargetTeamsDistributeDirective::Create( 7058 Context, StartLoc, EndLoc, NestedLoopCount, Clauses, AStmt, B); 7059 } 7060 7061 StmtResult Sema::ActOnOpenMPTargetTeamsDistributeParallelForDirective( 7062 ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc, 7063 SourceLocation EndLoc, 7064 llvm::DenseMap<ValueDecl *, Expr *> &VarsWithImplicitDSA) { 7065 if (!AStmt) 7066 return StmtError(); 7067 7068 CapturedStmt *CS = cast<CapturedStmt>(AStmt); 7069 // 1.2.2 OpenMP Language Terminology 7070 // Structured block - An executable statement with a single entry at the 7071 // top and a single exit at the bottom. 7072 // The point of exit cannot be a branch out of the structured block. 7073 // longjmp() and throw() must not violate the entry/exit criteria. 7074 CS->getCapturedDecl()->setNothrow(); 7075 7076 OMPLoopDirective::HelperExprs B; 7077 // In presence of clause 'collapse' with number of loops, it will 7078 // define the nested loops number. 7079 auto NestedLoopCount = CheckOpenMPLoop( 7080 OMPD_target_teams_distribute_parallel_for, 7081 getCollapseNumberExpr(Clauses), 7082 nullptr /*ordered not a clause on distribute*/, AStmt, *this, *DSAStack, 7083 VarsWithImplicitDSA, B); 7084 if (NestedLoopCount == 0) 7085 return StmtError(); 7086 7087 assert((CurContext->isDependentContext() || B.builtAll()) && 7088 "omp target teams distribute parallel for loop exprs were not built"); 7089 7090 if (!CurContext->isDependentContext()) { 7091 // Finalize the clauses that need pre-built expressions for CodeGen. 7092 for (auto C : Clauses) { 7093 if (auto *LC = dyn_cast<OMPLinearClause>(C)) 7094 if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef), 7095 B.NumIterations, *this, CurScope, 7096 DSAStack)) 7097 return StmtError(); 7098 } 7099 } 7100 7101 getCurFunction()->setHasBranchProtectedScope(); 7102 return OMPTargetTeamsDistributeParallelForDirective::Create( 7103 Context, StartLoc, EndLoc, NestedLoopCount, Clauses, AStmt, B); 7104 } 7105 7106 StmtResult Sema::ActOnOpenMPTargetTeamsDistributeParallelForSimdDirective( 7107 ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc, 7108 SourceLocation EndLoc, 7109 llvm::DenseMap<ValueDecl *, Expr *> &VarsWithImplicitDSA) { 7110 if (!AStmt) 7111 return StmtError(); 7112 7113 CapturedStmt *CS = cast<CapturedStmt>(AStmt); 7114 // 1.2.2 OpenMP Language Terminology 7115 // Structured block - An executable statement with a single entry at the 7116 // top and a single exit at the bottom. 7117 // The point of exit cannot be a branch out of the structured block. 7118 // longjmp() and throw() must not violate the entry/exit criteria. 7119 CS->getCapturedDecl()->setNothrow(); 7120 7121 OMPLoopDirective::HelperExprs B; 7122 // In presence of clause 'collapse' with number of loops, it will 7123 // define the nested loops number. 7124 auto NestedLoopCount = CheckOpenMPLoop( 7125 OMPD_target_teams_distribute_parallel_for_simd, 7126 getCollapseNumberExpr(Clauses), 7127 nullptr /*ordered not a clause on distribute*/, AStmt, *this, *DSAStack, 7128 VarsWithImplicitDSA, B); 7129 if (NestedLoopCount == 0) 7130 return StmtError(); 7131 7132 assert((CurContext->isDependentContext() || B.builtAll()) && 7133 "omp target teams distribute parallel for simd loop exprs were not " 7134 "built"); 7135 7136 if (!CurContext->isDependentContext()) { 7137 // Finalize the clauses that need pre-built expressions for CodeGen. 7138 for (auto C : Clauses) { 7139 if (auto *LC = dyn_cast<OMPLinearClause>(C)) 7140 if (FinishOpenMPLinearClause(*LC, cast<DeclRefExpr>(B.IterationVarRef), 7141 B.NumIterations, *this, CurScope, 7142 DSAStack)) 7143 return StmtError(); 7144 } 7145 } 7146 7147 getCurFunction()->setHasBranchProtectedScope(); 7148 return OMPTargetTeamsDistributeParallelForSimdDirective::Create( 7149 Context, StartLoc, EndLoc, NestedLoopCount, Clauses, AStmt, B); 7150 } 7151 7152 StmtResult Sema::ActOnOpenMPTargetTeamsDistributeSimdDirective( 7153 ArrayRef<OMPClause *> Clauses, Stmt *AStmt, SourceLocation StartLoc, 7154 SourceLocation EndLoc, 7155 llvm::DenseMap<ValueDecl *, Expr *> &VarsWithImplicitDSA) { 7156 if (!AStmt) 7157 return StmtError(); 7158 7159 auto *CS = cast<CapturedStmt>(AStmt); 7160 // 1.2.2 OpenMP Language Terminology 7161 // Structured block - An executable statement with a single entry at the 7162 // top and a single exit at the bottom. 7163 // The point of exit cannot be a branch out of the structured block. 7164 // longjmp() and throw() must not violate the entry/exit criteria. 7165 CS->getCapturedDecl()->setNothrow(); 7166 7167 OMPLoopDirective::HelperExprs B; 7168 // In presence of clause 'collapse' with number of loops, it will 7169 // define the nested loops number. 7170 auto NestedLoopCount = CheckOpenMPLoop( 7171 OMPD_target_teams_distribute_simd, getCollapseNumberExpr(Clauses), 7172 nullptr /*ordered not a clause on distribute*/, AStmt, *this, *DSAStack, 7173 VarsWithImplicitDSA, B); 7174 if (NestedLoopCount == 0) 7175 return StmtError(); 7176 7177 assert((CurContext->isDependentContext() || B.builtAll()) && 7178 "omp target teams distribute simd loop exprs were not built"); 7179 7180 getCurFunction()->setHasBranchProtectedScope(); 7181 return OMPTargetTeamsDistributeSimdDirective::Create( 7182 Context, StartLoc, EndLoc, NestedLoopCount, Clauses, AStmt, B); 7183 } 7184 7185 OMPClause *Sema::ActOnOpenMPSingleExprClause(OpenMPClauseKind Kind, Expr *Expr, 7186 SourceLocation StartLoc, 7187 SourceLocation LParenLoc, 7188 SourceLocation EndLoc) { 7189 OMPClause *Res = nullptr; 7190 switch (Kind) { 7191 case OMPC_final: 7192 Res = ActOnOpenMPFinalClause(Expr, StartLoc, LParenLoc, EndLoc); 7193 break; 7194 case OMPC_num_threads: 7195 Res = ActOnOpenMPNumThreadsClause(Expr, StartLoc, LParenLoc, EndLoc); 7196 break; 7197 case OMPC_safelen: 7198 Res = ActOnOpenMPSafelenClause(Expr, StartLoc, LParenLoc, EndLoc); 7199 break; 7200 case OMPC_simdlen: 7201 Res = ActOnOpenMPSimdlenClause(Expr, StartLoc, LParenLoc, EndLoc); 7202 break; 7203 case OMPC_collapse: 7204 Res = ActOnOpenMPCollapseClause(Expr, StartLoc, LParenLoc, EndLoc); 7205 break; 7206 case OMPC_ordered: 7207 Res = ActOnOpenMPOrderedClause(StartLoc, EndLoc, LParenLoc, Expr); 7208 break; 7209 case OMPC_device: 7210 Res = ActOnOpenMPDeviceClause(Expr, StartLoc, LParenLoc, EndLoc); 7211 break; 7212 case OMPC_num_teams: 7213 Res = ActOnOpenMPNumTeamsClause(Expr, StartLoc, LParenLoc, EndLoc); 7214 break; 7215 case OMPC_thread_limit: 7216 Res = ActOnOpenMPThreadLimitClause(Expr, StartLoc, LParenLoc, EndLoc); 7217 break; 7218 case OMPC_priority: 7219 Res = ActOnOpenMPPriorityClause(Expr, StartLoc, LParenLoc, EndLoc); 7220 break; 7221 case OMPC_grainsize: 7222 Res = ActOnOpenMPGrainsizeClause(Expr, StartLoc, LParenLoc, EndLoc); 7223 break; 7224 case OMPC_num_tasks: 7225 Res = ActOnOpenMPNumTasksClause(Expr, StartLoc, LParenLoc, EndLoc); 7226 break; 7227 case OMPC_hint: 7228 Res = ActOnOpenMPHintClause(Expr, StartLoc, LParenLoc, EndLoc); 7229 break; 7230 case OMPC_if: 7231 case OMPC_default: 7232 case OMPC_proc_bind: 7233 case OMPC_schedule: 7234 case OMPC_private: 7235 case OMPC_firstprivate: 7236 case OMPC_lastprivate: 7237 case OMPC_shared: 7238 case OMPC_reduction: 7239 case OMPC_task_reduction: 7240 case OMPC_in_reduction: 7241 case OMPC_linear: 7242 case OMPC_aligned: 7243 case OMPC_copyin: 7244 case OMPC_copyprivate: 7245 case OMPC_nowait: 7246 case OMPC_untied: 7247 case OMPC_mergeable: 7248 case OMPC_threadprivate: 7249 case OMPC_flush: 7250 case OMPC_read: 7251 case OMPC_write: 7252 case OMPC_update: 7253 case OMPC_capture: 7254 case OMPC_seq_cst: 7255 case OMPC_depend: 7256 case OMPC_threads: 7257 case OMPC_simd: 7258 case OMPC_map: 7259 case OMPC_nogroup: 7260 case OMPC_dist_schedule: 7261 case OMPC_defaultmap: 7262 case OMPC_unknown: 7263 case OMPC_uniform: 7264 case OMPC_to: 7265 case OMPC_from: 7266 case OMPC_use_device_ptr: 7267 case OMPC_is_device_ptr: 7268 llvm_unreachable("Clause is not allowed."); 7269 } 7270 return Res; 7271 } 7272 7273 // An OpenMP directive such as 'target parallel' has two captured regions: 7274 // for the 'target' and 'parallel' respectively. This function returns 7275 // the region in which to capture expressions associated with a clause. 7276 // A return value of OMPD_unknown signifies that the expression should not 7277 // be captured. 7278 static OpenMPDirectiveKind getOpenMPCaptureRegionForClause( 7279 OpenMPDirectiveKind DKind, OpenMPClauseKind CKind, 7280 OpenMPDirectiveKind NameModifier = OMPD_unknown) { 7281 OpenMPDirectiveKind CaptureRegion = OMPD_unknown; 7282 7283 switch (CKind) { 7284 case OMPC_if: 7285 switch (DKind) { 7286 case OMPD_target_parallel: 7287 // If this clause applies to the nested 'parallel' region, capture within 7288 // the 'target' region, otherwise do not capture. 7289 if (NameModifier == OMPD_unknown || NameModifier == OMPD_parallel) 7290 CaptureRegion = OMPD_target; 7291 break; 7292 case OMPD_cancel: 7293 case OMPD_parallel: 7294 case OMPD_parallel_sections: 7295 case OMPD_parallel_for: 7296 case OMPD_parallel_for_simd: 7297 case OMPD_target: 7298 case OMPD_target_simd: 7299 case OMPD_target_parallel_for: 7300 case OMPD_target_parallel_for_simd: 7301 case OMPD_target_teams: 7302 case OMPD_target_teams_distribute: 7303 case OMPD_target_teams_distribute_simd: 7304 case OMPD_target_teams_distribute_parallel_for: 7305 case OMPD_target_teams_distribute_parallel_for_simd: 7306 case OMPD_teams_distribute_parallel_for: 7307 case OMPD_teams_distribute_parallel_for_simd: 7308 case OMPD_distribute_parallel_for: 7309 case OMPD_distribute_parallel_for_simd: 7310 case OMPD_task: 7311 case OMPD_taskloop: 7312 case OMPD_taskloop_simd: 7313 case OMPD_target_data: 7314 case OMPD_target_enter_data: 7315 case OMPD_target_exit_data: 7316 case OMPD_target_update: 7317 // Do not capture if-clause expressions. 7318 break; 7319 case OMPD_threadprivate: 7320 case OMPD_taskyield: 7321 case OMPD_barrier: 7322 case OMPD_taskwait: 7323 case OMPD_cancellation_point: 7324 case OMPD_flush: 7325 case OMPD_declare_reduction: 7326 case OMPD_declare_simd: 7327 case OMPD_declare_target: 7328 case OMPD_end_declare_target: 7329 case OMPD_teams: 7330 case OMPD_simd: 7331 case OMPD_for: 7332 case OMPD_for_simd: 7333 case OMPD_sections: 7334 case OMPD_section: 7335 case OMPD_single: 7336 case OMPD_master: 7337 case OMPD_critical: 7338 case OMPD_taskgroup: 7339 case OMPD_distribute: 7340 case OMPD_ordered: 7341 case OMPD_atomic: 7342 case OMPD_distribute_simd: 7343 case OMPD_teams_distribute: 7344 case OMPD_teams_distribute_simd: 7345 llvm_unreachable("Unexpected OpenMP directive with if-clause"); 7346 case OMPD_unknown: 7347 llvm_unreachable("Unknown OpenMP directive"); 7348 } 7349 break; 7350 case OMPC_num_threads: 7351 switch (DKind) { 7352 case OMPD_target_parallel: 7353 CaptureRegion = OMPD_target; 7354 break; 7355 case OMPD_cancel: 7356 case OMPD_parallel: 7357 case OMPD_parallel_sections: 7358 case OMPD_parallel_for: 7359 case OMPD_parallel_for_simd: 7360 case OMPD_target: 7361 case OMPD_target_simd: 7362 case OMPD_target_parallel_for: 7363 case OMPD_target_parallel_for_simd: 7364 case OMPD_target_teams: 7365 case OMPD_target_teams_distribute: 7366 case OMPD_target_teams_distribute_simd: 7367 case OMPD_target_teams_distribute_parallel_for: 7368 case OMPD_target_teams_distribute_parallel_for_simd: 7369 case OMPD_teams_distribute_parallel_for: 7370 case OMPD_teams_distribute_parallel_for_simd: 7371 case OMPD_distribute_parallel_for: 7372 case OMPD_distribute_parallel_for_simd: 7373 case OMPD_task: 7374 case OMPD_taskloop: 7375 case OMPD_taskloop_simd: 7376 case OMPD_target_data: 7377 case OMPD_target_enter_data: 7378 case OMPD_target_exit_data: 7379 case OMPD_target_update: 7380 // Do not capture num_threads-clause expressions. 7381 break; 7382 case OMPD_threadprivate: 7383 case OMPD_taskyield: 7384 case OMPD_barrier: 7385 case OMPD_taskwait: 7386 case OMPD_cancellation_point: 7387 case OMPD_flush: 7388 case OMPD_declare_reduction: 7389 case OMPD_declare_simd: 7390 case OMPD_declare_target: 7391 case OMPD_end_declare_target: 7392 case OMPD_teams: 7393 case OMPD_simd: 7394 case OMPD_for: 7395 case OMPD_for_simd: 7396 case OMPD_sections: 7397 case OMPD_section: 7398 case OMPD_single: 7399 case OMPD_master: 7400 case OMPD_critical: 7401 case OMPD_taskgroup: 7402 case OMPD_distribute: 7403 case OMPD_ordered: 7404 case OMPD_atomic: 7405 case OMPD_distribute_simd: 7406 case OMPD_teams_distribute: 7407 case OMPD_teams_distribute_simd: 7408 llvm_unreachable("Unexpected OpenMP directive with num_threads-clause"); 7409 case OMPD_unknown: 7410 llvm_unreachable("Unknown OpenMP directive"); 7411 } 7412 break; 7413 case OMPC_num_teams: 7414 switch (DKind) { 7415 case OMPD_target_teams: 7416 CaptureRegion = OMPD_target; 7417 break; 7418 case OMPD_cancel: 7419 case OMPD_parallel: 7420 case OMPD_parallel_sections: 7421 case OMPD_parallel_for: 7422 case OMPD_parallel_for_simd: 7423 case OMPD_target: 7424 case OMPD_target_simd: 7425 case OMPD_target_parallel: 7426 case OMPD_target_parallel_for: 7427 case OMPD_target_parallel_for_simd: 7428 case OMPD_target_teams_distribute: 7429 case OMPD_target_teams_distribute_simd: 7430 case OMPD_target_teams_distribute_parallel_for: 7431 case OMPD_target_teams_distribute_parallel_for_simd: 7432 case OMPD_teams_distribute_parallel_for: 7433 case OMPD_teams_distribute_parallel_for_simd: 7434 case OMPD_distribute_parallel_for: 7435 case OMPD_distribute_parallel_for_simd: 7436 case OMPD_task: 7437 case OMPD_taskloop: 7438 case OMPD_taskloop_simd: 7439 case OMPD_target_data: 7440 case OMPD_target_enter_data: 7441 case OMPD_target_exit_data: 7442 case OMPD_target_update: 7443 case OMPD_teams: 7444 case OMPD_teams_distribute: 7445 case OMPD_teams_distribute_simd: 7446 // Do not capture num_teams-clause expressions. 7447 break; 7448 case OMPD_threadprivate: 7449 case OMPD_taskyield: 7450 case OMPD_barrier: 7451 case OMPD_taskwait: 7452 case OMPD_cancellation_point: 7453 case OMPD_flush: 7454 case OMPD_declare_reduction: 7455 case OMPD_declare_simd: 7456 case OMPD_declare_target: 7457 case OMPD_end_declare_target: 7458 case OMPD_simd: 7459 case OMPD_for: 7460 case OMPD_for_simd: 7461 case OMPD_sections: 7462 case OMPD_section: 7463 case OMPD_single: 7464 case OMPD_master: 7465 case OMPD_critical: 7466 case OMPD_taskgroup: 7467 case OMPD_distribute: 7468 case OMPD_ordered: 7469 case OMPD_atomic: 7470 case OMPD_distribute_simd: 7471 llvm_unreachable("Unexpected OpenMP directive with num_teams-clause"); 7472 case OMPD_unknown: 7473 llvm_unreachable("Unknown OpenMP directive"); 7474 } 7475 break; 7476 case OMPC_thread_limit: 7477 switch (DKind) { 7478 case OMPD_target_teams: 7479 CaptureRegion = OMPD_target; 7480 break; 7481 case OMPD_cancel: 7482 case OMPD_parallel: 7483 case OMPD_parallel_sections: 7484 case OMPD_parallel_for: 7485 case OMPD_parallel_for_simd: 7486 case OMPD_target: 7487 case OMPD_target_simd: 7488 case OMPD_target_parallel: 7489 case OMPD_target_parallel_for: 7490 case OMPD_target_parallel_for_simd: 7491 case OMPD_target_teams_distribute: 7492 case OMPD_target_teams_distribute_simd: 7493 case OMPD_target_teams_distribute_parallel_for: 7494 case OMPD_target_teams_distribute_parallel_for_simd: 7495 case OMPD_teams_distribute_parallel_for: 7496 case OMPD_teams_distribute_parallel_for_simd: 7497 case OMPD_distribute_parallel_for: 7498 case OMPD_distribute_parallel_for_simd: 7499 case OMPD_task: 7500 case OMPD_taskloop: 7501 case OMPD_taskloop_simd: 7502 case OMPD_target_data: 7503 case OMPD_target_enter_data: 7504 case OMPD_target_exit_data: 7505 case OMPD_target_update: 7506 case OMPD_teams: 7507 case OMPD_teams_distribute: 7508 case OMPD_teams_distribute_simd: 7509 // Do not capture thread_limit-clause expressions. 7510 break; 7511 case OMPD_threadprivate: 7512 case OMPD_taskyield: 7513 case OMPD_barrier: 7514 case OMPD_taskwait: 7515 case OMPD_cancellation_point: 7516 case OMPD_flush: 7517 case OMPD_declare_reduction: 7518 case OMPD_declare_simd: 7519 case OMPD_declare_target: 7520 case OMPD_end_declare_target: 7521 case OMPD_simd: 7522 case OMPD_for: 7523 case OMPD_for_simd: 7524 case OMPD_sections: 7525 case OMPD_section: 7526 case OMPD_single: 7527 case OMPD_master: 7528 case OMPD_critical: 7529 case OMPD_taskgroup: 7530 case OMPD_distribute: 7531 case OMPD_ordered: 7532 case OMPD_atomic: 7533 case OMPD_distribute_simd: 7534 llvm_unreachable("Unexpected OpenMP directive with thread_limit-clause"); 7535 case OMPD_unknown: 7536 llvm_unreachable("Unknown OpenMP directive"); 7537 } 7538 break; 7539 case OMPC_schedule: 7540 case OMPC_dist_schedule: 7541 case OMPC_firstprivate: 7542 case OMPC_lastprivate: 7543 case OMPC_reduction: 7544 case OMPC_task_reduction: 7545 case OMPC_in_reduction: 7546 case OMPC_linear: 7547 case OMPC_default: 7548 case OMPC_proc_bind: 7549 case OMPC_final: 7550 case OMPC_safelen: 7551 case OMPC_simdlen: 7552 case OMPC_collapse: 7553 case OMPC_private: 7554 case OMPC_shared: 7555 case OMPC_aligned: 7556 case OMPC_copyin: 7557 case OMPC_copyprivate: 7558 case OMPC_ordered: 7559 case OMPC_nowait: 7560 case OMPC_untied: 7561 case OMPC_mergeable: 7562 case OMPC_threadprivate: 7563 case OMPC_flush: 7564 case OMPC_read: 7565 case OMPC_write: 7566 case OMPC_update: 7567 case OMPC_capture: 7568 case OMPC_seq_cst: 7569 case OMPC_depend: 7570 case OMPC_device: 7571 case OMPC_threads: 7572 case OMPC_simd: 7573 case OMPC_map: 7574 case OMPC_priority: 7575 case OMPC_grainsize: 7576 case OMPC_nogroup: 7577 case OMPC_num_tasks: 7578 case OMPC_hint: 7579 case OMPC_defaultmap: 7580 case OMPC_unknown: 7581 case OMPC_uniform: 7582 case OMPC_to: 7583 case OMPC_from: 7584 case OMPC_use_device_ptr: 7585 case OMPC_is_device_ptr: 7586 llvm_unreachable("Unexpected OpenMP clause."); 7587 } 7588 return CaptureRegion; 7589 } 7590 7591 OMPClause *Sema::ActOnOpenMPIfClause(OpenMPDirectiveKind NameModifier, 7592 Expr *Condition, SourceLocation StartLoc, 7593 SourceLocation LParenLoc, 7594 SourceLocation NameModifierLoc, 7595 SourceLocation ColonLoc, 7596 SourceLocation EndLoc) { 7597 Expr *ValExpr = Condition; 7598 Stmt *HelperValStmt = nullptr; 7599 OpenMPDirectiveKind CaptureRegion = OMPD_unknown; 7600 if (!Condition->isValueDependent() && !Condition->isTypeDependent() && 7601 !Condition->isInstantiationDependent() && 7602 !Condition->containsUnexpandedParameterPack()) { 7603 ExprResult Val = CheckBooleanCondition(StartLoc, Condition); 7604 if (Val.isInvalid()) 7605 return nullptr; 7606 7607 ValExpr = MakeFullExpr(Val.get()).get(); 7608 7609 OpenMPDirectiveKind DKind = DSAStack->getCurrentDirective(); 7610 CaptureRegion = 7611 getOpenMPCaptureRegionForClause(DKind, OMPC_if, NameModifier); 7612 if (CaptureRegion != OMPD_unknown) { 7613 llvm::MapVector<Expr *, DeclRefExpr *> Captures; 7614 ValExpr = tryBuildCapture(*this, ValExpr, Captures).get(); 7615 HelperValStmt = buildPreInits(Context, Captures); 7616 } 7617 } 7618 7619 return new (Context) 7620 OMPIfClause(NameModifier, ValExpr, HelperValStmt, CaptureRegion, StartLoc, 7621 LParenLoc, NameModifierLoc, ColonLoc, EndLoc); 7622 } 7623 7624 OMPClause *Sema::ActOnOpenMPFinalClause(Expr *Condition, 7625 SourceLocation StartLoc, 7626 SourceLocation LParenLoc, 7627 SourceLocation EndLoc) { 7628 Expr *ValExpr = Condition; 7629 if (!Condition->isValueDependent() && !Condition->isTypeDependent() && 7630 !Condition->isInstantiationDependent() && 7631 !Condition->containsUnexpandedParameterPack()) { 7632 ExprResult Val = CheckBooleanCondition(StartLoc, Condition); 7633 if (Val.isInvalid()) 7634 return nullptr; 7635 7636 ValExpr = MakeFullExpr(Val.get()).get(); 7637 } 7638 7639 return new (Context) OMPFinalClause(ValExpr, StartLoc, LParenLoc, EndLoc); 7640 } 7641 ExprResult Sema::PerformOpenMPImplicitIntegerConversion(SourceLocation Loc, 7642 Expr *Op) { 7643 if (!Op) 7644 return ExprError(); 7645 7646 class IntConvertDiagnoser : public ICEConvertDiagnoser { 7647 public: 7648 IntConvertDiagnoser() 7649 : ICEConvertDiagnoser(/*AllowScopedEnumerations*/ false, false, true) {} 7650 SemaDiagnosticBuilder diagnoseNotInt(Sema &S, SourceLocation Loc, 7651 QualType T) override { 7652 return S.Diag(Loc, diag::err_omp_not_integral) << T; 7653 } 7654 SemaDiagnosticBuilder diagnoseIncomplete(Sema &S, SourceLocation Loc, 7655 QualType T) override { 7656 return S.Diag(Loc, diag::err_omp_incomplete_type) << T; 7657 } 7658 SemaDiagnosticBuilder diagnoseExplicitConv(Sema &S, SourceLocation Loc, 7659 QualType T, 7660 QualType ConvTy) override { 7661 return S.Diag(Loc, diag::err_omp_explicit_conversion) << T << ConvTy; 7662 } 7663 SemaDiagnosticBuilder noteExplicitConv(Sema &S, CXXConversionDecl *Conv, 7664 QualType ConvTy) override { 7665 return S.Diag(Conv->getLocation(), diag::note_omp_conversion_here) 7666 << ConvTy->isEnumeralType() << ConvTy; 7667 } 7668 SemaDiagnosticBuilder diagnoseAmbiguous(Sema &S, SourceLocation Loc, 7669 QualType T) override { 7670 return S.Diag(Loc, diag::err_omp_ambiguous_conversion) << T; 7671 } 7672 SemaDiagnosticBuilder noteAmbiguous(Sema &S, CXXConversionDecl *Conv, 7673 QualType ConvTy) override { 7674 return S.Diag(Conv->getLocation(), diag::note_omp_conversion_here) 7675 << ConvTy->isEnumeralType() << ConvTy; 7676 } 7677 SemaDiagnosticBuilder diagnoseConversion(Sema &, SourceLocation, QualType, 7678 QualType) override { 7679 llvm_unreachable("conversion functions are permitted"); 7680 } 7681 } ConvertDiagnoser; 7682 return PerformContextualImplicitConversion(Loc, Op, ConvertDiagnoser); 7683 } 7684 7685 static bool IsNonNegativeIntegerValue(Expr *&ValExpr, Sema &SemaRef, 7686 OpenMPClauseKind CKind, 7687 bool StrictlyPositive) { 7688 if (!ValExpr->isTypeDependent() && !ValExpr->isValueDependent() && 7689 !ValExpr->isInstantiationDependent()) { 7690 SourceLocation Loc = ValExpr->getExprLoc(); 7691 ExprResult Value = 7692 SemaRef.PerformOpenMPImplicitIntegerConversion(Loc, ValExpr); 7693 if (Value.isInvalid()) 7694 return false; 7695 7696 ValExpr = Value.get(); 7697 // The expression must evaluate to a non-negative integer value. 7698 llvm::APSInt Result; 7699 if (ValExpr->isIntegerConstantExpr(Result, SemaRef.Context) && 7700 Result.isSigned() && 7701 !((!StrictlyPositive && Result.isNonNegative()) || 7702 (StrictlyPositive && Result.isStrictlyPositive()))) { 7703 SemaRef.Diag(Loc, diag::err_omp_negative_expression_in_clause) 7704 << getOpenMPClauseName(CKind) << (StrictlyPositive ? 1 : 0) 7705 << ValExpr->getSourceRange(); 7706 return false; 7707 } 7708 } 7709 return true; 7710 } 7711 7712 OMPClause *Sema::ActOnOpenMPNumThreadsClause(Expr *NumThreads, 7713 SourceLocation StartLoc, 7714 SourceLocation LParenLoc, 7715 SourceLocation EndLoc) { 7716 Expr *ValExpr = NumThreads; 7717 Stmt *HelperValStmt = nullptr; 7718 OpenMPDirectiveKind CaptureRegion = OMPD_unknown; 7719 7720 // OpenMP [2.5, Restrictions] 7721 // The num_threads expression must evaluate to a positive integer value. 7722 if (!IsNonNegativeIntegerValue(ValExpr, *this, OMPC_num_threads, 7723 /*StrictlyPositive=*/true)) 7724 return nullptr; 7725 7726 OpenMPDirectiveKind DKind = DSAStack->getCurrentDirective(); 7727 CaptureRegion = getOpenMPCaptureRegionForClause(DKind, OMPC_num_threads); 7728 if (CaptureRegion != OMPD_unknown) { 7729 llvm::MapVector<Expr *, DeclRefExpr *> Captures; 7730 ValExpr = tryBuildCapture(*this, ValExpr, Captures).get(); 7731 HelperValStmt = buildPreInits(Context, Captures); 7732 } 7733 7734 return new (Context) OMPNumThreadsClause( 7735 ValExpr, HelperValStmt, CaptureRegion, StartLoc, LParenLoc, EndLoc); 7736 } 7737 7738 ExprResult Sema::VerifyPositiveIntegerConstantInClause(Expr *E, 7739 OpenMPClauseKind CKind, 7740 bool StrictlyPositive) { 7741 if (!E) 7742 return ExprError(); 7743 if (E->isValueDependent() || E->isTypeDependent() || 7744 E->isInstantiationDependent() || E->containsUnexpandedParameterPack()) 7745 return E; 7746 llvm::APSInt Result; 7747 ExprResult ICE = VerifyIntegerConstantExpression(E, &Result); 7748 if (ICE.isInvalid()) 7749 return ExprError(); 7750 if ((StrictlyPositive && !Result.isStrictlyPositive()) || 7751 (!StrictlyPositive && !Result.isNonNegative())) { 7752 Diag(E->getExprLoc(), diag::err_omp_negative_expression_in_clause) 7753 << getOpenMPClauseName(CKind) << (StrictlyPositive ? 1 : 0) 7754 << E->getSourceRange(); 7755 return ExprError(); 7756 } 7757 if (CKind == OMPC_aligned && !Result.isPowerOf2()) { 7758 Diag(E->getExprLoc(), diag::warn_omp_alignment_not_power_of_two) 7759 << E->getSourceRange(); 7760 return ExprError(); 7761 } 7762 if (CKind == OMPC_collapse && DSAStack->getAssociatedLoops() == 1) 7763 DSAStack->setAssociatedLoops(Result.getExtValue()); 7764 else if (CKind == OMPC_ordered) 7765 DSAStack->setAssociatedLoops(Result.getExtValue()); 7766 return ICE; 7767 } 7768 7769 OMPClause *Sema::ActOnOpenMPSafelenClause(Expr *Len, SourceLocation StartLoc, 7770 SourceLocation LParenLoc, 7771 SourceLocation EndLoc) { 7772 // OpenMP [2.8.1, simd construct, Description] 7773 // The parameter of the safelen clause must be a constant 7774 // positive integer expression. 7775 ExprResult Safelen = VerifyPositiveIntegerConstantInClause(Len, OMPC_safelen); 7776 if (Safelen.isInvalid()) 7777 return nullptr; 7778 return new (Context) 7779 OMPSafelenClause(Safelen.get(), StartLoc, LParenLoc, EndLoc); 7780 } 7781 7782 OMPClause *Sema::ActOnOpenMPSimdlenClause(Expr *Len, SourceLocation StartLoc, 7783 SourceLocation LParenLoc, 7784 SourceLocation EndLoc) { 7785 // OpenMP [2.8.1, simd construct, Description] 7786 // The parameter of the simdlen clause must be a constant 7787 // positive integer expression. 7788 ExprResult Simdlen = VerifyPositiveIntegerConstantInClause(Len, OMPC_simdlen); 7789 if (Simdlen.isInvalid()) 7790 return nullptr; 7791 return new (Context) 7792 OMPSimdlenClause(Simdlen.get(), StartLoc, LParenLoc, EndLoc); 7793 } 7794 7795 OMPClause *Sema::ActOnOpenMPCollapseClause(Expr *NumForLoops, 7796 SourceLocation StartLoc, 7797 SourceLocation LParenLoc, 7798 SourceLocation EndLoc) { 7799 // OpenMP [2.7.1, loop construct, Description] 7800 // OpenMP [2.8.1, simd construct, Description] 7801 // OpenMP [2.9.6, distribute construct, Description] 7802 // The parameter of the collapse clause must be a constant 7803 // positive integer expression. 7804 ExprResult NumForLoopsResult = 7805 VerifyPositiveIntegerConstantInClause(NumForLoops, OMPC_collapse); 7806 if (NumForLoopsResult.isInvalid()) 7807 return nullptr; 7808 return new (Context) 7809 OMPCollapseClause(NumForLoopsResult.get(), StartLoc, LParenLoc, EndLoc); 7810 } 7811 7812 OMPClause *Sema::ActOnOpenMPOrderedClause(SourceLocation StartLoc, 7813 SourceLocation EndLoc, 7814 SourceLocation LParenLoc, 7815 Expr *NumForLoops) { 7816 // OpenMP [2.7.1, loop construct, Description] 7817 // OpenMP [2.8.1, simd construct, Description] 7818 // OpenMP [2.9.6, distribute construct, Description] 7819 // The parameter of the ordered clause must be a constant 7820 // positive integer expression if any. 7821 if (NumForLoops && LParenLoc.isValid()) { 7822 ExprResult NumForLoopsResult = 7823 VerifyPositiveIntegerConstantInClause(NumForLoops, OMPC_ordered); 7824 if (NumForLoopsResult.isInvalid()) 7825 return nullptr; 7826 NumForLoops = NumForLoopsResult.get(); 7827 } else 7828 NumForLoops = nullptr; 7829 DSAStack->setOrderedRegion(/*IsOrdered=*/true, NumForLoops); 7830 return new (Context) 7831 OMPOrderedClause(NumForLoops, StartLoc, LParenLoc, EndLoc); 7832 } 7833 7834 OMPClause *Sema::ActOnOpenMPSimpleClause( 7835 OpenMPClauseKind Kind, unsigned Argument, SourceLocation ArgumentLoc, 7836 SourceLocation StartLoc, SourceLocation LParenLoc, SourceLocation EndLoc) { 7837 OMPClause *Res = nullptr; 7838 switch (Kind) { 7839 case OMPC_default: 7840 Res = 7841 ActOnOpenMPDefaultClause(static_cast<OpenMPDefaultClauseKind>(Argument), 7842 ArgumentLoc, StartLoc, LParenLoc, EndLoc); 7843 break; 7844 case OMPC_proc_bind: 7845 Res = ActOnOpenMPProcBindClause( 7846 static_cast<OpenMPProcBindClauseKind>(Argument), ArgumentLoc, StartLoc, 7847 LParenLoc, EndLoc); 7848 break; 7849 case OMPC_if: 7850 case OMPC_final: 7851 case OMPC_num_threads: 7852 case OMPC_safelen: 7853 case OMPC_simdlen: 7854 case OMPC_collapse: 7855 case OMPC_schedule: 7856 case OMPC_private: 7857 case OMPC_firstprivate: 7858 case OMPC_lastprivate: 7859 case OMPC_shared: 7860 case OMPC_reduction: 7861 case OMPC_task_reduction: 7862 case OMPC_in_reduction: 7863 case OMPC_linear: 7864 case OMPC_aligned: 7865 case OMPC_copyin: 7866 case OMPC_copyprivate: 7867 case OMPC_ordered: 7868 case OMPC_nowait: 7869 case OMPC_untied: 7870 case OMPC_mergeable: 7871 case OMPC_threadprivate: 7872 case OMPC_flush: 7873 case OMPC_read: 7874 case OMPC_write: 7875 case OMPC_update: 7876 case OMPC_capture: 7877 case OMPC_seq_cst: 7878 case OMPC_depend: 7879 case OMPC_device: 7880 case OMPC_threads: 7881 case OMPC_simd: 7882 case OMPC_map: 7883 case OMPC_num_teams: 7884 case OMPC_thread_limit: 7885 case OMPC_priority: 7886 case OMPC_grainsize: 7887 case OMPC_nogroup: 7888 case OMPC_num_tasks: 7889 case OMPC_hint: 7890 case OMPC_dist_schedule: 7891 case OMPC_defaultmap: 7892 case OMPC_unknown: 7893 case OMPC_uniform: 7894 case OMPC_to: 7895 case OMPC_from: 7896 case OMPC_use_device_ptr: 7897 case OMPC_is_device_ptr: 7898 llvm_unreachable("Clause is not allowed."); 7899 } 7900 return Res; 7901 } 7902 7903 static std::string 7904 getListOfPossibleValues(OpenMPClauseKind K, unsigned First, unsigned Last, 7905 ArrayRef<unsigned> Exclude = llvm::None) { 7906 std::string Values; 7907 unsigned Bound = Last >= 2 ? Last - 2 : 0; 7908 unsigned Skipped = Exclude.size(); 7909 auto S = Exclude.begin(), E = Exclude.end(); 7910 for (unsigned i = First; i < Last; ++i) { 7911 if (std::find(S, E, i) != E) { 7912 --Skipped; 7913 continue; 7914 } 7915 Values += "'"; 7916 Values += getOpenMPSimpleClauseTypeName(K, i); 7917 Values += "'"; 7918 if (i == Bound - Skipped) 7919 Values += " or "; 7920 else if (i != Bound + 1 - Skipped) 7921 Values += ", "; 7922 } 7923 return Values; 7924 } 7925 7926 OMPClause *Sema::ActOnOpenMPDefaultClause(OpenMPDefaultClauseKind Kind, 7927 SourceLocation KindKwLoc, 7928 SourceLocation StartLoc, 7929 SourceLocation LParenLoc, 7930 SourceLocation EndLoc) { 7931 if (Kind == OMPC_DEFAULT_unknown) { 7932 static_assert(OMPC_DEFAULT_unknown > 0, 7933 "OMPC_DEFAULT_unknown not greater than 0"); 7934 Diag(KindKwLoc, diag::err_omp_unexpected_clause_value) 7935 << getListOfPossibleValues(OMPC_default, /*First=*/0, 7936 /*Last=*/OMPC_DEFAULT_unknown) 7937 << getOpenMPClauseName(OMPC_default); 7938 return nullptr; 7939 } 7940 switch (Kind) { 7941 case OMPC_DEFAULT_none: 7942 DSAStack->setDefaultDSANone(KindKwLoc); 7943 break; 7944 case OMPC_DEFAULT_shared: 7945 DSAStack->setDefaultDSAShared(KindKwLoc); 7946 break; 7947 case OMPC_DEFAULT_unknown: 7948 llvm_unreachable("Clause kind is not allowed."); 7949 break; 7950 } 7951 return new (Context) 7952 OMPDefaultClause(Kind, KindKwLoc, StartLoc, LParenLoc, EndLoc); 7953 } 7954 7955 OMPClause *Sema::ActOnOpenMPProcBindClause(OpenMPProcBindClauseKind Kind, 7956 SourceLocation KindKwLoc, 7957 SourceLocation StartLoc, 7958 SourceLocation LParenLoc, 7959 SourceLocation EndLoc) { 7960 if (Kind == OMPC_PROC_BIND_unknown) { 7961 Diag(KindKwLoc, diag::err_omp_unexpected_clause_value) 7962 << getListOfPossibleValues(OMPC_proc_bind, /*First=*/0, 7963 /*Last=*/OMPC_PROC_BIND_unknown) 7964 << getOpenMPClauseName(OMPC_proc_bind); 7965 return nullptr; 7966 } 7967 return new (Context) 7968 OMPProcBindClause(Kind, KindKwLoc, StartLoc, LParenLoc, EndLoc); 7969 } 7970 7971 OMPClause *Sema::ActOnOpenMPSingleExprWithArgClause( 7972 OpenMPClauseKind Kind, ArrayRef<unsigned> Argument, Expr *Expr, 7973 SourceLocation StartLoc, SourceLocation LParenLoc, 7974 ArrayRef<SourceLocation> ArgumentLoc, SourceLocation DelimLoc, 7975 SourceLocation EndLoc) { 7976 OMPClause *Res = nullptr; 7977 switch (Kind) { 7978 case OMPC_schedule: 7979 enum { Modifier1, Modifier2, ScheduleKind, NumberOfElements }; 7980 assert(Argument.size() == NumberOfElements && 7981 ArgumentLoc.size() == NumberOfElements); 7982 Res = ActOnOpenMPScheduleClause( 7983 static_cast<OpenMPScheduleClauseModifier>(Argument[Modifier1]), 7984 static_cast<OpenMPScheduleClauseModifier>(Argument[Modifier2]), 7985 static_cast<OpenMPScheduleClauseKind>(Argument[ScheduleKind]), Expr, 7986 StartLoc, LParenLoc, ArgumentLoc[Modifier1], ArgumentLoc[Modifier2], 7987 ArgumentLoc[ScheduleKind], DelimLoc, EndLoc); 7988 break; 7989 case OMPC_if: 7990 assert(Argument.size() == 1 && ArgumentLoc.size() == 1); 7991 Res = ActOnOpenMPIfClause(static_cast<OpenMPDirectiveKind>(Argument.back()), 7992 Expr, StartLoc, LParenLoc, ArgumentLoc.back(), 7993 DelimLoc, EndLoc); 7994 break; 7995 case OMPC_dist_schedule: 7996 Res = ActOnOpenMPDistScheduleClause( 7997 static_cast<OpenMPDistScheduleClauseKind>(Argument.back()), Expr, 7998 StartLoc, LParenLoc, ArgumentLoc.back(), DelimLoc, EndLoc); 7999 break; 8000 case OMPC_defaultmap: 8001 enum { Modifier, DefaultmapKind }; 8002 Res = ActOnOpenMPDefaultmapClause( 8003 static_cast<OpenMPDefaultmapClauseModifier>(Argument[Modifier]), 8004 static_cast<OpenMPDefaultmapClauseKind>(Argument[DefaultmapKind]), 8005 StartLoc, LParenLoc, ArgumentLoc[Modifier], ArgumentLoc[DefaultmapKind], 8006 EndLoc); 8007 break; 8008 case OMPC_final: 8009 case OMPC_num_threads: 8010 case OMPC_safelen: 8011 case OMPC_simdlen: 8012 case OMPC_collapse: 8013 case OMPC_default: 8014 case OMPC_proc_bind: 8015 case OMPC_private: 8016 case OMPC_firstprivate: 8017 case OMPC_lastprivate: 8018 case OMPC_shared: 8019 case OMPC_reduction: 8020 case OMPC_task_reduction: 8021 case OMPC_in_reduction: 8022 case OMPC_linear: 8023 case OMPC_aligned: 8024 case OMPC_copyin: 8025 case OMPC_copyprivate: 8026 case OMPC_ordered: 8027 case OMPC_nowait: 8028 case OMPC_untied: 8029 case OMPC_mergeable: 8030 case OMPC_threadprivate: 8031 case OMPC_flush: 8032 case OMPC_read: 8033 case OMPC_write: 8034 case OMPC_update: 8035 case OMPC_capture: 8036 case OMPC_seq_cst: 8037 case OMPC_depend: 8038 case OMPC_device: 8039 case OMPC_threads: 8040 case OMPC_simd: 8041 case OMPC_map: 8042 case OMPC_num_teams: 8043 case OMPC_thread_limit: 8044 case OMPC_priority: 8045 case OMPC_grainsize: 8046 case OMPC_nogroup: 8047 case OMPC_num_tasks: 8048 case OMPC_hint: 8049 case OMPC_unknown: 8050 case OMPC_uniform: 8051 case OMPC_to: 8052 case OMPC_from: 8053 case OMPC_use_device_ptr: 8054 case OMPC_is_device_ptr: 8055 llvm_unreachable("Clause is not allowed."); 8056 } 8057 return Res; 8058 } 8059 8060 static bool checkScheduleModifiers(Sema &S, OpenMPScheduleClauseModifier M1, 8061 OpenMPScheduleClauseModifier M2, 8062 SourceLocation M1Loc, SourceLocation M2Loc) { 8063 if (M1 == OMPC_SCHEDULE_MODIFIER_unknown && M1Loc.isValid()) { 8064 SmallVector<unsigned, 2> Excluded; 8065 if (M2 != OMPC_SCHEDULE_MODIFIER_unknown) 8066 Excluded.push_back(M2); 8067 if (M2 == OMPC_SCHEDULE_MODIFIER_nonmonotonic) 8068 Excluded.push_back(OMPC_SCHEDULE_MODIFIER_monotonic); 8069 if (M2 == OMPC_SCHEDULE_MODIFIER_monotonic) 8070 Excluded.push_back(OMPC_SCHEDULE_MODIFIER_nonmonotonic); 8071 S.Diag(M1Loc, diag::err_omp_unexpected_clause_value) 8072 << getListOfPossibleValues(OMPC_schedule, 8073 /*First=*/OMPC_SCHEDULE_MODIFIER_unknown + 1, 8074 /*Last=*/OMPC_SCHEDULE_MODIFIER_last, 8075 Excluded) 8076 << getOpenMPClauseName(OMPC_schedule); 8077 return true; 8078 } 8079 return false; 8080 } 8081 8082 OMPClause *Sema::ActOnOpenMPScheduleClause( 8083 OpenMPScheduleClauseModifier M1, OpenMPScheduleClauseModifier M2, 8084 OpenMPScheduleClauseKind Kind, Expr *ChunkSize, SourceLocation StartLoc, 8085 SourceLocation LParenLoc, SourceLocation M1Loc, SourceLocation M2Loc, 8086 SourceLocation KindLoc, SourceLocation CommaLoc, SourceLocation EndLoc) { 8087 if (checkScheduleModifiers(*this, M1, M2, M1Loc, M2Loc) || 8088 checkScheduleModifiers(*this, M2, M1, M2Loc, M1Loc)) 8089 return nullptr; 8090 // OpenMP, 2.7.1, Loop Construct, Restrictions 8091 // Either the monotonic modifier or the nonmonotonic modifier can be specified 8092 // but not both. 8093 if ((M1 == M2 && M1 != OMPC_SCHEDULE_MODIFIER_unknown) || 8094 (M1 == OMPC_SCHEDULE_MODIFIER_monotonic && 8095 M2 == OMPC_SCHEDULE_MODIFIER_nonmonotonic) || 8096 (M1 == OMPC_SCHEDULE_MODIFIER_nonmonotonic && 8097 M2 == OMPC_SCHEDULE_MODIFIER_monotonic)) { 8098 Diag(M2Loc, diag::err_omp_unexpected_schedule_modifier) 8099 << getOpenMPSimpleClauseTypeName(OMPC_schedule, M2) 8100 << getOpenMPSimpleClauseTypeName(OMPC_schedule, M1); 8101 return nullptr; 8102 } 8103 if (Kind == OMPC_SCHEDULE_unknown) { 8104 std::string Values; 8105 if (M1Loc.isInvalid() && M2Loc.isInvalid()) { 8106 unsigned Exclude[] = {OMPC_SCHEDULE_unknown}; 8107 Values = getListOfPossibleValues(OMPC_schedule, /*First=*/0, 8108 /*Last=*/OMPC_SCHEDULE_MODIFIER_last, 8109 Exclude); 8110 } else { 8111 Values = getListOfPossibleValues(OMPC_schedule, /*First=*/0, 8112 /*Last=*/OMPC_SCHEDULE_unknown); 8113 } 8114 Diag(KindLoc, diag::err_omp_unexpected_clause_value) 8115 << Values << getOpenMPClauseName(OMPC_schedule); 8116 return nullptr; 8117 } 8118 // OpenMP, 2.7.1, Loop Construct, Restrictions 8119 // The nonmonotonic modifier can only be specified with schedule(dynamic) or 8120 // schedule(guided). 8121 if ((M1 == OMPC_SCHEDULE_MODIFIER_nonmonotonic || 8122 M2 == OMPC_SCHEDULE_MODIFIER_nonmonotonic) && 8123 Kind != OMPC_SCHEDULE_dynamic && Kind != OMPC_SCHEDULE_guided) { 8124 Diag(M1 == OMPC_SCHEDULE_MODIFIER_nonmonotonic ? M1Loc : M2Loc, 8125 diag::err_omp_schedule_nonmonotonic_static); 8126 return nullptr; 8127 } 8128 Expr *ValExpr = ChunkSize; 8129 Stmt *HelperValStmt = nullptr; 8130 if (ChunkSize) { 8131 if (!ChunkSize->isValueDependent() && !ChunkSize->isTypeDependent() && 8132 !ChunkSize->isInstantiationDependent() && 8133 !ChunkSize->containsUnexpandedParameterPack()) { 8134 SourceLocation ChunkSizeLoc = ChunkSize->getLocStart(); 8135 ExprResult Val = 8136 PerformOpenMPImplicitIntegerConversion(ChunkSizeLoc, ChunkSize); 8137 if (Val.isInvalid()) 8138 return nullptr; 8139 8140 ValExpr = Val.get(); 8141 8142 // OpenMP [2.7.1, Restrictions] 8143 // chunk_size must be a loop invariant integer expression with a positive 8144 // value. 8145 llvm::APSInt Result; 8146 if (ValExpr->isIntegerConstantExpr(Result, Context)) { 8147 if (Result.isSigned() && !Result.isStrictlyPositive()) { 8148 Diag(ChunkSizeLoc, diag::err_omp_negative_expression_in_clause) 8149 << "schedule" << 1 << ChunkSize->getSourceRange(); 8150 return nullptr; 8151 } 8152 } else if (isParallelOrTaskRegion(DSAStack->getCurrentDirective()) && 8153 !CurContext->isDependentContext()) { 8154 llvm::MapVector<Expr *, DeclRefExpr *> Captures; 8155 ValExpr = tryBuildCapture(*this, ValExpr, Captures).get(); 8156 HelperValStmt = buildPreInits(Context, Captures); 8157 } 8158 } 8159 } 8160 8161 return new (Context) 8162 OMPScheduleClause(StartLoc, LParenLoc, KindLoc, CommaLoc, EndLoc, Kind, 8163 ValExpr, HelperValStmt, M1, M1Loc, M2, M2Loc); 8164 } 8165 8166 OMPClause *Sema::ActOnOpenMPClause(OpenMPClauseKind Kind, 8167 SourceLocation StartLoc, 8168 SourceLocation EndLoc) { 8169 OMPClause *Res = nullptr; 8170 switch (Kind) { 8171 case OMPC_ordered: 8172 Res = ActOnOpenMPOrderedClause(StartLoc, EndLoc); 8173 break; 8174 case OMPC_nowait: 8175 Res = ActOnOpenMPNowaitClause(StartLoc, EndLoc); 8176 break; 8177 case OMPC_untied: 8178 Res = ActOnOpenMPUntiedClause(StartLoc, EndLoc); 8179 break; 8180 case OMPC_mergeable: 8181 Res = ActOnOpenMPMergeableClause(StartLoc, EndLoc); 8182 break; 8183 case OMPC_read: 8184 Res = ActOnOpenMPReadClause(StartLoc, EndLoc); 8185 break; 8186 case OMPC_write: 8187 Res = ActOnOpenMPWriteClause(StartLoc, EndLoc); 8188 break; 8189 case OMPC_update: 8190 Res = ActOnOpenMPUpdateClause(StartLoc, EndLoc); 8191 break; 8192 case OMPC_capture: 8193 Res = ActOnOpenMPCaptureClause(StartLoc, EndLoc); 8194 break; 8195 case OMPC_seq_cst: 8196 Res = ActOnOpenMPSeqCstClause(StartLoc, EndLoc); 8197 break; 8198 case OMPC_threads: 8199 Res = ActOnOpenMPThreadsClause(StartLoc, EndLoc); 8200 break; 8201 case OMPC_simd: 8202 Res = ActOnOpenMPSIMDClause(StartLoc, EndLoc); 8203 break; 8204 case OMPC_nogroup: 8205 Res = ActOnOpenMPNogroupClause(StartLoc, EndLoc); 8206 break; 8207 case OMPC_if: 8208 case OMPC_final: 8209 case OMPC_num_threads: 8210 case OMPC_safelen: 8211 case OMPC_simdlen: 8212 case OMPC_collapse: 8213 case OMPC_schedule: 8214 case OMPC_private: 8215 case OMPC_firstprivate: 8216 case OMPC_lastprivate: 8217 case OMPC_shared: 8218 case OMPC_reduction: 8219 case OMPC_task_reduction: 8220 case OMPC_in_reduction: 8221 case OMPC_linear: 8222 case OMPC_aligned: 8223 case OMPC_copyin: 8224 case OMPC_copyprivate: 8225 case OMPC_default: 8226 case OMPC_proc_bind: 8227 case OMPC_threadprivate: 8228 case OMPC_flush: 8229 case OMPC_depend: 8230 case OMPC_device: 8231 case OMPC_map: 8232 case OMPC_num_teams: 8233 case OMPC_thread_limit: 8234 case OMPC_priority: 8235 case OMPC_grainsize: 8236 case OMPC_num_tasks: 8237 case OMPC_hint: 8238 case OMPC_dist_schedule: 8239 case OMPC_defaultmap: 8240 case OMPC_unknown: 8241 case OMPC_uniform: 8242 case OMPC_to: 8243 case OMPC_from: 8244 case OMPC_use_device_ptr: 8245 case OMPC_is_device_ptr: 8246 llvm_unreachable("Clause is not allowed."); 8247 } 8248 return Res; 8249 } 8250 8251 OMPClause *Sema::ActOnOpenMPNowaitClause(SourceLocation StartLoc, 8252 SourceLocation EndLoc) { 8253 DSAStack->setNowaitRegion(); 8254 return new (Context) OMPNowaitClause(StartLoc, EndLoc); 8255 } 8256 8257 OMPClause *Sema::ActOnOpenMPUntiedClause(SourceLocation StartLoc, 8258 SourceLocation EndLoc) { 8259 return new (Context) OMPUntiedClause(StartLoc, EndLoc); 8260 } 8261 8262 OMPClause *Sema::ActOnOpenMPMergeableClause(SourceLocation StartLoc, 8263 SourceLocation EndLoc) { 8264 return new (Context) OMPMergeableClause(StartLoc, EndLoc); 8265 } 8266 8267 OMPClause *Sema::ActOnOpenMPReadClause(SourceLocation StartLoc, 8268 SourceLocation EndLoc) { 8269 return new (Context) OMPReadClause(StartLoc, EndLoc); 8270 } 8271 8272 OMPClause *Sema::ActOnOpenMPWriteClause(SourceLocation StartLoc, 8273 SourceLocation EndLoc) { 8274 return new (Context) OMPWriteClause(StartLoc, EndLoc); 8275 } 8276 8277 OMPClause *Sema::ActOnOpenMPUpdateClause(SourceLocation StartLoc, 8278 SourceLocation EndLoc) { 8279 return new (Context) OMPUpdateClause(StartLoc, EndLoc); 8280 } 8281 8282 OMPClause *Sema::ActOnOpenMPCaptureClause(SourceLocation StartLoc, 8283 SourceLocation EndLoc) { 8284 return new (Context) OMPCaptureClause(StartLoc, EndLoc); 8285 } 8286 8287 OMPClause *Sema::ActOnOpenMPSeqCstClause(SourceLocation StartLoc, 8288 SourceLocation EndLoc) { 8289 return new (Context) OMPSeqCstClause(StartLoc, EndLoc); 8290 } 8291 8292 OMPClause *Sema::ActOnOpenMPThreadsClause(SourceLocation StartLoc, 8293 SourceLocation EndLoc) { 8294 return new (Context) OMPThreadsClause(StartLoc, EndLoc); 8295 } 8296 8297 OMPClause *Sema::ActOnOpenMPSIMDClause(SourceLocation StartLoc, 8298 SourceLocation EndLoc) { 8299 return new (Context) OMPSIMDClause(StartLoc, EndLoc); 8300 } 8301 8302 OMPClause *Sema::ActOnOpenMPNogroupClause(SourceLocation StartLoc, 8303 SourceLocation EndLoc) { 8304 return new (Context) OMPNogroupClause(StartLoc, EndLoc); 8305 } 8306 8307 OMPClause *Sema::ActOnOpenMPVarListClause( 8308 OpenMPClauseKind Kind, ArrayRef<Expr *> VarList, Expr *TailExpr, 8309 SourceLocation StartLoc, SourceLocation LParenLoc, SourceLocation ColonLoc, 8310 SourceLocation EndLoc, CXXScopeSpec &ReductionIdScopeSpec, 8311 const DeclarationNameInfo &ReductionId, OpenMPDependClauseKind DepKind, 8312 OpenMPLinearClauseKind LinKind, OpenMPMapClauseKind MapTypeModifier, 8313 OpenMPMapClauseKind MapType, bool IsMapTypeImplicit, 8314 SourceLocation DepLinMapLoc) { 8315 OMPClause *Res = nullptr; 8316 switch (Kind) { 8317 case OMPC_private: 8318 Res = ActOnOpenMPPrivateClause(VarList, StartLoc, LParenLoc, EndLoc); 8319 break; 8320 case OMPC_firstprivate: 8321 Res = ActOnOpenMPFirstprivateClause(VarList, StartLoc, LParenLoc, EndLoc); 8322 break; 8323 case OMPC_lastprivate: 8324 Res = ActOnOpenMPLastprivateClause(VarList, StartLoc, LParenLoc, EndLoc); 8325 break; 8326 case OMPC_shared: 8327 Res = ActOnOpenMPSharedClause(VarList, StartLoc, LParenLoc, EndLoc); 8328 break; 8329 case OMPC_reduction: 8330 Res = ActOnOpenMPReductionClause(VarList, StartLoc, LParenLoc, ColonLoc, 8331 EndLoc, ReductionIdScopeSpec, ReductionId); 8332 break; 8333 case OMPC_task_reduction: 8334 Res = ActOnOpenMPTaskReductionClause(VarList, StartLoc, LParenLoc, ColonLoc, 8335 EndLoc, ReductionIdScopeSpec, 8336 ReductionId); 8337 break; 8338 case OMPC_in_reduction: 8339 Res = 8340 ActOnOpenMPInReductionClause(VarList, StartLoc, LParenLoc, ColonLoc, 8341 EndLoc, ReductionIdScopeSpec, ReductionId); 8342 break; 8343 case OMPC_linear: 8344 Res = ActOnOpenMPLinearClause(VarList, TailExpr, StartLoc, LParenLoc, 8345 LinKind, DepLinMapLoc, ColonLoc, EndLoc); 8346 break; 8347 case OMPC_aligned: 8348 Res = ActOnOpenMPAlignedClause(VarList, TailExpr, StartLoc, LParenLoc, 8349 ColonLoc, EndLoc); 8350 break; 8351 case OMPC_copyin: 8352 Res = ActOnOpenMPCopyinClause(VarList, StartLoc, LParenLoc, EndLoc); 8353 break; 8354 case OMPC_copyprivate: 8355 Res = ActOnOpenMPCopyprivateClause(VarList, StartLoc, LParenLoc, EndLoc); 8356 break; 8357 case OMPC_flush: 8358 Res = ActOnOpenMPFlushClause(VarList, StartLoc, LParenLoc, EndLoc); 8359 break; 8360 case OMPC_depend: 8361 Res = ActOnOpenMPDependClause(DepKind, DepLinMapLoc, ColonLoc, VarList, 8362 StartLoc, LParenLoc, EndLoc); 8363 break; 8364 case OMPC_map: 8365 Res = ActOnOpenMPMapClause(MapTypeModifier, MapType, IsMapTypeImplicit, 8366 DepLinMapLoc, ColonLoc, VarList, StartLoc, 8367 LParenLoc, EndLoc); 8368 break; 8369 case OMPC_to: 8370 Res = ActOnOpenMPToClause(VarList, StartLoc, LParenLoc, EndLoc); 8371 break; 8372 case OMPC_from: 8373 Res = ActOnOpenMPFromClause(VarList, StartLoc, LParenLoc, EndLoc); 8374 break; 8375 case OMPC_use_device_ptr: 8376 Res = ActOnOpenMPUseDevicePtrClause(VarList, StartLoc, LParenLoc, EndLoc); 8377 break; 8378 case OMPC_is_device_ptr: 8379 Res = ActOnOpenMPIsDevicePtrClause(VarList, StartLoc, LParenLoc, EndLoc); 8380 break; 8381 case OMPC_if: 8382 case OMPC_final: 8383 case OMPC_num_threads: 8384 case OMPC_safelen: 8385 case OMPC_simdlen: 8386 case OMPC_collapse: 8387 case OMPC_default: 8388 case OMPC_proc_bind: 8389 case OMPC_schedule: 8390 case OMPC_ordered: 8391 case OMPC_nowait: 8392 case OMPC_untied: 8393 case OMPC_mergeable: 8394 case OMPC_threadprivate: 8395 case OMPC_read: 8396 case OMPC_write: 8397 case OMPC_update: 8398 case OMPC_capture: 8399 case OMPC_seq_cst: 8400 case OMPC_device: 8401 case OMPC_threads: 8402 case OMPC_simd: 8403 case OMPC_num_teams: 8404 case OMPC_thread_limit: 8405 case OMPC_priority: 8406 case OMPC_grainsize: 8407 case OMPC_nogroup: 8408 case OMPC_num_tasks: 8409 case OMPC_hint: 8410 case OMPC_dist_schedule: 8411 case OMPC_defaultmap: 8412 case OMPC_unknown: 8413 case OMPC_uniform: 8414 llvm_unreachable("Clause is not allowed."); 8415 } 8416 return Res; 8417 } 8418 8419 ExprResult Sema::getOpenMPCapturedExpr(VarDecl *Capture, ExprValueKind VK, 8420 ExprObjectKind OK, SourceLocation Loc) { 8421 ExprResult Res = BuildDeclRefExpr( 8422 Capture, Capture->getType().getNonReferenceType(), VK_LValue, Loc); 8423 if (!Res.isUsable()) 8424 return ExprError(); 8425 if (OK == OK_Ordinary && !getLangOpts().CPlusPlus) { 8426 Res = CreateBuiltinUnaryOp(Loc, UO_Deref, Res.get()); 8427 if (!Res.isUsable()) 8428 return ExprError(); 8429 } 8430 if (VK != VK_LValue && Res.get()->isGLValue()) { 8431 Res = DefaultLvalueConversion(Res.get()); 8432 if (!Res.isUsable()) 8433 return ExprError(); 8434 } 8435 return Res; 8436 } 8437 8438 static std::pair<ValueDecl *, bool> 8439 getPrivateItem(Sema &S, Expr *&RefExpr, SourceLocation &ELoc, 8440 SourceRange &ERange, bool AllowArraySection = false) { 8441 if (RefExpr->isTypeDependent() || RefExpr->isValueDependent() || 8442 RefExpr->containsUnexpandedParameterPack()) 8443 return std::make_pair(nullptr, true); 8444 8445 // OpenMP [3.1, C/C++] 8446 // A list item is a variable name. 8447 // OpenMP [2.9.3.3, Restrictions, p.1] 8448 // A variable that is part of another variable (as an array or 8449 // structure element) cannot appear in a private clause. 8450 RefExpr = RefExpr->IgnoreParens(); 8451 enum { 8452 NoArrayExpr = -1, 8453 ArraySubscript = 0, 8454 OMPArraySection = 1 8455 } IsArrayExpr = NoArrayExpr; 8456 if (AllowArraySection) { 8457 if (auto *ASE = dyn_cast_or_null<ArraySubscriptExpr>(RefExpr)) { 8458 auto *Base = ASE->getBase()->IgnoreParenImpCasts(); 8459 while (auto *TempASE = dyn_cast<ArraySubscriptExpr>(Base)) 8460 Base = TempASE->getBase()->IgnoreParenImpCasts(); 8461 RefExpr = Base; 8462 IsArrayExpr = ArraySubscript; 8463 } else if (auto *OASE = dyn_cast_or_null<OMPArraySectionExpr>(RefExpr)) { 8464 auto *Base = OASE->getBase()->IgnoreParenImpCasts(); 8465 while (auto *TempOASE = dyn_cast<OMPArraySectionExpr>(Base)) 8466 Base = TempOASE->getBase()->IgnoreParenImpCasts(); 8467 while (auto *TempASE = dyn_cast<ArraySubscriptExpr>(Base)) 8468 Base = TempASE->getBase()->IgnoreParenImpCasts(); 8469 RefExpr = Base; 8470 IsArrayExpr = OMPArraySection; 8471 } 8472 } 8473 ELoc = RefExpr->getExprLoc(); 8474 ERange = RefExpr->getSourceRange(); 8475 RefExpr = RefExpr->IgnoreParenImpCasts(); 8476 auto *DE = dyn_cast_or_null<DeclRefExpr>(RefExpr); 8477 auto *ME = dyn_cast_or_null<MemberExpr>(RefExpr); 8478 if ((!DE || !isa<VarDecl>(DE->getDecl())) && 8479 (S.getCurrentThisType().isNull() || !ME || 8480 !isa<CXXThisExpr>(ME->getBase()->IgnoreParenImpCasts()) || 8481 !isa<FieldDecl>(ME->getMemberDecl()))) { 8482 if (IsArrayExpr != NoArrayExpr) 8483 S.Diag(ELoc, diag::err_omp_expected_base_var_name) << IsArrayExpr 8484 << ERange; 8485 else { 8486 S.Diag(ELoc, 8487 AllowArraySection 8488 ? diag::err_omp_expected_var_name_member_expr_or_array_item 8489 : diag::err_omp_expected_var_name_member_expr) 8490 << (S.getCurrentThisType().isNull() ? 0 : 1) << ERange; 8491 } 8492 return std::make_pair(nullptr, false); 8493 } 8494 return std::make_pair( 8495 getCanonicalDecl(DE ? DE->getDecl() : ME->getMemberDecl()), false); 8496 } 8497 8498 OMPClause *Sema::ActOnOpenMPPrivateClause(ArrayRef<Expr *> VarList, 8499 SourceLocation StartLoc, 8500 SourceLocation LParenLoc, 8501 SourceLocation EndLoc) { 8502 SmallVector<Expr *, 8> Vars; 8503 SmallVector<Expr *, 8> PrivateCopies; 8504 for (auto &RefExpr : VarList) { 8505 assert(RefExpr && "NULL expr in OpenMP private clause."); 8506 SourceLocation ELoc; 8507 SourceRange ERange; 8508 Expr *SimpleRefExpr = RefExpr; 8509 auto Res = getPrivateItem(*this, SimpleRefExpr, ELoc, ERange); 8510 if (Res.second) { 8511 // It will be analyzed later. 8512 Vars.push_back(RefExpr); 8513 PrivateCopies.push_back(nullptr); 8514 } 8515 ValueDecl *D = Res.first; 8516 if (!D) 8517 continue; 8518 8519 QualType Type = D->getType(); 8520 auto *VD = dyn_cast<VarDecl>(D); 8521 8522 // OpenMP [2.9.3.3, Restrictions, C/C++, p.3] 8523 // A variable that appears in a private clause must not have an incomplete 8524 // type or a reference type. 8525 if (RequireCompleteType(ELoc, Type, diag::err_omp_private_incomplete_type)) 8526 continue; 8527 Type = Type.getNonReferenceType(); 8528 8529 // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced 8530 // in a Construct] 8531 // Variables with the predetermined data-sharing attributes may not be 8532 // listed in data-sharing attributes clauses, except for the cases 8533 // listed below. For these exceptions only, listing a predetermined 8534 // variable in a data-sharing attribute clause is allowed and overrides 8535 // the variable's predetermined data-sharing attributes. 8536 DSAStackTy::DSAVarData DVar = DSAStack->getTopDSA(D, false); 8537 if (DVar.CKind != OMPC_unknown && DVar.CKind != OMPC_private) { 8538 Diag(ELoc, diag::err_omp_wrong_dsa) << getOpenMPClauseName(DVar.CKind) 8539 << getOpenMPClauseName(OMPC_private); 8540 ReportOriginalDSA(*this, DSAStack, D, DVar); 8541 continue; 8542 } 8543 8544 auto CurrDir = DSAStack->getCurrentDirective(); 8545 // Variably modified types are not supported for tasks. 8546 if (!Type->isAnyPointerType() && Type->isVariablyModifiedType() && 8547 isOpenMPTaskingDirective(CurrDir)) { 8548 Diag(ELoc, diag::err_omp_variably_modified_type_not_supported) 8549 << getOpenMPClauseName(OMPC_private) << Type 8550 << getOpenMPDirectiveName(CurrDir); 8551 bool IsDecl = 8552 !VD || 8553 VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly; 8554 Diag(D->getLocation(), 8555 IsDecl ? diag::note_previous_decl : diag::note_defined_here) 8556 << D; 8557 continue; 8558 } 8559 8560 // OpenMP 4.5 [2.15.5.1, Restrictions, p.3] 8561 // A list item cannot appear in both a map clause and a data-sharing 8562 // attribute clause on the same construct 8563 if (CurrDir == OMPD_target || CurrDir == OMPD_target_parallel || 8564 CurrDir == OMPD_target_teams || 8565 CurrDir == OMPD_target_teams_distribute || 8566 CurrDir == OMPD_target_teams_distribute_parallel_for || 8567 CurrDir == OMPD_target_teams_distribute_parallel_for_simd || 8568 CurrDir == OMPD_target_teams_distribute_simd || 8569 CurrDir == OMPD_target_parallel_for_simd || 8570 CurrDir == OMPD_target_parallel_for) { 8571 OpenMPClauseKind ConflictKind; 8572 if (DSAStack->checkMappableExprComponentListsForDecl( 8573 VD, /*CurrentRegionOnly=*/true, 8574 [&](OMPClauseMappableExprCommon::MappableExprComponentListRef, 8575 OpenMPClauseKind WhereFoundClauseKind) -> bool { 8576 ConflictKind = WhereFoundClauseKind; 8577 return true; 8578 })) { 8579 Diag(ELoc, diag::err_omp_variable_in_given_clause_and_dsa) 8580 << getOpenMPClauseName(OMPC_private) 8581 << getOpenMPClauseName(ConflictKind) 8582 << getOpenMPDirectiveName(CurrDir); 8583 ReportOriginalDSA(*this, DSAStack, D, DVar); 8584 continue; 8585 } 8586 } 8587 8588 // OpenMP [2.9.3.3, Restrictions, C/C++, p.1] 8589 // A variable of class type (or array thereof) that appears in a private 8590 // clause requires an accessible, unambiguous default constructor for the 8591 // class type. 8592 // Generate helper private variable and initialize it with the default 8593 // value. The address of the original variable is replaced by the address of 8594 // the new private variable in CodeGen. This new variable is not added to 8595 // IdResolver, so the code in the OpenMP region uses original variable for 8596 // proper diagnostics. 8597 Type = Type.getUnqualifiedType(); 8598 auto VDPrivate = buildVarDecl(*this, ELoc, Type, D->getName(), 8599 D->hasAttrs() ? &D->getAttrs() : nullptr); 8600 ActOnUninitializedDecl(VDPrivate); 8601 if (VDPrivate->isInvalidDecl()) 8602 continue; 8603 auto VDPrivateRefExpr = buildDeclRefExpr( 8604 *this, VDPrivate, RefExpr->getType().getUnqualifiedType(), ELoc); 8605 8606 DeclRefExpr *Ref = nullptr; 8607 if (!VD && !CurContext->isDependentContext()) 8608 Ref = buildCapture(*this, D, SimpleRefExpr, /*WithInit=*/false); 8609 DSAStack->addDSA(D, RefExpr->IgnoreParens(), OMPC_private, Ref); 8610 Vars.push_back((VD || CurContext->isDependentContext()) 8611 ? RefExpr->IgnoreParens() 8612 : Ref); 8613 PrivateCopies.push_back(VDPrivateRefExpr); 8614 } 8615 8616 if (Vars.empty()) 8617 return nullptr; 8618 8619 return OMPPrivateClause::Create(Context, StartLoc, LParenLoc, EndLoc, Vars, 8620 PrivateCopies); 8621 } 8622 8623 namespace { 8624 class DiagsUninitializedSeveretyRAII { 8625 private: 8626 DiagnosticsEngine &Diags; 8627 SourceLocation SavedLoc; 8628 bool IsIgnored; 8629 8630 public: 8631 DiagsUninitializedSeveretyRAII(DiagnosticsEngine &Diags, SourceLocation Loc, 8632 bool IsIgnored) 8633 : Diags(Diags), SavedLoc(Loc), IsIgnored(IsIgnored) { 8634 if (!IsIgnored) { 8635 Diags.setSeverity(/*Diag*/ diag::warn_uninit_self_reference_in_init, 8636 /*Map*/ diag::Severity::Ignored, Loc); 8637 } 8638 } 8639 ~DiagsUninitializedSeveretyRAII() { 8640 if (!IsIgnored) 8641 Diags.popMappings(SavedLoc); 8642 } 8643 }; 8644 } 8645 8646 OMPClause *Sema::ActOnOpenMPFirstprivateClause(ArrayRef<Expr *> VarList, 8647 SourceLocation StartLoc, 8648 SourceLocation LParenLoc, 8649 SourceLocation EndLoc) { 8650 SmallVector<Expr *, 8> Vars; 8651 SmallVector<Expr *, 8> PrivateCopies; 8652 SmallVector<Expr *, 8> Inits; 8653 SmallVector<Decl *, 4> ExprCaptures; 8654 bool IsImplicitClause = 8655 StartLoc.isInvalid() && LParenLoc.isInvalid() && EndLoc.isInvalid(); 8656 auto ImplicitClauseLoc = DSAStack->getConstructLoc(); 8657 8658 for (auto &RefExpr : VarList) { 8659 assert(RefExpr && "NULL expr in OpenMP firstprivate clause."); 8660 SourceLocation ELoc; 8661 SourceRange ERange; 8662 Expr *SimpleRefExpr = RefExpr; 8663 auto Res = getPrivateItem(*this, SimpleRefExpr, ELoc, ERange); 8664 if (Res.second) { 8665 // It will be analyzed later. 8666 Vars.push_back(RefExpr); 8667 PrivateCopies.push_back(nullptr); 8668 Inits.push_back(nullptr); 8669 } 8670 ValueDecl *D = Res.first; 8671 if (!D) 8672 continue; 8673 8674 ELoc = IsImplicitClause ? ImplicitClauseLoc : ELoc; 8675 QualType Type = D->getType(); 8676 auto *VD = dyn_cast<VarDecl>(D); 8677 8678 // OpenMP [2.9.3.3, Restrictions, C/C++, p.3] 8679 // A variable that appears in a private clause must not have an incomplete 8680 // type or a reference type. 8681 if (RequireCompleteType(ELoc, Type, 8682 diag::err_omp_firstprivate_incomplete_type)) 8683 continue; 8684 Type = Type.getNonReferenceType(); 8685 8686 // OpenMP [2.9.3.4, Restrictions, C/C++, p.1] 8687 // A variable of class type (or array thereof) that appears in a private 8688 // clause requires an accessible, unambiguous copy constructor for the 8689 // class type. 8690 auto ElemType = Context.getBaseElementType(Type).getNonReferenceType(); 8691 8692 // If an implicit firstprivate variable found it was checked already. 8693 DSAStackTy::DSAVarData TopDVar; 8694 if (!IsImplicitClause) { 8695 DSAStackTy::DSAVarData DVar = DSAStack->getTopDSA(D, false); 8696 TopDVar = DVar; 8697 OpenMPDirectiveKind CurrDir = DSAStack->getCurrentDirective(); 8698 bool IsConstant = ElemType.isConstant(Context); 8699 // OpenMP [2.4.13, Data-sharing Attribute Clauses] 8700 // A list item that specifies a given variable may not appear in more 8701 // than one clause on the same directive, except that a variable may be 8702 // specified in both firstprivate and lastprivate clauses. 8703 // OpenMP 4.5 [2.10.8, Distribute Construct, p.3] 8704 // A list item may appear in a firstprivate or lastprivate clause but not 8705 // both. 8706 if (DVar.CKind != OMPC_unknown && DVar.CKind != OMPC_firstprivate && 8707 (CurrDir == OMPD_distribute || DVar.CKind != OMPC_lastprivate) && 8708 DVar.RefExpr) { 8709 Diag(ELoc, diag::err_omp_wrong_dsa) 8710 << getOpenMPClauseName(DVar.CKind) 8711 << getOpenMPClauseName(OMPC_firstprivate); 8712 ReportOriginalDSA(*this, DSAStack, D, DVar); 8713 continue; 8714 } 8715 8716 // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced 8717 // in a Construct] 8718 // Variables with the predetermined data-sharing attributes may not be 8719 // listed in data-sharing attributes clauses, except for the cases 8720 // listed below. For these exceptions only, listing a predetermined 8721 // variable in a data-sharing attribute clause is allowed and overrides 8722 // the variable's predetermined data-sharing attributes. 8723 // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced 8724 // in a Construct, C/C++, p.2] 8725 // Variables with const-qualified type having no mutable member may be 8726 // listed in a firstprivate clause, even if they are static data members. 8727 if (!(IsConstant || (VD && VD->isStaticDataMember())) && !DVar.RefExpr && 8728 DVar.CKind != OMPC_unknown && DVar.CKind != OMPC_shared) { 8729 Diag(ELoc, diag::err_omp_wrong_dsa) 8730 << getOpenMPClauseName(DVar.CKind) 8731 << getOpenMPClauseName(OMPC_firstprivate); 8732 ReportOriginalDSA(*this, DSAStack, D, DVar); 8733 continue; 8734 } 8735 8736 // OpenMP [2.9.3.4, Restrictions, p.2] 8737 // A list item that is private within a parallel region must not appear 8738 // in a firstprivate clause on a worksharing construct if any of the 8739 // worksharing regions arising from the worksharing construct ever bind 8740 // to any of the parallel regions arising from the parallel construct. 8741 // OpenMP 4.5 [2.15.3.4, Restrictions, p.3] 8742 // A list item that is private within a teams region must not appear in a 8743 // firstprivate clause on a distribute construct if any of the distribute 8744 // regions arising from the distribute construct ever bind to any of the 8745 // teams regions arising from the teams construct. 8746 // OpenMP 4.5 [2.15.3.4, Restrictions, p.3] 8747 // A list item that appears in a reduction clause of a teams construct 8748 // must not appear in a firstprivate clause on a distribute construct if 8749 // any of the distribute regions arising from the distribute construct 8750 // ever bind to any of the teams regions arising from the teams construct. 8751 if ((isOpenMPWorksharingDirective(CurrDir) || 8752 isOpenMPDistributeDirective(CurrDir)) && 8753 !isOpenMPParallelDirective(CurrDir) && 8754 !isOpenMPTeamsDirective(CurrDir)) { 8755 DVar = DSAStack->getImplicitDSA(D, true); 8756 if (DVar.CKind != OMPC_shared && 8757 (isOpenMPParallelDirective(DVar.DKind) || 8758 isOpenMPTeamsDirective(DVar.DKind) || 8759 DVar.DKind == OMPD_unknown)) { 8760 Diag(ELoc, diag::err_omp_required_access) 8761 << getOpenMPClauseName(OMPC_firstprivate) 8762 << getOpenMPClauseName(OMPC_shared); 8763 ReportOriginalDSA(*this, DSAStack, D, DVar); 8764 continue; 8765 } 8766 } 8767 // OpenMP [2.9.3.4, Restrictions, p.3] 8768 // A list item that appears in a reduction clause of a parallel construct 8769 // must not appear in a firstprivate clause on a worksharing or task 8770 // construct if any of the worksharing or task regions arising from the 8771 // worksharing or task construct ever bind to any of the parallel regions 8772 // arising from the parallel construct. 8773 // OpenMP [2.9.3.4, Restrictions, p.4] 8774 // A list item that appears in a reduction clause in worksharing 8775 // construct must not appear in a firstprivate clause in a task construct 8776 // encountered during execution of any of the worksharing regions arising 8777 // from the worksharing construct. 8778 if (isOpenMPTaskingDirective(CurrDir)) { 8779 DVar = DSAStack->hasInnermostDSA( 8780 D, [](OpenMPClauseKind C) -> bool { return C == OMPC_reduction; }, 8781 [](OpenMPDirectiveKind K) -> bool { 8782 return isOpenMPParallelDirective(K) || 8783 isOpenMPWorksharingDirective(K) || 8784 isOpenMPTeamsDirective(K); 8785 }, 8786 /*FromParent=*/true); 8787 if (DVar.CKind == OMPC_reduction && 8788 (isOpenMPParallelDirective(DVar.DKind) || 8789 isOpenMPWorksharingDirective(DVar.DKind) || 8790 isOpenMPTeamsDirective(DVar.DKind))) { 8791 Diag(ELoc, diag::err_omp_parallel_reduction_in_task_firstprivate) 8792 << getOpenMPDirectiveName(DVar.DKind); 8793 ReportOriginalDSA(*this, DSAStack, D, DVar); 8794 continue; 8795 } 8796 } 8797 8798 // OpenMP 4.5 [2.15.5.1, Restrictions, p.3] 8799 // A list item cannot appear in both a map clause and a data-sharing 8800 // attribute clause on the same construct 8801 if (CurrDir == OMPD_target || CurrDir == OMPD_target_parallel || 8802 CurrDir == OMPD_target_teams || 8803 CurrDir == OMPD_target_teams_distribute || 8804 CurrDir == OMPD_target_teams_distribute_parallel_for || 8805 CurrDir == OMPD_target_teams_distribute_parallel_for_simd || 8806 CurrDir == OMPD_target_teams_distribute_simd || 8807 CurrDir == OMPD_target_parallel_for_simd || 8808 CurrDir == OMPD_target_parallel_for) { 8809 OpenMPClauseKind ConflictKind; 8810 if (DSAStack->checkMappableExprComponentListsForDecl( 8811 VD, /*CurrentRegionOnly=*/true, 8812 [&](OMPClauseMappableExprCommon::MappableExprComponentListRef, 8813 OpenMPClauseKind WhereFoundClauseKind) -> bool { 8814 ConflictKind = WhereFoundClauseKind; 8815 return true; 8816 })) { 8817 Diag(ELoc, diag::err_omp_variable_in_given_clause_and_dsa) 8818 << getOpenMPClauseName(OMPC_firstprivate) 8819 << getOpenMPClauseName(ConflictKind) 8820 << getOpenMPDirectiveName(DSAStack->getCurrentDirective()); 8821 ReportOriginalDSA(*this, DSAStack, D, DVar); 8822 continue; 8823 } 8824 } 8825 } 8826 8827 // Variably modified types are not supported for tasks. 8828 if (!Type->isAnyPointerType() && Type->isVariablyModifiedType() && 8829 isOpenMPTaskingDirective(DSAStack->getCurrentDirective())) { 8830 Diag(ELoc, diag::err_omp_variably_modified_type_not_supported) 8831 << getOpenMPClauseName(OMPC_firstprivate) << Type 8832 << getOpenMPDirectiveName(DSAStack->getCurrentDirective()); 8833 bool IsDecl = 8834 !VD || 8835 VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly; 8836 Diag(D->getLocation(), 8837 IsDecl ? diag::note_previous_decl : diag::note_defined_here) 8838 << D; 8839 continue; 8840 } 8841 8842 Type = Type.getUnqualifiedType(); 8843 auto VDPrivate = buildVarDecl(*this, ELoc, Type, D->getName(), 8844 D->hasAttrs() ? &D->getAttrs() : nullptr); 8845 // Generate helper private variable and initialize it with the value of the 8846 // original variable. The address of the original variable is replaced by 8847 // the address of the new private variable in the CodeGen. This new variable 8848 // is not added to IdResolver, so the code in the OpenMP region uses 8849 // original variable for proper diagnostics and variable capturing. 8850 Expr *VDInitRefExpr = nullptr; 8851 // For arrays generate initializer for single element and replace it by the 8852 // original array element in CodeGen. 8853 if (Type->isArrayType()) { 8854 auto VDInit = 8855 buildVarDecl(*this, RefExpr->getExprLoc(), ElemType, D->getName()); 8856 VDInitRefExpr = buildDeclRefExpr(*this, VDInit, ElemType, ELoc); 8857 auto Init = DefaultLvalueConversion(VDInitRefExpr).get(); 8858 ElemType = ElemType.getUnqualifiedType(); 8859 auto *VDInitTemp = buildVarDecl(*this, RefExpr->getExprLoc(), ElemType, 8860 ".firstprivate.temp"); 8861 InitializedEntity Entity = 8862 InitializedEntity::InitializeVariable(VDInitTemp); 8863 InitializationKind Kind = InitializationKind::CreateCopy(ELoc, ELoc); 8864 8865 InitializationSequence InitSeq(*this, Entity, Kind, Init); 8866 ExprResult Result = InitSeq.Perform(*this, Entity, Kind, Init); 8867 if (Result.isInvalid()) 8868 VDPrivate->setInvalidDecl(); 8869 else 8870 VDPrivate->setInit(Result.getAs<Expr>()); 8871 // Remove temp variable declaration. 8872 Context.Deallocate(VDInitTemp); 8873 } else { 8874 auto *VDInit = buildVarDecl(*this, RefExpr->getExprLoc(), Type, 8875 ".firstprivate.temp"); 8876 VDInitRefExpr = buildDeclRefExpr(*this, VDInit, RefExpr->getType(), 8877 RefExpr->getExprLoc()); 8878 AddInitializerToDecl(VDPrivate, 8879 DefaultLvalueConversion(VDInitRefExpr).get(), 8880 /*DirectInit=*/false); 8881 } 8882 if (VDPrivate->isInvalidDecl()) { 8883 if (IsImplicitClause) { 8884 Diag(RefExpr->getExprLoc(), 8885 diag::note_omp_task_predetermined_firstprivate_here); 8886 } 8887 continue; 8888 } 8889 CurContext->addDecl(VDPrivate); 8890 auto VDPrivateRefExpr = buildDeclRefExpr( 8891 *this, VDPrivate, RefExpr->getType().getUnqualifiedType(), 8892 RefExpr->getExprLoc()); 8893 DeclRefExpr *Ref = nullptr; 8894 if (!VD && !CurContext->isDependentContext()) { 8895 if (TopDVar.CKind == OMPC_lastprivate) 8896 Ref = TopDVar.PrivateCopy; 8897 else { 8898 Ref = buildCapture(*this, D, SimpleRefExpr, /*WithInit=*/true); 8899 if (!IsOpenMPCapturedDecl(D)) 8900 ExprCaptures.push_back(Ref->getDecl()); 8901 } 8902 } 8903 DSAStack->addDSA(D, RefExpr->IgnoreParens(), OMPC_firstprivate, Ref); 8904 Vars.push_back((VD || CurContext->isDependentContext()) 8905 ? RefExpr->IgnoreParens() 8906 : Ref); 8907 PrivateCopies.push_back(VDPrivateRefExpr); 8908 Inits.push_back(VDInitRefExpr); 8909 } 8910 8911 if (Vars.empty()) 8912 return nullptr; 8913 8914 return OMPFirstprivateClause::Create(Context, StartLoc, LParenLoc, EndLoc, 8915 Vars, PrivateCopies, Inits, 8916 buildPreInits(Context, ExprCaptures)); 8917 } 8918 8919 OMPClause *Sema::ActOnOpenMPLastprivateClause(ArrayRef<Expr *> VarList, 8920 SourceLocation StartLoc, 8921 SourceLocation LParenLoc, 8922 SourceLocation EndLoc) { 8923 SmallVector<Expr *, 8> Vars; 8924 SmallVector<Expr *, 8> SrcExprs; 8925 SmallVector<Expr *, 8> DstExprs; 8926 SmallVector<Expr *, 8> AssignmentOps; 8927 SmallVector<Decl *, 4> ExprCaptures; 8928 SmallVector<Expr *, 4> ExprPostUpdates; 8929 for (auto &RefExpr : VarList) { 8930 assert(RefExpr && "NULL expr in OpenMP lastprivate clause."); 8931 SourceLocation ELoc; 8932 SourceRange ERange; 8933 Expr *SimpleRefExpr = RefExpr; 8934 auto Res = getPrivateItem(*this, SimpleRefExpr, ELoc, ERange); 8935 if (Res.second) { 8936 // It will be analyzed later. 8937 Vars.push_back(RefExpr); 8938 SrcExprs.push_back(nullptr); 8939 DstExprs.push_back(nullptr); 8940 AssignmentOps.push_back(nullptr); 8941 } 8942 ValueDecl *D = Res.first; 8943 if (!D) 8944 continue; 8945 8946 QualType Type = D->getType(); 8947 auto *VD = dyn_cast<VarDecl>(D); 8948 8949 // OpenMP [2.14.3.5, Restrictions, C/C++, p.2] 8950 // A variable that appears in a lastprivate clause must not have an 8951 // incomplete type or a reference type. 8952 if (RequireCompleteType(ELoc, Type, 8953 diag::err_omp_lastprivate_incomplete_type)) 8954 continue; 8955 Type = Type.getNonReferenceType(); 8956 8957 OpenMPDirectiveKind CurrDir = DSAStack->getCurrentDirective(); 8958 // OpenMP [2.14.1.1, Data-sharing Attribute Rules for Variables Referenced 8959 // in a Construct] 8960 // Variables with the predetermined data-sharing attributes may not be 8961 // listed in data-sharing attributes clauses, except for the cases 8962 // listed below. 8963 // OpenMP 4.5 [2.10.8, Distribute Construct, p.3] 8964 // A list item may appear in a firstprivate or lastprivate clause but not 8965 // both. 8966 DSAStackTy::DSAVarData DVar = DSAStack->getTopDSA(D, false); 8967 if (DVar.CKind != OMPC_unknown && DVar.CKind != OMPC_lastprivate && 8968 (CurrDir == OMPD_distribute || DVar.CKind != OMPC_firstprivate) && 8969 (DVar.CKind != OMPC_private || DVar.RefExpr != nullptr)) { 8970 Diag(ELoc, diag::err_omp_wrong_dsa) 8971 << getOpenMPClauseName(DVar.CKind) 8972 << getOpenMPClauseName(OMPC_lastprivate); 8973 ReportOriginalDSA(*this, DSAStack, D, DVar); 8974 continue; 8975 } 8976 8977 // OpenMP [2.14.3.5, Restrictions, p.2] 8978 // A list item that is private within a parallel region, or that appears in 8979 // the reduction clause of a parallel construct, must not appear in a 8980 // lastprivate clause on a worksharing construct if any of the corresponding 8981 // worksharing regions ever binds to any of the corresponding parallel 8982 // regions. 8983 DSAStackTy::DSAVarData TopDVar = DVar; 8984 if (isOpenMPWorksharingDirective(CurrDir) && 8985 !isOpenMPParallelDirective(CurrDir) && 8986 !isOpenMPTeamsDirective(CurrDir)) { 8987 DVar = DSAStack->getImplicitDSA(D, true); 8988 if (DVar.CKind != OMPC_shared) { 8989 Diag(ELoc, diag::err_omp_required_access) 8990 << getOpenMPClauseName(OMPC_lastprivate) 8991 << getOpenMPClauseName(OMPC_shared); 8992 ReportOriginalDSA(*this, DSAStack, D, DVar); 8993 continue; 8994 } 8995 } 8996 8997 // OpenMP [2.14.3.5, Restrictions, C++, p.1,2] 8998 // A variable of class type (or array thereof) that appears in a 8999 // lastprivate clause requires an accessible, unambiguous default 9000 // constructor for the class type, unless the list item is also specified 9001 // in a firstprivate clause. 9002 // A variable of class type (or array thereof) that appears in a 9003 // lastprivate clause requires an accessible, unambiguous copy assignment 9004 // operator for the class type. 9005 Type = Context.getBaseElementType(Type).getNonReferenceType(); 9006 auto *SrcVD = buildVarDecl(*this, ERange.getBegin(), 9007 Type.getUnqualifiedType(), ".lastprivate.src", 9008 D->hasAttrs() ? &D->getAttrs() : nullptr); 9009 auto *PseudoSrcExpr = 9010 buildDeclRefExpr(*this, SrcVD, Type.getUnqualifiedType(), ELoc); 9011 auto *DstVD = 9012 buildVarDecl(*this, ERange.getBegin(), Type, ".lastprivate.dst", 9013 D->hasAttrs() ? &D->getAttrs() : nullptr); 9014 auto *PseudoDstExpr = buildDeclRefExpr(*this, DstVD, Type, ELoc); 9015 // For arrays generate assignment operation for single element and replace 9016 // it by the original array element in CodeGen. 9017 auto AssignmentOp = BuildBinOp(/*S=*/nullptr, ELoc, BO_Assign, 9018 PseudoDstExpr, PseudoSrcExpr); 9019 if (AssignmentOp.isInvalid()) 9020 continue; 9021 AssignmentOp = ActOnFinishFullExpr(AssignmentOp.get(), ELoc, 9022 /*DiscardedValue=*/true); 9023 if (AssignmentOp.isInvalid()) 9024 continue; 9025 9026 DeclRefExpr *Ref = nullptr; 9027 if (!VD && !CurContext->isDependentContext()) { 9028 if (TopDVar.CKind == OMPC_firstprivate) 9029 Ref = TopDVar.PrivateCopy; 9030 else { 9031 Ref = buildCapture(*this, D, SimpleRefExpr, /*WithInit=*/false); 9032 if (!IsOpenMPCapturedDecl(D)) 9033 ExprCaptures.push_back(Ref->getDecl()); 9034 } 9035 if (TopDVar.CKind == OMPC_firstprivate || 9036 (!IsOpenMPCapturedDecl(D) && 9037 Ref->getDecl()->hasAttr<OMPCaptureNoInitAttr>())) { 9038 ExprResult RefRes = DefaultLvalueConversion(Ref); 9039 if (!RefRes.isUsable()) 9040 continue; 9041 ExprResult PostUpdateRes = 9042 BuildBinOp(DSAStack->getCurScope(), ELoc, BO_Assign, SimpleRefExpr, 9043 RefRes.get()); 9044 if (!PostUpdateRes.isUsable()) 9045 continue; 9046 ExprPostUpdates.push_back( 9047 IgnoredValueConversions(PostUpdateRes.get()).get()); 9048 } 9049 } 9050 DSAStack->addDSA(D, RefExpr->IgnoreParens(), OMPC_lastprivate, Ref); 9051 Vars.push_back((VD || CurContext->isDependentContext()) 9052 ? RefExpr->IgnoreParens() 9053 : Ref); 9054 SrcExprs.push_back(PseudoSrcExpr); 9055 DstExprs.push_back(PseudoDstExpr); 9056 AssignmentOps.push_back(AssignmentOp.get()); 9057 } 9058 9059 if (Vars.empty()) 9060 return nullptr; 9061 9062 return OMPLastprivateClause::Create(Context, StartLoc, LParenLoc, EndLoc, 9063 Vars, SrcExprs, DstExprs, AssignmentOps, 9064 buildPreInits(Context, ExprCaptures), 9065 buildPostUpdate(*this, ExprPostUpdates)); 9066 } 9067 9068 OMPClause *Sema::ActOnOpenMPSharedClause(ArrayRef<Expr *> VarList, 9069 SourceLocation StartLoc, 9070 SourceLocation LParenLoc, 9071 SourceLocation EndLoc) { 9072 SmallVector<Expr *, 8> Vars; 9073 for (auto &RefExpr : VarList) { 9074 assert(RefExpr && "NULL expr in OpenMP lastprivate clause."); 9075 SourceLocation ELoc; 9076 SourceRange ERange; 9077 Expr *SimpleRefExpr = RefExpr; 9078 auto Res = getPrivateItem(*this, SimpleRefExpr, ELoc, ERange); 9079 if (Res.second) { 9080 // It will be analyzed later. 9081 Vars.push_back(RefExpr); 9082 } 9083 ValueDecl *D = Res.first; 9084 if (!D) 9085 continue; 9086 9087 auto *VD = dyn_cast<VarDecl>(D); 9088 // OpenMP [2.9.1.1, Data-sharing Attribute Rules for Variables Referenced 9089 // in a Construct] 9090 // Variables with the predetermined data-sharing attributes may not be 9091 // listed in data-sharing attributes clauses, except for the cases 9092 // listed below. For these exceptions only, listing a predetermined 9093 // variable in a data-sharing attribute clause is allowed and overrides 9094 // the variable's predetermined data-sharing attributes. 9095 DSAStackTy::DSAVarData DVar = DSAStack->getTopDSA(D, false); 9096 if (DVar.CKind != OMPC_unknown && DVar.CKind != OMPC_shared && 9097 DVar.RefExpr) { 9098 Diag(ELoc, diag::err_omp_wrong_dsa) << getOpenMPClauseName(DVar.CKind) 9099 << getOpenMPClauseName(OMPC_shared); 9100 ReportOriginalDSA(*this, DSAStack, D, DVar); 9101 continue; 9102 } 9103 9104 DeclRefExpr *Ref = nullptr; 9105 if (!VD && IsOpenMPCapturedDecl(D) && !CurContext->isDependentContext()) 9106 Ref = buildCapture(*this, D, SimpleRefExpr, /*WithInit=*/true); 9107 DSAStack->addDSA(D, RefExpr->IgnoreParens(), OMPC_shared, Ref); 9108 Vars.push_back((VD || !Ref || CurContext->isDependentContext()) 9109 ? RefExpr->IgnoreParens() 9110 : Ref); 9111 } 9112 9113 if (Vars.empty()) 9114 return nullptr; 9115 9116 return OMPSharedClause::Create(Context, StartLoc, LParenLoc, EndLoc, Vars); 9117 } 9118 9119 namespace { 9120 class DSARefChecker : public StmtVisitor<DSARefChecker, bool> { 9121 DSAStackTy *Stack; 9122 9123 public: 9124 bool VisitDeclRefExpr(DeclRefExpr *E) { 9125 if (VarDecl *VD = dyn_cast<VarDecl>(E->getDecl())) { 9126 DSAStackTy::DSAVarData DVar = Stack->getTopDSA(VD, false); 9127 if (DVar.CKind == OMPC_shared && !DVar.RefExpr) 9128 return false; 9129 if (DVar.CKind != OMPC_unknown) 9130 return true; 9131 DSAStackTy::DSAVarData DVarPrivate = Stack->hasDSA( 9132 VD, isOpenMPPrivate, [](OpenMPDirectiveKind) -> bool { return true; }, 9133 /*FromParent=*/true); 9134 if (DVarPrivate.CKind != OMPC_unknown) 9135 return true; 9136 return false; 9137 } 9138 return false; 9139 } 9140 bool VisitStmt(Stmt *S) { 9141 for (auto Child : S->children()) { 9142 if (Child && Visit(Child)) 9143 return true; 9144 } 9145 return false; 9146 } 9147 explicit DSARefChecker(DSAStackTy *S) : Stack(S) {} 9148 }; 9149 } // namespace 9150 9151 namespace { 9152 // Transform MemberExpression for specified FieldDecl of current class to 9153 // DeclRefExpr to specified OMPCapturedExprDecl. 9154 class TransformExprToCaptures : public TreeTransform<TransformExprToCaptures> { 9155 typedef TreeTransform<TransformExprToCaptures> BaseTransform; 9156 ValueDecl *Field; 9157 DeclRefExpr *CapturedExpr; 9158 9159 public: 9160 TransformExprToCaptures(Sema &SemaRef, ValueDecl *FieldDecl) 9161 : BaseTransform(SemaRef), Field(FieldDecl), CapturedExpr(nullptr) {} 9162 9163 ExprResult TransformMemberExpr(MemberExpr *E) { 9164 if (isa<CXXThisExpr>(E->getBase()->IgnoreParenImpCasts()) && 9165 E->getMemberDecl() == Field) { 9166 CapturedExpr = buildCapture(SemaRef, Field, E, /*WithInit=*/false); 9167 return CapturedExpr; 9168 } 9169 return BaseTransform::TransformMemberExpr(E); 9170 } 9171 DeclRefExpr *getCapturedExpr() { return CapturedExpr; } 9172 }; 9173 } // namespace 9174 9175 template <typename T> 9176 static T filterLookupForUDR(SmallVectorImpl<UnresolvedSet<8>> &Lookups, 9177 const llvm::function_ref<T(ValueDecl *)> &Gen) { 9178 for (auto &Set : Lookups) { 9179 for (auto *D : Set) { 9180 if (auto Res = Gen(cast<ValueDecl>(D))) 9181 return Res; 9182 } 9183 } 9184 return T(); 9185 } 9186 9187 static ExprResult 9188 buildDeclareReductionRef(Sema &SemaRef, SourceLocation Loc, SourceRange Range, 9189 Scope *S, CXXScopeSpec &ReductionIdScopeSpec, 9190 const DeclarationNameInfo &ReductionId, QualType Ty, 9191 CXXCastPath &BasePath, Expr *UnresolvedReduction) { 9192 if (ReductionIdScopeSpec.isInvalid()) 9193 return ExprError(); 9194 SmallVector<UnresolvedSet<8>, 4> Lookups; 9195 if (S) { 9196 LookupResult Lookup(SemaRef, ReductionId, Sema::LookupOMPReductionName); 9197 Lookup.suppressDiagnostics(); 9198 while (S && SemaRef.LookupParsedName(Lookup, S, &ReductionIdScopeSpec)) { 9199 auto *D = Lookup.getRepresentativeDecl(); 9200 do { 9201 S = S->getParent(); 9202 } while (S && !S->isDeclScope(D)); 9203 if (S) 9204 S = S->getParent(); 9205 Lookups.push_back(UnresolvedSet<8>()); 9206 Lookups.back().append(Lookup.begin(), Lookup.end()); 9207 Lookup.clear(); 9208 } 9209 } else if (auto *ULE = 9210 cast_or_null<UnresolvedLookupExpr>(UnresolvedReduction)) { 9211 Lookups.push_back(UnresolvedSet<8>()); 9212 Decl *PrevD = nullptr; 9213 for (auto *D : ULE->decls()) { 9214 if (D == PrevD) 9215 Lookups.push_back(UnresolvedSet<8>()); 9216 else if (auto *DRD = cast<OMPDeclareReductionDecl>(D)) 9217 Lookups.back().addDecl(DRD); 9218 PrevD = D; 9219 } 9220 } 9221 if (SemaRef.CurContext->isDependentContext() || Ty->isDependentType() || 9222 Ty->isInstantiationDependentType() || 9223 Ty->containsUnexpandedParameterPack() || 9224 filterLookupForUDR<bool>(Lookups, [](ValueDecl *D) -> bool { 9225 return !D->isInvalidDecl() && 9226 (D->getType()->isDependentType() || 9227 D->getType()->isInstantiationDependentType() || 9228 D->getType()->containsUnexpandedParameterPack()); 9229 })) { 9230 UnresolvedSet<8> ResSet; 9231 for (auto &Set : Lookups) { 9232 ResSet.append(Set.begin(), Set.end()); 9233 // The last item marks the end of all declarations at the specified scope. 9234 ResSet.addDecl(Set[Set.size() - 1]); 9235 } 9236 return UnresolvedLookupExpr::Create( 9237 SemaRef.Context, /*NamingClass=*/nullptr, 9238 ReductionIdScopeSpec.getWithLocInContext(SemaRef.Context), ReductionId, 9239 /*ADL=*/true, /*Overloaded=*/true, ResSet.begin(), ResSet.end()); 9240 } 9241 if (auto *VD = filterLookupForUDR<ValueDecl *>( 9242 Lookups, [&SemaRef, Ty](ValueDecl *D) -> ValueDecl * { 9243 if (!D->isInvalidDecl() && 9244 SemaRef.Context.hasSameType(D->getType(), Ty)) 9245 return D; 9246 return nullptr; 9247 })) 9248 return SemaRef.BuildDeclRefExpr(VD, Ty, VK_LValue, Loc); 9249 if (auto *VD = filterLookupForUDR<ValueDecl *>( 9250 Lookups, [&SemaRef, Ty, Loc](ValueDecl *D) -> ValueDecl * { 9251 if (!D->isInvalidDecl() && 9252 SemaRef.IsDerivedFrom(Loc, Ty, D->getType()) && 9253 !Ty.isMoreQualifiedThan(D->getType())) 9254 return D; 9255 return nullptr; 9256 })) { 9257 CXXBasePaths Paths(/*FindAmbiguities=*/true, /*RecordPaths=*/true, 9258 /*DetectVirtual=*/false); 9259 if (SemaRef.IsDerivedFrom(Loc, Ty, VD->getType(), Paths)) { 9260 if (!Paths.isAmbiguous(SemaRef.Context.getCanonicalType( 9261 VD->getType().getUnqualifiedType()))) { 9262 if (SemaRef.CheckBaseClassAccess(Loc, VD->getType(), Ty, Paths.front(), 9263 /*DiagID=*/0) != 9264 Sema::AR_inaccessible) { 9265 SemaRef.BuildBasePathArray(Paths, BasePath); 9266 return SemaRef.BuildDeclRefExpr(VD, Ty, VK_LValue, Loc); 9267 } 9268 } 9269 } 9270 } 9271 if (ReductionIdScopeSpec.isSet()) { 9272 SemaRef.Diag(Loc, diag::err_omp_not_resolved_reduction_identifier) << Range; 9273 return ExprError(); 9274 } 9275 return ExprEmpty(); 9276 } 9277 9278 namespace { 9279 /// Data for the reduction-based clauses. 9280 struct ReductionData { 9281 /// List of original reduction items. 9282 SmallVector<Expr *, 8> Vars; 9283 /// List of private copies of the reduction items. 9284 SmallVector<Expr *, 8> Privates; 9285 /// LHS expressions for the reduction_op expressions. 9286 SmallVector<Expr *, 8> LHSs; 9287 /// RHS expressions for the reduction_op expressions. 9288 SmallVector<Expr *, 8> RHSs; 9289 /// Reduction operation expression. 9290 SmallVector<Expr *, 8> ReductionOps; 9291 /// Taskgroup descriptors for the corresponding reduction items in 9292 /// in_reduction clauses. 9293 SmallVector<Expr *, 8> TaskgroupDescriptors; 9294 /// List of captures for clause. 9295 SmallVector<Decl *, 4> ExprCaptures; 9296 /// List of postupdate expressions. 9297 SmallVector<Expr *, 4> ExprPostUpdates; 9298 ReductionData() = delete; 9299 /// Reserves required memory for the reduction data. 9300 ReductionData(unsigned Size) { 9301 Vars.reserve(Size); 9302 Privates.reserve(Size); 9303 LHSs.reserve(Size); 9304 RHSs.reserve(Size); 9305 ReductionOps.reserve(Size); 9306 TaskgroupDescriptors.reserve(Size); 9307 ExprCaptures.reserve(Size); 9308 ExprPostUpdates.reserve(Size); 9309 } 9310 /// Stores reduction item and reduction operation only (required for dependent 9311 /// reduction item). 9312 void push(Expr *Item, Expr *ReductionOp) { 9313 Vars.emplace_back(Item); 9314 Privates.emplace_back(nullptr); 9315 LHSs.emplace_back(nullptr); 9316 RHSs.emplace_back(nullptr); 9317 ReductionOps.emplace_back(ReductionOp); 9318 TaskgroupDescriptors.emplace_back(nullptr); 9319 } 9320 /// Stores reduction data. 9321 void push(Expr *Item, Expr *Private, Expr *LHS, Expr *RHS, Expr *ReductionOp, 9322 Expr *TaskgroupDescriptor) { 9323 Vars.emplace_back(Item); 9324 Privates.emplace_back(Private); 9325 LHSs.emplace_back(LHS); 9326 RHSs.emplace_back(RHS); 9327 ReductionOps.emplace_back(ReductionOp); 9328 TaskgroupDescriptors.emplace_back(TaskgroupDescriptor); 9329 } 9330 }; 9331 } // namespace 9332 9333 static bool ActOnOMPReductionKindClause( 9334 Sema &S, DSAStackTy *Stack, OpenMPClauseKind ClauseKind, 9335 ArrayRef<Expr *> VarList, SourceLocation StartLoc, SourceLocation LParenLoc, 9336 SourceLocation ColonLoc, SourceLocation EndLoc, 9337 CXXScopeSpec &ReductionIdScopeSpec, const DeclarationNameInfo &ReductionId, 9338 ArrayRef<Expr *> UnresolvedReductions, ReductionData &RD) { 9339 auto DN = ReductionId.getName(); 9340 auto OOK = DN.getCXXOverloadedOperator(); 9341 BinaryOperatorKind BOK = BO_Comma; 9342 9343 ASTContext &Context = S.Context; 9344 // OpenMP [2.14.3.6, reduction clause] 9345 // C 9346 // reduction-identifier is either an identifier or one of the following 9347 // operators: +, -, *, &, |, ^, && and || 9348 // C++ 9349 // reduction-identifier is either an id-expression or one of the following 9350 // operators: +, -, *, &, |, ^, && and || 9351 switch (OOK) { 9352 case OO_Plus: 9353 case OO_Minus: 9354 BOK = BO_Add; 9355 break; 9356 case OO_Star: 9357 BOK = BO_Mul; 9358 break; 9359 case OO_Amp: 9360 BOK = BO_And; 9361 break; 9362 case OO_Pipe: 9363 BOK = BO_Or; 9364 break; 9365 case OO_Caret: 9366 BOK = BO_Xor; 9367 break; 9368 case OO_AmpAmp: 9369 BOK = BO_LAnd; 9370 break; 9371 case OO_PipePipe: 9372 BOK = BO_LOr; 9373 break; 9374 case OO_New: 9375 case OO_Delete: 9376 case OO_Array_New: 9377 case OO_Array_Delete: 9378 case OO_Slash: 9379 case OO_Percent: 9380 case OO_Tilde: 9381 case OO_Exclaim: 9382 case OO_Equal: 9383 case OO_Less: 9384 case OO_Greater: 9385 case OO_LessEqual: 9386 case OO_GreaterEqual: 9387 case OO_PlusEqual: 9388 case OO_MinusEqual: 9389 case OO_StarEqual: 9390 case OO_SlashEqual: 9391 case OO_PercentEqual: 9392 case OO_CaretEqual: 9393 case OO_AmpEqual: 9394 case OO_PipeEqual: 9395 case OO_LessLess: 9396 case OO_GreaterGreater: 9397 case OO_LessLessEqual: 9398 case OO_GreaterGreaterEqual: 9399 case OO_EqualEqual: 9400 case OO_ExclaimEqual: 9401 case OO_PlusPlus: 9402 case OO_MinusMinus: 9403 case OO_Comma: 9404 case OO_ArrowStar: 9405 case OO_Arrow: 9406 case OO_Call: 9407 case OO_Subscript: 9408 case OO_Conditional: 9409 case OO_Coawait: 9410 case NUM_OVERLOADED_OPERATORS: 9411 llvm_unreachable("Unexpected reduction identifier"); 9412 case OO_None: 9413 if (auto *II = DN.getAsIdentifierInfo()) { 9414 if (II->isStr("max")) 9415 BOK = BO_GT; 9416 else if (II->isStr("min")) 9417 BOK = BO_LT; 9418 } 9419 break; 9420 } 9421 SourceRange ReductionIdRange; 9422 if (ReductionIdScopeSpec.isValid()) 9423 ReductionIdRange.setBegin(ReductionIdScopeSpec.getBeginLoc()); 9424 else 9425 ReductionIdRange.setBegin(ReductionId.getBeginLoc()); 9426 ReductionIdRange.setEnd(ReductionId.getEndLoc()); 9427 9428 auto IR = UnresolvedReductions.begin(), ER = UnresolvedReductions.end(); 9429 bool FirstIter = true; 9430 for (auto RefExpr : VarList) { 9431 assert(RefExpr && "nullptr expr in OpenMP reduction clause."); 9432 // OpenMP [2.1, C/C++] 9433 // A list item is a variable or array section, subject to the restrictions 9434 // specified in Section 2.4 on page 42 and in each of the sections 9435 // describing clauses and directives for which a list appears. 9436 // OpenMP [2.14.3.3, Restrictions, p.1] 9437 // A variable that is part of another variable (as an array or 9438 // structure element) cannot appear in a private clause. 9439 if (!FirstIter && IR != ER) 9440 ++IR; 9441 FirstIter = false; 9442 SourceLocation ELoc; 9443 SourceRange ERange; 9444 Expr *SimpleRefExpr = RefExpr; 9445 auto Res = getPrivateItem(S, SimpleRefExpr, ELoc, ERange, 9446 /*AllowArraySection=*/true); 9447 if (Res.second) { 9448 // Try to find 'declare reduction' corresponding construct before using 9449 // builtin/overloaded operators. 9450 QualType Type = Context.DependentTy; 9451 CXXCastPath BasePath; 9452 ExprResult DeclareReductionRef = buildDeclareReductionRef( 9453 S, ELoc, ERange, Stack->getCurScope(), ReductionIdScopeSpec, 9454 ReductionId, Type, BasePath, IR == ER ? nullptr : *IR); 9455 Expr *ReductionOp = nullptr; 9456 if (S.CurContext->isDependentContext() && 9457 (DeclareReductionRef.isUnset() || 9458 isa<UnresolvedLookupExpr>(DeclareReductionRef.get()))) 9459 ReductionOp = DeclareReductionRef.get(); 9460 // It will be analyzed later. 9461 RD.push(RefExpr, ReductionOp); 9462 } 9463 ValueDecl *D = Res.first; 9464 if (!D) 9465 continue; 9466 9467 Expr *TaskgroupDescriptor = nullptr; 9468 QualType Type; 9469 auto *ASE = dyn_cast<ArraySubscriptExpr>(RefExpr->IgnoreParens()); 9470 auto *OASE = dyn_cast<OMPArraySectionExpr>(RefExpr->IgnoreParens()); 9471 if (ASE) 9472 Type = ASE->getType().getNonReferenceType(); 9473 else if (OASE) { 9474 auto BaseType = OMPArraySectionExpr::getBaseOriginalType(OASE->getBase()); 9475 if (auto *ATy = BaseType->getAsArrayTypeUnsafe()) 9476 Type = ATy->getElementType(); 9477 else 9478 Type = BaseType->getPointeeType(); 9479 Type = Type.getNonReferenceType(); 9480 } else 9481 Type = Context.getBaseElementType(D->getType().getNonReferenceType()); 9482 auto *VD = dyn_cast<VarDecl>(D); 9483 9484 // OpenMP [2.9.3.3, Restrictions, C/C++, p.3] 9485 // A variable that appears in a private clause must not have an incomplete 9486 // type or a reference type. 9487 if (S.RequireCompleteType(ELoc, Type, 9488 diag::err_omp_reduction_incomplete_type)) 9489 continue; 9490 // OpenMP [2.14.3.6, reduction clause, Restrictions] 9491 // A list item that appears in a reduction clause must not be 9492 // const-qualified. 9493 if (Type.getNonReferenceType().isConstant(Context)) { 9494 S.Diag(ELoc, diag::err_omp_const_reduction_list_item) << ERange; 9495 if (!ASE && !OASE) { 9496 bool IsDecl = !VD || VD->isThisDeclarationADefinition(Context) == 9497 VarDecl::DeclarationOnly; 9498 S.Diag(D->getLocation(), 9499 IsDecl ? diag::note_previous_decl : diag::note_defined_here) 9500 << D; 9501 } 9502 continue; 9503 } 9504 // OpenMP [2.9.3.6, Restrictions, C/C++, p.4] 9505 // If a list-item is a reference type then it must bind to the same object 9506 // for all threads of the team. 9507 if (!ASE && !OASE && VD) { 9508 VarDecl *VDDef = VD->getDefinition(); 9509 if (VD->getType()->isReferenceType() && VDDef && VDDef->hasInit()) { 9510 DSARefChecker Check(Stack); 9511 if (Check.Visit(VDDef->getInit())) { 9512 S.Diag(ELoc, diag::err_omp_reduction_ref_type_arg) 9513 << getOpenMPClauseName(ClauseKind) << ERange; 9514 S.Diag(VDDef->getLocation(), diag::note_defined_here) << VDDef; 9515 continue; 9516 } 9517 } 9518 } 9519 9520 // OpenMP [2.14.1.1, Data-sharing Attribute Rules for Variables Referenced 9521 // in a Construct] 9522 // Variables with the predetermined data-sharing attributes may not be 9523 // listed in data-sharing attributes clauses, except for the cases 9524 // listed below. For these exceptions only, listing a predetermined 9525 // variable in a data-sharing attribute clause is allowed and overrides 9526 // the variable's predetermined data-sharing attributes. 9527 // OpenMP [2.14.3.6, Restrictions, p.3] 9528 // Any number of reduction clauses can be specified on the directive, 9529 // but a list item can appear only once in the reduction clauses for that 9530 // directive. 9531 DSAStackTy::DSAVarData DVar; 9532 DVar = Stack->getTopDSA(D, false); 9533 if (DVar.CKind == OMPC_reduction) { 9534 S.Diag(ELoc, diag::err_omp_once_referenced) 9535 << getOpenMPClauseName(ClauseKind); 9536 if (DVar.RefExpr) 9537 S.Diag(DVar.RefExpr->getExprLoc(), diag::note_omp_referenced); 9538 continue; 9539 } else if (DVar.CKind != OMPC_unknown) { 9540 S.Diag(ELoc, diag::err_omp_wrong_dsa) 9541 << getOpenMPClauseName(DVar.CKind) 9542 << getOpenMPClauseName(OMPC_reduction); 9543 ReportOriginalDSA(S, Stack, D, DVar); 9544 continue; 9545 } 9546 9547 // OpenMP [2.14.3.6, Restrictions, p.1] 9548 // A list item that appears in a reduction clause of a worksharing 9549 // construct must be shared in the parallel regions to which any of the 9550 // worksharing regions arising from the worksharing construct bind. 9551 OpenMPDirectiveKind CurrDir = Stack->getCurrentDirective(); 9552 if (isOpenMPWorksharingDirective(CurrDir) && 9553 !isOpenMPParallelDirective(CurrDir) && 9554 !isOpenMPTeamsDirective(CurrDir)) { 9555 DVar = Stack->getImplicitDSA(D, true); 9556 if (DVar.CKind != OMPC_shared) { 9557 S.Diag(ELoc, diag::err_omp_required_access) 9558 << getOpenMPClauseName(OMPC_reduction) 9559 << getOpenMPClauseName(OMPC_shared); 9560 ReportOriginalDSA(S, Stack, D, DVar); 9561 continue; 9562 } 9563 } 9564 9565 // Try to find 'declare reduction' corresponding construct before using 9566 // builtin/overloaded operators. 9567 CXXCastPath BasePath; 9568 ExprResult DeclareReductionRef = buildDeclareReductionRef( 9569 S, ELoc, ERange, Stack->getCurScope(), ReductionIdScopeSpec, 9570 ReductionId, Type, BasePath, IR == ER ? nullptr : *IR); 9571 if (DeclareReductionRef.isInvalid()) 9572 continue; 9573 if (S.CurContext->isDependentContext() && 9574 (DeclareReductionRef.isUnset() || 9575 isa<UnresolvedLookupExpr>(DeclareReductionRef.get()))) { 9576 RD.push(RefExpr, DeclareReductionRef.get()); 9577 continue; 9578 } 9579 if (BOK == BO_Comma && DeclareReductionRef.isUnset()) { 9580 // Not allowed reduction identifier is found. 9581 S.Diag(ReductionId.getLocStart(), 9582 diag::err_omp_unknown_reduction_identifier) 9583 << Type << ReductionIdRange; 9584 continue; 9585 } 9586 9587 // OpenMP [2.14.3.6, reduction clause, Restrictions] 9588 // The type of a list item that appears in a reduction clause must be valid 9589 // for the reduction-identifier. For a max or min reduction in C, the type 9590 // of the list item must be an allowed arithmetic data type: char, int, 9591 // float, double, or _Bool, possibly modified with long, short, signed, or 9592 // unsigned. For a max or min reduction in C++, the type of the list item 9593 // must be an allowed arithmetic data type: char, wchar_t, int, float, 9594 // double, or bool, possibly modified with long, short, signed, or unsigned. 9595 if (DeclareReductionRef.isUnset()) { 9596 if ((BOK == BO_GT || BOK == BO_LT) && 9597 !(Type->isScalarType() || 9598 (S.getLangOpts().CPlusPlus && Type->isArithmeticType()))) { 9599 S.Diag(ELoc, diag::err_omp_clause_not_arithmetic_type_arg) 9600 << getOpenMPClauseName(ClauseKind) << S.getLangOpts().CPlusPlus; 9601 if (!ASE && !OASE) { 9602 bool IsDecl = !VD || VD->isThisDeclarationADefinition(Context) == 9603 VarDecl::DeclarationOnly; 9604 S.Diag(D->getLocation(), 9605 IsDecl ? diag::note_previous_decl : diag::note_defined_here) 9606 << D; 9607 } 9608 continue; 9609 } 9610 if ((BOK == BO_OrAssign || BOK == BO_AndAssign || BOK == BO_XorAssign) && 9611 !S.getLangOpts().CPlusPlus && Type->isFloatingType()) { 9612 S.Diag(ELoc, diag::err_omp_clause_floating_type_arg) 9613 << getOpenMPClauseName(ClauseKind); 9614 if (!ASE && !OASE) { 9615 bool IsDecl = !VD || VD->isThisDeclarationADefinition(Context) == 9616 VarDecl::DeclarationOnly; 9617 S.Diag(D->getLocation(), 9618 IsDecl ? diag::note_previous_decl : diag::note_defined_here) 9619 << D; 9620 } 9621 continue; 9622 } 9623 } 9624 9625 Type = Type.getNonLValueExprType(Context).getUnqualifiedType(); 9626 auto *LHSVD = buildVarDecl(S, ELoc, Type, ".reduction.lhs", 9627 D->hasAttrs() ? &D->getAttrs() : nullptr); 9628 auto *RHSVD = buildVarDecl(S, ELoc, Type, D->getName(), 9629 D->hasAttrs() ? &D->getAttrs() : nullptr); 9630 auto PrivateTy = Type; 9631 if (OASE || 9632 (!ASE && 9633 D->getType().getNonReferenceType()->isVariablyModifiedType())) { 9634 // For arrays/array sections only: 9635 // Create pseudo array type for private copy. The size for this array will 9636 // be generated during codegen. 9637 // For array subscripts or single variables Private Ty is the same as Type 9638 // (type of the variable or single array element). 9639 PrivateTy = Context.getVariableArrayType( 9640 Type, 9641 new (Context) OpaqueValueExpr(SourceLocation(), Context.getSizeType(), 9642 VK_RValue), 9643 ArrayType::Normal, /*IndexTypeQuals=*/0, SourceRange()); 9644 } else if (!ASE && !OASE && 9645 Context.getAsArrayType(D->getType().getNonReferenceType())) 9646 PrivateTy = D->getType().getNonReferenceType(); 9647 // Private copy. 9648 auto *PrivateVD = buildVarDecl(S, ELoc, PrivateTy, D->getName(), 9649 D->hasAttrs() ? &D->getAttrs() : nullptr); 9650 // Add initializer for private variable. 9651 Expr *Init = nullptr; 9652 auto *LHSDRE = buildDeclRefExpr(S, LHSVD, Type, ELoc); 9653 auto *RHSDRE = buildDeclRefExpr(S, RHSVD, Type, ELoc); 9654 if (DeclareReductionRef.isUsable()) { 9655 auto *DRDRef = DeclareReductionRef.getAs<DeclRefExpr>(); 9656 auto *DRD = cast<OMPDeclareReductionDecl>(DRDRef->getDecl()); 9657 if (DRD->getInitializer()) { 9658 Init = DRDRef; 9659 RHSVD->setInit(DRDRef); 9660 RHSVD->setInitStyle(VarDecl::CallInit); 9661 } 9662 } else { 9663 switch (BOK) { 9664 case BO_Add: 9665 case BO_Xor: 9666 case BO_Or: 9667 case BO_LOr: 9668 // '+', '-', '^', '|', '||' reduction ops - initializer is '0'. 9669 if (Type->isScalarType() || Type->isAnyComplexType()) 9670 Init = S.ActOnIntegerConstant(ELoc, /*Val=*/0).get(); 9671 break; 9672 case BO_Mul: 9673 case BO_LAnd: 9674 if (Type->isScalarType() || Type->isAnyComplexType()) { 9675 // '*' and '&&' reduction ops - initializer is '1'. 9676 Init = S.ActOnIntegerConstant(ELoc, /*Val=*/1).get(); 9677 } 9678 break; 9679 case BO_And: { 9680 // '&' reduction op - initializer is '~0'. 9681 QualType OrigType = Type; 9682 if (auto *ComplexTy = OrigType->getAs<ComplexType>()) 9683 Type = ComplexTy->getElementType(); 9684 if (Type->isRealFloatingType()) { 9685 llvm::APFloat InitValue = 9686 llvm::APFloat::getAllOnesValue(Context.getTypeSize(Type), 9687 /*isIEEE=*/true); 9688 Init = FloatingLiteral::Create(Context, InitValue, /*isexact=*/true, 9689 Type, ELoc); 9690 } else if (Type->isScalarType()) { 9691 auto Size = Context.getTypeSize(Type); 9692 QualType IntTy = Context.getIntTypeForBitwidth(Size, /*Signed=*/0); 9693 llvm::APInt InitValue = llvm::APInt::getAllOnesValue(Size); 9694 Init = IntegerLiteral::Create(Context, InitValue, IntTy, ELoc); 9695 } 9696 if (Init && OrigType->isAnyComplexType()) { 9697 // Init = 0xFFFF + 0xFFFFi; 9698 auto *Im = new (Context) ImaginaryLiteral(Init, OrigType); 9699 Init = S.CreateBuiltinBinOp(ELoc, BO_Add, Init, Im).get(); 9700 } 9701 Type = OrigType; 9702 break; 9703 } 9704 case BO_LT: 9705 case BO_GT: { 9706 // 'min' reduction op - initializer is 'Largest representable number in 9707 // the reduction list item type'. 9708 // 'max' reduction op - initializer is 'Least representable number in 9709 // the reduction list item type'. 9710 if (Type->isIntegerType() || Type->isPointerType()) { 9711 bool IsSigned = Type->hasSignedIntegerRepresentation(); 9712 auto Size = Context.getTypeSize(Type); 9713 QualType IntTy = 9714 Context.getIntTypeForBitwidth(Size, /*Signed=*/IsSigned); 9715 llvm::APInt InitValue = 9716 (BOK != BO_LT) ? IsSigned ? llvm::APInt::getSignedMinValue(Size) 9717 : llvm::APInt::getMinValue(Size) 9718 : IsSigned ? llvm::APInt::getSignedMaxValue(Size) 9719 : llvm::APInt::getMaxValue(Size); 9720 Init = IntegerLiteral::Create(Context, InitValue, IntTy, ELoc); 9721 if (Type->isPointerType()) { 9722 // Cast to pointer type. 9723 auto CastExpr = S.BuildCStyleCastExpr( 9724 SourceLocation(), Context.getTrivialTypeSourceInfo(Type, ELoc), 9725 SourceLocation(), Init); 9726 if (CastExpr.isInvalid()) 9727 continue; 9728 Init = CastExpr.get(); 9729 } 9730 } else if (Type->isRealFloatingType()) { 9731 llvm::APFloat InitValue = llvm::APFloat::getLargest( 9732 Context.getFloatTypeSemantics(Type), BOK != BO_LT); 9733 Init = FloatingLiteral::Create(Context, InitValue, /*isexact=*/true, 9734 Type, ELoc); 9735 } 9736 break; 9737 } 9738 case BO_PtrMemD: 9739 case BO_PtrMemI: 9740 case BO_MulAssign: 9741 case BO_Div: 9742 case BO_Rem: 9743 case BO_Sub: 9744 case BO_Shl: 9745 case BO_Shr: 9746 case BO_LE: 9747 case BO_GE: 9748 case BO_EQ: 9749 case BO_NE: 9750 case BO_AndAssign: 9751 case BO_XorAssign: 9752 case BO_OrAssign: 9753 case BO_Assign: 9754 case BO_AddAssign: 9755 case BO_SubAssign: 9756 case BO_DivAssign: 9757 case BO_RemAssign: 9758 case BO_ShlAssign: 9759 case BO_ShrAssign: 9760 case BO_Comma: 9761 llvm_unreachable("Unexpected reduction operation"); 9762 } 9763 } 9764 if (Init && DeclareReductionRef.isUnset()) 9765 S.AddInitializerToDecl(RHSVD, Init, /*DirectInit=*/false); 9766 else if (!Init) 9767 S.ActOnUninitializedDecl(RHSVD); 9768 if (RHSVD->isInvalidDecl()) 9769 continue; 9770 if (!RHSVD->hasInit() && DeclareReductionRef.isUnset()) { 9771 S.Diag(ELoc, diag::err_omp_reduction_id_not_compatible) 9772 << Type << ReductionIdRange; 9773 bool IsDecl = !VD || VD->isThisDeclarationADefinition(Context) == 9774 VarDecl::DeclarationOnly; 9775 S.Diag(D->getLocation(), 9776 IsDecl ? diag::note_previous_decl : diag::note_defined_here) 9777 << D; 9778 continue; 9779 } 9780 // Store initializer for single element in private copy. Will be used during 9781 // codegen. 9782 PrivateVD->setInit(RHSVD->getInit()); 9783 PrivateVD->setInitStyle(RHSVD->getInitStyle()); 9784 auto *PrivateDRE = buildDeclRefExpr(S, PrivateVD, PrivateTy, ELoc); 9785 ExprResult ReductionOp; 9786 if (DeclareReductionRef.isUsable()) { 9787 QualType RedTy = DeclareReductionRef.get()->getType(); 9788 QualType PtrRedTy = Context.getPointerType(RedTy); 9789 ExprResult LHS = S.CreateBuiltinUnaryOp(ELoc, UO_AddrOf, LHSDRE); 9790 ExprResult RHS = S.CreateBuiltinUnaryOp(ELoc, UO_AddrOf, RHSDRE); 9791 if (!BasePath.empty()) { 9792 LHS = S.DefaultLvalueConversion(LHS.get()); 9793 RHS = S.DefaultLvalueConversion(RHS.get()); 9794 LHS = ImplicitCastExpr::Create(Context, PtrRedTy, 9795 CK_UncheckedDerivedToBase, LHS.get(), 9796 &BasePath, LHS.get()->getValueKind()); 9797 RHS = ImplicitCastExpr::Create(Context, PtrRedTy, 9798 CK_UncheckedDerivedToBase, RHS.get(), 9799 &BasePath, RHS.get()->getValueKind()); 9800 } 9801 FunctionProtoType::ExtProtoInfo EPI; 9802 QualType Params[] = {PtrRedTy, PtrRedTy}; 9803 QualType FnTy = Context.getFunctionType(Context.VoidTy, Params, EPI); 9804 auto *OVE = new (Context) OpaqueValueExpr( 9805 ELoc, Context.getPointerType(FnTy), VK_RValue, OK_Ordinary, 9806 S.DefaultLvalueConversion(DeclareReductionRef.get()).get()); 9807 Expr *Args[] = {LHS.get(), RHS.get()}; 9808 ReductionOp = new (Context) 9809 CallExpr(Context, OVE, Args, Context.VoidTy, VK_RValue, ELoc); 9810 } else { 9811 ReductionOp = S.BuildBinOp( 9812 Stack->getCurScope(), ReductionId.getLocStart(), BOK, LHSDRE, RHSDRE); 9813 if (ReductionOp.isUsable()) { 9814 if (BOK != BO_LT && BOK != BO_GT) { 9815 ReductionOp = 9816 S.BuildBinOp(Stack->getCurScope(), ReductionId.getLocStart(), 9817 BO_Assign, LHSDRE, ReductionOp.get()); 9818 } else { 9819 auto *ConditionalOp = new (Context) ConditionalOperator( 9820 ReductionOp.get(), SourceLocation(), LHSDRE, SourceLocation(), 9821 RHSDRE, Type, VK_LValue, OK_Ordinary); 9822 ReductionOp = 9823 S.BuildBinOp(Stack->getCurScope(), ReductionId.getLocStart(), 9824 BO_Assign, LHSDRE, ConditionalOp); 9825 } 9826 if (ReductionOp.isUsable()) 9827 ReductionOp = S.ActOnFinishFullExpr(ReductionOp.get()); 9828 } 9829 if (!ReductionOp.isUsable()) 9830 continue; 9831 } 9832 9833 // OpenMP [2.15.4.6, Restrictions, p.2] 9834 // A list item that appears in an in_reduction clause of a task construct 9835 // must appear in a task_reduction clause of a construct associated with a 9836 // taskgroup region that includes the participating task in its taskgroup 9837 // set. The construct associated with the innermost region that meets this 9838 // condition must specify the same reduction-identifier as the in_reduction 9839 // clause. 9840 if (ClauseKind == OMPC_in_reduction) { 9841 SourceRange ParentSR; 9842 BinaryOperatorKind ParentBOK; 9843 const Expr *ParentReductionOp; 9844 Expr *ParentBOKTD, *ParentReductionOpTD; 9845 DSAStackTy::DSAVarData ParentBOKDSA = 9846 Stack->getTopMostTaskgroupReductionData(D, ParentSR, ParentBOK, 9847 ParentBOKTD); 9848 DSAStackTy::DSAVarData ParentReductionOpDSA = 9849 Stack->getTopMostTaskgroupReductionData( 9850 D, ParentSR, ParentReductionOp, ParentReductionOpTD); 9851 bool IsParentBOK = ParentBOKDSA.DKind != OMPD_unknown; 9852 bool IsParentReductionOp = ParentReductionOpDSA.DKind != OMPD_unknown; 9853 if (!IsParentBOK && !IsParentReductionOp) { 9854 S.Diag(ELoc, diag::err_omp_in_reduction_not_task_reduction); 9855 continue; 9856 } 9857 if ((DeclareReductionRef.isUnset() && IsParentReductionOp) || 9858 (DeclareReductionRef.isUsable() && IsParentBOK) || BOK != ParentBOK || 9859 IsParentReductionOp) { 9860 bool EmitError = true; 9861 if (IsParentReductionOp && DeclareReductionRef.isUsable()) { 9862 llvm::FoldingSetNodeID RedId, ParentRedId; 9863 ParentReductionOp->Profile(ParentRedId, Context, /*Canonical=*/true); 9864 DeclareReductionRef.get()->Profile(RedId, Context, 9865 /*Canonical=*/true); 9866 EmitError = RedId != ParentRedId; 9867 } 9868 if (EmitError) { 9869 S.Diag(ReductionId.getLocStart(), 9870 diag::err_omp_reduction_identifier_mismatch) 9871 << ReductionIdRange << RefExpr->getSourceRange(); 9872 S.Diag(ParentSR.getBegin(), 9873 diag::note_omp_previous_reduction_identifier) 9874 << ParentSR 9875 << (IsParentBOK ? ParentBOKDSA.RefExpr 9876 : ParentReductionOpDSA.RefExpr) 9877 ->getSourceRange(); 9878 continue; 9879 } 9880 } 9881 TaskgroupDescriptor = IsParentBOK ? ParentBOKTD : ParentReductionOpTD; 9882 assert(TaskgroupDescriptor && "Taskgroup descriptor must be defined."); 9883 } 9884 9885 DeclRefExpr *Ref = nullptr; 9886 Expr *VarsExpr = RefExpr->IgnoreParens(); 9887 if (!VD && !S.CurContext->isDependentContext()) { 9888 if (ASE || OASE) { 9889 TransformExprToCaptures RebuildToCapture(S, D); 9890 VarsExpr = 9891 RebuildToCapture.TransformExpr(RefExpr->IgnoreParens()).get(); 9892 Ref = RebuildToCapture.getCapturedExpr(); 9893 } else { 9894 VarsExpr = Ref = buildCapture(S, D, SimpleRefExpr, /*WithInit=*/false); 9895 } 9896 if (!S.IsOpenMPCapturedDecl(D)) { 9897 RD.ExprCaptures.emplace_back(Ref->getDecl()); 9898 if (Ref->getDecl()->hasAttr<OMPCaptureNoInitAttr>()) { 9899 ExprResult RefRes = S.DefaultLvalueConversion(Ref); 9900 if (!RefRes.isUsable()) 9901 continue; 9902 ExprResult PostUpdateRes = 9903 S.BuildBinOp(Stack->getCurScope(), ELoc, BO_Assign, SimpleRefExpr, 9904 RefRes.get()); 9905 if (!PostUpdateRes.isUsable()) 9906 continue; 9907 if (isOpenMPTaskingDirective(Stack->getCurrentDirective()) || 9908 Stack->getCurrentDirective() == OMPD_taskgroup) { 9909 S.Diag(RefExpr->getExprLoc(), 9910 diag::err_omp_reduction_non_addressable_expression) 9911 << RefExpr->getSourceRange(); 9912 continue; 9913 } 9914 RD.ExprPostUpdates.emplace_back( 9915 S.IgnoredValueConversions(PostUpdateRes.get()).get()); 9916 } 9917 } 9918 } 9919 // All reduction items are still marked as reduction (to do not increase 9920 // code base size). 9921 Stack->addDSA(D, RefExpr->IgnoreParens(), OMPC_reduction, Ref); 9922 if (CurrDir == OMPD_taskgroup) { 9923 if (DeclareReductionRef.isUsable()) 9924 Stack->addTaskgroupReductionData(D, ReductionIdRange, 9925 DeclareReductionRef.get()); 9926 else 9927 Stack->addTaskgroupReductionData(D, ReductionIdRange, BOK); 9928 } 9929 RD.push(VarsExpr, PrivateDRE, LHSDRE, RHSDRE, ReductionOp.get(), 9930 TaskgroupDescriptor); 9931 } 9932 return RD.Vars.empty(); 9933 } 9934 9935 OMPClause *Sema::ActOnOpenMPReductionClause( 9936 ArrayRef<Expr *> VarList, SourceLocation StartLoc, SourceLocation LParenLoc, 9937 SourceLocation ColonLoc, SourceLocation EndLoc, 9938 CXXScopeSpec &ReductionIdScopeSpec, const DeclarationNameInfo &ReductionId, 9939 ArrayRef<Expr *> UnresolvedReductions) { 9940 ReductionData RD(VarList.size()); 9941 9942 if (ActOnOMPReductionKindClause(*this, DSAStack, OMPC_reduction, VarList, 9943 StartLoc, LParenLoc, ColonLoc, EndLoc, 9944 ReductionIdScopeSpec, ReductionId, 9945 UnresolvedReductions, RD)) 9946 return nullptr; 9947 9948 return OMPReductionClause::Create( 9949 Context, StartLoc, LParenLoc, ColonLoc, EndLoc, RD.Vars, 9950 ReductionIdScopeSpec.getWithLocInContext(Context), ReductionId, 9951 RD.Privates, RD.LHSs, RD.RHSs, RD.ReductionOps, 9952 buildPreInits(Context, RD.ExprCaptures), 9953 buildPostUpdate(*this, RD.ExprPostUpdates)); 9954 } 9955 9956 OMPClause *Sema::ActOnOpenMPTaskReductionClause( 9957 ArrayRef<Expr *> VarList, SourceLocation StartLoc, SourceLocation LParenLoc, 9958 SourceLocation ColonLoc, SourceLocation EndLoc, 9959 CXXScopeSpec &ReductionIdScopeSpec, const DeclarationNameInfo &ReductionId, 9960 ArrayRef<Expr *> UnresolvedReductions) { 9961 ReductionData RD(VarList.size()); 9962 9963 if (ActOnOMPReductionKindClause(*this, DSAStack, OMPC_task_reduction, 9964 VarList, StartLoc, LParenLoc, ColonLoc, 9965 EndLoc, ReductionIdScopeSpec, ReductionId, 9966 UnresolvedReductions, RD)) 9967 return nullptr; 9968 9969 return OMPTaskReductionClause::Create( 9970 Context, StartLoc, LParenLoc, ColonLoc, EndLoc, RD.Vars, 9971 ReductionIdScopeSpec.getWithLocInContext(Context), ReductionId, 9972 RD.Privates, RD.LHSs, RD.RHSs, RD.ReductionOps, 9973 buildPreInits(Context, RD.ExprCaptures), 9974 buildPostUpdate(*this, RD.ExprPostUpdates)); 9975 } 9976 9977 OMPClause *Sema::ActOnOpenMPInReductionClause( 9978 ArrayRef<Expr *> VarList, SourceLocation StartLoc, SourceLocation LParenLoc, 9979 SourceLocation ColonLoc, SourceLocation EndLoc, 9980 CXXScopeSpec &ReductionIdScopeSpec, const DeclarationNameInfo &ReductionId, 9981 ArrayRef<Expr *> UnresolvedReductions) { 9982 ReductionData RD(VarList.size()); 9983 9984 if (ActOnOMPReductionKindClause(*this, DSAStack, OMPC_in_reduction, VarList, 9985 StartLoc, LParenLoc, ColonLoc, EndLoc, 9986 ReductionIdScopeSpec, ReductionId, 9987 UnresolvedReductions, RD)) 9988 return nullptr; 9989 9990 return OMPInReductionClause::Create( 9991 Context, StartLoc, LParenLoc, ColonLoc, EndLoc, RD.Vars, 9992 ReductionIdScopeSpec.getWithLocInContext(Context), ReductionId, 9993 RD.Privates, RD.LHSs, RD.RHSs, RD.ReductionOps, RD.TaskgroupDescriptors, 9994 buildPreInits(Context, RD.ExprCaptures), 9995 buildPostUpdate(*this, RD.ExprPostUpdates)); 9996 } 9997 9998 bool Sema::CheckOpenMPLinearModifier(OpenMPLinearClauseKind LinKind, 9999 SourceLocation LinLoc) { 10000 if ((!LangOpts.CPlusPlus && LinKind != OMPC_LINEAR_val) || 10001 LinKind == OMPC_LINEAR_unknown) { 10002 Diag(LinLoc, diag::err_omp_wrong_linear_modifier) << LangOpts.CPlusPlus; 10003 return true; 10004 } 10005 return false; 10006 } 10007 10008 bool Sema::CheckOpenMPLinearDecl(ValueDecl *D, SourceLocation ELoc, 10009 OpenMPLinearClauseKind LinKind, 10010 QualType Type) { 10011 auto *VD = dyn_cast_or_null<VarDecl>(D); 10012 // A variable must not have an incomplete type or a reference type. 10013 if (RequireCompleteType(ELoc, Type, diag::err_omp_linear_incomplete_type)) 10014 return true; 10015 if ((LinKind == OMPC_LINEAR_uval || LinKind == OMPC_LINEAR_ref) && 10016 !Type->isReferenceType()) { 10017 Diag(ELoc, diag::err_omp_wrong_linear_modifier_non_reference) 10018 << Type << getOpenMPSimpleClauseTypeName(OMPC_linear, LinKind); 10019 return true; 10020 } 10021 Type = Type.getNonReferenceType(); 10022 10023 // A list item must not be const-qualified. 10024 if (Type.isConstant(Context)) { 10025 Diag(ELoc, diag::err_omp_const_variable) 10026 << getOpenMPClauseName(OMPC_linear); 10027 if (D) { 10028 bool IsDecl = 10029 !VD || 10030 VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly; 10031 Diag(D->getLocation(), 10032 IsDecl ? diag::note_previous_decl : diag::note_defined_here) 10033 << D; 10034 } 10035 return true; 10036 } 10037 10038 // A list item must be of integral or pointer type. 10039 Type = Type.getUnqualifiedType().getCanonicalType(); 10040 const auto *Ty = Type.getTypePtrOrNull(); 10041 if (!Ty || (!Ty->isDependentType() && !Ty->isIntegralType(Context) && 10042 !Ty->isPointerType())) { 10043 Diag(ELoc, diag::err_omp_linear_expected_int_or_ptr) << Type; 10044 if (D) { 10045 bool IsDecl = 10046 !VD || 10047 VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly; 10048 Diag(D->getLocation(), 10049 IsDecl ? diag::note_previous_decl : diag::note_defined_here) 10050 << D; 10051 } 10052 return true; 10053 } 10054 return false; 10055 } 10056 10057 OMPClause *Sema::ActOnOpenMPLinearClause( 10058 ArrayRef<Expr *> VarList, Expr *Step, SourceLocation StartLoc, 10059 SourceLocation LParenLoc, OpenMPLinearClauseKind LinKind, 10060 SourceLocation LinLoc, SourceLocation ColonLoc, SourceLocation EndLoc) { 10061 SmallVector<Expr *, 8> Vars; 10062 SmallVector<Expr *, 8> Privates; 10063 SmallVector<Expr *, 8> Inits; 10064 SmallVector<Decl *, 4> ExprCaptures; 10065 SmallVector<Expr *, 4> ExprPostUpdates; 10066 if (CheckOpenMPLinearModifier(LinKind, LinLoc)) 10067 LinKind = OMPC_LINEAR_val; 10068 for (auto &RefExpr : VarList) { 10069 assert(RefExpr && "NULL expr in OpenMP linear clause."); 10070 SourceLocation ELoc; 10071 SourceRange ERange; 10072 Expr *SimpleRefExpr = RefExpr; 10073 auto Res = getPrivateItem(*this, SimpleRefExpr, ELoc, ERange, 10074 /*AllowArraySection=*/false); 10075 if (Res.second) { 10076 // It will be analyzed later. 10077 Vars.push_back(RefExpr); 10078 Privates.push_back(nullptr); 10079 Inits.push_back(nullptr); 10080 } 10081 ValueDecl *D = Res.first; 10082 if (!D) 10083 continue; 10084 10085 QualType Type = D->getType(); 10086 auto *VD = dyn_cast<VarDecl>(D); 10087 10088 // OpenMP [2.14.3.7, linear clause] 10089 // A list-item cannot appear in more than one linear clause. 10090 // A list-item that appears in a linear clause cannot appear in any 10091 // other data-sharing attribute clause. 10092 DSAStackTy::DSAVarData DVar = DSAStack->getTopDSA(D, false); 10093 if (DVar.RefExpr) { 10094 Diag(ELoc, diag::err_omp_wrong_dsa) << getOpenMPClauseName(DVar.CKind) 10095 << getOpenMPClauseName(OMPC_linear); 10096 ReportOriginalDSA(*this, DSAStack, D, DVar); 10097 continue; 10098 } 10099 10100 if (CheckOpenMPLinearDecl(D, ELoc, LinKind, Type)) 10101 continue; 10102 Type = Type.getNonReferenceType().getUnqualifiedType().getCanonicalType(); 10103 10104 // Build private copy of original var. 10105 auto *Private = buildVarDecl(*this, ELoc, Type, D->getName(), 10106 D->hasAttrs() ? &D->getAttrs() : nullptr); 10107 auto *PrivateRef = buildDeclRefExpr(*this, Private, Type, ELoc); 10108 // Build var to save initial value. 10109 VarDecl *Init = buildVarDecl(*this, ELoc, Type, ".linear.start"); 10110 Expr *InitExpr; 10111 DeclRefExpr *Ref = nullptr; 10112 if (!VD && !CurContext->isDependentContext()) { 10113 Ref = buildCapture(*this, D, SimpleRefExpr, /*WithInit=*/false); 10114 if (!IsOpenMPCapturedDecl(D)) { 10115 ExprCaptures.push_back(Ref->getDecl()); 10116 if (Ref->getDecl()->hasAttr<OMPCaptureNoInitAttr>()) { 10117 ExprResult RefRes = DefaultLvalueConversion(Ref); 10118 if (!RefRes.isUsable()) 10119 continue; 10120 ExprResult PostUpdateRes = 10121 BuildBinOp(DSAStack->getCurScope(), ELoc, BO_Assign, 10122 SimpleRefExpr, RefRes.get()); 10123 if (!PostUpdateRes.isUsable()) 10124 continue; 10125 ExprPostUpdates.push_back( 10126 IgnoredValueConversions(PostUpdateRes.get()).get()); 10127 } 10128 } 10129 } 10130 if (LinKind == OMPC_LINEAR_uval) 10131 InitExpr = VD ? VD->getInit() : SimpleRefExpr; 10132 else 10133 InitExpr = VD ? SimpleRefExpr : Ref; 10134 AddInitializerToDecl(Init, DefaultLvalueConversion(InitExpr).get(), 10135 /*DirectInit=*/false); 10136 auto InitRef = buildDeclRefExpr(*this, Init, Type, ELoc); 10137 10138 DSAStack->addDSA(D, RefExpr->IgnoreParens(), OMPC_linear, Ref); 10139 Vars.push_back((VD || CurContext->isDependentContext()) 10140 ? RefExpr->IgnoreParens() 10141 : Ref); 10142 Privates.push_back(PrivateRef); 10143 Inits.push_back(InitRef); 10144 } 10145 10146 if (Vars.empty()) 10147 return nullptr; 10148 10149 Expr *StepExpr = Step; 10150 Expr *CalcStepExpr = nullptr; 10151 if (Step && !Step->isValueDependent() && !Step->isTypeDependent() && 10152 !Step->isInstantiationDependent() && 10153 !Step->containsUnexpandedParameterPack()) { 10154 SourceLocation StepLoc = Step->getLocStart(); 10155 ExprResult Val = PerformOpenMPImplicitIntegerConversion(StepLoc, Step); 10156 if (Val.isInvalid()) 10157 return nullptr; 10158 StepExpr = Val.get(); 10159 10160 // Build var to save the step value. 10161 VarDecl *SaveVar = 10162 buildVarDecl(*this, StepLoc, StepExpr->getType(), ".linear.step"); 10163 ExprResult SaveRef = 10164 buildDeclRefExpr(*this, SaveVar, StepExpr->getType(), StepLoc); 10165 ExprResult CalcStep = 10166 BuildBinOp(CurScope, StepLoc, BO_Assign, SaveRef.get(), StepExpr); 10167 CalcStep = ActOnFinishFullExpr(CalcStep.get()); 10168 10169 // Warn about zero linear step (it would be probably better specified as 10170 // making corresponding variables 'const'). 10171 llvm::APSInt Result; 10172 bool IsConstant = StepExpr->isIntegerConstantExpr(Result, Context); 10173 if (IsConstant && !Result.isNegative() && !Result.isStrictlyPositive()) 10174 Diag(StepLoc, diag::warn_omp_linear_step_zero) << Vars[0] 10175 << (Vars.size() > 1); 10176 if (!IsConstant && CalcStep.isUsable()) { 10177 // Calculate the step beforehand instead of doing this on each iteration. 10178 // (This is not used if the number of iterations may be kfold-ed). 10179 CalcStepExpr = CalcStep.get(); 10180 } 10181 } 10182 10183 return OMPLinearClause::Create(Context, StartLoc, LParenLoc, LinKind, LinLoc, 10184 ColonLoc, EndLoc, Vars, Privates, Inits, 10185 StepExpr, CalcStepExpr, 10186 buildPreInits(Context, ExprCaptures), 10187 buildPostUpdate(*this, ExprPostUpdates)); 10188 } 10189 10190 static bool FinishOpenMPLinearClause(OMPLinearClause &Clause, DeclRefExpr *IV, 10191 Expr *NumIterations, Sema &SemaRef, 10192 Scope *S, DSAStackTy *Stack) { 10193 // Walk the vars and build update/final expressions for the CodeGen. 10194 SmallVector<Expr *, 8> Updates; 10195 SmallVector<Expr *, 8> Finals; 10196 Expr *Step = Clause.getStep(); 10197 Expr *CalcStep = Clause.getCalcStep(); 10198 // OpenMP [2.14.3.7, linear clause] 10199 // If linear-step is not specified it is assumed to be 1. 10200 if (Step == nullptr) 10201 Step = SemaRef.ActOnIntegerConstant(SourceLocation(), 1).get(); 10202 else if (CalcStep) { 10203 Step = cast<BinaryOperator>(CalcStep)->getLHS(); 10204 } 10205 bool HasErrors = false; 10206 auto CurInit = Clause.inits().begin(); 10207 auto CurPrivate = Clause.privates().begin(); 10208 auto LinKind = Clause.getModifier(); 10209 for (auto &RefExpr : Clause.varlists()) { 10210 SourceLocation ELoc; 10211 SourceRange ERange; 10212 Expr *SimpleRefExpr = RefExpr; 10213 auto Res = getPrivateItem(SemaRef, SimpleRefExpr, ELoc, ERange, 10214 /*AllowArraySection=*/false); 10215 ValueDecl *D = Res.first; 10216 if (Res.second || !D) { 10217 Updates.push_back(nullptr); 10218 Finals.push_back(nullptr); 10219 HasErrors = true; 10220 continue; 10221 } 10222 auto &&Info = Stack->isLoopControlVariable(D); 10223 Expr *InitExpr = *CurInit; 10224 10225 // Build privatized reference to the current linear var. 10226 auto *DE = cast<DeclRefExpr>(SimpleRefExpr); 10227 Expr *CapturedRef; 10228 if (LinKind == OMPC_LINEAR_uval) 10229 CapturedRef = cast<VarDecl>(DE->getDecl())->getInit(); 10230 else 10231 CapturedRef = 10232 buildDeclRefExpr(SemaRef, cast<VarDecl>(DE->getDecl()), 10233 DE->getType().getUnqualifiedType(), DE->getExprLoc(), 10234 /*RefersToCapture=*/true); 10235 10236 // Build update: Var = InitExpr + IV * Step 10237 ExprResult Update; 10238 if (!Info.first) { 10239 Update = 10240 BuildCounterUpdate(SemaRef, S, RefExpr->getExprLoc(), *CurPrivate, 10241 InitExpr, IV, Step, /* Subtract */ false); 10242 } else 10243 Update = *CurPrivate; 10244 Update = SemaRef.ActOnFinishFullExpr(Update.get(), DE->getLocStart(), 10245 /*DiscardedValue=*/true); 10246 10247 // Build final: Var = InitExpr + NumIterations * Step 10248 ExprResult Final; 10249 if (!Info.first) { 10250 Final = BuildCounterUpdate(SemaRef, S, RefExpr->getExprLoc(), CapturedRef, 10251 InitExpr, NumIterations, Step, 10252 /* Subtract */ false); 10253 } else 10254 Final = *CurPrivate; 10255 Final = SemaRef.ActOnFinishFullExpr(Final.get(), DE->getLocStart(), 10256 /*DiscardedValue=*/true); 10257 10258 if (!Update.isUsable() || !Final.isUsable()) { 10259 Updates.push_back(nullptr); 10260 Finals.push_back(nullptr); 10261 HasErrors = true; 10262 } else { 10263 Updates.push_back(Update.get()); 10264 Finals.push_back(Final.get()); 10265 } 10266 ++CurInit; 10267 ++CurPrivate; 10268 } 10269 Clause.setUpdates(Updates); 10270 Clause.setFinals(Finals); 10271 return HasErrors; 10272 } 10273 10274 OMPClause *Sema::ActOnOpenMPAlignedClause( 10275 ArrayRef<Expr *> VarList, Expr *Alignment, SourceLocation StartLoc, 10276 SourceLocation LParenLoc, SourceLocation ColonLoc, SourceLocation EndLoc) { 10277 10278 SmallVector<Expr *, 8> Vars; 10279 for (auto &RefExpr : VarList) { 10280 assert(RefExpr && "NULL expr in OpenMP linear clause."); 10281 SourceLocation ELoc; 10282 SourceRange ERange; 10283 Expr *SimpleRefExpr = RefExpr; 10284 auto Res = getPrivateItem(*this, SimpleRefExpr, ELoc, ERange, 10285 /*AllowArraySection=*/false); 10286 if (Res.second) { 10287 // It will be analyzed later. 10288 Vars.push_back(RefExpr); 10289 } 10290 ValueDecl *D = Res.first; 10291 if (!D) 10292 continue; 10293 10294 QualType QType = D->getType(); 10295 auto *VD = dyn_cast<VarDecl>(D); 10296 10297 // OpenMP [2.8.1, simd construct, Restrictions] 10298 // The type of list items appearing in the aligned clause must be 10299 // array, pointer, reference to array, or reference to pointer. 10300 QType = QType.getNonReferenceType().getUnqualifiedType().getCanonicalType(); 10301 const Type *Ty = QType.getTypePtrOrNull(); 10302 if (!Ty || (!Ty->isArrayType() && !Ty->isPointerType())) { 10303 Diag(ELoc, diag::err_omp_aligned_expected_array_or_ptr) 10304 << QType << getLangOpts().CPlusPlus << ERange; 10305 bool IsDecl = 10306 !VD || 10307 VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly; 10308 Diag(D->getLocation(), 10309 IsDecl ? diag::note_previous_decl : diag::note_defined_here) 10310 << D; 10311 continue; 10312 } 10313 10314 // OpenMP [2.8.1, simd construct, Restrictions] 10315 // A list-item cannot appear in more than one aligned clause. 10316 if (Expr *PrevRef = DSAStack->addUniqueAligned(D, SimpleRefExpr)) { 10317 Diag(ELoc, diag::err_omp_aligned_twice) << 0 << ERange; 10318 Diag(PrevRef->getExprLoc(), diag::note_omp_explicit_dsa) 10319 << getOpenMPClauseName(OMPC_aligned); 10320 continue; 10321 } 10322 10323 DeclRefExpr *Ref = nullptr; 10324 if (!VD && IsOpenMPCapturedDecl(D)) 10325 Ref = buildCapture(*this, D, SimpleRefExpr, /*WithInit=*/true); 10326 Vars.push_back(DefaultFunctionArrayConversion( 10327 (VD || !Ref) ? RefExpr->IgnoreParens() : Ref) 10328 .get()); 10329 } 10330 10331 // OpenMP [2.8.1, simd construct, Description] 10332 // The parameter of the aligned clause, alignment, must be a constant 10333 // positive integer expression. 10334 // If no optional parameter is specified, implementation-defined default 10335 // alignments for SIMD instructions on the target platforms are assumed. 10336 if (Alignment != nullptr) { 10337 ExprResult AlignResult = 10338 VerifyPositiveIntegerConstantInClause(Alignment, OMPC_aligned); 10339 if (AlignResult.isInvalid()) 10340 return nullptr; 10341 Alignment = AlignResult.get(); 10342 } 10343 if (Vars.empty()) 10344 return nullptr; 10345 10346 return OMPAlignedClause::Create(Context, StartLoc, LParenLoc, ColonLoc, 10347 EndLoc, Vars, Alignment); 10348 } 10349 10350 OMPClause *Sema::ActOnOpenMPCopyinClause(ArrayRef<Expr *> VarList, 10351 SourceLocation StartLoc, 10352 SourceLocation LParenLoc, 10353 SourceLocation EndLoc) { 10354 SmallVector<Expr *, 8> Vars; 10355 SmallVector<Expr *, 8> SrcExprs; 10356 SmallVector<Expr *, 8> DstExprs; 10357 SmallVector<Expr *, 8> AssignmentOps; 10358 for (auto &RefExpr : VarList) { 10359 assert(RefExpr && "NULL expr in OpenMP copyin clause."); 10360 if (isa<DependentScopeDeclRefExpr>(RefExpr)) { 10361 // It will be analyzed later. 10362 Vars.push_back(RefExpr); 10363 SrcExprs.push_back(nullptr); 10364 DstExprs.push_back(nullptr); 10365 AssignmentOps.push_back(nullptr); 10366 continue; 10367 } 10368 10369 SourceLocation ELoc = RefExpr->getExprLoc(); 10370 // OpenMP [2.1, C/C++] 10371 // A list item is a variable name. 10372 // OpenMP [2.14.4.1, Restrictions, p.1] 10373 // A list item that appears in a copyin clause must be threadprivate. 10374 DeclRefExpr *DE = dyn_cast<DeclRefExpr>(RefExpr); 10375 if (!DE || !isa<VarDecl>(DE->getDecl())) { 10376 Diag(ELoc, diag::err_omp_expected_var_name_member_expr) 10377 << 0 << RefExpr->getSourceRange(); 10378 continue; 10379 } 10380 10381 Decl *D = DE->getDecl(); 10382 VarDecl *VD = cast<VarDecl>(D); 10383 10384 QualType Type = VD->getType(); 10385 if (Type->isDependentType() || Type->isInstantiationDependentType()) { 10386 // It will be analyzed later. 10387 Vars.push_back(DE); 10388 SrcExprs.push_back(nullptr); 10389 DstExprs.push_back(nullptr); 10390 AssignmentOps.push_back(nullptr); 10391 continue; 10392 } 10393 10394 // OpenMP [2.14.4.1, Restrictions, C/C++, p.1] 10395 // A list item that appears in a copyin clause must be threadprivate. 10396 if (!DSAStack->isThreadPrivate(VD)) { 10397 Diag(ELoc, diag::err_omp_required_access) 10398 << getOpenMPClauseName(OMPC_copyin) 10399 << getOpenMPDirectiveName(OMPD_threadprivate); 10400 continue; 10401 } 10402 10403 // OpenMP [2.14.4.1, Restrictions, C/C++, p.2] 10404 // A variable of class type (or array thereof) that appears in a 10405 // copyin clause requires an accessible, unambiguous copy assignment 10406 // operator for the class type. 10407 auto ElemType = Context.getBaseElementType(Type).getNonReferenceType(); 10408 auto *SrcVD = 10409 buildVarDecl(*this, DE->getLocStart(), ElemType.getUnqualifiedType(), 10410 ".copyin.src", VD->hasAttrs() ? &VD->getAttrs() : nullptr); 10411 auto *PseudoSrcExpr = buildDeclRefExpr( 10412 *this, SrcVD, ElemType.getUnqualifiedType(), DE->getExprLoc()); 10413 auto *DstVD = 10414 buildVarDecl(*this, DE->getLocStart(), ElemType, ".copyin.dst", 10415 VD->hasAttrs() ? &VD->getAttrs() : nullptr); 10416 auto *PseudoDstExpr = 10417 buildDeclRefExpr(*this, DstVD, ElemType, DE->getExprLoc()); 10418 // For arrays generate assignment operation for single element and replace 10419 // it by the original array element in CodeGen. 10420 auto AssignmentOp = BuildBinOp(/*S=*/nullptr, DE->getExprLoc(), BO_Assign, 10421 PseudoDstExpr, PseudoSrcExpr); 10422 if (AssignmentOp.isInvalid()) 10423 continue; 10424 AssignmentOp = ActOnFinishFullExpr(AssignmentOp.get(), DE->getExprLoc(), 10425 /*DiscardedValue=*/true); 10426 if (AssignmentOp.isInvalid()) 10427 continue; 10428 10429 DSAStack->addDSA(VD, DE, OMPC_copyin); 10430 Vars.push_back(DE); 10431 SrcExprs.push_back(PseudoSrcExpr); 10432 DstExprs.push_back(PseudoDstExpr); 10433 AssignmentOps.push_back(AssignmentOp.get()); 10434 } 10435 10436 if (Vars.empty()) 10437 return nullptr; 10438 10439 return OMPCopyinClause::Create(Context, StartLoc, LParenLoc, EndLoc, Vars, 10440 SrcExprs, DstExprs, AssignmentOps); 10441 } 10442 10443 OMPClause *Sema::ActOnOpenMPCopyprivateClause(ArrayRef<Expr *> VarList, 10444 SourceLocation StartLoc, 10445 SourceLocation LParenLoc, 10446 SourceLocation EndLoc) { 10447 SmallVector<Expr *, 8> Vars; 10448 SmallVector<Expr *, 8> SrcExprs; 10449 SmallVector<Expr *, 8> DstExprs; 10450 SmallVector<Expr *, 8> AssignmentOps; 10451 for (auto &RefExpr : VarList) { 10452 assert(RefExpr && "NULL expr in OpenMP linear clause."); 10453 SourceLocation ELoc; 10454 SourceRange ERange; 10455 Expr *SimpleRefExpr = RefExpr; 10456 auto Res = getPrivateItem(*this, SimpleRefExpr, ELoc, ERange, 10457 /*AllowArraySection=*/false); 10458 if (Res.second) { 10459 // It will be analyzed later. 10460 Vars.push_back(RefExpr); 10461 SrcExprs.push_back(nullptr); 10462 DstExprs.push_back(nullptr); 10463 AssignmentOps.push_back(nullptr); 10464 } 10465 ValueDecl *D = Res.first; 10466 if (!D) 10467 continue; 10468 10469 QualType Type = D->getType(); 10470 auto *VD = dyn_cast<VarDecl>(D); 10471 10472 // OpenMP [2.14.4.2, Restrictions, p.2] 10473 // A list item that appears in a copyprivate clause may not appear in a 10474 // private or firstprivate clause on the single construct. 10475 if (!VD || !DSAStack->isThreadPrivate(VD)) { 10476 auto DVar = DSAStack->getTopDSA(D, false); 10477 if (DVar.CKind != OMPC_unknown && DVar.CKind != OMPC_copyprivate && 10478 DVar.RefExpr) { 10479 Diag(ELoc, diag::err_omp_wrong_dsa) 10480 << getOpenMPClauseName(DVar.CKind) 10481 << getOpenMPClauseName(OMPC_copyprivate); 10482 ReportOriginalDSA(*this, DSAStack, D, DVar); 10483 continue; 10484 } 10485 10486 // OpenMP [2.11.4.2, Restrictions, p.1] 10487 // All list items that appear in a copyprivate clause must be either 10488 // threadprivate or private in the enclosing context. 10489 if (DVar.CKind == OMPC_unknown) { 10490 DVar = DSAStack->getImplicitDSA(D, false); 10491 if (DVar.CKind == OMPC_shared) { 10492 Diag(ELoc, diag::err_omp_required_access) 10493 << getOpenMPClauseName(OMPC_copyprivate) 10494 << "threadprivate or private in the enclosing context"; 10495 ReportOriginalDSA(*this, DSAStack, D, DVar); 10496 continue; 10497 } 10498 } 10499 } 10500 10501 // Variably modified types are not supported. 10502 if (!Type->isAnyPointerType() && Type->isVariablyModifiedType()) { 10503 Diag(ELoc, diag::err_omp_variably_modified_type_not_supported) 10504 << getOpenMPClauseName(OMPC_copyprivate) << Type 10505 << getOpenMPDirectiveName(DSAStack->getCurrentDirective()); 10506 bool IsDecl = 10507 !VD || 10508 VD->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly; 10509 Diag(D->getLocation(), 10510 IsDecl ? diag::note_previous_decl : diag::note_defined_here) 10511 << D; 10512 continue; 10513 } 10514 10515 // OpenMP [2.14.4.1, Restrictions, C/C++, p.2] 10516 // A variable of class type (or array thereof) that appears in a 10517 // copyin clause requires an accessible, unambiguous copy assignment 10518 // operator for the class type. 10519 Type = Context.getBaseElementType(Type.getNonReferenceType()) 10520 .getUnqualifiedType(); 10521 auto *SrcVD = 10522 buildVarDecl(*this, RefExpr->getLocStart(), Type, ".copyprivate.src", 10523 D->hasAttrs() ? &D->getAttrs() : nullptr); 10524 auto *PseudoSrcExpr = buildDeclRefExpr(*this, SrcVD, Type, ELoc); 10525 auto *DstVD = 10526 buildVarDecl(*this, RefExpr->getLocStart(), Type, ".copyprivate.dst", 10527 D->hasAttrs() ? &D->getAttrs() : nullptr); 10528 auto *PseudoDstExpr = buildDeclRefExpr(*this, DstVD, Type, ELoc); 10529 auto AssignmentOp = BuildBinOp(DSAStack->getCurScope(), ELoc, BO_Assign, 10530 PseudoDstExpr, PseudoSrcExpr); 10531 if (AssignmentOp.isInvalid()) 10532 continue; 10533 AssignmentOp = ActOnFinishFullExpr(AssignmentOp.get(), ELoc, 10534 /*DiscardedValue=*/true); 10535 if (AssignmentOp.isInvalid()) 10536 continue; 10537 10538 // No need to mark vars as copyprivate, they are already threadprivate or 10539 // implicitly private. 10540 assert(VD || IsOpenMPCapturedDecl(D)); 10541 Vars.push_back( 10542 VD ? RefExpr->IgnoreParens() 10543 : buildCapture(*this, D, SimpleRefExpr, /*WithInit=*/false)); 10544 SrcExprs.push_back(PseudoSrcExpr); 10545 DstExprs.push_back(PseudoDstExpr); 10546 AssignmentOps.push_back(AssignmentOp.get()); 10547 } 10548 10549 if (Vars.empty()) 10550 return nullptr; 10551 10552 return OMPCopyprivateClause::Create(Context, StartLoc, LParenLoc, EndLoc, 10553 Vars, SrcExprs, DstExprs, AssignmentOps); 10554 } 10555 10556 OMPClause *Sema::ActOnOpenMPFlushClause(ArrayRef<Expr *> VarList, 10557 SourceLocation StartLoc, 10558 SourceLocation LParenLoc, 10559 SourceLocation EndLoc) { 10560 if (VarList.empty()) 10561 return nullptr; 10562 10563 return OMPFlushClause::Create(Context, StartLoc, LParenLoc, EndLoc, VarList); 10564 } 10565 10566 OMPClause * 10567 Sema::ActOnOpenMPDependClause(OpenMPDependClauseKind DepKind, 10568 SourceLocation DepLoc, SourceLocation ColonLoc, 10569 ArrayRef<Expr *> VarList, SourceLocation StartLoc, 10570 SourceLocation LParenLoc, SourceLocation EndLoc) { 10571 if (DSAStack->getCurrentDirective() == OMPD_ordered && 10572 DepKind != OMPC_DEPEND_source && DepKind != OMPC_DEPEND_sink) { 10573 Diag(DepLoc, diag::err_omp_unexpected_clause_value) 10574 << "'source' or 'sink'" << getOpenMPClauseName(OMPC_depend); 10575 return nullptr; 10576 } 10577 if (DSAStack->getCurrentDirective() != OMPD_ordered && 10578 (DepKind == OMPC_DEPEND_unknown || DepKind == OMPC_DEPEND_source || 10579 DepKind == OMPC_DEPEND_sink)) { 10580 unsigned Except[] = {OMPC_DEPEND_source, OMPC_DEPEND_sink}; 10581 Diag(DepLoc, diag::err_omp_unexpected_clause_value) 10582 << getListOfPossibleValues(OMPC_depend, /*First=*/0, 10583 /*Last=*/OMPC_DEPEND_unknown, Except) 10584 << getOpenMPClauseName(OMPC_depend); 10585 return nullptr; 10586 } 10587 SmallVector<Expr *, 8> Vars; 10588 DSAStackTy::OperatorOffsetTy OpsOffs; 10589 llvm::APSInt DepCounter(/*BitWidth=*/32); 10590 llvm::APSInt TotalDepCount(/*BitWidth=*/32); 10591 if (DepKind == OMPC_DEPEND_sink) { 10592 if (auto *OrderedCountExpr = DSAStack->getParentOrderedRegionParam()) { 10593 TotalDepCount = OrderedCountExpr->EvaluateKnownConstInt(Context); 10594 TotalDepCount.setIsUnsigned(/*Val=*/true); 10595 } 10596 } 10597 if ((DepKind != OMPC_DEPEND_sink && DepKind != OMPC_DEPEND_source) || 10598 DSAStack->getParentOrderedRegionParam()) { 10599 for (auto &RefExpr : VarList) { 10600 assert(RefExpr && "NULL expr in OpenMP shared clause."); 10601 if (isa<DependentScopeDeclRefExpr>(RefExpr)) { 10602 // It will be analyzed later. 10603 Vars.push_back(RefExpr); 10604 continue; 10605 } 10606 10607 SourceLocation ELoc = RefExpr->getExprLoc(); 10608 auto *SimpleExpr = RefExpr->IgnoreParenCasts(); 10609 if (DepKind == OMPC_DEPEND_sink) { 10610 if (DepCounter >= TotalDepCount) { 10611 Diag(ELoc, diag::err_omp_depend_sink_unexpected_expr); 10612 continue; 10613 } 10614 ++DepCounter; 10615 // OpenMP [2.13.9, Summary] 10616 // depend(dependence-type : vec), where dependence-type is: 10617 // 'sink' and where vec is the iteration vector, which has the form: 10618 // x1 [+- d1], x2 [+- d2 ], . . . , xn [+- dn] 10619 // where n is the value specified by the ordered clause in the loop 10620 // directive, xi denotes the loop iteration variable of the i-th nested 10621 // loop associated with the loop directive, and di is a constant 10622 // non-negative integer. 10623 if (CurContext->isDependentContext()) { 10624 // It will be analyzed later. 10625 Vars.push_back(RefExpr); 10626 continue; 10627 } 10628 SimpleExpr = SimpleExpr->IgnoreImplicit(); 10629 OverloadedOperatorKind OOK = OO_None; 10630 SourceLocation OOLoc; 10631 Expr *LHS = SimpleExpr; 10632 Expr *RHS = nullptr; 10633 if (auto *BO = dyn_cast<BinaryOperator>(SimpleExpr)) { 10634 OOK = BinaryOperator::getOverloadedOperator(BO->getOpcode()); 10635 OOLoc = BO->getOperatorLoc(); 10636 LHS = BO->getLHS()->IgnoreParenImpCasts(); 10637 RHS = BO->getRHS()->IgnoreParenImpCasts(); 10638 } else if (auto *OCE = dyn_cast<CXXOperatorCallExpr>(SimpleExpr)) { 10639 OOK = OCE->getOperator(); 10640 OOLoc = OCE->getOperatorLoc(); 10641 LHS = OCE->getArg(/*Arg=*/0)->IgnoreParenImpCasts(); 10642 RHS = OCE->getArg(/*Arg=*/1)->IgnoreParenImpCasts(); 10643 } else if (auto *MCE = dyn_cast<CXXMemberCallExpr>(SimpleExpr)) { 10644 OOK = MCE->getMethodDecl() 10645 ->getNameInfo() 10646 .getName() 10647 .getCXXOverloadedOperator(); 10648 OOLoc = MCE->getCallee()->getExprLoc(); 10649 LHS = MCE->getImplicitObjectArgument()->IgnoreParenImpCasts(); 10650 RHS = MCE->getArg(/*Arg=*/0)->IgnoreParenImpCasts(); 10651 } 10652 SourceLocation ELoc; 10653 SourceRange ERange; 10654 auto Res = getPrivateItem(*this, LHS, ELoc, ERange, 10655 /*AllowArraySection=*/false); 10656 if (Res.second) { 10657 // It will be analyzed later. 10658 Vars.push_back(RefExpr); 10659 } 10660 ValueDecl *D = Res.first; 10661 if (!D) 10662 continue; 10663 10664 if (OOK != OO_Plus && OOK != OO_Minus && (RHS || OOK != OO_None)) { 10665 Diag(OOLoc, diag::err_omp_depend_sink_expected_plus_minus); 10666 continue; 10667 } 10668 if (RHS) { 10669 ExprResult RHSRes = VerifyPositiveIntegerConstantInClause( 10670 RHS, OMPC_depend, /*StrictlyPositive=*/false); 10671 if (RHSRes.isInvalid()) 10672 continue; 10673 } 10674 if (!CurContext->isDependentContext() && 10675 DSAStack->getParentOrderedRegionParam() && 10676 DepCounter != DSAStack->isParentLoopControlVariable(D).first) { 10677 ValueDecl* VD = DSAStack->getParentLoopControlVariable( 10678 DepCounter.getZExtValue()); 10679 if (VD) { 10680 Diag(ELoc, diag::err_omp_depend_sink_expected_loop_iteration) 10681 << 1 << VD; 10682 } else { 10683 Diag(ELoc, diag::err_omp_depend_sink_expected_loop_iteration) << 0; 10684 } 10685 continue; 10686 } 10687 OpsOffs.push_back({RHS, OOK}); 10688 } else { 10689 auto *ASE = dyn_cast<ArraySubscriptExpr>(SimpleExpr); 10690 if (!RefExpr->IgnoreParenImpCasts()->isLValue() || 10691 (ASE && 10692 !ASE->getBase() 10693 ->getType() 10694 .getNonReferenceType() 10695 ->isPointerType() && 10696 !ASE->getBase()->getType().getNonReferenceType()->isArrayType())) { 10697 Diag(ELoc, diag::err_omp_expected_addressable_lvalue_or_array_item) 10698 << RefExpr->getSourceRange(); 10699 continue; 10700 } 10701 bool Suppress = getDiagnostics().getSuppressAllDiagnostics(); 10702 getDiagnostics().setSuppressAllDiagnostics(/*Val=*/true); 10703 ExprResult Res = CreateBuiltinUnaryOp(SourceLocation(), UO_AddrOf, 10704 RefExpr->IgnoreParenImpCasts()); 10705 getDiagnostics().setSuppressAllDiagnostics(Suppress); 10706 if (!Res.isUsable() && !isa<OMPArraySectionExpr>(SimpleExpr)) { 10707 Diag(ELoc, diag::err_omp_expected_addressable_lvalue_or_array_item) 10708 << RefExpr->getSourceRange(); 10709 continue; 10710 } 10711 } 10712 Vars.push_back(RefExpr->IgnoreParenImpCasts()); 10713 } 10714 10715 if (!CurContext->isDependentContext() && DepKind == OMPC_DEPEND_sink && 10716 TotalDepCount > VarList.size() && 10717 DSAStack->getParentOrderedRegionParam() && 10718 DSAStack->getParentLoopControlVariable(VarList.size() + 1)) { 10719 Diag(EndLoc, diag::err_omp_depend_sink_expected_loop_iteration) << 1 10720 << DSAStack->getParentLoopControlVariable(VarList.size() + 1); 10721 } 10722 if (DepKind != OMPC_DEPEND_source && DepKind != OMPC_DEPEND_sink && 10723 Vars.empty()) 10724 return nullptr; 10725 } 10726 auto *C = OMPDependClause::Create(Context, StartLoc, LParenLoc, EndLoc, 10727 DepKind, DepLoc, ColonLoc, Vars); 10728 if (DepKind == OMPC_DEPEND_sink || DepKind == OMPC_DEPEND_source) 10729 DSAStack->addDoacrossDependClause(C, OpsOffs); 10730 return C; 10731 } 10732 10733 OMPClause *Sema::ActOnOpenMPDeviceClause(Expr *Device, SourceLocation StartLoc, 10734 SourceLocation LParenLoc, 10735 SourceLocation EndLoc) { 10736 Expr *ValExpr = Device; 10737 Stmt *HelperValStmt = nullptr; 10738 10739 // OpenMP [2.9.1, Restrictions] 10740 // The device expression must evaluate to a non-negative integer value. 10741 if (!IsNonNegativeIntegerValue(ValExpr, *this, OMPC_device, 10742 /*StrictlyPositive=*/false)) 10743 return nullptr; 10744 10745 OpenMPDirectiveKind DKind = DSAStack->getCurrentDirective(); 10746 if (isOpenMPTargetExecutionDirective(DKind) && 10747 !CurContext->isDependentContext()) { 10748 llvm::MapVector<Expr *, DeclRefExpr *> Captures; 10749 ValExpr = tryBuildCapture(*this, ValExpr, Captures).get(); 10750 HelperValStmt = buildPreInits(Context, Captures); 10751 } 10752 10753 return new (Context) 10754 OMPDeviceClause(ValExpr, HelperValStmt, StartLoc, LParenLoc, EndLoc); 10755 } 10756 10757 static bool CheckTypeMappable(SourceLocation SL, SourceRange SR, Sema &SemaRef, 10758 DSAStackTy *Stack, QualType QTy) { 10759 NamedDecl *ND; 10760 if (QTy->isIncompleteType(&ND)) { 10761 SemaRef.Diag(SL, diag::err_incomplete_type) << QTy << SR; 10762 return false; 10763 } 10764 return true; 10765 } 10766 10767 /// \brief Return true if it can be proven that the provided array expression 10768 /// (array section or array subscript) does NOT specify the whole size of the 10769 /// array whose base type is \a BaseQTy. 10770 static bool CheckArrayExpressionDoesNotReferToWholeSize(Sema &SemaRef, 10771 const Expr *E, 10772 QualType BaseQTy) { 10773 auto *OASE = dyn_cast<OMPArraySectionExpr>(E); 10774 10775 // If this is an array subscript, it refers to the whole size if the size of 10776 // the dimension is constant and equals 1. Also, an array section assumes the 10777 // format of an array subscript if no colon is used. 10778 if (isa<ArraySubscriptExpr>(E) || (OASE && OASE->getColonLoc().isInvalid())) { 10779 if (auto *ATy = dyn_cast<ConstantArrayType>(BaseQTy.getTypePtr())) 10780 return ATy->getSize().getSExtValue() != 1; 10781 // Size can't be evaluated statically. 10782 return false; 10783 } 10784 10785 assert(OASE && "Expecting array section if not an array subscript."); 10786 auto *LowerBound = OASE->getLowerBound(); 10787 auto *Length = OASE->getLength(); 10788 10789 // If there is a lower bound that does not evaluates to zero, we are not 10790 // covering the whole dimension. 10791 if (LowerBound) { 10792 llvm::APSInt ConstLowerBound; 10793 if (!LowerBound->EvaluateAsInt(ConstLowerBound, SemaRef.getASTContext())) 10794 return false; // Can't get the integer value as a constant. 10795 if (ConstLowerBound.getSExtValue()) 10796 return true; 10797 } 10798 10799 // If we don't have a length we covering the whole dimension. 10800 if (!Length) 10801 return false; 10802 10803 // If the base is a pointer, we don't have a way to get the size of the 10804 // pointee. 10805 if (BaseQTy->isPointerType()) 10806 return false; 10807 10808 // We can only check if the length is the same as the size of the dimension 10809 // if we have a constant array. 10810 auto *CATy = dyn_cast<ConstantArrayType>(BaseQTy.getTypePtr()); 10811 if (!CATy) 10812 return false; 10813 10814 llvm::APSInt ConstLength; 10815 if (!Length->EvaluateAsInt(ConstLength, SemaRef.getASTContext())) 10816 return false; // Can't get the integer value as a constant. 10817 10818 return CATy->getSize().getSExtValue() != ConstLength.getSExtValue(); 10819 } 10820 10821 // Return true if it can be proven that the provided array expression (array 10822 // section or array subscript) does NOT specify a single element of the array 10823 // whose base type is \a BaseQTy. 10824 static bool CheckArrayExpressionDoesNotReferToUnitySize(Sema &SemaRef, 10825 const Expr *E, 10826 QualType BaseQTy) { 10827 auto *OASE = dyn_cast<OMPArraySectionExpr>(E); 10828 10829 // An array subscript always refer to a single element. Also, an array section 10830 // assumes the format of an array subscript if no colon is used. 10831 if (isa<ArraySubscriptExpr>(E) || (OASE && OASE->getColonLoc().isInvalid())) 10832 return false; 10833 10834 assert(OASE && "Expecting array section if not an array subscript."); 10835 auto *Length = OASE->getLength(); 10836 10837 // If we don't have a length we have to check if the array has unitary size 10838 // for this dimension. Also, we should always expect a length if the base type 10839 // is pointer. 10840 if (!Length) { 10841 if (auto *ATy = dyn_cast<ConstantArrayType>(BaseQTy.getTypePtr())) 10842 return ATy->getSize().getSExtValue() != 1; 10843 // We cannot assume anything. 10844 return false; 10845 } 10846 10847 // Check if the length evaluates to 1. 10848 llvm::APSInt ConstLength; 10849 if (!Length->EvaluateAsInt(ConstLength, SemaRef.getASTContext())) 10850 return false; // Can't get the integer value as a constant. 10851 10852 return ConstLength.getSExtValue() != 1; 10853 } 10854 10855 // Return the expression of the base of the mappable expression or null if it 10856 // cannot be determined and do all the necessary checks to see if the expression 10857 // is valid as a standalone mappable expression. In the process, record all the 10858 // components of the expression. 10859 static Expr *CheckMapClauseExpressionBase( 10860 Sema &SemaRef, Expr *E, 10861 OMPClauseMappableExprCommon::MappableExprComponentList &CurComponents, 10862 OpenMPClauseKind CKind) { 10863 SourceLocation ELoc = E->getExprLoc(); 10864 SourceRange ERange = E->getSourceRange(); 10865 10866 // The base of elements of list in a map clause have to be either: 10867 // - a reference to variable or field. 10868 // - a member expression. 10869 // - an array expression. 10870 // 10871 // E.g. if we have the expression 'r.S.Arr[:12]', we want to retrieve the 10872 // reference to 'r'. 10873 // 10874 // If we have: 10875 // 10876 // struct SS { 10877 // Bla S; 10878 // foo() { 10879 // #pragma omp target map (S.Arr[:12]); 10880 // } 10881 // } 10882 // 10883 // We want to retrieve the member expression 'this->S'; 10884 10885 Expr *RelevantExpr = nullptr; 10886 10887 // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, p.2] 10888 // If a list item is an array section, it must specify contiguous storage. 10889 // 10890 // For this restriction it is sufficient that we make sure only references 10891 // to variables or fields and array expressions, and that no array sections 10892 // exist except in the rightmost expression (unless they cover the whole 10893 // dimension of the array). E.g. these would be invalid: 10894 // 10895 // r.ArrS[3:5].Arr[6:7] 10896 // 10897 // r.ArrS[3:5].x 10898 // 10899 // but these would be valid: 10900 // r.ArrS[3].Arr[6:7] 10901 // 10902 // r.ArrS[3].x 10903 10904 bool AllowUnitySizeArraySection = true; 10905 bool AllowWholeSizeArraySection = true; 10906 10907 while (!RelevantExpr) { 10908 E = E->IgnoreParenImpCasts(); 10909 10910 if (auto *CurE = dyn_cast<DeclRefExpr>(E)) { 10911 if (!isa<VarDecl>(CurE->getDecl())) 10912 break; 10913 10914 RelevantExpr = CurE; 10915 10916 // If we got a reference to a declaration, we should not expect any array 10917 // section before that. 10918 AllowUnitySizeArraySection = false; 10919 AllowWholeSizeArraySection = false; 10920 10921 // Record the component. 10922 CurComponents.push_back(OMPClauseMappableExprCommon::MappableComponent( 10923 CurE, CurE->getDecl())); 10924 continue; 10925 } 10926 10927 if (auto *CurE = dyn_cast<MemberExpr>(E)) { 10928 auto *BaseE = CurE->getBase()->IgnoreParenImpCasts(); 10929 10930 if (isa<CXXThisExpr>(BaseE)) 10931 // We found a base expression: this->Val. 10932 RelevantExpr = CurE; 10933 else 10934 E = BaseE; 10935 10936 if (!isa<FieldDecl>(CurE->getMemberDecl())) { 10937 SemaRef.Diag(ELoc, diag::err_omp_expected_access_to_data_field) 10938 << CurE->getSourceRange(); 10939 break; 10940 } 10941 10942 auto *FD = cast<FieldDecl>(CurE->getMemberDecl()); 10943 10944 // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, C/C++, p.3] 10945 // A bit-field cannot appear in a map clause. 10946 // 10947 if (FD->isBitField()) { 10948 SemaRef.Diag(ELoc, diag::err_omp_bit_fields_forbidden_in_clause) 10949 << CurE->getSourceRange() << getOpenMPClauseName(CKind); 10950 break; 10951 } 10952 10953 // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, C++, p.1] 10954 // If the type of a list item is a reference to a type T then the type 10955 // will be considered to be T for all purposes of this clause. 10956 QualType CurType = BaseE->getType().getNonReferenceType(); 10957 10958 // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, C/C++, p.2] 10959 // A list item cannot be a variable that is a member of a structure with 10960 // a union type. 10961 // 10962 if (auto *RT = CurType->getAs<RecordType>()) 10963 if (RT->isUnionType()) { 10964 SemaRef.Diag(ELoc, diag::err_omp_union_type_not_allowed) 10965 << CurE->getSourceRange(); 10966 break; 10967 } 10968 10969 // If we got a member expression, we should not expect any array section 10970 // before that: 10971 // 10972 // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, p.7] 10973 // If a list item is an element of a structure, only the rightmost symbol 10974 // of the variable reference can be an array section. 10975 // 10976 AllowUnitySizeArraySection = false; 10977 AllowWholeSizeArraySection = false; 10978 10979 // Record the component. 10980 CurComponents.push_back( 10981 OMPClauseMappableExprCommon::MappableComponent(CurE, FD)); 10982 continue; 10983 } 10984 10985 if (auto *CurE = dyn_cast<ArraySubscriptExpr>(E)) { 10986 E = CurE->getBase()->IgnoreParenImpCasts(); 10987 10988 if (!E->getType()->isAnyPointerType() && !E->getType()->isArrayType()) { 10989 SemaRef.Diag(ELoc, diag::err_omp_expected_base_var_name) 10990 << 0 << CurE->getSourceRange(); 10991 break; 10992 } 10993 10994 // If we got an array subscript that express the whole dimension we 10995 // can have any array expressions before. If it only expressing part of 10996 // the dimension, we can only have unitary-size array expressions. 10997 if (CheckArrayExpressionDoesNotReferToWholeSize(SemaRef, CurE, 10998 E->getType())) 10999 AllowWholeSizeArraySection = false; 11000 11001 // Record the component - we don't have any declaration associated. 11002 CurComponents.push_back( 11003 OMPClauseMappableExprCommon::MappableComponent(CurE, nullptr)); 11004 continue; 11005 } 11006 11007 if (auto *CurE = dyn_cast<OMPArraySectionExpr>(E)) { 11008 E = CurE->getBase()->IgnoreParenImpCasts(); 11009 11010 auto CurType = 11011 OMPArraySectionExpr::getBaseOriginalType(E).getCanonicalType(); 11012 11013 // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, C++, p.1] 11014 // If the type of a list item is a reference to a type T then the type 11015 // will be considered to be T for all purposes of this clause. 11016 if (CurType->isReferenceType()) 11017 CurType = CurType->getPointeeType(); 11018 11019 bool IsPointer = CurType->isAnyPointerType(); 11020 11021 if (!IsPointer && !CurType->isArrayType()) { 11022 SemaRef.Diag(ELoc, diag::err_omp_expected_base_var_name) 11023 << 0 << CurE->getSourceRange(); 11024 break; 11025 } 11026 11027 bool NotWhole = 11028 CheckArrayExpressionDoesNotReferToWholeSize(SemaRef, CurE, CurType); 11029 bool NotUnity = 11030 CheckArrayExpressionDoesNotReferToUnitySize(SemaRef, CurE, CurType); 11031 11032 if (AllowWholeSizeArraySection) { 11033 // Any array section is currently allowed. Allowing a whole size array 11034 // section implies allowing a unity array section as well. 11035 // 11036 // If this array section refers to the whole dimension we can still 11037 // accept other array sections before this one, except if the base is a 11038 // pointer. Otherwise, only unitary sections are accepted. 11039 if (NotWhole || IsPointer) 11040 AllowWholeSizeArraySection = false; 11041 } else if (AllowUnitySizeArraySection && NotUnity) { 11042 // A unity or whole array section is not allowed and that is not 11043 // compatible with the properties of the current array section. 11044 SemaRef.Diag( 11045 ELoc, diag::err_array_section_does_not_specify_contiguous_storage) 11046 << CurE->getSourceRange(); 11047 break; 11048 } 11049 11050 // Record the component - we don't have any declaration associated. 11051 CurComponents.push_back( 11052 OMPClauseMappableExprCommon::MappableComponent(CurE, nullptr)); 11053 continue; 11054 } 11055 11056 // If nothing else worked, this is not a valid map clause expression. 11057 SemaRef.Diag(ELoc, 11058 diag::err_omp_expected_named_var_member_or_array_expression) 11059 << ERange; 11060 break; 11061 } 11062 11063 return RelevantExpr; 11064 } 11065 11066 // Return true if expression E associated with value VD has conflicts with other 11067 // map information. 11068 static bool CheckMapConflicts( 11069 Sema &SemaRef, DSAStackTy *DSAS, ValueDecl *VD, Expr *E, 11070 bool CurrentRegionOnly, 11071 OMPClauseMappableExprCommon::MappableExprComponentListRef CurComponents, 11072 OpenMPClauseKind CKind) { 11073 assert(VD && E); 11074 SourceLocation ELoc = E->getExprLoc(); 11075 SourceRange ERange = E->getSourceRange(); 11076 11077 // In order to easily check the conflicts we need to match each component of 11078 // the expression under test with the components of the expressions that are 11079 // already in the stack. 11080 11081 assert(!CurComponents.empty() && "Map clause expression with no components!"); 11082 assert(CurComponents.back().getAssociatedDeclaration() == VD && 11083 "Map clause expression with unexpected base!"); 11084 11085 // Variables to help detecting enclosing problems in data environment nests. 11086 bool IsEnclosedByDataEnvironmentExpr = false; 11087 const Expr *EnclosingExpr = nullptr; 11088 11089 bool FoundError = DSAS->checkMappableExprComponentListsForDecl( 11090 VD, CurrentRegionOnly, 11091 [&](OMPClauseMappableExprCommon::MappableExprComponentListRef 11092 StackComponents, 11093 OpenMPClauseKind) -> bool { 11094 11095 assert(!StackComponents.empty() && 11096 "Map clause expression with no components!"); 11097 assert(StackComponents.back().getAssociatedDeclaration() == VD && 11098 "Map clause expression with unexpected base!"); 11099 11100 // The whole expression in the stack. 11101 auto *RE = StackComponents.front().getAssociatedExpression(); 11102 11103 // Expressions must start from the same base. Here we detect at which 11104 // point both expressions diverge from each other and see if we can 11105 // detect if the memory referred to both expressions is contiguous and 11106 // do not overlap. 11107 auto CI = CurComponents.rbegin(); 11108 auto CE = CurComponents.rend(); 11109 auto SI = StackComponents.rbegin(); 11110 auto SE = StackComponents.rend(); 11111 for (; CI != CE && SI != SE; ++CI, ++SI) { 11112 11113 // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, p.3] 11114 // At most one list item can be an array item derived from a given 11115 // variable in map clauses of the same construct. 11116 if (CurrentRegionOnly && 11117 (isa<ArraySubscriptExpr>(CI->getAssociatedExpression()) || 11118 isa<OMPArraySectionExpr>(CI->getAssociatedExpression())) && 11119 (isa<ArraySubscriptExpr>(SI->getAssociatedExpression()) || 11120 isa<OMPArraySectionExpr>(SI->getAssociatedExpression()))) { 11121 SemaRef.Diag(CI->getAssociatedExpression()->getExprLoc(), 11122 diag::err_omp_multiple_array_items_in_map_clause) 11123 << CI->getAssociatedExpression()->getSourceRange(); 11124 SemaRef.Diag(SI->getAssociatedExpression()->getExprLoc(), 11125 diag::note_used_here) 11126 << SI->getAssociatedExpression()->getSourceRange(); 11127 return true; 11128 } 11129 11130 // Do both expressions have the same kind? 11131 if (CI->getAssociatedExpression()->getStmtClass() != 11132 SI->getAssociatedExpression()->getStmtClass()) 11133 break; 11134 11135 // Are we dealing with different variables/fields? 11136 if (CI->getAssociatedDeclaration() != SI->getAssociatedDeclaration()) 11137 break; 11138 } 11139 // Check if the extra components of the expressions in the enclosing 11140 // data environment are redundant for the current base declaration. 11141 // If they are, the maps completely overlap, which is legal. 11142 for (; SI != SE; ++SI) { 11143 QualType Type; 11144 if (auto *ASE = 11145 dyn_cast<ArraySubscriptExpr>(SI->getAssociatedExpression())) { 11146 Type = ASE->getBase()->IgnoreParenImpCasts()->getType(); 11147 } else if (auto *OASE = dyn_cast<OMPArraySectionExpr>( 11148 SI->getAssociatedExpression())) { 11149 auto *E = OASE->getBase()->IgnoreParenImpCasts(); 11150 Type = 11151 OMPArraySectionExpr::getBaseOriginalType(E).getCanonicalType(); 11152 } 11153 if (Type.isNull() || Type->isAnyPointerType() || 11154 CheckArrayExpressionDoesNotReferToWholeSize( 11155 SemaRef, SI->getAssociatedExpression(), Type)) 11156 break; 11157 } 11158 11159 // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, p.4] 11160 // List items of map clauses in the same construct must not share 11161 // original storage. 11162 // 11163 // If the expressions are exactly the same or one is a subset of the 11164 // other, it means they are sharing storage. 11165 if (CI == CE && SI == SE) { 11166 if (CurrentRegionOnly) { 11167 if (CKind == OMPC_map) 11168 SemaRef.Diag(ELoc, diag::err_omp_map_shared_storage) << ERange; 11169 else { 11170 assert(CKind == OMPC_to || CKind == OMPC_from); 11171 SemaRef.Diag(ELoc, diag::err_omp_once_referenced_in_target_update) 11172 << ERange; 11173 } 11174 SemaRef.Diag(RE->getExprLoc(), diag::note_used_here) 11175 << RE->getSourceRange(); 11176 return true; 11177 } else { 11178 // If we find the same expression in the enclosing data environment, 11179 // that is legal. 11180 IsEnclosedByDataEnvironmentExpr = true; 11181 return false; 11182 } 11183 } 11184 11185 QualType DerivedType = 11186 std::prev(CI)->getAssociatedDeclaration()->getType(); 11187 SourceLocation DerivedLoc = 11188 std::prev(CI)->getAssociatedExpression()->getExprLoc(); 11189 11190 // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, C++, p.1] 11191 // If the type of a list item is a reference to a type T then the type 11192 // will be considered to be T for all purposes of this clause. 11193 DerivedType = DerivedType.getNonReferenceType(); 11194 11195 // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, C/C++, p.1] 11196 // A variable for which the type is pointer and an array section 11197 // derived from that variable must not appear as list items of map 11198 // clauses of the same construct. 11199 // 11200 // Also, cover one of the cases in: 11201 // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, p.5] 11202 // If any part of the original storage of a list item has corresponding 11203 // storage in the device data environment, all of the original storage 11204 // must have corresponding storage in the device data environment. 11205 // 11206 if (DerivedType->isAnyPointerType()) { 11207 if (CI == CE || SI == SE) { 11208 SemaRef.Diag( 11209 DerivedLoc, 11210 diag::err_omp_pointer_mapped_along_with_derived_section) 11211 << DerivedLoc; 11212 } else { 11213 assert(CI != CE && SI != SE); 11214 SemaRef.Diag(DerivedLoc, diag::err_omp_same_pointer_derreferenced) 11215 << DerivedLoc; 11216 } 11217 SemaRef.Diag(RE->getExprLoc(), diag::note_used_here) 11218 << RE->getSourceRange(); 11219 return true; 11220 } 11221 11222 // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, p.4] 11223 // List items of map clauses in the same construct must not share 11224 // original storage. 11225 // 11226 // An expression is a subset of the other. 11227 if (CurrentRegionOnly && (CI == CE || SI == SE)) { 11228 if (CKind == OMPC_map) 11229 SemaRef.Diag(ELoc, diag::err_omp_map_shared_storage) << ERange; 11230 else { 11231 assert(CKind == OMPC_to || CKind == OMPC_from); 11232 SemaRef.Diag(ELoc, diag::err_omp_once_referenced_in_target_update) 11233 << ERange; 11234 } 11235 SemaRef.Diag(RE->getExprLoc(), diag::note_used_here) 11236 << RE->getSourceRange(); 11237 return true; 11238 } 11239 11240 // The current expression uses the same base as other expression in the 11241 // data environment but does not contain it completely. 11242 if (!CurrentRegionOnly && SI != SE) 11243 EnclosingExpr = RE; 11244 11245 // The current expression is a subset of the expression in the data 11246 // environment. 11247 IsEnclosedByDataEnvironmentExpr |= 11248 (!CurrentRegionOnly && CI != CE && SI == SE); 11249 11250 return false; 11251 }); 11252 11253 if (CurrentRegionOnly) 11254 return FoundError; 11255 11256 // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, p.5] 11257 // If any part of the original storage of a list item has corresponding 11258 // storage in the device data environment, all of the original storage must 11259 // have corresponding storage in the device data environment. 11260 // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, p.6] 11261 // If a list item is an element of a structure, and a different element of 11262 // the structure has a corresponding list item in the device data environment 11263 // prior to a task encountering the construct associated with the map clause, 11264 // then the list item must also have a corresponding list item in the device 11265 // data environment prior to the task encountering the construct. 11266 // 11267 if (EnclosingExpr && !IsEnclosedByDataEnvironmentExpr) { 11268 SemaRef.Diag(ELoc, 11269 diag::err_omp_original_storage_is_shared_and_does_not_contain) 11270 << ERange; 11271 SemaRef.Diag(EnclosingExpr->getExprLoc(), diag::note_used_here) 11272 << EnclosingExpr->getSourceRange(); 11273 return true; 11274 } 11275 11276 return FoundError; 11277 } 11278 11279 namespace { 11280 // Utility struct that gathers all the related lists associated with a mappable 11281 // expression. 11282 struct MappableVarListInfo final { 11283 // The list of expressions. 11284 ArrayRef<Expr *> VarList; 11285 // The list of processed expressions. 11286 SmallVector<Expr *, 16> ProcessedVarList; 11287 // The mappble components for each expression. 11288 OMPClauseMappableExprCommon::MappableExprComponentLists VarComponents; 11289 // The base declaration of the variable. 11290 SmallVector<ValueDecl *, 16> VarBaseDeclarations; 11291 11292 MappableVarListInfo(ArrayRef<Expr *> VarList) : VarList(VarList) { 11293 // We have a list of components and base declarations for each entry in the 11294 // variable list. 11295 VarComponents.reserve(VarList.size()); 11296 VarBaseDeclarations.reserve(VarList.size()); 11297 } 11298 }; 11299 } 11300 11301 // Check the validity of the provided variable list for the provided clause kind 11302 // \a CKind. In the check process the valid expressions, and mappable expression 11303 // components and variables are extracted and used to fill \a Vars, 11304 // \a ClauseComponents, and \a ClauseBaseDeclarations. \a MapType and 11305 // \a IsMapTypeImplicit are expected to be valid if the clause kind is 'map'. 11306 static void 11307 checkMappableExpressionList(Sema &SemaRef, DSAStackTy *DSAS, 11308 OpenMPClauseKind CKind, MappableVarListInfo &MVLI, 11309 SourceLocation StartLoc, 11310 OpenMPMapClauseKind MapType = OMPC_MAP_unknown, 11311 bool IsMapTypeImplicit = false) { 11312 // We only expect mappable expressions in 'to', 'from', and 'map' clauses. 11313 assert((CKind == OMPC_map || CKind == OMPC_to || CKind == OMPC_from) && 11314 "Unexpected clause kind with mappable expressions!"); 11315 11316 // Keep track of the mappable components and base declarations in this clause. 11317 // Each entry in the list is going to have a list of components associated. We 11318 // record each set of the components so that we can build the clause later on. 11319 // In the end we should have the same amount of declarations and component 11320 // lists. 11321 11322 for (auto &RE : MVLI.VarList) { 11323 assert(RE && "Null expr in omp to/from/map clause"); 11324 SourceLocation ELoc = RE->getExprLoc(); 11325 11326 auto *VE = RE->IgnoreParenLValueCasts(); 11327 11328 if (VE->isValueDependent() || VE->isTypeDependent() || 11329 VE->isInstantiationDependent() || 11330 VE->containsUnexpandedParameterPack()) { 11331 // We can only analyze this information once the missing information is 11332 // resolved. 11333 MVLI.ProcessedVarList.push_back(RE); 11334 continue; 11335 } 11336 11337 auto *SimpleExpr = RE->IgnoreParenCasts(); 11338 11339 if (!RE->IgnoreParenImpCasts()->isLValue()) { 11340 SemaRef.Diag(ELoc, 11341 diag::err_omp_expected_named_var_member_or_array_expression) 11342 << RE->getSourceRange(); 11343 continue; 11344 } 11345 11346 OMPClauseMappableExprCommon::MappableExprComponentList CurComponents; 11347 ValueDecl *CurDeclaration = nullptr; 11348 11349 // Obtain the array or member expression bases if required. Also, fill the 11350 // components array with all the components identified in the process. 11351 auto *BE = 11352 CheckMapClauseExpressionBase(SemaRef, SimpleExpr, CurComponents, CKind); 11353 if (!BE) 11354 continue; 11355 11356 assert(!CurComponents.empty() && 11357 "Invalid mappable expression information."); 11358 11359 // For the following checks, we rely on the base declaration which is 11360 // expected to be associated with the last component. The declaration is 11361 // expected to be a variable or a field (if 'this' is being mapped). 11362 CurDeclaration = CurComponents.back().getAssociatedDeclaration(); 11363 assert(CurDeclaration && "Null decl on map clause."); 11364 assert( 11365 CurDeclaration->isCanonicalDecl() && 11366 "Expecting components to have associated only canonical declarations."); 11367 11368 auto *VD = dyn_cast<VarDecl>(CurDeclaration); 11369 auto *FD = dyn_cast<FieldDecl>(CurDeclaration); 11370 11371 assert((VD || FD) && "Only variables or fields are expected here!"); 11372 (void)FD; 11373 11374 // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, p.10] 11375 // threadprivate variables cannot appear in a map clause. 11376 // OpenMP 4.5 [2.10.5, target update Construct] 11377 // threadprivate variables cannot appear in a from clause. 11378 if (VD && DSAS->isThreadPrivate(VD)) { 11379 auto DVar = DSAS->getTopDSA(VD, false); 11380 SemaRef.Diag(ELoc, diag::err_omp_threadprivate_in_clause) 11381 << getOpenMPClauseName(CKind); 11382 ReportOriginalDSA(SemaRef, DSAS, VD, DVar); 11383 continue; 11384 } 11385 11386 // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, p.9] 11387 // A list item cannot appear in both a map clause and a data-sharing 11388 // attribute clause on the same construct. 11389 11390 // Check conflicts with other map clause expressions. We check the conflicts 11391 // with the current construct separately from the enclosing data 11392 // environment, because the restrictions are different. We only have to 11393 // check conflicts across regions for the map clauses. 11394 if (CheckMapConflicts(SemaRef, DSAS, CurDeclaration, SimpleExpr, 11395 /*CurrentRegionOnly=*/true, CurComponents, CKind)) 11396 break; 11397 if (CKind == OMPC_map && 11398 CheckMapConflicts(SemaRef, DSAS, CurDeclaration, SimpleExpr, 11399 /*CurrentRegionOnly=*/false, CurComponents, CKind)) 11400 break; 11401 11402 // OpenMP 4.5 [2.10.5, target update Construct] 11403 // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, C++, p.1] 11404 // If the type of a list item is a reference to a type T then the type will 11405 // be considered to be T for all purposes of this clause. 11406 QualType Type = CurDeclaration->getType().getNonReferenceType(); 11407 11408 // OpenMP 4.5 [2.10.5, target update Construct, Restrictions, p.4] 11409 // A list item in a to or from clause must have a mappable type. 11410 // OpenMP 4.5 [2.15.5.1, map Clause, Restrictions, p.9] 11411 // A list item must have a mappable type. 11412 if (!CheckTypeMappable(VE->getExprLoc(), VE->getSourceRange(), SemaRef, 11413 DSAS, Type)) 11414 continue; 11415 11416 if (CKind == OMPC_map) { 11417 // target enter data 11418 // OpenMP [2.10.2, Restrictions, p. 99] 11419 // A map-type must be specified in all map clauses and must be either 11420 // to or alloc. 11421 OpenMPDirectiveKind DKind = DSAS->getCurrentDirective(); 11422 if (DKind == OMPD_target_enter_data && 11423 !(MapType == OMPC_MAP_to || MapType == OMPC_MAP_alloc)) { 11424 SemaRef.Diag(StartLoc, diag::err_omp_invalid_map_type_for_directive) 11425 << (IsMapTypeImplicit ? 1 : 0) 11426 << getOpenMPSimpleClauseTypeName(OMPC_map, MapType) 11427 << getOpenMPDirectiveName(DKind); 11428 continue; 11429 } 11430 11431 // target exit_data 11432 // OpenMP [2.10.3, Restrictions, p. 102] 11433 // A map-type must be specified in all map clauses and must be either 11434 // from, release, or delete. 11435 if (DKind == OMPD_target_exit_data && 11436 !(MapType == OMPC_MAP_from || MapType == OMPC_MAP_release || 11437 MapType == OMPC_MAP_delete)) { 11438 SemaRef.Diag(StartLoc, diag::err_omp_invalid_map_type_for_directive) 11439 << (IsMapTypeImplicit ? 1 : 0) 11440 << getOpenMPSimpleClauseTypeName(OMPC_map, MapType) 11441 << getOpenMPDirectiveName(DKind); 11442 continue; 11443 } 11444 11445 // OpenMP 4.5 [2.15.5.1, Restrictions, p.3] 11446 // A list item cannot appear in both a map clause and a data-sharing 11447 // attribute clause on the same construct 11448 if ((DKind == OMPD_target || DKind == OMPD_target_teams || 11449 DKind == OMPD_target_teams_distribute || 11450 DKind == OMPD_target_teams_distribute_parallel_for || 11451 DKind == OMPD_target_teams_distribute_parallel_for_simd || 11452 DKind == OMPD_target_teams_distribute_simd) && VD) { 11453 auto DVar = DSAS->getTopDSA(VD, false); 11454 if (isOpenMPPrivate(DVar.CKind)) { 11455 SemaRef.Diag(ELoc, diag::err_omp_variable_in_given_clause_and_dsa) 11456 << getOpenMPClauseName(DVar.CKind) 11457 << getOpenMPClauseName(OMPC_map) 11458 << getOpenMPDirectiveName(DSAS->getCurrentDirective()); 11459 ReportOriginalDSA(SemaRef, DSAS, CurDeclaration, DVar); 11460 continue; 11461 } 11462 } 11463 } 11464 11465 // Save the current expression. 11466 MVLI.ProcessedVarList.push_back(RE); 11467 11468 // Store the components in the stack so that they can be used to check 11469 // against other clauses later on. 11470 DSAS->addMappableExpressionComponents(CurDeclaration, CurComponents, 11471 /*WhereFoundClauseKind=*/OMPC_map); 11472 11473 // Save the components and declaration to create the clause. For purposes of 11474 // the clause creation, any component list that has has base 'this' uses 11475 // null as base declaration. 11476 MVLI.VarComponents.resize(MVLI.VarComponents.size() + 1); 11477 MVLI.VarComponents.back().append(CurComponents.begin(), 11478 CurComponents.end()); 11479 MVLI.VarBaseDeclarations.push_back(isa<MemberExpr>(BE) ? nullptr 11480 : CurDeclaration); 11481 } 11482 } 11483 11484 OMPClause * 11485 Sema::ActOnOpenMPMapClause(OpenMPMapClauseKind MapTypeModifier, 11486 OpenMPMapClauseKind MapType, bool IsMapTypeImplicit, 11487 SourceLocation MapLoc, SourceLocation ColonLoc, 11488 ArrayRef<Expr *> VarList, SourceLocation StartLoc, 11489 SourceLocation LParenLoc, SourceLocation EndLoc) { 11490 MappableVarListInfo MVLI(VarList); 11491 checkMappableExpressionList(*this, DSAStack, OMPC_map, MVLI, StartLoc, 11492 MapType, IsMapTypeImplicit); 11493 11494 // We need to produce a map clause even if we don't have variables so that 11495 // other diagnostics related with non-existing map clauses are accurate. 11496 return OMPMapClause::Create(Context, StartLoc, LParenLoc, EndLoc, 11497 MVLI.ProcessedVarList, MVLI.VarBaseDeclarations, 11498 MVLI.VarComponents, MapTypeModifier, MapType, 11499 IsMapTypeImplicit, MapLoc); 11500 } 11501 11502 QualType Sema::ActOnOpenMPDeclareReductionType(SourceLocation TyLoc, 11503 TypeResult ParsedType) { 11504 assert(ParsedType.isUsable()); 11505 11506 QualType ReductionType = GetTypeFromParser(ParsedType.get()); 11507 if (ReductionType.isNull()) 11508 return QualType(); 11509 11510 // [OpenMP 4.0], 2.15 declare reduction Directive, Restrictions, C\C++ 11511 // A type name in a declare reduction directive cannot be a function type, an 11512 // array type, a reference type, or a type qualified with const, volatile or 11513 // restrict. 11514 if (ReductionType.hasQualifiers()) { 11515 Diag(TyLoc, diag::err_omp_reduction_wrong_type) << 0; 11516 return QualType(); 11517 } 11518 11519 if (ReductionType->isFunctionType()) { 11520 Diag(TyLoc, diag::err_omp_reduction_wrong_type) << 1; 11521 return QualType(); 11522 } 11523 if (ReductionType->isReferenceType()) { 11524 Diag(TyLoc, diag::err_omp_reduction_wrong_type) << 2; 11525 return QualType(); 11526 } 11527 if (ReductionType->isArrayType()) { 11528 Diag(TyLoc, diag::err_omp_reduction_wrong_type) << 3; 11529 return QualType(); 11530 } 11531 return ReductionType; 11532 } 11533 11534 Sema::DeclGroupPtrTy Sema::ActOnOpenMPDeclareReductionDirectiveStart( 11535 Scope *S, DeclContext *DC, DeclarationName Name, 11536 ArrayRef<std::pair<QualType, SourceLocation>> ReductionTypes, 11537 AccessSpecifier AS, Decl *PrevDeclInScope) { 11538 SmallVector<Decl *, 8> Decls; 11539 Decls.reserve(ReductionTypes.size()); 11540 11541 LookupResult Lookup(*this, Name, SourceLocation(), LookupOMPReductionName, 11542 ForRedeclaration); 11543 // [OpenMP 4.0], 2.15 declare reduction Directive, Restrictions 11544 // A reduction-identifier may not be re-declared in the current scope for the 11545 // same type or for a type that is compatible according to the base language 11546 // rules. 11547 llvm::DenseMap<QualType, SourceLocation> PreviousRedeclTypes; 11548 OMPDeclareReductionDecl *PrevDRD = nullptr; 11549 bool InCompoundScope = true; 11550 if (S != nullptr) { 11551 // Find previous declaration with the same name not referenced in other 11552 // declarations. 11553 FunctionScopeInfo *ParentFn = getEnclosingFunction(); 11554 InCompoundScope = 11555 (ParentFn != nullptr) && !ParentFn->CompoundScopes.empty(); 11556 LookupName(Lookup, S); 11557 FilterLookupForScope(Lookup, DC, S, /*ConsiderLinkage=*/false, 11558 /*AllowInlineNamespace=*/false); 11559 llvm::DenseMap<OMPDeclareReductionDecl *, bool> UsedAsPrevious; 11560 auto Filter = Lookup.makeFilter(); 11561 while (Filter.hasNext()) { 11562 auto *PrevDecl = cast<OMPDeclareReductionDecl>(Filter.next()); 11563 if (InCompoundScope) { 11564 auto I = UsedAsPrevious.find(PrevDecl); 11565 if (I == UsedAsPrevious.end()) 11566 UsedAsPrevious[PrevDecl] = false; 11567 if (auto *D = PrevDecl->getPrevDeclInScope()) 11568 UsedAsPrevious[D] = true; 11569 } 11570 PreviousRedeclTypes[PrevDecl->getType().getCanonicalType()] = 11571 PrevDecl->getLocation(); 11572 } 11573 Filter.done(); 11574 if (InCompoundScope) { 11575 for (auto &PrevData : UsedAsPrevious) { 11576 if (!PrevData.second) { 11577 PrevDRD = PrevData.first; 11578 break; 11579 } 11580 } 11581 } 11582 } else if (PrevDeclInScope != nullptr) { 11583 auto *PrevDRDInScope = PrevDRD = 11584 cast<OMPDeclareReductionDecl>(PrevDeclInScope); 11585 do { 11586 PreviousRedeclTypes[PrevDRDInScope->getType().getCanonicalType()] = 11587 PrevDRDInScope->getLocation(); 11588 PrevDRDInScope = PrevDRDInScope->getPrevDeclInScope(); 11589 } while (PrevDRDInScope != nullptr); 11590 } 11591 for (auto &TyData : ReductionTypes) { 11592 auto I = PreviousRedeclTypes.find(TyData.first.getCanonicalType()); 11593 bool Invalid = false; 11594 if (I != PreviousRedeclTypes.end()) { 11595 Diag(TyData.second, diag::err_omp_declare_reduction_redefinition) 11596 << TyData.first; 11597 Diag(I->second, diag::note_previous_definition); 11598 Invalid = true; 11599 } 11600 PreviousRedeclTypes[TyData.first.getCanonicalType()] = TyData.second; 11601 auto *DRD = OMPDeclareReductionDecl::Create(Context, DC, TyData.second, 11602 Name, TyData.first, PrevDRD); 11603 DC->addDecl(DRD); 11604 DRD->setAccess(AS); 11605 Decls.push_back(DRD); 11606 if (Invalid) 11607 DRD->setInvalidDecl(); 11608 else 11609 PrevDRD = DRD; 11610 } 11611 11612 return DeclGroupPtrTy::make( 11613 DeclGroupRef::Create(Context, Decls.begin(), Decls.size())); 11614 } 11615 11616 void Sema::ActOnOpenMPDeclareReductionCombinerStart(Scope *S, Decl *D) { 11617 auto *DRD = cast<OMPDeclareReductionDecl>(D); 11618 11619 // Enter new function scope. 11620 PushFunctionScope(); 11621 getCurFunction()->setHasBranchProtectedScope(); 11622 getCurFunction()->setHasOMPDeclareReductionCombiner(); 11623 11624 if (S != nullptr) 11625 PushDeclContext(S, DRD); 11626 else 11627 CurContext = DRD; 11628 11629 PushExpressionEvaluationContext( 11630 ExpressionEvaluationContext::PotentiallyEvaluated); 11631 11632 QualType ReductionType = DRD->getType(); 11633 // Create 'T* omp_parm;T omp_in;'. All references to 'omp_in' will 11634 // be replaced by '*omp_parm' during codegen. This required because 'omp_in' 11635 // uses semantics of argument handles by value, but it should be passed by 11636 // reference. C lang does not support references, so pass all parameters as 11637 // pointers. 11638 // Create 'T omp_in;' variable. 11639 auto *OmpInParm = 11640 buildVarDecl(*this, D->getLocation(), ReductionType, "omp_in"); 11641 // Create 'T* omp_parm;T omp_out;'. All references to 'omp_out' will 11642 // be replaced by '*omp_parm' during codegen. This required because 'omp_out' 11643 // uses semantics of argument handles by value, but it should be passed by 11644 // reference. C lang does not support references, so pass all parameters as 11645 // pointers. 11646 // Create 'T omp_out;' variable. 11647 auto *OmpOutParm = 11648 buildVarDecl(*this, D->getLocation(), ReductionType, "omp_out"); 11649 if (S != nullptr) { 11650 PushOnScopeChains(OmpInParm, S); 11651 PushOnScopeChains(OmpOutParm, S); 11652 } else { 11653 DRD->addDecl(OmpInParm); 11654 DRD->addDecl(OmpOutParm); 11655 } 11656 } 11657 11658 void Sema::ActOnOpenMPDeclareReductionCombinerEnd(Decl *D, Expr *Combiner) { 11659 auto *DRD = cast<OMPDeclareReductionDecl>(D); 11660 DiscardCleanupsInEvaluationContext(); 11661 PopExpressionEvaluationContext(); 11662 11663 PopDeclContext(); 11664 PopFunctionScopeInfo(); 11665 11666 if (Combiner != nullptr) 11667 DRD->setCombiner(Combiner); 11668 else 11669 DRD->setInvalidDecl(); 11670 } 11671 11672 VarDecl *Sema::ActOnOpenMPDeclareReductionInitializerStart(Scope *S, Decl *D) { 11673 auto *DRD = cast<OMPDeclareReductionDecl>(D); 11674 11675 // Enter new function scope. 11676 PushFunctionScope(); 11677 getCurFunction()->setHasBranchProtectedScope(); 11678 11679 if (S != nullptr) 11680 PushDeclContext(S, DRD); 11681 else 11682 CurContext = DRD; 11683 11684 PushExpressionEvaluationContext( 11685 ExpressionEvaluationContext::PotentiallyEvaluated); 11686 11687 QualType ReductionType = DRD->getType(); 11688 // Create 'T* omp_parm;T omp_priv;'. All references to 'omp_priv' will 11689 // be replaced by '*omp_parm' during codegen. This required because 'omp_priv' 11690 // uses semantics of argument handles by value, but it should be passed by 11691 // reference. C lang does not support references, so pass all parameters as 11692 // pointers. 11693 // Create 'T omp_priv;' variable. 11694 auto *OmpPrivParm = 11695 buildVarDecl(*this, D->getLocation(), ReductionType, "omp_priv"); 11696 // Create 'T* omp_parm;T omp_orig;'. All references to 'omp_orig' will 11697 // be replaced by '*omp_parm' during codegen. This required because 'omp_orig' 11698 // uses semantics of argument handles by value, but it should be passed by 11699 // reference. C lang does not support references, so pass all parameters as 11700 // pointers. 11701 // Create 'T omp_orig;' variable. 11702 auto *OmpOrigParm = 11703 buildVarDecl(*this, D->getLocation(), ReductionType, "omp_orig"); 11704 if (S != nullptr) { 11705 PushOnScopeChains(OmpPrivParm, S); 11706 PushOnScopeChains(OmpOrigParm, S); 11707 } else { 11708 DRD->addDecl(OmpPrivParm); 11709 DRD->addDecl(OmpOrigParm); 11710 } 11711 return OmpPrivParm; 11712 } 11713 11714 void Sema::ActOnOpenMPDeclareReductionInitializerEnd(Decl *D, Expr *Initializer, 11715 VarDecl *OmpPrivParm) { 11716 auto *DRD = cast<OMPDeclareReductionDecl>(D); 11717 DiscardCleanupsInEvaluationContext(); 11718 PopExpressionEvaluationContext(); 11719 11720 PopDeclContext(); 11721 PopFunctionScopeInfo(); 11722 11723 if (Initializer != nullptr) { 11724 DRD->setInitializer(Initializer, OMPDeclareReductionDecl::CallInit); 11725 } else if (OmpPrivParm->hasInit()) { 11726 DRD->setInitializer(OmpPrivParm->getInit(), 11727 OmpPrivParm->isDirectInit() 11728 ? OMPDeclareReductionDecl::DirectInit 11729 : OMPDeclareReductionDecl::CopyInit); 11730 } else { 11731 DRD->setInvalidDecl(); 11732 } 11733 } 11734 11735 Sema::DeclGroupPtrTy Sema::ActOnOpenMPDeclareReductionDirectiveEnd( 11736 Scope *S, DeclGroupPtrTy DeclReductions, bool IsValid) { 11737 for (auto *D : DeclReductions.get()) { 11738 if (IsValid) { 11739 auto *DRD = cast<OMPDeclareReductionDecl>(D); 11740 if (S != nullptr) 11741 PushOnScopeChains(DRD, S, /*AddToContext=*/false); 11742 } else 11743 D->setInvalidDecl(); 11744 } 11745 return DeclReductions; 11746 } 11747 11748 OMPClause *Sema::ActOnOpenMPNumTeamsClause(Expr *NumTeams, 11749 SourceLocation StartLoc, 11750 SourceLocation LParenLoc, 11751 SourceLocation EndLoc) { 11752 Expr *ValExpr = NumTeams; 11753 Stmt *HelperValStmt = nullptr; 11754 OpenMPDirectiveKind CaptureRegion = OMPD_unknown; 11755 11756 // OpenMP [teams Constrcut, Restrictions] 11757 // The num_teams expression must evaluate to a positive integer value. 11758 if (!IsNonNegativeIntegerValue(ValExpr, *this, OMPC_num_teams, 11759 /*StrictlyPositive=*/true)) 11760 return nullptr; 11761 11762 OpenMPDirectiveKind DKind = DSAStack->getCurrentDirective(); 11763 CaptureRegion = getOpenMPCaptureRegionForClause(DKind, OMPC_num_teams); 11764 if (CaptureRegion != OMPD_unknown) { 11765 llvm::MapVector<Expr *, DeclRefExpr *> Captures; 11766 ValExpr = tryBuildCapture(*this, ValExpr, Captures).get(); 11767 HelperValStmt = buildPreInits(Context, Captures); 11768 } 11769 11770 return new (Context) OMPNumTeamsClause(ValExpr, HelperValStmt, CaptureRegion, 11771 StartLoc, LParenLoc, EndLoc); 11772 } 11773 11774 OMPClause *Sema::ActOnOpenMPThreadLimitClause(Expr *ThreadLimit, 11775 SourceLocation StartLoc, 11776 SourceLocation LParenLoc, 11777 SourceLocation EndLoc) { 11778 Expr *ValExpr = ThreadLimit; 11779 Stmt *HelperValStmt = nullptr; 11780 OpenMPDirectiveKind CaptureRegion = OMPD_unknown; 11781 11782 // OpenMP [teams Constrcut, Restrictions] 11783 // The thread_limit expression must evaluate to a positive integer value. 11784 if (!IsNonNegativeIntegerValue(ValExpr, *this, OMPC_thread_limit, 11785 /*StrictlyPositive=*/true)) 11786 return nullptr; 11787 11788 OpenMPDirectiveKind DKind = DSAStack->getCurrentDirective(); 11789 CaptureRegion = getOpenMPCaptureRegionForClause(DKind, OMPC_thread_limit); 11790 if (CaptureRegion != OMPD_unknown) { 11791 llvm::MapVector<Expr *, DeclRefExpr *> Captures; 11792 ValExpr = tryBuildCapture(*this, ValExpr, Captures).get(); 11793 HelperValStmt = buildPreInits(Context, Captures); 11794 } 11795 11796 return new (Context) OMPThreadLimitClause( 11797 ValExpr, HelperValStmt, CaptureRegion, StartLoc, LParenLoc, EndLoc); 11798 } 11799 11800 OMPClause *Sema::ActOnOpenMPPriorityClause(Expr *Priority, 11801 SourceLocation StartLoc, 11802 SourceLocation LParenLoc, 11803 SourceLocation EndLoc) { 11804 Expr *ValExpr = Priority; 11805 11806 // OpenMP [2.9.1, task Constrcut] 11807 // The priority-value is a non-negative numerical scalar expression. 11808 if (!IsNonNegativeIntegerValue(ValExpr, *this, OMPC_priority, 11809 /*StrictlyPositive=*/false)) 11810 return nullptr; 11811 11812 return new (Context) OMPPriorityClause(ValExpr, StartLoc, LParenLoc, EndLoc); 11813 } 11814 11815 OMPClause *Sema::ActOnOpenMPGrainsizeClause(Expr *Grainsize, 11816 SourceLocation StartLoc, 11817 SourceLocation LParenLoc, 11818 SourceLocation EndLoc) { 11819 Expr *ValExpr = Grainsize; 11820 11821 // OpenMP [2.9.2, taskloop Constrcut] 11822 // The parameter of the grainsize clause must be a positive integer 11823 // expression. 11824 if (!IsNonNegativeIntegerValue(ValExpr, *this, OMPC_grainsize, 11825 /*StrictlyPositive=*/true)) 11826 return nullptr; 11827 11828 return new (Context) OMPGrainsizeClause(ValExpr, StartLoc, LParenLoc, EndLoc); 11829 } 11830 11831 OMPClause *Sema::ActOnOpenMPNumTasksClause(Expr *NumTasks, 11832 SourceLocation StartLoc, 11833 SourceLocation LParenLoc, 11834 SourceLocation EndLoc) { 11835 Expr *ValExpr = NumTasks; 11836 11837 // OpenMP [2.9.2, taskloop Constrcut] 11838 // The parameter of the num_tasks clause must be a positive integer 11839 // expression. 11840 if (!IsNonNegativeIntegerValue(ValExpr, *this, OMPC_num_tasks, 11841 /*StrictlyPositive=*/true)) 11842 return nullptr; 11843 11844 return new (Context) OMPNumTasksClause(ValExpr, StartLoc, LParenLoc, EndLoc); 11845 } 11846 11847 OMPClause *Sema::ActOnOpenMPHintClause(Expr *Hint, SourceLocation StartLoc, 11848 SourceLocation LParenLoc, 11849 SourceLocation EndLoc) { 11850 // OpenMP [2.13.2, critical construct, Description] 11851 // ... where hint-expression is an integer constant expression that evaluates 11852 // to a valid lock hint. 11853 ExprResult HintExpr = VerifyPositiveIntegerConstantInClause(Hint, OMPC_hint); 11854 if (HintExpr.isInvalid()) 11855 return nullptr; 11856 return new (Context) 11857 OMPHintClause(HintExpr.get(), StartLoc, LParenLoc, EndLoc); 11858 } 11859 11860 OMPClause *Sema::ActOnOpenMPDistScheduleClause( 11861 OpenMPDistScheduleClauseKind Kind, Expr *ChunkSize, SourceLocation StartLoc, 11862 SourceLocation LParenLoc, SourceLocation KindLoc, SourceLocation CommaLoc, 11863 SourceLocation EndLoc) { 11864 if (Kind == OMPC_DIST_SCHEDULE_unknown) { 11865 std::string Values; 11866 Values += "'"; 11867 Values += getOpenMPSimpleClauseTypeName(OMPC_dist_schedule, 0); 11868 Values += "'"; 11869 Diag(KindLoc, diag::err_omp_unexpected_clause_value) 11870 << Values << getOpenMPClauseName(OMPC_dist_schedule); 11871 return nullptr; 11872 } 11873 Expr *ValExpr = ChunkSize; 11874 Stmt *HelperValStmt = nullptr; 11875 if (ChunkSize) { 11876 if (!ChunkSize->isValueDependent() && !ChunkSize->isTypeDependent() && 11877 !ChunkSize->isInstantiationDependent() && 11878 !ChunkSize->containsUnexpandedParameterPack()) { 11879 SourceLocation ChunkSizeLoc = ChunkSize->getLocStart(); 11880 ExprResult Val = 11881 PerformOpenMPImplicitIntegerConversion(ChunkSizeLoc, ChunkSize); 11882 if (Val.isInvalid()) 11883 return nullptr; 11884 11885 ValExpr = Val.get(); 11886 11887 // OpenMP [2.7.1, Restrictions] 11888 // chunk_size must be a loop invariant integer expression with a positive 11889 // value. 11890 llvm::APSInt Result; 11891 if (ValExpr->isIntegerConstantExpr(Result, Context)) { 11892 if (Result.isSigned() && !Result.isStrictlyPositive()) { 11893 Diag(ChunkSizeLoc, diag::err_omp_negative_expression_in_clause) 11894 << "dist_schedule" << ChunkSize->getSourceRange(); 11895 return nullptr; 11896 } 11897 } else if (isParallelOrTaskRegion(DSAStack->getCurrentDirective()) && 11898 !CurContext->isDependentContext()) { 11899 llvm::MapVector<Expr *, DeclRefExpr *> Captures; 11900 ValExpr = tryBuildCapture(*this, ValExpr, Captures).get(); 11901 HelperValStmt = buildPreInits(Context, Captures); 11902 } 11903 } 11904 } 11905 11906 return new (Context) 11907 OMPDistScheduleClause(StartLoc, LParenLoc, KindLoc, CommaLoc, EndLoc, 11908 Kind, ValExpr, HelperValStmt); 11909 } 11910 11911 OMPClause *Sema::ActOnOpenMPDefaultmapClause( 11912 OpenMPDefaultmapClauseModifier M, OpenMPDefaultmapClauseKind Kind, 11913 SourceLocation StartLoc, SourceLocation LParenLoc, SourceLocation MLoc, 11914 SourceLocation KindLoc, SourceLocation EndLoc) { 11915 // OpenMP 4.5 only supports 'defaultmap(tofrom: scalar)' 11916 if (M != OMPC_DEFAULTMAP_MODIFIER_tofrom || Kind != OMPC_DEFAULTMAP_scalar) { 11917 std::string Value; 11918 SourceLocation Loc; 11919 Value += "'"; 11920 if (M != OMPC_DEFAULTMAP_MODIFIER_tofrom) { 11921 Value += getOpenMPSimpleClauseTypeName(OMPC_defaultmap, 11922 OMPC_DEFAULTMAP_MODIFIER_tofrom); 11923 Loc = MLoc; 11924 } else { 11925 Value += getOpenMPSimpleClauseTypeName(OMPC_defaultmap, 11926 OMPC_DEFAULTMAP_scalar); 11927 Loc = KindLoc; 11928 } 11929 Value += "'"; 11930 Diag(Loc, diag::err_omp_unexpected_clause_value) 11931 << Value << getOpenMPClauseName(OMPC_defaultmap); 11932 return nullptr; 11933 } 11934 DSAStack->setDefaultDMAToFromScalar(StartLoc); 11935 11936 return new (Context) 11937 OMPDefaultmapClause(StartLoc, LParenLoc, MLoc, KindLoc, EndLoc, Kind, M); 11938 } 11939 11940 bool Sema::ActOnStartOpenMPDeclareTargetDirective(SourceLocation Loc) { 11941 DeclContext *CurLexicalContext = getCurLexicalContext(); 11942 if (!CurLexicalContext->isFileContext() && 11943 !CurLexicalContext->isExternCContext() && 11944 !CurLexicalContext->isExternCXXContext() && 11945 !isa<CXXRecordDecl>(CurLexicalContext) && 11946 !isa<ClassTemplateDecl>(CurLexicalContext) && 11947 !isa<ClassTemplatePartialSpecializationDecl>(CurLexicalContext) && 11948 !isa<ClassTemplateSpecializationDecl>(CurLexicalContext)) { 11949 Diag(Loc, diag::err_omp_region_not_file_context); 11950 return false; 11951 } 11952 if (IsInOpenMPDeclareTargetContext) { 11953 Diag(Loc, diag::err_omp_enclosed_declare_target); 11954 return false; 11955 } 11956 11957 IsInOpenMPDeclareTargetContext = true; 11958 return true; 11959 } 11960 11961 void Sema::ActOnFinishOpenMPDeclareTargetDirective() { 11962 assert(IsInOpenMPDeclareTargetContext && 11963 "Unexpected ActOnFinishOpenMPDeclareTargetDirective"); 11964 11965 IsInOpenMPDeclareTargetContext = false; 11966 } 11967 11968 void Sema::ActOnOpenMPDeclareTargetName(Scope *CurScope, 11969 CXXScopeSpec &ScopeSpec, 11970 const DeclarationNameInfo &Id, 11971 OMPDeclareTargetDeclAttr::MapTypeTy MT, 11972 NamedDeclSetType &SameDirectiveDecls) { 11973 LookupResult Lookup(*this, Id, LookupOrdinaryName); 11974 LookupParsedName(Lookup, CurScope, &ScopeSpec, true); 11975 11976 if (Lookup.isAmbiguous()) 11977 return; 11978 Lookup.suppressDiagnostics(); 11979 11980 if (!Lookup.isSingleResult()) { 11981 if (TypoCorrection Corrected = 11982 CorrectTypo(Id, LookupOrdinaryName, CurScope, nullptr, 11983 llvm::make_unique<VarOrFuncDeclFilterCCC>(*this), 11984 CTK_ErrorRecovery)) { 11985 diagnoseTypo(Corrected, PDiag(diag::err_undeclared_var_use_suggest) 11986 << Id.getName()); 11987 checkDeclIsAllowedInOpenMPTarget(nullptr, Corrected.getCorrectionDecl()); 11988 return; 11989 } 11990 11991 Diag(Id.getLoc(), diag::err_undeclared_var_use) << Id.getName(); 11992 return; 11993 } 11994 11995 NamedDecl *ND = Lookup.getAsSingle<NamedDecl>(); 11996 if (isa<VarDecl>(ND) || isa<FunctionDecl>(ND)) { 11997 if (!SameDirectiveDecls.insert(cast<NamedDecl>(ND->getCanonicalDecl()))) 11998 Diag(Id.getLoc(), diag::err_omp_declare_target_multiple) << Id.getName(); 11999 12000 if (!ND->hasAttr<OMPDeclareTargetDeclAttr>()) { 12001 Attr *A = OMPDeclareTargetDeclAttr::CreateImplicit(Context, MT); 12002 ND->addAttr(A); 12003 if (ASTMutationListener *ML = Context.getASTMutationListener()) 12004 ML->DeclarationMarkedOpenMPDeclareTarget(ND, A); 12005 checkDeclIsAllowedInOpenMPTarget(nullptr, ND); 12006 } else if (ND->getAttr<OMPDeclareTargetDeclAttr>()->getMapType() != MT) { 12007 Diag(Id.getLoc(), diag::err_omp_declare_target_to_and_link) 12008 << Id.getName(); 12009 } 12010 } else 12011 Diag(Id.getLoc(), diag::err_omp_invalid_target_decl) << Id.getName(); 12012 } 12013 12014 static void checkDeclInTargetContext(SourceLocation SL, SourceRange SR, 12015 Sema &SemaRef, Decl *D) { 12016 if (!D) 12017 return; 12018 Decl *LD = nullptr; 12019 if (isa<TagDecl>(D)) { 12020 LD = cast<TagDecl>(D)->getDefinition(); 12021 } else if (isa<VarDecl>(D)) { 12022 LD = cast<VarDecl>(D)->getDefinition(); 12023 12024 // If this is an implicit variable that is legal and we do not need to do 12025 // anything. 12026 if (cast<VarDecl>(D)->isImplicit()) { 12027 Attr *A = OMPDeclareTargetDeclAttr::CreateImplicit( 12028 SemaRef.Context, OMPDeclareTargetDeclAttr::MT_To); 12029 D->addAttr(A); 12030 if (ASTMutationListener *ML = SemaRef.Context.getASTMutationListener()) 12031 ML->DeclarationMarkedOpenMPDeclareTarget(D, A); 12032 return; 12033 } 12034 12035 } else if (isa<FunctionDecl>(D)) { 12036 const FunctionDecl *FD = nullptr; 12037 if (cast<FunctionDecl>(D)->hasBody(FD)) 12038 LD = const_cast<FunctionDecl *>(FD); 12039 12040 // If the definition is associated with the current declaration in the 12041 // target region (it can be e.g. a lambda) that is legal and we do not need 12042 // to do anything else. 12043 if (LD == D) { 12044 Attr *A = OMPDeclareTargetDeclAttr::CreateImplicit( 12045 SemaRef.Context, OMPDeclareTargetDeclAttr::MT_To); 12046 D->addAttr(A); 12047 if (ASTMutationListener *ML = SemaRef.Context.getASTMutationListener()) 12048 ML->DeclarationMarkedOpenMPDeclareTarget(D, A); 12049 return; 12050 } 12051 } 12052 if (!LD) 12053 LD = D; 12054 if (LD && !LD->hasAttr<OMPDeclareTargetDeclAttr>() && 12055 (isa<VarDecl>(LD) || isa<FunctionDecl>(LD))) { 12056 // Outlined declaration is not declared target. 12057 if (LD->isOutOfLine()) { 12058 SemaRef.Diag(LD->getLocation(), diag::warn_omp_not_in_target_context); 12059 SemaRef.Diag(SL, diag::note_used_here) << SR; 12060 } else { 12061 DeclContext *DC = LD->getDeclContext(); 12062 while (DC) { 12063 if (isa<FunctionDecl>(DC) && 12064 cast<FunctionDecl>(DC)->hasAttr<OMPDeclareTargetDeclAttr>()) 12065 break; 12066 DC = DC->getParent(); 12067 } 12068 if (DC) 12069 return; 12070 12071 // Is not declared in target context. 12072 SemaRef.Diag(LD->getLocation(), diag::warn_omp_not_in_target_context); 12073 SemaRef.Diag(SL, diag::note_used_here) << SR; 12074 } 12075 // Mark decl as declared target to prevent further diagnostic. 12076 Attr *A = OMPDeclareTargetDeclAttr::CreateImplicit( 12077 SemaRef.Context, OMPDeclareTargetDeclAttr::MT_To); 12078 D->addAttr(A); 12079 if (ASTMutationListener *ML = SemaRef.Context.getASTMutationListener()) 12080 ML->DeclarationMarkedOpenMPDeclareTarget(D, A); 12081 } 12082 } 12083 12084 static bool checkValueDeclInTarget(SourceLocation SL, SourceRange SR, 12085 Sema &SemaRef, DSAStackTy *Stack, 12086 ValueDecl *VD) { 12087 if (VD->hasAttr<OMPDeclareTargetDeclAttr>()) 12088 return true; 12089 if (!CheckTypeMappable(SL, SR, SemaRef, Stack, VD->getType())) 12090 return false; 12091 return true; 12092 } 12093 12094 void Sema::checkDeclIsAllowedInOpenMPTarget(Expr *E, Decl *D) { 12095 if (!D || D->isInvalidDecl()) 12096 return; 12097 SourceRange SR = E ? E->getSourceRange() : D->getSourceRange(); 12098 SourceLocation SL = E ? E->getLocStart() : D->getLocation(); 12099 // 2.10.6: threadprivate variable cannot appear in a declare target directive. 12100 if (VarDecl *VD = dyn_cast<VarDecl>(D)) { 12101 if (DSAStack->isThreadPrivate(VD)) { 12102 Diag(SL, diag::err_omp_threadprivate_in_target); 12103 ReportOriginalDSA(*this, DSAStack, VD, DSAStack->getTopDSA(VD, false)); 12104 return; 12105 } 12106 } 12107 if (ValueDecl *VD = dyn_cast<ValueDecl>(D)) { 12108 // Problem if any with var declared with incomplete type will be reported 12109 // as normal, so no need to check it here. 12110 if ((E || !VD->getType()->isIncompleteType()) && 12111 !checkValueDeclInTarget(SL, SR, *this, DSAStack, VD)) { 12112 // Mark decl as declared target to prevent further diagnostic. 12113 if (isa<VarDecl>(VD) || isa<FunctionDecl>(VD)) { 12114 Attr *A = OMPDeclareTargetDeclAttr::CreateImplicit( 12115 Context, OMPDeclareTargetDeclAttr::MT_To); 12116 VD->addAttr(A); 12117 if (ASTMutationListener *ML = Context.getASTMutationListener()) 12118 ML->DeclarationMarkedOpenMPDeclareTarget(VD, A); 12119 } 12120 return; 12121 } 12122 } 12123 if (!E) { 12124 // Checking declaration inside declare target region. 12125 if (!D->hasAttr<OMPDeclareTargetDeclAttr>() && 12126 (isa<VarDecl>(D) || isa<FunctionDecl>(D))) { 12127 Attr *A = OMPDeclareTargetDeclAttr::CreateImplicit( 12128 Context, OMPDeclareTargetDeclAttr::MT_To); 12129 D->addAttr(A); 12130 if (ASTMutationListener *ML = Context.getASTMutationListener()) 12131 ML->DeclarationMarkedOpenMPDeclareTarget(D, A); 12132 } 12133 return; 12134 } 12135 checkDeclInTargetContext(E->getExprLoc(), E->getSourceRange(), *this, D); 12136 } 12137 12138 OMPClause *Sema::ActOnOpenMPToClause(ArrayRef<Expr *> VarList, 12139 SourceLocation StartLoc, 12140 SourceLocation LParenLoc, 12141 SourceLocation EndLoc) { 12142 MappableVarListInfo MVLI(VarList); 12143 checkMappableExpressionList(*this, DSAStack, OMPC_to, MVLI, StartLoc); 12144 if (MVLI.ProcessedVarList.empty()) 12145 return nullptr; 12146 12147 return OMPToClause::Create(Context, StartLoc, LParenLoc, EndLoc, 12148 MVLI.ProcessedVarList, MVLI.VarBaseDeclarations, 12149 MVLI.VarComponents); 12150 } 12151 12152 OMPClause *Sema::ActOnOpenMPFromClause(ArrayRef<Expr *> VarList, 12153 SourceLocation StartLoc, 12154 SourceLocation LParenLoc, 12155 SourceLocation EndLoc) { 12156 MappableVarListInfo MVLI(VarList); 12157 checkMappableExpressionList(*this, DSAStack, OMPC_from, MVLI, StartLoc); 12158 if (MVLI.ProcessedVarList.empty()) 12159 return nullptr; 12160 12161 return OMPFromClause::Create(Context, StartLoc, LParenLoc, EndLoc, 12162 MVLI.ProcessedVarList, MVLI.VarBaseDeclarations, 12163 MVLI.VarComponents); 12164 } 12165 12166 OMPClause *Sema::ActOnOpenMPUseDevicePtrClause(ArrayRef<Expr *> VarList, 12167 SourceLocation StartLoc, 12168 SourceLocation LParenLoc, 12169 SourceLocation EndLoc) { 12170 MappableVarListInfo MVLI(VarList); 12171 SmallVector<Expr *, 8> PrivateCopies; 12172 SmallVector<Expr *, 8> Inits; 12173 12174 for (auto &RefExpr : VarList) { 12175 assert(RefExpr && "NULL expr in OpenMP use_device_ptr clause."); 12176 SourceLocation ELoc; 12177 SourceRange ERange; 12178 Expr *SimpleRefExpr = RefExpr; 12179 auto Res = getPrivateItem(*this, SimpleRefExpr, ELoc, ERange); 12180 if (Res.second) { 12181 // It will be analyzed later. 12182 MVLI.ProcessedVarList.push_back(RefExpr); 12183 PrivateCopies.push_back(nullptr); 12184 Inits.push_back(nullptr); 12185 } 12186 ValueDecl *D = Res.first; 12187 if (!D) 12188 continue; 12189 12190 QualType Type = D->getType(); 12191 Type = Type.getNonReferenceType().getUnqualifiedType(); 12192 12193 auto *VD = dyn_cast<VarDecl>(D); 12194 12195 // Item should be a pointer or reference to pointer. 12196 if (!Type->isPointerType()) { 12197 Diag(ELoc, diag::err_omp_usedeviceptr_not_a_pointer) 12198 << 0 << RefExpr->getSourceRange(); 12199 continue; 12200 } 12201 12202 // Build the private variable and the expression that refers to it. 12203 auto VDPrivate = buildVarDecl(*this, ELoc, Type, D->getName(), 12204 D->hasAttrs() ? &D->getAttrs() : nullptr); 12205 if (VDPrivate->isInvalidDecl()) 12206 continue; 12207 12208 CurContext->addDecl(VDPrivate); 12209 auto VDPrivateRefExpr = buildDeclRefExpr( 12210 *this, VDPrivate, RefExpr->getType().getUnqualifiedType(), ELoc); 12211 12212 // Add temporary variable to initialize the private copy of the pointer. 12213 auto *VDInit = 12214 buildVarDecl(*this, RefExpr->getExprLoc(), Type, ".devptr.temp"); 12215 auto *VDInitRefExpr = buildDeclRefExpr(*this, VDInit, RefExpr->getType(), 12216 RefExpr->getExprLoc()); 12217 AddInitializerToDecl(VDPrivate, 12218 DefaultLvalueConversion(VDInitRefExpr).get(), 12219 /*DirectInit=*/false); 12220 12221 // If required, build a capture to implement the privatization initialized 12222 // with the current list item value. 12223 DeclRefExpr *Ref = nullptr; 12224 if (!VD) 12225 Ref = buildCapture(*this, D, SimpleRefExpr, /*WithInit=*/true); 12226 MVLI.ProcessedVarList.push_back(VD ? RefExpr->IgnoreParens() : Ref); 12227 PrivateCopies.push_back(VDPrivateRefExpr); 12228 Inits.push_back(VDInitRefExpr); 12229 12230 // We need to add a data sharing attribute for this variable to make sure it 12231 // is correctly captured. A variable that shows up in a use_device_ptr has 12232 // similar properties of a first private variable. 12233 DSAStack->addDSA(D, RefExpr->IgnoreParens(), OMPC_firstprivate, Ref); 12234 12235 // Create a mappable component for the list item. List items in this clause 12236 // only need a component. 12237 MVLI.VarBaseDeclarations.push_back(D); 12238 MVLI.VarComponents.resize(MVLI.VarComponents.size() + 1); 12239 MVLI.VarComponents.back().push_back( 12240 OMPClauseMappableExprCommon::MappableComponent(SimpleRefExpr, D)); 12241 } 12242 12243 if (MVLI.ProcessedVarList.empty()) 12244 return nullptr; 12245 12246 return OMPUseDevicePtrClause::Create( 12247 Context, StartLoc, LParenLoc, EndLoc, MVLI.ProcessedVarList, 12248 PrivateCopies, Inits, MVLI.VarBaseDeclarations, MVLI.VarComponents); 12249 } 12250 12251 OMPClause *Sema::ActOnOpenMPIsDevicePtrClause(ArrayRef<Expr *> VarList, 12252 SourceLocation StartLoc, 12253 SourceLocation LParenLoc, 12254 SourceLocation EndLoc) { 12255 MappableVarListInfo MVLI(VarList); 12256 for (auto &RefExpr : VarList) { 12257 assert(RefExpr && "NULL expr in OpenMP is_device_ptr clause."); 12258 SourceLocation ELoc; 12259 SourceRange ERange; 12260 Expr *SimpleRefExpr = RefExpr; 12261 auto Res = getPrivateItem(*this, SimpleRefExpr, ELoc, ERange); 12262 if (Res.second) { 12263 // It will be analyzed later. 12264 MVLI.ProcessedVarList.push_back(RefExpr); 12265 } 12266 ValueDecl *D = Res.first; 12267 if (!D) 12268 continue; 12269 12270 QualType Type = D->getType(); 12271 // item should be a pointer or array or reference to pointer or array 12272 if (!Type.getNonReferenceType()->isPointerType() && 12273 !Type.getNonReferenceType()->isArrayType()) { 12274 Diag(ELoc, diag::err_omp_argument_type_isdeviceptr) 12275 << 0 << RefExpr->getSourceRange(); 12276 continue; 12277 } 12278 12279 // Check if the declaration in the clause does not show up in any data 12280 // sharing attribute. 12281 auto DVar = DSAStack->getTopDSA(D, false); 12282 if (isOpenMPPrivate(DVar.CKind)) { 12283 Diag(ELoc, diag::err_omp_variable_in_given_clause_and_dsa) 12284 << getOpenMPClauseName(DVar.CKind) 12285 << getOpenMPClauseName(OMPC_is_device_ptr) 12286 << getOpenMPDirectiveName(DSAStack->getCurrentDirective()); 12287 ReportOriginalDSA(*this, DSAStack, D, DVar); 12288 continue; 12289 } 12290 12291 Expr *ConflictExpr; 12292 if (DSAStack->checkMappableExprComponentListsForDecl( 12293 D, /*CurrentRegionOnly=*/true, 12294 [&ConflictExpr]( 12295 OMPClauseMappableExprCommon::MappableExprComponentListRef R, 12296 OpenMPClauseKind) -> bool { 12297 ConflictExpr = R.front().getAssociatedExpression(); 12298 return true; 12299 })) { 12300 Diag(ELoc, diag::err_omp_map_shared_storage) << RefExpr->getSourceRange(); 12301 Diag(ConflictExpr->getExprLoc(), diag::note_used_here) 12302 << ConflictExpr->getSourceRange(); 12303 continue; 12304 } 12305 12306 // Store the components in the stack so that they can be used to check 12307 // against other clauses later on. 12308 OMPClauseMappableExprCommon::MappableComponent MC(SimpleRefExpr, D); 12309 DSAStack->addMappableExpressionComponents( 12310 D, MC, /*WhereFoundClauseKind=*/OMPC_is_device_ptr); 12311 12312 // Record the expression we've just processed. 12313 MVLI.ProcessedVarList.push_back(SimpleRefExpr); 12314 12315 // Create a mappable component for the list item. List items in this clause 12316 // only need a component. We use a null declaration to signal fields in 12317 // 'this'. 12318 assert((isa<DeclRefExpr>(SimpleRefExpr) || 12319 isa<CXXThisExpr>(cast<MemberExpr>(SimpleRefExpr)->getBase())) && 12320 "Unexpected device pointer expression!"); 12321 MVLI.VarBaseDeclarations.push_back( 12322 isa<DeclRefExpr>(SimpleRefExpr) ? D : nullptr); 12323 MVLI.VarComponents.resize(MVLI.VarComponents.size() + 1); 12324 MVLI.VarComponents.back().push_back(MC); 12325 } 12326 12327 if (MVLI.ProcessedVarList.empty()) 12328 return nullptr; 12329 12330 return OMPIsDevicePtrClause::Create( 12331 Context, StartLoc, LParenLoc, EndLoc, MVLI.ProcessedVarList, 12332 MVLI.VarBaseDeclarations, MVLI.VarComponents); 12333 } 12334