diff options
| author | Boris Fomitchev <bfomitchev@nvidia.com> | 2016-05-13 03:17:07 (GMT) |
|---|---|---|
| committer | Brad King <brad.king@kitware.com> | 2016-05-19 14:52:32 (GMT) |
| commit | cc95001b1cc20367ea127ea09ea91960782e65a3 (patch) | |
| tree | efdd5356e029549ea24504a0c186f66de30de81f /Modules | |
| parent | e0da6c3b562f7fd25b83b00c432f016439c24c22 (diff) | |
| download | CMake-cc95001b1cc20367ea127ea09ea91960782e65a3.zip CMake-cc95001b1cc20367ea127ea09ea91960782e65a3.tar.gz CMake-cc95001b1cc20367ea127ea09ea91960782e65a3.tar.bz2 | |
FindCUDA: Detect and report FP16 support
Reviewed-by: James Bigler <jamesbigler@gmail.com>
Diffstat (limited to 'Modules')
| -rw-r--r-- | Modules/FindCUDA.cmake | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/Modules/FindCUDA.cmake b/Modules/FindCUDA.cmake index 4f52d25..0718116 100644 --- a/Modules/FindCUDA.cmake +++ b/Modules/FindCUDA.cmake @@ -264,6 +264,7 @@ # CUDA_VERSION_MINOR -- The minor version. # CUDA_VERSION # CUDA_VERSION_STRING -- CUDA_VERSION_MAJOR.CUDA_VERSION_MINOR +# CUDA_HAS_FP16 -- Whether a short float (float16,fp16) is supported. # # CUDA_TOOLKIT_ROOT_DIR -- Path to the CUDA Toolkit (defined if not set). # CUDA_SDK_ROOT_DIR -- Path to the CUDA SDK. Use this to find files in the @@ -683,6 +684,12 @@ find_path(CUDA_TOOLKIT_INCLUDE find_path(CUDA_TOOLKIT_INCLUDE device_functions.h) mark_as_advanced(CUDA_TOOLKIT_INCLUDE) +if (CUDA_VERSION VERSION_GREATER "7.0" OR EXISTS "${CUDA_TOOLKIT_INCLUDE}/cuda_fp16.h") + set(CUDA_HAS_FP16 TRUE) +else() + set(CUDA_HAS_FP16 FALSE) +endif() + # Set the user list of include dir to nothing to initialize it. set (CUDA_NVCC_INCLUDE_ARGS_USER "") set (CUDA_INCLUDE_DIRS ${CUDA_TOOLKIT_INCLUDE}) |