forked from BVLC/caffe
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathFindMKL.cmake
113 lines (92 loc) · 3.47 KB
/
FindMKL.cmake
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
# - Find Intel MKL
# Find the MKL libraries
#
# Options:
#
# MKL_STATAIC : use static linking
# MKL_MULTI_THREADED: use multi-threading
# MKL_SDL : Single Dynamic Library interface
#
# This module defines the following variables:
#
# MKL_FOUND : True if MKL_INCLUDE_DIR are found
# MKL_INCLUDE_DIR : where to find mkl.h, etc.
# MKL_INCLUDE_DIRS : set when MKL_INCLUDE_DIR found
# MKL_LIBRARIES : the library to link against.
include(FindPackageHandleStandardArgs)
set(INTEL_ROOT "/opt/intel" CACHE PATH "Folder contains intel libs")
set(MKL_ROOT ${INTEL_ROOT}/mkl CACHE PATH "Folder contains MKL")
# Find include dir
find_path(MKL_INCLUDE_DIR mkl.h
PATHS ${MKL_ROOT}/include)
# Find include directory
# There is no include folder under linux
if(WIN32)
find_path(INTEL_INCLUDE_DIR omp.h
PATHS ${INTEL_ROOT}/include)
set(MKL_INCLUDE_DIR ${MKL_INCLUDE_DIR} ${INTEL_INCLUDE_DIR})
endif()
# Find libraries
# Handle suffix
set(_MKL_ORIG_CMAKE_FIND_LIBRARY_SUFFIXES ${CMAKE_FIND_LIBRARY_SUFFIXES})
if(WIN32)
if(MKL_STATAIC)
set(CMAKE_FIND_LIBRARY_SUFFIXES .lib)
else()
set(CMAKE_FIND_LIBRARY_SUFFIXES _dll.lib)
endif()
else()
if(MKL_STATAIC)
set(CMAKE_FIND_LIBRARY_SUFFIXES .a)
else()
set(CMAKE_FIND_LIBRARY_SUFFIXES .so)
endif()
endif()
# MKL is composed by four layers: Interface, Threading, Computational and RTL
if(MKL_SDL)
find_library(MKL_LIBRARY mkl_rt
PATHS ${MKL_ROOT}/lib/ia32/)
set(MKL_MINIMAL_LIBRARY ${MKL_LIBRARY})
else()
######################### Interface layer #######################
if(WIN32)
set(MKL_INTERFACE_LIBNAME mkl_intel_c)
else()
set(MKL_INTERFACE_LIBNAME mkl_intel)
endif()
find_library(MKL_INTERFACE_LIBRARY ${MKL_INTERFACE_LIBNAME}
PATHS ${MKL_ROOT}/lib/ia32/)
######################## Threading layer ########################
if(MKL_MULTI_THREADED)
set(MKL_THREADING_LIBNAME mkl_intel_thread)
else()
set(MKL_THREADING_LIBNAME mkl_sequential)
endif()
find_library(MKL_THREADING_LIBRARY ${MKL_THREADING_LIBNAME}
PATHS ${MKL_ROOT}/lib/ia32/)
####################### Computational layer #####################
find_library(MKL_CORE_LIBRARY mkl_core
PATHS ${MKL_ROOT}/lib/ia32/)
find_library(MKL_FFT_LIBRARY mkl_cdft_core
PATHS ${MKL_ROOT}/lib/ia32/)
find_library(MKL_SCALAPACK_LIBRARY mkl_scalapack_core
PATHS ${MKL_ROOT}/lib/ia32/)
############################ RTL layer ##########################
if(WIN32)
set(MKL_RTL_LIBNAME libiomp5md)
else()
set(MKL_RTL_LIBNAME libiomp5)
endif()
find_library(MKL_RTL_LIBRARY ${MKL_RTL_LIBNAME}
PATHS ${INTEL_RTL_ROOT}/lib)
set(MKL_LIBRARY ${MKL_INTERFACE_LIBRARY} ${MKL_THREADING_LIBRARY} ${MKL_CORE_LIBRARY} ${MKL_FFT_LIBRARY} ${MKL_SCALAPACK_LIBRARY} ${MKL_RTL_LIBRARY})
set(MKL_MINIMAL_LIBRARY ${MKL_INTERFACE_LIBRARY} ${MKL_THREADING_LIBRARY} ${MKL_CORE_LIBRARY} ${MKL_RTL_LIBRARY})
endif()
set(CMAKE_FIND_LIBRARY_SUFFIXES ${_MKL_ORIG_CMAKE_FIND_LIBRARY_SUFFIXES})
find_package_handle_standard_args(MKL DEFAULT_MSG
MKL_INCLUDE_DIR MKL_LIBRARY MKL_MINIMAL_LIBRARY)
if(MKL_FOUND)
set(MKL_INCLUDE_DIRS ${MKL_INCLUDE_DIR})
set(MKL_LIBRARIES ${MKL_LIBRARY})
set(MKL_MINIMAL_LIBRARIES ${MKL_LIBRARY})
endif()