mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Ivan Krysak
60 lines
1.6 KiB
C++
60 lines
1.6 KiB
C++
#pragma once
|
|
|
|
//
|
|
// This file is distributed under the MIT License. See LICENSE.md for details.
|
|
//
|
|
|
|
#include "mlir/Pass/PassRegistry.h"
|
|
|
|
#include "revng/PipeboxCommon/CliftContainers.h"
|
|
#include "revng/PipeboxCommon/Common.h"
|
|
#include "revng/PipeboxCommon/Model.h"
|
|
|
|
namespace revng::pypeline::pipes {
|
|
|
|
/// Pipe that allows running non-model-aware MLIR passes on a
|
|
/// CliftFunctionContainer. Just like the PureLLVMPassesPipe, the passes to be
|
|
/// run are set by the static configuration, e.g.:
|
|
/// ```yaml
|
|
/// passes:
|
|
/// - pass1
|
|
/// - pass2
|
|
/// ```
|
|
/// However MLIR passes can also be initialized with configuration, so in that
|
|
/// case you have to specify the parts separately:
|
|
/// ```yaml
|
|
/// passes:
|
|
/// - name: pass1
|
|
/// options: option1=value1 option2=value2
|
|
/// - pass2
|
|
/// ```
|
|
class PureMLIRPassesPipe {
|
|
public:
|
|
static constexpr llvm::StringRef Name = "pure-mlir-passes-pipe";
|
|
using Arguments = TypeList<
|
|
PipeArgument<"Module", "MLIR Modules to apply the MLIR passes to">>;
|
|
|
|
private:
|
|
std::string TaskName;
|
|
|
|
struct PassInfo {
|
|
const mlir::PassInfo *PassInfo;
|
|
std::string Options;
|
|
};
|
|
std::vector<PassInfo> PassInfos;
|
|
|
|
public:
|
|
struct Configuration;
|
|
const std::string StaticConfiguration;
|
|
PureMLIRPassesPipe(llvm::StringRef StaticConfiguration);
|
|
|
|
public:
|
|
PipeOutput run(const Model &TheModel,
|
|
const revng::pypeline::Request &Incoming,
|
|
const revng::pypeline::Request &Outgoing,
|
|
llvm::StringRef Configuration,
|
|
CliftFunctionContainer &Container);
|
|
};
|
|
|
|
} // namespace revng::pypeline::pipes
|