2424# *****************************************************************************
2525
2626# 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
2829#
2930# The following are set after configuration is done:
3031# MATHLIB_FOUND
3334
3435include (FindPackageHandleStandardArgs)
3536
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 ()
3942
4043if (UNIX )
4144 set (MATHLIB_SYCL_LIB
@@ -51,14 +54,14 @@ endif()
5154
5255find_path (
5356 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
5659 DOC "Path to mathlib include files" )
5760
5861find_path (
5962 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
6265 DOC "Path to mathlib library files" )
6366
6467# 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
6871
6972if (MathLib_FOUND)
7073 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)
7275endif ()
0 commit comments