[mlir] NFC: Expose tiled_loop->scf pattern.

Differential Revision: https://reviews.llvm.org/D102921

mlir/include/mlir/Dialect/Linalg/Transforms/Transforms.h

@@ -81,6 +81,9 @@ void populateFoldUnitExtentDimsPatterns(RewritePatternSet &patterns);
 /// Patterns that are used to inline constant operands into linalg generic ops.
 void populateInlineConstantOperandsPatterns(RewritePatternSet &patterns);
 
+/// Pattern to convert TiledLoopOp to SCF loops.
+void populateTiledLoopToSCFPattern(RewritePatternSet &patterns);
+
 /// Options that control fusion of elementwise operations.
 struct LinalgElementwiseFusionOptions {
   /// Enable fusion of reshapes into the shape with elementwise operations. By

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

@@ -597,12 +597,16 @@ struct LowerTiledLoopsToSCF
   void runOnFunction() override {
     MLIRContext *context = &getContext();
     RewritePatternSet patterns(context);
-    patterns.add<TiledLoopToSCFPattern>(context);
+    populateTiledLoopToSCFPattern(patterns);
     (void)applyPatternsAndFoldGreedily(getFunction(), std::move(patterns));
   }
 };
 } // namespace
 
+void mlir::linalg::populateTiledLoopToSCFPattern(RewritePatternSet &patterns) {
+  patterns.add<TiledLoopToSCFPattern>(patterns.getContext());
+}
+
 std::unique_ptr<OperationPass<FuncOp>>
 mlir::createConvertLinalgTiledLoopsToSCFPass() {
   return std::make_unique<LowerTiledLoopsToSCF>();