diff options
author | Rui Guo <guorui.xt@gmail.com> | 2016-11-28 09:02:21 +0300 |
---|---|---|
committer | Rui Guo <guorui.xt@gmail.com> | 2016-11-28 09:02:21 +0300 |
commit | 5a80d2eca40099066dd6ecd5a40a9ab53c634845 (patch) | |
tree | d07f68c5c85880c2557eb2269701c23dbd137e33 | |
parent | 1793771859ec96dd9a94b7d5df51401506fc7fb6 (diff) |
use local modified select_compute_arch.cmake for msvc
-rw-r--r-- | lib/THCUNN/CMakeLists.txt | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/lib/THCUNN/CMakeLists.txt b/lib/THCUNN/CMakeLists.txt index d86dfdc..bce4cbd 100644 --- a/lib/THCUNN/CMakeLists.txt +++ b/lib/THCUNN/CMakeLists.txt @@ -10,7 +10,7 @@ IF(NOT CUDA_FOUND) ENDIF() # Detect CUDA architecture and get best NVCC flags -IF(NOT COMMAND CUDA_SELECT_NVCC_ARCH_FLAGS) +IF(NOT COMMAND CUDA_SELECT_NVCC_ARCH_FLAGS OR MSVC) INCLUDE(${CMAKE_CURRENT_SOURCE_DIR}/cmake/select_compute_arch.cmake) ENDIF() CUDA_SELECT_NVCC_ARCH_FLAGS(NVCC_FLAGS_EXTRA $ENV{TORCH_CUDA_ARCH_LIST}) |