1# Try to find the CUB library and headers. 2# CUB_FOUND - system has CUB 3# CUB_INCLUDE_DIRS - the CUB include directory 4 5find_path(CUB_INCLUDE_DIR 6 HINTS "${CUDA_TOOLKIT_INCLUDE}" 7 NAMES cub/cub.cuh 8 DOC "The directory where CUB includes reside" 9) 10 11set(CUB_INCLUDE_DIRS ${CUB_INCLUDE_DIR}) 12 13include(FindPackageHandleStandardArgs) 14find_package_handle_standard_args(CUB 15 FOUND_VAR CUB_FOUND 16 REQUIRED_VARS CUB_INCLUDE_DIR 17) 18 19mark_as_advanced(CUB_FOUND) 20