[Mlir-commits] [mlir] 584f401 - [mlir][linalg][transform] Register linalg dialect patterns

[Matthias Springer]

Author: Matthias Springer
Date: 2023-06-05T11:36:59+02:00
New Revision: 584f401730da87cf54b536320439d0875ddb3d63

URL: https://github.com/llvm/llvm-project/commit/584f401730da87cf54b536320439d0875ddb3d63
DIFF: https://github.com/llvm/llvm-project/commit/584f401730da87cf54b536320439d0875ddb3d63.diff

LOG: [mlir][linalg][transform] Register linalg dialect patterns

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

Added: 
    

Modified: 
    mlir/lib/Dialect/Linalg/TransformOps/DialectExtension.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/lib/Dialect/Linalg/TransformOps/DialectExtension.cpp b/mlir/lib/Dialect/Linalg/TransformOps/DialectExtension.cpp
index 9578648d6363e..ab28ad32e350d 100644
--- a/mlir/lib/Dialect/Linalg/TransformOps/DialectExtension.cpp
+++ b/mlir/lib/Dialect/Linalg/TransformOps/DialectExtension.cpp
@@ -13,9 +13,11 @@
 #include "mlir/Dialect/Linalg/IR/Linalg.h"
 #include "mlir/Dialect/Linalg/TransformOps/LinalgMatchOps.h"
 #include "mlir/Dialect/Linalg/TransformOps/LinalgTransformOps.h"
+#include "mlir/Dialect/Linalg/Transforms/Transforms.h"
 #include "mlir/Dialect/SCF/IR/SCF.h"
 #include "mlir/Dialect/Tensor/IR/Tensor.h"
 #include "mlir/Dialect/Transform/IR/TransformDialect.h"
+#include "mlir/Dialect/Transform/IR/TransformOps.h"
 #include "mlir/Dialect/Vector/IR/VectorOps.h"
 
 using namespace mlir;
@@ -47,6 +49,22 @@ class LinalgTransformDialectExtension
 #define GET_OP_LIST
 #include "mlir/Dialect/Linalg/TransformOps/LinalgMatchOps.cpp.inc"
         >();
+
+    addDialectDataInitializer<transform::PatternRegistry>(
+        [&](transform::PatternRegistry &registry) {
+          registry.registerPatterns(
+              "linalg.erase_unnecessary_inputs",
+              linalg::populateEraseUnnecessaryInputsPatterns);
+          registry.registerPatterns(
+              "linalg.fold_unit_extent_dims_via_slices",
+              linalg::populateFoldUnitExtentDimsViaSlicesPatterns);
+          registry.registerPatterns(
+              "linalg.fold_unit_extent_dims_via_reshapes",
+              linalg::populateFoldUnitExtentDimsViaReshapesPatterns);
+          registry.registerPatterns(
+              "linalg.tiling_canonicalization",
+              linalg::populateLinalgTilingCanonicalizationPatterns);
+        });
   }
 };
 } // namespace