devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Disabled the use of half2 on cuda devices of compute capability < 5.3

Browse Source
This commit is contained in:
Benoit Steiner
2016-04-08 14:43:36 -07:00
parent 8d22967bd9
commit 995f202cea
3 changed files with 5 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
unsupported/test/CMakeLists.txt
View File

@@ -210,10 +210,7 @@ if(CUDA_FOUND AND EIGEN_TEST_CUDA)
ei_add_test(cxx11_tensor_random_cuda)
endif()
# Operations other that casting of half floats are only supported starting with arch 5.3
if (${EIGEN_CUDA_COMPUTE_ARCH} GREATER 52)
ei_add_test(cxx11_tensor_of_float16_cuda)
endif()
ei_add_test(cxx11_tensor_of_float16_cuda)
unset(EIGEN_ADD_TEST_FILENAME_EXTENSION)
endif()