想要什麼:
int add1(int p1) { return 1 + p1; }
IR:
define i32 @add1(i32 %p1) {
entry:
%retVal = add i32 1, %p1
ret i32 %retVal
}
#include "llvm/ExecutionEngine/Orc/LLJIT.h"
#include "llvm/IR/Function.h"
#include "llvm/IR/IRBuilder.h"
#include "llvm/IR/Module.h"
#include "llvm/Support/CommandLine.h"
#include "llvm/Support/InitLLVM.h"
#include "llvm/Support/TargetSelect.h"
#include "llvm/Support/raw_ostream.h"
using namespace llvm;
using namespace llvm::orc;
ExitOnError ExitOnErr;
ThreadSafeModule createTSM() {
// context 擁有許多核心的 LLVM 數據結構,例如類型和常量值表
auto Context = std::make_unique<LLVMContext>();
// module 包含函數和全局變量的LLVM構造,它是 LLVM IR 用來包含代碼的頂級結構。它將擁有我們生成的所有 IR 的內存
auto M = std::make_unique<Module>("test", *Context);
// 創建一個構建器,可以輕鬆生成 LLVM 指令
IRBuilder<> builder(*Context);
// 聲明一個函數 int add1(int) 添加到module中
auto funcType = FunctionType::get(Type::getInt32Ty(*Context), { Type::getInt32Ty(*Context) }, false);
Function* Add1F = Function::Create(funcType, Function::ExternalLinkage, "add1", M.get());
// 創建一個block添加到Add1F
BasicBlock* BB = BasicBlock::Create(*Context, "entry", Add1F);
// 構建器接下來的指令將插入到BB
builder.SetInsertPoint(BB);
// 獲取指向常量“1”的指針
Value* One = builder.getInt32(1);
// 獲取指向 add1 函數的整數參數的指針
assert(Add1F->arg_size() == 1); // 確保有一個 arg
Argument* p1 = Add1F->getArg(0); // 獲取參數
p1->setName("p1"); // 設置arg name, int add1(int p1);
// 創建 add 指令,將其插入到 BB
// 創建返回指令並將其添加到基本塊
builder.CreateRet(builder.CreateAdd(One, p1, "retVal")); // ret 1 + p1
// 打印出所有生成的代碼
M->print(errs(), nullptr);
// 返回線程模塊
return ThreadSafeModule(std::move(M), std::move(Context));
}
int main() {
InitializeNativeTarget();
InitializeNativeTargetAsmPrinter();
// Create an LLJIT instance.
auto _jit = ExitOnErr(LLJITBuilder().create());
auto M = createTSM();
ExitOnErr(_jit->addIRModule(std::move(M)));
// 查找 JIT 函數,將其轉換爲函數指針,然後調用它.
auto Add1Sym = ExitOnErr(_jit->lookup("add1"));
int (*Add1)(int) = (int (*)(int))Add1Sym.getAddress();
int Result = Add1(42);
outs() << "add1(42) = " << Result << "\n";
return 0;
}