Ensure the Cuda context is initialized correctly as long as the setDevice is not called in a multi-thread environment.

2016-02-08 13:17:08 +01:00 · 2016-02-08 13:17:08 +01:00 · b030ac0433
commit b030ac0433
parent f9f5313670
1 changed files with 1 additions and 0 deletions
--- a/modules/core/src/cuda_info.cpp
+++ b/modules/core/src/cuda_info.cpp
@ -70,6 +70,7 @@ void cv::cuda::setDevice(int device)
    (void) device;
    throw_no_cuda();
 #else
+    cudaFree(0);
    cudaSafeCall( cudaSetDevice(device) );
 #endif
 }