diff options
| author |  Gael Guennebaud <g.gael@free.fr> | 2016-09-05 17:14:20 +0200 |
|---|---|---|
| committer | Gael Guennebaud <g.gael@free.fr> | 2016-09-05 17:14:20 +0200 |
| commit | dabc81751fe2a1eed16967e140e65558e52b5a5f (patch) | |
| tree | bd92ebae15688f37395f6c40463a43cc46f3d5cb /unsupported/test/cxx11_tensor_device.cu | |
| parent | e13071dd13233b5d07e725453617c40421cf7501 (diff) | |
Fix compilation when cuda_fp16.h does not exist.
Diffstat (limited to 'unsupported/test/cxx11_tensor_device.cu')
| -rw-r--r-- | unsupported/test/cxx11_tensor_device.cu | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/unsupported/test/cxx11_tensor_device.cu b/unsupported/test/cxx11_tensor_device.cu index bbda95906..fde20ddf2 100644 --- a/unsupported/test/cxx11_tensor_device.cu +++ b/unsupported/test/cxx11_tensor_device.cu @@ -13,7 +13,9 @@ #define EIGEN_DEFAULT_DENSE_INDEX_TYPE int #define EIGEN_USE_GPU +#if defined __CUDACC_VER__ && __CUDACC_VER__ >= 70500 #include <cuda_fp16.h> +#endif #include "main.h" #include <unsupported/Eigen/CXX11/Tensor> |