You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
53 lines
1.5 KiB
53 lines
1.5 KiB
#
|
|
# FindThrust
|
|
#
|
|
# This module finds the Thrust header files and extrats their version. It
|
|
# sets the following variables.
|
|
#
|
|
# THRUST_INCLUDE_DIR - Include directory for thrust header files. (All header
|
|
# files will actually be in the thrust subdirectory.)
|
|
# THRUST_VERSION - Version of thrust in the form "major.minor.patch".
|
|
#
|
|
|
|
find_path( THRUST_INCLUDE_DIR
|
|
HINTS
|
|
/usr/include/cuda
|
|
/usr/local/include
|
|
/usr/local/cuda/include
|
|
${CUDA_INCLUDE_DIRS}
|
|
NAMES thrust/version.h
|
|
DOC "Thrust headers"
|
|
)
|
|
if( THRUST_INCLUDE_DIR )
|
|
list( REMOVE_DUPLICATES THRUST_INCLUDE_DIR )
|
|
endif( THRUST_INCLUDE_DIR )
|
|
|
|
# Find thrust version
|
|
file( STRINGS ${THRUST_INCLUDE_DIR}/thrust/version.h
|
|
version
|
|
REGEX "#define THRUST_VERSION[ \t]+([0-9x]+)"
|
|
)
|
|
string( REGEX REPLACE
|
|
"#define THRUST_VERSION[ \t]+"
|
|
""
|
|
version
|
|
"${version}"
|
|
)
|
|
|
|
string( REGEX MATCH "^[0-9]" major ${version} )
|
|
string( REGEX REPLACE "^${major}00" "" version "${version}" )
|
|
string( REGEX MATCH "^[0-9]" minor ${version} )
|
|
string( REGEX REPLACE "^${minor}0" "" version "${version}" )
|
|
set( THRUST_VERSION "${major}.${minor}.${version}")
|
|
set( THRUST_MAJOR_VERSION "${major}")
|
|
set( THRUST_MINOR_VERSION "${minor}")
|
|
|
|
# Check for required components
|
|
include( FindPackageHandleStandardArgs )
|
|
find_package_handle_standard_args( Thrust
|
|
REQUIRED_VARS THRUST_INCLUDE_DIR
|
|
VERSION_VAR THRUST_VERSION
|
|
)
|
|
|
|
set(THRUST_INCLUDE_DIRS ${THRUST_INCLUDE_DIR})
|
|
mark_as_advanced(THRUST_INCLUDE_DIR)
|