diff options
| author |  Benoit Steiner <benoit.steiner.goog@gmail.com> | 2017-08-31 02:49:39 +0000 |
|---|---|---|
| committer | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2017-08-31 02:49:39 +0000 |
| commit | a4089991eb6bdb9e8ebfef93d81ca7b5e67ea77d (patch) | |
| tree | 49a9b6c0c4ec6d006debe862cf209a8f252cfe78 /unsupported/test/cxx11_tensor_argmax_cuda.cu | |
| parent | 304ef2957134be386e50592ad7120177c5f3a7c0 (diff) | |
Added support for CUDA 9.0.
Diffstat (limited to 'unsupported/test/cxx11_tensor_argmax_cuda.cu')
| -rw-r--r-- | unsupported/test/cxx11_tensor_argmax_cuda.cu | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/unsupported/test/cxx11_tensor_argmax_cuda.cu b/unsupported/test/cxx11_tensor_argmax_cuda.cu index 3d73d491a..0e8b8125d 100644 --- a/unsupported/test/cxx11_tensor_argmax_cuda.cu +++ b/unsupported/test/cxx11_tensor_argmax_cuda.cu @@ -15,6 +15,12 @@ #include "main.h" #include <unsupported/Eigen/CXX11/Tensor> +// The EIGEN_CUDACC_VER macro is provided by +// unsupported/Eigen/CXX11/Tensor included above +#if defined EIGEN_CUDACC_VER && EIGEN_CUDACC_VER >= 70500 +#include <cuda_fp16.h> +#endif + using Eigen::Tensor; template <int Layout> |