FindOpenCL.cmake 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384
  1. # - Try to find OpenCL
  2. # This module tries to find an OpenCL implementation on your system. It supports
  3. # AMD / ATI, Apple and NVIDIA implementations.
  4. #
  5. # To set the paths manually, define these environment variables:
  6. # OpenCL_INCPATH - Include path (e.g. OpenCL_INCPATH=/opt/cuda/4.0/cuda/include)
  7. # OpenCL_LIBPATH - Library path (e.h. OpenCL_LIBPATH=/usr/lib64/nvidia)
  8. #
  9. # Once done this will define
  10. # OpenCL_FOUND - system has OpenCL
  11. # OpenCL_INCLUDE_DIRS - the OpenCL include directory
  12. # OpenCL_LIBRARIES - link these to use OpenCL
  13. FIND_PACKAGE(PackageHandleStandardArgs)
  14. SET (OpenCL_VERSION_STRING "0.1.0")
  15. SET (OpenCL_VERSION_MAJOR 0)
  16. SET (OpenCL_VERSION_MINOR 1)
  17. SET (OpenCL_VERSION_PATCH 0)
  18. IF (APPLE)
  19. FIND_LIBRARY(OpenCL_LIBRARIES OpenCL DOC "OpenCL lib for OSX")
  20. FIND_PATH(OpenCL_INCLUDE_DIRS OpenCL/cl.h DOC "Include for OpenCL on OSX")
  21. FIND_PATH(_OpenCL_CPP_INCLUDE_DIRS OpenCL/cl.hpp DOC "Include for OpenCL CPP bindings on OSX")
  22. ELSE (APPLE)
  23. IF (WIN32)
  24. # The AMD SDK currently installs both x86 and x86_64 libraries
  25. # This is only a hack to find out architecture
  26. IF( CMAKE_SIZEOF_VOID_P EQUAL 8 )
  27. SET(OpenCL_LIB_DIR "$ENV{AMDAPPSDKROOT}/lib/x86_64")
  28. ELSE( CMAKE_SIZEOF_VOID_P EQUAL 8 )
  29. SET(OpenCL_LIB_DIR "$ENV{AMDAPPSDKROOT}/lib/x86")
  30. ENDIF( CMAKE_SIZEOF_VOID_P EQUAL 8 )
  31. FIND_LIBRARY(OpenCL_LIBRARIES OpenCL.lib PATHS
  32. ${OpenCL_LIB_DIR} $ENV{OpenCL_LIBPATH} $ENV{CUDA_LIB_PATH})
  33. GET_FILENAME_COMPONENT(_OpenCL_INC_CAND ${OpenCL_LIB_DIR}/../../include ABSOLUTE)
  34. # On Win32 search relative to the library
  35. FIND_PATH(OpenCL_INCLUDE_DIRS CL/cl.h PATHS
  36. ${_OpenCL_INC_CAND} $ENV{OpenCL_INCPATH} $ENV{CUDA_INC_PATH})
  37. FIND_PATH(_OpenCL_CPP_INCLUDE_DIRS CL/cl.hpp PATHS
  38. ${_OpenCL_INC_CAND} $ENV{OpenCL_INCPATH} $ENV{CUDA_INC_PATH})
  39. ELSE (WIN32)
  40. # Unix style platforms
  41. FIND_LIBRARY(OpenCL_LIBRARIES OpenCL
  42. PATHS ENV LD_LIBRARY_PATH ENV OpenCL_LIBPATH
  43. )
  44. GET_FILENAME_COMPONENT(OpenCL_LIB_DIR ${OpenCL_LIBRARIES} PATH)
  45. GET_FILENAME_COMPONENT(_OpenCL_INC_CAND ${OpenCL_LIB_DIR}/../../include ABSOLUTE)
  46. # The AMD SDK currently does not place its headers
  47. # in /usr/include, therefore also search relative
  48. # to the library
  49. FIND_PATH(OpenCL_INCLUDE_DIRS CL/cl.h PATHS "$ENV{CUDADIR}/include" ${_OpenCL_INC_CAND} "/usr/local/cuda/include" "/opt/cuda/include" "/opt/AMDAPP/include" ENV OpenCL_INCPATH)
  50. FIND_PATH(_OpenCL_CPP_INCLUDE_DIRS CL/cl.hpp PATHS "$ENV{CUDADIR}/include" ${_OpenCL_INC_CAND} "/usr/local/cuda/include" "/opt/cuda/include" "/opt/AMDAPP/include" ENV OpenCL_INCPATH)
  51. # for Mali GPUs, libmali must be linked
  52. FIND_LIBRARY(OpenCL_LIBRARIES_MALI mali
  53. PATHS ENV LD_LIBRARY_PATH ENV OpenCL_LIBPATH "/usr/lib64" "$ENV{CUDADIR}/lib64"
  54. )
  55. IF (OpenCL_LIBRARIES_MALI)
  56. SET(OpenCL_LIBRARIES ${OpenCL_LIBRARIES} ${OpenCL_LIBRARIES_MALI})
  57. ENDIF (OpenCL_LIBRARIES_MALI)
  58. ENDIF (WIN32)
  59. ENDIF (APPLE)
  60. FIND_PACKAGE_HANDLE_STANDARD_ARGS(OpenCL DEFAULT_MSG OpenCL_LIBRARIES OpenCL_INCLUDE_DIRS)
  61. IF(_OpenCL_CPP_INCLUDE_DIRS)
  62. SET( OpenCL_HAS_CPP_BINDINGS TRUE )
  63. LIST( APPEND OpenCL_INCLUDE_DIRS ${_OpenCL_CPP_INCLUDE_DIRS} )
  64. # This is often the same, so clean up
  65. LIST( REMOVE_DUPLICATES OpenCL_INCLUDE_DIRS )
  66. ENDIF(_OpenCL_CPP_INCLUDE_DIRS)
  67. MARK_AS_ADVANCED( OpenCL_INCLUDE_DIRS )