24
24
# *****************************************************************************
25
25
26
26
# The following variables are optionally searched for defaults
27
- # MATHLIB_ROOT_DIR: Base directory where all components are found
27
+ # MKLROOT: Environment variable to specify custom search place
28
+ # ONEAPI_ROOT: Environment variable to specify search place from oneAPI
28
29
#
29
30
# The following are set after configuration is done:
30
31
# MATHLIB_FOUND
33
34
34
35
include (FindPackageHandleStandardArgs)
35
36
36
- set (MATHLIB_ROOT_DIR
37
- "${DPNP_ONEAPI_ROOT} /mkl"
38
- CACHE PATH "Folder contains mathlib" )
37
+ set (DPNP_ONEAPI_MKL "$ENV{DPNP_ONEAPI_ROOT} /mkl/latest" CACHE PATH "Folder contains Math Lib files from ONEAPI_ROOT" )
38
+
39
+ if (DEFINED ENV{MKLROOT})
40
+ set (DPNP_MKLROOT "$ENV{MKLROOT} " CACHE PATH "Folder contains Math Lib files from MKLROOT" )
41
+ endif ()
39
42
40
43
if (UNIX )
41
44
set (MATHLIB_SYCL_LIB
@@ -51,14 +54,14 @@ endif()
51
54
52
55
find_path (
53
56
MATHLIB_INCLUDE_DIR oneapi/mkl.hpp
54
- HINTS ENV CONDA_PREFIX ENV PREFIX ${MATHLIB_ROOT_DIR} # search order is important
55
- PATH_SUFFIXES include latest/ include
57
+ HINTS ${DPNP_MKLROOT} ${DPNP_ONEAPI_MKL} ENV CONDA_PREFIX ENV PREFIX # search order is important
58
+ PATH_SUFFIXES include include
56
59
DOC "Path to mathlib include files" )
57
60
58
61
find_path (
59
62
MATHLIB_LIBRARY_DIR ${MATHLIB_SYCL_LIB}
60
- HINTS ENV CONDA_PREFIX ENV PREFIX ${MATHLIB_ROOT_DIR} # search order is important
61
- PATH_SUFFIXES lib latest/ lib/intel64
63
+ HINTS ${DPNP_MKLROOT} ${DPNP_ONEAPI_MKL} ENV CONDA_PREFIX ENV PREFIX # search order is important
64
+ PATH_SUFFIXES lib lib/intel64
62
65
DOC "Path to mathlib library files" )
63
66
64
67
# TODO implement recurcive searching file (GLOB_RECURSE MY_PATH "/opt/intel/*/mkl.hpp")
@@ -68,5 +71,5 @@ find_package_handle_standard_args(MathLib DEFAULT_MSG MATHLIB_INCLUDE_DIR MATHLI
68
71
69
72
if (MathLib_FOUND)
70
73
message (STATUS "Found MathLib: (include: ${MATHLIB_INCLUDE_DIR} , library: ${MATHLIB_LIBRARY_DIR} )" )
71
- mark_as_advanced (MATHLIB_ROOT_DIR MATHLIB_INCLUDE_DIR MATHLIB_LIBRARY_DIR)
74
+ # mark_as_advanced(DPNP_MKLROOT MATHLIB_INCLUDE_DIR MATHLIB_LIBRARY_DIR)
72
75
endif ()
0 commit comments