diff options
Diffstat (limited to 'Eigen/src/Core/arch/Default/BFloat16.h')
-rw-r--r-- | Eigen/src/Core/arch/Default/BFloat16.h | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/Eigen/src/Core/arch/Default/BFloat16.h b/Eigen/src/Core/arch/Default/BFloat16.h index abf2ac933..34a4f0ced 100644 --- a/Eigen/src/Core/arch/Default/BFloat16.h +++ b/Eigen/src/Core/arch/Default/BFloat16.h @@ -23,6 +23,13 @@ limitations under the License. #define EIGEN_EXPLICIT_CAST(tgt_type) operator tgt_type() #endif +#define BF16_PACKET_FUNCTION(PACKET_F, PACKET_BF16, METHOD) \ + template <> \ + EIGEN_DEFINE_FUNCTION_ALLOWING_MULTIPLE_DEFINITIONS EIGEN_UNUSED \ + PACKET_BF16 METHOD<PACKET_BF16>(const PACKET_BF16& _x) { \ + return F32ToBf16(METHOD<PACKET_F>(Bf16ToF32(_x))); \ + } + namespace Eigen { struct bfloat16; |