20 lines
494 B
CMake
20 lines
494 B
CMake
# Try to find the CUB library and headers.
|
|
# CUB_FOUND - system has CUB
|
|
# CUB_INCLUDE_DIRS - the CUB include directory
|
|
|
|
find_path(CUB_INCLUDE_DIR
|
|
HINTS "${CUDA_TOOLKIT_INCLUDE}"
|
|
NAMES cub/cub.cuh
|
|
DOC "The directory where CUB includes reside"
|
|
)
|
|
|
|
set(CUB_INCLUDE_DIRS ${CUB_INCLUDE_DIR})
|
|
|
|
include(FindPackageHandleStandardArgs)
|
|
find_package_handle_standard_args(CUB
|
|
FOUND_VAR CUB_FOUND
|
|
REQUIRED_VARS CUB_INCLUDE_DIR
|
|
)
|
|
|
|
mark_as_advanced(CUB_FOUND)
|