-
Notifications
You must be signed in to change notification settings - Fork 195
/
CMakeLists.txt
84 lines (73 loc) · 2.56 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
cmake_minimum_required(VERSION 2.8)
project(OpenTracker)
set( LIBRARY_OUTPUT_PATH ${PROJECT_BINARY_DIR}/lib )
set( EXECUTABLE_OUTPUT_PATH ${PROJECT_BINARY_DIR}/bin )
link_directories( ${PROJECT_BINARY_DIR}/lib )
set(CMAKE_ALLOW_LOOSE_LOOP_CONSTRUCTS ON)
option(GPU "use GPU" OFF)
option(OPENCV "use OPENCV" ON)
option(OPENMP "use OPENMP" OFF)
option(CUDNN "use CUDNN" OFF)
option(MULTI_THREAD "use MULTI_THREAD" ON)
if(NOT CMAKE_BUILD_TYPE)
set(CMAKE_BUILD_TYPE RELEASE)
endif()
message(STATUS "compile type:" ${CMAKE_BUILD_TYPE})
#0:no-simd;1:sse-Intel;2:neon-TX2;
set(SIMD 0)
if(SIMD EQUAL 1)
message(STATUS "use sse-Intel to compile")
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -msse4")
add_definitions(-DUSE_SIMD)
elseif(SIMD EQUAL 2)
message(STATUS "use neon-TX2 to compile")
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -ffast-math -flto -march=armv8-a+crypto -mcpu=cortex-a57+crypto")
add_definitions(-DUSE_SIMD)
add_definitions(-DUSE_NENOS)
elseif(SIMD EQUAL 0)
message(STATUS "no-simd to compile")
endif()
if(MULTI_THREAD)
add_definitions(-DUSE_MULTI_THREAD)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -pthread")
endif()
if(GPU)
message(STATUS "use gpu to compile")
find_package( CUDA QUIET REQUIRED )
include(FindCUDA)
add_definitions(-DGPU)
set(CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS}
-gencode arch=compute_50,code=sm_50
-gencode arch=compute_52,code=sm_52
-gencode arch=compute_53,code=sm_53
-gencode arch=compute_61,code=sm_61
-gencode arch=compute_62,code=sm_62)
set(CUDA_NVCC_FLGAS_DEBUG "-g -G -Og")
set(CUDA_NVCC_FLGAS_RELEASE "-Ofast -lineinfo")
include_directories( ${CUDA_INCLUDE_DIRS} )
link_directories( /usr/local/cuda/lib64 )
set(LIBS "-lcuda -lcudart -lcublas -lcurand -lstdc++")
if(CUDNN)
message(STATUS "use cudnn to compile")
add_definitions(-DCUDNN)
set(LIBS ${LIBS} "-lcudnn")
endif()
endif()
if(OPENCV)
message(STATUS "use opencv to compile")
find_package( OpenCV REQUIRED )
add_definitions(-DOPENCV)
include_directories(${OpenCV_INCLUDE_DIRS})
link_directories ( ${OpenCV_LIB_DIR})
endif()
if(OPENMP)
find_package(OpenMP REQUIRED)
SET(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${OpenMP_C_FLAGS}")
endif()
set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${OPTIMAZE_FLAG} -Wall -Wno-unknown-pragmas -Wfatal-errors -fPIC")
set(CMAKE_C_FLAGS_DEBUG "-O0 -g")
set(CMAKE_C_FLAGS_RELEASE "-O3")
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -lm -pthread -Wall -lstdc++ -std=c++0x")
set(CMAKE_CXX_FLAGS_DEBUG "-O0 -g")
set(CMAKE_CXX_FLAGS_RELEASE "-O3")
add_subdirectory(eco)