forked from apache/mxnet
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathChooseBlas.cmake
194 lines (189 loc) · 8.36 KB
/
ChooseBlas.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
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements. See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership. The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License. You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.
set(BLAS "Open" CACHE STRING "Selected BLAS library")
set_property(CACHE BLAS PROPERTY STRINGS "Atlas;Open;MKL")
# ---[ Root folders
set(INTEL_HOME_ROOT "$ENV{HOME}/intel" CACHE PATH "Folder contains user-installed intel libs")
set(INTEL_OPT_ROOT "/opt/intel" CACHE PATH "Folder contains root-installed intel libs")
if(DEFINED USE_BLAS)
set(BLAS "${USE_BLAS}")
endif()
if(USE_BLAS MATCHES "MKL" OR USE_BLAS MATCHES "mkl" OR NOT DEFINED USE_BLAS)
find_path(MKL_INCLUDE_DIR mkl_version.h
PATHS $ENV{MKLROOT} ${INTEL_HOME_ROOT}/mkl ${INTEL_OPT_ROOT}/mkl ${INTEL_OPT_ROOT}/oneapi/mkl/latest
PATH_SUFFIXES mkl latest include)
if(NOT MKL_INCLUDE_DIR STREQUAL "MKL_INCLUDE_DIR-NOTFOUND")
set(BLAS "MKL")
endif()
endif()
if(BLAS STREQUAL "Atlas" OR BLAS STREQUAL "atlas")
find_package(Atlas REQUIRED)
include_directories(SYSTEM ${Atlas_INCLUDE_DIR})
list(APPEND mshadow_LINKER_LIBS ${Atlas_LIBRARIES})
add_definitions(-DMSHADOW_USE_CBLAS=1)
add_definitions(-DMSHADOW_USE_MKL=0)
add_definitions(-DMXNET_USE_BLAS_ATLAS=1)
elseif(BLAS STREQUAL "Open" OR BLAS STREQUAL "open")
find_package(OpenBLAS REQUIRED)
include_directories(SYSTEM ${OpenBLAS_INCLUDE_DIR})
list(APPEND mshadow_LINKER_LIBS ${OpenBLAS_LIB})
add_definitions(-DMSHADOW_USE_CBLAS=1)
add_definitions(-DMSHADOW_USE_MKL=0)
add_definitions(-DMXNET_USE_BLAS_OPEN=1)
if(NOT MSVC)
# check if we need to link to omp
execute_process(COMMAND ${CMAKE_NM} -g ${OpenBLAS_LIB}
COMMAND grep omp_get_num_threads
WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}
OUTPUT_VARIABLE OPENBLAS_USES_OMP_OUT
RESULT_VARIABLE OPENBLAS_USES_OMP_RET)
if(NOT OPENBLAS_USES_OMP_OUT STREQUAL "" AND NOT OPENBLAS_USES_OMP_RET AND NOT USE_OPENMP)
message("Openblas uses OMP, automatically linking to it")
find_package(OpenMP REQUIRED)
message("OpenMP_CXX_LIBRARIES is ${OpenMP_CXX_LIBRARIES}")
list(APPEND mshadow_LINKER_LIBS "${OpenMP_CXX_LIBRARIES}")
endif()
# check if we need to link to gfortran
execute_process(COMMAND ${CMAKE_NM} -g ${OpenBLAS_LIB}
COMMAND grep gfortran
WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}
OUTPUT_VARIABLE OPENBLAS_USES_GFORTRAN_OUT
RESULT_VARIABLE OPENBLAS_USES_GFORTRAN_RET)
if(NOT OPENBLAS_USES_GFORTRAN_OUT STREQUAL "" AND NOT OPENBLAS_USES_GFORTRAN_RET)
message("Openblas uses GFortran, automatically linking to it")
file(WRITE "${CMAKE_CURRENT_BINARY_DIR}/temp/CMakeLists.txt"
"cmake_minimum_required(VERSION ${CMAKE_VERSION})
project(CheckFortran Fortran)
set(CMAKE_Fortran_COMPILER gfortran)
file(WRITE \"${CMAKE_CURRENT_BINARY_DIR}/temp/FortranDir.cmake\"
\"
set(FORTRAN_DIR \\\"\$\{CMAKE_Fortran_IMPLICIT_LINK_DIRECTORIES\}\\\")
\")
")
execute_process(
WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}/temp/
COMMAND ${CMAKE_COMMAND} .
)
set(FORTRAN_DIR "")
include(${CMAKE_CURRENT_BINARY_DIR}/temp/FortranDir.cmake)
find_library(FORTRAN_LIB NAMES gfortran HINTS ${FORTRAN_DIR})
message("FORTRAN_DIR is ${FORTRAN_DIR}")
message("FORTRAN_LIB is ${FORTRAN_LIB}")
list(APPEND mshadow_LINKER_LIBS ${FORTRAN_LIB})
file(REMOVE_RECURSE "${CMAKE_CURRENT_BINARY_DIR}/temp/")
endif()
# check the lapack flavor of openblas
include(CheckSymbolExists)
check_symbol_exists(OPENBLAS_USE64BITINT "${OpenBLAS_INCLUDE_DIR}/openblas_config.h" OPENBLAS_ILP64)
if(OPENBLAS_ILP64)
message("Using ILP64 OpenBLAS")
if(NOT USE_INT64_TENSOR_SIZE)
message(FATAL_ERROR "Must set USE_INT64_TENSOR_SIZE=1 when using ILP64 OpenBLAS")
endif()
else()
message("Using LP64 OpenBLAS")
endif()
if(USE_LAPACK)
if(EXISTS "${OpenBLAS_INCLUDE_DIR}/lapacke.h")
message("Detected lapacke.h, automatically using the LAPACKE interface")
add_definitions(-DMXNET_USE_LAPACKE_INTERFACE=1)
set(USE_LAPACKE_INTERFACE 1)
if(OPENBLAS_ILP64)
message("Detected ILP64 LAPACKE")
add_definitions(-DMXNET_USE_ILP64_LAPACKE=1)
endif()
else()
execute_process(COMMAND ${CMAKE_NM} -g ${OpenBLAS_LIB}
COMMAND grep sgetri_
WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}
OUTPUT_VARIABLE OPENBLAS_CONTAINS_C_LAPACK_OUT
RESULT_VARIABLE OPENBLAS_CONTAINS_C_LAPACK_RET)
if(OPENBLAS_CONTAINS_C_LAPACK_OUT STREQUAL ""
AND NOT OPENBLAS_CONTAINS_C_LAPACK_RET)
list(APPEND mshadow_LINKER_LIBS lapack)
endif()
endif()
endif()
endif()
elseif(BLAS STREQUAL "MKL" OR BLAS STREQUAL "mkl")
# ---[ MKL Options
file(STRINGS ${MKL_INCLUDE_DIR}/mkl_version.h MKL_VERSION_DEF REGEX "INTEL_MKL_VERSION")
string(REGEX MATCH "([0-9]+)" MKL_VERSION ${MKL_VERSION_DEF})
if(UNIX)
# Single dynamic library interface leads to conflicts between intel omp and llvm omp
# https://github.com/apache/incubator-mxnet/issues/17641
# Fixed in oneMKL 2021.3: [MKLD-11109] MKL is opening libgomp.so instead of
# libgomp.so.1 while SDL=1 & MKL_THREADING_LAYER=GNU
cmake_dependent_option(MKL_USE_SINGLE_DYNAMIC_LIBRARY "Use single dynamic library interface" ON
"NOT BLA_STATIC;MKL_VERSION GREATER_EQUAL 20210003" OFF)
else()
option(MKL_USE_SINGLE_DYNAMIC_LIBRARY "Use single dynamic library interface" ON)
endif()
cmake_dependent_option(BLA_STATIC "Use static libraries" ON "NOT MKL_USE_SINGLE_DYNAMIC_LIBRARY" OFF)
option(MKL_MULTI_THREADED "Use multi-threading" ON)
if(BLA_VENDOR)
message(FATAL_ERROR "Do not set BLA_VENDOR manually. MKL version (BLA_VENDOR) is selected based on MKL_USE_SINGLE_DYNAMIC_LIBRARY, "
"MKL_MULTI_THREADED and USE_INT64_TENSOR_SIZE flags. If you want to select specific MKL library version "
"please set the above-mentioned flags instead.")
endif()
if(MKL_USE_SINGLE_DYNAMIC_LIBRARY)
set(BLA_VENDOR Intel10_64_dyn)
add_definitions(-DMKL_USE_SINGLE_DYNAMIC_LIBRARY=1)
else()
if(CMAKE_SIZEOF_VOID_P EQUAL 4)
set(BLA_VENDOR Intel10_32)
else()
if(MKL_MULTI_THREADED)
if(USE_INT64_TENSOR_SIZE)
set(BLA_VENDOR Intel10_64ilp)
else()
set(BLA_VENDOR Intel10_64lp)
endif()
else()
if(USE_INT64_TENSOR_SIZE)
set(BLA_VENDOR Intel10_64ilp_seq)
else()
set(BLA_VENDOR Intel10_64lp_seq)
endif()
endif()
endif()
endif()
# In case of oneAPI 2021.3 if MKL_INCLUDE_DIR points to the subdirectory 'include',
# use the parent directory 'latest' instead
file(TO_CMAKE_PATH "${MKL_INCLUDE_DIR}" BLAS_mkl_MKLROOT)
get_filename_component(BLAS_mkl_MKLROOT_LAST_DIR "${BLAS_mkl_MKLROOT}" NAME)
if(BLAS_mkl_MKLROOT_LAST_DIR STREQUAL "include")
get_filename_component(BLAS_mkl_MKLROOT "${BLAS_mkl_MKLROOT}" DIRECTORY)
endif()
find_package(BLAS)
include_directories(SYSTEM ${MKL_INCLUDE_DIR})
list(APPEND mshadow_LINKER_LIBS ${BLAS_LIBRARIES})
if(USE_INT64_TENSOR_SIZE)
add_definitions(-DUSE_INT64_TENSOR_SIZE=1)
endif()
add_definitions(-DMSHADOW_USE_CBLAS=0)
add_definitions(-DMSHADOW_USE_MKL=1)
add_definitions(-DMXNET_USE_BLAS_MKL=1)
message("-- Found MKL (version: ${MKL_VERSION})")
elseif(BLAS STREQUAL "apple")
find_package(Accelerate REQUIRED)
include_directories(SYSTEM ${Accelerate_INCLUDE_DIR})
list(APPEND mshadow_LINKER_LIBS ${Accelerate_LIBRARIES})
add_definitions(-DMSHADOW_USE_MKL=0)
add_definitions(-DMSHADOW_USE_CBLAS=1)
add_definitions(-DMXNET_USE_BLAS_APPLE=1)
endif()