diff options
| author | Clement Farabet <clement.farabet@gmail.com> | 2014-04-10 06:56:25 +0400 |
|---|---|---|
| committer | Clement Farabet <clement.farabet@gmail.com> | 2014-04-10 06:56:25 +0400 |
| commit | 0768e0187f84e3331afab0c6e3936e4a16bd0fc2 (patch) | |
| tree | ec9b55e1eff3d01260a984dff8dfb84ef601c2d5 /CMakeLists.txt | |
| parent | 95905b50beede0f4b8f38a45df7bc1fe62895e55 (diff) | |
Re-imposing sm_20 (missing from the torch-distro split)
Diffstat (limited to 'CMakeLists.txt')
| -rw-r--r-- | CMakeLists.txt | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/CMakeLists.txt b/CMakeLists.txt index b6d1edb..d8a4c3a 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -16,6 +16,8 @@ ENDIF() FIND_PACKAGE(CUDA 4.0 REQUIRED) +LIST(APPEND CUDA_NVCC_FLAGS "-arch=sm_20") + INCLUDE_DIRECTORIES("${Torch_INSTALL_INCLUDE}/THC") LINK_DIRECTORIES("${Torch_INSTALL_LIB}") |