diff options
| author | Rui Guo <guorui.xt@gmail.com> | 2016-11-28 08:23:24 +0300 |
|---|---|---|
| committer | Rui Guo <guorui.xt@gmail.com> | 2016-11-28 08:23:24 +0300 |
| commit | 8d8bbc3168b7a6fb0bd3ba41b1288cc0fcfdae72 (patch) | |
| tree | 6360d0309f83bc83bd99fa9e445e85309456eff3 | |
| parent | e2051b652d5b1f5182a1bfce11da9f53c2f92bd8 (diff) | |
use local modified select_compute_arch.cmake for msvc
| -rw-r--r-- | lib/THC/CMakeLists.txt | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/lib/THC/CMakeLists.txt b/lib/THC/CMakeLists.txt index 08e6fbc..bc3cbfb 100644 --- a/lib/THC/CMakeLists.txt +++ b/lib/THC/CMakeLists.txt @@ -87,7 +87,7 @@ IF(APPLE) ENDIF(APPLE) # 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() LIST(APPEND CUDA_NVCC_FLAGS $ENV{TORCH_NVCC_FLAGS}) |