Added the ability to compute the absolute value of a complex number on GPU, as well as a test to catch the problem.

This commit is contained in:
Benoit Steiner
2016-09-12 13:46:13 -07:00
parent 8321dcce76
commit 5f50f12d2c
3 changed files with 89 additions and 0 deletions

View File

@@ -227,6 +227,7 @@ if(CUDA_FOUND AND EIGEN_TEST_CUDA)
ei_add_test(cxx11_tensor_device)
ei_add_test(cxx11_tensor_cuda)
ei_add_test(cxx11_tensor_complex_cuda)
ei_add_test(cxx11_tensor_contract_cuda)
ei_add_test(cxx11_tensor_reduction_cuda)
ei_add_test(cxx11_tensor_argmax_cuda)