diff --git a/Eigen/Core b/Eigen/Core index c7192b037..ac58f10c3 100644 --- a/Eigen/Core +++ b/Eigen/Core @@ -214,10 +214,14 @@ #include #if defined __CUDACC_VER__ && __CUDACC_VER__ >= 70500 #define EIGEN_HAS_CUDA_FP16 - #include #endif #endif +#if defined EIGEN_HAS_CUDA_FP16 + #include + #include +#endif + #if (defined _OPENMP) && (!defined EIGEN_DONT_PARALLELIZE) #define EIGEN_HAS_OPENMP #endif