commit d5166106ebfc11b1db832a708892e12edeaf997e
author Matthew Jackson <matthew.jackson@arm.com> Wed Jul 31 14:06:28 2019 +0100
committer Matteo Martincigh <matteo.martincigh@arm.com> Thu Aug 01 09:42:03 2019 +0000
tree 8f4be789711ed50b0bddf50a813caf4b823a59e9
parent 2208b604dea6919525a12779df9aa5e49ea6a539

IVGCVSW-3539 Add CL support and tests for Stack

* Added CL backend support for Stack
* Added unit tests for Stack on the CL backend
* Refactored unit tests to support generic data types

Signed-off-by: Matthew Jackson <matthew.jackson@arm.com>
Change-Id: I38ee3e9d8947ea98a3104c982698001e704d7d89

src/backends/cl/ClWorkloadFactory.cpp

diff --git a/src/backends/cl/ClWorkloadFactory.cpp b/src/backends/cl/ClWorkloadFactory.cpp
index 6ce87d8..4a593aa 100644
--- a/src/backends/cl/ClWorkloadFactory.cpp
+++ b/src/backends/cl/ClWorkloadFactory.cpp
@@ -433,4 +433,10 @@
     return MakeWorkload<ClSpaceToDepthWorkload>(descriptor, info);
 }
 
+std::unique_ptr<IWorkload> ClWorkloadFactory::CreateStack(const StackQueueDescriptor& descriptor,
+                                                          const WorkloadInfo& info) const
+{
+    return MakeWorkload<ClStackWorkload>(descriptor, info);
+}
+
 } // namespace armnn