From bc5a5d59e07d2365c2c88b3430b4a311d186ebbc Mon Sep 17 00:00:00 2001
From: Tracy Narine <tracy.narine@arm.com>
Date: Tue, 6 Feb 2024 15:22:41 +0000
Subject: IVGCVSW-7622 GpuFsa Op: Add Cast operator

  * Added cast operator support

Signed-off-by: Tracy Narine <tracy.narine@arm.com>
Change-Id: Ie12cb1559a7a059ff35e1c395bc77243499243cd
---
 .../test/layerTests/CastTestImpl.hpp               | 40 +++++++++++++++++++++-
 1 file changed, 39 insertions(+), 1 deletion(-)

(limited to 'src/backends/backendsCommon/test/layerTests')

diff --git a/src/backends/backendsCommon/test/layerTests/CastTestImpl.hpp b/src/backends/backendsCommon/test/layerTests/CastTestImpl.hpp
index a727950bb8..13512ad700 100644
--- a/src/backends/backendsCommon/test/layerTests/CastTestImpl.hpp
+++ b/src/backends/backendsCommon/test/layerTests/CastTestImpl.hpp
@@ -1,10 +1,12 @@
 //
-// Copyright © 2021 Arm Ltd and Contributors. All rights reserved.
+// Copyright © 2021, 2024 Arm Ltd and Contributors. All rights reserved.
 // SPDX-License-Identifier: MIT
 //
 
 #pragma once
 
+#include <backendsCommon/test/EndToEndTestImpl.hpp>
+
 #include <armnnTestUtils/LayerTestResult.hpp>
 
 #include <ResolveType.hpp>
@@ -82,3 +84,39 @@ LayerTestResult<uint8_t , 4> CastFloat32ToUInt82dTest(
         armnn::IWorkloadFactory& workloadFactory,
         const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
         const armnn::ITensorHandleFactory& tensorHandleFactory);
+
+template<armnn::DataType inputDataType, armnn::DataType outputDataType, typename TInput, typename TOutput>
+void CastSimpleTest(const std::vector<armnn::BackendId>& backends,
+                    const std::vector<unsigned int>& shape,
+                    const std::vector<TInput>& inputValues,
+                    const std::vector<TOutput>& outputValues,
+                    float qScale = 1.0f,
+                    int32_t qOffset = 0)
+{
+    using namespace armnn;
+
+    const TensorShape inputShape(static_cast<unsigned int>(shape.size()), shape.data());
+    const TensorShape outputShape(static_cast<unsigned int>(shape.size()), shape.data());
+
+    TensorInfo inputTensorInfo(inputShape, inputDataType, qScale, qOffset, true);
+    TensorInfo outputTensorInfo(outputShape, outputDataType, qScale, qOffset);
+
+    IRuntime::CreationOptions options;
+    IRuntimePtr runtime(IRuntime::Create(options));
+    INetworkPtr network(INetwork::Create());
+
+    IConnectableLayer* input     = network->AddInputLayer(0, "input");
+    IConnectableLayer* castLayer = network->AddCastLayer("cast");
+    IConnectableLayer* output    = network->AddOutputLayer(0, "output");
+
+    Connect(input, castLayer, inputTensorInfo, 0, 0);
+    Connect(castLayer, output, outputTensorInfo, 0, 0);
+
+    std::map<int, std::vector<TInput>> inputTensorData     = {{ 0, inputValues }};
+    std::map<int, std::vector<TOutput>> expectedOutputData = {{ 0, outputValues }};
+
+    EndToEndLayerTestImpl<inputDataType, outputDataType>(std::move(network),
+                                                         inputTensorData,
+                                                         expectedOutputData,
+                                                         backends);
+}
\ No newline at end of file
-- 
cgit v1.2.1