diff options
| author | Soumith Chintala <soumith@gmail.com> | 2017-06-08 01:16:42 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2017-06-08 01:16:42 +0300 |
| commit | 5c1d3cfda8101123628a45e70435d545ae1bc771 (patch) | |
| tree | 252617dbd8d50032de635eb825a843c399702146 | |
| parent | 54a378af2b2eef0a2e1e1a0396b26f5973d7c36f (diff) | |
| parent | 296022dac400bf7b80698f409e822b4c824a1817 (diff) | |
Merge pull request #227 from anirudh2290/p2_build_support
Fix to Support building on p2(AWS EC2 Instance type) for multiple CUDA Architectures
| -rw-r--r-- | cmake/3.6/Modules/FindCUDA/select_compute_arch.cmake | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/cmake/3.6/Modules/FindCUDA/select_compute_arch.cmake b/cmake/3.6/Modules/FindCUDA/select_compute_arch.cmake index 5376e5b..0c02a8a 100644 --- a/cmake/3.6/Modules/FindCUDA/select_compute_arch.cmake +++ b/cmake/3.6/Modules/FindCUDA/select_compute_arch.cmake @@ -88,7 +88,9 @@ endfunction() # Usage: # SELECT_NVCC_ARCH_FLAGS(out_variable [list of CUDA compute archs]) function(CUDA_SELECT_NVCC_ARCH_FLAGS out_variable) - set(CUDA_ARCH_LIST "${ARGN}") + if(ARGN) + set(CUDA_ARCH_LIST "${ARGN}") + endif() if("X${CUDA_ARCH_LIST}" STREQUAL "X" ) set(CUDA_ARCH_LIST "Auto") |