在llvmlite中链接C.

时间:2016-04-16 00:13:58

标签: python c compiler-construction llvm llvmlite

我在Python中编写编译器,使用llvmlite生成中间LLVM IR。 Lexer和解析器已经完成,现在我正在进行代码生成。编译器将是动态和弱类型的,所以我需要在运行时做一些事情,比如分配。为此,我已经在C中实现了一些功能,现在我想使用llvmlite中的builder.call来调用这些函数。

我没有找到如何执行此操作的文档或示例。

这个函数只是一个简单的例子,真正的函数要大得多。

C:

int some_function(int a)
{
     return a + 4;
}

的Python:

...

    main_ty = ir.FunctionType(ir.IntType(32), [])
    func = ir.Function(module, main_ty, 'main')
    block = func.append_basic_block('entry')
    builder = ir.IRBuilder(block)

    # I want to do something like this...

    ret = builder.call(some_function, [ir.Constant(ir.IntType(32), 34)]);

...

我可以使用llvmlite构建器直接编写函数,但是在C中可以快速,简洁地使用它。欢迎任何帮助!

3 个答案:

答案 0 :(得分:4)

您可以导入包含运行时的动态库。

import zipfile
with zipfile.ZipFile("file.zip","r") as zip_ref:
    zip_ref.extractall("targetdir")

然后你可以简单地生成正常的函数调用。

答案 1 :(得分:4)

在LLVM IR方面,您可以使用正确的签名(并且没有正文)声明函数,并像其他任何函数一样插入对它们的调用。这就像在C中你可以调用另一个文件中定义的函数一样。

从那里,你必须以某种方式链接你的C函数。此处的详细信息取决于您打算如何使用生成的IR代码。例如,您可以使用clang将其转换为目标文件,然后像任何其他程序一样链接它。或者您可以使用llvm JIT,在这种情况下,@ Coder3000的答案(llvmlite.binding.load_library_permanently)应该可以让LLVM解析您的函数调用。

答案 2 :(得分:2)

@ Coder3000和@Ismail Badawi的答案很完美,但没有代码。有我的代码

    #!/usr/bin/env python
    # coding=utf-8
    from __future__ import print_function
    from ctypes import CFUNCTYPE, c_double,cdll,c_int
    import llvmlite.binding as llvm
    import llvmlite.ir as  ir

    llvm.initialize()
    llvm.initialize_native_target()
    llvm.initialize_native_asmprinter()
    #load shared library
    llvm.load_library_permanently('./TestLib.so')
    # Create some useful types
    double = ir.DoubleType()
    fnty = ir.FunctionType(double, (double, double))

    # Create an empty module...
    module = ir.Module("fdadd")
    # and declare a function named "fpadd" inside it
    func = ir.Function(module, fnty, name="fpadd")

    # Now implement the function
    block = func.append_basic_block(name="entry")
    builder = ir.IRBuilder(block)
    a, b = func.args
    result = builder.fadd(a, b, name="res")
    builder.ret(result)
    #function call in llvm ir internal
    func2=ir.Function(module,fnty,name="fdadd")
    block2=func2.append_basic_block(name="entry")
    builder=ir.IRBuilder(block2)
    a,b=func2.args
    result2=builder.call(func,(a,b))
    builder.ret(result2)
    # llvm IR call external C++ function
    outfunc=ir.Function(module,fnty,name="SampleAddDouble")
    #just declare shared library function in module
    outaddfunc=ir.Function(module,fnty,name="outadd")
    builder=ir.IRBuilder(outaddfunc.append_basic_block(name="entry"))
    a,b=outaddfunc.args
    outresult=builder.call(outfunc,(a,b))
    builder.ret(outresult)
    strmod=str(module)
    # Print the module IR
    print(strmod)
    print("-----------------------------------------")
    #assembly llvm ir
    assmod = llvm.parse_assembly(strmod)
    assmod.verify()
    print("--parse assembly")
    target = llvm.Target.from_default_triple()
    target_machine = target.create_target_machine()
    engine = llvm.create_mcjit_compiler(assmod, target_machine)
    engine.finalize_object()
    print(engine)

    # Look up the function pointer (a Python int) 
    #llvm execution engine call llvm IR function(or dsl function)
    func_ptr = engine.get_function_address("fpadd")
    print('func_ptr is:',func_ptr)

    # Run the function via ctypes
    cfunc = CFUNCTYPE(c_double, c_double, c_double)(func_ptr)
    res = cfunc(1.0, 3.5)
    print("fpadd(...) =", res)

    #llvm binding layer call shared library function
    add_int_addr = llvm.address_of_symbol("SampleAddInt")
    print(add_int_addr)
    add_int_func=CFUNCTYPE(c_int,c_int,c_int)(add_int_addr)
    res2=add_int_func(23,34)
    print(res2)

    #llvm execution engine call shared library function ,by llvm IR 
    add_double_addr=engine.get_function_address("outadd")
    print(add_double_addr)
    add_double_func=CFUNCTYPE(c_double,c_double,c_double)(add_double_addr)
    res3=add_double_func(1.21,1.12)
    print(res3)

和Testlib.cpp $ reference https://helloacm.com/calling-c-shared-library-from-python-code-linux-version/

关于这个问题,这里是其他参考

- Call C/C++ functions from the ExecutionEngine - Call Python code from LLVM JIT - http://eli.thegreenplace.net/2015/calling-back-into-python-from-llvmlite-jited-code/