[Openmp-commits] [clang] [flang] [llvm] [openmp] [Clang][OpenMP][LoopTransformations] Add support for "#pragma omp fuse" loop transformation directive and "looprange" clause (PR #139293)

[Michael Kruse via Openmp-commits]

================
@@ -508,6 +512,43 @@ OMPInterchangeDirective::CreateEmpty(const ASTContext &C, unsigned NumClauses,
       SourceLocation(), SourceLocation(), NumLoops);
 }
 
+OMPFuseDirective *OMPFuseDirective::Create(
+    const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
+    ArrayRef<OMPClause *> Clauses, unsigned NumLoops, unsigned NumLoopNests,
+    Stmt *AssociatedStmt, Stmt *TransformedStmt, Stmt *PreInits) {
+
+  OMPFuseDirective *Dir = createDirective<OMPFuseDirective>(
+      C, Clauses, AssociatedStmt, TransformedStmtOffset + 1, StartLoc, EndLoc,
+      NumLoops);
+  Dir->setTransformedStmt(TransformedStmt);
+  Dir->setPreInits(PreInits);
+  // The number of top level canonical nests could
+  // not match the total number of generated loops
+  // Example:
+  // Before fusion:
+  //   for (int i = 0; i < N; ++i)
+  //     for (int j = 0; j < M; ++j)
+  //       A[i][j] = i + j;
+  //
+  //   for (int k = 0; k < P; ++k)
+  //     B[k] = k * 2;
+  // Here, NumLoopNests = 2, but NumLoops = 3.
----------------
Meinersbur wrote:

I don't think the NumLoops as in the total number loops is useful. It is either a loop sequence, then you have the number of loop nests; a single loop nest, then you have the number of nested loops; or you do an analysis of the entire loop nest structure, then you need a graph as in `LoopSequenceAnalysis`.

https://github.com/llvm/llvm-project/pull/139293