3

cudaSetDevice次の使用法が正しいかどうかは誰にもわかりませんか? 任意のホスト スレッドで、いつでも異なるデバイスで作成されたリソースを繰り返し呼び出したい。CUDAでこれを行う方法はありますか?

 cudaSetDevice(0);
 /...create cuda streams and do some memory allocation on gpu.../
 cudaSetDevice(1);
 /...create cuda streams and do some memory allocation on gpu.../
 #pragma omp parallel num_threads(2)
 { 
   int omp_threadID=omp_get_thread_num();
    ....
   if (omp_threadID==0)
   {
    cudaSetDevice(0);
    /...calling streams/memory created on device 0.../
   }
   else
   {
    cudaSetDevice(1);
    /...calling streams/memory created on device 1.../
    }; 
  };
4

1 に答える 1