IVGCVSW-4420 Failing Unit Test Cl: Quantize layer
* Fix bug with missing ValidateInputsOutputs in Cl Workload
* Renamed data type tests to be more specific
* Future proofing
Signed-off-by: Keith Davis <keith.davis@arm.com>
Change-Id: I157acc318e40d727fcfb3f36f7f577fc87472f3d
diff --git a/src/backends/cl/workloads/ClDequantizeWorkload.cpp b/src/backends/cl/workloads/ClDequantizeWorkload.cpp
index 67a555a..eca795d 100644
--- a/src/backends/cl/workloads/ClDequantizeWorkload.cpp
+++ b/src/backends/cl/workloads/ClDequantizeWorkload.cpp
@@ -32,6 +32,8 @@
const WorkloadInfo& workloadInfo)
: BaseWorkload<DequantizeQueueDescriptor>(descriptor, workloadInfo)
{
+ m_Data.ValidateInputsOutputs("ClDequantizeWorkload", 1, 1);
+
arm_compute::ICLTensor& input = boost::polymorphic_pointer_downcast<IClTensorHandle>(
m_Data.m_Inputs[0])->GetTensor();
diff --git a/src/backends/cl/workloads/ClQuantizeWorkload.cpp b/src/backends/cl/workloads/ClQuantizeWorkload.cpp
index 230e346..263065a 100644
--- a/src/backends/cl/workloads/ClQuantizeWorkload.cpp
+++ b/src/backends/cl/workloads/ClQuantizeWorkload.cpp
@@ -32,6 +32,8 @@
ClQuantizeWorkload::ClQuantizeWorkload(const QuantizeQueueDescriptor& descriptor, const WorkloadInfo& info)
: BaseWorkload<QuantizeQueueDescriptor>(descriptor, info)
{
+ m_Data.ValidateInputsOutputs("ClQuantizeWorkload", 1, 1);
+
arm_compute::ICLTensor& input = static_cast<IClTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
arm_compute::ICLTensor& output = static_cast<IClTensorHandle*>(m_Data.m_Outputs[0])->GetTensor();
diff --git a/src/backends/cl/workloads/ClWorkloadUtils.hpp b/src/backends/cl/workloads/ClWorkloadUtils.hpp
index d3c6df5..b4bcc1c 100644
--- a/src/backends/cl/workloads/ClWorkloadUtils.hpp
+++ b/src/backends/cl/workloads/ClWorkloadUtils.hpp
@@ -101,6 +101,7 @@
case DataType::Float32:
CopyArmComputeClTensorData(clTensor, handle->GetConstTensor<float>());
break;
+ case DataType::QAsymmS8:
case DataType::QAsymmU8:
CopyArmComputeClTensorData(clTensor, handle->GetConstTensor<uint8_t>());
break;