diff options
| author | Soumith Chintala <soumith@fb.com> | 2019-04-02 09:36:05 -0700 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2019-04-02 09:39:03 -0700 |
| commit | a799751e33809459584e476ade99a110f4b5500c (patch) | |
| tree | afa300cd87fa952554aba4c27328bbcf29059525 /cmake | |
| parent | 1f5a46ab052df45a603271f1a7b2b32081a42ef8 (diff) | |
| download | pytorch-a799751e33809459584e476ade99a110f4b5500c.tar.gz pytorch-a799751e33809459584e476ade99a110f4b5500c.tar.bz2 pytorch-a799751e33809459584e476ade99a110f4b5500c.zip | |
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
Diffstat (limited to 'cmake')
| -rw-r--r-- | cmake/External/nccl.cmake | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/cmake/External/nccl.cmake b/cmake/External/nccl.cmake index 56fdbd5287..47c6027477 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}") |