llvm function expr

想要什麼:

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;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章