FindCUDAToolkit: Add support for finding Tegra cudla library

Fixes #24575
This commit is contained in:
Robert Maynard
2023-03-07 13:47:16 -05:00
parent 3d6075da4d
commit 21b102c77d
2 changed files with 23 additions and 0 deletions

View File

@@ -0,0 +1,4 @@
FindCUDAToolkit-target-for-cudla
--------------------------------
* The :module:`FindCUDAToolkit` module now provides an imported target for ``cudla``, if found.

View File

@@ -109,6 +109,7 @@ of the following libraries that are part of the CUDAToolkit:
- :ref:`CUDA Runtime Library<cuda_toolkit_rt_lib>`
- :ref:`CUDA Driver Library<cuda_toolkit_driver_lib>`
- :ref:`cuBLAS<cuda_toolkit_cuBLAS>`
- :ref:`cuDLA<cuda_toolkit_cuDLA>`
- :ref:`cuFile<cuda_toolkit_cuFile>`
- :ref:`cuFFT<cuda_toolkit_cuFFT>`
- :ref:`cuRAND<cuda_toolkit_cuRAND>`
@@ -166,6 +167,19 @@ Targets Created:
- ``CUDA::cublasLt`` starting in CUDA 10.1
- ``CUDA::cublasLt_static`` starting in CUDA 10.1
.. _`cuda_toolkit_cuDLA`:
cuDLA
""""""
.. versionadded:: 3.27
The NVIDIA Tegra Deep Learning Accelerator `cuDLA <https://docs.nvidia.com/cuda/cublas/index.html>`_ library.
Targets Created:
- ``CUDA::cudla`` starting in CUDA 11.6
.. _`cuda_toolkit_cuFile`:
cuFile
@@ -1046,6 +1060,11 @@ if(CUDAToolkit_FOUND)
_CUDAToolkit_find_and_add_import_lib(cuFile_rdma_static DEPS cuFile_static culibos)
endif()
if(CUDAToolkit_VERSION VERSION_GREATER_EQUAL 11.6)
_CUDAToolkit_find_and_add_import_lib(cudla)
endif()
# cuFFTW depends on cuFFT
_CUDAToolkit_find_and_add_import_lib(cufftw DEPS cufft)
_CUDAToolkit_find_and_add_import_lib(cufftw_static DEPS cufft_static)