From a799751e33809459584e476ade99a110f4b5500c Mon Sep 17 00:00:00 2001 From: Soumith Chintala Date: Tue, 2 Apr 2019 09:36:05 -0700 Subject: [PATCH] Revert D14717015: [pytorch][PR] fix nccl compilation to make sure it compiles for architectures that pytorch compiles for Differential Revision: D14717015 Original commit changeset: 4aac036f57e5 fbshipit-source-id: c820b8dfb27564271e6b80e133fe655658a7c25c --- cmake/External/nccl.cmake | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cmake/External/nccl.cmake b/cmake/External/nccl.cmake index 56fdbd5..47c6027 100644 --- a/cmake/External/nccl.cmake +++ b/cmake/External/nccl.cmake @@ -20,8 +20,8 @@ if (NOT __NCCL_INCLUDED) endif() endif() else() - torch_cuda_get_nvcc_gencode_flag(NVCC_GENCODE) if (TORCH_CUDA_ARCH_LIST) + torch_cuda_get_nvcc_gencode_flag(NVCC_GENCODE) string(REPLACE "-gencode;" "-gencode=" NVCC_GENCODE "${NVCC_GENCODE}") # this second replacement is needed when there are multiple archs string(REPLACE ";-gencode" " -gencode" NVCC_GENCODE "${NVCC_GENCODE}") -- 2.7.4