mirror of
https://gitlab.com/libeigen/eigen.git
synced 2024-12-21 07:19:46 +08:00
229 lines
8.5 KiB
CMake
229 lines
8.5 KiB
CMake
|
#.rst:
|
||
|
|
# FindComputeCpp
|
||
|
|
#---------------
|
||
|
|
|
||
|
|
#########################
|
||
|
|
# FindComputeCpp.cmake
|
||
|
|
#########################
|
||
|
|
#
|
||
|
|
# Tools for finding and building with ComputeCpp.
|
||
|
|
#
|
||
|
|
|
||
|
|
# Require CMake version 3.2.2 or higher
|
||
|
|
cmake_minimum_required(VERSION 3.2.2)
|
||
|
|
|
||
|
|
# Check that a supported host compiler can be found
|
||
|
|
if(CMAKE_COMPILER_IS_GNUCXX)
|
||
|
|
# Require at least gcc 4.8
|
||
|
|
if (CMAKE_CXX_COMPILER_VERSION VERSION_LESS 4.8)
|
||
|
|
message(FATAL_ERROR
|
||
|
|
"host compiler - Not found! (gcc version must be at least 4.8)")
|
||
|
|
# Require the GCC dual ABI to be disabled for 5.1 or higher
|
||
|
|
elseif (CMAKE_CXX_COMPILER_VERSION VERSION_GREATER 5.1)
|
||
|
|
set(COMPUTECPP_DISABLE_GCC_DUAL_ABI "True")
|
||
|
|
message(STATUS
|
||
|
|
"host compiler - gcc ${CMAKE_CXX_COMPILER_VERSION} (note pre 5.1 gcc ABI enabled)")
|
||
|
|
else()
|
||
|
|
message(STATUS "host compiler - gcc ${CMAKE_CXX_COMPILER_VERSION}")
|
||
|
|
endif()
|
||
|
|
elseif ("${CMAKE_CXX_COMPILER_ID}" STREQUAL "Clang")
|
||
|
|
# Require at least clang 3.6
|
||
|
|
if (${CMAKE_CXX_COMPILER_VERSION} VERSION_LESS 3.6)
|
||
|
|
message(FATAL_ERROR
|
||
|
|
"host compiler - Not found! (clang version must be at least 3.6)")
|
||
|
|
else()
|
||
|
|
set(COMPUTECPP_DISABLE_GCC_DUAL_ABI "True")
|
||
|
|
message(STATUS "host compiler - clang ${CMAKE_CXX_COMPILER_VERSION}")
|
||
|
|
endif()
|
||
|
|
else()
|
||
|
|
message(WARNING
|
||
|
|
"host compiler - Not found! (ComputeCpp supports GCC and Clang, see readme)")
|
||
|
|
endif()
|
||
|
|
|
||
|
|
set(COMPUTECPP_64_BIT_DEFAULT ON)
|
||
|
|
option(COMPUTECPP_64_BIT_CODE "Compile device code in 64 bit mode"
|
||
|
|
${COMPUTECPP_64_BIT_DEFAULT})
|
||
|
|
mark_as_advanced(COMPUTECPP_64_BIT_CODE)
|
||
|
|
|
||
|
|
# Find OpenCL package
|
||
|
|
find_package(OpenCL REQUIRED)
|
||
|
|
|
||
|
|
# Find ComputeCpp package
|
||
|
|
if(EXISTS ${COMPUTECPP_PACKAGE_ROOT_DIR})
|
||
|
|
message(STATUS "ComputeCpp package - Found (${COMPUTECPP_PACKAGE_ROOT_DIR})")
|
||
|
|
else()
|
||
|
|
message(FATAL_ERROR "ComputeCpp package - Not found! (please set COMPUTECPP_PACKAGE_ROOT_DIR) (${COMPUTECPP_PACKAGE_ROOT_DIR})")
|
||
|
|
endif()
|
||
|
|
option(COMPUTECPP_PACKAGE_ROOT_DIR "Path to the ComputeCpp Package")
|
||
|
|
|
||
|
|
# Obtain the path to compute++
|
||
|
|
find_program(COMPUTECPP_DEVICE_COMPILER compute++ PATHS
|
||
|
|
${COMPUTECPP_PACKAGE_ROOT_DIR} PATH_SUFFIXES bin)
|
||
|
|
if (EXISTS ${COMPUTECPP_DEVICE_COMPILER})
|
||
|
|
mark_as_advanced(COMPUTECPP_DEVICE_COMPILER)
|
||
|
|
message(STATUS "compute++ - Found (${COMPUTECPP_PACKAGE_ROOT_DIR})")
|
||
|
|
else()
|
||
|
|
message(FATAL_ERROR "compute++ - Not found! (${COMPUTECPP_DEVICE_COMPILER}) (${COMPUTECPP_PACKAGE_ROOT_DIR})")
|
||
|
|
endif()
|
||
|
|
|
||
|
|
# Obtain the path to computecpp_info
|
||
|
|
find_program(COMPUTECPP_INFO_TOOL computecpp_info PATHS
|
||
|
|
${COMPUTECPP_PACKAGE_ROOT_DIR} PATH_SUFFIXES bin)
|
||
|
|
if (EXISTS ${COMPUTECPP_INFO_TOOL})
|
||
|
|
mark_as_advanced(${COMPUTECPP_INFO_TOOL})
|
||
|
|
message(STATUS "computecpp_info - Found (${COMPUTECPP_PACKAGE_ROOT_DIR})")
|
||
|
|
else()
|
||
|
|
message(FATAL_ERROR "computecpp_info - Not found! (${COMPUTECPP_INFO_TOOL}) (${COMPUTECPP_PACKAGE_ROOT_DIR})")
|
||
|
|
endif()
|
||
|
|
|
||
|
|
# Obtain the path to the ComputeCpp runtime library
|
||
|
|
find_library(COMPUTECPP_RUNTIME_LIBRARY ComputeCpp PATHS ${COMPUTECPP_PACKAGE_ROOT_DIR}
|
||
|
|
HINTS ${COMPUTECPP_PACKAGE_ROOT_DIR}/lib PATH_SUFFIXES lib
|
||
|
|
DOC "ComputeCpp Runtime Library" NO_DEFAULT_PATH)
|
||
|
|
|
||
|
|
if (EXISTS ${COMPUTECPP_RUNTIME_LIBRARY})
|
||
|
|
mark_as_advanced(COMPUTECPP_RUNTIME_LIBRARY)
|
||
|
|
message(STATUS "libComputeCpp.so - Found")
|
||
|
|
else()
|
||
|
|
message(FATAL_ERROR "libComputeCpp.so - Not found! (${COMPUTECPP_PACKAGE_ROOT_DIR})")
|
||
|
|
endif()
|
||
|
|
|
||
|
|
# Obtain the ComputeCpp include directory
|
||
|
|
set(COMPUTECPP_INCLUDE_DIRECTORY ${COMPUTECPP_PACKAGE_ROOT_DIR}/include/)
|
||
|
|
if (NOT EXISTS ${COMPUTECPP_INCLUDE_DIRECTORY})
|
||
|
|
message(FATAL_ERROR "ComputeCpp includes - Not found! (${COMPUTECPP_PACKAGE_ROOT_DIR}/include/)")
|
||
|
|
else()
|
||
|
|
message(STATUS "ComputeCpp includes - Found (${COMPUTECPP_PACKAGE_ROOT_DIR})")
|
||
|
|
endif()
|
||
|
|
|
||
|
|
# Obtain the package version
|
||
|
|
execute_process(COMMAND ${COMPUTECPP_INFO_TOOL} "--dump-version"
|
||
|
|
OUTPUT_VARIABLE COMPUTECPP_PACKAGE_VERSION
|
||
|
|
RESULT_VARIABLE COMPUTECPP_INFO_TOOL_RESULT OUTPUT_STRIP_TRAILING_WHITESPACE)
|
||
|
|
if(NOT COMPUTECPP_INFO_TOOL_RESULT EQUAL "0")
|
||
|
|
message(FATAL_ERROR "Package version - Error obtaining version!")
|
||
|
|
else()
|
||
|
|
mark_as_advanced(COMPUTECPP_PACKAGE_VERSION)
|
||
|
|
message(STATUS "Package version - ${COMPUTECPP_PACKAGE_VERSION}")
|
||
|
|
endif()
|
||
|
|
|
||
|
|
# Obtain the device compiler flags
|
||
|
|
execute_process(COMMAND ${COMPUTECPP_INFO_TOOL} "--dump-device-compiler-flags"
|
||
|
|
OUTPUT_VARIABLE COMPUTECPP_DEVICE_COMPILER_FLAGS
|
||
|
|
RESULT_VARIABLE COMPUTECPP_INFO_TOOL_RESULT OUTPUT_STRIP_TRAILING_WHITESPACE)
|
||
|
|
if(NOT COMPUTECPP_INFO_TOOL_RESULT EQUAL "0")
|
||
|
|
message(FATAL_ERROR "compute++ flags - Error obtaining compute++ flags!")
|
||
|
|
else()
|
||
|
|
mark_as_advanced(COMPUTECPP_COMPILER_FLAGS)
|
||
|
|
message(STATUS "compute++ flags - ${COMPUTECPP_DEVICE_COMPILER_FLAGS}")
|
||
|
|
endif()
|
||
|
|
|
||
|
|
set(COMPUTECPP_DEVICE_COMPILER_FLAGS ${COMPUTECPP_DEVICE_COMPILER_FLAGS} -sycl-compress-name -no-serial-memop -DEIGEN_NO_ASSERTION_CHECKING=1)
|
||
|
|
|
||
|
|
# Check if the platform is supported
|
||
|
|
execute_process(COMMAND ${COMPUTECPP_INFO_TOOL} "--dump-is-supported"
|
||
|
|
OUTPUT_VARIABLE COMPUTECPP_PLATFORM_IS_SUPPORTED
|
||
|
|
RESULT_VARIABLE COMPUTECPP_INFO_TOOL_RESULT OUTPUT_STRIP_TRAILING_WHITESPACE)
|
||
|
|
if(NOT COMPUTECPP_INFO_TOOL_RESULT EQUAL "0")
|
||
|
|
message(FATAL_ERROR "platform - Error checking platform support!")
|
||
|
|
else()
|
||
|
|
mark_as_advanced(COMPUTECPP_PLATFORM_IS_SUPPORTED)
|
||
|
|
if (COMPUTECPP_PLATFORM_IS_SUPPORTED)
|
||
|
|
message(STATUS "platform - your system can support ComputeCpp")
|
||
|
|
else()
|
||
|
|
message(STATUS "platform - your system CANNOT support ComputeCpp")
|
||
|
|
endif()
|
||
|
|
endif()
|
||
|
|
|
||
|
|
####################
|
||
|
|
# __build_sycl
|
||
|
|
####################
|
||
|
|
#
|
||
|
|
# Adds a custom target for running compute++ and adding a dependency for the
|
||
|
|
# resulting integration header.
|
||
|
|
#
|
||
|
|
# targetName : Name of the target.
|
||
|
|
# sourceFile : Source file to be compiled.
|
||
|
|
# binaryDir : Intermediate output directory for the integration header.
|
||
|
|
#
|
||
|
|
function(__build_spir targetName sourceFile binaryDir)
|
||
|
|
|
||
|
|
# Retrieve source file name.
|
||
|
|
get_filename_component(sourceFileName ${sourceFile} NAME)
|
||
|
|
|
||
|
|
# Set the path to the Sycl file.
|
||
|
|
set(outputSyclFile ${binaryDir}/${sourceFileName}.sycl)
|
||
|
|
|
||
|
|
# Add any user-defined include to the device compiler
|
||
|
|
get_property(includeDirectories DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR} PROPERTY
|
||
|
|
INCLUDE_DIRECTORIES)
|
||
|
|
set(device_compiler_includes "")
|
||
|
|
foreach(directory ${includeDirectories})
|
||
|
|
set(device_compiler_includes "-I${directory}" ${device_compiler_includes})
|
||
|
|
endforeach()
|
||
|
|
if (CMAKE_INCLUDE_PATH)
|
||
|
|
foreach(directory ${CMAKE_INCLUDE_PATH})
|
||
|
|
set(device_compiler_includes "-I${directory}"
|
||
|
|
${device_compiler_includes})
|
||
|
|
endforeach()
|
||
|
|
endif()
|
||
|
|
|
||
|
|
# Convert argument list format
|
||
|
|
separate_arguments(COMPUTECPP_DEVICE_COMPILER_FLAGS)
|
||
|
|
|
||
|
|
# Add custom command for running compute++
|
||
|
|
add_custom_command(
|
||
|
|
OUTPUT ${outputSyclFile}
|
||
|
|
COMMAND ${COMPUTECPP_DEVICE_COMPILER}
|
||
|
|
${COMPUTECPP_DEVICE_COMPILER_FLAGS}
|
||
|
|
-I${COMPUTECPP_INCLUDE_DIRECTORY}
|
||
|
|
${COMPUTECPP_PLATFORM_SPECIFIC_ARGS}
|
||
|
|
${device_compiler_includes}
|
||
|
|
-o ${outputSyclFile}
|
||
|
|
-c ${CMAKE_CURRENT_SOURCE_DIR}/${sourceFile}
|
||
|
|
DEPENDS ${sourceFile}
|
||
|
|
COMMENT "Building ComputeCpp integration header file ${outputSyclFile}")
|
||
|
|
|
||
|
|
# Add a custom target for the generated integration header
|
||
|
|
add_custom_target(${targetName}_integration_header DEPENDS ${outputSyclFile})
|
||
|
|
|
||
|
|
# Add a dependency on the integration header
|
||
|
|
add_dependencies(${targetName} ${targetName}_integration_header)
|
||
|
|
|
||
|
|
# Force inclusion of the integration header for the host compiler
|
||
|
|
#set(compileFlags -include ${include_file} "-Wall")
|
||
|
|
target_compile_options(${targetName} PUBLIC ${compileFlags})
|
||
|
|
|
||
|
|
# Set the host compiler C++ standard to C++11
|
||
|
|
set_property(TARGET ${targetName} PROPERTY CXX_STANDARD 11)
|
||
|
|
|
||
|
|
# Disable GCC dual ABI on GCC 5.1 and higher
|
||
|
|
if(COMPUTECPP_DISABLE_GCC_DUAL_ABI)
|
||
|
|
set_property(TARGET ${targetName} APPEND PROPERTY COMPILE_DEFINITIONS
|
||
|
|
"_GLIBCXX_USE_CXX11_ABI=0")
|
||
|
|
endif()
|
||
|
|
|
||
|
|
endfunction()
|
||
|
|
|
||
|
|
#######################
|
||
|
|
# add_sycl_to_target
|
||
|
|
#######################
|
||
|
|
#
|
||
|
|
# Adds a SYCL compilation custom command associated with an existing
|
||
|
|
# target and sets a dependency on that new command.
|
||
|
|
#
|
||
|
|
# targetName : Name of the target to add a SYCL to.
|
||
|
|
# sourceFile : Source file to be compiled for SYCL.
|
||
|
|
# binaryDir : Intermediate output directory for the integration header.
|
||
|
|
#
|
||
|
|
function(add_sycl_to_target targetName sourceFile binaryDir)
|
||
|
|
|
||
|
|
# Add custom target to run compute++ and generate the integration header
|
||
|
|
__build_spir(${targetName} ${sourceFile} ${binaryDir})
|
||
|
|
|
||
|
|
# Link with the ComputeCpp runtime library
|
||
|
|
target_link_libraries(${targetName} PUBLIC ${COMPUTECPP_RUNTIME_LIBRARY}
|
||
|
|
PUBLIC ${OpenCL_LIBRARIES})
|
||
|
|
|
||
|
|
endfunction(add_sycl_to_target)
|