-
Notifications
You must be signed in to change notification settings - Fork 92
/
Copy pathCMakeLists.txt
203 lines (184 loc) · 7.01 KB
/
CMakeLists.txt
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
195
196
197
198
199
200
201
202
203
cmake_minimum_required(VERSION 3.5)
enable_testing()
set(CMAKE_MODULE_PATH ${CMAKE_MODULE_PATH} ${CMAKE_CURRENT_SOURCE_DIR}/cmake)
set(CMAKE_EXPORT_COMPILE_COMMANDS ON)
####################
# Project
project(onnc)
set(ONNC_ROOT_PATH ${CMAKE_CURRENT_LIST_DIR})
file(READ ${ONNC_ROOT_PATH}/VERSION PACKAGE_VERSION)
string(REPLACE "\n" " " PACKAGE_VERSION ${PACKAGE_VERSION})
set(PACKAGE "onnc")
set(PACKAGE_BUGREPORT "maillist.onnc.ai")
set(PACKAGE_NAME "ONNC")
set(PACKAGE_STRING "ONNC ${PACKAGE_VERSION}")
set(PACKAGE_TARNAME "onnc")
set(PACKAGE_URL "onnc.ai")
####################
# ONNC RT
if(ONNC_RT_ONLY)
SET(ONNC_RUNTIME_LIB_NAME "onnc-rt")
set(LIB_BASE_PATH ${CMAKE_CURRENT_LIST_DIR})
cmake_policy(SET CMP0048 NEW)
add_library(${ONNC_RUNTIME_LIB_NAME}
lib/Runtime/onnc-runtime.c
)
add_subdirectory(lib/Runtime/operator)
target_include_directories(${ONNC_RUNTIME_LIB_NAME} PUBLIC
${LIB_BASE_PATH}/include
)
file(GLOB_RECURSE PUBLIC_HEADER_FILES ${LIB_BASE_PATH}/include/onnc-runtime.h)
set_target_properties(${ONNC_RUNTIME_LIB_NAME} PROPERTIES
PUBLIC_HEADER "${PUBLIC_HEADER_FILES}"
)
install(TARGETS ${ONNC_RUNTIME_LIB_NAME}
ARCHIVE DESTINATION lib
PUBLIC_HEADER DESTINATION include/onnc/Runtime
)
RETURN()
endif()
# Provide function for static link
function(find_static_library OUT)
cmake_parse_arguments(
PARSED_ARGS
""
""
"NAMES;HINTS;PATH_SUFFIXES;PATHS"
${ARGN}
)
if (WIN32 OR MSVC)
set(CMAKE_FIND_LIBRARY_SUFFIXES ".lib")
elseif (UNIX)
set(CMAKE_FIND_LIBRARY_SUFFIXES ".a ${CMAKE_FIND_LIBRARY_SUFFIXES}")
endif()
unset(FOUND CACHE)
list(GET PARSED_ARGS_NAMES 0 NAME)
find_library(
FOUND_${NAME}_STATIC
NAMES
${PARSED_ARGS_NAMES}
HINTS
${PARSED_ARGS_HINTS}
PATH_SUFFIXES
${PARSED_ARGS_PATH_SUFFIXES}
PATHS
${PARSED_ARGS_PATHS}
)
if (FOUND_${NAME}_STATIC)
get_filename_component(ABS_FILE ${FOUND_${NAME}_STATIC} ABSOLUTE)
set(${OUT} ${FOUND_${NAME}_STATIC} PARENT_SCOPE)
else()
message(SEND_ERROR "Unable to find library ${NAME}")
endif()
set(${OUT} ${ABS_FILE} PARENT_SCOPE)
endfunction()
####################
# General settings
include(option)
option_enum(NAME CMAKE_BUILD_TYPE HELP "Choose the type of build" VALUE Normal Debug Release)
option(ENABLE_PTHREAD "use pthreads" ON)
set(HAVE_PTHREADS ${ENABLE_PTHREAD})
option(ENABLE_CLOCK_GETTIME "enable clock_gettime()" ON)
option(ENABLE_GETTIMEOFDAY "enable gettimeofday()" ON)
option(ENABLE_UNITTEST "enable unittest" ON)
####################
# Language
set(STDC_HEADERS 1)
set(CMAKE_CXX_STANDARD 14)
set(HAVE_CXX14 TRUE)
set(CMAKE_CXX_STANDARD_REQUIRED ON)
set(CMAKE_CXX_EXTENSIONS OFF)
set(CMAKE_CXX_FLAGS_DEBUG "-O0 -g3" CACHE STRING "Flags used by the CXX compiler during DEBUG builds.")
set(CMAKE_CXX_FLAGS_RELEASE "-O2 -DNDEBUG" CACHE STRING "Flags used by the CXX compiler during RELEASE builds.")
set(CMAKE_CXX_FLAGS_RELWITHDEBINFO "-O2 -DNDEBUG -g3" CACHE STRING "Flags used by the CXX compiler during RELWITHDEBINFO builds.")
option(ENABLE_WERROR "Compile with -Werror enabled (default is OFF)" OFF)
if (ENABLE_WERROR)
add_compile_options(-Werror)
endif()
####################
# Check
include(check)
message(STATUS "Default target quadruple: ${ONNC_DEFAULT_TARGET_QUADRUPLE}")
####################
# Scan targets from config
file(STRINGS ${ONNC_ROOT_PATH}/include/onnc/Config/Config.h.cmake.in CONFIG_IN_CONTENT)
string(REGEX MATCHALL "ENABLE_[A-Za-z0-9_]*_TARGET" CONFIG_IN_TARGETS ${CONFIG_IN_CONTENT})
file(GLOB target_dir_dump RELATIVE ${ONNC_ROOT_PATH}/lib/Target ${ONNC_ROOT_PATH}/lib/Target/* )
foreach(content IN LISTS target_dir_dump)
string(TOUPPER ${content} content)
set(target_dir_upper ${target_dir_upper} ${content})
endforeach(content IN LISTS target_dir_dump)
foreach(EACH_TARGET IN LISTS CONFIG_IN_TARGETS)
# Get target name
string(LENGTH ${EACH_TARGET} EACH_LEN)
math(EXPR SUBSTR_LEN "${EACH_LEN} - 14")
string(SUBSTRING ${EACH_TARGET} 7 ${SUBSTR_LEN} EACH_TARGET_NAME)
# Set option
string(TOUPPER ${EACH_TARGET_NAME} EACH_TARGET_NAME_upper)
if(EACH_TARGET_NAME_upper IN_LIST target_dir_upper)
option(${EACH_TARGET} "Enable ${EACH_TARGET_NAME}" ON)
if(${EACH_TARGET})
list(FIND target_dir_upper ${EACH_TARGET_NAME_upper} target_dir_index)
list(GET target_dir_dump ${target_dir_index} target_name_real)
set(ONNC_TARGET_PLATFORMS "${ONNC_TARGET_PLATFORMS} ONNC_PLATFORM(${target_name_real})")
set(ONNC_TARGET_BACKENDS "${ONNC_TARGET_BACKENDS} ONNC_BACKEND(${target_name_real})")
set(ONNC_TARGETS ${ONNC_TARGETS} ${target_name_real})
set(ONNC_TARGET_ENTRIES ${ONNC_TARGET_ENTRIES} ${EACH_TARGET})
endif(${EACH_TARGET})
endif(EACH_TARGET_NAME_upper IN_LIST target_dir_upper)
endforeach(EACH_TARGET IN CONFIG_IN_TARGETS)
unset(target_dir_upper)
####################
# Include
set(ONNC_INCLUDE_DIRS
${CMAKE_CURRENT_SOURCE_DIR}/include
${CMAKE_CURRENT_BINARY_DIR}/include
${CMAKE_CURRENT_BINARY_DIR}/lib
)
####################
# External
# onnx
set(ONNX_NAMESPACE "onnx" CACHE STRING "Namespace of onnx")
add_definitions(-DONNX_NAMESPACE=${ONNX_NAMESPACE})
find_package(onnx REQUIRED)
message(STATUS "[${PACKAGE}] Using onnx include at ${ONNX_INCLUDE_DIR}")
message(STATUS "[${PACKAGE}] Using onnx library at ${ONNX_LIBRARY_DIR}")
include_directories(${ONNX_INCLUDE_DIR})
link_directories(${ONNX_LIBRARY_DIR})
# SkyPat
find_package(SkyPat REQUIRED)
message(STATUS "[${PACKAGE}] Using SkyPat include at ${SKYPAT_INCLUDE_DIR}")
include_directories(${SKYPAT_INCLUDE_DIR})
link_directories(${SKYPAT_LIBRARY_DIR})
# llvm
OPTION(USE_LLVM "Use llvm" ON)
if(USE_LLVM)
if(LLVM_ROOT_DIR)
set(LLVM_DIR ${LLVM_ROOT_DIR}/lib/cmake/llvm)
endif(LLVM_ROOT_DIR)
find_package(LLVM REQUIRED CONFIG)
message(STATUS "[${PACKAGE}] Using llvm include at ${LLVM_INCLUDE_DIR}")
include_directories(${LLVM_INCLUDE_DIRS})
add_definitions(${LLVM_DEFINITIONS})
endif(USE_LLVM)
# Protobuf
find_package(Protobuf REQUIRED)
# glog
find_package(Glog REQUIRED)
message(STATUS "[${PACKAGE}] Using glog include at ${GLOG_INCLUDE_DIR}")
include_directories(${GLOG_INCLUDE_DIR})
link_directories(${GLOG_LIBRARY_DIR})
####################
# Other
add_custom_target(code_gen)
add_subdirectory(lib)
add_subdirectory(tools)
add_subdirectory(include)
####################
# OUTPUT
configure_file(include/onnc/Support/DataTypes.h.cmake.in include/onnc/Support/DataTypes.h)
configure_file(include/onnc/Config/Config.h.cmake.in include/onnc/Config/Config.h)
configure_file(include/onnc/Config/ONNX.h.in include/onnc/Config/ONNX.h @ONLY)
configure_file(include/onnc/Config/Platforms.def.in include/onnc/Config/Platforms.def @ONLY)
configure_file(include/onnc/Config/Backends.def.in include/onnc/Config/Backends.def @ONLY)
configure_file(cmake/ONNCConfig.cmake.in cmake/ONNCConfig.cmake @ONLY)