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