43 lines
1.1 KiB
CMake
43 lines
1.1 KiB
CMake
|
function(_thrust_find_thrust_multiconfig)
|
||
|
|
# Check which systems are enabled by multiconfig:
|
||
|
|
set(req_systems)
|
||
|
|
if (THRUST_MULTICONFIG_ENABLE_SYSTEM_CUDA)
|
||
|
|
list(APPEND req_systems CUDA)
|
||
|
|
endif()
|
||
|
|
if (THRUST_MULTICONFIG_ENABLE_SYSTEM_CPP)
|
||
|
|
list(APPEND req_systems CPP)
|
||
|
|
endif()
|
||
|
|
if (THRUST_MULTICONFIG_ENABLE_SYSTEM_TBB)
|
||
|
|
list(APPEND req_systems TBB)
|
||
|
|
endif()
|
||
|
|
if (THRUST_MULTICONFIG_ENABLE_SYSTEM_OMP)
|
||
|
|
list(APPEND req_systems OMP)
|
||
|
|
endif()
|
||
|
|
|
||
|
|
find_package(
|
||
|
|
Thrust
|
||
|
|
REQUIRED
|
||
|
|
CONFIG
|
||
|
|
NO_DEFAULT_PATH # Only check the explicit path in HINTS:
|
||
|
|
HINTS "${CCCL_SOURCE_DIR}/lib/cmake/thrust/"
|
||
|
|
COMPONENTS ${req_systems}
|
||
|
|
)
|
||
|
|
endfunction()
|
||
|
|
|
||
|
|
function(_thrust_find_thrust_singleconfig)
|
||
|
|
cccl_get_thrust()
|
||
|
|
|
||
|
|
# Create target now to prepare system found flags:
|
||
|
|
thrust_create_target(thrust.config FROM_OPTIONS ${THRUST_TARGET_FLAGS})
|
||
|
|
thrust_debug_target(thrust.config "${THRUST_VERSION}")
|
||
|
|
endfunction()
|
||
|
|
|
||
|
|
# Find thrust along with all requested backends.
|
||
|
|
function(thrust_find_thrust)
|
||
|
|
if (THRUST_ENABLE_MULTICONFIG)
|
||
|
|
_thrust_find_thrust_multiconfig()
|
||
|
|
else()
|
||
|
|
_thrust_find_thrust_singleconfig()
|
||
|
|
endif()
|
||
|
|
endfunction()
|