diff options
| author | James Ward <james.ward@arm.com> | 2022-10-19 12:20:31 +0100 |
|---|---|---|
| committer | Jeremy Johnson <jeremy.johnson@arm.com> | 2022-11-09 12:19:51 +0000 |
| commit | 24dbc420aae556649f50e645bd94489dab2cc75a (patch) | |
| tree | 490345da43e9c5bae0f450ba05ffe85874077e0a /reference_model/src/ops/template_types.h | |
| parent | 3b0544c1e7463295c49a48a162ebb9a546326829 (diff) | |
| download | reference_model-24dbc420aae556649f50e645bd94489dab2cc75a.tar.gz | |
Add BF16 support to reference model
* Upgrade Eigen to 3.4.0 (for bfloat16 support) and add work-
arounds for reduce.any() and reduce.all() bugs (introduced
between 3.3.7 and 3.4.0)
* Truncation to bfloat16 now performed in eval() methods
Signed-off-by: James Ward <james.ward@arm.com>
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: If5f5c988d76d3d30790acf3b97081726b89205fe
Diffstat (limited to 'reference_model/src/ops/template_types.h')
| -rw-r--r-- | reference_model/src/ops/template_types.h | 14 |
1 files changed, 8 insertions, 6 deletions
diff --git a/reference_model/src/ops/template_types.h b/reference_model/src/ops/template_types.h index 3de4899..647ca84 100644 --- a/reference_model/src/ops/template_types.h +++ b/reference_model/src/ops/template_types.h @@ -19,6 +19,8 @@ #include "tosa_generated.h" #include <Eigen/CXX11/Tensor> #include "half.hpp" +#include <Eigen/Core> +#include "arith_util.h" using namespace tosa; @@ -76,6 +78,12 @@ struct GetEigenType<DType_FP16> using type = float; }; template <> +struct GetEigenType<DType_BF16> +{ + // NOTE: full precision used + using type = float; +}; +template <> struct GetEigenType<DType_INT32> { using type = int32_t; @@ -132,12 +140,6 @@ struct GetAccEigenType using type = typename GetEigenType<Dtype>::type; }; -template <DType Dtype> -struct GetHalfEigenType -{ - using type = half_float::half; -}; - // Meta function to get number of bits template <DType T> struct GetNumBits |