diff options
| author |  Luke Iwanski <luke@codeplay.com> | 2016-09-19 14:03:54 +0100 |
|---|---|---|
| committer | Luke Iwanski <luke@codeplay.com> | 2016-09-19 14:03:54 +0100 |
| commit | b91e0211727b9ea5d7c30908ed86afc4e50d4c6c (patch) | |
| tree | 4dd857d1aabdd4065e8b7dec1369a4bd06bac5e7 /unsupported/test/cxx11_tensor_contract_cuda.cu | |
| parent | cb81975714a96ecb2faf33ca242feeee3543b1db (diff) | |
| parent | ff47717f25aeede4878f65b214cdce264b8314e8 (diff) | |
Merged with default.
Diffstat (limited to 'unsupported/test/cxx11_tensor_contract_cuda.cu')
| -rw-r--r-- | unsupported/test/cxx11_tensor_contract_cuda.cu | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/unsupported/test/cxx11_tensor_contract_cuda.cu b/unsupported/test/cxx11_tensor_contract_cuda.cu index 98ac180ef..767e9c678 100644 --- a/unsupported/test/cxx11_tensor_contract_cuda.cu +++ b/unsupported/test/cxx11_tensor_contract_cuda.cu @@ -14,7 +14,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> |