forked from OSchip/llvm-project
[mlir][Linalg] NFC - Cleanup Linalg Pass locations and namespacing
Summary: This diff moves the conversion pass declaration closer to its definition and makes the namespacing of passes consistent with the rest of the infrastructure (i.e. `mlir::linalg::createXXXPass` -> `mlir::createXXXPass`). Reviewers: ftynse, jpienaar, mehdi_amini Subscribers: rriddle, burmako, shauheen, antiagainst, arpith-jacob, mgester, lucyrfox, aartbik, liufengdb, llvm-commits Tags: #llvm Differential Revision: https://reviews.llvm.org/D72766
This commit is contained in:
parent
e9e26c01cd
commit
7741de9435
|
@ -13,6 +13,8 @@
|
|||
|
||||
namespace mlir {
|
||||
class MLIRContext;
|
||||
class ModuleOp;
|
||||
template <typename T> class OpPassBase;
|
||||
|
||||
class LinalgTypeConverter : public LLVMTypeConverter {
|
||||
public:
|
||||
|
@ -25,6 +27,9 @@ void populateLinalgToLLVMConversionPatterns(LinalgTypeConverter &converter,
|
|||
OwningRewritePatternList &patterns,
|
||||
MLIRContext *ctx);
|
||||
|
||||
/// Create a pass to convert Linalg operations to the LLVMIR dialect.
|
||||
std::unique_ptr<OpPassBase<ModuleOp>> createConvertLinalgToLLVMPass();
|
||||
|
||||
} // namespace mlir
|
||||
|
||||
#endif // MLIR_CONVERSION_LINALGTOLLVM_LINALGTOLLVM_H_
|
||||
|
|
|
@ -21,7 +21,6 @@ class FuncOp;
|
|||
class ModuleOp;
|
||||
template <typename T> class OpPassBase;
|
||||
|
||||
namespace linalg {
|
||||
std::unique_ptr<OpPassBase<FuncOp>> createLinalgFusionPass();
|
||||
|
||||
std::unique_ptr<OpPassBase<FuncOp>>
|
||||
|
@ -39,10 +38,6 @@ std::unique_ptr<OpPassBase<FuncOp>> createConvertLinalgToLoopsPass();
|
|||
/// Placeholder for now, this is NYI.
|
||||
std::unique_ptr<OpPassBase<FuncOp>> createConvertLinalgToAffineLoopsPass();
|
||||
|
||||
/// Create a pass to convert Linalg operations to the LLVMIR dialect.
|
||||
std::unique_ptr<OpPassBase<ModuleOp>> createConvertLinalgToLLVMPass();
|
||||
|
||||
} // namespace linalg
|
||||
} // namespace mlir
|
||||
|
||||
#endif // MLIR_DIALECT_LINALG_PASSES_H_
|
||||
|
|
|
@ -594,8 +594,7 @@ void ConvertLinalgToLLVMPass::runOnModule() {
|
|||
signalPassFailure();
|
||||
}
|
||||
|
||||
std::unique_ptr<OpPassBase<ModuleOp>>
|
||||
mlir::linalg::createConvertLinalgToLLVMPass() {
|
||||
std::unique_ptr<OpPassBase<ModuleOp>> mlir::createConvertLinalgToLLVMPass() {
|
||||
return std::make_unique<ConvertLinalgToLLVMPass>();
|
||||
}
|
||||
|
||||
|
|
|
@ -360,7 +360,7 @@ struct LinalgFusionPass : public FunctionPass<LinalgFusionPass> {
|
|||
};
|
||||
} // namespace
|
||||
|
||||
std::unique_ptr<OpPassBase<FuncOp>> mlir::linalg::createLinalgFusionPass() {
|
||||
std::unique_ptr<OpPassBase<FuncOp>> mlir::createLinalgFusionPass() {
|
||||
return std::make_unique<LinalgFusionPass>();
|
||||
}
|
||||
|
||||
|
|
|
@ -562,8 +562,7 @@ void LowerLinalgToLoopsPass<LoopType, IndexedValueType>::runOnFunction() {
|
|||
|
||||
/// Create a pass to convert Linalg operations to loop.for loops and
|
||||
/// std.load/std.store accesses.
|
||||
std::unique_ptr<OpPassBase<FuncOp>>
|
||||
mlir::linalg::createConvertLinalgToLoopsPass() {
|
||||
std::unique_ptr<OpPassBase<FuncOp>> mlir::createConvertLinalgToLoopsPass() {
|
||||
return std::make_unique<
|
||||
LowerLinalgToLoopsPass<loop::ForOp, IndexedStdValue>>();
|
||||
}
|
||||
|
@ -572,7 +571,7 @@ mlir::linalg::createConvertLinalgToLoopsPass() {
|
|||
/// affine_load/affine_store accesses.
|
||||
/// Placeholder for now, this is NYI.
|
||||
std::unique_ptr<OpPassBase<FuncOp>>
|
||||
mlir::linalg::createConvertLinalgToAffineLoopsPass() {
|
||||
mlir::createConvertLinalgToAffineLoopsPass() {
|
||||
return std::make_unique<
|
||||
LowerLinalgToLoopsPass<AffineForOp, IndexedAffineValue>>();
|
||||
}
|
||||
|
|
|
@ -238,7 +238,7 @@ struct LinalgPromotionPass : public FunctionPass<LinalgPromotionPass> {
|
|||
} // namespace
|
||||
|
||||
std::unique_ptr<OpPassBase<FuncOp>>
|
||||
mlir::linalg::createLinalgPromotionPass(bool dynamicBuffers) {
|
||||
mlir::createLinalgPromotionPass(bool dynamicBuffers) {
|
||||
return std::make_unique<LinalgPromotionPass>(dynamicBuffers);
|
||||
}
|
||||
|
||||
|
|
|
@ -457,7 +457,7 @@ LinalgTilingPass::LinalgTilingPass(ArrayRef<int64_t> sizes) {
|
|||
}
|
||||
|
||||
std::unique_ptr<OpPassBase<FuncOp>>
|
||||
mlir::linalg::createLinalgTilingPass(ArrayRef<int64_t> tileSizes) {
|
||||
mlir::createLinalgTilingPass(ArrayRef<int64_t> tileSizes) {
|
||||
return std::make_unique<LinalgTilingPass>(tileSizes);
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in New Issue