Browse Source

cuda: prevent ptxas crash with -n

master
Tanguy Pruvot 8 years ago
parent
commit
ce6a8da188
  1. 4
      cuda.cpp

4
cuda.cpp

@ -122,7 +122,9 @@ void cuda_print_devices() @@ -122,7 +122,9 @@ void cuda_print_devices()
void cuda_shutdown()
{
cudaDeviceSynchronize();
// require gpu init first
if (thr_info != NULL)
cudaDeviceSynchronize();
cudaDeviceReset();
}

Loading…
Cancel
Save