diff options
Diffstat (limited to 'unsupported/test/cxx11_tensor_device.cu')
-rw-r--r-- | unsupported/test/cxx11_tensor_device.cu | 2 |
1 files changed, 1 insertions, 1 deletions
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> |