diff options
Diffstat (limited to 'mlir/examples/Linalg/Linalg3')
-rw-r--r-- | mlir/examples/Linalg/Linalg3/include/linalg3/Transforms.h | 4 | ||||
-rw-r--r-- | mlir/examples/Linalg/Linalg3/lib/Transforms.cpp | 2 |
2 files changed, 3 insertions, 3 deletions
diff --git a/mlir/examples/Linalg/Linalg3/include/linalg3/Transforms.h b/mlir/examples/Linalg/Linalg3/include/linalg3/Transforms.h index 849d65a6b6f..5381734721c 100644 --- a/mlir/examples/Linalg/Linalg3/include/linalg3/Transforms.h +++ b/mlir/examples/Linalg/Linalg3/include/linalg3/Transforms.h @@ -31,7 +31,6 @@ class Operation; class Value; template <typename T> class OpPassBase; -using FunctionPassBase = OpPassBase<FuncOp>; } // namespace mlir namespace linalg { @@ -75,7 +74,8 @@ void lowerToLoops(mlir::FuncOp f); /// Creates a pass that rewrites linalg.load and linalg.store to affine.load and /// affine.store operations. -std::unique_ptr<mlir::FunctionPassBase> createLowerLinalgLoadStorePass(); +std::unique_ptr<mlir::OpPassBase<mlir::FuncOp>> +createLowerLinalgLoadStorePass(); } // namespace linalg diff --git a/mlir/examples/Linalg/Linalg3/lib/Transforms.cpp b/mlir/examples/Linalg/Linalg3/lib/Transforms.cpp index ce2656520fa..184f1da528f 100644 --- a/mlir/examples/Linalg/Linalg3/lib/Transforms.cpp +++ b/mlir/examples/Linalg/Linalg3/lib/Transforms.cpp @@ -300,6 +300,6 @@ Rewriter<linalg::StoreOp>::matchAndRewrite(linalg::StoreOp store, } } // namespace -std::unique_ptr<FunctionPassBase> linalg::createLowerLinalgLoadStorePass() { +std::unique_ptr<OpPassBase<FuncOp>> linalg::createLowerLinalgLoadStorePass() { return std::make_unique<LowerLinalgLoadStorePass>(); } |