You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
85 lines
3.1 KiB
85 lines
3.1 KiB
cmake_minimum_required(VERSION 3.15)
|
|
project(Toolkit CXX)
|
|
|
|
#Goal for this example:
|
|
# Validate that we can use CUDAToolkit to find cuda include paths
|
|
find_package(CUDAToolkit REQUIRED)
|
|
|
|
if(NOT DEFINED CUDAToolkit_VERSION)
|
|
message(FATAL_ERROR "expected CUDAToolkit variable CUDAToolkit_VERSION not found")
|
|
endif()
|
|
|
|
message(STATUS "CUDAToolkit_VERSION: ${CUDAToolkit_VERSION}")
|
|
message(STATUS "CUDAToolkit_VERSION_MAJOR: ${CUDAToolkit_VERSION_MAJOR}")
|
|
message(STATUS "CUDAToolkit_VERSION_MINOR: ${CUDAToolkit_VERSION_MINOR}")
|
|
message(STATUS "CUDAToolkit_VERSION_PATCH: ${CUDAToolkit_VERSION_PATCH}")
|
|
message(STATUS "CUDAToolkit_BIN_DIR: ${CUDAToolkit_BIN_DIR}")
|
|
message(STATUS "CUDAToolkit_INCLUDE_DIRS: ${CUDAToolkit_INCLUDE_DIRS}")
|
|
message(STATUS "CUDAToolkit_LIBRARY_DIR: ${CUDAToolkit_LIBRARY_DIR}")
|
|
message(STATUS "CUDAToolkit_LIBRARY_ROOT: ${CUDAToolkit_LIBRARY_ROOT}")
|
|
message(STATUS "CUDAToolkit_NVCC_EXECUTABLE ${CUDAToolkit_NVCC_EXECUTABLE}")
|
|
|
|
set(should_exist
|
|
CUDAToolkit_BIN_DIR
|
|
CUDAToolkit_INCLUDE_DIRS
|
|
CUDAToolkit_LIBRARY_DIR
|
|
CUDAToolkit_LIBRARY_ROOT
|
|
)
|
|
foreach (cuda_loc_var IN LISTS should_exist)
|
|
if(NOT EXISTS "${${cuda_loc_var}}")
|
|
message(FATAL_ERROR "${cuda_loc_var} variable is expected to be set to valid path")
|
|
endif()
|
|
endforeach()
|
|
|
|
|
|
set(cuda_libs cudart cuda_driver cublas cufft cufftw curand cusolver cusparse)
|
|
if(CUDAToolkit_VERSION VERSION_GREATER_EQUAL 10.1)
|
|
list(APPEND cuda_libs cublasLt)
|
|
endif()
|
|
if(CUDAToolkit_VERSION_MAJOR VERSION_GREATER 11)
|
|
list(APPEND cuda_libs nvJitLink)
|
|
endif()
|
|
if(CUDAToolkit_VERSION_MAJOR VERSION_LESS 11)
|
|
list(APPEND cuda_libs nvgraph)
|
|
endif()
|
|
|
|
# Verify that all the CUDA:: targets exist even when the CUDA language isn't enabled
|
|
foreach (cuda_lib IN LISTS cuda_libs)
|
|
if(NOT CUDA_${cuda_lib}_LIBRARY)
|
|
message(FATAL_ERROR "expected CUDAToolkit variable CUDA_${cuda_lib}_LIBRARY not found")
|
|
elseif(CUDA_${cuda_lib}_LIBRARY MATCHES [[\.\./]])
|
|
message(FATAL_ERROR "expected CUDAToolkit variable CUDA_${cuda_lib}_LIBRARY[\"${CUDA_${cuda_lib}_LIBRARY}\"] to not contain /..")
|
|
endif()
|
|
if(NOT TARGET CUDA::${cuda_lib})
|
|
message(FATAL_ERROR "expected CUDAToolkit target CUDA::${cuda_lib} not found")
|
|
endif()
|
|
endforeach()
|
|
|
|
set(npp_libs nppc nppial nppicc nppidei nppif nppig nppim nppist nppitc npps nppisu)
|
|
if(CUDAToolkit_VERSION_MAJOR VERSION_LESS 11)
|
|
list(APPEND npp_libs nppicom)
|
|
endif()
|
|
foreach (cuda_lib IN LISTS npp_libs)
|
|
if(CUDA_${cuda_lib}_LIBRARY MATCHES [[\.\./]])
|
|
message(FATAL_ERROR "expected CUDAToolkit variable CUDA_${cuda_lib}_LIBRARY[\"${CUDA_${cuda_lib}_LIBRARY}\"] to not contain /..")
|
|
endif()
|
|
if(NOT TARGET CUDA::${cuda_lib})
|
|
message(FATAL_ERROR "The CUDA::${cuda_lib} target was expected but couldn't be found")
|
|
endif()
|
|
endforeach()
|
|
|
|
foreach (cuda_lib nvrtc nvToolsExt OpenCL)
|
|
if(NOT TARGET CUDA::${cuda_lib})
|
|
message(FATAL_ERROR "The CUDA::${cuda_lib} target was expected but couldn't be found")
|
|
endif()
|
|
endforeach()
|
|
|
|
add_executable(Toolkit main.cpp)
|
|
target_link_libraries(Toolkit PRIVATE CUDA::toolkit)
|
|
|
|
# cupti is an optional component of the CUDA toolkit
|
|
if(TARGET CUDA::cupti)
|
|
add_executable(cupti cupti.cpp)
|
|
target_link_libraries(cupti PRIVATE CUDA::cupti)
|
|
endif()
|