diff options
Diffstat (limited to 'Eigen/Core')
| -rw-r--r-- | Eigen/Core | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/Eigen/Core b/Eigen/Core index f36031557..f44b77831 100644 --- a/Eigen/Core +++ b/Eigen/Core @@ -51,6 +51,10 @@ #define EIGEN_HAS_GPU_FP16 #endif +#if defined(EIGEN_HAS_CUDA_BF16) || defined(EIGEN_HAS_HIP_BF16) + #define EIGEN_HAS_GPU_BF16 +#endif + #if (defined _OPENMP) && (!defined EIGEN_DONT_PARALLELIZE) #define EIGEN_HAS_OPENMP #endif @@ -163,6 +167,7 @@ using std::ptrdiff_t; #include "src/Core/arch/Default/ConjHelper.h" // Generic half float support #include "src/Core/arch/Default/Half.h" +#include "src/Core/arch/Default/BFloat16.h" #include "src/Core/arch/Default/TypeCasting.h" #include "src/Core/arch/Default/GenericPacketMathFunctionsFwd.h" |