GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-08-14 12:46:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fixed the flags passed to nvcc to compile the tensor code.

Browse Source
This commit is contained in:
Benoit Steiner 2016-01-27 14:46:34 -08:00
parent 4865e1e732
commit 55a5204319
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
unsupported/test/CMakeLists.txt
View File

@ -157,15 +157,16 @@ if(CUDA_FOUND)
if("${CMAKE_CXX_COMPILER_ID}" STREQUAL "Clang") if("${CMAKE_CXX_COMPILER_ID}" STREQUAL "Clang")
set(CUDA_NVCC_FLAGS "-ccbin /usr/bin/clang" CACHE STRING "nvcc flags" FORCE) set(CUDA_NVCC_FLAGS "-ccbin /usr/bin/clang" CACHE STRING "nvcc flags" FORCE)
endif() endif()
set(CUDA_NVCC_FLAGS "-std=c++11 -arch compute_30")
cuda_include_directories("${CMAKE_CURRENT_BINARY_DIR}" "${CUDA_TOOLKIT_ROOT_DIR}/include") cuda_include_directories("${CMAKE_CURRENT_BINARY_DIR}" "${CUDA_TOOLKIT_ROOT_DIR}/include")
set(EIGEN_ADD_TEST_FILENAME_EXTENSION "cu") set(EIGEN_ADD_TEST_FILENAME_EXTENSION "cu")
ei_add_test(cxx11_tensor_device "-std=c++11") ei_add_test(cxx11_tensor_device)
ei_add_test(cxx11_tensor_cuda "-std=c++11") ei_add_test(cxx11_tensor_cuda)
ei_add_test(cxx11_tensor_contract_cuda "-std=c++11") ei_add_test(cxx11_tensor_contract_cuda)
ei_add_test(cxx11_tensor_reduction_cuda "-std=c++11") ei_add_test(cxx11_tensor_reduction_cuda)
ei_add_test(cxx11_tensor_random_cuda "-std=c++11") ei_add_test(cxx11_tensor_random_cuda)
ei_add_test(cxx11_tensor_argmax_cuda "-std=c++11 -I/opt-cuda-7.0/include") ei_add_test(cxx11_tensor_argmax_cuda)
unset(EIGEN_ADD_TEST_FILENAME_EXTENSION) unset(EIGEN_ADD_TEST_FILENAME_EXTENSION)
endif(CUDA_FOUND) endif(CUDA_FOUND)