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 "llvm/Support/FileSystem.h"
#include "mlir/Pass/PassInstrumentation.h"
 
#include "passes.h"
 
namespace picceler {
 
struct PassLogger : public mlir::PassInstrumentation {
  void runBeforePass(mlir::Pass *pass, mlir::Operation *op) override {
    spdlog::info("Started pass: {}", pass->getName().str());
  }
  void runAfterPass(mlir::Pass *pass, mlir::Operation *op) override {
    spdlog::info("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 run(mlir::ModuleOp module);
 
private:
  void addPasses();

 
  void addHighLevelOptimizationPasses();
  void addAffineLoweringPasses();
  void addRuntimeLoweringPasses();
  void addBackendLoweringPasses();
 
private:
  std::unique_ptr<llvm::raw_fd_ostream> _outStream;
  mlir::PassManager _passManager;
};
 
} // namespace picceler