fix test bug

taichi-dev · Jun 16, 2022 · 73ebb92 · 73ebb92
1 parent 9ae6722
commit 73ebb92
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/taichi/backends/cuda/cuda_driver.cpp b/taichi/backends/cuda/cuda_driver.cpp
@@ -20,7 +20,7 @@ bool CUDADriver::detected() {
 }
 
 CUDADriver::CUDADriver() {
-  load_lib("libcuda.so", "nvcuda.dll");
+  if (!load_lib("libcuda.so", "nvcuda.dll")) return;
 
   loader_->load_function("cuGetErrorName", get_error_name);
   loader_->load_function("cuGetErrorString", get_error_string);