xo-jit: move IR improvement pipeline to dedicated class
This commit is contained in:
parent
537e178e09
commit
f3af5d27bf
5 changed files with 126 additions and 40 deletions
75
include/xo/jit/IrPipeline.hpp
Normal file
75
include/xo/jit/IrPipeline.hpp
Normal file
|
|
@ -0,0 +1,75 @@
|
|||
/** @file IrPipeline.hpp
|
||||
*
|
||||
* Author: Roland Conybeare
|
||||
**/
|
||||
|
||||
#pragma once
|
||||
|
||||
#include "xo/refcnt/Refcounted.hpp"
|
||||
|
||||
/* stuff from kaleidoscope.cpp */
|
||||
#include "llvm/ADT/APFloat.h"
|
||||
#include "llvm/ADT/STLExtras.h"
|
||||
#include "llvm/IR/BasicBlock.h"
|
||||
#include "llvm/IR/Constants.h"
|
||||
#include "llvm/IR/DerivedTypes.h"
|
||||
#include "llvm/IR/Function.h"
|
||||
#include "llvm/IR/IRBuilder.h"
|
||||
#include "llvm/IR/LLVMContext.h"
|
||||
#include "llvm/IR/Module.h"
|
||||
#include "llvm/IR/PassManager.h"
|
||||
#include "llvm/IR/Type.h"
|
||||
#include "llvm/IR/Verifier.h"
|
||||
#include "llvm/Passes/PassBuilder.h"
|
||||
#include "llvm/Passes/StandardInstrumentations.h"
|
||||
#include "llvm/Support/TargetSelect.h"
|
||||
#include "llvm/Target/TargetMachine.h"
|
||||
#include "llvm/Transforms/InstCombine/InstCombine.h"
|
||||
#include "llvm/Transforms/Scalar.h"
|
||||
#include "llvm/Transforms/Scalar/GVN.h"
|
||||
#include "llvm/Transforms/Scalar/Reassociate.h"
|
||||
#include "llvm/Transforms/Scalar/SimplifyCFG.h"
|
||||
|
||||
//#include <cstdint>
|
||||
|
||||
namespace xo {
|
||||
namespace jit {
|
||||
/** @class IrPipeline
|
||||
* @brief represent an LLVM IR pipeline
|
||||
*
|
||||
* Represents analysis/transformation short of generating
|
||||
* machine-code. For now pipeline stages are hardwired;
|
||||
* adapted from the LLVM Kaleidoscope example project.
|
||||
*
|
||||
* Conversely, pipeline *starts* with code already that has
|
||||
* already been expressed in LLVM IR
|
||||
**/
|
||||
class IrPipeline : public ref::Refcount {
|
||||
public:
|
||||
explicit IrPipeline(llvm::LLVMContext & llvm_cx);
|
||||
|
||||
void run_pipeline(llvm::Function & fn);
|
||||
|
||||
private:
|
||||
// ----- transforms (also adapted from kaleidescope.cpp) ------
|
||||
|
||||
/** manages all the passes+analaysis (?) **/
|
||||
std::unique_ptr<llvm::FunctionPassManager> llvm_fpmgr_;
|
||||
/** loop analysis (?) **/
|
||||
std::unique_ptr<llvm::LoopAnalysisManager> llvm_lamgr_;
|
||||
/** function-level analysis (?) **/
|
||||
std::unique_ptr<llvm::FunctionAnalysisManager> llvm_famgr_;
|
||||
/** cgscc (?) analysis **/
|
||||
std::unique_ptr<llvm::CGSCCAnalysisManager> llvm_cgamgr_;
|
||||
/** module analsyis (?) **/
|
||||
std::unique_ptr<llvm::ModuleAnalysisManager> llvm_mamgr_;
|
||||
/** pass instrumentation **/
|
||||
std::unique_ptr<llvm::PassInstrumentationCallbacks> llvm_pic_;
|
||||
/** standard instrumentation **/
|
||||
std::unique_ptr<llvm::StandardInstrumentations> llvm_si_;
|
||||
}; /*IrPipeline*/
|
||||
} /*namespace jit*/
|
||||
} /*namespace xo*/
|
||||
|
||||
|
||||
/** end IrPipeline.hpp **/
|
||||
|
|
@ -8,6 +8,7 @@
|
|||
//#include <cstdint>
|
||||
|
||||
#include "xo/refcnt/Refcounted.hpp"
|
||||
#include "IrPipeline.hpp"
|
||||
#include "xo/expression/Expression.hpp"
|
||||
#include "xo/expression/ConstantInterface.hpp"
|
||||
#include "xo/expression/PrimitiveInterface.hpp"
|
||||
|
|
@ -150,6 +151,7 @@ namespace xo {
|
|||
* kal_jit_.addModule()
|
||||
* Note that this makes the module itself unavailable to us
|
||||
**/
|
||||
xo::ref::rp<IrPipeline> ir_pipeline_;
|
||||
|
||||
/** owns + manages core "global" llvm data,
|
||||
* including type- and constant- unique-ing tables.
|
||||
|
|
@ -176,22 +178,6 @@ namespace xo {
|
|||
**/
|
||||
std::map<std::string, llvm::Value*> nested_env_;
|
||||
|
||||
// ----- transforms (also adapted from kaleidescope.cpp) ------
|
||||
|
||||
/** manages all the passes+analaysis (?) **/
|
||||
std::unique_ptr<llvm::FunctionPassManager> llvm_fpmgr_;
|
||||
/** loop analysis (?) **/
|
||||
std::unique_ptr<llvm::LoopAnalysisManager> llvm_lamgr_;
|
||||
/** function-level analysis (?) **/
|
||||
std::unique_ptr<llvm::FunctionAnalysisManager> llvm_famgr_;
|
||||
/** cgscc (?) analysis **/
|
||||
std::unique_ptr<llvm::CGSCCAnalysisManager> llvm_cgamgr_;
|
||||
/** module analsyis (?) **/
|
||||
std::unique_ptr<llvm::ModuleAnalysisManager> llvm_mamgr_;
|
||||
/** pass instrumentation **/
|
||||
std::unique_ptr<llvm::PassInstrumentationCallbacks> llvm_pic_;
|
||||
/** standard instrumentation **/
|
||||
std::unique_ptr<llvm::StandardInstrumentations> llvm_si_;
|
||||
}; /*Jit*/
|
||||
|
||||
inline std::ostream &
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue