cuda: prevent ptxas crash with -n

8 years ago · ce6a8da188
1 changed files with 3 additions and 1 deletions
--- a/cuda.cpp
+++ b/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();
 }