From 44179c372eea9f17c96cbf50ee383e57e14d70a6 Mon Sep 17 00:00:00 2001 From: Narumol Prangnawarat Date: Wed, 11 Mar 2020 14:51:27 +0000 Subject: IVGCVSW-4511 Add BFloat16 to RefLayerSupport and unit tests Signed-off-by: Narumol Prangnawarat Change-Id: Ifaae4d5aac468ba927b2c6a4bf31b8c8522aeb2e --- src/backends/backendsCommon/MakeWorkloadHelper.hpp | 2 ++ 1 file changed, 2 insertions(+) (limited to 'src/backends/backendsCommon/MakeWorkloadHelper.hpp') diff --git a/src/backends/backendsCommon/MakeWorkloadHelper.hpp b/src/backends/backendsCommon/MakeWorkloadHelper.hpp index 7ef140e453..8abc8a6ef5 100644 --- a/src/backends/backendsCommon/MakeWorkloadHelper.hpp +++ b/src/backends/backendsCommon/MakeWorkloadHelper.hpp @@ -52,6 +52,7 @@ std::unique_ptr MakeWorkloadHelper(const QueueDescriptorType& descrip switch (dataType) { + case DataType::Float16: return MakeWorkloadForType::Func(descriptor, info, std::forward(args)...); case DataType::Float32: @@ -65,6 +66,7 @@ std::unique_ptr MakeWorkloadHelper(const QueueDescriptorType& descrip return MakeWorkloadForType::Func(descriptor, info, std::forward(args)...); case DataType::Boolean: return MakeWorkloadForType::Func(descriptor, info, std::forward(args)...); + case DataType::BFloat16: case DataType::QSymmS16: return nullptr; default: -- cgit v1.2.1