-
Notifications
You must be signed in to change notification settings - Fork 14
/
Copy pathCMakeLists.txt
113 lines (85 loc) · 3.93 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.12 FATAL_ERROR)
project(bude_kokkos)
# XXX CMAKE_INCLUDE_SYSTEM_FLAG_* must be set after entry into this script as the flag gets overwritten at CMake initialisaiton.
# so we can't just pass -DCMAKE_INCLUDE_SYSTEM_FLAG_CXX=...
if(CUSTOM_SYSTEM_INCLUDE_FLAG)
set(CMAKE_INCLUDE_SYSTEM_FLAG_CXX "${CUSTOM_SYSTEM_INCLUDE_FLAG}")
set(CMAKE_INCLUDE_SYSTEM_FLAG_C "${CUSTOM_SYSTEM_INCLUDE_FLAG}")
endif()
set(CMAKE_CXX_STANDARD 14)
set(CMAKE_CXX_EXTENSIONS OFF)
set(CMAKE_CXX_STANDARD_REQUIRED ON)
cmake_policy(SET CMP0074 NEW) #see https://github.com/kokkos/kokkos/blob/master/BUILD.md
if (KOKKOS_IN_TREE)
message(STATUS "Building using in-tree kokkos source at `${KOKKOS_IN_TREE}`")
if(FORWARD_CXX_EXTRA_FLAGS_TO_KOKKOS)
message(STATUS "Forwarding CXX_EXTRA_FLAGS to Kokkos...")
set(OLD_CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}") # save it for now
set(CMAKE_CXX_FLAGS ${CXX_EXTRA_FLAGS})
else()
message(STATUS "Not forwarding CXX_EXTRA_FLAGS to Kokkos by default, set FORWARD_CXX_EXTRA_FLAGS_TO_KOKKOS=ON to override")
endif()
if(KOKKOS_CXX_EXTRA_FLAGS)
message(STATUS "Setting KOKKOS_CXX_EXTRA_FLAGS for Kokkos...")
set(OLD_CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}") # save it for now
set(CMAKE_CXX_FLAGS ${KOKKOS_CXX_EXTRA_FLAGS})
set(OLD_CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS}") # save it for now
set(CMAKE_EXE_LINKER_FLAGS ${KOKKOS_CXX_EXTRA_FLAGS})
endif()
add_subdirectory(${KOKKOS_IN_TREE} ${CMAKE_BINARY_DIR}/kokkos)
add_definitions(-DKOKKOS_IN_TREE=${KOKKOS_IN_TREE})
else ()
find_package(Kokkos REQUIRED)
endif ()
set(SOURCES
src/bude.cpp
src/vec-pose-inner.cpp)
include_directories(src)
add_executable(bude ${SOURCES})
separate_arguments(CXX_EXTRA_FLAGS)
separate_arguments(CXX_EXTRA_LINKER_FLAGS)
target_compile_options(bude
PUBLIC
-Wall
-Wextra
-Wcast-align
-Wfatal-errors
-Werror=return-type
-Wno-unused-parameter
-Wno-unused-variable
-Wno-ignored-attributes
${EXTRA_FLAGS}
)
if (NOT DEFINED WG_SIZE)
message(STATUS "WG_SIZE unspecified, using source default (available with --help in the compiled binary)")
else ()
if (NOT WG_SIZE MATCHES "^[0-9]+$")
message(FATAL_ERROR "WG_SIZE must be an integer, got ${WG_SIZE}")
else ()
add_definitions(-DWG_SIZE=${WG_SIZE})
endif ()
endif ()
set(DEBUG_OPTIONS -O2 -fno-omit-frame-pointer ${CXX_EXTRA_FLAGS})
set(RELEASE_OPTIONS -O3 -ffast-math ${CXX_EXTRA_FLAGS}) #nvcc can't handle -Ofast, must be -O<n>
target_compile_options(bude PUBLIC "$<$<CONFIG:RelWithDebInfo>:${RELEASE_OPTIONS}>")
target_compile_options(bude PUBLIC "$<$<CONFIG:Release>:${RELEASE_OPTIONS}>")
target_compile_options(bude PUBLIC "$<$<CONFIG:Debug>:${DEBUG_OPTIONS}>")
if(DEFINED OLD_CMAKE_CXX_FLAGS) # restore if overwritten before, as required by Kokkos to pick up CXX_EXTRA_FLAGS as CMAKE_CXX_FLAGS
set(CMAKE_CXX_FLAGS ${OLD_CMAKE_CXX_FLAGS})
endif()
if(DEFINED OLD_CMAKE_EXE_LINKER_FLAGS)
set(CMAKE_EXE_LINKER_FLAGS ${OLD_CMAKE_EXE_LINKER_FLAGS})
endif()
target_link_libraries(bude Kokkos::kokkos)
if (${CMAKE_VERSION} VERSION_LESS "3.13.0")
message(WARNING "target_link_options is only available in CMake >= 3.13.0, using fallback target_link_libraries, this may cause issues with some compilers")
message(WARNING "whitespaces are not supported for CXX_EXTRA_LINKER_FLAGS/CXX_EXTRA_FLAGS in this mode as they are treated as libraries arguments (CMake splits them)")
if (DEFINED CXX_EXTRA_LINKER_FLAGS)
list(APPEND EXTRA_LINK_FLAGS "-Wl,${CXX_EXTRA_LINKER_FLAGS}")
endif ()
target_link_libraries(bude PUBLIC ${EXTRA_LINK_FLAGS})
target_link_libraries(bude PUBLIC ${CXX_EXTRA_FLAGS})
else ()
target_link_options(bude PUBLIC LINKER:${CXX_EXTRA_LINKER_FLAGS})
target_link_options(bude PUBLIC ${CXX_EXTRA_FLAGS})
endif ()