cuda: prevent ptxas crash with -n

2016-12-18 03:23:11 +01:00 · 2016-12-18 03:23:11 +01:00 · ce6a8da188
commit ce6a8da188
parent 397472818d
1 changed files with 3 additions and 1 deletions
--- a/cuda.cpp
+++ b/cuda.cpp
@ -122,7 +122,9 @@ void cuda_print_devices()

 void cuda_shutdown()
 {
-	cudaDeviceSynchronize();
+	// require gpu init first
+	if (thr_info != NULL)
+		cudaDeviceSynchronize();
 	cudaDeviceReset();
 }