mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
45c9ed825a
Summary:
Running commands:
```bash
shopt -s globstar
sed -e 's/IF (/if(/g' -e 's/IF(/if(/g' -e 's/if (/if(/g' -e 's/ELSE (/else(/g' -e 's/ELSE(/else(/g' -e 's/else (/else(/g' -e 's/ENDif(/endif(/g' -e 's/ELSEif(/elseif(/g' -i CMakeLists.txt
sed -e 's/IF (/if(/g' -e 's/IF(/if(/g' -e 's/if (/if(/g' -e 's/ELSE (/else(/g' -e 's/ELSE(/else(/g' -e 's/else (/else(/g' -e 's/ENDif(/endif(/g' -e 's/ELSEif(/elseif(/g' -i caffe2/**/CMakeLists.txt
sed -e 's/IF (/if(/g' -e 's/IF(/if(/g' -e 's/if (/if(/g' -e 's/ELSE (/else(/g' -e 's/ELSE(/else(/g' -e 's/else (/else(/g' -e 's/ENDif(/endif(/g' -e 's/ELSEif(/elseif(/g' -i torch/**/CMakeLists.txt
sed -e 's/IF (/if(/g' -e 's/IF(/if(/g' -e 's/if (/if(/g' -e 's/ELSE (/else(/g' -e 's/ELSE(/else(/g' -e 's/else (/else(/g' -e 's/ENDif(/endif(/g' -e 's/ELSEif(/elseif(/g' -i c10/**/CMakeLists.txt
sed -e 's/IF (/if(/g' -e 's/IF(/if(/g' -e 's/if (/if(/g' -e 's/ELSE (/else(/g' -e 's/ELSE(/else(/g' -e 's/else (/else(/g' -e 's/ENDif(/endif(/g' -e 's/ELSEif(/elseif(/g' -i cmake/**/*.cmake
sed -e 's/IF (/if(/g' -e 's/IF(/if(/g' -e 's/if (/if(/g' -e 's/ELSE (/else(/g' -e 's/ELSE(/else(/g' -e 's/else (/else(/g' -e 's/ENDif(/endif(/g' -e 's/ELSEif(/elseif(/g' -i cmake/**/*.cmake.in
```
We may further convert all the commands into lowercase according to the following issue: 77543bde41.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/35521
Differential Revision: D20704382
Pulled By: malfet
fbshipit-source-id: 42186b9b1660c34428ab7ceb8d3f7a0ced5d2e80
82 lines
3.1 KiB
CMake
82 lines
3.1 KiB
CMake
project(libshm C CXX)
|
|
CMAKE_MINIMUM_REQUIRED(VERSION 2.6 FATAL_ERROR)
|
|
CMAKE_POLICY(VERSION 2.6)
|
|
|
|
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)
|
|
endif()
|
|
|
|
ADD_LIBRARY(shm SHARED core.cpp)
|
|
|
|
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"
|
|
${CMAKE_BINARY_DIR}/caffe2/aten/src # provides <TH/THGeneral.h> to THC.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 Threads::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 Threads::Threads)
|
|
target_link_libraries(torch_shm_manager rt Threads::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")
|