| # 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) |