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
|
project(libshm C CXX)
cmake_minimum_required(VERSION 2.8.12 FATAL_ERROR)
cmake_policy(VERSION 2.8.12)
set(TORCH_ROOT ${CMAKE_CURRENT_LIST_DIR}/../../../)
include(${TORCH_ROOT}/cmake/public/threads.cmake)
if(NOT LIBSHM_INSTALL_LIB_SUBDIR)
set(LIBSHM_INSTALL_LIB_SUBDIR "lib" CACHE PATH "libshm install library directory")
endif()
# Flags
# When using MSVC
if(MSVC)
# we want to respect the standard, and we are bored of those **** .
add_definitions(-D_CRT_SECURE_NO_DEPRECATE=1)
endif(MSVC)
if(CMAKE_VERSION VERSION_LESS "3.1")
set(CMAKE_CXX_FLAGS "-std=c++14 ${CMAKE_CXX_FLAGS}")
else()
set(CMAKE_CXX_STANDARD 14 CACHE STRING "The C++ standard whose features are requested to build this target.")
endif()
add_library(shm SHARED core.cpp)
if(HAVE_SOVERSION)
set_target_properties(shm PROPERTIES
VERSION ${TORCH_VERSION} SOVERSION ${TORCH_SOVERSION})
endif()
target_include_directories(shm PUBLIC
${CMAKE_BINARY_DIR}/aten/src # provides "ATen/TypeExtendedInterface.h" to ATen.h
${TORCH_ROOT}/torch/lib # provides "libshm/libshm.h"
)
add_executable(torch_shm_manager manager.cpp)
target_link_libraries(torch_shm_manager shm)
set_target_properties(torch_shm_manager PROPERTIES INSTALL_RPATH "${_rpath_portable_origin}/../lib")
### Torch packages supposes libraries prefix is "lib"
set_target_properties(shm PROPERTIES
PREFIX "lib"
IMPORT_PREFIX "lib")
target_link_libraries(shm torch c10)
if(UNIX AND NOT APPLE)
include(CheckLibraryExists)
# https://github.com/libgit2/libgit2/issues/2128#issuecomment-35649830
check_library_exists(rt clock_gettime "time.h" NEED_LIBRT)
if(NEED_LIBRT)
target_link_libraries(shm rt)
target_link_libraries(torch_shm_manager rt)
else()
message(STATUS "Checking if rt requires pthread")
# Sometimes, rt won't be available unless you also link against
# pthreads. In this case, the NEED_LIBRT test will fail, because
# check_library_exists isn't going to build the C file with the
# pthread file, and the build will fail, setting NEED_LIBRT to
# false (this is TOTALLY BOGUS, this situation should be an error
# situation, not a "oh, I guess rt is not supported", but it's
# not too easy to distinguish between the two situations). So,
# if it fails, we try again, but this time also with a dependency
# on pthread. If it succeeds this time, we know we not only need
# an rt dependency, but we also need pthread.
#
# BTW, this test looks for shm_open, because that's what we
# really care about (not clock_gettime). I didn't change the
# site above though in case there was a reason we were testing
# against clock_gettime. In principle, the choice of symbol you
# test for shouldn't matter.
set(CMAKE_REQUIRED_LIBRARIES caffe2::Threads)
check_library_exists(rt shm_open "sys/mman.h" NEED_RT_AND_PTHREAD)
unset(CMAKE_REQUIRED_LIBRARIES)
if(NEED_RT_AND_PTHREAD)
message(STATUS "Needs it, linking against pthread and rt")
target_link_libraries(shm rt caffe2::Threads)
target_link_libraries(torch_shm_manager rt caffe2::Threads)
endif()
endif()
endif()
install(TARGETS shm LIBRARY DESTINATION ${LIBSHM_INSTALL_LIB_SUBDIR})
install(FILES libshm.h DESTINATION "include")
install(TARGETS torch_shm_manager DESTINATION "bin")
|
