diff options
author | Sheng Yang <yang.sheng@intel.com> | 2020-07-14 01:34:03 +0000 |
---|---|---|
committer | Rasmus Munk Larsen <rmlarsen@google.com> | 2020-07-14 01:34:03 +0000 |
commit | 56b3e3f3f8ca9972ca390c8296fde363bdab271c (patch) | |
tree | 5d06bf0995ed07dd232e346369e71f70561b5d9c /Eigen/src/Core/arch/Default/BFloat16.h | |
parent | 4ab32e2de2511746e2108563a43cbbeb1922fbf2 (diff) |
AVX path for BF16
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; |