forked from kokkos/kokkos-remote-spaces
-
Notifications
You must be signed in to change notification settings - Fork 0
/
CMakeLists.txt
163 lines (137 loc) · 4.89 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
cmake_minimum_required (VERSION 3.12)
project(KokkosRemote
LANGUAGES CXX
VERSION 0.1.0)
list(APPEND CMAKE_MODULE_PATH ${CMAKE_CURRENT_SOURCE_DIR}/modules)
find_package(Kokkos REQUIRED)
option(Kokkos_ENABLE_NVSHMEMSPACE "Whether to build with NVSHMEM space" OFF)
option(Kokkos_ENABLE_SHMEMSPACE "Whether to build with SHMEMS space" OFF)
option(Kokkos_ENABLE_MPISPACE "Whether to build with MPI space" OFF)
option(Kokkos_ENABLE_TESTS "Whether to enable tests" OFF)
option(Kokkos_ENABLE_DEBUG "Whether to enable debugging output" OFF)
set(SOURCE_DIRS)
set(PUBLIC_DEPS)
set(PRIVATE_DEPS)
set(BACKENDS)
set(PREFIX_BACKEND_SRC_PATH "src/impl")
set(NVSHMEMSPACE_PATH "${PREFIX_BACKEND_SRC_PATH}/nvshmemspace")
set(SHMEMSPACE_PATH "${PREFIX_BACKEND_SRC_PATH}/shmemspace")
set(MPISPACE_PATH "${PREFIX_BACKEND_SRC_PATH}/mpispace")
find_package(MPI REQUIRED)
add_library(MPI INTERFACE)
list(APPEND MPI_CXX_LINK_FLAGS ${MPI_CXX_LIBRARIES})
set_target_properties(MPI PROPERTIES
INTERFACE_COMPILE_OPTIONS "${MPI_CXX_COMPILE_FLAGS}"
INTERFACE_INCLUDE_DIRECTORIES "${MPI_CXX_INCLUDE_PATH}"
INTERFACE_LINK_LIBRARIES "${MPI_CXX_LINK_FLAGS}"
)
list(APPEND PUBLIC_DEPS MPI)
if (Kokkos_ENABLE_NVSHMEMSPACE)
# Make sure kokkos has RDC and lambda support
kokkos_check(
OPTIONS CUDA_LAMBDA CUDA_RELOCATABLE_DEVICE_CODE
DEVICES CUDA)
find_package(NVSHMEM REQUIRED)
set(BACKEND_NAME NVSHMEMSPACE)
list(APPEND SOURCE_DIRS ${NVSHMEMSPACE_PATH})
list(APPEND PUBLIC_DEPS ${BACKEND_NAME})
list(APPEND BACKENDS ${BACKEND_NAME})
endif()
if (Kokkos_ENABLE_SHMEMSPACE)
find_package(SHMEM REQUIRED)
set(BACKEND_NAME SHMEMSPACE)
list(APPEND SOURCE_DIRS ${SHMEMSPACE_PATH})
list(APPEND PUBLIC_DEPS ${BACKEND_NAME})
list(APPEND BACKENDS ${BACKEND_NAME})
endif()
if (Kokkos_ENABLE_MPISPACE)
add_library(MPISPACE INTERFACE)
set(BACKEND_NAME MPISPACE)
list(APPEND SOURCE_DIRS ${MPISPACE_PATH})
list(APPEND PUBLIC_DEPS ${BACKEND_NAME})
list(APPEND BACKENDS ${BACKEND_NAME})
endif()
if (Kokkos_ENABLE_RACERLIB)
find_package(IBVERBS REQUIRED)
list(APPEND PUBLIC_DEPS IBVERBS)
endif()
message(STATUS "Enabled remote spaces: ${BACKENDS}")
list(LENGTH BACKENDS N_BACKENDS)
if (NOT ${N_BACKENDS} EQUAL "1")
message(FATAL_ERROR "Must give a single valid backend")
endif()
set(SOURCES)
set(HEADERS)
foreach(DIR ${SOURCE_DIRS})
file(GLOB DIR_SRCS ${DIR}/*.cpp)
file(GLOB DIR_HDRS ${DIR}/*.hpp)
list(APPEND SOURCES ${DIR_SRCS})
list(APPEND HEADERS ${DIR_HDRS})
endforeach()
if (Kokkos_ENABLE_RACERLIB)
file(GLOB DIR_SRCS ${CMAKE_CURRENT_SOURCE_DIR}/src/features/racerlib/*.cpp)
file(GLOB DIR_HDRS ${CMAKE_CURRENT_SOURCE_DIR}/src/features/racerlib/*.hpp)
list(APPEND SOURCES ${DIR_SRCS})
list(APPEND HEADERS ${DIR_HDRS})
endif()
add_library(kokkosremote ${SOURCES} ${HEADERS})
add_library(Kokkos::kokkosremote ALIAS kokkosremote)
target_link_libraries(kokkosremote PUBLIC Kokkos::kokkos)
foreach(DIR ${SOURCE_DIRS})
target_include_directories(kokkosremote PUBLIC $<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/${DIR}>)
foreach(BACKEND_NAME ${PUBLIC_DEPS})
target_compile_definitions(kokkosremote PUBLIC KOKKOS_ENABLE_${BACKEND_NAME})
endforeach()
endforeach()
if (Kokkos_ENABLE_RACERLIB)
target_include_directories(kokkosremote PUBLIC $<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/src/features/racerlib>)
target_compile_definitions(kokkosremote PUBLIC KOKKOS_ENABLE_RACERLIB)
endif()
target_include_directories(kokkosremote PUBLIC $<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/src/core>)
target_include_directories(kokkosremote PUBLIC $<INSTALL_INTERFACE:include>)
if(Kokkos_ENABLE_DEBUG)
target_compile_definitions(kokkosremote PUBLIC KOKKOS_IBV_DEBUG)
endif()
if (PRIVATE_DEPS)
target_link_libraries(kokkosremote PRIVATE ${PRIVATE_DEPS})
endif()
if (PUBLIC_DEPS)
target_link_libraries(kokkosremote PUBLIC ${PUBLIC_DEPS})
endif()
include(GNUInstallDirs)
include(CMakePackageConfigHelpers)
configure_package_config_file(
KokkosRemoteConfig.cmake.in
"${CMAKE_CURRENT_BINARY_DIR}/KokkosRemoteConfig.cmake"
INSTALL_DESTINATION ${CMAKE_INSTALL_FULL_LIBDIR}/cmake
)
write_basic_package_version_file(
"${CMAKE_CURRENT_BINARY_DIR}/KokkosRemoteConfigVersion.cmake"
VERSION "${KokkosRemote_VERSION}"
COMPATIBILITY AnyNewerVersion
)
install(FILES
"${CMAKE_CURRENT_BINARY_DIR}/KokkosRemoteConfig.cmake"
"${CMAKE_CURRENT_BINARY_DIR}/KokkosRemoteConfigVersion.cmake"
DESTINATION ${CMAKE_INSTALL_LIBDIR}/cmake/KokkosRemote)
install(FILES
${HEADERS}
DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}
)
install(
TARGETS kokkosremote ${PUBLIC_DEPS}
EXPORT KokkosRemoteTargets
RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR}
LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR}
ARCHIVE DESTINATION ${CMAKE_INSTALL_LIBDIR}
)
install(EXPORT
KokkosRemoteTargets
NAMESPACE Kokkos::
DESTINATION ${CMAKE_INSTALL_LIBDIR}/cmake/KokkosRemote
)
add_subdirectory(examples)
IF (Kokkos_ENABLE_TESTS)
enable_testing()
add_subdirectory(unit_tests)
ENDIF()