[MLIR][Linalg] Avoid forward declaration in `Loops.cpp`

[frgossen]

Summary

Depends On D101768

[frgossen]

Rebase

mlir/lib/Dialect/Linalg/Transforms/Loops.cpp

scf::buildLoopNest(rewriter, loc, tiledLoop.lowerBound(),
                                               newBlockArgs);
                          rewriter.eraseOp(newBody->getTerminator());
                        });
     rewriter.eraseOp(tiledLoop);
     return success();
   }
 };
 
-struct FoldAffineOp;
-} // namespace
-
-template <typename LoopType>
-static void lowerLinalgToLoopsImpl(FuncOp funcOp) {
-  MLIRContext *context = funcOp.getContext();
-  RewritePatternSet patterns(context);
-  patterns.add<LinalgRewritePattern<LoopType>>(context);
-  memref::DimOp::getCanonicalizationPatterns(patterns, context);
-  AffineApplyOp::getCanonicalizationPatterns(patterns, context);
-  patterns.add<FoldAffineOp>(context);
-  // Just apply the patterns greedily.
-  (void)applyPatternsAndFoldGreedily(funcOp, std::move(patterns));
-}
-
-namespace {
 /// Local folding pattern for AffineApplyOp that we can apply greedily.
 /// This replaces AffineApplyOp by the proper value in cases where the
 /// associated map is trivial.
 /// A trivial map here is defined as a map with a single result and either:
 ///   1. Zero operand + returns a single AffineConstantExpr
 ///   2. One operand + returns a single AffineDimExpr
 ///   3. One operand + returns a single AffineSymbolExpr
 //
LogicalResult matchAndRewrite(Operation *op,
     if (expr.dyn_cast<AffineDimExpr>() || expr.dyn_cast<AffineSymbolExpr>()) {
       rewriter.replaceOp(op, op->getOperand(0));
       return success();
     }
     return failure();
   }
 };
 
+template <typename LoopType>
+static void lowerLinalgToLoopsImpl(FuncOp funcOp) {
+  MLIRContext *context = funcOp.getContext();
+  RewritePatternSet patterns(context);
+  patterns.add<LinalgRewritePattern<LoopType>>(context);
+  memref::DimOp::getCanonicalizationPatterns(patterns, context);
+  AffineApplyOp::getCanonicalizationPatterns(patterns, context);
+  patterns.add<FoldAffineOp>(context);
+  // Just apply the patterns greedily.
+  (void)applyPatternsAndFoldGreedily(funcOp, std::move(patterns));
+}
+
 struct LowerToAffineLoops
     : public LinalgLowerToAffineLoopsBase<LowerToAffineLoops> {
   void getDependentDialects(DialectRegistry &registry) const override {
     registry.insert<memref::MemRefDialect>();
   }
   void runOnFunction() override {
     lowerLinalgToLoopsImpl<AffineForOp>(getFunction());
   }