From 1a763dd350e80f26a661dc5216fdef1256e5b32a Mon Sep 17 00:00:00 2001
From: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Date: Tue, 10 Sep 2019 12:32:08 +0100
Subject: IVGCVSW-3829 Add CL workload for RSQRT

Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: Idf40ed0ebf2af3170137bf3a3cb5a9635a203d4a
---
 src/armnn/test/CreateWorkload.hpp | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

(limited to 'src/armnn')

diff --git a/src/armnn/test/CreateWorkload.hpp b/src/armnn/test/CreateWorkload.hpp
index b50135aea5..d6d1ab3c5b 100644
--- a/src/armnn/test/CreateWorkload.hpp
+++ b/src/armnn/test/CreateWorkload.hpp
@@ -129,6 +129,32 @@ std::unique_ptr<WorkloadType> CreateElementwiseWorkloadTest(armnn::IWorkloadFact
     return workload;
 }
 
+template <typename WorkloadType,
+          typename DescriptorType,
+          typename LayerType,
+          armnn::DataType DataType>
+std::unique_ptr<WorkloadType> CreateElementwiseUnaryWorkloadTest(armnn::IWorkloadFactory & factory,
+                                                                 armnn::Graph & graph)
+{
+    Layer* const layer = graph.AddLayer<LayerType>("layer");
+
+    Layer* const input  = graph.AddLayer<InputLayer>(0, "input");
+    Layer* const output = graph.AddLayer<OutputLayer>(0, "output");
+
+    armnn::TensorInfo tensorInfo({ 2, 3 }, DataType);
+    Connect(input, layer, tensorInfo, 0, 0);
+    Connect(layer, output, tensorInfo, 0, 0);
+    CreateTensorHandles(graph, factory);
+
+    auto workload = MakeAndCheckWorkload<WorkloadType>(*layer, graph, factory);
+    DescriptorType queueDescriptor = workload->GetData();
+
+    BOOST_TEST(queueDescriptor.m_Inputs.size()  == 1);
+    BOOST_TEST(queueDescriptor.m_Outputs.size() == 1);
+
+    return workload;
+}
+
 template <typename BatchNormalizationWorkloadType, armnn::DataType DataType>
 std::unique_ptr<BatchNormalizationWorkloadType> CreateBatchNormalizationWorkloadTest(
     armnn::IWorkloadFactory& factory, armnn::Graph& graph, DataLayout dataLayout = DataLayout::NCHW)
-- 
cgit v1.2.1