0

我正在尝试使用 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)

4

1 回答 1

0

对于您正在使用的驱动程序 API,错误代码 1 表示CUDA_ERROR_INVALID_VALUE

这个错误的原因是:

*(void **)(&_cuMemcpyDtoH) = dlsym(dlhandle, "cuMemcpyHtoD");

--- 您正在尝试使用错误的功能:HtoD而不是DtoH.

于 2012-05-21T18:20:54.927 回答