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
75 lines
2.5 KiB
CMake
75 lines
2.5 KiB
CMake
# Build file for the C10 CUDA.
|
|
#
|
|
# C10 CUDA is a minimal library, but it does depend on CUDA.
|
|
|
|
include(../../cmake/public/utils.cmake)
|
|
include(../../cmake/public/cuda.cmake)
|
|
|
|
# ---[ Configure macro file.
|
|
set(C10_CUDA_BUILD_SHARED_LIBS ${BUILD_SHARED_LIBS}) # used in cmake_macros.h.in
|
|
# Probably have to do this :(
|
|
configure_file(
|
|
${CMAKE_CURRENT_LIST_DIR}/impl/cuda_cmake_macros.h.in
|
|
${CMAKE_BINARY_DIR}/c10/cuda/impl/cuda_cmake_macros.h)
|
|
|
|
# Note: if you want to add ANY dependency to the c10 library, make sure you
|
|
# check with the core PyTorch developers as the dependendency will be
|
|
# transitively passed on to all libraries dependent on PyTorch.
|
|
|
|
# Note: if you add a new source file/header, you will need to update
|
|
# torch/utils/hipify/cuda_to_hip_mappings.py for new files
|
|
# and headers you add
|
|
set(C10_CUDA_SRCS
|
|
CUDAStream.cpp
|
|
CUDACachingAllocator.cpp
|
|
impl/CUDAGuardImpl.cpp
|
|
impl/CUDATest.cpp
|
|
)
|
|
set(C10_CUDA_HEADERS
|
|
CUDAException.h
|
|
CUDAGuard.h
|
|
CUDAMacros.h
|
|
CUDAMathCompat.h
|
|
CUDAStream.h
|
|
impl/CUDAGuardImpl.h
|
|
impl/CUDATest.h
|
|
)
|
|
set(CUDA_LINK_LIBRARIES_KEYWORD PRIVATE)
|
|
torch_cuda_based_add_library(c10_cuda ${C10_CUDA_SRCS} ${C10_CUDA_HEADERS})
|
|
set(CUDA_LINK_LIBRARIES_KEYWORD)
|
|
# If building shared library, set dllimport/dllexport proper.
|
|
target_compile_options(c10_cuda PRIVATE "-DC10_CUDA_BUILD_MAIN_LIB")
|
|
# Enable hidden visibility if compiler supports it.
|
|
if(${COMPILER_SUPPORTS_HIDDEN_VISIBILITY})
|
|
target_compile_options(c10_cuda PRIVATE "-fvisibility=hidden")
|
|
endif()
|
|
|
|
# ---[ Dependency of c10_cuda
|
|
target_link_libraries(c10_cuda PUBLIC c10)
|
|
|
|
target_link_libraries(c10_cuda INTERFACE torch::cudart)
|
|
|
|
target_include_directories(
|
|
c10_cuda PUBLIC
|
|
$<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/../..>
|
|
$<BUILD_INTERFACE:${CMAKE_BINARY_DIR}>
|
|
$<INSTALL_INTERFACE:include>)
|
|
|
|
add_subdirectory(test)
|
|
|
|
# ---[ Installation
|
|
# Note: for now, we will put all export path into one single Caffe2Targets group
|
|
# to deal with the cmake deployment need. Inside the Caffe2Targets set, the
|
|
# individual libraries like libc10.so and libcaffe2.so are still self-contained.
|
|
install(TARGETS c10_cuda EXPORT Caffe2Targets DESTINATION lib)
|
|
foreach(file ${C10_CUDA_HEADERS})
|
|
get_filename_component( dir ${file} DIRECTORY )
|
|
install( FILES ${file} DESTINATION include/c10/cuda/${dir} )
|
|
endforeach()
|
|
install(FILES ${CMAKE_BINARY_DIR}/c10/cuda/impl/cuda_cmake_macros.h
|
|
DESTINATION include/c10/cuda/impl)
|
|
|
|
if(MSVC AND C10_CUDA_BUILD_SHARED_LIBS)
|
|
install(FILES $<TARGET_PDB_FILE:c10_cuda> DESTINATION lib OPTIONAL)
|
|
endif()
|