wehub-resource-sync
43 lines
1.0 KiB
CMake
43 lines
1.0 KiB
CMake
##########################
|
|
# FindDPCPP.cmake
|
|
##########################
|
|
|
|
include_guard()
|
|
|
|
include(FindPackageHandleStandardArgs)
|
|
|
|
# set DPCPP_ROOT
|
|
if(IS_DIRECTORY ${CINN_WITH_SYCL})
|
|
set(DPCPP_ROOT ${CINN_WITH_SYCL})
|
|
else()
|
|
execute_process(
|
|
COMMAND which sycl-ls
|
|
TIMEOUT 2
|
|
OUTPUT_VARIABLE sycl_ls_path
|
|
OUTPUT_STRIP_TRAILING_WHITESPACE)
|
|
if(EXISTS ${sycl_ls_path})
|
|
get_filename_component(DPCPP_ROOT ${sycl_ls_path}/../.. ABSOLUTE)
|
|
else()
|
|
message(
|
|
FATAL_ERROR
|
|
"Failed in SYCL path auto search, please set CINN_WITH_SYCL to sycl root path."
|
|
)
|
|
endif()
|
|
endif()
|
|
# find libsycl.so
|
|
find_library(
|
|
DPCPP_LIB
|
|
NAMES sycl
|
|
PATHS "${DPCPP_ROOT}/lib")
|
|
find_package_handle_standard_args(
|
|
DPCPP
|
|
FOUND_VAR DPCPP_FOUND
|
|
REQUIRED_VARS DPCPP_LIB)
|
|
if(NOT DPCPP_FOUND)
|
|
return()
|
|
endif()
|
|
message(STATUS "Enable SYCL: " ${DPCPP_ROOT})
|
|
include_directories("${DPCPP_ROOT}/include/sycl;${DPCPP_ROOT}/include")
|
|
link_libraries(${DPCPP_LIB})
|
|
add_definitions(-DSYCL_CXX_COMPILER="${DPCPP_ROOT}/bin/clang++")
|