diff options
| author |  Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-04-19 17:44:12 -0700 |
|---|---|---|
| committer | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-04-19 17:44:12 -0700 |
| commit | 1d0238375d53e168eda40139979f8d128e72f4e8 (patch) | |
| tree | aa144da5a23d9a7fd8c4df98f403332f491fc41c /Eigen/Core | |
| parent | 2b7216302864715eb6950448fc71b0e4b51d617c (diff) | |
Made sure all the required header files are included when trying to use fp16
Diffstat (limited to 'Eigen/Core')
| -rw-r--r-- | Eigen/Core | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/Eigen/Core b/Eigen/Core index c7192b037..ac58f10c3 100644 --- a/Eigen/Core +++ b/Eigen/Core @@ -214,10 +214,14 @@ #include <vector_types.h> #if defined __CUDACC_VER__ && __CUDACC_VER__ >= 70500 #define EIGEN_HAS_CUDA_FP16 - #include <cuda_fp16.h> #endif #endif +#if defined EIGEN_HAS_CUDA_FP16 + #include <host_defines.h> + #include <cuda_fp16.h> +#endif + #if (defined _OPENMP) && (!defined EIGEN_DONT_PARALLELIZE) #define EIGEN_HAS_OPENMP #endif |