带有数组参数的 LLVM IR 函数

Question

我想从两个基本的 C++ 函数生成 LLVM IR 代码，如下所示。

int newFun2(int x){
    int z = x + x;
    return z;
}

int newFun(int *y){
    int first = y[3]; //How to define it using the LLVM API?
    int num = newFun2(first);
    return num;
}

我的问题是使用 LLVM API 获取数组参数的索引。有任何想法吗 ？非常感谢

已编辑

这是我使用 API 的代码：

llvm::LLVMContext &context = llvm::getGlobalContext();
llvm::Module *module = new llvm::Module("AST", context);
llvm::IRBuilder<> builder(context);

//newFun2
llvm::FunctionType *newFunc2Type = llvm::FunctionType::get(builder.getInt32Ty(), builder.getInt32Ty(), false);
llvm::Function *newFunc2 = llvm::Function::Create(newFunc2Type, llvm::Function::ExternalLinkage, "newFun2", module);

llvm::Function::arg_iterator argsFun2 = newFunc2->arg_begin();
llvm::Value* x = argsFun2++;
x->setName("x");

llvm::BasicBlock* block = llvm::BasicBlock::Create(context, "entry", newFunc2);
llvm::IRBuilder<> builder2(block);

llvm::Value* tmp = builder2.CreateBinOp(llvm::Instruction::Add,
                                 x, x, "tmp");

builder2.CreateRet(tmp);

//newFun
llvm::FunctionType *newFuncType = llvm::FunctionType::get(builder.getInt32Ty(), builder.getInt32Ty()->getPointerTo(), false);
llvm::Function *newFunc = llvm::Function::Create(newFuncType, llvm::Function::ExternalLinkage, "newFun", module);

llvm::BasicBlock* block2 = llvm::BasicBlock::Create(context, "entry", newFunc);
llvm::IRBuilder<> builder3(block2);

module->dump();

这是生成的 LLVM IR：

; ModuleID = 'AST'

define i32 @newFun2(i32 %x) {
entry:
  %tmp = add i32 %x, %x
  ret i32 %tmp
}

define i32 @newFun(i32*) {
entry:
}

由于数组访问，我被困在 newFun 的主体上。

Answer 1

我认为您首先需要了解 IR 应该是什么样子。可以通过查看语言规范或使用 Clang 将 C 代码编译为 IR并查看结果来完成。

在任何情况下，访问给定索引处的数组元素的方法是使用extractvalue（仅接受常量索引）或使用gep。这两个都有对应的构造函数/工厂方法和IRBuilder构造它们的方法，例如

builder.CreateExtractValue(y, 3);

创建 gep 稍微复杂一些；我建议看一下gep 指南。

但是，查看如何调用 LLVM API 以创建所需 IR 的一个好方法是使用llc（LLVM 命令行工具之一）从 IR 文件中生成包含这些调用本身的源文件，请参阅这两个相关问题：

• 可以从 LLVM-IR 自动生成 llvm c++ api 代码吗？
• 生成 LLVM C++ API 代码作为后端