[Mlir-commits] [mlir] 609c0e1 - [mlir] Register linalg external TilingInterface models in InitAllDialects

[Christopher Bate]

Author: Christopher Bate
Date: 2022-07-11T15:54:37-06:00
New Revision: 609c0e1b9b1b9768d91cfed9b444c308b8836ef9

URL: https://github.com/llvm/llvm-project/commit/609c0e1b9b1b9768d91cfed9b444c308b8836ef9
DIFF: https://github.com/llvm/llvm-project/commit/609c0e1b9b1b9768d91cfed9b444c308b8836ef9.diff

LOG: [mlir] Register linalg external TilingInterface models in InitAllDialects

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

Added: 
    

Modified: 
    mlir/include/mlir/InitAllDialects.h

Removed: 
    


################################################################################
diff  --git a/mlir/include/mlir/InitAllDialects.h b/mlir/include/mlir/InitAllDialects.h
index a3c39ae1bf62..0eceaa18312c 100644
--- a/mlir/include/mlir/InitAllDialects.h
+++ b/mlir/include/mlir/InitAllDialects.h
@@ -37,6 +37,7 @@
 #include "mlir/Dialect/Linalg/IR/Linalg.h"
 #include "mlir/Dialect/Linalg/TransformOps/LinalgTransformOps.h"
 #include "mlir/Dialect/Linalg/Transforms/BufferizableOpInterfaceImpl.h"
+#include "mlir/Dialect/Linalg/Transforms/TilingInterfaceImpl.h"
 #include "mlir/Dialect/MLProgram/IR/MLProgram.h"
 #include "mlir/Dialect/Math/IR/Math.h"
 #include "mlir/Dialect/MemRef/IR/MemRef.h"
@@ -118,6 +119,7 @@ inline void registerAllDialects(DialectRegistry &registry) {
   bufferization::func_ext::registerBufferizableOpInterfaceExternalModels(
       registry);
   linalg::registerBufferizableOpInterfaceExternalModels(registry);
+  linalg::registerTilingInterfaceExternalModels(registry);
   scf::registerBufferizableOpInterfaceExternalModels(registry);
   shape::registerBufferizableOpInterfaceExternalModels(registry);
   sparse_tensor::registerBufferizableOpInterfaceExternalModels(registry);