FindCUDAToolkit: Add support for nvfatbin added in 12.4

This commit is contained in:
Robert Maynard
2024-02-09 10:24:35 -05:00
parent c481ddda14
commit d420ee6b43
2 changed files with 26 additions and 0 deletions
@@ -0,0 +1,5 @@
FindCUDAToolkit-nvfatbin
------------------------
* The :module:`FindCUDAToolkit` module now provides a target for
``libnvfatbin`` and ``libnvfatbin_static``, if found.
+21
View File
@@ -123,6 +123,8 @@ of the following libraries that are part of the CUDAToolkit:
- :ref:`nvidia-ML<cuda_toolkit_nvML>`
- :ref:`nvPTX Compiler<cuda_toolkit_nvptx>`
- :ref:`nvRTC<cuda_toolkit_nvRTC>`
- :ref:`nvJitLink<cuda_toolkit_nvJitLink>`
- :ref:`nvFatBin<cuda_toolkit_nvfatbin>`
- :ref:`nvToolsExt<cuda_toolkit_nvToolsExt>`
- :ref:`nvtx3<cuda_toolkit_nvtx3>`
- :ref:`OpenCL<cuda_toolkit_opencl>`
@@ -418,6 +420,20 @@ Targets Created:
- ``CUDA::nvJitLink`` starting in CUDA 12.0
- ``CUDA::nvJitLink_static`` starting in CUDA 12.0
.. _`cuda_toolkit_nvfatbin`:
nvFatBin
"""""""""
.. versionadded:: 3.30
The `nvFatBin <https://docs.nvidia.com/cuda/>`_ (Runtime fatbin creation) library.
Targets Created:
- ``CUDA::nvfatbin`` starting in CUDA 12.4
- ``CUDA::nvfatbin_static`` starting in CUDA 12.4
.. _`cuda_toolkit_nvml`:
nvidia-ML
@@ -1168,6 +1184,11 @@ if(CUDAToolkit_FOUND)
_CUDAToolkit_find_and_add_import_lib(nvJitLink_static DEPS cudart_static_deps)
endif()
if(CUDAToolkit_VERSION VERSION_GREATER_EQUAL 12.4.0)
_CUDAToolkit_find_and_add_import_lib(nvfatbin DEPS cudart_static_deps)
_CUDAToolkit_find_and_add_import_lib(nvfatbin_static DEPS cudart_static_deps)
endif()
_CUDAToolkit_find_and_add_import_lib(culibos) # it's a static library
foreach (cuda_lib cublasLt cufft nvjpeg)
_CUDAToolkit_find_and_add_import_lib(${cuda_lib})