diff options
author | Keith Davis <keith.davis@arm.com> | 2020-02-19 10:08:33 +0000 |
---|---|---|
committer | Keith Davis <keith.davis@arm.com> | 2020-02-19 10:08:33 +0000 |
commit | 67e6c54b4c4ff1e46a9ed14014d279794a666969 (patch) | |
tree | 80f272650985f9cd7ae684791dc54d6f68bfd1ca /src/backends/reference | |
parent | 48d709356269f4f4131d319f6638ab4f7b083931 (diff) | |
download | armnn-67e6c54b4c4ff1e46a9ed14014d279794a666969.tar.gz |
IVGCVSW-4472 Yolo v3 ExecuteNetwork failing
* Add Debug workload for QAsymmS8/U8
* Change Dequantize tests to test AsymmS8 instead of SymmS8
* Fix incorrect supportedness within RefLayerSupport
Signed-off-by: Keith Davis <keith.davis@arm.com>
Change-Id: Ie51f1e33c564d46c86bf0150b1addda3fc093d13
Diffstat (limited to 'src/backends/reference')
-rw-r--r-- | src/backends/reference/RefLayerSupport.cpp | 19 | ||||
-rw-r--r-- | src/backends/reference/RefWorkloadFactory.cpp | 18 | ||||
-rw-r--r-- | src/backends/reference/workloads/RefDebugWorkload.cpp | 1 | ||||
-rw-r--r-- | src/backends/reference/workloads/RefDebugWorkload.hpp | 1 |
4 files changed, 28 insertions, 11 deletions
diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp index bba83e23d4..5cb36c4299 100644 --- a/src/backends/reference/RefLayerSupport.cpp +++ b/src/backends/reference/RefLayerSupport.cpp @@ -82,7 +82,6 @@ bool RefLayerSupport::IsActivationSupported(const TensorInfo& input, std::array<DataType,6> supportedTypes = { DataType::Float32, DataType::Float16, - DataType::QSymmS8, DataType::QAsymmS8, DataType::QAsymmU8, DataType::QSymmS16 @@ -147,7 +146,6 @@ bool RefLayerSupport::IsAdditionSupported(const TensorInfo& input0, std::array<DataType,6> supportedTypes = { DataType::Float32, DataType::Float16, - DataType::QSymmS8, DataType::QAsymmS8, DataType::QAsymmU8, DataType::QSymmS16 @@ -332,8 +330,8 @@ bool RefLayerSupport::IsConcatSupported(const std::vector<const TensorInfo*> inp { DataType::Float32, DataType::Float16, - DataType::QSymmS8, DataType::QAsymmU8, + DataType::QAsymmS8, DataType::QSymmS16 }; @@ -355,11 +353,12 @@ bool RefLayerSupport::IsConcatSupported(const std::vector<const TensorInfo*> inp bool RefLayerSupport::IsConstantSupported(const TensorInfo& output, Optional<std::string&> reasonIfUnsupported) const { - std::array<DataType,5> supportedTypes = + std::array<DataType,6> supportedTypes = { DataType::Float32, DataType::Signed32, DataType::QAsymmU8, + DataType::QAsymmS8, DataType::QSymmS8, DataType::QSymmS16 }; @@ -1120,7 +1119,7 @@ bool RefLayerSupport::IsMaximumSupported(const TensorInfo& input0, std::array<DataType,5> supportedTypes = { DataType::Float32, DataType::Float16, - DataType::QSymmS8, + DataType::QAsymmS8, DataType::QAsymmU8, DataType::QSymmS16 }; @@ -1284,11 +1283,11 @@ bool RefLayerSupport::IsMultiplicationSupported(const TensorInfo& input0, { bool supported = true; - std::array<DataType,5> supportedTypes = { + std::array<DataType,6> supportedTypes = { DataType::Float32, DataType::Float16, - DataType::QSymmS8, DataType::QAsymmU8, + DataType::QAsymmS8, DataType::QSymmS16 }; @@ -1416,11 +1415,10 @@ bool RefLayerSupport::IsPooling2dSupported(const TensorInfo& input, bool supported = true; // Define supported output and inputs types. - std::array<DataType,6> supportedTypes = + std::array<DataType,5> supportedTypes = { DataType::Float32, DataType::Float16, - DataType::QSymmS8, DataType::QAsymmS8, DataType::QAsymmU8, DataType::QSymmS16 @@ -1488,7 +1486,6 @@ bool RefLayerSupport::IsReshapeSupported(const TensorInfo& input, DataType::Signed32, DataType::QAsymmS8, DataType::QAsymmU8, - DataType::QSymmS8, DataType::QSymmS16 }; @@ -1532,8 +1529,8 @@ bool RefLayerSupport::IsResizeSupported(const TensorInfo& input, { DataType::Float32, DataType::Float16, - DataType::QSymmS8, DataType::QAsymmU8, + DataType::QAsymmS8, DataType::QSymmS16 }; diff --git a/src/backends/reference/RefWorkloadFactory.cpp b/src/backends/reference/RefWorkloadFactory.cpp index dadb456104..02dbbabf9f 100644 --- a/src/backends/reference/RefWorkloadFactory.cpp +++ b/src/backends/reference/RefWorkloadFactory.cpp @@ -65,6 +65,16 @@ bool IsQSymmS8(const WorkloadInfo& info) return IsDataType<DataType::QSymmS8>(info); } +bool IsQAsymmS8(const WorkloadInfo& info) +{ + return IsDataType<DataType::QAsymmS8>(info); +} + +bool IsQAsymmU8(const WorkloadInfo& info) +{ + return IsDataType<DataType::QAsymmU8>(info); +} + RefWorkloadFactory::RefWorkloadFactory(const std::shared_ptr<RefMemoryManager>& memoryManager) : m_MemoryManager(memoryManager) { @@ -200,6 +210,14 @@ std::unique_ptr<IWorkload> RefWorkloadFactory::CreateDebug(const DebugQueueDescr { return std::make_unique<RefDebugQSymmS8Workload>(descriptor, info); } + if (IsQAsymmU8(info)) + { + return std::make_unique<RefDebugQAsymmU8Workload>(descriptor, info); + } + if (IsQAsymmS8(info)) + { + return std::make_unique<RefDebugQAsymmS8Workload>(descriptor, info); + } if (IsSigned32(info)) { return std::make_unique<RefDebugSigned32Workload>(descriptor, info); diff --git a/src/backends/reference/workloads/RefDebugWorkload.cpp b/src/backends/reference/workloads/RefDebugWorkload.cpp index 2f0b4276c0..af714a3ca7 100644 --- a/src/backends/reference/workloads/RefDebugWorkload.cpp +++ b/src/backends/reference/workloads/RefDebugWorkload.cpp @@ -47,6 +47,7 @@ void RefDebugWorkload<DataType>::RegisterDebugCallback(const DebugCallbackFuncti template class RefDebugWorkload<DataType::Float16>; template class RefDebugWorkload<DataType::Float32>; template class RefDebugWorkload<DataType::QAsymmU8>; +template class RefDebugWorkload<DataType::QAsymmS8>; template class RefDebugWorkload<DataType::QSymmS16>; template class RefDebugWorkload<DataType::QSymmS8>; template class RefDebugWorkload<DataType::Signed32>; diff --git a/src/backends/reference/workloads/RefDebugWorkload.hpp b/src/backends/reference/workloads/RefDebugWorkload.hpp index 4966ca3432..5a2a1cdf1b 100644 --- a/src/backends/reference/workloads/RefDebugWorkload.hpp +++ b/src/backends/reference/workloads/RefDebugWorkload.hpp @@ -40,6 +40,7 @@ private: using RefDebugFloat16Workload = RefDebugWorkload<DataType::Float16>; using RefDebugFloat32Workload = RefDebugWorkload<DataType::Float32>; using RefDebugQAsymmU8Workload = RefDebugWorkload<DataType::QAsymmU8>; +using RefDebugQAsymmS8Workload = RefDebugWorkload<DataType::QAsymmS8>; using RefDebugQSymmS16Workload = RefDebugWorkload<DataType::QSymmS16>; using RefDebugQSymmS8Workload = RefDebugWorkload<DataType::QSymmS8>; using RefDebugSigned32Workload = RefDebugWorkload<DataType::Signed32>; |