-
Notifications
You must be signed in to change notification settings - Fork 9
/
CMakeLists.txt
101 lines (82 loc) · 3.44 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
cmake_minimum_required(VERSION 3.20 FATAL_ERROR)
project(
cumm
LANGUAGES CXX
VERSION 1.0)
option(CUMM_DISABLE_CMAKE_INSTALL "disable cumm install" OFF)
message (STATUS "AWDSAF ${CUMM_DISABLE_CMAKE_INSTALL}")
set(CUMM_INCLUDE_PATH ${${PROJECT_NAME}_SOURCE_DIR}/include PARENT_SCOPE)
add_library(${PROJECT_NAME}_cpu INTERFACE)
add_library(${PROJECT_NAME}::${PROJECT_NAME}_cpu ALIAS ${PROJECT_NAME}_cpu)
target_include_directories(
${PROJECT_NAME}_cpu
INTERFACE $<BUILD_INTERFACE:${${PROJECT_NAME}_SOURCE_DIR}/include>
$<INSTALL_INTERFACE:${CMAKE_INSTALL_INCLUDEDIR}>)
target_compile_features(${PROJECT_NAME}_cpu INTERFACE cxx_std_14)
include(GNUInstallDirs)
find_package(CUDAToolkit)
find_package(OpenMP)
if(OpenMP_CXX_FOUND)
target_link_libraries(${PROJECT_NAME}_cpu INTERFACE OpenMP::OpenMP_CXX)
endif()
if(${CUDAToolkit_FOUND})
add_library(${PROJECT_NAME} INTERFACE)
add_library(${PROJECT_NAME}::${PROJECT_NAME} ALIAS ${PROJECT_NAME})
target_include_directories(
${PROJECT_NAME}
INTERFACE $<BUILD_INTERFACE:${${PROJECT_NAME}_SOURCE_DIR}/include>
$<INSTALL_INTERFACE:${CMAKE_INSTALL_INCLUDEDIR}>)
target_compile_features(${PROJECT_NAME} INTERFACE cxx_std_14)
if(OpenMP_CXX_FOUND)
target_link_libraries(${PROJECT_NAME} INTERFACE OpenMP::OpenMP_CXX CUDA::cudart)
if (WIN32)
target_compile_options(${PROJECT_NAME} INTERFACE $<$<COMPILE_LANGUAGE:CUDA>:--extended-lambda -Xcompiler="/openmp">)
else()
target_compile_options(${PROJECT_NAME} INTERFACE $<$<COMPILE_LANGUAGE:CUDA>:--extended-lambda -Xcompiler="-fopenmp">)
endif()
else()
target_link_libraries(${PROJECT_NAME} INTERFACE CUDA::cudart)
target_compile_options(${PROJECT_NAME} INTERFACE $<$<COMPILE_LANGUAGE:CUDA>:--extended-lambda>)
endif()
target_compile_definitions(${PROJECT_NAME} INTERFACE TV_ENABLE_HARDWARE_ACC)
if (NOT CUMM_DISABLE_CMAKE_INSTALL)
install(
TARGETS ${PROJECT_NAME} ${PROJECT_NAME}_cpu
EXPORT ${PROJECT_NAME}_Targets
ARCHIVE DESTINATION ${CMAKE_INSTALL_LIBDIR}
LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR}
RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR})
endif()
else()
if (NOT CUMM_DISABLE_CMAKE_INSTALL)
install(
TARGETS ${PROJECT_NAME}_cpu
EXPORT ${PROJECT_NAME}_Targets
ARCHIVE DESTINATION ${CMAKE_INSTALL_LIBDIR}
LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR}
RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR})
endif()
endif()
set(CUMM_CMAKECONFIG_INSTALL_DIR
"${CMAKE_INSTALL_DATAROOTDIR}/cmake/${PROJECT_NAME}"
CACHE STRING "install path for cummConfig.cmake")
include(CMakePackageConfigHelpers)
write_basic_package_version_file(
"${PROJECT_NAME}ConfigVersion.cmake"
VERSION ${PROJECT_VERSION}
COMPATIBILITY SameMajorVersion)
configure_package_config_file(
"${PROJECT_SOURCE_DIR}/cmake/${PROJECT_NAME}Config.cmake.in"
"${PROJECT_BINARY_DIR}/${PROJECT_NAME}Config.cmake"
INSTALL_DESTINATION ${CMAKE_INSTALL_DATAROOTDIR}/${PROJECT_NAME}/cmake)
if (NOT CUMM_DISABLE_CMAKE_INSTALL)
install(
EXPORT ${PROJECT_NAME}_Targets
FILE ${PROJECT_NAME}Targets.cmake
NAMESPACE ${PROJECT_NAME}::
DESTINATION ${CUMM_CMAKECONFIG_INSTALL_DIR})
install(FILES "${PROJECT_BINARY_DIR}/${PROJECT_NAME}Config.cmake"
"${PROJECT_BINARY_DIR}/${PROJECT_NAME}ConfigVersion.cmake"
DESTINATION ${CUMM_CMAKECONFIG_INSTALL_DIR})
install(DIRECTORY ${PROJECT_SOURCE_DIR}/include/tensorview DESTINATION include)
endif()