実行時に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で返されます。これは、エラー1(= cudaErrorMissingConfiguration)を返す最後の_cuMemcpyDtoHを除いて、すべてが正常に行われたことを意味します。誰かがこのエラーの意味と、なぜそれが発生するのかを説明できますか?それを解決する方法はありますか?そして、実行時に関数をロードするとなぜ表示されるのですか?
ありがとう。
私のシステム:nvccリリース4.1 GPU:GTX 480 NVRMバージョン:NVIDIA UNIXx86_64カーネルモジュール285.05.32GCCバージョン:gccバージョン4.5.2(Ubuntu / Linaro 4.5.2-8ubuntu4)