Browse Source

FindCUDAToolkit: Add support for CUDA::nvrtc_static

Robert Maynard 3 years ago
parent
commit
2a94c762ed
2 changed files with 18 additions and 1 deletions
  1. 5 0
      Help/release/dev/FindCUDAToolkit-nvrtc.rst
  2. 13 1
      Modules/FindCUDAToolkit.cmake

+ 5 - 0
Help/release/dev/FindCUDAToolkit-nvrtc.rst

@@ -0,0 +1,5 @@
+FindCUDAToolkit-nvrtc
+---------------------
+
+* The :module:`FindCUDAToolkit` module now provides a target for
+  ``libnvrtc_static``, if found.

+ 13 - 1
Modules/FindCUDAToolkit.cmake

@@ -380,6 +380,12 @@ Targets Created:
 
 - ``CUDA::nvrtc``
 
+.. versionadded:: 3.26
+
+  - ``CUDA::nvrtc_builtins``
+  - ``CUDA::nvrtc_static`` starting in CUDA 11.5
+  - ``CUDA::nvrtc_builtins_static`` starting in CUDA 11.5
+
 .. _`cuda_toolkit_nvjitlink`:
 
 nvJitLink
@@ -1099,7 +1105,6 @@ if(CUDAToolkit_FOUND)
                                         EXTRA_INCLUDE_DIRS "${CUDAToolkit_CUPTI_INCLUDE_DIR}")
   endif()
 
-  _CUDAToolkit_find_and_add_import_lib(nvrtc DEPS cuda_driver)
   if(CUDAToolkit_VERSION VERSION_GREATER_EQUAL 11.1.0)
     if(NOT TARGET CUDA::nvptxcompiler_static)
       _CUDAToolkit_find_and_add_import_lib(nvptxcompiler_static DEPS cuda_driver)
@@ -1114,6 +1119,13 @@ if(CUDAToolkit_FOUND)
     _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)
+    _CUDAToolkit_find_and_add_import_lib(nvrtc_builtins_static DEPS cuda_driver)
+    _CUDAToolkit_find_and_add_import_lib(nvrtc_static DEPS nvrtc_builtins_static nvptxcompiler_static nvJitLink_static)
+  endif()
+
   _CUDAToolkit_find_and_add_import_lib(nvml ALT nvidia-ml nvml)
 
   if(WIN32)