diff options
author | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-01-27 14:46:34 -0800 |
---|---|---|
committer | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-01-27 14:46:34 -0800 |
commit | 55a5204319f119f3609e0522302ff24e69ab60ae (patch) | |
tree | 0155961f3d35afdb03e15787bd00d80e8fbfdc0a /unsupported/test/CMakeLists.txt | |
parent | 4865e1e73265e12d564f8b4d9069a2159f777d90 (diff) |
Fixed the flags passed to nvcc to compile the tensor code.
Diffstat (limited to 'unsupported/test/CMakeLists.txt')
-rw-r--r-- | unsupported/test/CMakeLists.txt | 17 |
1 files changed, 9 insertions, 8 deletions
diff --git a/unsupported/test/CMakeLists.txt b/unsupported/test/CMakeLists.txt index 5c383aab6..3a90a5371 100644 --- a/unsupported/test/CMakeLists.txt +++ b/unsupported/test/CMakeLists.txt @@ -157,15 +157,16 @@ if(CUDA_FOUND) if("${CMAKE_CXX_COMPILER_ID}" STREQUAL "Clang") set(CUDA_NVCC_FLAGS "-ccbin /usr/bin/clang" CACHE STRING "nvcc flags" FORCE) endif() + set(CUDA_NVCC_FLAGS "-std=c++11 -arch compute_30") cuda_include_directories("${CMAKE_CURRENT_BINARY_DIR}" "${CUDA_TOOLKIT_ROOT_DIR}/include") - set(EIGEN_ADD_TEST_FILENAME_EXTENSION "cu") - - ei_add_test(cxx11_tensor_device "-std=c++11") - ei_add_test(cxx11_tensor_cuda "-std=c++11") - ei_add_test(cxx11_tensor_contract_cuda "-std=c++11") - ei_add_test(cxx11_tensor_reduction_cuda "-std=c++11") - ei_add_test(cxx11_tensor_random_cuda "-std=c++11") - ei_add_test(cxx11_tensor_argmax_cuda "-std=c++11 -I/opt-cuda-7.0/include") + set(EIGEN_ADD_TEST_FILENAME_EXTENSION "cu") + + ei_add_test(cxx11_tensor_device) + ei_add_test(cxx11_tensor_cuda) + ei_add_test(cxx11_tensor_contract_cuda) + ei_add_test(cxx11_tensor_reduction_cuda) + ei_add_test(cxx11_tensor_random_cuda) + ei_add_test(cxx11_tensor_argmax_cuda) unset(EIGEN_ADD_TEST_FILENAME_EXTENSION) endif(CUDA_FOUND) |