From 38b72e8de898d84a1481e242803da61009719891 Mon Sep 17 00:00:00 2001
From: Teresa Charlin <teresa.charlinreyes@arm.com>
Date: Wed, 4 May 2022 17:54:19 +0100
Subject: IVGCVSW-6937 Add INT32 support to FLOOR

Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I6f7cddb2d23c67ae682132d18f98776c074dcb3b
---
 src/backends/backendsCommon/WorkloadData.cpp                  |  3 ++-
 src/backends/backendsCommon/test/layerTests/FloorTestImpl.cpp | 11 ++++++++---
 2 files changed, 10 insertions(+), 4 deletions(-)

(limited to 'src/backends/backendsCommon')

diff --git a/src/backends/backendsCommon/WorkloadData.cpp b/src/backends/backendsCommon/WorkloadData.cpp
index 962ecde24b..70d7641f41 100644
--- a/src/backends/backendsCommon/WorkloadData.cpp
+++ b/src/backends/backendsCommon/WorkloadData.cpp
@@ -1953,7 +1953,8 @@ void FloorQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const
         DataType::BFloat16,
         DataType::Float32,
         DataType::Float16,
-        DataType::QSymmS16
+        DataType::QSymmS16,
+        DataType::Signed32
     };
 
     ValidateDataTypes(inputTensorInfo,  supportedTypes, descriptorName);
diff --git a/src/backends/backendsCommon/test/layerTests/FloorTestImpl.cpp b/src/backends/backendsCommon/test/layerTests/FloorTestImpl.cpp
index 2bf8c97af5..377c2aaf1a 100644
--- a/src/backends/backendsCommon/test/layerTests/FloorTestImpl.cpp
+++ b/src/backends/backendsCommon/test/layerTests/FloorTestImpl.cpp
@@ -19,10 +19,10 @@ LayerTestResult<T, 4> SimpleFloorTest(
 {
     IgnoreUnused(memoryManager);
     armnn::TensorInfo inputTensorInfo({1, 3, 2, 3}, ArmnnType);
-    inputTensorInfo.SetQuantizationScale(0.1f);
+    inputTensorInfo.SetQuantizationScale(1.0f);
 
     armnn::TensorInfo outputTensorInfo(inputTensorInfo);
-    outputTensorInfo.SetQuantizationScale(0.1f);
+    outputTensorInfo.SetQuantizationScale(1.0f);
 
     std::vector<T> input = ConvertToDataType<ArmnnType>(
         {
@@ -80,9 +80,14 @@ SimpleFloorTest<armnn::DataType::Float16>(
     const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
     const armnn::ITensorHandleFactory& tensorHandleFactory);
 
-
 template LayerTestResult<armnn::ResolveType<armnn::DataType::QSymmS16>, 4>
 SimpleFloorTest<armnn::DataType::QSymmS16>(
     armnn::IWorkloadFactory& workloadFactory,
     const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
     const armnn::ITensorHandleFactory& tensorHandleFactory);
+
+template LayerTestResult<armnn::ResolveType<armnn::DataType::Signed32>, 4>
+SimpleFloorTest<armnn::DataType::Signed32>(
+        armnn::IWorkloadFactory& workloadFactory,
+        const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
+        const armnn::ITensorHandleFactory& tensorHandleFactory);
\ No newline at end of file
-- 
cgit v1.2.1