[mlir][SCF][transform] Register SCF dialect patterns

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

diff --git a/mlir/lib/Dialect/SCF/TransformOps/SCFTransformOps.cpp b/mlir/lib/Dialect/SCF/TransformOps/SCFTransformOps.cpp
index 2da7473..4c64c54 100644 (file)
--- a/mlir/lib/Dialect/SCF/TransformOps/SCFTransformOps.cpp
+++ b/mlir/lib/Dialect/SCF/TransformOps/SCFTransformOps.cpp
@@ -309,6 +309,13 @@ public:
 #define GET_OP_LIST
 #include "mlir/Dialect/SCF/TransformOps/SCFTransformOps.cpp.inc"
         >();
+
+    addDialectDataInitializer<transform::PatternRegistry>(
+        [&](transform::PatternRegistry &registry) {
+          registry.registerPatterns(
+              "scf.for_loop_canonicalization",
+              scf::populateSCFForLoopCanonicalizationPatterns);
+        });
   }
 };
 } // namespace