Added preliminary support for half floats on CUDA GPU. For now we can simply convert floats into half floats and vice versa

This commit is contained in:
Benoit Steiner
2016-02-19 06:16:07 +00:00
parent 8ce46f9d89
commit 17b9fbed34
4 changed files with 109 additions and 3 deletions

View File

@@ -37,9 +37,9 @@ if (NOT CMAKE_CXX_COMPILER MATCHES "clang\\+\\+$")
ei_add_test(BVH)
endif()
ei_add_test(matrix_exponential)
#ei_add_test(matrix_exponential)
ei_add_test(matrix_function)
ei_add_test(matrix_power)
#ei_add_test(matrix_power)
ei_add_test(matrix_square_root)
ei_add_test(alignedvector3)
@@ -173,5 +173,9 @@ if(CUDA_FOUND)
ei_add_test(cxx11_tensor_random_cuda)
ei_add_test(cxx11_tensor_argmax_cuda)
set(CUDA_NVCC_FLAGS "-std=c++11 --relaxed-constexpr -arch compute_53 -Xcudafe \"--display_error_number\"")
ei_add_test(cxx11_tensor_of_float16_cuda)
unset(EIGEN_ADD_TEST_FILENAME_EXTENSION)
endif()