Question

我正在编写代码，使用LLVM Go bindings从自定义VM字节码生成LLVM字节码;然后对代码进行JIT处理并在进程中执行。

自定义VM字节码有几个操作无法在LLVM中直接实现，因为它们需要更改外部状态;这些操作码的功能实现为Go函数。

从Go开始生成LLVM字节码有excellent guides，但没有一个解决回调或导出函数的问题。是否可以生成LLVM指令以回调Go函数？如果是这样，怎么样？

我已经尝试过@arrowd下面描述的方式，但它似乎不起作用。源代码，改编自Felix Angell的博客文章：

package main

import (
    "C"
    "fmt"
    "llvm.org/llvm/final/bindings/go/llvm"
)

// export AddInts
func AddInts(arg1, arg2 int) int {
    return arg1 + arg2;
}

func main() {
    // setup our builder and module
    builder := llvm.NewBuilder()
    mod := llvm.NewModule("my_module")

    // create our function prologue
    main := llvm.FunctionType(llvm.Int32Type(), []llvm.Type{}, false)
    llvm.AddFunction(mod, "main", main)
    block := llvm.AddBasicBlock(mod.NamedFunction("main"), "entry")
    builder.SetInsertPoint(block, block.FirstInstruction())

    // int a = 32
    a := builder.CreateAlloca(llvm.Int32Type(), "a")
    builder.CreateStore(llvm.ConstInt(llvm.Int32Type(), 32, false), a)

    // int b = 16
    b := builder.CreateAlloca(llvm.Int32Type(), "b")
    builder.CreateStore(llvm.ConstInt(llvm.Int32Type(), 16, false), b)

    // return a + b
    bVal := builder.CreateLoad(b, "b_val")
    aVal := builder.CreateLoad(a, "a_val")
    addIntsType := llvm.FunctionType(llvm.Int32Type(), []llvm.Type{llvm.Int32Type(), llvm.Int32Type()}, false)
    addInts := llvm.AddFunction(mod, "AddInts", addIntsType)
    call := builder.CreateCall(addInts, []llvm.Value{aVal, bVal}, "AddInts")
    builder.CreateRet(call)

    // verify it's all good
    if ok := llvm.VerifyModule(mod, llvm.ReturnStatusAction); ok != nil {
        fmt.Println(ok.Error())
    }
    mod.Dump()

    // create our exe engine
    engine, err := llvm.NewExecutionEngine(mod)
    if err != nil {
        fmt.Println(err.Error())
    }

    // run the function!
    funcResult := engine.RunFunction(mod.NamedFunction("main"), []llvm.GenericValue{})
    fmt.Printf("%d\n", funcResult.Int(false))
}

返回：

; ModuleID = 'my_module'

define i32 @main() {
entry:
  %a = alloca i32
  store i32 32, i32* %a
  %b = alloca i32
  store i32 16, i32* %b
  %b_val = load i32* %b
  %a_val = load i32* %a
  %AddInts = call i32 @AddInts(i32 %a_val, i32 %b_val)
  ret i32 %AddInts
}

declare i32 @AddInts(i32, i32)
LLVM ERROR: Tried to execute an unknown external function: AddInts
exit status 1

Answer 1

根据this answer，您可以通过添加“导出”注释将Go函数导出为C函数。如果为函数执行此操作，LLVM JIT应该能够在运行时链接期间解析此符号。答案没有提供调用导出函数的C代码，但假设它只是goProgressCB(args);，那么你可以为这个函数创建CallInst并JIT它。

<强>更新

看看this answer。看来最近的Go可以为你生成C标题来调用Go。有了这个，您可以使用clang -S -emit-llvm或clang -march=cpp编译它们，并获取对构成调用指令的LLVM API的文本IR或C ++调用。

Answer 2

主要问题是你实际上并没有在这里使用JIT，而是使用了解释器。如果JIT编译器可用，llvm.NewExecutionEngine将构造一个JIT编译器，否则将回退到解释器。

您应该使用llvm.NewMCJITCompiler。如果没有其他更改，这将失败，原因与NewExecutionEngine首先没有产生JIT的原因相同。您需要将以下内容添加到“func main（）”：

llvm.LinkInMCJIT()               
llvm.InitializeNativeTarget()    
llvm.InitializeNativeAsmPrinter()

你的代码的另一个更小的问题是（缺少）空格在“// export”魔术中很重要。

JITed LLVM代码如何回调到Go函数？

2 个答案: