我在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中可以快速,简洁地使用它。欢迎任何帮助!
答案 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/