diff options
| author | Brad King <brad.king@kitware.com> | 2023-03-08 13:58:38 (GMT) |
|---|---|---|
| committer | Kitware Robot <kwrobot@kitware.com> | 2023-03-08 13:58:46 (GMT) |
| commit | 9942115714ca521c5d4c0aa4141706bc8e3b35b5 (patch) | |
| tree | a6e1ff068ec7c6ea8e2eb6be86481b6f1fd1858d | |
| parent | 9b44cc5febe2260f5e365b913185c093b03116f1 (diff) | |
| parent | 21b102c77d85897c2500488180e58de077447b4c (diff) | |
| download | CMake-9942115714ca521c5d4c0aa4141706bc8e3b35b5.zip CMake-9942115714ca521c5d4c0aa4141706bc8e3b35b5.tar.gz CMake-9942115714ca521c5d4c0aa4141706bc8e3b35b5.tar.bz2 | |
Merge topic 'find_cudatoolkit_add_cudla_targets'
21b102c77d FindCUDAToolkit: Add support for finding Tegra cudla library
Acked-by: Kitware Robot <kwrobot@kitware.com>
Merge-request: !8295
| -rw-r--r-- | Help/release/dev/FindCUDAToolkit-target-for-cudla.rst | 4 | ||||
| -rw-r--r-- | Modules/FindCUDAToolkit.cmake | 19 |
2 files changed, 23 insertions, 0 deletions
diff --git a/Help/release/dev/FindCUDAToolkit-target-for-cudla.rst b/Help/release/dev/FindCUDAToolkit-target-for-cudla.rst new file mode 100644 index 0000000..9de456e --- /dev/null +++ b/Help/release/dev/FindCUDAToolkit-target-for-cudla.rst @@ -0,0 +1,4 @@ +FindCUDAToolkit-target-for-cudla +-------------------------------- + +* The :module:`FindCUDAToolkit` module now provides an imported target for ``cudla``, if found. diff --git a/Modules/FindCUDAToolkit.cmake b/Modules/FindCUDAToolkit.cmake index d7c6d5e..713f3b8 100644 --- a/Modules/FindCUDAToolkit.cmake +++ b/Modules/FindCUDAToolkit.cmake @@ -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) |