132 lines
4.1 KiB
CMake
132 lines
4.1 KiB
CMake
|
# This will define the following variables:
|
|||
|
|
# SYCL_FOUND : True if the system has the SYCL library.
|
|||
|
|
# SYCL_INCLUDE_DIR : Include directories needed to use SYCL.
|
|||
|
|
# SYCL_LIBRARY_DIR ：The path to the SYCL library.
|
|||
|
|
# SYCL_LIBRARY : SYCL library fullname.
|
|||
|
|
# SYCL_COMPILER_VERSION : SYCL compiler version.
|
|||
|
|
|
|||
|
|
include(FindPackageHandleStandardArgs)
|
|||
|
|
|
|||
|
|
set(SYCL_ROOT "")
|
|||
|
|
if(DEFINED ENV{SYCL_ROOT})
|
|||
|
|
set(SYCL_ROOT $ENV{SYCL_ROOT})
|
|||
|
|
elseif(DEFINED ENV{CMPLR_ROOT})
|
|||
|
|
set(SYCL_ROOT $ENV{CMPLR_ROOT})
|
|||
|
|
endif()
|
|||
|
|
|
|||
|
|
string(COMPARE EQUAL "${SYCL_ROOT}" "" nosyclfound)
|
|||
|
|
if(nosyclfound)
|
|||
|
|
set(SYCL_FOUND False)
|
|||
|
|
set(SYCL_REASON_FAILURE "SYCL library not set!!")
|
|||
|
|
set(SYCL_NOT_FOUND_MESSAGE "${SYCL_REASON_FAILURE}")
|
|||
|
|
return()
|
|||
|
|
endif()
|
|||
|
|
|
|||
|
|
# Find SYCL compiler executable.
|
|||
|
|
find_program(
|
|||
|
|
SYCL_COMPILER
|
|||
|
|
NAMES icx
|
|||
|
|
PATHS "${SYCL_ROOT}"
|
|||
|
|
PATH_SUFFIXES bin bin64
|
|||
|
|
NO_DEFAULT_PATH
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
function(parse_sycl_compiler_version version_number)
|
|||
|
|
# Execute the SYCL compiler with the --version flag to match the version string.
|
|||
|
|
execute_process(COMMAND ${SYCL_COMPILER} --version OUTPUT_VARIABLE SYCL_VERSION_STRING)
|
|||
|
|
string(REGEX REPLACE "Intel\\(R\\) (.*) Compiler ([0-9]+\\.[0-9]+\\.[0-9]+) (.*)" "\\2"
|
|||
|
|
SYCL_VERSION_STRING_MATCH ${SYCL_VERSION_STRING})
|
|||
|
|
string(REPLACE "." ";" SYCL_VERSION_LIST ${SYCL_VERSION_STRING_MATCH})
|
|||
|
|
# Split the version number list into major, minor, and patch components.
|
|||
|
|
list(GET SYCL_VERSION_LIST 0 VERSION_MAJOR)
|
|||
|
|
list(GET SYCL_VERSION_LIST 1 VERSION_MINOR)
|
|||
|
|
list(GET SYCL_VERSION_LIST 2 VERSION_PATCH)
|
|||
|
|
# Calculate the version number in the format XXXXYYZZ, using the formula (major * 10000 + minor * 100 + patch).
|
|||
|
|
math(EXPR VERSION_NUMBER_MATCH "${VERSION_MAJOR} * 10000 + ${VERSION_MINOR} * 100 + ${VERSION_PATCH}")
|
|||
|
|
set(${version_number} "${VERSION_NUMBER_MATCH}" PARENT_SCOPE)
|
|||
|
|
endfunction()
|
|||
|
|
|
|||
|
|
parse_sycl_compiler_version(SYCL_COMPILER_VERSION)
|
|||
|
|
|
|||
|
|
if(NOT SYCL_COMPILER_VERSION)
|
|||
|
|
set(SYCL_FOUND False)
|
|||
|
|
set(SYCL_REASON_FAILURE "Cannot parse sycl compiler version to get SYCL_COMPILER_VERSION!")
|
|||
|
|
set(SYCL_NOT_FOUND_MESSAGE "${SYCL_REASON_FAILURE}")
|
|||
|
|
return()
|
|||
|
|
endif()
|
|||
|
|
|
|||
|
|
# Find include path from binary.
|
|||
|
|
find_file(
|
|||
|
|
SYCL_INCLUDE_DIR
|
|||
|
|
NAMES include
|
|||
|
|
HINTS ${SYCL_ROOT}
|
|||
|
|
NO_DEFAULT_PATH
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# Find include/sycl path from include path.
|
|||
|
|
find_file(
|
|||
|
|
SYCL_INCLUDE_SYCL_DIR
|
|||
|
|
NAMES sycl
|
|||
|
|
HINTS ${SYCL_ROOT}/include/
|
|||
|
|
NO_DEFAULT_PATH
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# Due to the unrecognized compilation option `-fsycl` in other compiler.
|
|||
|
|
list(APPEND SYCL_INCLUDE_DIR ${SYCL_INCLUDE_SYCL_DIR})
|
|||
|
|
|
|||
|
|
# Find library directory from binary.
|
|||
|
|
find_file(
|
|||
|
|
SYCL_LIBRARY_DIR
|
|||
|
|
NAMES lib lib64
|
|||
|
|
HINTS ${SYCL_ROOT}
|
|||
|
|
NO_DEFAULT_PATH
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# Define the old version of SYCL toolkit that is compatible with the current version of PyTorch.
|
|||
|
|
set(PYTORCH_2_5_SYCL_TOOLKIT_VERSION 20249999)
|
|||
|
|
|
|||
|
|
# By default, we use libsycl.so on Linux and sycl.lib on Windows as the SYCL library name.
|
|||
|
|
if (SYCL_COMPILER_VERSION VERSION_LESS_EQUAL PYTORCH_2_5_SYCL_TOOLKIT_VERSION)
|
|||
|
|
# Don't use if(LINUX) here since this requires cmake>=3.25 and file is installed
|
|||
|
|
# and used by other projects.
|
|||
|
|
# See: https://cmake.org/cmake/help/v3.25/variable/LINUX.html
|
|||
|
|
if(CMAKE_SYSTEM_NAME MATCHES "Linux")
|
|||
|
|
set(sycl_lib_suffix "-preview")
|
|||
|
|
elseif(CMAKE_SYSTEM_NAME MATCHES "Windows")
|
|||
|
|
# On Windows, the SYCL library is named sycl7.lib until PYTORCH_2_5_SYCL_TOOLKIT_VERSION.
|
|||
|
|
# sycl.lib is supported in the later version.
|
|||
|
|
set(sycl_lib_suffix "7")
|
|||
|
|
endif()
|
|||
|
|
endif()
|
|||
|
|
|
|||
|
|
# Find SYCL library fullname.
|
|||
|
|
find_library(
|
|||
|
|
SYCL_LIBRARY
|
|||
|
|
NAMES "sycl${sycl_lib_suffix}"
|
|||
|
|
HINTS ${SYCL_LIBRARY_DIR}
|
|||
|
|
NO_DEFAULT_PATH
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# Find OpenCL library fullname, which is a dependency of oneDNN.
|
|||
|
|
find_library(
|
|||
|
|
OCL_LIBRARY
|
|||
|
|
NAMES OpenCL
|
|||
|
|
HINTS ${SYCL_LIBRARY_DIR}
|
|||
|
|
NO_DEFAULT_PATH
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
if((NOT SYCL_LIBRARY) OR (NOT OCL_LIBRARY))
|
|||
|
|
set(SYCL_FOUND False)
|
|||
|
|
set(SYCL_REASON_FAILURE "SYCL library is incomplete!!")
|
|||
|
|
set(SYCL_NOT_FOUND_MESSAGE "${SYCL_REASON_FAILURE}")
|
|||
|
|
return()
|
|||
|
|
endif()
|
|||
|
|
|
|||
|
|
find_package_handle_standard_args(
|
|||
|
|
SYCL
|
|||
|
|
FOUND_VAR SYCL_FOUND
|
|||
|
|
REQUIRED_VARS SYCL_INCLUDE_DIR SYCL_LIBRARY_DIR SYCL_LIBRARY
|
|||
|
|
REASON_FAILURE_MESSAGE "${SYCL_REASON_FAILURE}"
|
|||
|
|
VERSION_VAR SYCL_COMPILER_VERSION
|
|||
|
|
)
|