commit 693acb46c5829f32082501e6c3956206efc9f3c6
author Colm Donelan <colm.donelan@arm.com> Fri Apr 19 15:02:09 2024 +0100
committer Colm Donelan <colm.donelan@arm.com> Fri Apr 19 15:02:09 2024 +0100
tree 8e6a11761ee4e90e7dbf6a7d97028cd746693443
parent 02300aa6460441891e54342286358afe42f432c8

IVGCVSW-8314 Add Boolean data type to Debug layer support.

Signed-off-by: Colm Donelan <colm.donelan@arm.com>
Change-Id: Ie2054393188c6099ecee2e09b7008860b9a7178c

src/backends/reference/RefWorkloadFactory.cpp

diff --git a/src/backends/reference/RefWorkloadFactory.cpp b/src/backends/reference/RefWorkloadFactory.cpp
index df458c1..086f8ee 100644
--- a/src/backends/reference/RefWorkloadFactory.cpp
+++ b/src/backends/reference/RefWorkloadFactory.cpp
@@ -77,6 +77,10 @@
 {
     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 @@
             {
                 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:

src/backends/reference/workloads/RefDebugWorkload.cpp

diff --git a/src/backends/reference/workloads/RefDebugWorkload.cpp b/src/backends/reference/workloads/RefDebugWorkload.cpp
index 94eed4f..23df873 100644
--- a/src/backends/reference/workloads/RefDebugWorkload.cpp
+++ b/src/backends/reference/workloads/RefDebugWorkload.cpp
@@ -1,5 +1,5 @@
 //
-// Copyright © 2018-2023 Arm Ltd and Contributors. All rights reserved.
+// Copyright © 2018-2024 Arm Ltd and Contributors. All rights reserved.
 // SPDX-License-Identifier: MIT
 //
 
@@ -66,5 +66,6 @@
 template class RefDebugWorkload<DataType::QSymmS8>;
 template class RefDebugWorkload<DataType::Signed32>;
 template class RefDebugWorkload<DataType::Signed64>;
+template class RefDebugWorkload<DataType::Boolean>;
 
 } // namespace armnn

src/backends/reference/workloads/RefDebugWorkload.hpp

diff --git a/src/backends/reference/workloads/RefDebugWorkload.hpp b/src/backends/reference/workloads/RefDebugWorkload.hpp
index 4c99990..1645782 100644
--- a/src/backends/reference/workloads/RefDebugWorkload.hpp
+++ b/src/backends/reference/workloads/RefDebugWorkload.hpp
@@ -1,5 +1,5 @@
 //
-// Copyright © 2018-2023 Arm Ltd and Contributors. All rights reserved.
+// Copyright © 2018-2024 Arm Ltd and Contributors. All rights reserved.
 // SPDX-License-Identifier: MIT
 //
 
@@ -48,5 +48,6 @@
 using RefDebugQSymmS8Workload   = RefDebugWorkload<DataType::QSymmS8>;
 using RefDebugSigned32Workload  = RefDebugWorkload<DataType::Signed32>;
 using RefDebugSigned64Workload  = RefDebugWorkload<DataType::Signed64>;
+using RefDebugBooleanWorkload   = RefDebugWorkload<DataType::Boolean>;
 
 } // namespace armnn