我正在使用一个使用LLVM的JIT.该语言有一个用C++编写的小运行时,我使用clang编译成LLVM IR
clang++ runtime.cu --cuda-gpu-arch=sm_50 -c -emit-llvm
然后加载*.bc文件,生成额外的IR,并动态执行.CUDA的原因是我想在运行时添加一些GPU加速.但是,这会引入CUDA特定的外部函数,这些函数会产生以下错误:
LLVM ERROR: Program used external function 'cudaSetupArgument' which could not be resolved!
正如这里所讨论的,这通常通过在编译程序时包含适当的库来解决:
g++ main.c cmal.o -L/usr/local/cuda/lib64 -lcudart
但是,我不确定如何使用LLVM在JITed模块中包含库.我发现这个问题表明以前可以将库添加到JIT中的模块,如下所示:
[your module]->addLibrary("m");
不幸的是,这已被弃用.现在有人能告诉我最好的方法吗？如果我需要提供更多信息,请告诉我们!
此外,我不确定这是否是将GPU卸载合并到我的JIT中的最佳方式,所以如果有人能指出我更好的方法那么请做:)
谢谢!
编辑:我正在使用LLVM 5.0和我正在使用的JIT引擎llvm/ExecutionEngine/ExecutionEngine.h,更具体地说,我创建它像这样:
EngineBuilder EB(std::move(module));
ExecutionEngine *EE = EB.create(targetMachine);