forked from epifanovsky/libtensor
-
Notifications
You must be signed in to change notification settings - Fork 4
Expand file tree
/
Copy pathCMakeLists.txt
More file actions
156 lines (130 loc) · 5.52 KB
/
CMakeLists.txt
File metadata and controls
156 lines (130 loc) · 5.52 KB
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
cmake_minimum_required(VERSION 3.5.0)
##########################################################################
# Project setup
list(APPEND CMAKE_MODULE_PATH ${CMAKE_CURRENT_SOURCE_DIR}/cmake)
option(BUILD_SHARED_LIBS "Build shared instead of static libraries" ON)
set(CMAKE_BUILD_TYPE Release CACHE STRING "Build type (Release or Debug)")
set(CMAKE_EXPORT_COMPILE_COMMANDS ON)
project(libtensorlight VERSION 3.0.1 LANGUAGES C CXX)
enable_testing()
##########################################################################
# Setup compiler flags
set(CMAKE_CXX_STANDARD 11)
set(CMAKE_CXX_STANDARD_REQUIRED ON)
set(CMAKE_CXX_EXTENSIONS OFF)
find_program(CCACHE_FOUND ccache)
if(CCACHE_FOUND)
set_property(GLOBAL PROPERTY RULE_LAUNCH_COMPILE ccache)
set_property(GLOBAL PROPERTY RULE_LAUNCH_LINK ccache)
endif(CCACHE_FOUND)
include(GNUInstallDirs)
include(CMakePackageConfigHelpers)
include(SetupCompiler)
set(CMAKE_POSITION_INDEPENDENT_CODE ${BUILD_SHARED_LIBS})
##########################################################################
# Threading and stdlib
if(NOT UNIX)
message(FATAL_ERROR "Only unix platforms are supported.")
endif()
include(InspectThreads)
include(InspectStdlib)
##########################################################################
# Blas
option(BLA_STATIC "Statically link to BLAS and LAPACK" OFF)
if (UNIX AND NOT APPLE)
set(BLA_VENDOR "OpenBLAS" CACHE STRING "BLAS Vendor to use (see CMake documentation)")
elseif (APPLE)
set(BLA_VENDOR "Apple" CACHE STRING "BLAS Vendor to use (see CMake documentation)")
endif()
find_package(BLAS REQUIRED)
include_directories(cblas)
if (BLA_VENDOR STREQUAL "Apple")
elseif(BLA_VENDOR STREQUAL "OpenBLAS")
elseif(BLA_VENDOR STREQUAL "Intel10_64lp_seq" OR BLA_VENDOR STREQUAL "Intel10_64lp")
# Find the directory with mkl.h
list(GET BLAS_LIBRARIES 1 BLAS_LIB)
get_filename_component(BLAS_DIR ${BLAS_LIB} DIRECTORY)
find_file(MKLINCLUDE mkl.h HINTS "${BLAS_DIR}/../include" "${BLAS_DIR}/../../include")
get_filename_component(BLAS_INCLUDE_DIR ${MKLINCLUDE} DIRECTORY)
else()
# TODO If more blas vendors are supported do not forget to add them to
# libtensor/linalg/BlasSequential.C and libtensor/CMakeLists.txt
message(WARNING "BLAS vendor ${BLA_VENDOR} has not been tested with libtensorlight")
endif()
set(BLAS_LAPACK_LIBRARIES ${BLAS_LIBRARIES})
##########################################################################
# Libxm dependencies
option(WITH_LIBXM "Compile libxm support into libtensor" OFF)
option(WITH_MPI "Compile MPI support into libtensor" OFF)
if (WITH_LIBXM)
find_package(OpenMP REQUIRED)
if (OPENMP_FOUND)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${OpenMP_CXX_FLAGS}")
set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} ${OpenMP_EXE_LINKER_FLAGS}")
endif()
if (WITH_MPI)
find_package(MPI REQUIRED)
endif()
else()
if(WITH_MPI)
message(FATAL_ERROR "Can't use MPI if WITH_LIBXM is off.")
endif()
endif()
##########################################################################
# Pass on to subdirectories
add_subdirectory(libutil)
add_subdirectory(libtensor)
option(LIBTENSOR_TESTS "Build libtensor tests" ON)
if (LIBTENSOR_TESTS)
add_subdirectory(tests)
endif()
##########################################################################
# Installation
# TODO: needs to be optimized, taking almost everything for now
install(DIRECTORY "libtensor/core"
DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/libtensor
FILES_MATCHING PATTERN "*.h")
install(DIRECTORY "libtensor/btod"
DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/libtensor
FILES_MATCHING PATTERN "*.h")
install(DIRECTORY "libtensor/block_tensor"
DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/libtensor
FILES_MATCHING PATTERN "*.h")
install(DIRECTORY "libtensor/gen_block_tensor"
DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/libtensor
FILES_MATCHING PATTERN "*.h")
install(DIRECTORY "libtensor/dense_tensor"
DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/libtensor
FILES_MATCHING PATTERN "*.h")
install(DIRECTORY "libtensor/symmetry"
DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/libtensor
FILES_MATCHING PATTERN "*.h")
install(DIRECTORY "libtensor/expr"
DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/libtensor
FILES_MATCHING PATTERN "*.h")
install(DIRECTORY "libtensor/kernels"
DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/libtensor
FILES_MATCHING PATTERN "*.h")
install(DIRECTORY "libtensor/linalg"
DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/libtensor
FILES_MATCHING PATTERN "*.h")
install(DIRECTORY "libutil"
DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}
FILES_MATCHING PATTERN "*.h")
install(FILES libtensor/libtensor.h libtensor/defs.h libtensor/exception.h
libtensor/metadata.h libtensor/timings.h libtensor/not_implemented.h
DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/libtensor)
install(TARGETS tensorlight
EXPORT "libtensorlightTargets"
ARCHIVE DESTINATION lib
LIBRARY DESTINATION lib)
# Build string of required compile definitions
get_directory_property(LIBTENSOR_COMPILE_DEFINITIONS_RAW COMPILE_DEFINITIONS)
list(JOIN LIBTENSOR_COMPILE_DEFINITIONS_RAW " -D" LIBTENSOR_COMPILE_DEFINITIONS)
if (LIBTENSOR_COMPILE_DEFINITIONS)
set(LIBTENSOR_COMPILE_DEFINITIONS "-D${LIBTENSOR_COMPILE_DEFINITIONS}")
endif()
# Build the pkg-config file
configure_file("${libtensorlight_SOURCE_DIR}/libtensor/libtensorlight.pc.in" "libtensorlight.pc" @ONLY)
install(FILES ${CMAKE_CURRENT_BINARY_DIR}/libtensorlight.pc
DESTINATION lib/pkgconfig)