diff options
author | Ryan OShea <ryan.oshea3@arm.com> | 2022-11-07 16:20:48 +0000 |
---|---|---|
committer | ryan.oshea3 <ryan.oshea3@arm.com> | 2022-11-16 15:22:50 +0000 |
commit | 31441595009182c985dacbedc70c41ee6664d070 (patch) | |
tree | 248a85295aeff4022c9b395fc97748b0a0aa6b35 /src/armnn/optimizations/ConvertConstants.hpp | |
parent | bd18eab07a8f30492de1e462b1815189014cb8d5 (diff) | |
download | armnn-31441595009182c985dacbedc70c41ee6664d070.tar.gz |
IVGCVSW-7214 Disable BF16-Turbo-Mode and remove conversion layers
- Remove Bf16ToFp32 Conversion Layer
- Remove Fp32ToBf16 Conversion Layer
- Remove B16 Conversion tests
* Throw exception if m_ReduceFp32ToBf16 optimzer option is set to true
* Provide comments to enable fast math in order to use bf16
* Update docs to inform users to enable fast math for bf16
Execute Network Changes
* Require bf16_turbo_mode to also have fast_math_enabled set to true
- Remove setting m_ReduceFp32ToBf16 optimizer option
Signed-off-by: Ryan OShea <ryan.oshea3@arm.com>
Change-Id: Ibaa6da9d29c96a1ce32ff5196b0847fde9f04a1c
Diffstat (limited to 'src/armnn/optimizations/ConvertConstants.hpp')
-rw-r--r-- | src/armnn/optimizations/ConvertConstants.hpp | 54 |
1 files changed, 0 insertions, 54 deletions
diff --git a/src/armnn/optimizations/ConvertConstants.hpp b/src/armnn/optimizations/ConvertConstants.hpp index 54c14e5c89..7b2f1fd291 100644 --- a/src/armnn/optimizations/ConvertConstants.hpp +++ b/src/armnn/optimizations/ConvertConstants.hpp @@ -11,7 +11,6 @@ #include <armnn/backends/TensorHandle.hpp> #include <armnn/utility/IgnoreUnused.hpp> -#include <BFloat16.hpp> #include <Half.hpp> namespace armnn @@ -19,27 +18,6 @@ namespace armnn namespace optimizations { -struct BFloat16ToFloat32 -{ - static void Func(std::shared_ptr<ConstTensorHandle>& handle) - { - const TensorInfo& info = handle->GetTensorInfo(); - - if (info.GetDataType() == DataType::BFloat16) - { - std::vector<float> newValues(info.GetNumElements()); - - armnnUtils::FloatingPointConverter::ConvertBFloat16ToFloat32(handle->GetConstTensor<BFloat16>(), - info.GetNumElements(), - newValues.data()); - - TensorInfo newInfo(info.GetShape(), DataType::Float32, 0.0f, 0, true); - ConstTensor newInput(newInfo, newValues); - handle.reset(new ScopedTensorHandle(newInput)); - } - } -}; - struct Float16ToFloat32 { static void Func(std::shared_ptr<ConstTensorHandle>& handle) @@ -61,27 +39,6 @@ struct Float16ToFloat32 } }; -struct Float32ToBFloat16 -{ - static void Func(std::shared_ptr<ConstTensorHandle>& handle) - { - const TensorInfo& info = handle->GetTensorInfo(); - - if (info.GetDataType() == DataType::Float32) - { - std::vector<BFloat16> newValues(info.GetNumElements()); - - armnnUtils::FloatingPointConverter::ConvertFloat32ToBFloat16(handle->GetConstTensor<float>(), - info.GetNumElements(), - newValues.data()); - - TensorInfo newInfo(info.GetShape(), DataType::BFloat16, 0.0f, 0, true); - ConstTensor newInput(newInfo, newValues); - handle.reset(new ScopedTensorHandle(newInput)); - } - } -}; - struct Float32ToFloat16 { static void Func(std::shared_ptr<ConstTensorHandle>& handle) @@ -138,17 +95,6 @@ struct IsFloat16Layer } }; -struct IsBFloat16Layer -{ - static bool Test(const Layer& layer) - { - return layer.GetDataType() == DataType::BFloat16; - } -}; - -using ConvertConstantsBFloatToFloat = ConvertConstants<BFloat16ToFloat32, IsFloat32Layer>; -using ConvertConstantsFloatToBFloat = ConvertConstants<Float32ToBFloat16, IsBFloat16Layer>; - using ConvertConstantsHalfToFloat = ConvertConstants<Float16ToFloat32, IsFloat32Layer>; using ConvertConstantsFloatToHalf = ConvertConstants<Float32ToFloat16, IsFloat16Layer>; |