From fc6296d7775a44b65ff83ae64d9804b69e08f5d3 Mon Sep 17 00:00:00 2001 From: Soumith Chintala Date: Mon, 1 Apr 2019 17:07:08 -0700 Subject: [PATCH] fix nccl compilation to make sure it compiles for architectures that pytorch compiles for (#18704) Summary: cc: t-vi gchanan zou3519 This fixes https://github.com/pytorch/pytorch/issues/18359 Pull Request resolved: https://github.com/pytorch/pytorch/pull/18704 Differential Revision: D14717015 Pulled By: soumith fbshipit-source-id: 4aac036f57e564b05d759662e8ad7a80170901c0 --- 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 47c6027..56fdbd5 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