aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/RefWorkloadFactory.cpp
diff options
context:
space:
mode:
authorColm Donelan <colm.donelan@arm.com>2024-04-19 15:02:09 +0100
committerColm Donelan <colm.donelan@arm.com>2024-04-19 15:02:09 +0100
commit693acb46c5829f32082501e6c3956206efc9f3c6 (patch)
tree8e6a11761ee4e90e7dbf6a7d97028cd746693443 /src/backends/reference/RefWorkloadFactory.cpp
parent02300aa6460441891e54342286358afe42f432c8 (diff)
downloadarmnn-693acb46c5829f32082501e6c3956206efc9f3c6.tar.gz
IVGCVSW-8314 Add Boolean data type to Debug layer support.
Signed-off-by: Colm Donelan <colm.donelan@arm.com> Change-Id: Ie2054393188c6099ecee2e09b7008860b9a7178c
Diffstat (limited to 'src/backends/reference/RefWorkloadFactory.cpp')
-rw-r--r--src/backends/reference/RefWorkloadFactory.cpp8
1 files changed, 8 insertions, 0 deletions
diff --git a/src/backends/reference/RefWorkloadFactory.cpp b/src/backends/reference/RefWorkloadFactory.cpp
index df458c1a6d..086f8eea8d 100644
--- a/src/backends/reference/RefWorkloadFactory.cpp
+++ b/src/backends/reference/RefWorkloadFactory.cpp
@@ -77,6 +77,10 @@ bool IsQAsymmU8(const WorkloadInfo& info)
{
return IsDataType<DataType::QAsymmU8>(info);
}
+bool IsBoolean(const WorkloadInfo& info)
+{
+ return IsDataType<DataType::Boolean>(info);
+}
RefWorkloadFactory::RefWorkloadFactory(const std::shared_ptr<RefMemoryManager>& memoryManager)
: m_MemoryManager(memoryManager)
@@ -271,6 +275,10 @@ std::unique_ptr<IWorkload> RefWorkloadFactory::CreateWorkload(LayerType type,
{
return std::make_unique<RefDebugSigned64Workload>(*debugQueueDescriptor, info);
}
+ if (IsBoolean(info))
+ {
+ return std::make_unique<RefDebugBooleanWorkload>(*debugQueueDescriptor, info);
+ }
return MakeWorkload<RefDebugFloat32Workload, RefDebugQAsymmU8Workload>(*debugQueueDescriptor, info);
}
case LayerType::DepthToSpace: