Commit 6a0d3b3e authored by Mathieu Barnachon's avatar Mathieu Barnachon

Called after setDevice.

Wrap in a cudaSafeCall.
parent b030ac04
...@@ -70,8 +70,8 @@ void cv::cuda::setDevice(int device) ...@@ -70,8 +70,8 @@ void cv::cuda::setDevice(int device)
(void) device; (void) device;
throw_no_cuda(); throw_no_cuda();
#else #else
cudaFree(0);
cudaSafeCall( cudaSetDevice(device) ); cudaSafeCall( cudaSetDevice(device) );
cudaSafeCall( cudaFree(0) );
#endif #endif
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment