diff options
| author | Anders Papitto <anderspapitto@gmail.com> | 2018-10-29 12:02:37 -0700 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2018-10-29 12:04:32 -0700 |
| commit | 380d2dfb272bb6f56d157acc26d21e0bfb2857f4 (patch) | |
| tree | 45a6a3f3c89232866a68c2711934efa73e95533a /third_party/nccl | |
| parent | 1c8a823b3b3b67e4e1bcf5d1102b6cfbda7cef15 (diff) | |
| download | pytorch-380d2dfb272bb6f56d157acc26d21e0bfb2857f4.tar.gz pytorch-380d2dfb272bb6f56d157acc26d21e0bfb2857f4.tar.bz2 pytorch-380d2dfb272bb6f56d157acc26d21e0bfb2857f4.zip | |
absorb nccl (#13150)
Summary:
always build nccl from within the main cmake build, rather than via a separate invocation in build_pytorch_libs.sh. Use the existing caffe2 codepaths
Pull Request resolved: https://github.com/pytorch/pytorch/pull/13150
Differential Revision: D12815674
Pulled By: anderspapitto
fbshipit-source-id: a710b6f242d159b9816911a25ee2c4b8c3f855aa
Diffstat (limited to 'third_party/nccl')
| -rw-r--r-- | third_party/nccl/CMakeLists.txt | 21 |
1 files changed, 0 insertions, 21 deletions
diff --git a/third_party/nccl/CMakeLists.txt b/third_party/nccl/CMakeLists.txt deleted file mode 100644 index 2ac095ff1d..0000000000 --- a/third_party/nccl/CMakeLists.txt +++ /dev/null @@ -1,21 +0,0 @@ -CMAKE_MINIMUM_REQUIRED(VERSION 2.8 FATAL_ERROR) -CMAKE_POLICY(VERSION 2.8) - -IF(NOT CUDA_FOUND) - FIND_PACKAGE(CUDA 7.0 REQUIRED) -ENDIF() - -include("${CMAKE_UTILS_PATH}") -torch_cuda_get_nvcc_gencode_flag(NVCC_GENCODE) -string(REPLACE "-gencode;" "-gencode=" NVCC_GENCODE "${NVCC_GENCODE}") -message(STATUS "Set NVCC_GENCODE for building NCCL: ${NVCC_GENCODE}") - -ADD_CUSTOM_COMMAND( - WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}/nccl - OUTPUT ${CMAKE_CURRENT_BINARY_DIR}/lib/libnccl.so - COMMAND env CUDA_HOME=${CUDA_TOOLKIT_ROOT_DIR} NVCC=${CUDA_NVCC_EXECUTABLE} BUILDDIR=${CMAKE_CURRENT_BINARY_DIR} NVCC_GENCODE="${NVCC_GENCODE}" make -j${NUM_JOBS} -) - -ADD_CUSTOM_TARGET(nccl ALL DEPENDS ${CMAKE_CURRENT_BINARY_DIR}/lib/libnccl.so) - -INSTALL(FILES ${CMAKE_CURRENT_BINARY_DIR}/include/nccl.h DESTINATION "include") |