Jelajahi Sumber

Merge topic 'FindCUDAToolkit-cusparse-deps'

516d807423 FindCUDAToolkit: Add dependency between cusparse and nvJitLink

Acked-by: Kitware Robot <[email protected]>
Merge-request: !8397
Brad King 2 tahun lalu
induk
melakukan
d797c3c8e9
1 mengubah file dengan 9 tambahan dan 6 penghapusan
  1. 9 6
      Modules/FindCUDAToolkit.cmake

+ 9 - 6
Modules/FindCUDAToolkit.cmake

@@ -1036,12 +1036,20 @@ if(CUDAToolkit_FOUND)
     endif()
   endif()
 
+  if(CUDAToolkit_VERSION VERSION_GREATER_EQUAL 12.0.0)
+    _CUDAToolkit_find_and_add_import_lib(nvJitLink)
+    _CUDAToolkit_find_and_add_import_lib(nvJitLink_static DEPS cudart_static_deps)
+  endif()
+
   _CUDAToolkit_find_and_add_import_lib(culibos) # it's a static library
-  foreach (cuda_lib cublasLt cufft curand cusparse nppc nvjpeg)
+  foreach (cuda_lib cublasLt cufft curand nppc nvjpeg)
     _CUDAToolkit_find_and_add_import_lib(${cuda_lib})
     _CUDAToolkit_find_and_add_import_lib(${cuda_lib}_static DEPS culibos)
   endforeach()
 
+  _CUDAToolkit_find_and_add_import_lib(cusparse DEPS nvJitLink)
+  _CUDAToolkit_find_and_add_import_lib(cusparse_static DEPS nvJitLink_static culibos)
+
   if(CUDAToolkit_VERSION VERSION_GREATER_EQUAL 11.0.0)
     # cublas depends on cublasLt
     # https://docs.nvidia.com/cuda/archive/11.0/cublas/index.html#static-library
@@ -1134,11 +1142,6 @@ if(CUDAToolkit_FOUND)
     endif()
   endif()
 
-  if(CUDAToolkit_VERSION VERSION_GREATER_EQUAL 12.0.0)
-    _CUDAToolkit_find_and_add_import_lib(nvJitLink DEPS cuda_driver)
-    _CUDAToolkit_find_and_add_import_lib(nvJitLink_static DEPS cuda_driver)
-  endif()
-
   _CUDAToolkit_find_and_add_import_lib(nvrtc_builtins DEPS cuda_driver)
   _CUDAToolkit_find_and_add_import_lib(nvrtc DEPS nvrtc_builtins nvJitLink)
   if(CUDAToolkit_VERSION VERSION_GREATER_EQUAL 11.5.0)