cmake/FindThrust.cmake

##=============================================================================
##
##  Copyright (c) Kitware, Inc.
##  All rights reserved.
##  See LICENSE.txt for details.
##
##  This software is distributed WITHOUT ANY WARRANTY; without even
##  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
##  PURPOSE.  See the above copyright notice for more information.
##
##  Copyright 2012 Sandia Corporation.
##  Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
##  the U.S. Government retains certain rights in this software.
##
##=============================================================================

#
# 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}
	      ./thrust
	      ../thrust
	NAMES thrust/version.h
)

if (THRUST_INCLUDE_DIR)
  set(THRUST_FOUND TRUE)
endif ()