pass_manager.h

Go to the documentation of this file.

#pragma once
 
#include <memory>
 
#include "spdlog/spdlog.h"
 
#include "mlir/Pass/PassManager.h"
#include "llvm/Support/raw_ostream.h"
#include "mlir/Pass/PassInstrumentation.h"
#include "mlir/Pass/Pass.h"
#include "mlir/IR/BuiltinOps.h"
 
namespace picceler {

struct PassLogger : public mlir::PassInstrumentation {
  void runBeforePass(mlir::Pass *pass, mlir::Operation *op) override {
    spdlog::debug("Started pass: {}", pass->getName().str());
  }
  void runAfterPass(mlir::Pass *pass, mlir::Operation *op) override {
    spdlog::debug("Finished pass: {}", pass->getName().str());
  }
  void runAfterPassFailed(mlir::Pass *pass, mlir::Operation *op) override {
    spdlog::error("Failed pass: {}", pass->getName().str());
  }
};

class IRPassManager {
public:
  IRPassManager(mlir::MLIRContext *context);

  void addPasses(bool profile);

  bool run(mlir::ModuleOp module);
 
private:

  void addHighLevelOptimizationPasses(bool profile);

  void addAffineLoweringPasses();

  void addRuntimeLoweringPasses();

  void addBackendLoweringPasses();

 
private:
  std::unique_ptr<llvm::raw_fd_ostream> _outStream;
  mlir::PassManager _passManager;
};
 
} // namespace picceler