我正在尝试使用 dlsym 在运行时加载 cuda 驱动程序 api 函数,但遇到了一个奇怪的错误。我的代码可以在我的系统上顺利运行(使用 nvcc 等编译):
#include <cuda.h>
#include <stdio.h>
int main(int argc,char *argv[]){
if(argc<3){
printf("Usage: ./test.cu <ptx_file> <cuda_device>\n");
exit(0);
}
// Error code
CUresult error;
// My number
unsigned int h_var=7;
// Initialize driver API
error = cuInit(0);
if((int)error!=0){
printf("Error! cuInit returned: %d\n",(int)error);
exit(0);
}
// Get Cuda Device and give handle
CUdevice cu_device;
error = cuDeviceGet(&cu_device,atoi(argv[2]));
if((int)error!=0){
printf("Error! cuDeviceGet returned: %d\n",(int)error);
exit(0);
}
// Create context to run on device
CUcontext cu_context;
error = cuCtxCreate(&cu_context, 0, cu_device);
if((int)error!=0){
printf("Error! cuCtxCreate returned: %d\n",(int)error);
exit(0);
}
// Load ptx code
CUmodule cu_module;
error = cuModuleLoad(&cu_module,argv[1]);
if((int)error!=0){
printf("Error! cuModuleLoad returned: %d\n",(int)error);
exit(0);
}
// Get kernel function
CUfunction func;
error = cuModuleGetFunction(&func,cu_module,"testing");
if((int)error!=0){
printf("Error! cuModuleGetFunction returned: %d\n",(int)error);
exit(0);
}
CUdeviceptr var;
// Allocate device memory
unsigned int size = sizeof(unsigned int);
error = cuMemAlloc(&var, size);
if((int)error!=0){
printf("Error! cuMemAlloc returned: %d\n",(int)error);
exit(0);
}
// Copy variable to host
error = cuMemcpyHtoD(var,&h_var,size);
if((int)error!=0){
printf("Error! cuMemcpyHtoD returned: %d\n",(int)error);
exit(0);
}
// Lauch kernel
void *args[] = {&var};
error = cuLaunchKernel(func, 1, 1, 1, 1, 1, 1, 0, NULL, args, NULL);
if((int)error!=0){
printf("Error! cuLaunchKernel returned: %d\n",(int)error);
exit(0);
}
// Get result to host
error = cuMemcpyDtoH(&h_var,var,size);
if((int)error!=0){
printf("Error! cuMemcpyDtoH returned: %d\n",(int)error);
exit(0);
}
// Free device memory
error = cuMemFree(var);
if((int)error!=0){
printf("Error! cuMemFree returned: %d\n",(int)error);
exit(0);
}
// Destroy context
error = cuCtxDestroy(cu_context);
if((int)error!=0){
printf("Error! cuCtxDestroy returned: %d\n",(int)error);
exit(0);
}
// Print result
printf("var: %d\n",h_var);
}
使用我编写的 ptx 代码:(这是一个简单的添加,只是为了测试它是否有效)
.version 1.4 .target sm_10, map_f64_to_f32
.entry testing (
.param .u64 mynum)
{
.reg .u64 %r;
.reg .u64 %i;
ld.param.u64 %r,[mynum];
ld.global.u64 %i,[%r];
add.u64 %i,%i,3;
st.global.u64 [%r+0],%i;
exit;
}
到现在为止还挺好。然后我获取代码并使用 dlsym 加载每个函数,如下所示:
/* Lauch kernel */
void *args[] = {&var};
int (*_cuLaunchKernel)( void *, unsigned int, unsigned int, unsigned int, unsigned int, unsigned int, unsigned int, unsigned int, void *, void **, void ** );
*(void **)(&_cuLaunchKernel) = dlsym(dlhandle, "cuLaunchKernel");
(*_cuLaunchKernel)(cu_func,1 , 1, 1, 1, 1, 1, 0, NULL, args, NULL);
/* Get result to host */
int (*_cuMemcpyDtoH)( void *, void *, size_t );
*(void **)(&_cuMemcpyDtoH) = dlsym(dlhandle, "cuMemcpyHtoD");
error = (*_cuMemcpyDtoH)(&h_var,var,size);
(其余代码的逻辑相同)。我所有的函数都返回 0,这意味着一切正常,除了最后一个 _cuMemcpyDtoH ,它返回错误 1 ( = cudaErrorMissingConfiguration )。有人可以解释这个错误是什么意思,为什么会发生?有没有办法解决它?为什么当我在运行时加载函数时它会出现?
谢谢。
我的系统:nvcc 版本 4.1 GPU:GTX 480 NVRM 版本:NVIDIA UNIX x86_64 内核模块 285.05.32 GCC 版本:gcc 版本 4.5.2(Ubuntu/Linaro 4.5.2-8ubuntu4)