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,6 +122,8 @@ void cuda_print_devices()
				@@ -122,6 +122,8 @@ void cuda_print_devices()

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