diff --git a/Modules/FindCUDAToolkit.cmake b/Modules/FindCUDAToolkit.cmake index 2485427784..ca05aee0a6 100644 --- a/Modules/FindCUDAToolkit.cmake +++ b/Modules/FindCUDAToolkit.cmake @@ -120,6 +120,7 @@ of the following libraries that are part of the CUDAToolkit: - :ref:`nvGRAPH` - :ref:`nvJPEG` - :ref:`nvidia-ML` +- :ref:`nvPTX Compiler` - :ref:`nvRTC` - :ref:`nvToolsExt` - :ref:`nvtx3` @@ -351,6 +352,22 @@ Targets Created: - ``CUDA::nvjpeg`` - ``CUDA::nvjpeg_static`` +.. _`cuda_toolkit_nvPTX`: + +nvPTX Compiler +"""""""""""""" + +.. versionadded:: 3.25 + +The `nvPTX `_ (PTX Compilation) library. +The PTX Compiler APIs are a set of APIs which can be used to compile a PTX program into GPU assembly code. +Introduced in CUDA 11.1 +This is a static library only. + +Targets Created: + +- ``CUDA::nvptxcompiler`` starting in CUDA 11.1 + .. _`cuda_toolkit_nvRTC`: nvRTC @@ -1040,6 +1057,9 @@ if(CUDAToolkit_FOUND) endif() _CUDAToolkit_find_and_add_import_lib(nvrtc DEPS cuda_driver) + if(CUDAToolkit_VERSION VERSION_GREATER_EQUAL 11.1.0) + _CUDAToolkit_find_and_add_import_lib(nvptxcompiler_static DEPS cuda_driver) + endif() _CUDAToolkit_find_and_add_import_lib(nvml ALT nvidia-ml nvml)