5e2afe5c66
Implementation of Pass and Dialect Plugins that mirrors LLVM Pass Plugin implementation from the new pass manager. Currently the implementation only supports using the pass-pipeline option for adding passes. This restriction is imposed by the `PassPipelineCLParser` variable in mlir/lib/Tools/mlir-opt/MlirOptMain.cpp:114 that loads the parse options statically before parsing the cmd line args. ``` mlir-opt stanalone-plugin.mlir --load-dialect-plugin=lib/libStandalonePlugin.so --pass-pipeline="builtin.module(standalone-switch-bar-foo)" ``` Reviewed By: rriddle, mehdi_amini Differential Revision: https://reviews.llvm.org/D147053
40 lines
1.5 KiB
C++
40 lines
1.5 KiB
C++
//===- standalone-plugin.cpp ------------------------------------*- C++ -*-===//
|
|
//
|
|
// This file is licensed under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/Dialect/Arith/IR/Arith.h"
|
|
#include "mlir/IR/Dialect.h"
|
|
#include "mlir/IR/MLIRContext.h"
|
|
#include "mlir/InitAllDialects.h"
|
|
#include "mlir/InitAllPasses.h"
|
|
#include "mlir/Pass/Pass.h"
|
|
#include "mlir/Tools/Plugins/DialectPlugin.h"
|
|
|
|
#include "Standalone/StandaloneDialect.h"
|
|
#include "Standalone/StandalonePasses.h"
|
|
|
|
using namespace mlir;
|
|
|
|
/// Dialect plugin registration mechanism.
|
|
/// Observe that it also allows to register passes.
|
|
/// Necessary symbol to register the dialect plugin.
|
|
extern "C" LLVM_ATTRIBUTE_WEAK DialectPluginLibraryInfo
|
|
mlirGetDialectPluginInfo() {
|
|
return {MLIR_PLUGIN_API_VERSION, "Standalone", LLVM_VERSION_STRING,
|
|
[](DialectRegistry *registry) {
|
|
registry->insert<mlir::standalone::StandaloneDialect>();
|
|
mlir::standalone::registerPasses();
|
|
}};
|
|
}
|
|
|
|
/// Pass plugin registration mechanism.
|
|
/// Necessary symbol to register the pass plugin.
|
|
extern "C" LLVM_ATTRIBUTE_WEAK PassPluginLibraryInfo mlirGetPassPluginInfo() {
|
|
return {MLIR_PLUGIN_API_VERSION, "StandalonePasses", LLVM_VERSION_STRING,
|
|
[]() { mlir::standalone::registerPasses(); }};
|
|
}
|