mirror/CMake
1
0
Fork 0
mirror of https://github.com/Kitware/CMake.git synced 2025-12-31 02:39:48 -06:00
Code Issues Packages Projects Releases Wiki Activity

FindCUDAToolkit: Add support for CUDA::nvrtc_static

Browse Source
This commit is contained in:
Robert Maynard
2023-01-20 10:38:27 -05:00
parent 06c7e88b91
commit 2a94c762ed
2 changed files with 18 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
Help/release/dev/FindCUDAToolkit-nvrtc.rst Normal file
View File

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

14
Modules/FindCUDAToolkit.cmake
View File

@@ -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)