diff options
author | Gael Guennebaud <g.gael@free.fr> | 2016-09-01 13:39:33 +0200 |
---|---|---|
committer | Gael Guennebaud <g.gael@free.fr> | 2016-09-01 13:39:33 +0200 |
commit | 72a4d49315ee25c64e8e457735c494e77863a3e0 (patch) | |
tree | 6e621ab41c2bb1af14650166b99a42d541e64a9b | |
parent | f9f32e9e2d2b6a9c89b83232d5290c956b175561 (diff) |
Fix compilation with CUDA 8
-rw-r--r-- | unsupported/test/cxx11_tensor_argmax_cuda.cu | 1 | ||||
-rw-r--r-- | unsupported/test/cxx11_tensor_cast_float16_cuda.cu | 2 | ||||
-rw-r--r-- | unsupported/test/cxx11_tensor_contract_cuda.cu | 2 | ||||
-rw-r--r-- | unsupported/test/cxx11_tensor_cuda.cu | 2 | ||||
-rw-r--r-- | unsupported/test/cxx11_tensor_device.cu | 2 | ||||
-rw-r--r-- | unsupported/test/cxx11_tensor_random_cuda.cu | 1 | ||||
-rw-r--r-- | unsupported/test/cxx11_tensor_reduction_cuda.cu | 1 | ||||
-rw-r--r-- | unsupported/test/cxx11_tensor_scan_cuda.cu | 2 |
8 files changed, 8 insertions, 5 deletions
diff --git a/unsupported/test/cxx11_tensor_argmax_cuda.cu b/unsupported/test/cxx11_tensor_argmax_cuda.cu index 41ccbe974..c3f4bfa53 100644 --- a/unsupported/test/cxx11_tensor_argmax_cuda.cu +++ b/unsupported/test/cxx11_tensor_argmax_cuda.cu @@ -12,6 +12,7 @@ #define EIGEN_TEST_FUNC cxx11_tensor_cuda #define EIGEN_USE_GPU +#include <cuda_fp16.h> #include "main.h" #include <unsupported/Eigen/CXX11/Tensor> diff --git a/unsupported/test/cxx11_tensor_cast_float16_cuda.cu b/unsupported/test/cxx11_tensor_cast_float16_cuda.cu index f22b99de8..559aae269 100644 --- a/unsupported/test/cxx11_tensor_cast_float16_cuda.cu +++ b/unsupported/test/cxx11_tensor_cast_float16_cuda.cu @@ -13,7 +13,7 @@ #define EIGEN_DEFAULT_DENSE_INDEX_TYPE int #define EIGEN_USE_GPU - +#include <cuda_fp16.h> #include "main.h" #include <unsupported/Eigen/CXX11/Tensor> diff --git a/unsupported/test/cxx11_tensor_contract_cuda.cu b/unsupported/test/cxx11_tensor_contract_cuda.cu index 98ac180ef..5282dc3b9 100644 --- a/unsupported/test/cxx11_tensor_contract_cuda.cu +++ b/unsupported/test/cxx11_tensor_contract_cuda.cu @@ -14,7 +14,7 @@ #define EIGEN_DEFAULT_DENSE_INDEX_TYPE int #define EIGEN_USE_GPU - +#include <cuda_fp16.h> #include "main.h" #include <unsupported/Eigen/CXX11/Tensor> diff --git a/unsupported/test/cxx11_tensor_cuda.cu b/unsupported/test/cxx11_tensor_cuda.cu index 284b46803..29ea76857 100644 --- a/unsupported/test/cxx11_tensor_cuda.cu +++ b/unsupported/test/cxx11_tensor_cuda.cu @@ -13,7 +13,7 @@ #define EIGEN_DEFAULT_DENSE_INDEX_TYPE int #define EIGEN_USE_GPU - +#include <cuda_fp16.h> #include "main.h" #include <unsupported/Eigen/CXX11/Tensor> diff --git a/unsupported/test/cxx11_tensor_device.cu b/unsupported/test/cxx11_tensor_device.cu index b6ca54d93..bbda95906 100644 --- a/unsupported/test/cxx11_tensor_device.cu +++ b/unsupported/test/cxx11_tensor_device.cu @@ -13,7 +13,7 @@ #define EIGEN_DEFAULT_DENSE_INDEX_TYPE int #define EIGEN_USE_GPU - +#include <cuda_fp16.h> #include "main.h" #include <unsupported/Eigen/CXX11/Tensor> diff --git a/unsupported/test/cxx11_tensor_random_cuda.cu b/unsupported/test/cxx11_tensor_random_cuda.cu index fa1a46732..32f71e808 100644 --- a/unsupported/test/cxx11_tensor_random_cuda.cu +++ b/unsupported/test/cxx11_tensor_random_cuda.cu @@ -13,6 +13,7 @@ #define EIGEN_DEFAULT_DENSE_INDEX_TYPE int #define EIGEN_USE_GPU +#include <cuda_fp16.h> #include "main.h" #include <Eigen/CXX11/Tensor> diff --git a/unsupported/test/cxx11_tensor_reduction_cuda.cu b/unsupported/test/cxx11_tensor_reduction_cuda.cu index 6d8f01c02..9b5e60e99 100644 --- a/unsupported/test/cxx11_tensor_reduction_cuda.cu +++ b/unsupported/test/cxx11_tensor_reduction_cuda.cu @@ -12,6 +12,7 @@ #define EIGEN_TEST_FUNC cxx11_tensor_reduction_cuda #define EIGEN_USE_GPU +#include <cuda_fp16.h> #include "main.h" #include <unsupported/Eigen/CXX11/Tensor> diff --git a/unsupported/test/cxx11_tensor_scan_cuda.cu b/unsupported/test/cxx11_tensor_scan_cuda.cu index 35e19e51c..c8a99f62b 100644 --- a/unsupported/test/cxx11_tensor_scan_cuda.cu +++ b/unsupported/test/cxx11_tensor_scan_cuda.cu @@ -13,7 +13,7 @@ #define EIGEN_DEFAULT_DENSE_INDEX_TYPE int #define EIGEN_USE_GPU - +#include <cuda_fp16.h> #include "main.h" #include <unsupported/Eigen/CXX11/Tensor> |