summaryrefslogtreecommitdiffstats
path: root/mlir/examples/Linalg/Linalg3
diff options
context:
space:
mode:
Diffstat (limited to 'mlir/examples/Linalg/Linalg3')
-rw-r--r--mlir/examples/Linalg/Linalg3/include/linalg3/Transforms.h4
-rw-r--r--mlir/examples/Linalg/Linalg3/lib/Transforms.cpp2
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>();
}
OpenPOWER on IntegriCloud