diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2019-07-31 16:36:27 +0100 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2019-07-31 16:36:27 +0100 |
commit | 2208b604dea6919525a12779df9aa5e49ea6a539 (patch) | |
tree | fbe5bfa905412afdf97e9bcedb1f225bdf3caaa1 /src/backends | |
parent | a77d80e757630e5cbc62f39690df99907ecaf1a6 (diff) | |
download | armnn-2208b604dea6919525a12779df9aa5e49ea6a539.tar.gz |
IVGCVSW-3571 Add Gpu backend to CI tests for Android Q
* Fixed Quantized FP16 failures on GpuAcc
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: Ia6372da1305c3e826f56ba537e11acf7875c2d74
Diffstat (limited to 'src/backends')
-rw-r--r-- | src/backends/backendsCommon/WorkloadData.cpp | 18 |
1 files changed, 12 insertions, 6 deletions
diff --git a/src/backends/backendsCommon/WorkloadData.cpp b/src/backends/backendsCommon/WorkloadData.cpp index 2000ce4a57..6667eabdc5 100644 --- a/src/backends/backendsCommon/WorkloadData.cpp +++ b/src/backends/backendsCommon/WorkloadData.cpp @@ -1806,10 +1806,13 @@ void QuantizeQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const const TensorInfo& inputTensorInfo = workloadInfo.m_InputTensorInfos[0]; const TensorInfo& outputTensorInfo = workloadInfo.m_OutputTensorInfos[0]; - if (inputTensorInfo.GetDataType() != DataType::Float32) + std::vector<DataType> supportedTypes = { - throw InvalidArgumentException(descriptorName + ": Quantize only accepts Float32 inputs."); - } + DataType::Float32, + DataType::Float16 + }; + + ValidateDataTypes(inputTensorInfo, supportedTypes, descriptorName); if (outputTensorInfo.GetDataType() != DataType::QuantisedAsymm8 && outputTensorInfo.GetDataType() != DataType::QuantisedSymm16) @@ -2117,10 +2120,13 @@ void DequantizeQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const throw InvalidArgumentException(descriptorName + ": Input to dequantize layer must be quantized type."); } - if (outputTensorInfo.GetDataType() != DataType::Float32) + std::vector<DataType> supportedTypes = { - throw InvalidArgumentException(descriptorName + ": Output of dequantize layer must be Float32 type."); - } + DataType::Float32, + DataType::Float16 + }; + + ValidateDataTypes(outputTensorInfo, supportedTypes, descriptorName); } void MergeQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const |