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.../
};
};