diff options
| author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-07-04 10:29:00 +0100 |
|---|---|---|
| committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-07-08 07:49:31 +0000 |
| commit | 47cfee9358be7a5c1c71e4ab7e94bec2557248c1 (patch) | |
| tree | 6608cbbba1377e0384c2517fee2a4b34509356c0 /src/backends/reference/RefWorkloadFactory.cpp | |
| parent | d2aa85ee2a0f574bcba50a376721f15c7e99308d (diff) | |
| download | armnn-47cfee9358be7a5c1c71e4ab7e94bec2557248c1.tar.gz | |
IVGCVSW-3399 Add support of QuantisedSymm16 to Debug and unit tests,
refactor to be done
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I6dbf6a2aa328b8c843632e1231931b4e1c6888bb
Diffstat (limited to 'src/backends/reference/RefWorkloadFactory.cpp')
| -rw-r--r-- | src/backends/reference/RefWorkloadFactory.cpp | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/src/backends/reference/RefWorkloadFactory.cpp b/src/backends/reference/RefWorkloadFactory.cpp index 7ae5b97dcf..d1189a6542 100644 --- a/src/backends/reference/RefWorkloadFactory.cpp +++ b/src/backends/reference/RefWorkloadFactory.cpp @@ -50,9 +50,9 @@ bool IsFloat16(const WorkloadInfo& info) return IsDataType<DataType::Float16>(info); } -bool IsUint8(const WorkloadInfo& info) +bool IsQSymm16(const WorkloadInfo& info) { - return IsDataType<DataType::QuantisedAsymm8>(info); + return IsDataType<DataType::QuantisedSymm16>(info); } RefWorkloadFactory::RefWorkloadFactory(const std::shared_ptr<RefMemoryManager>& memoryManager) @@ -432,7 +432,11 @@ std::unique_ptr<IWorkload> RefWorkloadFactory::CreateGreater(const GreaterQueueD std::unique_ptr<IWorkload> RefWorkloadFactory::CreateDebug(const DebugQueueDescriptor& descriptor, const WorkloadInfo& info) const { - return MakeWorkload<RefDebugFloat32Workload, RefDebugUint8Workload>(descriptor, info); + if (IsQSymm16(info)) + { + return std::make_unique<RefDebugQSymm16Workload>(descriptor, info); + } + return MakeWorkload<RefDebugFloat32Workload, RefDebugQAsymm8Workload>(descriptor, info); } std::unique_ptr<IWorkload> RefWorkloadFactory::CreateRsqrt(const RsqrtQueueDescriptor& descriptor, |