-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathCMakeLists.txt
113 lines (96 loc) · 5.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
cmake_minimum_required(VERSION 3.28)
set(TORCH_CUDA_ARCH_LIST 8.6)
project(DiffusionModelC++ C CXX)
############################################################################################################
# Compile Options
############################################################################################################
set(CMAKE_CXX_STANDARD 17)
set(CMAKE_POSITION_INDEPENDENT_CODE on)
# Set default build type if not specified
if (NOT CMAKE_BUILD_TYPE AND NOT CMAKE_CONFIGURATION_TYPES)
message(STATUS "Setting build type to 'Release' as none was specified.")
set(CMAKE_BUILD_TYPE Release CACHE STRING "Choose the type of build." FORCE)
# Set the possible values of build type for cmake-gui
set_property(CACHE CMAKE_BUILD_TYPE PROPERTY STRINGS "Debug" "Release" "MinSizeRel" "RelWithDebInfo")
endif ()
if (MSVC)
set(CMAKE_CXX_FLAGS_RELEASE "${CMAKE_CXX_FLAGS_RELEASE} /O2")
elseif (UNIX)
set(CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG} -O0 -Wall")
set(CMAKE_CXX_FLAGS_RELEASE "${CMAKE_CXX_FLAGS_RELEASE}")
endif ()
############################################################################################################
# Environment Variables
############################################################################################################
set(PROJECT_INCLUDE_DIR "${PROJECT_SOURCE_DIR}/include")
set(PROJECT_EXTERNAL_DIR "${PROJECT_SOURCE_DIR}/external")
############################################################################################################
# External Libraries
############################################################################################################
# OpenMP
find_package(OpenMP REQUIRED)
if (OpenMP_FOUND)
set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${OpenMP_C_FLAGS}")
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${OpenMP_CXX_FLAGS}")
endif ()
# Torch
set(CAFFE2_USE_CUDNN 1)
find_package(Torch REQUIRED)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${TORCH_CXX_FLAGS}")
# OpenCV
find_package(OpenCV REQUIRED)
# Spdlog
set(SPD_LOG_INCLUDE_DIR ${PROJECT_EXTERNAL_DIR}/spdlog/include)
# Pico JSON
set(PICO_JSON_INCLUDE_DIR ${PROJECT_EXTERNAL_DIR}/picojson)
############################################################################################################
# External libralies
############################################################################################################
set(
EXTERNAL_INCLUDE_DIR
${TORCH_INCLUDE_DIRS}
${OpenCV_INCLUDE_DIRS}
${SPD_LOG_INCLUDE_DIR}
${PICO_JSON_INCLUDE_DIR}
)
set(
PROJECT_LIBS
${TORCH_LIBRARIES}
${OpenCV_LIBS}
)
############################################################################################################
# Add subdirectory
############################################################################################################
add_subdirectory(src)
add_subdirectory(external)
add_subdirectory(tests)
############################################################################################################
# Message
############################################################################################################
message(STATUS "# =======================================================================================================")
message(STATUS "# '${PROJECT_NAME}' build configuration ")
message(STATUS "# =======================================================================================================")
message(STATUS "# [CMake]")
message(STATUS "# CMake version : ${CMAKE_VERSION}")
message(STATUS "# CMake build type : ${CMAKE_BUILD_TYPE}")
message(STATUS "# CMAKE_CXX_FLAGS : ${CMAKE_CXX_FLAGS}")
message(STATUS "# CMAKE_CXX_FLAGS_DEBUG : ${CMAKE_CXX_FLAGS_DEBUG}")
message(STATUS "# CMAKE_CXX_FLAGS_RELEASE : ${CMAKE_CXX_FLAGS_RELEASE}")
message(STATUS "# ")
message(STATUS "# [C/C++]")
message(STATUS "# C Compiler : ${CMAKE_C_COMPILER_ID} | ${CMAKE_C_COMPILER_VERSION} | ${CMAKE_C_COMPILER}")
message(STATUS "# C++ Compiler : ${CMAKE_CXX_COMPILER_ID} | ${CMAKE_CXX_COMPILER_VERSION} | ${CMAKE_CXX_COMPILER}")
message(STATUS "# CXX STANDARD : ${CMAKE_CXX_STANDARD}")
message(STATUS "# ")
message(STATUS "# [CUDA]")
message(STATUS "# CMAKE_CUDA_COMPILER : ${CMAKE_CUDA_COMPILER}")
message(STATUS "# CMAKE_CUDA_ARCHITECTURES : ${CMAKE_CUDA_ARCHITECTURES}")
message(STATUS "# ")
message(STATUS "# [OpenMP]")
message(STATUS "# OpenMP_C_FLAGS : ${OpenMP_C_FLAGS}")
message(STATUS "# OpenMP_CXX_FLAGS : ${OpenMP_CXX_FLAGS}")
message(STATUS "# ")
message(STATUS "# [Environment Variables]")
message(STATUS "# PROJECT_INCLUDE_DIR : ${PROJECT_INCLUDE_DIR}")
message(STATUS "# PROJECT_EXTERNAL_DIR : ${PROJECT_EXTERNAL_DIR}")
message(STATUS "# =======================================================================================================")