================
@@ -15783,6 +15828,100 @@ static Expr *makeFloorIVRef(Sema &SemaRef,
ArrayRef<VarDecl *> FloorIndVars,
OrigCntVar->getExprLoc());
}
+/// Whether \p Counter is compiler generated (e.g. a fuse index).
+static bool isGeneratedLoopCounter(const Expr *Counter) {
+ const auto *DRE = dyn_cast<DeclRefExpr>(Counter);
+ const auto *VD = DRE ? dyn_cast<VarDecl>(DRE->getDecl()) : nullptr;
+ return VD && VD->isImplicit() && !isa<OMPCapturedExprDecl>(VD);
+}
+
+/// Wrap \p S so it runs only if every loop in \p Loops has iterations.
+static Stmt *
+guardByLoopPreConds(ASTContext &Context,
+ ArrayRef<OMPLoopBasedDirective::HelperExprs> Loops,
+ Stmt *S) {
+ for (const auto &Loop : llvm::reverse(Loops))
+ S = IfStmt::Create(Context, SourceLocation(), IfStatementKind::Ordinary,
+ nullptr, nullptr, Loop.PreCond, SourceLocation(),
+ SourceLocation(), S, SourceLocation(), nullptr);
+ return S;
+}
+
+/// Run loop \p I's inits only if all enclosing loops have iterations.
+/// Declarations are left as is.
+static void
+guardNestedLoopInit(ASTContext &Context,
+ ArrayRef<OMPLoopBasedDirective::HelperExprs> LoopHelpers,
+ unsigned I, SmallVectorImpl<Stmt *> &OriginalInit) {
+ if (I == 0)
+ return;
+ for (Stmt *&Init : OriginalInit)
+ if (Init && !isa<DeclStmt>(Init))
+ Init = guardByLoopPreConds(Context, LoopHelpers.take_front(I), Init);
+}
+
+/// Build loop variable finalization statement from HelperExprs.Finals.
+/// Each Final runs only if its loop, and with \p IsLoopNest all enclosing
+/// loops, have iterations; \p NestedFinals[I] is emitted with loop I's.
+/// Returns nullptr if there is nothing to finalize.
+/// Compiler-generated counters and class-type iterators are skipped.
+static Stmt *
+buildLoopFinalization(ASTContext &Context,
+ ArrayRef<OMPLoopBasedDirective::HelperExprs> LoopHelpers,
+ bool IsLoopNest = false,
----------------
alexey-bataev wrote:
```suggestion
```
https://github.com/llvm/llvm-project/pull/214717
_______________________________________________
cfe-commits mailing list
[email protected]
https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
