diff options
author | Teng Lu <teng.lu@intel.com> | 2020-06-20 19:16:24 +0000 |
---|---|---|
committer | Rasmus Munk Larsen <rmlarsen@google.com> | 2020-06-20 19:16:24 +0000 |
commit | 386d809bde475c65b7940f290efe80e6a05878c4 (patch) | |
tree | c38e161a53393d15be0ddb02a7a4e22dec738484 /Eigen/Core | |
parent | 6b9c92fe7eff0dedb031cec38004c9c3667f3057 (diff) |
Support BFloat16 in Eigen
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" |