mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Giacomo Vercesi
11f1e532c6
Add a `Dirty` boolean to the `ContainerBase` class. This will be set to true when the container mutates itself (e.g. remove, mergeBack, clear). The dirty bit is reset when running `PipelineManager.store`.
62 lines
1.7 KiB
C++
62 lines
1.7 KiB
C++
#pragma once
|
|
|
|
//
|
|
// This file is distributed under the MIT License. See LICENSE.md for details.
|
|
//
|
|
|
|
#include "llvm/ADT/DenseMap.h"
|
|
|
|
#include "mlir/IR/BuiltinOps.h"
|
|
#include "mlir/IR/FunctionInterfaces.h"
|
|
#include "mlir/IR/MLIRContext.h"
|
|
|
|
#include "revng/Pipeline/Container.h"
|
|
#include "revng/Pipes/Kinds.h"
|
|
|
|
namespace revng::pipes {
|
|
|
|
class MLIRContainer : public pipeline::Container<MLIRContainer> {
|
|
public:
|
|
static const char ID;
|
|
static constexpr auto Name = "mlir-module";
|
|
static constexpr auto MIMEType = "application/x.mlir.bc";
|
|
|
|
private:
|
|
// unique_ptr is used to allow moving the context.
|
|
std::unique_ptr<mlir::MLIRContext> Context;
|
|
mlir::OwningOpRef<mlir::ModuleOp> Module;
|
|
|
|
public:
|
|
explicit MLIRContainer(const llvm::StringRef Name) :
|
|
pipeline::Container<MLIRContainer>(Name) {
|
|
clearImpl();
|
|
}
|
|
|
|
mlir::MLIRContext *getContext() { return Context.get(); }
|
|
mlir::ModuleOp getModule() { return *Module; }
|
|
void setModule(mlir::OwningOpRef<mlir::ModuleOp> &&NewModule);
|
|
|
|
std::unique_ptr<pipeline::ContainerBase>
|
|
cloneFiltered(const pipeline::TargetsList &Targets) const override;
|
|
|
|
void mergeBackImpl(MLIRContainer &&Container) override;
|
|
|
|
pipeline::TargetsList enumerate() const override;
|
|
|
|
bool removeImpl(const pipeline::TargetsList &Targets) override;
|
|
|
|
void clearImpl() override;
|
|
|
|
llvm::Error serialize(llvm::raw_ostream &OS) const override;
|
|
llvm::Error deserializeImpl(const llvm::MemoryBuffer &Buffer) override;
|
|
|
|
llvm::Error extractOne(llvm::raw_ostream &OS,
|
|
const pipeline::Target &Target) const override;
|
|
|
|
static std::vector<pipeline::Kind *> possibleKinds() {
|
|
return { &kinds::MLIRFunctionKind };
|
|
}
|
|
};
|
|
|
|
} // namespace revng::pipes
|